Repository navigation

[Main] Fix FusedAdam.use_decoupled_grad mis-set for Megatron-FSDP. - #4427

Merged

cspades merged 4 commits into

NVIDIA:mainfrom

cspades:cye/mfsdp-decgrad-segfault

Apr 27, 2026

Merged

[Main] Fix FusedAdam.use_decoupled_grad mis-set for Megatron-FSDP.#4427
cspades merged 4 commits into
NVIDIA:mainfrom
cspades:cye/mfsdp-decgrad-segfault

Commits on Apr 22, 2026
- Fix segfault caused by not overriding use_decoupled_grad for Megatron-FSDP.
  cspades
  committed
  a1038a6
Commits on Apr 24, 2026
- Merge branch 'main' into cye/mfsdp-decgrad-segfault
  cspades
  authored
  874917b
- Add more assert tests that guard against the FusedAdam.step() bug.
  cspades
  committed
  1a5fcfa
Commits on Apr 27, 2026
- Merge branch 'main' into cye/mfsdp-decgrad-segfault
  cspades
  authored
  881fa37