Procházet zdrojové kódy

cuda : add missing support check for xielu (#17895)

Sigbjørn Skjæret před 1 měsícem
rodič
revize
4df6e859e9
1 změnil soubory, kde provedl 1 přidání a 0 odebrání
  1. 1 0
      ggml/src/ggml-cuda/ggml-cuda.cu

+ 1 - 0
ggml/src/ggml-cuda/ggml-cuda.cu

@@ -4313,6 +4313,7 @@ static bool ggml_backend_cuda_device_supports_op(ggml_backend_dev_t dev, const g
                 case GGML_UNARY_OP_EXPM1:
                 case GGML_UNARY_OP_SOFTPLUS:
                 case GGML_UNARY_OP_ELU:
+                case GGML_UNARY_OP_XIELU:
                 case GGML_UNARY_OP_FLOOR:
                 case GGML_UNARY_OP_CEIL:
                 case GGML_UNARY_OP_ROUND: