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

slaren · web-flow · commit e5b89a441af2 · 2024-03-27T15:07:50.000+01:00
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) {

Original file line number	Diff line number	Diff line change
`@@ -2938,7 +2938,7 @@ static struct ggml_tensor * ggml_new_tensor_impl(`
`2938`	`2938`	`data_size *= ne[i];`
`2939`	`2939`	`}`
`2940`	`2940`
`2941`		`- GGML_ASSERT(view_src == NULL \|\| data_size + view_offs <= ggml_nbytes(view_src));`
	`2941`	`+ GGML_ASSERT(view_src == NULL \|\| data_size == 0 \|\| data_size + view_offs <= ggml_nbytes(view_src));`
`2942`	`2942`
`2943`	`2943`	`void * data = view_src != NULL ? view_src->data : NULL;`
`2944`	`2944`	`if (data != NULL) {`