Sfoglia il codice sorgente

ggml-alloc : enlarge size of parse_seq (#2776)

Since we also store barriers in this array, we need to double its size.
Shouzheng Liu 2 anni fa
parent
commit
b91ad7f461
1 ha cambiato i file con 2 aggiunte e 1 eliminazioni
  1. 2 1
      ggml-alloc.c

+ 2 - 1
ggml-alloc.c

@@ -8,6 +8,7 @@
 
 #define UNUSED(x) (void)(x)
 #define MAX(a, b) ((a) > (b) ? (a) : (b))
+#define GGML_MAX_CONCUR (2*GGML_MAX_NODES)
 
 //#define GGML_ALLOCATOR_DEBUG
 
@@ -67,7 +68,7 @@ struct ggml_allocr {
     struct hash_node hash_table[GGML_GRAPH_HASHTABLE_SIZE];
     size_t max_size;
     bool measure;
-    int parse_seq[GGML_MAX_NODES];
+    int parse_seq[GGML_MAX_CONCUR];
     int parse_seq_len;
 
 #ifdef GGML_ALLOCATOR_DEBUG