Skip to content

Commit 8b76ed8

Browse files
NFC: Fix CICD build script
1 parent 54fcb2b commit 8b76ed8

File tree

3 files changed

+4559
-3
lines changed

3 files changed

+4559
-3
lines changed

mlir-tensorrt/build_tools/scripts/cicd_build.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,9 @@ export ENABLE_ASAN=${ENABLE_ASAN:-OFF}
2727
export CPM_SOURCE_CACHE=${CPM_SOURCE_CACHE:-/.cache.cpm}
2828
export CCACHE_DIR=${CCACHE_DIR:-/ccache}
2929

30-
python3 -m pip install -r integrations/python/requirements-dev.txt --use-pep517
30+
python3 -m pip install uv
31+
python3 -m uv sync --extra cu12
32+
source .venv/bin/activate
3133

3234
ccache --zero-stats || true
3335
rm -rf ${BUILD_DIR} || true

mlir-tensorrt/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@ dependencies = [
99
"pandas>=2.3.2",
1010
"sh>=2.2.2",
1111
"flax",
12-
"torch==2.9.0.dev20250909",
13-
"torchvision==0.24.0.dev20250909",
12+
"torch==2.10.0.dev20251110",
13+
"torchvision==0.25.0.dev20251111",
1414
"nanobind>=2.4, <3.0",
1515
"numpy>=1.19.5, <=2.1.2",
1616
"pybind11>=2.10.0, <=2.13.6",

0 commit comments

Comments
 (0)