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

2025-09-03 16:46:01 +08:00 · 2025-09-03 16:46:01 +08:00 · 51bc843f3a
parent 75f739c7c8
commit 51bc843f3a
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: {