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

formatting

parent 90967138
...@@ -68,7 +68,8 @@ void auto_contiguous::apply(module& m) const ...@@ -68,7 +68,8 @@ void auto_contiguous::apply(module& m) const
if(not outputs.empty()) if(not outputs.empty())
// if contiguous was already inserted, skip // if contiguous was already inserted, skip
if(std::all_of(outputs.begin(), outputs.end(), [](auto output) { if(std::all_of(outputs.begin(), outputs.end(), [](auto output) {
return output->name() == "contiguous";})) return output->name() == "contiguous";
}))
continue; continue;
shape s = ins->get_shape(); shape s = ins->get_shape();
if(s.dynamic()) if(s.dynamic())
...@@ -79,7 +80,6 @@ void auto_contiguous::apply(module& m) const ...@@ -79,7 +80,6 @@ void auto_contiguous::apply(module& m) const
continue; continue;
if(s.scalar() and not contains(ins->name(), "broadcast")) if(s.scalar() and not contains(ins->name(), "broadcast"))
continue; continue;
auto c = m.insert_instruction(std::next(ins), make_op("contiguous"), ins); auto c = m.insert_instruction(std::next(ins), make_op("contiguous"), ins);
m.replace_instruction(ins, c); m.replace_instruction(ins, c);
......
...@@ -202,7 +202,7 @@ TEST_CASE(standard_reshape) ...@@ -202,7 +202,7 @@ TEST_CASE(standard_reshape)
auto data = m2.add_parameter("2x2", {migraphx::shape::float_type, {2, 3, 4, 5}}); auto data = m2.add_parameter("2x2", {migraphx::shape::float_type, {2, 3, 4, 5}});
auto add = m2.add_instruction(migraphx::make_op("add"), data, data); auto add = m2.add_instruction(migraphx::make_op("add"), data, data);
auto ca = m2.add_instruction(migraphx::make_op("contiguous"), add); auto ca = m2.add_instruction(migraphx::make_op("contiguous"), add);
auto r = m2.add_instruction(migraphx::make_op("reshape", {{"dims", {2, 1, 12, 5}}}), ca); auto r = m2.add_instruction(migraphx::make_op("reshape", {{"dims", {2, 1, 12, 5}}}), ca);
auto cr = m2.add_instruction(migraphx::make_op("contiguous"), r); auto cr = m2.add_instruction(migraphx::make_op("contiguous"), r);
m2.add_return({cr}); m2.add_return({cr});
} }
......
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