diff --git a/.github/workflows/build-magma-linux.yml b/.github/workflows/build-magma-linux.yml index 939040475..f6398ae2b 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: ["117", "116", "115"] + cuda_version: ["118", "117", "116", "115"] steps: - name: Checkout PyTorch builder uses: actions/checkout@v2 diff --git a/magma/Makefile b/magma/Makefile index 2d690c14c..9fe83673a 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-cuda118 all: magma-cuda117 all: magma-cuda116 all: magma-cuda115 @@ -25,6 +26,13 @@ clean: $(RM) -r magma-* $(RM) -r output +.PHONY: magma-cuda118 +magma-cuda118: DESIRED_CUDA := 11.8 +magma-cuda118: PACKAGE_NAME := magma-cuda118 +magma-cuda118: CUDA_ARCH_LIST += -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86 +magma-cuda118: + $(DOCKER_RUN) + .PHONY: magma-cuda117 magma-cuda117: DESIRED_CUDA := 11.7 magma-cuda117: PACKAGE_NAME := magma-cuda117