TritonGPU verifier

2022-05-24 19:48:56 +08:00
parent 36c45ec687
commit a2c9f919a8
10 changed files with 131 additions and 4 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -1347,6 +1347,9 @@ void init_triton_ir(py::module &&m) {
      .def("add_triton_gpu_combine_pass", [](mlir::PassManager &self) {
        self.addPass(mlir::triton::gpu::createCombineOpsPass());
      })
+      .def("add_triton_gpu_verifier_pass", [](mlir::PassManager &self) {
+        self.addPass(mlir::triton::gpu::createTritonGPUVerifier());
+      })
      ;
 }