Add magma build for CUDA12.4 (#1722)

* Add magma build for CUDA12.4 * Bug fix
pytorch · Mar 12, 2024 · 04881ed · 04881ed
1 parent cd7a668
commit 04881ed
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git 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: ["121", "118"]
+        cuda_version: ["124", "121", "118"]
     steps:
       - name: Checkout PyTorch builder
         uses: actions/checkout@v3

diff --git 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-cuda124
 all: magma-cuda121
 all: magma-cuda118
 
@@ -22,6 +23,12 @@ clean:
 	$(RM) -r magma-*
 	$(RM) -r output
 
+.PHONY: magma-cuda124
+magma-cuda124: DESIRED_CUDA := 12.4
+magma-cuda124: PACKAGE_NAME := magma-cuda124
+magma-cuda124:
+	$(DOCKER_RUN)
+
 .PHONY: magma-cuda121
 magma-cuda121: DESIRED_CUDA := 12.1
 magma-cuda121: PACKAGE_NAME := magma-cuda121