TritonGPU combiner

2022-05-16 19:16:01 +08:00
parent 0e68e6eb59
commit e3916c3a46
9 changed files with 109 additions and 118 deletions
--- a/python/triton/code_gen.py
+++ b/python/triton/code_gen.py
@@ -1316,6 +1316,7 @@ class JITFunction:
        pm.add_convert_triton_to_tritongpu_pass()
        pm.add_tritongpu_pipeline_pass()
        pm.add_canonicalizer_pass()
+        pm.add_triton_gpu_combine_pass()
        pm.run(mod)
        return mod