"docker/Dockerfile.cu125" did not exist on "39fc5a6d94ea6cc714ec08c828d8be9e1edf2610"
Commit b07f5e4e authored by turneram's avatar turneram
Browse files

Remove extra mul op

parent 14329696
...@@ -51,8 +51,7 @@ struct find_gelu_erf ...@@ -51,8 +51,7 @@ struct find_gelu_erf
one = m.insert_instruction( one = m.insert_instruction(
ins, make_op("multibroadcast", {{"out_lens", x->get_shape().lens()}}), one); ins, make_op("multibroadcast", {{"out_lens", x->get_shape().lens()}}), one);
sig = m.insert_instruction(ins, make_op("add"), sig, one); sig = m.insert_instruction(ins, make_op("add"), sig, one);
sig = m.insert_instruction(ins, make_op("div"), one, sig); sig = m.insert_instruction(ins, make_op("div"), x, sig);
sig = m.insert_instruction(ins, make_op("mul"), x, sig);
m.replace_instruction(ins, sig); m.replace_instruction(ins, sig);
} }
}; };
......
...@@ -77,8 +77,7 @@ TEST_CASE(bias_gelu) ...@@ -77,8 +77,7 @@ TEST_CASE(bias_gelu)
auto l2 = m2.add_literal(migraphx::literal{s2, {1.0f}}); auto l2 = m2.add_literal(migraphx::literal{s2, {1.0f}});
l2 = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", s1.lens()}}), l2); l2 = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", s1.lens()}}), l2);
sig = m2.add_instruction(migraphx::make_op("add"), sig, l2); sig = m2.add_instruction(migraphx::make_op("add"), sig, l2);
sig = m2.add_instruction(migraphx::make_op("div"), l2, sig); sig = m2.add_instruction(migraphx::make_op("div"), add, sig);
sig = m2.add_instruction(migraphx::make_op("mul"), add, sig);
m2.add_return({sig}); m2.add_return({sig});
} }
...@@ -125,8 +124,7 @@ TEST_CASE(non_bias_gelu) ...@@ -125,8 +124,7 @@ TEST_CASE(non_bias_gelu)
auto l2 = m2.add_literal(migraphx::literal{s2, {1.0f}}); auto l2 = m2.add_literal(migraphx::literal{s2, {1.0f}});
l2 = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", s1.lens()}}), l2); l2 = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", s1.lens()}}), l2);
sig = m2.add_instruction(migraphx::make_op("add"), sig, l2); sig = m2.add_instruction(migraphx::make_op("add"), sig, l2);
sig = m2.add_instruction(migraphx::make_op("div"), l2, sig); sig = m2.add_instruction(migraphx::make_op("div"), sub, sig);
sig = m2.add_instruction(migraphx::make_op("mul"), sub, sig);
m2.add_return({sig}); m2.add_return({sig});
} }
......
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