Format

src/fuse_reduce.cpp

@@ -52,15 +52,15 @@ struct fused_reduce
         {
             MIGRAPHX_THROW("should have one submodule.");
         }
-        auto* sm    = mods.front();
+        auto* sm = mods.front();
         check_shapes{inputs, *this}.has(sm->get_parameter_shapes().size()).same_dims();
-        auto s = inputs.at(0);
-        auto lens       = s.lens();
+        auto s    = inputs.at(0);
+        auto lens = s.lens();
         for(const auto& axis : axes)
         {
             lens[axis] = 1;
         }
-        if (sm->get_output_shapes().size() != 1)
+        if(sm->get_output_shapes().size() != 1)
             MIGRAPHX_THROW("Only one output supported");
         return inputs[0].with_lens(sm->get_output_shapes().front().type(), lens);
     }
@@ -75,15 +75,16 @@ static void create_reduce_modules(module_pass_manager& mpm)
     {
         if(not ins->get_operator().attributes().get("reduce", false))
             continue;
-        if (ins->inputs().size() != 1)
+        if(ins->inputs().size() != 1)
             continue;
 
-        auto* rm = mpm.create_module(mpm.get_module().name() + ":" + ins->name() + std::to_string(n++));
+        auto* rm =
+            mpm.create_module(mpm.get_module().name() + ":" + ins->name() + std::to_string(n++));
         rm->set_bypass();
 
         // TODO: Ensure standard shape
         auto x0 = rm->add_parameter("x0", ins->inputs().front()->get_shape());
-        auto r = rm->add_instruction(ins->get_operator(), x0);
+        auto r  = rm->add_instruction(ins->get_operator(), x0);
         rm->add_return({r});
 
         // TODO: Set axes
@@ -91,22 +92,27 @@ static void create_reduce_modules(module_pass_manager& mpm)
     }
 }
 
-static std::unordered_map<instruction_ref, instruction_ref> get_ins_param_map(const std::vector<instruction_ref>& inputs, const_module_ref sm)
+static std::unordered_map<instruction_ref, instruction_ref>
+get_ins_param_map(const std::vector<instruction_ref>& inputs, const_module_ref sm)
 {
     std::unordered_map<instruction_ref, instruction_ref> result;
     auto names = sm->get_parameter_names();
     std::sort(names.begin(), names.end());
     assert(names.size() == inputs.size());
-    std::transform(names.begin(), names.end(), inputs.begin(), std::inserter(result, result.end()), [&](const auto& name, auto input) {
-        return std::make_pair(input, sm->get_parameter(name));
-    });
+    std::transform(names.begin(),
+                   names.end(),
+                   inputs.begin(),
+                   std::inserter(result, result.end()),
+                   [&](const auto& name, auto input) {
+                       return std::make_pair(input, sm->get_parameter(name));
+                   });
     return result;
 }
 
 static std::vector<instruction_ref> get_returns(module& m)
 {
     auto last = std::prev(m.end());
-    if (last->name() == "@return")
+    if(last->name() == "@return")
         return last->inputs();
     return {last};
 }
@@ -115,31 +121,33 @@ struct find_reduce_pointwise
 {
     auto matcher() const
     {
-        return match::name("pointwise")(match::any_of[match::inputs()](match::name("fused_reduce")(match::used_once()).bind("reduce")));
+        return match::name("pointwise")(match::any_of[match::inputs()](
+            match::name("fused_reduce")(match::used_once()).bind("reduce")));
     }
 
     void apply(module_pass_manager& mpm, const match::matcher_result& r) const
     {
-        auto ins = r.result;
+        auto ins    = r.result;
         auto reduce = r.instructions["reduce"];
 
         auto* old_rm = reduce->module_inputs().front();
-        auto* rm = mpm.create_module(old_rm->name() + ":pointwise");
+        auto* rm     = mpm.create_module(old_rm->name() + ":pointwise");
         // Copy module
-        *rm = *old_rm;
-        auto map_ins = get_ins_param_map(reduce->inputs(), rm);
+        *rm             = *old_rm;
+        auto map_ins    = get_ins_param_map(reduce->inputs(), rm);
         auto new_inputs = reduce->inputs();
-        for(auto input:ins->inputs())
+        for(auto input : ins->inputs())
         {
             if(contains(map_ins, input))
                 continue;
-            if (input == reduce)
+            if(input == reduce)
             {
                 map_ins[input] = get_returns(*rm).front();
             }
             else
             {
-                map_ins[input] = rm->add_parameter("x" + std::to_string(new_inputs.size()), input->get_shape());
+                map_ins[input] =
+                    rm->add_parameter("x" + std::to_string(new_inputs.size()), input->get_shape());
                 new_inputs.push_back(input);
             }
         }