diff --git a/onnxruntime/contrib_ops/cuda/layer_norm.cc b/onnxruntime/contrib_ops/cuda/layer_norm.cc index ab4cabc3c6..f295990817 100644 --- a/onnxruntime/contrib_ops/cuda/layer_norm.cc +++ b/onnxruntime/contrib_ops/cuda/layer_norm.cc @@ -32,7 +32,7 @@ LayerNorm::LayerNorm(const OpKernelInfo& op_kernel_info) : CudaKernel(op_k ORT_ENFORCE(op_kernel_info.GetAttr("axis", &axis_).IsOK()); float tmp_epsilon; ORT_ENFORCE(op_kernel_info.GetAttr("epsilon", &tmp_epsilon).IsOK()); - epsilon_ = ClampCudnnBatchNormEpsilon(tmp_epsilon); + epsilon_ = tmp_epsilon; } template