cp: Fix: nccl-ub in ddp path (3181) into main - #3182

cp: Fix: nccl-ub in ddp path (3181) into main#3182

youngeunkwon0405 merged 2 commits into

mainNVIDIA/Megatron-LM:mainfrom

cherry-pick-3181-into-mainNVIDIA/Megatron-LM:cherry-pick-3181-into-mainCopy head branch name to clipboard