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/triton/code_gen.py
+++ b/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):