update mma encoding & triton-opt

2022-06-06 21:03:58 +08:00
parent 7807f64ef3
commit 366dddc3bc
13 changed files with 88 additions and 28 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -1353,10 +1353,10 @@ void init_triton_ir(py::module &&m) {
        self.addPass(mlir::createTritonGPUPipelinePass(numStages));
      })
      .def("add_triton_gpu_combine_pass", [](mlir::PassManager &self) {
-        self.addPass(mlir::triton::gpu::createCombineOpsPass());
+        self.addPass(mlir::createTritonGPUCombineOpsPass());
      })
      .def("add_triton_gpu_verifier_pass", [](mlir::PassManager &self) {
-        self.addPass(mlir::triton::gpu::createTritonGPUVerifier());
+        self.addPass(mlir::createTritonGPUVerifier());
      })
      ;
 }