Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
pfultz2 committed May 4, 2024
1 parent 56ee9b3 commit 58000cf
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions test/verify/test_step_dot.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,10 @@ struct test_step_dot : verify_program<test_step_dot>
auto* mm = p.get_main_module();
migraphx::shape as{migraphx::shape::float_type, {128, 4, 64, 196}};
migraphx::shape bs{migraphx::shape::float_type, {128, 4, 196, 196}};
auto a = mm->add_parameter("input", as);
auto b = mm->add_literal(migraphx::generate_literal(bs));
auto step = mm->add_instruction(migraphx::make_op("step", {{"axes", {2}}, {"steps", {2}}}), a);
auto a = mm->add_parameter("input", as);
auto b = mm->add_literal(migraphx::generate_literal(bs));
auto step =
mm->add_instruction(migraphx::make_op("step", {{"axes", {2}}, {"steps", {2}}}), a);
auto dot = mm->add_instruction(migraphx::make_op("dot"), step, b);
mm->add_return({dot});
return p;
Expand Down

0 comments on commit 58000cf

Please sign in to comment.