Commit ef3a45c2 authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent 8639a349
...@@ -104,7 +104,7 @@ void transform_convolutions(module& m, bool skip_elim_contiguous) ...@@ -104,7 +104,7 @@ void transform_convolutions(module& m, bool skip_elim_contiguous)
// auto c = conv; // auto c = conv;
// auto nchw = m.insert_instruction(ins, make_op("layout", {{"permutation", {0, 1, 2, 3}}}), // auto nchw = m.insert_instruction(ins, make_op("layout", {{"permutation", {0, 1, 2, 3}}}),
// conv); m.debug_print(); if(not skip_elim_contiguous) // conv); m.debug_print(); if(not skip_elim_contiguous)
auto c = m.insert_instruction(ins, make_op("contiguous"), conv); auto c = m.insert_instruction(ins, make_op("contiguous"), conv);
m.replace_instruction(ins, c); m.replace_instruction(ins, c);
} }
} }
...@@ -144,12 +144,12 @@ void layout_nhwc::apply(module_pass_manager& mpm) const ...@@ -144,12 +144,12 @@ void layout_nhwc::apply(module_pass_manager& mpm) const
mpm.run_pass(dead_code_elimination{}); mpm.run_pass(dead_code_elimination{});
// mpm.get_module().debug_print(); // mpm.get_module().debug_print();
transform_convolutions(mpm.get_module(), this->skip_elim_contiguous); transform_convolutions(mpm.get_module(), this->skip_elim_contiguous);
mpm.run_pass(dead_code_elimination{}); mpm.run_pass(dead_code_elimination{});
// std::cout << "after layout" << std::endl; // std::cout << "after layout" << std::endl;
// mpm.get_module().debug_print(); // mpm.get_module().debug_print();
// if(not this->skip_elim_contiguous) // if(not this->skip_elim_contiguous)
mpm.run_pass(eliminate_contiguous{"contiguous"}); mpm.run_pass(eliminate_contiguous{"contiguous"});
mpm.run_pass(dead_code_elimination{}); mpm.run_pass(dead_code_elimination{});
mpm.run_pass(auto_contiguous{}); mpm.run_pass(auto_contiguous{});
mpm.run_pass(dead_code_elimination{}); mpm.run_pass(dead_code_elimination{});
......
...@@ -39,7 +39,7 @@ struct test_conv_pooling : verify_program<test_conv_pooling> ...@@ -39,7 +39,7 @@ struct test_conv_pooling : verify_program<test_conv_pooling>
auto weights = auto weights =
mm->add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}}); mm->add_parameter("w", migraphx::shape{migraphx::shape::float_type, {4, 3, 3, 3}});
auto conv = mm->add_instruction(migraphx::make_op("convolution"), input, weights); auto conv = mm->add_instruction(migraphx::make_op("convolution"), input, weights);
auto relu = mm->add_instruction(migraphx::make_op("relu"), conv); auto relu = mm->add_instruction(migraphx::make_op("relu"), conv);
auto pooling = mm->add_instruction( auto pooling = mm->add_instruction(
migraphx::make_op("pooling", {{"mode", migraphx::op::pooling_mode::max}}), relu); migraphx::make_op("pooling", {{"mode", migraphx::op::pooling_mode::max}}), relu);
mm->add_instruction(migraphx::make_op("relu"), pooling); mm->add_instruction(migraphx::make_op("relu"), pooling);
......
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