ViperEkura e8e228d035 feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention ... Structure: csrc/ -- .cu sources + build.py registry astrai/extension/ -- compiled .so + __init__.py (import dispatcher) setup.py -- CUDAExtension from csrc/build.py REGISTRY Control: CSRC_KERNELS=true|false env var at install time. Fallback: astrai.extension.available dict for runtime detection.		2026-07-06 12:09:58 +08:00
..
config	fix : SFT pipeline position_ids default & doc boundary preservation	2026-07-04 15:59:11 +08:00
dataset	feat: add JSONL dataset store with on-the-fly tokenization	2026-07-04 15:42:33 +08:00
extension	feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention	2026-07-06 12:09:58 +08:00
inference	refactor: clean up inference design patterns	2026-07-05 11:41:54 +08:00
model	perf: replace paged KV cache with contiguous ContiguousCache, decode all groups	2026-07-05 11:34:36 +08:00
parallel	refactor : grad_norm 指标简化，clip_grad_norm 移至 executor	2026-06-30 14:59:43 +08:00
preprocessing	fix : SFT pipeline position_ids default & doc boundary preservation	2026-07-04 15:59:11 +08:00
serialization	feat: add JSONL dataset store with on-the-fly tokenization	2026-07-04 15:42:33 +08:00
tokenize	refactor : Pipeline 去除去重，ids 重命名为 sequence，泛型透传	2026-05-31 15:14:27 +08:00
trainer	fix: scheduler division-by-zero, loss_mask bool	2026-07-03 22:04:55 +08:00
__init__.py	feat: add optional CUDA kernel system (csrc/) + fused GQA decode attention	2026-07-06 12:09:58 +08:00
factory.py	fix: ForwardRef._evaluate Python 3.12 compatibility	2026-07-03 18:41:19 +08:00
protocols.py	refactor: 重构训练后端为 Executor 模式	2026-05-24 20:35:44 +08:00