diff --git a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp index ae1260a8c8..cf1877b961 100644 --- a/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp +++ b/ggml/src/ggml-zdnn/ggml-zdnn-rewrite.cpp @@ -10,6 +10,10 @@ static bool ggml_backend_zdnn_compute_forward(struct ggml_backend_zdnn_context * ctx, struct ggml_tensor * dst) { switch (dst->op) { + case GGML_OP_MUL_MAT: + { + std::raise(SIGINT); + } break; default: return false; } @@ -154,6 +158,9 @@ static bool ggml_zdnn_supports_op(const struct ggml_backend_zdnn_device_context case GGML_OP_CONCAT: return true; + case GGML_OP_MUL_MAT: + return true; + default: return false; }