frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix torchaudio build when TORCH_CUDA_ARCH_LIST is not set (#161084)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/160988.  The root cause can be found in the same issue.  This fix ensures that when reuse old wheel is on and `torchaudio` wheel is not there, the inductor test job can still rebuild the wheel it needs
Pull Request resolved: https://github.com/pytorch/pytorch/pull/161084
Approved by: https://github.com/malfet, https://github.com/zou3519
This commit is contained in:
Huy Do
2025-08-21 03:47:15 +00:00
committed by PyTorch MergeBot
parent 117f11adb4
commit cfdaaaaa26
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.ci/pytorch/common_utils.sh
View File

@ -152,6 +152,12 @@ function get_pinned_commit() {
function install_torchaudio() {
local commit
commit=$(get_pinned_commit audio)
# TODO (huydhn): PyTorch CI docker image set the default TORCH_CUDA_ARCH_LIST
# to Maxwell. This default doesn't make sense anymore and should be cleaned up
if [[ "${BUILD_ENVIRONMENT}" == *cuda* ]]; then
TORCH_CUDA_ARCH_LIST=$(nvidia-smi --query-gpu=compute_cap --format=csv | tail -n 1)
export TORCH_CUDA_ARCH_LIST
fi
pip_build_and_install "git+https://github.com/pytorch/audio.git@${commit}" dist/audio
}