ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
AstrAI/csrc/tests
History
ViperEkura 2c3cef1c87 feat: wire up paged decode CUDA kernel to Python extension 
- Add attn_paged_decode wrapper in ops.py with gather fallback
- Register kernel in loader.py and export from __init__.py
- Extract test_utils.cuh shared by all attention unit tests
- Rename attn_paged_vs_contiguous.cu to attn_paged_decode_test.cu
- Refactor decode/prefill tests to use common bf16 helpers and cpu ref
- Fix k_cache dim check in attn_paged_decode.cu
 		2026-07-11 18:40:49 +08:00
..
attn_decode_test.cu feat: wire up paged decode CUDA kernel to Python extension 2026-07-11 18:40:49 +08:00
attn_paged_decode_test.cu feat: wire up paged decode CUDA kernel to Python extension 2026-07-11 18:40:49 +08:00
attn_prefill_test.cu feat: wire up paged decode CUDA kernel to Python extension 2026-07-11 18:40:49 +08:00
test_utils.cuh feat: wire up paged decode CUDA kernel to Python extension 2026-07-11 18:40:49 +08:00