Skip to content

GitLab

"...Megatron-LM-v1.1.5-3D_parallelism/megatron/indexer.py" did not exist on "aebde649e30016aa33b2e1345cb22210a2e49b04"
Commit 34a51892 authored by Paul's avatar Paul
Browse files

Format

parent ab48f476
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
test/verify/gemm_add_broadcast1.cpp
View file @ 34a51892
......@@ -39,7 +39,8 @@ struct gemm_add_broadcast1 : verify_program<gemm_add_broadcast1>
auto l1 = mm->add_parameter("1", m1_shape);
auto l2 = mm->add_parameter("2", m2_shape);
auto l3 = mm->add_parameter("3", m3_shape);
auto l3_b = mm->add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {1, 2, 4}}}), l3);
auto l3_b =
mm->add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {1, 2, 4}}}), l3);
auto dot = mm->add_instruction(migraphx::make_op("dot"), l1, l2);
mm->add_instruction(migraphx::make_op("add"), dot, l3_b);
......
test/verify/gemm_add_broadcast2.cpp
View file @ 34a51892
......@@ -39,7 +39,8 @@ struct gemm_add_broadcast2 : verify_program<gemm_add_broadcast2>
auto l1 = mm->add_parameter("1", m1_shape);
auto l2 = mm->add_parameter("2", m2_shape);
auto l3 = mm->add_parameter("3", m3_shape);
auto l3_b = mm->add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {1, 2, 4}}}), l3);
auto l3_b =
mm->add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {1, 2, 4}}}), l3);
auto dot = mm->add_instruction(migraphx::make_op("dot"), l1, l2);
mm->add_instruction(migraphx::make_op("add"), dot, l3_b);
......
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