File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -13,10 +13,10 @@ concurrency:
13
13
14
14
jobs :
15
15
test :
16
- name : test-cuda12.6 -py${{ matrix.python-version }}-${{ matrix.alias }}
16
+ name : test-cuda12.9 -py${{ matrix.python-version }}-${{ matrix.alias }}
17
17
18
18
container :
19
- image : nvidia/cuda:12.6.3 -devel-ubuntu24.04
19
+ image : nvidia/cuda:12.9.1 -devel-ubuntu24.04
20
20
options : --gpus all
21
21
22
22
strategy :
45
45
ref-mode : " none"
46
46
runner : " linux.aws.h100"
47
47
alias : " h100"
48
+ # B200
49
+ - python-version : " 3.12"
50
+ ref-mode : " none"
51
+ runner : " linux.dgx.b200"
52
+ alias : " b200"
48
53
49
54
runs-on : ${{ matrix.runner }}
50
55
82
87
- name : Install PyTorch
83
88
run : |
84
89
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
86
91
87
92
- name : Install Triton
88
93
if : steps.cache.outputs.cache-hit != 'true'
You can’t perform that action at this time.
0 commit comments