[OPTIMIZER] Made layout simplification pass efficient for fused attention kernels (#790)

2022-10-21 16:52:15 -07:00
parent c4726333bf
commit bb0f9235d1
26 changed files with 683 additions and 229 deletions
--- a/python/tests/test_type.py
+++ b/python/tests/test_type.py
@@ -54,7 +54,7 @@ def test_binop_type_check():
    kernel = triton.compiler._compile(binop_type_check,
                                      signature="*fp32",
                                      device=0,
-                                      output="ttgir")
+                                      output="ttir")
    assert (kernel)
    # TODO: Check types of the results

@@ -75,6 +75,6 @@ def test_reduce_type_check():
    kernel = triton.compiler._compile(reduce_type_check,
                                      signature="*fp32",
                                      device=0,
-                                      output="ttgir")
+                                      output="ttir")
    assert (kernel)
    # TODO: Check types of the results