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
if(not outputs.empty())
// if contiguous was already inserted, skip
if(std::all_of(outputs.begin(), outputs.end(), [](auto output) {
return output->name() == "contiguous";}))
return output->name() == "contiguous";
}))
continue;
shape s = ins->get_shape();
if(s.dynamic())
......@@ -79,7 +80,6 @@ void auto_contiguous::apply(module& m) const
continue;
if(s.scalar() and not contains(ins->name(), "broadcast"))
continue;
auto c = m.insert_instruction(std::next(ins), make_op("contiguous"), ins);
m.replace_instruction(ins, c);
......
......@@ -202,7 +202,7 @@ TEST_CASE(standard_reshape)
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 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);
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