ggml : fix bounds checking of zero size views (ggerganov#6347)

teleprint-me · Mar 27, 2024 · e5b89a4 · e5b89a4
1 parent 3a03459
commit e5b89a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ggml.c b/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) {