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