Yongqi ZhuoandClaude Fable 5 30c1904d3a [CUDA] Unify TMA copy lowering on CuTe algebra (#3106) ...

* [CUDA] Unify TMA copy lowering on CuTe algebra and support general swizzled layouts

Every descriptor-based TMA path (bulk copy, atomic add, gather4/scatter4,
im2col) now shares one CuTe-algebra box decomposition. Boxes follow the
driver contract (inner box bytes at most the swizzle span, per-dim caps),
and the recovered swizzle is widened on demand under an equivalence proof,
so general annotated layouts -- including phase-shifted swizzles -- lower
correctly.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

* Address review concerns

---------

Co-authored-by: Claude Fable 5 <noreply@anthropic.com>

2026-08-29 15:43:29 +08:00

..

test_tilelang_annotate_loop_layout.py

[Refactor][Pipeline] Run pipeline rewriting before layout inference and stabilize tiled WS (#2002)

2026-04-07 23:54:55 +08:00

test_tilelang_bank_swizzle_expand.py

[TIR][IR] Update to use tirx (#2216)

2026-05-20 15:38:13 +08:00

test_tilelang_cute.py

[CUDA] Unify TMA copy lowering on CuTe algebra (#3106)

2026-08-29 15:43:29 +08:00

test_tilelang_layout_equal.py

[TIR][IR] Update to use tirx (#2216)

2026-05-20 15:38:13 +08:00

test_tilelang_layout_expand.py

[Layout] Add Layout.expand to lift a layout into higher dimensions and improve repeat errors (#1876)

2026-02-25 12:54:17 +08:00

test_tilelang_layout_fused_replicate.py

[Enhancement] Update examples and tests for improved type handling functionality (#1448)

2025-12-17 14:41:25 +08:00

test_tilelang_layout_inference_z3_context.py

[JIT][FFI] Allocate kernel outputs through the packed API (#2937)

2026-08-11 10:47:08 +08:00

test_tilelang_layout_inference.py

[Transform][Layout] Avoid thread-indexed replicated fragment readback (#2514)

2026-07-03 14:52:24 +08:00

test_tilelang_layout_inverse.py

[Layout] Reject non-round-tripping inferred inverses (#3090)

2026-08-27 18:04:12 +08:00

test_tilelang_layout_linearized_index.py

[CUDA][Pipeline] Fix 1D TMA selection for versioned layouts (#2737)

2026-07-22 21:38:18 +08:00

test_tilelang_layout_repeat.py

[Layout] Add Layout.expand to lift a layout into higher dimensions and improve repeat errors (#1876)

2026-02-25 12:54:17 +08:00

test_tilelang_tcgen05_expand.py

[Layout] Support half-subpartition (M=64) TMEM tiles in tcgen05.ld/st (#2880)

2026-08-11 15:35:15 +08:00