Răsfoiți Sursa

cann : fix ops broken by circular padding guard (#17825)

Sigbjørn Skjæret 1 lună în urmă
părinte
comite
dcb7d17758
1 a modificat fișierele cu 1 adăugiri și 0 ștergeri
  1. 1 0
      ggml/src/ggml-cann/ggml-cann.cpp

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

@@ -2548,6 +2548,7 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev, const ggml_ten
         case GGML_OP_ARGSORT:
         case GGML_OP_ACC:
         case GGML_OP_GROUP_NORM:
+            return true;
         case GGML_OP_PAD:
             // TODO: add circular padding support for cann, see https://github.com/ggml-org/llama.cpp/pull/16985
             return ggml_get_op_params_i32(op, 8) == 0;