8000 add +PTX to CUDA 12.4 nightly binaries (#1932) · pytorch/builder@047ed9e · GitHub
[go: up one dir, main page]
Skip to content

Commit 047ed9e

Browse files
ptrblckptrblck
authored
add +PTX to CUDA 12.4 nightly binaries (#1932)
1 parent d9e7bdb commit 047ed9e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎manywheel/build_cuda.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ case ${CUDA_VERSION} in
6363
if [[ "$GPU_ARCH_TYPE" = "cuda-aarch64" ]]; then
6464
TORCH_CUDA_ARCH_LIST="9.0"
6565
else
66-
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0"
66+
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0+PTX"
6767
fi
6868
EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
6969
;;

0 commit comments

Comments
 (0)
0