Skip to content

Commit 714dec2

Browse files
committed
CI: Add cupy gpu tests
1 parent 279d985 commit 714dec2

File tree

1 file changed

+48
-0
lines changed

1 file changed

+48
-0
lines changed

.github/workflows/gpu_tests.yml

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
name: GPU jobs
2+
3+
on: [ push, pull_request ]
4+
5+
permissions:
6+
contents: read # to fetch code (actions/checkout)
7+
8+
env:
9+
CCACHE_DIR: "${{ github.workspace }}/.ccache"
10+
11+
concurrency:
12+
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
13+
cancel-in-progress: true
14+
15+
jobs:
16+
cupy_tests:
17+
name: CuPy GPU
18+
runs-on: ghcr.io/cirruslabs/ubuntu-runner-amd64-gpu:22.04
19+
steps:
20+
- name: Checkout xsf repo
21+
uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
22+
with:
23+
submodules: recursive
24+
25+
- name: Setup compiler cache
26+
uses: cirruslabs/cache@v4 #caa3ad0624c6c2acd8ba50ad452d1f44bba078bb # v4
27+
with:
28+
path: ${{ env.CCACHE_DIR }}
29+
# Make primary key unique by using `run_id`, this ensures the cache
30+
# is always saved at the end.
31+
key: ${{ runner.os }}-gpu-ccache-${{ github.run_id }}
32+
restore-keys: |
33+
${{ runner.os }}-gpu-ccache
34+
35+
- name: run nvidia-smi
36+
run: nvidia-smi
37+
38+
- name: run nvidia-smi --query
39+
run: nvidia-smi --query
40+
41+
- uses: prefix-dev/setup-pixi@ba3bb36eb2066252b2363392b7739741bb777659 # v0.8.1
42+
with:
43+
pixi-version: v0.39.2
44+
manifest-path: pixi.toml
45+
cache: false
46+
47+
- name: Run CuPy tests
48+
run: pixi run test-cupy

0 commit comments

Comments
 (0)