Skip to content

N
ngraph
Commit ad7e1ea1 authored by Pruthvi's avatar Pruthvi Committed by Scott Cyphers
Browse files
Options

disable rnn && lstm fusion pass for DEX (#1371)

parent 58aa4746
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
src/ngraph/runtime/cpu/cpu_external_function.cpp
View file @ ad7e1ea1
...@@ -1068,9 +1068,11 @@ void runtime::cpu::CPU_ExternalFunction::build() ...@@ -1068,9 +1068,11 @@ void runtime::cpu::CPU_ExternalFunction::build()
//in which case they should run this pass(CPUWorkspaceInsertion) explicitly //in which case they should run this pass(CPUWorkspaceInsertion) explicitly
NodeVector nv_cwi; NodeVector nv_cwi;
pass_manager.register_pass<ngraph::pass::NopElimination>(); pass_manager.register_pass<ngraph::pass::NopElimination>();
pass_manager.register_pass<runtime::cpu::pass::LSTMFusion>(); // TODO (pruthvi): Enable all the disabeled RNN fusion graph pass after fixing
pass_manager.register_pass<runtime::cpu::pass::RNNFusion>(); // failing mxnet unit tests.
pass_manager.register_pass<runtime::cpu::pass::ConcatInputs>(); //pass_manager.register_pass<runtime::cpu::pass::LSTMFusion>();
//pass_manager.register_pass<runtime::cpu::pass::RNNFusion>();
//pass_manager.register_pass<runtime::cpu::pass::ConcatInputs>();
pass_manager.register_pass<ngraph::pass::AlgebraicSimplification>(); pass_manager.register_pass<ngraph::pass::AlgebraicSimplification>();
pass_manager.register_pass<ngraph::pass::CommonSubexpressionElimination>(); pass_manager.register_pass<ngraph::pass::CommonSubexpressionElimination>();
pass_manager.register_pass<ngraph::pass::CoreFusion>(); pass_manager.register_pass<ngraph::pass::CoreFusion>();
......
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