Răsfoiți Sursa

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

slaren 2 ani în urmă
părinte
comite
95bd60a0a6
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  1. 1 1
      ggml-alloc.c

+ 1 - 1
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);
 }