Skip to content

Commit 6c3ae0b

Browse files
tanmayv25mc-nv
andauthored
Change default flag value for OpenVINO (#252) (#255)
* Change default flag value for OpenVINO * Change default flag value for OpenVINO Co-authored-by: Misha Chornyi <99709299+mc-nv@users.noreply.github.com>
1 parent 48cc4f1 commit 6c3ae0b

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
@@ -224,7 +224,7 @@ def dockerfile_for_linux(output_file):
224224
ep_flags += " --allow_running_as_root"
225225

226226
if FLAGS.ort_openvino is not None:
227-
ep_flags += " --use_openvino CPU_FP32"
227+
ep_flags += " --use_openvino CPU"
228228

229229
if target_platform() == "igpu":
230230
ep_flags += (
@@ -418,7 +418,7 @@ def dockerfile_for_windows(output_file):
418418
if FLAGS.tensorrt_home is not None:
419419
ep_flags += ' --tensorrt_home "{}"'.format(FLAGS.tensorrt_home)
420420
if FLAGS.ort_openvino is not None:
421-
ep_flags += " --use_openvino CPU_FP32"
421+
ep_flags += " --use_openvino CPU"
422422

423423
df += """
424424
WORKDIR /workspace/onnxruntime

0 commit comments

Comments
 (0)