Преглед изворни кода

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

Sigbjørn Skjæret пре 1 месец
родитељ
комит
4df6e859e9
1 измењених фајлова са 1 додато и 0 уклоњено
  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_EXPM1:
                 case GGML_UNARY_OP_SOFTPLUS:
                 case GGML_UNARY_OP_SOFTPLUS:
                 case GGML_UNARY_OP_ELU:
                 case GGML_UNARY_OP_ELU:
+                case GGML_UNARY_OP_XIELU:
                 case GGML_UNARY_OP_FLOOR:
                 case GGML_UNARY_OP_FLOOR:
                 case GGML_UNARY_OP_CEIL:
                 case GGML_UNARY_OP_CEIL:
                 case GGML_UNARY_OP_ROUND:
                 case GGML_UNARY_OP_ROUND: