[main] [2/5] Qwen3.5 support: FSDP DTensor Bridge checkpoint compatibility - #4753

Open

wplf wants to merge 10 commits into

NVIDIA:mainfrom

wplf:fix/fsdp-dtensor-bridge-checkpoint-main

Commits on May 13, 2026

fix(fsdp): support Megatron-Bridge checkpoints under fsdp_dtensor

wplf
and
BestJuly
committed
9c93a67
compute SWiGLU/GDN split in item coordinates for non-DTensor optimizer states

xuwchen
authored and
wplf
committed
bb5dbe7
chore(lint): apply black formatting to split_gdn_fused TP-local fast-path

wplf
and
claude
committed
8223d35

Commits on May 14, 2026

feat(fsdp): implement model->main param copy for finetune path

wplf
and
claude
committed
f4ba23b
Revert "feat(fsdp): implement model->main param copy for finetune path"

wplf
and
claude
committed
95986b7

Commits on May 15, 2026

drop TP-local GDN DTensor fast-path; will land via #4799

wplf
and
claude
committed
947c91e

Commits on May 18, 2026

Revert "drop TP-local GDN DTensor fast-path; will land via #4799"

wplf
and
claude
committed
8b21572
fix(fsdp): assert plain Tensor is FSDP-local shard in SwiGLU/GDN splitters

xuwchen
authored and
wplf
committed
20c9f2f
docs: replace stale megatron_fsdp.py line refs with symbol names

wplf
and
claude
committed
7590102

Commits on Jun 4, 2026

Merge remote-tracking branch 'origin/main' into resolve-4753
wplf
committed
6fd7b35