Format

src/simplify_algebra.cpp

@@ -421,17 +421,19 @@ struct find_dot_broadcast
 
         auto insert_squeeze = [&](instruction_ref b_ins) -> instruction_ref {
             auto input = b_ins->inputs()[0];
-            std::vector<std::size_t> lens(b_ins->get_shape().lens().begin() + naxes, b_ins->get_shape().lens().end());
-            if (b_ins->name() == "multibroadcast")
+            std::vector<std::size_t> lens(b_ins->get_shape().lens().begin() + naxes,
+                                          b_ins->get_shape().lens().end());
+            if(b_ins->name() == "multibroadcast")
             {
-                return m.insert_instruction(ins, make_op("multibroadcast", {{"out_lens", lens}}), input);
+                return m.insert_instruction(
+                    ins, make_op("multibroadcast", {{"out_lens", lens}}), input);
             }
-            else if (b_ins->name() == "broadcast")
+            else if(b_ins->name() == "broadcast")
             {
                 auto v    = b_ins->get_operator().to_value();
                 auto axis = v.at("axis").to<std::size_t>() - naxes;
-                return m.insert_instruction(ins, make_op("broadcast", {{"axis", axis}, {"out_lens", lens}}), input);
-
+                return m.insert_instruction(
+                    ins, make_op("broadcast", {{"axis", axis}, {"out_lens", lens}}), input);
             }
             assert(false);
             return m.end();

test/simplify_algebra_test.cpp

@@ -3009,8 +3009,10 @@ TEST_CASE(dot_broadcast_different_rank)
     {
         auto x  = m1.add_parameter("x", {migraphx::shape::float_type, {768}});
         auto y  = m1.add_parameter("y", {migraphx::shape::float_type, {768, 3072}});
-        auto xb = m1.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {2, 384, 768}}}), x);
-        auto yb = m1.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {2, 768, 3072}}}), y);
+        auto xb = m1.add_instruction(
+            migraphx::make_op("multibroadcast", {{"out_lens", {2, 384, 768}}}), x);
+        auto yb = m1.add_instruction(
+            migraphx::make_op("multibroadcast", {{"out_lens", {2, 768, 3072}}}), y);
         auto dot = m1.add_instruction(migraphx::make_op("dot"), xb, yb);
         m1.add_return({dot});
     };
@@ -3019,10 +3021,13 @@ TEST_CASE(dot_broadcast_different_rank)
     {
         auto x = m2.add_parameter("x", {migraphx::shape::float_type, {768}});
         auto y = m2.add_parameter("y", {migraphx::shape::float_type, {768, 3072}});
-        auto xb = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {384, 768}}}), x);
-        auto yb = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {768, 3072}}}), y);
+        auto xb =
+            m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {384, 768}}}), x);
+        auto yb =
+            m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {768, 3072}}}), y);
         auto dot       = m2.add_instruction(migraphx::make_op("dot"), xb, yb);
-        auto broadcast = m2.add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {2, 384, 3072}}}), dot);
+        auto broadcast = m2.add_instruction(
+            migraphx::make_op("multibroadcast", {{"out_lens", {2, 384, 3072}}}), dot);
         m2.add_return({broadcast});
     };