Fix for hetero multi python binding with new shared library (microsof…

…t#5895) Co-authored-by: sfatimar <sahar.fatima@intel/com>
jjsbear · Nov 23, 2020 · 9164101 · 9164101
1 parent 58ea7b3
commit 9164101
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onnxruntime/python/onnxruntime_pybind_state.cc b/onnxruntime/python/onnxruntime_pybind_state.cc
@@ -663,8 +663,10 @@ static void RegisterExecutionProviders(InferenceSession* sess, const std::vector
       auto it = provider_options_map.find(type);
       if (it != provider_options_map.end()) {
         for (auto option : it->second) {
-          if (option.first == "device_type")
+          if (option.first == "device_type") {
             openvino_device_type = option.second;
+            params.device_type = openvino_device_type.c_str();
+          }  
           else if (option.first == "enable_vpu_fast_compile") {
             if (option.second == "True") {
               params.enable_vpu_fast_compile = true;