-
-
Notifications
You must be signed in to change notification settings - Fork 14.9k
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
magma: 2.0.2 -> 2.5.0 #61347
magma: 2.0.2 -> 2.5.0 #61347
Conversation
@wavewave let me know if you have any input! (If you are the former “ianwookim” that is) |
Heads-up, MAGMA also allows for building with MKL support and I might try to shoe-horn this build optimization into #65041 or layer on an additional PR here. |
Now that PyTorch 1.2 #65041 (comment) is ready, this is blocking. Would someone be able to review? Thanks! Edit: I plan to remove the changes to pytorch, and just have the addition of magma, but would be great to make all changes in one swoop |
It's worth pointing out that I have been using this code and running the full pytorch test suite with it (on cudatoolkit 10 and 9.2), so you can consider this functional and tested. Code changes get a +1 from me. |
@tbenst thank you for upgrading this! 👍 |
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.
I checked this works. btw, please take the maintainership since I am not very familar with recent updates of magma. :-)
(and it is more desirable to be listed as ianwookim
as listed in /maintainers/maintainer-list.nix
.)
Thanks again! @tbenst
undefined variable 'wavewave' at /var/lib/ofborg/checkout/repo/38dca4e3aa6bca43ea96d2fcc04e8229/mr-est/eval-2-shlevy.ewr1.nix.ci/pkgs/development/libraries/science/math/magma/default.nix:46:39 |
I'm probably not a good maintainer for magma -- it's a slightly magical build system which I haven't fully figured out. I'd be happy to be added as a maintainer as a second pair of eyes, however. |
Ok, just wanted to make sure that I acknowledged your contributions to this pull request :). In that case I think this is good to go! |
Motivation for this change
Without magma, many linalg functions in pytorch are broken on GPU. I also had to update magma to support newer versions of CUDA.
Things done
sandbox
innix.conf
on non-NixOS)nix-shell -p nix-review --run "nix-review wip"
./result/bin/
)nix path-info -S
before and after)