From 6ae469a17ebe517325cdf6acdf0e2a8b4d464734 Mon Sep 17 00:00:00 2001
From: ciyong <ciyong.chen@intel.com>
Date: Tue, 18 Aug 2020 15:10:57 +0800
Subject: [PATCH] fix gelu to use erf based algorithm (#18827) (#18946)

Co-authored-by: Tao Lv <tao.a.lv@intel.com>
---
 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 de5c0149e612..29ff8d92b2da 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;