From bdf40ec6cf2fa33234f86cd3bc43338d7b14492f Mon Sep 17 00:00:00 2001 From: Tao Lv Date: Thu, 30 Jul 2020 21:23:55 +0800 Subject: [PATCH] fix gelu to use erf based algorithm --- src/operator/nn/mkldnn/mkldnn_act.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/operator/nn/mkldnn/mkldnn_act.cc b/src/operator/nn/mkldnn/mkldnn_act.cc index 4654c24547cd..e76a0629cc01 100644 --- a/src/operator/nn/mkldnn/mkldnn_act.cc +++ b/src/operator/nn/mkldnn/mkldnn_act.cc @@ -100,7 +100,7 @@ mkldnn::algorithm GetMKLDNNActAlgo(const LeakyReLUParam& param) { case leakyrelu::kELU: return mkldnn::algorithm::eltwise_elu; case leakyrelu::kGELU: - return mkldnn::algorithm::eltwise_gelu; + return mkldnn::algorithm::eltwise_gelu_erf; default: LOG(FATAL) << "unknown activation type for LeakyReLU: " << param.act_type; return mkldnn::algorithm::eltwise_relu;