tilelang

mirror of https://github.com/tile-ai/tilelang.git synced 2026-10-22 06:50:25 +08:00

Files

T

cklxx 2a17fffd44 [TileOP] Add SM70 GEMM FMA fallback (#2339 )

* [TileOP] Add SM70 GEMM FMA fallback

* [TileOP] GemmFMA: drop redundant pre-staging sync and add docstrings

Address PR #2279 review:
- coderabbitai nit: the T.sync_threads() before the cooperative A/B stage
  loads is unnecessary because A_stage / B_stage are freshly allocated
  shared buffers — no other thread holds references to them, and the
  later sync before the GEMM compute is sufficient.
- Docstring Coverage pre-merge check: add module-, helper-, method-, and
  prim_func-level docstrings so coverage clears the 80% threshold.

No behavior change; the prim_func still stages through shared memory and
accumulates via scalar FMA. SM70 pytest trio still expected to pass.

* [TileOP] Harden SM70 GemmFMA fallback

* [TileOP] Fix SM70 FMA staging dtypes

* [TileOP] Tighten SM70 MMA dispatch for transposed A

2026-07-27 20:30:42 +08:00

test_tilelang_kernel_bf16_gemm_tcgen5_ts.py

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

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

test_tilelang_kernel_gemm_batched.py

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

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

test_tilelang_kernel_gemm_simt.py

[Backend] Support TMA lowering for arbitrary (swizzled) SMEM layout (#2380 )

2026-06-18 18:42:01 +08:00

test_tilelang_kernel_gemm_sm75.py

[CUDA] Add SM75 MMA dispatchers for FP16 accumulation and UINT8 (#2392 )