[Auto Parallel][PIR] Refine grad merge when using amp_master_grad#68472

JZ-LIANG merged 1 commit intoPaddlePaddle:developPaddlePaddle/Paddle:developfrom

AndSonder:fix_grad_merge_2AndSonder/Paddle:fix_grad_merge_2Copy head branch name to clipboard