[Auto-parallel] fix loss base of sharding fuse optimize test

[waliwali777]

Before submitting

• Lint code. If there are lint issues, please format the code first.

# Install and register `pre-commit` in the project folder
pip install pre-commit && pre-commit install

# Process previous code files separately
pre-commit run --file XXXX.py

• Add test cases into tests folder. If there are codecov issues, please add tests cases first.

PR types

Others

PR changes

Others

Description

该修复PR增加边界情况的处理，导致 测试 llm_gpt_dygraph_auto_bs8_fp16_DP2-MP2-PP2_intermediate 的 loss_base 需要进行更新

[paddle-bot]

Thanks for your contribution!