Add magma cuda build 12.8 (#145765)

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

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145765
Approved by: https://github.com/malfet

.ci/magma/Makefile

@@ -16,9 +16,9 @@ DOCKER_RUN = set -eou pipefail; ${DOCKER_CMD} run --rm -i \
 	magma/build_magma.sh
 
 .PHONY: all
+all: magma-cuda128
 all: magma-cuda126
 all: magma-cuda124
-all: magma-cuda121
 all: magma-cuda118
 
 .PHONY:
@@ -26,6 +26,12 @@ clean:
 	$(RM) -r magma-*
 	$(RM) -r output
 
+.PHONY: magma-cuda128
+magma-cuda128: DESIRED_CUDA := 12.8
+magma-cuda128: CUDA_ARCH_LIST += -gencode arch=compute_100,code=sm_100 -gencode arch=compute_120,code=sm_120
+magma-cuda128:
+	$(DOCKER_RUN)
+
 .PHONY: magma-cuda126
 magma-cuda126: DESIRED_CUDA := 12.6
 magma-cuda126:
@@ -36,11 +42,6 @@ magma-cuda124: DESIRED_CUDA := 12.4
 magma-cuda124:
 	$(DOCKER_RUN)
 
-.PHONY: magma-cuda121
-magma-cuda121: DESIRED_CUDA := 12.1
-magma-cuda121:
-	$(DOCKER_RUN)
-
 .PHONY: magma-cuda118
 magma-cuda118: DESIRED_CUDA := 11.8
 magma-cuda118: CUDA_ARCH_LIST += -gencode arch=compute_37,code=sm_37

.github/workflows/build-magma-linux.yml

@@ -34,7 +34,7 @@ jobs:
       id-token: write
     strategy:
       matrix:
-        cuda_version: ["126", "124", "121", "118"]  # There is no pytorch/manylinux-cuda126 yet
+        cuda_version: ["128", "126", "124", "118"]
     steps:
       - name: Checkout PyTorch
         uses: actions/checkout@v4