More on the pipeline pass

2022-05-11 20:31:08 +08:00
parent 1a4fbed25b
commit d23d7b244c
4 changed files with 99 additions and 2 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -1314,6 +1314,7 @@ class JITFunction:
        pm.add_triton_combine_pass()
        pm.add_canonicalizer_pass()
        pm.add_convert_triton_to_tritongpu_pass()
+        pm.add_tritongpu_pipeline_pass()
        pm.run(mod)
        return mod