Skip to content

Commit 667158b

Browse files
committed
ci: use CPU torch compatible with default-device reset
1 parent 6a2e59e commit 667158b

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

‎.github/workflows/ci.yml‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ jobs:
9090
--no-install-package nvidia-nvjitlink-cu12 \
9191
--no-install-package nvidia-nvtx-cu12 \
9292
--no-install-package triton
93-
uv pip install torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu
93+
uv pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
9494
- name: Check
9595
run: uv run --no-sync mypy src/unilab
9696

@@ -126,7 +126,7 @@ jobs:
126126
--no-install-package nvidia-nvjitlink-cu12 \
127127
--no-install-package nvidia-nvtx-cu12 \
128128
--no-install-package triton
129-
uv pip install torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu
129+
uv pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
130130
- name: Check
131131
run: uv run --no-sync pyright
132132

@@ -163,7 +163,7 @@ jobs:
163163
--no-install-package nvidia-nvjitlink-cu12 \
164164
--no-install-package nvidia-nvtx-cu12 \
165165
--no-install-package triton
166-
uv pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu
166+
uv pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
167167
- name: Benchmark entrypoint smoke test
168168
run: uv run --no-sync python scripts/benchmark/smoke_test.py
169169

@@ -209,7 +209,7 @@ jobs:
209209
--no-install-package nvidia-nvjitlink-cu12 \
210210
--no-install-package nvidia-nvtx-cu12 \
211211
--no-install-package triton
212-
uv pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu
212+
uv pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
213213
- name: Cache robot assets
214214
uses: actions/cache@v4.2.3
215215
with:

‎.github/workflows/tensor-runtime-ci.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ jobs:
6060
--no-install-package nvidia-nvjitlink-cu12 \
6161
--no-install-package nvidia-nvtx-cu12 \
6262
--no-install-package triton
63-
uv pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu
63+
uv pip install torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu
6464
- name: Focused contract tests
6565
run: >
6666
uv run --no-sync pytest -q

0 commit comments

Comments
 (0)