-
Notifications
You must be signed in to change notification settings - Fork 661
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
[ROCm] add rocm wheels #1477
[ROCm] add rocm wheels #1477
Conversation
@micmelesse Let us know when this PR is ready. This looks like a good addition. |
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.
LGTM - CI passing for new ROCm jobs.
@michaelauli Can you rebase on the latest master, just in case? I fixed some of the macOS issues. |
Thank you for the change. PR looks good to me, but let's not merge it until we understand why those builds are failing. |
f57b031
to
987506a
Compare
@mthrok I have rebased and force pushed. |
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 think this PR disables CPU builds (and also tried to build ROCM on Windows)
Let's rebase it on top of #1497 when it passes CI sanity
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.
LGTM on green
HI @micmelesse #1497 has been merged. Could you rebase ? |
987506a
to
437612c
Compare
use the term cu_version update pkg_helpers add docker wheel option update config yml add force_cuda option
42a7fed
to
23cf3fb
Compare
Co-authored-by: Teng Gao <tegao@microsoft.com>
This PR adds wheels for rocm. It is based on this PR for pytorch vision(pytorch/vision#3575) that adds rocm wheels .
It works by creating a rocm version of the binary_linux_wheel build jobs that use pytorch/manylinux-rocm instead of manylinux-cuda102.