ktransformers

mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-06-06 15:54:02 +00:00

Files

mrhaoxx 020eb929f7 refactor(sft): unify KTConfig field names with kt_ prefix, add share_cache_pool, remove dead code

- KTConfig fields all use kt_ prefix matching dict keys — eliminates
  _OLD_TO_NEW mapping and prefix-stripping in wrapper.py
- Add kt_share_cache_pool field, auto-enabled when gradient_checkpointing
  is on (via training_args.py), flows through to C++ cache allocation
- Remove dead checkpoint detection code: in_ckpt_recompute,
  in_ckpt_first_forward vars (assigned but never read), fallback
  _is_in_checkpoint_first_forward() function, unused inspect import
- Remove redundant env var fallbacks in wrapper.py for share_backward_bb
  and share_cache_pool (KTConfig.__post_init__ already handles env vars)
- Simplify layer.py checkpoint logic to single _checkpoint_hook_mode() check

Verified: Qwen3-235B 3-step training on sap4, loss matches baseline
(1.2886 / 1.9824 / 1.377 vs 1.2886 / 1.9766 / 1.3809)

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

2026-04-09 14:17:50 +08:00

cli

[fix] improve Sglang kt-kernel detect time duration (#1887 )

2026-03-18 23:07:40 +08:00

sft

refactor(sft): unify KTConfig field names with kt_ prefix, add share_cache_pool, remove dead code

2026-04-09 14:17:50 +08:00

utils

merge: integrate origin/main into sft branch

2026-04-08 23:19:28 +08:00

__init__.py

merge: integrate origin/main into sft branch