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

Update onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc

Browse source 
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
This commit is contained in:
jingyanwangms 2025-01-30 21:52:12 -08:00 committed by GitHub
parent afe0a2695e
commit 169e13dd8a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -2497,7 +2497,7 @@ TensorrtExecutionProvider::GetCapability(const GraphViewer& graph,
const auto& node = graph.GetNode(node_index[i]);
const bool is_context_node = node && !node->OpType().empty() && node->OpType() == "EPContext";
if (is_context_node) {
SubGraph_t supported_node_vector (std::vector<long unsigned int>{i}, true);
SubGraph_t supported_node_vector(std::vector<long unsigned int>{i}, true);
std::unique_ptr<IndexedSubGraph> sub_graph = GetSubGraph(supported_node_vector, graph, model_hash, subgraph_idx++);
result.push_back(ComputeCapability::Create(std::move(sub_graph)));
}