-
Notifications
You must be signed in to change notification settings - Fork 24.9k
Add torchao to install_inductor_benchmark_deps cleanup stage #160191
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
Add torchao to install_inductor_benchmark_deps cleanup stage #160191
Conversation
@pytorchbot label "ciflow/inductor-perf-nightly-aarch64" |
Didn't find following labels among repository labels: ciflow/inductor-perf-nightly-aarch64 |
@pytorchbot label "topic: not user facing" |
@pytorchbot label "ciflow/inductor-perf-test-nightly-aarch64" |
Didn't find following labels among repository labels: ciflow/inductor-perf-test-nightly-aarch64 |
Looks like it has fixed the issue - > https://github.com/pytorch/pytorch/actions/runs/16833390878/job/47686603854 |
@@ -48,4 +48,4 @@ install_huggingface | |||
install_timm | |||
|
|||
# Clean up | |||
conda_run pip uninstall -y torch torchvision torchaudio triton | |||
conda_run pip uninstall -y torch torchvision torchaudio triton torchao |
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 just realize that there is one more permission error bug in this script. After installing TorchBench on line 37, the script runs under root as part of the Docker build, so another chown -R jenkins /opt/conda
is needed similar to https://github.com/pytorch/pytorch/blob/main/.ci/docker/common/install_conda.sh#L21. This will fix the remaining error in this script
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.
Thank you for the fix! Uninstalling torchao is the right thing to do here. There is another permission bug from my comment, but feel free to fix it here together with this change, or I could submit another PR to fix it after this lands. It's up to you
@pytorchbot merge -f 'Let land this to unblock #159291, I will submit a fix for the rest of these failures in a separate PR' |
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 |
…#160191) It looks like `torcho` was missed from the cleanup during torchbench setup. Fixes pytorch#160188 Pull Request resolved: pytorch#160191 Approved by: https://github.com/huydhn
Thanks for the approval and for merging @huydhn . I would have fixed the permissions in this PR, but we have a process to follow and it was Fri PM in UK, so I wouldn't be able to make the change until Monday, so better to merge as-is 👍 |
It looks like
torcho
was missed from the cleanup during torchbench setup.Stack from ghstack (oldest at bottom):
Fixes #160188
cc @seemethere @malfet @pytorch/pytorch-dev-infra @snadampal @milpuz01 @aditew01 @nikhil-arm @fadara01