Skip to content

Commit 4190d78

Browse files
.github/workflows/windows-trt.yml: update to tensorrt 10.14.1 and cuda 13.0.2
1 parent ae2ceaf commit 4190d78

3 files changed

Lines changed: 13 additions & 16 deletions

File tree

.github/workflows/windows-trt.yml

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,17 @@ jobs:
4646
uses: actions/cache@v4
4747
with:
4848
path: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
49-
key: ${{ runner.os }}-vstrt-cuda-13.0.1
49+
key: ${{ runner.os }}-vstrt-cuda-13.0.2
5050

5151
- name: Setup CUDA
5252
if: steps.cache-cuda.outputs.cache-hit != 'true'
5353
run: |
54-
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/13.0.1/network_installers/cuda_13.0.1_windows_network.exe
54+
curl -s -o cuda_installer.exe -L https://developer.download.nvidia.com/compute/cuda/13.0.2/network_installers/cuda_13.0.2_windows_network.exe
5555
cuda_installer.exe -s nvcc_13.0 cudart_13.0 cuda_profiler_api_13.0 crt_13.0 nvptxcompiler_13.0
5656
5757
- name: Download TensorRT
5858
run: |
59-
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.13.3/zip/TensorRT-10.13.3.9.Windows.win10.cuda-13.0.zip
59+
curl -L -o trt.zip https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.14.1/zip/TensorRT-10.14.1.48.Windows.win10.cuda-13.0.zip
6060
unzip trt.zip
6161
mv TensorRT-*/ tensorrt/
6262
@@ -87,15 +87,12 @@ jobs:
8787
uses: actions/checkout@v4
8888
with:
8989
repository: NVIDIA/TensorRT
90-
ref: v10.1.0
90+
ref: v10.14
9191
fetch-depth: 1
9292
path: tensorrt-oss
9393

9494
- name: Override trtexec CMake file
9595
run: |
96-
cp -f -r -v tensorrt/samples ../tensorrt-oss
97-
cp -f -r -v tensorrt/include ../tensorrt-oss
98-
9996
mv trtexec/CMakeLists.txt ../tensorrt-oss/samples/trtexec
10097
mv trtexec/*.cpp ../tensorrt-oss/samples/trtexec
10198
mv longpath.manifest ../tensorrt-oss/samples/trtexec
@@ -105,11 +102,11 @@ jobs:
105102
mv trtexec/trtexec.patch ../tensorrt-oss
106103
cd ../tensorrt-oss
107104
108-
copy samples\utils\fileLock.cpp samples\utils\fileLock-utf16le.cpp
109-
powershell "Get-Content samples\utils\fileLock-utf16le.cpp | Out-File samples\utils\fileLock.cpp -Encoding ascii"
110105
git apply trtexec.patch --verbose
111-
copy samples\utils\fileLock.cpp samples\utils\fileLock-utf8.cpp
112-
powershell "Get-Content samples\utils\fileLock-utf8.cpp | Out-File samples\utils\fileLock.cpp"
106+
107+
- name: Apply patch2
108+
shell: bash
109+
run: sed -i 's/fp16 || bf16 || int8 || fp8 || int4 || best/0/g' ../tensorrt-oss/samples/common/sampleOptions.cpp
113110

114111
- name: Configure trtexec
115112
run: cmake -S ../tensorrt-oss/samples/trtexec -B build_trtexec -G Ninja

vstrt/trtexec/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ add_executable(trtexec
1717
../common/sampleOptions.cpp
1818
../common/sampleReporting.cpp
1919
../common/sampleUtils.cpp
20-
../utils/fileLock.cpp
21-
../utils/cacheUtils.cpp
20+
../../shared/utils/fileLock.cpp
21+
../../shared/utils/cacheUtils.cpp
2222
)
2323

2424
target_include_directories(trtexec PRIVATE

vstrt/trtexec/trtexec.patch

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
diff --git a/samples/utils/fileLock.cpp b/samples/utils/fileLock.cpp
1+
diff --git a/shared/utils/fileLock.cpp b/shared/utils/fileLock.cpp
22
index e155c0b..de6bce2 100644
3-
--- a/samples/utils/fileLock.cpp
4-
+++ b/samples/utils/fileLock.cpp
3+
--- a/shared/utils/fileLock.cpp
4+
+++ b/shared/utils/fileLock.cpp
55
@@ -35,8 +35,11 @@ FileLock::FileLock(ILogger& logger, std::string const& fileName)
66
ss << "Trying to set exclusive file lock " << lockFileName << std::endl;
77
mLogger.log(ILogger::Severity::kVERBOSE, ss.str().c_str());

0 commit comments

Comments
 (0)