mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-17 09:08:35 +00:00
- Add BlockSize_ template parameter to all kernel trait structs, enabling block_size=32 instances for d64/GQA-8 (medium/small/tiny tiers). - 12 new instance files for block_size=32 (bf16/fp16 x mask/nmask x 3 tiers). - Dispatch branches on page_blk_size to select bs32 vs bs64 instances. - Fix potential int32 overflow in KV cache tensor view by using long_index_t for sizes and strides (overflow threshold was ~66K blocks for d64/GQA-8). Made-with: Cursor