Update onnxruntime/test/perftest/ort_test_session.cc

Co-authored-by: Scott McKay <[email protected]>
microsoft · Nov 22, 2024 · 07f5772 · 07f5772
1 parent c49f250
commit 07f5772
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/onnxruntime/test/perftest/ort_test_session.cc b/onnxruntime/test/perftest/ort_test_session.cc
@@ -359,7 +359,11 @@ select from 'TF8', 'TF16', 'UINT8', 'FLOAT', 'ITENSOR'. \n)");
 #ifdef __APPLE__
 #ifdef USE_COREML
     std::string ov_string = performance_test_config.run_config.ep_runtime_config_string;
-    if (!ParseSessionConfigs(ov_string, provider_options, {kCoremlProviderOption_MLComputeUnits, kCoremlProviderOption_ModelFormat, kCoremlProviderOption_RequireStaticInputShapes, kCoremlProviderOption_EnableOnSubgraphs})) {
+    static const std::unordered_set<std::string> available_keys = {kCoremlProviderOption_MLComputeUnits,
+                                                                   kCoremlProviderOption_ModelFormat,
+                                                                   kCoremlProviderOption_RequireStaticInputShapes,
+                                                                   kCoremlProviderOption_EnableOnSubgraphs};
+    if (!ParseSessionConfigs(ov_string, provider_options, available_keys)) {
       ORT_THROW(
           "[ERROR] Use a '|' to separate the key and value for the "
           "run-time option you are trying to use.\n");