Fix bug of reuse_grad_buf_for_mxfp8_param_ag#2802

chtruong814 merged 2 commits intoNVIDIA:mainNVIDIA/Megatron-LM:mainfrom

kunlunl:mxfp8_fix_mainkunlunl/Megatron-LM:mxfp8_fix_mainCopy head branch name to clipboard