Ye Wang d7bb3b10cc [fmha-bwd] Flat cu_id remap for arbitrary CTA_NUM + grid Y/Z override env ...

The persistent bwd group kernel remapped block->CU as bx/8 + (bx%8)*32,
which is only bijective for N=256 -> wrong gradients when ROCM_FLASH_ATTN_CU_NUM
is set to any other value. Replace with flat blockIdx (bx + by*gridDim.x +
bz*gridDim.x*gridDim.y), a bijection for any 1D/3D grid. Also add
get_persistent_grid_override() honoring ROCM_FLASH_ATTN_GRID_Y/Z to reshape
the persistent grid for 1D-vs-3D scheduling experiments.

2026-06-11 13:19:51 +00:00

..

ck

[CK][CK Tile] Conv Bwd Data flush cache and profiling improvements (#6090)

2026-04-04 00:22:22 +00:00

ck_tile

[fmha-bwd] Flat cu_id remap for arbitrary CTA_NUM + grid Y/Z override env

2026-06-11 13:19:51 +00:00

rapidjson

Update pre-commit to fixed versions, run remod for ck_tile (#2895)

2025-10-16 15:29:17 -07:00