add the rewrite_gru pass.

e3df7b35 · Shucai Xiao · 76a4fbbc · e3df7b35 · e3df7b35
Commit e3df7b35 authored Feb 04, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/targets/cpu/target.cpp src/targets/cpu/target.cpp +2 -0

src/targets/gpu/target.cpp src/targets/gpu/target.cpp +1 -0

No files found.
--- a/src/targets/cpu/target.cpp
+++ b/src/targets/cpu/target.cpp
@@ -17,6 +17,8 @@ std::vector<pass> target::get_passes(migraphx::context&) const
    return {auto_contiguous{},
            rewrite_rnn{},
            dead_code_elimination{},
+            rewrite_gru{},
+            dead_code_elimination{},
            lowering{},
            dead_code_elimination{}};
 }

--- a/src/targets/gpu/target.cpp
+++ b/src/targets/gpu/target.cpp
@@ -34,6 +34,7 @@ std::vector<pass> target::get_passes(migraphx::context& gctx) const
        fwd_conv_batchnorm_rewrite{},
        dead_code_elimination{},
        rewrite_rnn{},
+        dead_code_elimination{},
        rewrite_gru{},
        dead_code_elimination{},
        //common_subexpression_elimination{},