add magma build for CUDA11.6 (#997)

Co-authored-by: pbialecki <pbialecki@nvidia.com>

.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

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