ggml-alloc : fix assert in debug builds (ggerganov#3555)

yxq321 · Oct 9, 2023 · 95bd60a · 95bd60a
1 parent fcca0a7
commit 95bd60a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ggml-alloc.c b/ggml-alloc.c
@@ -386,7 +386,7 @@ static void init_view(struct ggml_allocr * alloc, struct ggml_tensor * view) {
 
     // FIXME: the view should be initialized by the owning buffer, but currently this breaks the CUDA backend
     // due to the ggml_tensor_extra_gpu ring buffer overwriting the KV cache extras
-    assert(ggml_allocr_is_measure(alloc) || view->buffer->backend == alloc->buffer->backend);
+    assert(ggml_allocr_is_measure(alloc) || !view->buffer || view->buffer->backend == alloc->buffer->backend);
     ggml_backend_buffer_init_tensor(alloc->buffer, view);
 }