ggml : fix bounds checking of zero size views (llama/6347)

surper518 · Apr 6, 2024 · 40d3da9 · 40d3da9
1 parent 092c047
commit 40d3da9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ggml.c b/src/ggml.c
@@ -2938,7 +2938,7 @@ static struct ggml_tensor * ggml_new_tensor_impl(
         data_size *= ne[i];
     }
 
-    GGML_ASSERT(view_src == NULL || data_size + view_offs <= ggml_nbytes(view_src));
+    GGML_ASSERT(view_src == NULL || data_size == 0 || data_size + view_offs <= ggml_nbytes(view_src));
 
     void * data = view_src != NULL ? view_src->data : NULL;
     if (data != NULL) {