From 29b3c184449c10c9e2f79395a5507f031489cdbf Mon Sep 17 00:00:00 2001 From: ptrblck Date: Mon, 21 Mar 2022 14:01:09 -0700 Subject: [PATCH] add magma build for CUDA11.6 (#997) Co-authored-by: pbialecki --- .github/workflows/build-magma-linux.yml | 2 +- magma/Makefile | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build-magma-linux.yml b/.github/workflows/build-magma-linux.yml index d26f64b6f..cfd138941 100644 --- a/.github/workflows/build-magma-linux.yml +++ b/.github/workflows/build-magma-linux.yml @@ -28,7 +28,7 @@ jobs: runs-on: linux.2xlarge strategy: matrix: - cuda_version: ["115"] + cuda_version: ["116", "115"] steps: - name: Checkout PyTorch builder uses: actions/checkout@v2 diff --git a/magma/Makefile b/magma/Makefile index 7728c32e7..5a09111e7 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-cuda116 all: magma-cuda115 all: magma-cuda113 all: magma-cuda102 @@ -23,6 +24,13 @@ clean: $(RM) -r magma-* $(RM) -r output +.PHONY: magma-cuda116 +magma-cuda116: DESIRED_CUDA := 11.6 +magma-cuda116: PACKAGE_NAME := magma-cuda116 +magma-cuda116: CUDA_ARCH_LIST += -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 +magma-cuda116: + $(DOCKER_RUN) + .PHONY: magma-cuda115 magma-cuda115: DESIRED_CUDA := 11.5 magma-cuda115: PACKAGE_NAME := magma-cuda115