fix

microsoft · Oct 21, 2024 · 86d3019 · 86d3019
1 parent 0ab620a
commit 86d3019
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/onnxruntime/test/onnx/main.cc b/onnxruntime/test/onnx/main.cc
@@ -107,11 +107,11 @@ inline void THROW_ON_ERROR(OrtStatus* status) {
 }
 
 void TestTensorRTEp(const OrtApi* g_ort, OrtEnv* env, OrtSessionOptions* so, int device_id) {
-  THROW_ON_ERROR(g_ort->RegisterOrtExecutionProviderLibrary("/home/yifanl/onnxruntime/samples/tensorRTEp/build/libTensorRTEp.so", env, "tensorrtEp"));
+  THROW_ON_ERROR(g_ort->RegisterPluginExecutionProviderLibrary("/home/yifanl/onnxruntime/samples/tensorRTEp/build/libTensorRTEp.so", env, "tensorrtEp"));
   std::vector<const char*> keys{"device_id", "str_property"};
   std::string device_id_str = std::to_string(device_id);
   std::vector<const char*> values{device_id_str.c_str(), "strvalue"};
-  THROW_ON_ERROR(g_ort->SessionOptionsAppendOrtExecutionProvider(so, "tensorrtEp", env, keys.data(), values.data(), keys.size()));
+  THROW_ON_ERROR(g_ort->SessionOptionsAppendPluginExecutionProvider(so, "tensorrtEp", env, keys.data(), values.data(), keys.size()));
 }
 
 static TestTolerances LoadTestTolerances(bool enable_cuda, bool enable_openvino, bool useCustom, double atol, double rtol) {

diff --git a/samples/b.sh b/samples/b.sh
@@ -16,7 +16,7 @@ cd ../../
 
 cd tensorRTEp && rm -r build && mkdir -p build && cd build
 # Download TensorRT with correct cuda version selected
-cmake -S ../ -B ./ -DCMAKE_BUILD_TYPE=Debug -DCMAKE_CUDA_ARCHITECTURES=80 -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc -DTENSORRT_HOME=/home/yifanl/TensorRT-10.4.0.26/ -DORT_HOME=/home/yifanl/onnxruntime/
+cmake -S ../ -B ./ -DCMAKE_BUILD_TYPE=Debug -DCMAKE_CUDA_ARCHITECTURES=80 -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc -DTENSORRT_HOME=/home/yifanl/TensorRT-10.5.0.18/ -DORT_HOME=/home/yifanl/onnxruntime/
 cmake --build ./
 cd ../../