From 2872276d8a1495d3397b144af3f0b03c82b9eb39 Mon Sep 17 00:00:00 2001
From: Aaron Teo <aaron.teo1@ibm.com>
Date: Tue, 29 Jul 2025 00:09:00 +0800
Subject: [PATCH] ggml-zdnn: fix invalid ztensor buffer release

Signed-off-by: Aaron Teo <aaron.teo1@ibm.com>
---
 ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp
index ff099fffed..f4d121ec4b 100644
--- a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp
+++ b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp
@@ -321,6 +321,9 @@ static void ggml_backend_zdnn_buffer_free_buffer(ggml_backend_buffer_t buffer) {
     ggml_backend_zdnn_buffer_context * ctx = (ggml_backend_zdnn_buffer_context *)buffer->context;
 
     for (int i = 0; i < ctx->n_buffers; i++) {
+        if (&ctx->buffers[i].ztensor.buffer == NULL) {
+            continue;
+        }
         ZDNN_CHECK(zdnn_free_ztensor_buffer(&ctx->buffers[i].ztensor));
     }