Commit 24005e2f authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 70320bbd
...@@ -576,22 +576,24 @@ struct miopen_apply ...@@ -576,22 +576,24 @@ struct miopen_apply
shape s32{shape::float_type, s.lens()}; shape s32{shape::float_type, s.lens()};
auto cout32 = insert_allocation(ins, s32); auto cout32 = insert_allocation(ins, s32);
auto cop32 = make_op("convert", {{"target_type", shape::float_type}}); auto cop32 = make_op("convert", {{"target_type", shape::float_type}});
auto convert32 = mod->insert_instruction(ins, make_op("gpu::convert", cop32.to_value()), in, cout32); auto convert32 = mod->insert_instruction(
ins, make_op("gpu::convert", cop32.to_value()), in, cout32);
auto lout32 = insert_allocation(ins, s32); auto lout32 = insert_allocation(ins, s32);
auto lrn32 = mod->insert_instruction(ins, make_op("gpu::lrn", ins->get_operator().to_value()), convert32, lout32); auto lrn32 = mod->insert_instruction(
ins, make_op("gpu::lrn", ins->get_operator().to_value()), convert32, lout32);
auto cop16 = make_op("convert", {{"target_type", shape::half_type}}); auto cop16 = make_op("convert", {{"target_type", shape::half_type}});
auto lout16 = mod->insert_instruction(ins, make_op("gpu::convert", cop16.to_value()), lrn32, output); auto lout16 = mod->insert_instruction(
ins, make_op("gpu::convert", cop16.to_value()), lrn32, output);
return mod->replace_instruction(ins, lout16); return mod->replace_instruction(ins, lout16);
} }
else else
{ {
auto lrn16 = mod->insert_instruction(ins, make_op("gpu::lrn", ins->get_operator().to_value()), in, output); auto lrn16 = mod->insert_instruction(
ins, make_op("gpu::lrn", ins->get_operator().to_value()), in, output);
return mod->replace_instruction(ins, lrn16); return mod->replace_instruction(ins, lrn16);
} }
}); });
} }
}; };
void lowering::apply(module& m) const { miopen_apply{&m, this}.apply(); } void lowering::apply(module& m) const { miopen_apply{&m, this}.apply(); }
......
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