[Release-Only] Remove ptx from Linux CUDA 12.6 binary builds (#145616)

atalman · web-flow · commit ca3c3a63b808 · 2025-01-24T08:39:52.000-08:00
Cuda 12.6 remove +ptx
diff --git a/.ci/manywheel/build_cuda.sh b/.ci/manywheel/build_cuda.sh
@@ -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")
         ;;
