ViperEkura f7a16efc9d refactor: extract shared dispatcher header, unify MMA/scalar dispatch format ...

- Merge 3 duplicated dispatch blocks into single attn_dispatchers.cuh
- Merge compute_num_splits from attn_utils.cuh into dispatcher header
- All dim3 grid/block declarations and <<<>>> launches are single-line
- Production .cu files (35-42 loc) only handle torch wrapping + pybind11
- Test files include dispatcher header directly, removing all #ifndef ASTRAI_NO_MMA duplication

2026-07-21 22:21:39 +08:00

..

attn_decode_test.cu

refactor: extract shared dispatcher header, unify MMA/scalar dispatch format

2026-07-21 22:21:39 +08:00

attn_paged_decode_test.cu

refactor: extract shared dispatcher header, unify MMA/scalar dispatch format

2026-07-21 22:21:39 +08:00

attn_prefill_test.cu

refactor: extract shared dispatcher header, unify MMA/scalar dispatch format

2026-07-21 22:21:39 +08:00

test_utils.cuh

refactor: extract shared dispatcher header, unify MMA/scalar dispatch format

2026-07-21 22:21:39 +08:00