Skip to content

Fix TEGroupedMLP pre-backward unshard in fine-grained FSDP hooks for … - #4990

Merged
rapatel merged 7 commits into
NVIDIA:mainfrom
rapatel:ripatel/fsdp_1f1b_pre_bwd_fix_main
Jun 30, 2026
Merged

Fix TEGroupedMLP pre-backward unshard in fine-grained FSDP hooks for …#4990
rapatel merged 7 commits into
NVIDIA:mainfrom
rapatel:ripatel/fsdp_1f1b_pre_bwd_fix_main

Fix line-too-long linting error.

cc78b37
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jun 30, 2026 in 0s

DCO

All commits are signed off!