From b7a77cf683a0c5dc7dc236e0adb3500f48482d11 Mon Sep 17 00:00:00 2001 From: Aaron Teo Date: Wed, 30 Jul 2025 16:15:20 +0800 Subject: [PATCH] ggml-zdnn: add guards to prevent loading ztensor if transformed Signed-off-by: Aaron Teo --- ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp index 4fbe6924fa..b011683eee 100644 --- a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp +++ b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp @@ -133,8 +133,13 @@ static void ggml_zdnn_mul_mat_op(ggml_backend_zdnn_context * ctx, const ggml_ten // ggml_zdnn_create_tensor(ptd_output, td_output, zt_output, output, output_dim, ZDNN_2D); void * bias_data = (void *)calloc(ne0, ggml_element_size(output)); - ggml_zdnn_load_tensor(weights_extra->ztensor, weights->data); - ggml_zdnn_load_tensor(inputs_extra->ztensor, inputs->data); + if (weights_extra->ztensor.is_transformed == false) { + ggml_zdnn_load_tensor(weights_extra->ztensor, weights->data); + } + + if (inputs_extra->ztensor.is_transformed == false) { + ggml_zdnn_load_tensor(inputs_extra->ztensor, inputs->data); + } ggml_zdnn_load_tensor(zt_bias, bias_data); // ggml_zdnn_load_tensor(output_extra->ztensor, output->data);