TritonGPU verifier

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());
+      })
       ;
 }
 

python/triton/code_gen.py

@@ -1320,8 +1320,8 @@ class JITFunction:
         pm.add_tritongpu_pipeline_pass(num_stages)
         pm.add_canonicalizer_pass()
         pm.add_triton_gpu_combine_pass()
-        pm.run(mod)
-        return mod
+        pm.add_triton_gpu_verifier_pass()
+        return pm.run(mod)
 
 
     def __getitem__(self, grid):