-
Notifications
You must be signed in to change notification settings - Fork 61
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
2.4x avg speedup on PMG struct to CHGNET graph conversion #40
Conversation
2. rewrite arg names nad docstrings 3. added arg in model initialization to select converter algorithm
pyproject.toml
Outdated
@@ -15,6 +15,7 @@ dependencies = [ | |||
"numpy>=1.21.6", | |||
"pymatgen>=2022.4.19", | |||
"torch>=1.11.0", | |||
"cython>=0.29.26" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
pip
is case insensitive. Also, hyphens and underscores are equivalent in package names.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looking great! I think this is ready to go.
…rter(verbose=True)
Converted
graph
data structure creation (not be confused with theCrystal_Graph
object) into c (using Cython as an interfacing layer). Still supports legacy graph creation if c-supported graph creation library import fails.Overall average
graph
creation speedup: 3xOverall average
Crystal_Graph
creation speedup: 2.4xOverall average MD speedup (on GPU): ~2x (may be a noisy benchmark however)
Benchmark on Ac2AgPb supercell: