From 143ae98a377b99167d5a9417e9da998313f7d07a Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Tue, 5 Nov 2019 15:04:50 -0800 Subject: [PATCH] Fix a bug in onnxruntime_pybind_state.cc when TENSORRT is enabled (#2326) --- onnxruntime/python/onnxruntime_pybind_state.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/onnxruntime_pybind_state.cc b/onnxruntime/python/onnxruntime_pybind_state.cc index ea3339c5d6..410e9a989f 100644 --- a/onnxruntime/python/onnxruntime_pybind_state.cc +++ b/onnxruntime/python/onnxruntime_pybind_state.cc @@ -408,7 +408,7 @@ void addGlobalMethods(py::module& m) { onnxruntime::CreateExecutionProviderFactory_OpenVINO("CPU"), #endif #ifdef USE_TENSORRT - onnxruntime::CreateExecutionProviderFactory_Tensorrt() + onnxruntime::CreateExecutionProviderFactory_Tensorrt(0) #endif };