From 69d392ae9cba76fe328f1db07ce986097e0636de Mon Sep 17 00:00:00 2001 From: slaren Date: Sun, 19 May 2024 00:39:15 +0200 Subject: [PATCH] cuda : clear error after buffer allocation failure --- ggml-cuda.cu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ggml-cuda.cu b/ggml-cuda.cu index 04b6e5285..c4e572c13 100644 --- a/ggml-cuda.cu +++ b/ggml-cuda.cu @@ -499,6 +499,8 @@ GGML_CALL static ggml_backend_buffer_t ggml_backend_cuda_buffer_type_alloc_buffe void * dev_ptr; cudaError_t err = cudaMalloc(&dev_ptr, size); if (err != cudaSuccess) { + // clear the error + cudaGetLastError(); fprintf(stderr, "%s: allocating %.2f MiB on device %d: cudaMalloc failed: %s\n", __func__, size/1024.0/1024.0, buft_ctx->device, cudaGetErrorString(err)); return nullptr; }