Skip to content

Commit

Permalink
add magma build for CUDA 12.6 (#2020)
Browse files Browse the repository at this point in the history
* add magma build for cuda 12.6

* use v4 actions/upload-artifact

* add target magma-126
  • Loading branch information
tinglvv authored Oct 24, 2024
1 parent fabfde6 commit cb75d7d
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/build-magma-linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ jobs:
runs-on: linux.2xlarge
strategy:
matrix:
cuda_version: ["124", "121", "118"]
cuda_version: ["126", "124", "121", "118"]
steps:
- name: Checkout PyTorch builder
uses: actions/checkout@v3
Expand All @@ -39,7 +39,7 @@ jobs:
run: |
make magma-cuda${{ matrix.cuda_version }}
- name: Save as artifact
uses: actions/upload-artifact@v2
uses: actions/upload-artifact@v4
with:
path: magma/output/linux-64/magma-cuda*.bz2
- name: Install conda
Expand Down
7 changes: 7 additions & 0 deletions magma/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ DOCKER_RUN = set -eou pipefail; docker run --rm -i \
magma/build_magma.sh

.PHONY: all
all: magma-cuda126
all: magma-cuda124
all: magma-cuda121
all: magma-cuda118
Expand All @@ -23,6 +24,12 @@ clean:
$(RM) -r magma-*
$(RM) -r output

.PHONY: magma-cuda126
magma-cuda126: DESIRED_CUDA := 12.6
magma-cuda126: PACKAGE_NAME := magma-cuda126
magma-cuda126:
$(DOCKER_RUN)

.PHONY: magma-cuda124
magma-cuda124: DESIRED_CUDA := 12.4
magma-cuda124: PACKAGE_NAME := magma-cuda124
Expand Down

0 comments on commit cb75d7d

Please sign in to comment.