Skip to content

Commit b45edfb

Browse files
authored
[NvTensorRTRTX EP]Disable Fast GELU operator in base model used for NV EP Unit Tests (microsoft#25323)
### Description Remove fast_gelu operator from the base model created in NV TRT RTX EP unit tests. ### Motivation and Context The operator was added in the model to partition the model into subgraphs which can be assigned to NV TRT RTX EP and CUDA EP, which supports fast_gelu. But CUDA EP is not built when building ORT with NV TRT RTX EP hence the unit tests fail with unsupported op error. @ishwar-raut1 @ankan-ban
1 parent 033ca86 commit b45edfb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxruntime/test/providers/nv_tensorrt_rtx/nv_basic_test.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ TEST(NvExecutionProviderTest, ContextEmbedAndReloadDataDynamic) {
322322
std::string graph_name = "test";
323323
std::vector<int> dims = {1, -1, -1};
324324

325-
CreateBaseModel(model_name, graph_name, dims, true);
325+
CreateBaseModel(model_name, graph_name, dims);
326326

327327
auto env = Ort::Env();
328328
auto logging_level = OrtLoggingLevel::ORT_LOGGING_LEVEL_WARNING;
@@ -371,7 +371,7 @@ TYPED_TEST(NvExecutionProviderTest, IOTypeTests) {
371371
std::string graph_name = "test" + dtype_name;
372372
std::vector<int> dims = {1, -1, -1};
373373

374-
CreateBaseModel(model_name, graph_name, dims, true);
374+
CreateBaseModel(model_name, graph_name, dims);
375375

376376
auto env = Ort::Env();
377377
auto logging_level = OrtLoggingLevel::ORT_LOGGING_LEVEL_WARNING;

0 commit comments

Comments
 (0)