The pipeline pass is now functional

2022-05-15 22:29:27 +08:00
parent 7e0e7ec365
commit 7027af9666
4 changed files with 119 additions and 98 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -1315,6 +1315,7 @@ class JITFunction:
        pm.add_canonicalizer_pass()
        pm.add_convert_triton_to_tritongpu_pass()
        pm.add_tritongpu_pipeline_pass()
+        pm.add_canonicalizer_pass()
        pm.run(mod)
        return mod