Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fix gelu to use erf based algorithm #18827

Merged
merged 1 commit into from
Jul 31, 2020
Merged

Fix gelu to use erf based algorithm #18827

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/operator/nn/mkldnn/mkldnn_act.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down