fix parameter calculation in auto_parallel mode

[zhiqiu]

PR types

Others

PR changes

Others

Description

fix parameter calculation in auto_parallel mode， use local_shape when parameter is dist

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

Attention: Patch coverage is 33.33333% with 4 lines in your changes missing coverage. Please review.

Project coverage is 52.90%. Comparing base (05ae377) to head (dce969f).
Report is 256 commits behind head on develop.

Files with missing lines	Patch %	Lines
paddlenlp/trainer/trainer.py	33.33%	4 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #9327      +/-   ##
===========================================
+ Coverage    52.74%   52.90%   +0.16%     
===========================================
  Files          669      669              
  Lines       107240   107862     +622     
===========================================
+ Hits         56559    57068     +509     
- Misses       50681    50794     +113     

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