Skip to content

Commit

Permalink
Revert "add +PTX to CUDA 12.4 nightly binaries (#1932)"
Browse files Browse the repository at this point in the history
This reverts commit 047ed9e.
  • Loading branch information
kit1980 committed Oct 4, 2024
1 parent 3d7d875 commit f7b390a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion manywheel/build_cuda.sh
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ case ${CUDA_VERSION} in
if [[ "$GPU_ARCH_TYPE" = "cuda-aarch64" ]]; then
TORCH_CUDA_ARCH_LIST="9.0"
else
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0+PTX"
TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};9.0"
fi
EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
;;
Expand Down

0 comments on commit f7b390a

Please sign in to comment.