"vscode:/vscode.git/clone" did not exist on "b1d772d127fcff4cc01d8fa1b4ea6e07da12193d"
Commit 06349ea9 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

refine the code change

parent 91f635a2
...@@ -574,11 +574,11 @@ struct miopen_apply ...@@ -574,11 +574,11 @@ struct miopen_apply
if(type == shape::half_type) if(type == shape::half_type)
{ {
shape s32{shape::float_type, s.lens()}; shape s32{shape::float_type, s.lens()};
auto cout32 = insert_allocation(ins, s32); auto cout32 = mod->insert_instruction(ins, make_op("hip::allocate", {{"shape", to_value(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( auto convert32 = mod->insert_instruction(
ins, make_op("gpu::convert", cop32.to_value()), in, cout32); ins, make_op("gpu::convert", cop32.to_value()), in, cout32);
auto lout32 = insert_allocation(ins, s32); auto lout32 = mod->insert_instruction(ins, make_op("hip::allocate", {{"shape", to_value(s32)}}));
auto lrn32 = mod->insert_instruction( auto lrn32 = mod->insert_instruction(
ins, make_op("gpu::lrn", ins->get_operator().to_value()), convert32, lout32); 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}});
......
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