-
Notifications
You must be signed in to change notification settings - Fork 25.6k
[ROCm] Updating ROCM_HOME handling for >ROCm 4.0 #55968
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
Conversation
💊 CI failures summary and remediationsAs of commit 652f662 (more details on the Dr. CI page):
🕵️ 1 new failure recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
@jeffdaily @jithunnair-amd @sunway513 |
319f4d3
to
8f64472
Compare
8f64472
to
22d09b2
Compare
torch/utils/cpp_extension.py
Outdated
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.
nit: need a space after comma
I'm not the ROCm expert but I don't like this patch. You can't assume just because hipcc is in usr that you're on a linux distribution with Conventionally, if there is a directory associated with a compiler, there is some command line flag you can pass to the compiler to get the paths. For example, you can do |
Codecov Report
@@ Coverage Diff @@
## master #55968 +/- ##
==========================================
- Coverage 77.14% 76.78% -0.36%
==========================================
Files 1909 1909
Lines 189104 189110 +6
==========================================
- Hits 145878 145215 -663
- Misses 43226 43895 +669 |
We discussed this and decided to use |
- This change is required to handle the case when hipcc is updated to the latest using update-alternatives. - This change doesnt not affect any previous versions of ROCm.
22d09b2
to
652f662
Compare
As discussed I have updated the patch. |
Currently in hipcc we dont have any options to get the installed dirs. I have updated the commit to NOT check the 'usr' directory and use readlink to get the final binary path, this works also when update-alternatives is used. |
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 if CI passes.
The CI failure, doesnt seems to be related to this change
|
@ezyang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
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
Summary: - This change is required to handle the case when hipcc is updated to the latest using update-alternatives. - Update-alternatives support for few ROCm binaries is available from ROCm 4.1 onwards. - This change doesnt not affect any previous versions of ROCm. Pull Request resolved: pytorch#55968 Reviewed By: mruberry Differential Revision: D27785123 Pulled By: ezyang fbshipit-source-id: 8467e468d8d51277fab9b0c8cbd57e80bbcfc7f7
updated to the latest using update-alternatives.
from ROCm 4.1 onwards.