saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: handle setBindingDimensions return value in TensorRT EP (#11929)

Browse source
This commit is contained in:
Arseny 2022-06-22 04:30:27 +07:00 committed by GitHub
parent 5646410f65
commit 8c8a781cdb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
View file

@ -1626,7 +1626,11 @@ common::Status TensorrtExecutionProvider::Compile(const std::vector<FusedNodeAnd
for (int j = 0, end = nb_dims; j < end; ++j) {
dimensions.d[j] = static_cast<int32_t>(tensor_shapes[j]);
}
trt_context->setBindingDimensions(binding_index, dimensions);
const bool status = trt_context->setBindingDimensions(binding_index, dimensions);
if (!status) {
ORT_THROW_IF_ERROR(ORT_MAKE_STATUS(ONNXRUNTIME, EP_FAIL,
"TensorRT EP cannot set the dynamic dimensions of a binding"));
}
}
}