Commit 72e56e7
authored
[TRT RTX EP] Fix bug for generating the correct subgraph in GetCapability (#26132)
### Description
In current TRT RTX EP/ TRT EP implementation, when constructing the
`IndexedSubGraph`, for some cases, it will include the node's unused
output as the SubGraph's output. So, it will return the incorrect
`IndexedSubGraph` from its GetCapability to ORT.
Add the logic to prevent adding the unused node's output.
With this fix, we can avoid generating the incorrect EPContext model
where the EPContext node has unused output.1 parent bdffd76 commit 72e56e7
File tree
2 files changed
+10
-4
lines changed- onnxruntime/core/providers
- nv_tensorrt_rtx
- tensorrt
2 files changed
+10
-4
lines changedLines changed: 5 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1466 | 1466 | | |
1467 | 1467 | | |
1468 | 1468 | | |
1469 | | - | |
| 1469 | + | |
1470 | 1470 | | |
1471 | 1471 | | |
1472 | 1472 | | |
1473 | 1473 | | |
1474 | | - | |
| 1474 | + | |
| 1475 | + | |
| 1476 | + | |
| 1477 | + | |
1475 | 1478 | | |
1476 | 1479 | | |
1477 | 1480 | | |
| |||
Lines changed: 5 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
2114 | 2114 | | |
2115 | 2115 | | |
2116 | 2116 | | |
2117 | | - | |
| 2117 | + | |
2118 | 2118 | | |
2119 | 2119 | | |
2120 | 2120 | | |
2121 | 2121 | | |
2122 | | - | |
| 2122 | + | |
| 2123 | + | |
| 2124 | + | |
| 2125 | + | |
2123 | 2126 | | |
2124 | 2127 | | |
2125 | 2128 | | |
| |||
0 commit comments