Skip to content

Commit cbb5937

Browse files
committed
CANN: Add RoPE contiguous check for 310I DUP device
1 parent 25f1045 commit cbb5937

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ggml/src/ggml-cann/ggml-cann.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2417,7 +2417,11 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev,
24172417
if (mode & GGML_ROPE_TYPE_VISION) {
24182418
return false;
24192419
}
2420-
2420+
#ifdef ASCEND_310P
2421+
if(!ggml_is_contiguous(op->src[0])){
2422+
return false;
2423+
}
2424+
#endif
24212425
return true;
24222426
}
24232427
case GGML_OP_UPSCALE: {

0 commit comments

Comments
 (0)