AstrAI/csrc/tests
ViperEkura fd6d25ad86 refactor: extract bench_kernel and dispatch_by_head_dim into test_utils 2026-07-12 00:05:04 +08:00
..
attn_decode_test.cu refactor: extract bench_kernel and dispatch_by_head_dim into test_utils 2026-07-12 00:05:04 +08:00
attn_paged_decode_test.cu refactor: extract bench_kernel and dispatch_by_head_dim into test_utils 2026-07-12 00:05:04 +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 refactor: extract bench_kernel and dispatch_by_head_dim into test_utils 2026-07-12 00:05:04 +08:00