Skip to content

Commit d3f62ac

Browse files
authored
add magma build for CUDA11.8 (#1189)
1 parent 975e05d commit d3f62ac

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
@@ -30,7 +30,7 @@ jobs:
3030
runs-on: linux.2xlarge
3131
strategy:
3232
matrix:
33-
cuda_version: ["117", "116", "115"]
33+
cuda_version: ["118", "117", "116", "115"]
3434
steps:
3535
- name: Checkout PyTorch builder
3636
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-cuda118
1718
all: magma-cuda117
1819
all: magma-cuda116
1920
all: magma-cuda115
@@ -25,6 +26,13 @@ clean:
2526
$(RM) -r magma-*
2627
$(RM) -r output
2728

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

0 commit comments

Comments
 (0)