saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Apply onnx-tensorrt bug fixes (#3785)

Browse source 
* merge latest onnx-tensorrt parser

* differentiate kernel names between graph and subgraph

* merge more TRT parser bug fixes

* merge more onnx-tensorrt bug fixes

* fix merge issue

Co-authored-by: stevenlix <stevenlix>
This commit is contained in:
stevenlix 2020-05-01 01:51:48 -07:00 committed by GitHub
parent e42e0d4787
commit 99ec93ea42
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 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 177a2e470ca28433b8d8ca30e44f79b373f2407e
Subproject commit 77a8d522e582efaac3166e912e96215acbf3f129

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

@ -335,7 +335,8 @@ std::unique_ptr<IndexedSubGraph> TensorrtExecutionProvider::GetSubGraph(SubGraph
// Assign inputs and outputs to subgraph's meta_def
auto meta_def = onnxruntime::make_unique<::onnxruntime::IndexedSubGraph::MetaDef>();
meta_def->name = "TRTKernel_" + graph.Name() + "_" + std::to_string(kernels_index++);
const std::string graph_type = graph.IsSubgraph() ? "subgraph" : "graph";
meta_def->name = "TRTKernel_" + graph_type + "_" + graph.Name() + "_" + std::to_string(kernels_index++);
meta_def->domain = kMSDomain;
for (const auto& input : inputs) {