fix gelu to use erf based algorithm (#18827) (#18946)

Co-authored-by: Tao Lv <[email protected]>
apache · Aug 18, 2020 · 6ae469a · 6ae469a
1 parent 5e0db7a
commit 6ae469a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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;