File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ runners :
2+ - name : aws-gpu-runner
3+ cloud : aws
4+ instance_type : g4dn.xlarge
5+ machine_image : ami-067a4ba2816407ee9
6+ region : eu-north-1
7+ preemptible :
8+ - true
9+ - false
10+ labels :
11+ - cirun-aws-gpu
Original file line number Diff line number Diff line change @@ -32,15 +32,16 @@ jobs:
3232 - {name: "pertpy", extras: "de"}
3333 - {name: "decoupler", extras: ""}
3434 - {name: "SnapATAC2", extras: ""}
35+ - {name: "rapids_singlecell", extras: "rapids12", gpu: true}
3536 exclude :
3637 - { python: 3.12, is_pre: true }
3738
3839 defaults :
3940 run :
4041 shell : bash -el {0}
4142
42- runs-on : ubuntu-latest
43-
43+ runs-on : ${{ matrix.package.gpu && format('cirun-aws-gpu--{0}', github.run_id) || ' ubuntu-latest' }}
44+
4445 env :
4546 # This env variable is used by the `gh` CLI
4647 GH_TOKEN : ${{ secrets.TOKEN_FOR_ISSUE_WRITE }}
6061 filter : blob:none
6162 path : ${{ matrix.package.name }}
6263
64+ - name : Nvidia SMI sanity check
65+ if : matrix.package.gpu
66+ run : nvidia-smi
67+
6368 - name : Install Python
6469 uses : actions/setup-python@v5
6570 with :
7782 uv pip install --system toml-cli
7883
7984 - name : Install AnnData and scverse package
85+ env :
86+ UV_EXTRA_INDEX_URL : ${{ matrix.package.gpu && 'https://pypi.nvidia.com' || '' }}
8087 run : |
8188 EXTRAS=""
8289 GROUP=""
Original file line number Diff line number Diff line change @@ -20,6 +20,7 @@ The following packages are tested:
2020- pertpy
2121- decoupler
2222- SnapATAC2
23+ - rapids-singlecell
2324
2425## How it Works
2526
You can’t perform that action at this time.
0 commit comments