Skip to content

More efficient Orb state_to_atoms_graph calculation#165

Merged
AdeeshKolluru merged 1 commit intomainfrom
faster_orb_batching
Apr 18, 2025
Merged

More efficient Orb state_to_atoms_graph calculation#165
AdeeshKolluru merged 1 commit intomainfrom
faster_orb_batching

Conversation

@AdeeshKolluru
Copy link
Copy Markdown
Contributor

Summary

Makes Orb's state_to_atoms_graph calculation slightly more efficient with a single loop across the batch by reverting from batch_compute_pbc_radius_graph which internally was already looping across the batch.

Copy link
Copy Markdown
Collaborator

@orionarcher orionarcher left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@AdeeshKolluru AdeeshKolluru merged commit 61fa79c into main Apr 18, 2025
91 checks passed
@AdeeshKolluru AdeeshKolluru deleted the faster_orb_batching branch April 18, 2025 14:44
@janosh janosh added the perf Performance issues or improvements label May 2, 2025
@janosh janosh changed the title Improves Orb's state_to_atoms_graph calculation More efficient Orb state_to_atoms_graph calculation May 2, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

perf Performance issues or improvements

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants