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

ad7e1ea1 · Pruthvi · Scott Cyphers · 58aa4746 · ad7e1ea1
Commit ad7e1ea1 authored Aug 08, 2018 by Pruthvi Committed by Scott Cyphers Aug 08, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

cpu_external_function.cpp src/ngraph/runtime/cpu/cpu_external_function.cpp +5 -3

No files found.
--- a/src/ngraph/runtime/cpu/cpu_external_function.cpp
+++ b/src/ngraph/runtime/cpu/cpu_external_function.cpp
@@ -1068,9 +1068,11 @@ void runtime::cpu::CPU_ExternalFunction::build()
    //in which case they should run this pass(CPUWorkspaceInsertion) explicitly
    NodeVector nv_cwi;
    pass_manager.register_pass<ngraph::pass::NopElimination>();
-    pass_manager.register_pass<runtime::cpu::pass::LSTMFusion>();
-    pass_manager.register_pass<runtime::cpu::pass::RNNFusion>();
-    pass_manager.register_pass<runtime::cpu::pass::ConcatInputs>();
+    // TODO (pruthvi): Enable all the disabeled RNN fusion graph pass after fixing
+    // failing mxnet unit tests.
+    //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::CommonSubexpressionElimination>();
    pass_manager.register_pass<ngraph::pass::CoreFusion>();