slaren commited on
Commit
80db462
·
unverified ·
1 Parent(s): e7ddd12

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

Browse files
Files changed (1) hide show
  1. ggml.c +1 -1
ggml.c CHANGED
@@ -2938,7 +2938,7 @@ static struct ggml_tensor * ggml_new_tensor_impl(
2938
  data_size *= ne[i];
2939
  }
2940
 
2941
- GGML_ASSERT(view_src == NULL || data_size + view_offs <= ggml_nbytes(view_src));
2942
 
2943
  void * data = view_src != NULL ? view_src->data : NULL;
2944
  if (data != NULL) {
 
2938
  data_size *= ne[i];
2939
  }
2940
 
2941
+ GGML_ASSERT(view_src == NULL || data_size == 0 || data_size + view_offs <= ggml_nbytes(view_src));
2942
 
2943
  void * data = view_src != NULL ? view_src->data : NULL;
2944
  if (data != NULL) {