SinkConversionsFromShared template

2023-01-06 13:01:08 -08:00
parent 874ee11ab5
commit a81345f7c1
5 changed files with 49 additions and 29 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -1348,6 +1348,11 @@ void init_triton_ir(py::module &&m) {
            [](mlir::PassManager &self) {
              self.addPass(mlir::createTritonGPUOptimizeLoadConvertPass());
            })
+      .def("add_tritongpu_sink_conversions_from_shared_pass",
+            [](mlir::PassManager &self) {
+              self.addPass(
+                  mlir::createTritonGPUSinkConversionsFromSharedPass());
+            })
      .def("add_triton_gpu_to_llvm",
           [](mlir::PassManager &self) {
             self.addPass(mlir::triton::createConvertTritonGPUToLLVMPass());