Skip to content

Commit 206a03f

Browse files
authored
Add B200 CI (#436)
1 parent 334095f commit 206a03f

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

.github/workflows/test.yml

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ concurrency:
1313

1414
jobs:
1515
test:
16-
name: test-cuda12.6-py${{ matrix.python-version }}-${{ matrix.alias }}
16+
name: test-cuda12.9-py${{ matrix.python-version }}-${{ matrix.alias }}
1717

1818
container:
19-
image: nvidia/cuda:12.6.3-devel-ubuntu24.04
19+
image: nvidia/cuda:12.9.1-devel-ubuntu24.04
2020
options: --gpus all
2121

2222
strategy:
@@ -45,6 +45,11 @@ jobs:
4545
ref-mode: "none"
4646
runner: "linux.aws.h100"
4747
alias: "h100"
48+
# B200
49+
- python-version: "3.12"
50+
ref-mode: "none"
51+
runner: "linux.dgx.b200"
52+
alias: "b200"
4853

4954
runs-on: ${{ matrix.runner }}
5055

@@ -82,7 +87,7 @@ jobs:
8287
- name: Install PyTorch
8388
run: |
8489
source .venv/bin/activate
85-
uv pip install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126
90+
uv pip install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu129
8691
8792
- name: Install Triton
8893
if: steps.cache.outputs.cache-hit != 'true'

0 commit comments

Comments
 (0)