Whitelist operators

src/targets/gpu/fuse_mlir.cpp

@@ -28,6 +28,11 @@ struct find_conv_pointwise
         auto x_ins    = r.instructions["x"]; // input after contiguous
         auto pm       = ins->module_inputs().front();
         auto names    = pm->get_parameter_names();
+        // Whitelist pointwise operators
+        if (std::any_of(pm->begin(), pm->end(), [](const auto& i) {
+            return not contains({"@literal", "@param", "@return", "convolution", "add", "relu"}, i.name());
+        }))
+            return;
         std::sort(names.begin(), names.end());
         module mm{};
         std::unordered_map<instruction_ref, instruction_ref> param_map;

src/targets/gpu/mlir.cpp

@@ -468,7 +468,8 @@ struct mlir_program
         std::string tname = get_device_name();
         // HACK: Since MLIR can't handle the full target name
         auto hacked_tname = tname.substr(0, tname.find(":"));
-        mlirMIGraphXAddBackendPipeline(pm.get(), hacked_tname.c_str(), nullptr, nullptr);
+        auto hacked_features = tname.substr(tname.find(":"));
+        mlirMIGraphXAddBackendPipeline(pm.get(), hacked_tname.c_str(), "", hacked_features.c_str());
         mlirPassManagerRun(pm.get(), mmodule.get());
 
         code_object_op op;