From cb75d7d626f959d556b2c49e5da88b08731c375e Mon Sep 17 00:00:00 2001
From: Ting Lu <92425201+tinglvv@users.noreply.github.com>
Date: Fri, 25 Oct 2024 02:03:50 +0800
Subject: [PATCH] add magma build for CUDA 12.6 (#2020)

* add magma build for cuda 12.6

* use v4 actions/upload-artifact

* add target magma-126
---
 .github/workflows/build-magma-linux.yml | 4 ++--
 magma/Makefile                          | 7 +++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/build-magma-linux.yml b/.github/workflows/build-magma-linux.yml
index a97070ae3..e7da485ca 100644
--- a/.github/workflows/build-magma-linux.yml
+++ b/.github/workflows/build-magma-linux.yml
@@ -30,7 +30,7 @@ jobs:
     runs-on: linux.2xlarge
     strategy:
       matrix:
-        cuda_version: ["124", "121", "118"]
+        cuda_version: ["126", "124", "121", "118"]
     steps:
       - name: Checkout PyTorch builder
         uses: actions/checkout@v3
@@ -39,7 +39,7 @@ jobs:
         run: |
           make magma-cuda${{ matrix.cuda_version }}
       - name: Save as artifact
-        uses: actions/upload-artifact@v2
+        uses: actions/upload-artifact@v4
         with:
           path: magma/output/linux-64/magma-cuda*.bz2
       - name: Install conda
diff --git a/magma/Makefile b/magma/Makefile
index 54a9bc0e2..93d78aaf8 100644
--- a/magma/Makefile
+++ b/magma/Makefile
@@ -14,6 +14,7 @@ DOCKER_RUN = set -eou pipefail; docker run --rm -i \
 	magma/build_magma.sh
 
 .PHONY: all
+all: magma-cuda126
 all: magma-cuda124
 all: magma-cuda121
 all: magma-cuda118
@@ -23,6 +24,12 @@ clean:
 	$(RM) -r magma-*
 	$(RM) -r output
 
+.PHONY: magma-cuda126
+magma-cuda126: DESIRED_CUDA := 12.6
+magma-cuda126: PACKAGE_NAME := magma-cuda126
+magma-cuda126:
+	$(DOCKER_RUN)
+
 .PHONY: magma-cuda124
 magma-cuda124: DESIRED_CUDA := 12.4
 magma-cuda124: PACKAGE_NAME := magma-cuda124