From 5c814e2527b3f5797488bf57d9d5425e63dcc1ac Mon Sep 17 00:00:00 2001
From: Nikita Shulga <nshulga@meta.com>
Date: Tue, 13 Feb 2024 10:58:30 -0800
Subject: [PATCH] Bundle PTXAS into 11.8 wheel

---
 manywheel/build_cuda.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/manywheel/build_cuda.sh b/manywheel/build_cuda.sh
index 318273ba4..39dafe8b1 100644
--- a/manywheel/build_cuda.sh
+++ b/manywheel/build_cuda.sh
@@ -197,6 +197,8 @@ elif [[ $CUDA_VERSION == "11.8" ]]; then
     export USE_STATIC_CUDNN=0
     # Try parallelizing nvcc as well
     export TORCH_NVCC_FLAGS="-Xfatbin -compress-all --threads 2"
+    # Bundle ptxas into the wheel, see https://github.com/pytorch/pytorch/pull/119750
+    export BUILD_BUNDLE_PTXAS=1
 
     if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
         echo "Bundling with cudnn and cublas."