Reflect the changes made by #1920 in RL#2780

tdene merged 3 commits into

NVIDIA:mainNVIDIA/Megatron-LM:mainfrom

tdene:tde/fix_1920_in_rltdene/Megatron-LM:tde/fix_1920_in_rlCopy head branch name to clipboard