]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
ggml : fix bounds checking of zero size views (llama/6347)
authorslaren <redacted>
Wed, 27 Mar 2024 14:07:50 +0000 (15:07 +0100)
committerGeorgi Gerganov <redacted>
Sat, 6 Apr 2024 15:27:03 +0000 (18:27 +0300)
src/ggml.c

index eb469d0f7953d514bdf14a99009fdda0c2b00453..81100e27766b7d574c6b8b37167c6c1d418a1e19 100644 (file)
@@ -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) {