Skip to content

Commit a2b0fe8

Browse files
authored
CANN: Disable Ger operator of OUT_PROD on 310p device (#17563)
1 parent 7f3a72a commit a2b0fe8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2564,6 +2564,10 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
25642564
return true;
25652565
case GGML_OP_OUT_PROD:
25662566
{
2567+
#ifdef ASCEND_310P
2568+
// Ger is not supported on 310p device
2569+
return false;
2570+
#endif
25672571
switch (op->src[0]->type) {
25682572
case GGML_TYPE_F16:
25692573
case GGML_TYPE_F32:

0 commit comments

Comments
 (0)