can push SWIZZLE through reduce both ways (#6453)

This commit is contained in:
qazal
2024-09-10 16:00:50 +08:00
committed by GitHub
parent 1347e49e82
commit f4f705a07c
2 changed files with 16 additions and 6 deletions

View File

@@ -1723,7 +1723,6 @@ class TestScheduleRewrite(unittest.TestCase):
change = tms[-1] / tms[0]
assert change <= SZ, f"bad complexity, time increased by {change:4.2f}x while input only grew {SZ}x"
@unittest.skip("TODO: this can swizzle twice, once up to LOAD and then down to the STORE")
def test_swizzle_rewrite(self):
# graph rewrite
sink = UOp(UOps.SINK, None, arg=None, src=(