Skip to content

Commit e5eef90

Browse files
Updates MLIR-TRT to use TensorRT 10.9 (#602)
Updates the build_wheels.sh script to use TensorRT 10.9 by default. Also bumps the version number so we can make a new release.
1 parent 3f8d14e commit e5eef90

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

mlir-tensorrt/Version.cmake

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
set(MLIR_TENSORRT_VERSION_MAJOR "0")
22
set(MLIR_TENSORRT_VERSION_MINOR "1")
3-
set(MLIR_TENSORRT_VERSION_PATCH "39")
3+
set(MLIR_TENSORRT_VERSION_PATCH "40")
44
set(MLIR_TENSORRT_VERSION
55
"${MLIR_TENSORRT_VERSION_MAJOR}.${MLIR_TENSORRT_VERSION_MINOR}.${MLIR_TENSORRT_VERSION_PATCH}")
6-

mlir-tensorrt/build_tools/scripts/build_wheels.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Builds Python wheels for MLIR-TensorRT packages using the specified python version number.
33
# e.g.
44
# PY_VERSION=3.10 build_wheels.sh
5-
# PY_VERSION=3.10 DOWNLOAD_TENSORRT_VERSION=10.7 build_wheels.sh
5+
# PY_VERSION=3.10 DOWNLOAD_TENSORRT_VERSION=10.9 build_wheels.sh
66
set -e
77
py_version=${PY_VERSION:-3.10}
88

@@ -11,7 +11,7 @@ mkdir -p .private.wheels || true
1111
pyenv local ${py_version}
1212
python${py_version} -m pip install -r python/requirements-dev.txt
1313

14-
export DOWNLOAD_TENSORRT_VERSION=${DOWNLOAD_TENSORRT_VERSION:-10.7}
14+
export DOWNLOAD_TENSORRT_VERSION=${DOWNLOAD_TENSORRT_VERSION:-10.9}
1515

1616
rm -rf build || true
1717
cmake --preset ninja-release-wheels

0 commit comments

Comments
 (0)