[RL] Update reshard

[gongel]

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

Bug fixes

PR changes

APIs

Description

• 适配新的改动

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 48.97%. Comparing base (97b7af4) to head (2537b17).
Report is 82 commits behind head on develop.

❌ Your project status has failed because the head coverage (48.97%) is below the target coverage (58.00%). You can increase the head coverage or adjust the target coverage.

Additional details and impacted files

@@             Coverage Diff             @@
##           develop   #10472      +/-   ##
===========================================
- Coverage    48.99%   48.97%   -0.02%     
===========================================
  Files          765      765              
  Lines       125973   126048      +75     
===========================================
+ Hits         61716    61728      +12     
- Misses       64257    64320      +63     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[gongel]

Move to #10606