From f4ec752d0ebdf4e55b496ce48ffb621dbeb25fc0 Mon Sep 17 00:00:00 2001 From: Aaron Teo Date: Sat, 6 Sep 2025 19:11:00 +0800 Subject: [PATCH] ggml-zdnn: clean up set_tensor Signed-off-by: Aaron Teo --- ggml/src/ggml-zdnn/ggml-zdnn.cpp | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/ggml/src/ggml-zdnn/ggml-zdnn.cpp b/ggml/src/ggml-zdnn/ggml-zdnn.cpp index 335eab9298..24b514bccf 100644 --- a/ggml/src/ggml-zdnn/ggml-zdnn.cpp +++ b/ggml/src/ggml-zdnn/ggml-zdnn.cpp @@ -425,12 +425,8 @@ static void ggml_backend_zdnn_buffer_set_tensor(ggml_backend_buffer_t buffer, gg memcpy((char *)tensor->data + offset, data, size); ggml_backend_zdnn_buffer * extra = (ggml_backend_zdnn_buffer *)tensor->extra; - size_t total_size = ggml_nbytes(tensor); - // WARNING: this check might not be thread-safe. need to verify. - if (offset + size == total_size) { - if (extra->ztensor.is_transformed) zdnn_reset_ztensor(&extra->ztensor); - ggml_zdnn_load_tensor(extra->ztensor, tensor->data); - } + if (extra->ztensor.is_transformed) zdnn_reset_ztensor(&extra->ztensor); + ggml_zdnn_load_tensor(extra->ztensor, tensor->data); GGML_UNUSED(buffer); }