Skip to content

Commit 7f3abd0

Browse files
nieubankCopilot
andcommitted
Fix TensorRT RTX provider name matching
Use the real Windows TensorRT RTX execution provider name in shared helpers and the canonical ONNX Runtime GenAI provider name in GenAI helpers. ADO: 61791041 Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
1 parent 6e81dc6 commit 7f3abd0

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

Samples/WindowsML/Shared/cpp/PerformanceConfigurator.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ namespace Shared
3434
{
3535
return GetMiGraphXOptions(mode);
3636
}
37-
if (normalized == "TENSORRTRTXEXECUTIONPROVIDER")
37+
if (normalized == "NVTENSORRTRTXEXECUTIONPROVIDER")
3838
{
3939
return GetTensorRtRtxOptions(mode);
4040
}

Samples/WindowsML/Shared/cs/PerformanceConfigurator.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ public static Dictionary<string, string> GetEpOptions(string epName, Performance
1717
"QNNEXECUTIONPROVIDER" => GetQnnOptions(mode),
1818
"VITISAIEXECUTIONPROVIDER" => GetVitisAiOptions(mode),
1919
"MIGRAPHXEXECUTIONPROVIDER" => GetMiGraphxOptions(mode),
20-
"TENSORRTRTXEXECUTIONPROVIDER" => GetTensorRtRtxOptions(mode),
20+
"NVTENSORRTRTXEXECUTIONPROVIDER" or "NVTENSORRTRTX" => GetTensorRtRtxOptions(mode),
2121
_ => new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase)
2222
};
2323
}

Samples/WindowsML/cpp/CppConsoleDesktop.GenAI/GenAIPerformanceConfigurator.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ class GenAIPerformanceConfigurator
4343
{
4444
return GetMiGraphXOptions(mode);
4545
}
46-
if (normalized == "TENSORRTRTXEXECUTIONPROVIDER")
46+
if (normalized == "NVTENSORRTRTX")
4747
{
4848
return GetTensorRtRtxOptions(mode);
4949
}

0 commit comments

Comments
 (0)