Pārlūkot izejas kodu

CANN: Disable Ger operator of OUT_PROD on 310p device (#17563)

TianHao324 1 mēnesi atpakaļ
vecāks
revīzija
a2b0fe8d37
1 mainītis faili ar 4 papildinājumiem un 0 dzēšanām
  1. 4 0
      ggml/src/ggml-cann/ggml-cann.cpp

+ 4 - 0
ggml/src/ggml-cann/ggml-cann.cpp

@@ -2564,6 +2564,10 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
             return true;
         case GGML_OP_OUT_PROD:
             {
+#ifdef ASCEND_310P
+                // Ger is not supported on 310p device
+                return false;
+#endif
                 switch (op->src[0]->type) {
                     case GGML_TYPE_F16:
                     case GGML_TYPE_F32: