Skip to content

[Arith] Preserve non-divisible nested floormod - #64

Closed
haoyang9804 wants to merge 1 commit into
tile-ai:tilelang_mainfrom
haoyang9804:fix/2955
Closed

[Arith] Preserve non-divisible nested floormod#64
haoyang9804 wants to merge 1 commit into
tile-ai:tilelang_mainfrom
haoyang9804:fix/2955

Conversation

@haoyang9804

Copy link
Copy Markdown

Summary

  • keep a nested floormod when the inner split can wrap and its extent is not divisible by the outer modulus
  • preserve the existing simplification for non-wrapping or provably divisible cases
  • add regression coverage for both x % 64 % 7 and x % 64 % 8

Padding the inner extent cannot justify dropping an active inner modulo because the wrapped expression never produces the padded values.

Related to tile-ai/tilelang#2955.

Testing

  • cmake --build build -j12
  • pytest tests/python/arith/test_arith_iter_affine_map.py -q — 39 passed
  • TileLang mirrored arithmetic suite — 39 passed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants