Skip to content

Conversation

@chilo-ms
Copy link
Contributor

@chilo-ms chilo-ms commented Oct 29, 2025

Description

For TRT EP's GetCapability(), in some case, the GetSubGraph() won't add graph's output to the ComputeCapability/IndexedSubGraph returning to ORT.

The issue if from following code:

...
if (node->GetOutputEdgesCount() > node->OutputDefs().size()) {
 ... // execute here
} else {
  ...
          if (graph_output_names.find(output->Name()) != graph_output_names.end()) {
            graph_outputs_to_add[output] = output_order; // missing this
          }
}

Update TRT RTX EP as well.

Motivation and Context

#25373

@chilo-ms chilo-ms marked this pull request as ready for review October 30, 2025 21:18
Copy link
Contributor

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can commit the suggested changes from lintrunner.

@chilo-ms chilo-ms changed the title [TensorRT EP] Fix bug for missing outputs in the returning ComputeCapability/IndexedSubGraph [TRT/TRT RTX EP] Fix bug for missing outputs in the returning ComputeCapability/IndexedSubGraph Oct 31, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants