We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 11e91d6 commit bf4474dCopy full SHA for bf4474d
core/runtime/TRTEngine.cpp
@@ -111,7 +111,7 @@ TRTEngine::TRTEngine(
111
for (size_t pyt_idx = 0; pyt_idx < inputs_size; pyt_idx++) {
112
auto binding_name = _in_binding_names[pyt_idx];
113
auto trt_idx = cuda_engine->getBindingIndex(binding_name.c_str());
114
- std::string engine_binded_name = cuda_engine->getIOTensorName(pyt_idx);
+ std::string engine_binded_name = cuda_engine->getIOTensorName(trt_idx);
115
TORCHTRT_CHECK(
116
(binding_name == engine_binded_name),
117
"Could not find a TensorRT engine binding for input named " << binding_name);
0 commit comments