-
Notifications
You must be signed in to change notification settings - Fork 25.6k
ROCm: Remove static specifier for allow_tf32 variable. #147186
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: Remove static specifier for allow_tf32 variable. #147186
Conversation
Since the env variable HIPBLASLT_ALLOW_TF32 can change, remove const static type for allow_tf32 variable so that it captures the current value of env variable HIPBLASLT_ALLOW_TF32. Signed-off-by: Jagadish Krishnamoorthy <jagadish.krishnamoorthy@amd.com>
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/147186
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 1 Unrelated FailureAs of commit 641ce99 with merge base 331d5cf ( NEW FAILURE - The following job has failed:
BROKEN TRUNK - The following job failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
@pytorchbot label "topic: not user facing" |
I am working on PR to enable tf32 testing on test_nn for ROCm. |
Signed-off-by: Jagadish Krishnamoorthy <jagadish.krishnamoorthy@amd.com>
Signed-off-by: Jagadish Krishnamoorthy <jagadish.krishnamoorthy@amd.com>
else: | ||
del os.environ["HIPBLASLT_ALLOW_TF32"] | ||
|
||
@unittest.skipIf(not TEST_WITH_ROCM, "not relevant for CUDA testing") |
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.
Test only runs on MI300, so we need to do:
from torch.testing._internal.common_utils import runOnRocmArch, MI300_ARCH
Then we need to add the decorator below to the test.
@runOnRocmArch(MI300_ARCH)
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.
Actually, since we don't execute any test. You can just ignore my last comment.
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 disagree. The new test is not arch-specific, it only tests whether the env var controls the returned value. No kernels are run.
@pytorch merge -f "unrelated rocm ci failures" |
@pytorchbot merge -f "unrelated rocm ci failures" |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Since the env variable HIPBLASLT_ALLOW_TF32 can change, remove static type for allow_tf32 variable so that it captures the current value of env variable HIPBLASLT_ALLOW_TF32. Pull Request resolved: #147186 Approved by: https://github.com/jeffdaily, https://github.com/naromero77amd
Since the env variable HIPBLASLT_ALLOW_TF32 can change, remove static type for allow_tf32 variable so that it captures the current value of env variable HIPBLASLT_ALLOW_TF32.
cc @jeffdaily @sunway513 @jithunnair-amd @pruthvistony @ROCmSupport @dllehr-amd @jataylo @hongxiayang @naromero77amd