Skip to content

Commit 6bbdc01

Browse files
[CI] Update CI with latest config (#41)
1 parent 9beea3c commit 6bbdc01

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

.github/workflows/unit_tests.yml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,11 @@ on:
1919

2020
env:
2121
ref: ${{ github.event.inputs.ref || github.ref }}
22-
CUDA_VERSION: 128
23-
UV_TORCH_BACKEND: cu128
24-
TORCH_VERSION: 2.8.0
25-
PYTHON_VERSION: 3.12 # use 312, no need to recompile libs like flash_attn...
26-
UV_PYTHON: python3.12
22+
CUDA_VERSION: 130
23+
UV_TORCH_BACKEND: cu130
24+
TORCH_VERSION: 2.11.0
25+
PYTHON_VERSION: 3.14
26+
UV_PYTHON: python3.14
2727

2828
concurrency:
2929
group: ${{ github.ref }}-workflow-unit-tests
@@ -69,7 +69,7 @@ jobs:
6969
- list-test-files
7070
runs-on: [ self-hosted, xeon5 ]
7171
container:
72-
image: 10.0.13.31:5000/nvidia/cuda:128-ubuntu22.04_0206
72+
image: 10.0.13.31:5000/nvidia/cuda:128-ubuntu24.04_0325
7373
options: --device /dev/dri --ipc=host --runtime=nvidia --gpus all
7474
volumes:
7575
- /monster/ci/env/entrypoint.sh:/entrypoint.sh
@@ -151,9 +151,9 @@ jobs:
151151
152152
- name: Install
153153
run: |
154-
pip install pip pytest regex setuptools build wheel -U
154+
uv pip install pip pytest regex setuptools build wheel -U
155155
uv pip install -r requirements.txt
156-
pip install -v .
156+
uv pip install -v .
157157
158158
- name: Test ${{ matrix.test_script }}
159159
run: |

0 commit comments

Comments
 (0)