Commit f854fd62 authored by gaurides's avatar gaurides Committed by Scott Cyphers

Force Gelu decompose on CPU (#3902)

parent bcd1f46f
......@@ -1210,7 +1210,8 @@ void runtime::cpu::CPU_ExternalFunction::register_common_passes(
else if (typeid(ngraph::op::GeluBackpropFactor) == typeid(node))
{
#if MKLDNN_VERSION_MAJOR < 1
return ((node.input(0).get_element_type() == element::f32) ? true : false);
// TODO: (gauri): need to differentiate which implementation : erf vs tanh
return false;
#else
// TODO: will be supported in mkldnn v1.1
return false;
......@@ -1219,7 +1220,8 @@ void runtime::cpu::CPU_ExternalFunction::register_common_passes(
else if (typeid(ngraph::op::Gelu) == typeid(node))
{
#if MKLDNN_VERSION_MAJOR < 1
return ((node.input(0).get_element_type() == element::f32) ? true : false);
// TODO: (gauri): need to differentiate which implementation : erf vs tanh
return false;
#else
// TODO: will be supported in mkldnn v1.1
return false;
......
......@@ -22,9 +22,5 @@ lrn_across_nw
lrn_across_empty
lrn_6D_across_2_axes
# Gelu tests not supported in CPU backend, we use mkldnn gelubackprop (and not factor)
gelu_backprop_factor_f32
backwards_gelu_f32
# ONNX TopK with dynamic K
top_k_opset_10
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment