From 732df731ba5974babe9fae38388607b7b11f5ab9 Mon Sep 17 00:00:00 2001 From: Aaron Teo Date: Thu, 31 Jul 2025 12:10:14 +0800 Subject: [PATCH] ggml-zdnn: disable batched matmul Signed-off-by: Aaron Teo --- ggml/src/ggml-zdnn/ggml-zdnn.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ggml/src/ggml-zdnn/ggml-zdnn.cpp b/ggml/src/ggml-zdnn/ggml-zdnn.cpp index 767b33df0f..ab88a149e3 100644 --- a/ggml/src/ggml-zdnn/ggml-zdnn.cpp +++ b/ggml/src/ggml-zdnn/ggml-zdnn.cpp @@ -275,7 +275,9 @@ static bool ggml_zdnn_supports_op(const ggml_backend_zdnn_device_context * ctx_d const int64_t max_batch = ctx_dev->max_size; - return ggml_is_contiguous(src0) && + return ggml_is_matrix(src0) && + ggml_is_matrix(src1) && + ggml_is_contiguous(src0) && ggml_is_contiguous(src1) && src0->type == GGML_TYPE_F32 && src1->type == GGML_TYPE_F32 && (ne0 <= max_batch && ne1 <= max_batch && ne10 <= max_batch);