From e3ec30efb65d596a6aab5dea4b8edbdb8c1a1370 Mon Sep 17 00:00:00 2001 From: Dwayne Robinson Date: Tue, 14 Jun 2022 23:28:15 -0700 Subject: [PATCH] Add missing GELU to ApiHelpers.h --- .../src/External/DirectMLHelpers/ApiHelpers.h | 2 ++ 1 file changed, 2 insertions(+) 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 };