Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

pt: Add parallel implementation for LKF #3436

Merged
merged 11 commits into from
Mar 11, 2024

Conversation

CaRoLZhangxy
Copy link
Collaborator

Add parallel implementation for LKF, enabling distributed storage of the P matrix across multiple GPUs, reducing memory overhead.

@github-actions github-actions bot added the Python label Mar 8, 2024
deepmd/pt/optimizer/LKF.py Fixed Show fixed Hide fixed
deepmd/pt/optimizer/LKF.py Fixed Show fixed Hide fixed
Copy link

codecov bot commented Mar 8, 2024

Codecov Report

Attention: Patch coverage is 29.88506% with 61 lines in your changes are missing coverage. Please review.

Project coverage is 77.36%. Comparing base (804848a) to head (a5583c6).

Files Patch % Lines
deepmd/pt/optimizer/LKF.py 29.88% 61 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##            devel    #3436      +/-   ##
==========================================
- Coverage   77.39%   77.36%   -0.03%     
==========================================
  Files         427      394      -33     
  Lines       36755    32028    -4727     
  Branches     1621      909     -712     
==========================================
- Hits        28446    24780    -3666     
+ Misses       7441     6720     -721     
+ Partials      868      528     -340     

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

@wanghan-iapcm wanghan-iapcm requested a review from zjgemi March 8, 2024 08:25
@wanghan-iapcm wanghan-iapcm enabled auto-merge March 9, 2024 02:33
auto-merge was automatically disabled March 11, 2024 07:46

Head branch was pushed to by a user without write access

@wanghan-iapcm wanghan-iapcm enabled auto-merge March 11, 2024 14:39
@wanghan-iapcm wanghan-iapcm added this pull request to the merge queue Mar 11, 2024
Merged via the queue into deepmodeling:devel with commit 24d02b7 Mar 11, 2024
48 checks passed
@njzjz njzjz mentioned this pull request Apr 2, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants