CANN: Disable Ger operator of OUT_PROD on 310p device (#17563)

2026-05-01 22:54:05 +00:00 · 2025-12-02 20:35:23 +08:00
parent 7f3a72a8ed
commit a2b0fe8d37
1 changed files with 4 additions and 0 deletions
--- a/ggml/src/ggml-cann/ggml-cann.cpp
+++ b/ggml/src/ggml-cann/ggml-cann.cpp
@@ -2564,6 +2564,10 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
            return true;
        case GGML_OP_OUT_PROD:
            {
+#ifdef ASCEND_310P
+                // Ger is not supported on 310p device
+                return false;
+#endif
                switch (op->src[0]->type) {
                    case GGML_TYPE_F16:
                    case GGML_TYPE_F32: