Use OpenMP to parallelize Autograd::backward in rascaline-torch

Luthaf / rascaline

Computing representations for atomistic machine learning

https://luthaf.fr/rascaline/

BSD 3-Clause "New" or "Revised" License

44 stars 13 forks source link

Use OpenMP to parallelize Autograd::backward in rascaline-torch #307

Closed Luthaf closed 2 months ago

Luthaf commented 3 months ago

This is working on my machine, now to debug the build on other platforms!

📚 Documentation preview 📚: https://rascaline--307.org.readthedocs.build/en/307/

github-actions[bot] commented 3 months ago

Here is a pre-built version of the code in this pull request: wheels.zip, you can install it locally by unzipping wheels.zip and using pip to install the file matching your system

Luthaf commented 3 months ago

I'm surprised it worked everywhere (almost) first try!