Commit a48c41a9 authored by turneram's avatar turneram
Browse files

Formatting

parent fb6de954
...@@ -38,7 +38,7 @@ struct find_gelu_erf ...@@ -38,7 +38,7 @@ struct find_gelu_erf
{ {
auto ins = r.result; auto ins = r.result;
auto x = r.instructions["x"]; auto x = r.instructions["x"];
if (x->get_shape().type() != migraphx::shape::half_type) if(x->get_shape().type() != migraphx::shape::half_type)
return; return;
auto lit = m.add_literal(literal{shape{x->get_shape().type()}, {1.702f}}); auto lit = m.add_literal(literal{shape{x->get_shape().type()}, {1.702f}});
......
...@@ -34,12 +34,12 @@ struct test_add_gelu_half : verify_program<test_add_gelu_half> ...@@ -34,12 +34,12 @@ struct test_add_gelu_half : verify_program<test_add_gelu_half>
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
std::vector<size_t> input_lens{1, 1, 5}; std::vector<size_t> input_lens{1, 1, 5};
auto x = mm->add_parameter("x", {migraphx::shape::half_type, input_lens}); auto x = mm->add_parameter("x", {migraphx::shape::half_type, input_lens});
auto y = mm->add_parameter("y", {migraphx::shape::half_type, input_lens}); auto y = mm->add_parameter("y", {migraphx::shape::half_type, input_lens});
auto half = mm->add_literal(migraphx::literal{{migraphx::shape::half_type}, {0.5f}}); auto half = mm->add_literal(migraphx::literal{{migraphx::shape::half_type}, {0.5f}});
auto one = mm->add_literal(migraphx::literal{{migraphx::shape::half_type}, {1.0f}}); auto one = mm->add_literal(migraphx::literal{{migraphx::shape::half_type}, {1.0f}});
auto sqrt2 = mm->add_literal(migraphx::literal{{migraphx::shape::half_type}, {M_SQRT2}}); auto sqrt2 = mm->add_literal(migraphx::literal{{migraphx::shape::half_type}, {M_SQRT2}});
auto add = mm->add_instruction(migraphx::make_op("add"), x, y); auto add = mm->add_instruction(migraphx::make_op("add"), x, y);
auto half_mbcast = mm->add_instruction( auto half_mbcast = mm->add_instruction(
migraphx::make_op("multibroadcast", {{"out_lens", input_lens}}), half); migraphx::make_op("multibroadcast", {{"out_lens", input_lens}}), half);
auto mul_half = mm->add_instruction(migraphx::make_op("mul"), add, half_mbcast); auto mul_half = mm->add_instruction(migraphx::make_op("mul"), add, half_mbcast);
......
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