Skip to content

Commit 5c814e2

Browse files
committed
Bundle PTXAS into 11.8 wheel
1 parent fa8b6d6 commit 5c814e2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

manywheel/build_cuda.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,8 @@ elif [[ $CUDA_VERSION == "11.8" ]]; then
197197
export USE_STATIC_CUDNN=0
198198
# Try parallelizing nvcc as well
199199
export TORCH_NVCC_FLAGS="-Xfatbin -compress-all --threads 2"
200+
# Bundle ptxas into the wheel, see https://github.com/pytorch/pytorch/pull/119750
201+
export BUILD_BUNDLE_PTXAS=1
200202

201203
if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
202204
echo "Bundling with cudnn and cublas."

0 commit comments

Comments
 (0)