Repository navigation

[AutoTP] Support frozen AutoTP models and manage AutoTP mode per model - #8851

Open

jinyouzhi wants to merge 6 commits into

deepspeedai:masterfrom

jinyouzhi:autotp-frozen-replicated-hook

Open

[AutoTP] Support frozen AutoTP models and manage AutoTP mode per model#8851
jinyouzhi wants to merge 6 commits into
deepspeedai:masterfrom
jinyouzhi:autotp-frozen-replicated-hook

Commits on Oct 17, 2026
- Support frozen AutoTP models and keep the AutoTP mode per model
  
  jinyouzhi
  and
  Copilot
  committed
  6b6f4b8
- Reject unfreezing AutoTP replicated params after initialize
  
  jinyouzhi
  and
  Copilot
  committed
  53777bf
- Carry the AutoTP training mode per model in AutoTPMeta
  
  jinyouzhi
  and
  Copilot
  committed
  68647c9
Commits on Oct 21, 2026
- Keep layers without an explicit AutoTP mode following set_autotp_mode
  
  jinyouzhi
  and
  Copilot
  committed
  63bb7f5
Commits on Oct 23, 2026
- Keep training_mode=None in AutoTP layer metadata
  
  jinyouzhi
  and
  Copilot
  committed
  11a3cb3
Commits on Oct 29, 2026
- Resolve an unspecified AutoTP mode when it is used, not at construction
  
  jinyouzhi
  and
  Copilot
  committed
  d93285f