From 13c64448bdeefdbf4d264e1273912ee23daeb032 Mon Sep 17 00:00:00 2001
From: Aaron Teo <aaron.teo1@ibm.com>
Date: Thu, 24 Jul 2025 00:48:32 +0800
Subject: [PATCH] ggml-zdnn: assign tensor->extra to buffer

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

diff --git a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp
index cf1877b961..9e0aa977ec 100644
--- a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp
+++ b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp
@@ -266,6 +266,8 @@ static enum ggml_status ggml_backend_zdnn_buffer_init_tensor(ggml_backend_buffer
         return GGML_STATUS_FAILED;
     }
 
+    tensor->extra = &ctx->buffers[buffer_idx];
+
     return GGML_STATUS_SUCCESS;
 }