From 08d88127fe4b839bdfee63b60ceb8d1a44540cf8 Mon Sep 17 00:00:00 2001
From: Ting Lu <tingl@nvidia.com>
Date: Fri, 31 Jan 2025 02:19:20 +0000
Subject: [PATCH] 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
---
 .ci/docker/common/install_magma.sh | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/.ci/docker/common/install_magma.sh b/.ci/docker/common/install_magma.sh
index 8d14bd2c324..1d8db6154e5 100644
--- a/.ci/docker/common/install_magma.sh
+++ b/.ci/docker/common/install_magma.sh
@@ -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"