ViperEkura 16a55bb474 refactor: reorganize CUDA kernels into per-family directories ...

- move attention kernels to csrc/kernels/attention/ and rotary to rotary/
- add shared common/mma.cuh (mma_sync, ldmatrix) and device.cuh (sm checks)
- split fp8_mm into three-layer fp8/common.h, gemm.cuh, mm.cu
- fix fused FP8 GEMM ldmatrix lane indexing to fix OOB shared reads
- update extension ops, loader, and kernel tests

2026-08-22 20:40:31 +08:00

..

config

refactor: simplify training and inference interfaces

2026-08-19 20:55:13 +08:00

dataset

refactor: deduplicate low-risk code paths

2026-08-19 16:17:40 +08:00

extension

refactor: reorganize CUDA kernels into per-family directories

2026-08-22 20:40:31 +08:00

inference

test: refactor tests and fix inference edge cases

2026-08-21 22:59:51 +08:00

model

feat: load HuggingFace checkpoints via key/config conversion

2026-08-20 11:34:59 +08:00

optim

fix: use torch.optim.AdamW in ManoAdamW instead of NAdamW

2026-08-01 09:20:54 +08:00

parallel

refactor: deduplicate low-risk code paths

2026-08-19 16:17:40 +08:00

preprocessing

fix : correct prefill sampling and record alignment

2026-08-05 22:20:29 +08:00

serialization

test: refactor tests and fix inference edge cases

2026-08-21 22:59:51 +08:00

tokenize

refactor: simplify BaseFactory and separate ModelFactory from AutoModel

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

trainer

feat: load HuggingFace checkpoints via key/config conversion

2026-08-20 11:34:59 +08:00

__init__.py

refactor: simplify training and inference interfaces

2026-08-19 20:55:13 +08:00

factory.py

refactor: simplify BaseFactory and separate ModelFactory from AutoModel

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

logging.py

feat: add distributed rank to logs

2026-08-20 01:37:00 +08:00

protocols.py

refactor: 重构训练后端为 Executor 模式

2026-05-24 20:35:44 +08:00

signal_handler.py

refactor: move signal_handler from parallel/ to top-level for broader reuse

2026-07-28 10:36:17 +08:00