ViperEkura 2667b8116d refactor: unify paged and contiguous attention kernels via KVSource policy ...

- merge AttentionParams and PagedAttentionParams into one struct
- add attn_kv_source.cuh with ContigKV/PagedKV addressing policies
- template prefill/decode kernels (MMA + scalar) on the KV policy, deleting the four duplicated attn_paged_*.cuh variants
- template dispatcher launchers on KV; single combine kernel
- verify: all correctness tests pass and SASS matches baseline (no perf regression)

2026-08-05 14:06:13 +08:00

..

kernels

refactor: unify paged and contiguous attention kernels via KVSource policy

2026-08-05 14:06:13 +08:00

tests

refactor: unify paged and contiguous attention kernels via KVSource policy

2026-08-05 14:06:13 +08:00

__init__.py

feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention

2026-07-06 12:09:58 +08:00

CMakeLists.txt

build: migrate CUDA kernel build to CMake

2026-08-04 21:27:22 +08:00