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

This commit is contained in:
hipudding 2025-09-03 16:46:01 +08:00 committed by GitHub
parent 05c0380f2a
commit 5eae934883
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions

View File

@ -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: {