Skip to content
Projects
Groups
Snippets
Help
Loading...
Sign in / Register
Toggle navigation
N
ngraph
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
submodule
ngraph
Commits
aa359461
Commit
aa359461
authored
5 years ago
by
gaurides
Committed by
Scott Cyphers
5 years ago
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Force Gelu decompose on CPU (#3887)
parent
beaf154e
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
6 deletions
+4
-6
cpu_external_function.cpp
src/ngraph/runtime/cpu/cpu_external_function.cpp
+4
-2
unit_test.manifest
src/ngraph/runtime/cpu/unit_test.manifest
+0
-4
No files found.
src/ngraph/runtime/cpu/cpu_external_function.cpp
View file @
aa359461
...
...
@@ -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
;
...
...
This diff is collapsed.
Click to expand it.
src/ngraph/runtime/cpu/unit_test.manifest
View file @
aa359461
...
...
@@ -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
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment