]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
Fix assert when free invalid cuda pointer (#2005)
authorHoward Su <redacted>
Mon, 26 Jun 2023 15:15:47 +0000 (23:15 +0800)
committerGitHub <redacted>
Mon, 26 Jun 2023 15:15:47 +0000 (23:15 +0800)
Fix assert via initializing extra structure always.
CUDA error 1 at C:\GPT\llama.cpp\ggml-cuda.cu:2536: invalid argument

ggml-cuda.cu

index 010682edb703cbbf1282b0ab76e5298132662f49..5e2fbc72442d5487c88d9b2df81217fb47e96c10 100644 (file)
@@ -2553,6 +2553,7 @@ void ggml_cuda_assign_buffers_impl(struct ggml_tensor * tensor, bool scratch) {
 
     tensor->backend = GGML_BACKEND_GPU;
     struct ggml_tensor_extra_gpu * extra = new ggml_tensor_extra_gpu;
+    memset(extra, 0, sizeof(*extra));
 
     const bool inplace = (tensor->src0 != nullptr && tensor->src0->data == tensor->data) ||
         tensor->op == GGML_OP_VIEW;