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

Scatter add refactor #314

Merged
merged 6 commits into from
Nov 8, 2024
Merged

Conversation

laserkelvin
Copy link
Collaborator

This PR closes #313 by refactoring the code base to use the scatter_sum ported over by MACE. This allows FAENet to be used without the hard torch_scatter dependency, along with SchNet.

@laserkelvin laserkelvin added enhancement New feature or request dependencies Pull requests that update a dependency file code maintenance Issue/PR for refactors, code clean up, etc. labels Nov 8, 2024
@melo-gonzo
Copy link
Collaborator

Should we also purge from here as part of the PR?

@laserkelvin
Copy link
Collaborator Author

laserkelvin commented Nov 8, 2024

We can - but I felt like I wanted to purge the unused code alongside it too in the rest of the module, and just reduce it down to the wrapper in a separate PR.

@laserkelvin laserkelvin merged commit 2caaee3 into IntelLabs:main Nov 8, 2024
2 of 3 checks passed
@laserkelvin laserkelvin deleted the scatter-add-refactor branch November 8, 2024 22:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
code maintenance Issue/PR for refactors, code clean up, etc. dependencies Pull requests that update a dependency file enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Feature request]: Refactor FAENet to not require torch_scatter
2 participants