Skip to content

Commit f36536b

Browse files
committed
CANN: The Ger operator of OUT_PROD is not supported on the 310p device
1 parent b78db3b commit f36536b

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
@@ -2561,6 +2561,10 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
25612561
return true;
25622562
case GGML_OP_OUT_PROD:
25632563
{
2564+
#ifdef ASCEND_310P
2565+
// Ger is not supported on 310p device
2566+
return false;
2567+
#endif
25642568
switch (op->src[0]->type) {
25652569
case GGML_TYPE_F16:
25662570
case GGML_TYPE_F32:

0 commit comments

Comments
 (0)