CANN: Add RoPE contiguous check for 310I DUP device (#15735)

2025-10-27 08:21:30 +00:00 · 2025-09-03 16:46:01 +08:00
parent 05c0380f2a
commit 5eae934883
1 changed files with 5 additions and 1 deletions
--- a/ggml/src/ggml-cann/ggml-cann.cpp
+++ b/ggml/src/ggml-cann/ggml-cann.cpp
@@ -2417,7 +2417,11 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev,
            if (mode & GGML_ROPE_TYPE_VISION) {
                return false;
            }
-
+#ifdef ASCEND_310P
+            if(!ggml_is_contiguous(op->src[0])){
+                return false;
+            }
+#endif
            return true;
        }
        case GGML_OP_UPSCALE: {