Skip to content

Commit 8585617

Browse files
authored
add magma build for CUDA11.7 (#1049)
Co-authored-by: pbialecki <[email protected]>
1 parent 6bbab18 commit 8585617

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ jobs:
2828
runs-on: linux.2xlarge
2929
strategy:
3030
matrix:
31-
cuda_version: ["116", "115"]
31+
cuda_version: ["117", "116", "115"]
3232
steps:
3333
- name: Checkout PyTorch builder
3434
uses: actions/checkout@v2

magma/Makefile

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ DOCKER_RUN = set -eou pipefail; docker run --rm -i \
1414
magma/build_magma.sh
1515

1616
.PHONY: all
17+
all: magma-cuda117
1718
all: magma-cuda116
1819
all: magma-cuda115
1920
all: magma-cuda113
@@ -24,6 +25,13 @@ clean:
2425
$(RM) -r magma-*
2526
$(RM) -r output
2627

28+
.PHONY: magma-cuda117
29+
magma-cuda117: DESIRED_CUDA := 11.7
30+
magma-cuda117: PACKAGE_NAME := magma-cuda117
31+
magma-cuda117: CUDA_ARCH_LIST += -gencode arch=compute_80,code=sm_80 -gencode arch=compute_86,code=sm_86
32+
magma-cuda117:
33+
$(DOCKER_RUN)
34+
2735
.PHONY: magma-cuda116
2836
magma-cuda116: DESIRED_CUDA := 11.6
2937
magma-cuda116: PACKAGE_NAME := magma-cuda116

0 commit comments

Comments
 (0)