ViperEkura 7feeb0b93e refactor: replace magic layout ints with TensorLayout enum ...

- Add TensorLayout enum (C++ + Python) to replace magic layout ints
- Add C10_CUDA_CHECK post-launch error checking to all kernel entries
- Add CUDAGuard + freqs_cis shape validation to rotary_emb.cu
- Cache SM count to eliminate per-call cudaDeviceGetAttribute
- Add DISPATCH_CAUSAL_MASK macro to deduplicate dispatcher if/else
- Convert mask type hints from X|None to Optional[X]

2026-08-01 11:05:52 +08:00

..

attn_common.h

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00

attn_decode_split_kv_mma.cuh

perf: remove split partials memset and overlap decode tile loads

2026-07-31 22:37:44 +08:00

attn_decode_split_kv.cuh

style: unify decode expf to __expf

2026-07-31 00:19:18 +08:00

attn_decode.cu

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00

attn_dispatchers.cuh

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00

attn_entry_utils.cuh

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00

attn_mma_utils.cuh

build: parametrize CUDA version for wheels and docker

2026-07-31 14:10:55 +08:00

attn_paged_decode_split_kv_mma.cuh

perf: remove split partials memset and overlap decode tile loads

2026-07-31 22:37:44 +08:00

attn_paged_decode_split_kv.cuh

fix: stabilize paged decode attention kernels

2026-07-31 21:01:12 +08:00

attn_paged_decode.cu

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00

attn_prefill_split_q_mma.cuh

feat: unify attention backend with multi-dim mask support

2026-07-30 20:38:34 +08:00

attn_prefill_split_q.cuh

feat: unify attention backend with multi-dim mask support

2026-07-30 20:38:34 +08:00

attn_prefill.cu

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00

attn_warp_utils.cuh

perf: extend MMA decode to arbitrary GQA ratio, add launch bounds, vectorize combine

2026-07-27 00:35:34 +08:00

rotary_emb.cu

refactor: replace magic layout ints with TensorLayout enum

2026-08-01 11:05:52 +08:00