TritonGPU combiner

2022-05-16 19:16:01 +08:00
parent 0e68e6eb59
commit e3916c3a46
9 changed files with 109 additions and 118 deletions
--- a/python/src/triton.cc
+++ b/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());
+      })
      ;
 }