Skip to content

Commit c77391f

Browse files
committed
fix to use correct API
1 parent a65908f commit c77391f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

plugin_execution_providers/tensorrt/tensorrt_execution_provider.cc

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1717,7 +1717,9 @@ OrtStatus* ORT_API_CALL TensorrtExecutionProvider::GetCapabilityImpl(OrtEp* this
17171717
int number_of_trt_nodes = 0;
17181718
for (const auto& group : supported_nodes_vector) {
17191719
if (!group.first.empty()) {
1720-
std::vector<const OrtNode*> supported_nodes(group.first.size());
1720+
std::vector<const OrtNode*> supported_nodes;
1721+
supported_nodes.reserve(group.first.size());
1722+
17211723
for (const auto& index : group.first) {
17221724
const OrtNode* supported_node = nodes[index];
17231725

@@ -1782,7 +1784,7 @@ OrtStatus* ORT_API_CALL TensorrtExecutionProvider::CompileImpl(_In_ OrtEp* this_
17821784

17831785
// Gets number of node's outputs
17841786
size_t num_node_outputs = 0;
1785-
RETURN_IF_ERROR(ort_api.Node_GetNumInputs(fused_node, &num_node_outputs));
1787+
RETURN_IF_ERROR(ort_api.Node_GetNumOutputs(fused_node, &num_node_outputs));
17861788

17871789
std::vector<const OrtValueInfo*> node_outputs(num_node_outputs);
17881790
RETURN_IF_ERROR(ort_api.Node_GetOutputs(fused_node, node_outputs.data(), node_outputs.size()));

0 commit comments

Comments
 (0)