Add an opt-in fused weighted restore for AutoEP - #8326

Merged

hwchen2017 merged 15 commits into

deepspeedai:masterfrom

yh0903:yh0903/autoep-fused-weighted-restore

Sep 2, 2026

Commits on Aug 24, 2026

Add an opt-in fused local token engine for AutoEP
yh0903
committed
104719d

Commits on Aug 25, 2026

Compare the fused and eager local token backends through a full step
yh0903
committed
0c93d4d
Narrow the fused path to the weighted restore, which is where the win is
yh0903
committed
7871a46

Commits on Aug 26, 2026

Reject fused restore for folded tensor parallelism early

yh0903
and
Copilot
committed
5c14b01
Validate fused restore inputs and preserve higher-order gradients

yh0903
and
Copilot
committed
9a4b79a
Avoid importing Triton on ROCm builds

yh0903
and
Copilot
committed
248257b
Merge branch 'master' into yh0903/autoep-fused-weighted-restore
yh0903
authored
e5b7b6a
Tighten fused restore comments

yh0903
and
Copilot
committed
c02ea19

Commits on Aug 27, 2026

Merge branch 'master' into yh0903/autoep-fused-weighted-restore
yh0903
authored
31bbca7

Commits on Aug 29, 2026

Commits on Sep 2, 2026

Merge remote-tracking branch 'upstream/master' into yh0903/autoep-fused-weighted-restore
yh0903
committed
58694a7