mirrors/pytorch
0
0 Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2026-10-28 18:56:51 +08:00
Code Activity

[release-only] Remove +ptx from cuda 13.0 builds (#175567)

Browse Source
This commit is contained in:
Andrey Talman authored and GitHub committed 2026-02-23 17:29:28 -05:00
1 parent 5decbe0079
commit a2813af656
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
.ci/manywheel/build_cuda.sh
+1 -1
View File
@@ -115,7 +115,7 @@ case ${CUDA_VERSION} in
fi
;;
13.0)
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};$([[ "$ARCH" == "aarch64" ]] && echo "11.0;" || echo "")12.0+PTX"
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};$([[ "$ARCH" == "aarch64" ]] && echo "11.0;" || echo "")12.0"
export TORCH_NVCC_FLAGS="-compress-mode=size"
export BUILD_BUNDLE_PTXAS=1
;;