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

[ROCm] add rocm wheels #1477

Merged
merged 3 commits into from
May 28, 2021
Merged

[ROCm] add rocm wheels #1477

merged 3 commits into from
May 28, 2021

Conversation

micmelesse
Copy link
Contributor

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.

@mthrok
Copy link
Collaborator

mthrok commented May 6, 2021

@micmelesse Let us know when this PR is ready. This looks like a good addition.

Copy link
Collaborator

@jeffdaily jeffdaily left a 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.

@micmelesse micmelesse marked this pull request as ready for review May 6, 2021 18:38
@mthrok
Copy link
Collaborator

mthrok commented May 6, 2021

@michaelauli Can you rebase on the latest master, just in case? I fixed some of the macOS issues.

@malfet
Copy link
Contributor

malfet commented May 6, 2021

Thank you for the change. PR looks good to me, but let's not merge it until we understand why those builds are failing.

@micmelesse micmelesse force-pushed the build_rocm_wheels branch from f57b031 to 987506a Compare May 6, 2021 19:24
@micmelesse
Copy link
Contributor Author

@mthrok I have rebased and force pushed.

Copy link
Contributor

@malfet malfet left a 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

Copy link
Member

@seemethere seemethere left a comment

Choose a reason for hiding this comment

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

LGTM on green

@micmelesse
Copy link
Contributor Author

@malfet Is #1497 done?

@mthrok
Copy link
Collaborator

mthrok commented May 25, 2021

HI @micmelesse

#1497 has been merged. Could you rebase ?

@mthrok mthrok mentioned this pull request May 25, 2021
14 tasks
@micmelesse micmelesse force-pushed the build_rocm_wheels branch from 987506a to 437612c Compare May 27, 2021 18:42
use the term cu_version

update pkg_helpers

add docker wheel option

update config yml

add force_cuda option
@micmelesse micmelesse force-pushed the build_rocm_wheels branch 2 times, most recently from 42a7fed to 23cf3fb Compare May 27, 2021 19:48
@micmelesse
Copy link
Contributor Author

@mthrok @malfet I have rebased and everything passes.

@malfet malfet merged commit 3bb3287 into pytorch:master May 28, 2021
mthrok pushed a commit to mthrok/audio that referenced this pull request Dec 13, 2022
Co-authored-by: Teng Gao <tegao@microsoft.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

6 participants