ViperEkura ddc4bd1cf6 feat: tensor-core mma prefill with build-time dispatch ...

- add register-resident flash-attention kernel using mma.sync.m16n8k16
- dispatch mma vs scalar at build time: pre-sm_80 defines
  -DASTRAI_NO_MMA, else defaults to mma
- scalar path vectorized with float4 smem loads (ld8)

2026-07-06 20:33:24 +08:00

..

kernels

feat: tensor-core mma prefill with build-time dispatch

2026-07-06 20:33:24 +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