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