Skip to content

Commit 5e54793

Browse files
Conditionally include NV TRT RTX EP in list of known provider types when ORT is built with the provider-bridge version of the EP
1 parent 5c91f19 commit 5e54793

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

onnxruntime/test/unittest_util/base_tester.cc

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -659,7 +659,12 @@ void BaseTester::RunWithConfig(size_t* number_of_pre_packed_weights_counter,
659659
#endif
660660
kDnnlExecutionProvider,
661661
kTensorrtExecutionProvider,
662+
#ifdef USE_NV
663+
// Only include NV TRT RTX EP when is ORT is built with the provider-bridge
664+
// version of the EP (i.e., USE_NV is defined). This allows use of the plugin EP version of the EP
665+
// when ORT is not built any provider-bridge EPs.
662666
kNvTensorRTRTXExecutionProvider,
667+
#endif
663668
kOpenVINOExecutionProvider,
664669
kDmlExecutionProvider,
665670
kAclExecutionProvider,
@@ -835,7 +840,10 @@ void BaseTester::ExecuteModelForEps(
835840

836841
constexpr std::array kEpsThatCompileNodes{
837842
kOpenVINOExecutionProvider,
838-
kTensorrtExecutionProvider, // uses kernel registry for Memcpy* nodes, but compiles all others.
843+
kTensorrtExecutionProvider, // uses kernel registry for Memcpy* nodes only
844+
#ifdef USE_NV
845+
kNvTensorRTRTXExecutionProvider, // uses kernel registry for Memcpy* nodes only
846+
#endif
839847
kNnapiExecutionProvider,
840848
kVSINPUExecutionProvider,
841849
kCoreMLExecutionProvider,

0 commit comments

Comments
 (0)