Format

src/targets/gpu/fuse_mlir.cpp

@@ -16,7 +16,8 @@ struct find_conv_pointwise
     // Find a convolution followed by a pointwise operation.
     auto matcher() const
     {
-        auto convolution = match::skip(match::name("contiguous"))(match::name("convolution").bind("convolution"));
+        auto convolution =
+            match::skip(match::name("contiguous"))(match::name("convolution").bind("convolution"));
         return match::name("pointwise")(match::any_of[match::inputs()](convolution.bind("x")));
     }
 
@@ -30,31 +31,35 @@ struct find_conv_pointwise
         std::sort(names.begin(), names.end());
         module mm{};
         std::unordered_map<instruction_ref, instruction_ref> param_map;
-        auto x = mm.add_parameter("x"+std::to_string(names.size()), conv_ins->inputs().at(0)->get_shape());
-        auto w = mm.add_parameter("x"+std::to_string(names.size()+1), conv_ins->inputs().at(1)->get_shape());
+        auto x    = mm.add_parameter("x" + std::to_string(names.size()),
+                                  conv_ins->inputs().at(0)->get_shape());
+        auto w    = mm.add_parameter("x" + std::to_string(names.size() + 1),
+                                  conv_ins->inputs().at(1)->get_shape());
         auto conv = mm.add_instruction(conv_ins->get_operator(), {x, w});
-        std::transform(names.begin(), names.end(), ins->inputs().begin(), std::inserter(param_map, param_map.end()), [&](auto name, auto input) {
-            if (input == x_ins)
+        std::transform(names.begin(),
+                       names.end(),
+                       ins->inputs().begin(),
+                       std::inserter(param_map, param_map.end()),
+                       [&](auto name, auto input) {
+                           if(input == x_ins)
                                return std::make_pair(pm->get_parameter(name), conv);
-            return std::make_pair(pm->get_parameter(name), mm.add_parameter(name, input->get_shape()));
+                           return std::make_pair(pm->get_parameter(name),
+                                                 mm.add_parameter(name, input->get_shape()));
                        });
         mm.add_return(mm.insert_module_instructions(mm.end(), pm, param_map));
 
         auto inputs = ins->inputs();
         inputs.insert(inputs.end(), conv_ins->inputs().begin(), conv_ins->inputs().end());
-        inputs.push_back(m.insert_instruction(ins, make_op("hip::allocate", {{"shape", to_value(ins->get_shape())}})));
+        inputs.push_back(m.insert_instruction(
+            ins, make_op("hip::allocate", {{"shape", to_value(ins->get_shape())}})));
         auto mlir = insert_mlir(m, ins, mm, inputs);
         m.replace_instruction(ins, mlir);
     }
 };
 
-void fuse_mlir::apply(module& m) const
-{
-    match::find_matches(m, find_conv_pointwise{});    
-
-}
+void fuse_mlir::apply(module& m) const { match::find_matches(m, find_conv_pointwise{}); }
 
-}// namespace gpu
+} // namespace gpu
 
 } // namespace MIGRAPHX_INLINE_NS
 } // namespace migraphx