Fix access violation in ggml_cuda_free_data if tensor->extra is NULL (#4554)

diff --git a/ggml-cuda.cu b/ggml-cuda.cu
index 9f4b188cbb0d6f24b8c783fe2b134889f336a960..28d3787843800d9cbe2f0f090d44cd5c3a517a46 100644 (file)
--- a/ggml-cuda.cu
+++ b/ggml-cuda.cu
@@ -9091,7 +9091,7 @@ void ggml_cuda_transform_tensor(void * data, struct ggml_tensor * tensor) {
 }
 
 void ggml_cuda_free_data(struct ggml_tensor * tensor) {
-    if (!tensor || (tensor->backend != GGML_BACKEND_GPU && tensor->backend != GGML_BACKEND_GPU_SPLIT) ) {
+    if (!tensor || !tensor->extra || (tensor->backend != GGML_BACKEND_GPU && tensor->backend != GGML_BACKEND_GPU_SPLIT) ) {
         return;
     }