AstrAI/csrc
ViperEkura fd6d25ad86 refactor: extract bench_kernel and dispatch_by_head_dim into test_utils 2026-07-12 00:05:04 +08:00
..
kernels feat: wire up paged decode CUDA kernel to Python extension 2026-07-11 18:40:49 +08:00
tests refactor: extract bench_kernel and dispatch_by_head_dim into test_utils 2026-07-12 00:05:04 +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: paged decode attention with split-KV (scalar + MMA) 2026-07-11 18:14:51 +08:00