Optimize parallel matmul

[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

Function optimization

PR changes

APIs

Description

[paddle-bot]

Thanks for your contribution!

[codecov-commenter]

Codecov Report

❌ Patch coverage is 66.66667% with 2 lines in your changes missing coverage. Please review.
⚠️ Please upload report for BASE (develop@8074e23). Learn more about missing BASE report.

Files with missing lines	Patch %	Lines
...addleformers/transformers/tensor_parallel_utils.py	33.33%	2 Missing ⚠️

❌ Your patch status has failed because the patch coverage (66.66%) is below the target coverage (80.00%). You can increase the patch coverage or adjust the target coverage.

Additional details and impacted files

@@            Coverage Diff             @@
##             develop    #2897   +/-   ##
==========================================
  Coverage           ?   31.60%           
==========================================
  Files              ?      371           
  Lines              ?    60787           
  Branches           ?        0           
==========================================
  Hits               ?    19209           
  Misses             ?    41578           
  Partials           ?        0           

☔ 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.