Commit 1b4ebf8e authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fixed a bug and added one more tests.

parent 55f8e435
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
#include <migraphx/cpu/lowering.hpp> #include <migraphx/cpu/lowering.hpp>
#include <migraphx/auto_contiguous.hpp> #include <migraphx/auto_contiguous.hpp>
#include <migraphx/rewrite_rnn.hpp> #include <migraphx/rewrite_rnn.hpp>
#include <migraphx/dead_code_elimination.hpp>
namespace migraphx { namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS { inline namespace MIGRAPHX_INLINE_NS {
...@@ -12,7 +13,13 @@ std::string target::name() const { return "cpu"; } ...@@ -12,7 +13,13 @@ std::string target::name() const { return "cpu"; }
std::vector<pass> target::get_passes(migraphx::context&) const std::vector<pass> target::get_passes(migraphx::context&) const
{ {
return {auto_contiguous{}, rewrite_rnn{}, lowering{}}; return {
auto_contiguous{},
rewrite_rnn{},
dead_code_elimination{},
lowering{},
dead_code_elimination{}
};
} }
} // namespace cpu } // namespace cpu
......
...@@ -34,8 +34,8 @@ std::vector<pass> target::get_passes(migraphx::context& gctx) const ...@@ -34,8 +34,8 @@ std::vector<pass> target::get_passes(migraphx::context& gctx) const
dead_code_elimination{}, dead_code_elimination{},
rewrite_rnn{}, rewrite_rnn{},
dead_code_elimination{}, dead_code_elimination{},
common_subexpression_elimination{}, //common_subexpression_elimination{},
dead_code_elimination{}, //dead_code_elimination{},
simplify_algebra{}, simplify_algebra{},
dead_code_elimination{}, dead_code_elimination{},
constant_propagate{}, constant_propagate{},
......
...@@ -1542,5 +1542,5 @@ int main() ...@@ -1542,5 +1542,5 @@ int main()
verify_program<test_rnn_5args>(); verify_program<test_rnn_5args>();
verify_program<test_rnn_bidirectional>(); verify_program<test_rnn_bidirectional>();
verify_program<test_rnn_bidirectional10>(); verify_program<test_rnn_bidirectional10>();
// verify_program<test_rnn_bi_3args>(); verify_program<test_rnn_bi_3args>();
} }
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