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

update onnx-tensorrt submodule (#5442)

Browse source
This commit is contained in:
stevenlix 2020-10-09 21:49:40 -07:00 committed by Tianlei Wu
parent 8206ff4a90
commit bcbcb2552a
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cmake/external/onnx-tensorrt vendored

@ -1 +1 @@
Subproject commit fbebb144744b3be8defecd8478d74940056df305
Subproject commit a3a4e38b2dfa7a62b6dcae33c0d1678b3bb5ef2a

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

@ -1337,6 +1337,7 @@ common::Status TensorrtExecutionProvider::Provider_Compile(const std::vector<onn
}
return ORT_MAKE_STATUS(ONNXRUNTIME, FAIL, "TensorRT EP execution context enqueue failed.");
}
cudaDeviceSynchronize();
// Cast INT64 input to INT32 because TensorRT doesn't fully support INT64
for (int i = 0, end = output_binding_names.size(); i < end; ++i) {
@ -1355,7 +1356,6 @@ common::Status TensorrtExecutionProvider::Provider_Compile(const std::vector<onn
}
}
cudaDeviceSynchronize();
for (const auto& binding_index : binding_buffers_to_freeup) {
cudaFree(buffers[binding_index]);
}