Selaa lähdekoodia

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

slaren 1 vuosi sitten
vanhempi
sitoutus
e5b89a441a
1 muutettua tiedostoa jossa 1 lisäystä ja 1 poistoa
  1. 1 1
      ggml.c

+ 1 - 1
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) {