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

..

lib

refactor: move compiled kernel .so files into extension/lib

2026-07-31 15:36:32 +08:00

__init__.py

refactor: replace magic layout ints with TensorLayout enum

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

attention_backend.py

perf: remove split partials memset and overlap decode tile loads

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

attention_ops.py

refactor: replace magic layout ints with TensorLayout enum

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

loader.py

refactor: move compiled kernel .so files into extension/lib

2026-07-31 15:36:32 +08:00

rotary_backend.py

refactor: unify rotary embedding interface and update docs

2026-07-31 16:52:25 +08:00

rotary_ops.py

refactor: unify rotary embedding interface and update docs

2026-07-31 16:52:25 +08:00