Use Magma-cuda 12.8 for libtorch (#146019)

https://github.com/pytorch/pytorch/issues/145570

Build failure for libtorch wheel
`CUDAContext.cpp:(.text+0x157): additional relocation overflows omitted from the output
/usr/bin/ld: failed to convert GOTPCREL relocation; relink with --no-relax
collect2: error: ld returned 1 exit status`

Unsure if this is related, fixing as a start
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146019
Approved by: https://github.com/eqy
This commit is contained in:
Ting Lu 2025-01-31 02:19:20 +00:00 committed by PyTorch MergeBot
parent 2811f33d12
commit 08d88127fe

View File

@ -7,12 +7,6 @@ function do_install() {
cuda_version=$1
cuda_version_nodot=${1/./}
# Temporary WAR to be updated for CUDA 12.8
if [ "$cuda_version_nodot" == "128" ]; then
# Set it to 12.6 if it matches
cuda_version_nodot="126"
fi
MAGMA_VERSION="2.6.1"
magma_archive="magma-cuda${cuda_version_nodot}-${MAGMA_VERSION}-1.tar.bz2"