[Dev] Align decoupled Muon FP8 parameter-gather paths with main - #6015

[Dev] Align decoupled Muon FP8 parameter-gather paths with main#6015

Wohox wants to merge 2 commits into

NVIDIA:devNVIDIA/Megatron-LM:devfrom

Wohox:wohox/fix-dp-reshardable-empty-bucketWohox/Megatron-LM:wohox/fix-dp-reshardable-empty-bucketCopy head branch name to clipboard