AstrAI/csrc
ViperEkura f1cc7cedce feat: ldmatrix + smem padding for mma prefill kernel
- replace scalar fragment loads with ldmatrix.sync.x4/x2
- add smem row-stride padding (LD = HEAD_DIM + 8) to eliminate 8-way bank conflicts from HEAD_DIM being a 32-bank multiple
- switch build flag from positive to negative: -DASTRAI_NO_MMA for pre-sm_80 only; mma is the default path
- vectorize scalar path smem loads with float4 ld8
- fix pure-C test configs for ld8 alignment
2026-07-06 20:55:22 +08:00
..
kernels feat: ldmatrix + smem padding for mma prefill kernel 2026-07-06 20:55:22 +08:00
tests feat: tensor-core mma prefill with build-time dispatch 2026-07-06 20:33:24 +08:00
__init__.py feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention 2026-07-06 12:09:58 +08:00
build.py feat: tensor-core mma prefill with build-time dispatch 2026-07-06 20:33:24 +08:00