Spaces:
Sleeping
Sleeping
slaren
commited on
ggml : fix bounds checking of zero size views (llama/6347)
Browse files
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) {
|