Просмотр исходного кода

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

Sigbjørn Skjæret 1 месяц назад
Родитель
Сommit
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_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: