TritonGPU combiner

python/src/triton.cc

@@ -1344,6 +1344,9 @@ void init_triton_ir(py::module &&m) {
       .def("add_tritongpu_pipeline_pass", [](mlir::PassManager &self) {
         self.addPass(mlir::createTritonGPUPipelinePass());
       })
+      .def("add_triton_gpu_combine_pass", [](mlir::PassManager &self) {
+        self.addPass(mlir::triton::gpu::createCombineOpsPass());
+      })
       ;
 }
 

python/triton/code_gen.py

@@ -1316,6 +1316,7 @@ class JITFunction:
         pm.add_convert_triton_to_tritongpu_pass()
         pm.add_tritongpu_pipeline_pass()
         pm.add_canonicalizer_pass()
+        pm.add_triton_gpu_combine_pass()
         pm.run(mod)
         return mod