瀏覽代碼

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

slaren 1 年之前
父節點
當前提交
e5b89a441a
共有 1 個文件被更改,包括 1 次插入1 次删除
  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) {