Skip to content

Commit 4f208b3

Browse files
authored
[NV TensorRt RTX EP] : Fix Domain check. (#24816)
### Description <!-- Describe your changes. --> Small change to remove the MS Domain check on onnx model nodes ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> The check returns unsupported for some nodes having an MS Domain. Trt RTX supports some MS domain ops. if return unsupported these ops falls back to CPU EP @ankan-ban @chilo-ms @gedoensmax @jywu-msft Co-authored-by: iraut <iraut@nvidia.com>
1 parent fbc6b23 commit 4f208b3

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

onnxruntime/core/providers/nv_tensorrt_rtx/nv_execution_provider.cc

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1989,10 +1989,6 @@ NvExecutionProvider::GetCapability(const GraphViewer& graph,
19891989
if (exclude_ops_set.find(node->OpType()) != exclude_ops_set.end()) {
19901990
supported_node = false;
19911991
}
1992-
// Exclude contrib ops
1993-
if (node->Domain() == kMSDomain) {
1994-
supported_node = false;
1995-
}
19961992

19971993
if (supported_node) {
19981994
if (new_subgraph) {

0 commit comments

Comments
 (0)