diff --git a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/External/DirectMLHelpers/ApiHelpers.h b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/External/DirectMLHelpers/ApiHelpers.h index 8c85e4ec1d..76ca37bd05 100644 --- a/onnxruntime/core/providers/dml/DmlExecutionProvider/src/External/DirectMLHelpers/ApiHelpers.h +++ b/onnxruntime/core/providers/dml/DmlExecutionProvider/src/External/DirectMLHelpers/ApiHelpers.h @@ -28,6 +28,7 @@ union ActivationOperatorDescUnion DML_ACTIVATION_TANH_OPERATOR_DESC tanh; DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC thresholdedRelu; DML_ACTIVATION_SHRINK_OPERATOR_DESC shrink; + DML_ACTIVATION_GELU_OPERATOR_DESC gelu; }; struct ActivationOperatorDesc @@ -64,6 +65,7 @@ struct ActivationOperatorDesc case DML_OPERATOR_ACTIVATION_TANH: return { activationType, ¶ms.tanh }; case DML_OPERATOR_ACTIVATION_THRESHOLDED_RELU: return { activationType, ¶ms.thresholdedRelu }; case DML_OPERATOR_ACTIVATION_SHRINK: return { activationType, ¶ms.shrink }; + case DML_OPERATOR_ACTIVATION_GELU: return { activationType, ¶ms.gelu }; default: ORT_THROW_HR(E_INVALIDARG); return { activationType, ¶ms.relu };