Skip to content

Commit d83c736

Browse files
mc-nvkrishung5
authored andcommitted
Change default flag value for OpenVINO (#252)
* Change default flag value for OpenVINO * Change default flag value for OpenVINO
1 parent 33aa258 commit d83c736

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/onnxruntime.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -783,7 +783,7 @@ ModelState::LoadModel(
783783
need_lock = true;
784784
OrtOpenVINOProviderOptions openvino_options;
785785
openvino_options.device_type =
786-
"CPU_FP32"; // device_type default is CPU_FP32
786+
"CPU"; // device_type default is CPU
787787

788788
RETURN_IF_ORT_ERROR(
789789
ort_api->SessionOptionsAppendExecutionProvider_OpenVINO(

tools/gen_ort_dockerfile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def dockerfile_for_linux(output_file):
248248
ep_flags += " --allow_running_as_root"
249249

250250
if FLAGS.ort_openvino is not None:
251-
ep_flags += " --use_openvino CPU_FP32"
251+
ep_flags += " --use_openvino CPU"
252252

253253
if target_platform() == "igpu":
254254
ep_flags += (
@@ -442,7 +442,7 @@ def dockerfile_for_windows(output_file):
442442
if FLAGS.tensorrt_home is not None:
443443
ep_flags += ' --tensorrt_home "{}"'.format(FLAGS.tensorrt_home)
444444
if FLAGS.ort_openvino is not None:
445-
ep_flags += " --use_openvino CPU_FP32"
445+
ep_flags += " --use_openvino CPU"
446446

447447
df += """
448448
WORKDIR /workspace/onnxruntime

0 commit comments

Comments
 (0)