From 717a9c39776b3d23da897c4c00867b6d360389ba Mon Sep 17 00:00:00 2001 From: ankitm3k Date: Thu, 26 Sep 2024 12:02:56 +0530 Subject: [PATCH] update: Enable Python API for load_config in provider options --- onnxruntime/core/session/provider_bridge_ort.cc | 1 + onnxruntime/python/onnxruntime_pybind_state.cc | 3 +++ 2 files changed, 4 insertions(+) diff --git a/onnxruntime/core/session/provider_bridge_ort.cc b/onnxruntime/core/session/provider_bridge_ort.cc index f0cf20a2b513f..f10af05408717 100644 --- a/onnxruntime/core/session/provider_bridge_ort.cc +++ b/onnxruntime/core/session/provider_bridge_ort.cc @@ -1816,6 +1816,7 @@ ProviderOptions OrtOpenVINOProviderOptionsToOrtOpenVINOProviderOptionsV2(const O // Add new provider option below ov_options_converted_map["num_streams"] = "1"; + ov_options_converted_map["load_config"] = ""; ov_options_converted_map["model_priority"] = "DEFAULT"; ov_options_converted_map["enable_qdq_optimizer"] = "false"; return ov_options_converted_map; diff --git a/onnxruntime/python/onnxruntime_pybind_state.cc b/onnxruntime/python/onnxruntime_pybind_state.cc index e8bf61612c89b..cbb3954b670ea 100644 --- a/onnxruntime/python/onnxruntime_pybind_state.cc +++ b/onnxruntime/python/onnxruntime_pybind_state.cc @@ -1087,6 +1087,9 @@ std::unique_ptr CreateExecutionProviderInstance( } else if (option.first == "num_streams") { OV_provider_options_map[option.first] = option.second; continue; + } else if (option.first == "load_config") { + OV_provider_options_map[option.first] = option.second; + continue; } else if (option.first == "cache_dir") { OV_provider_options_map[option.first] = option.second; continue;