Skip to content

Commit 44d74cc

Browse files
RyanMetcalfeInt8ankitm3k
authored andcommitted
openvino_provider_factory: Add 'num_of_threads' to block_and_advise_entries
1 parent d006a5d commit 44d74cc

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

onnxruntime/core/providers/openvino/openvino_provider_factory.cc

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -454,14 +454,15 @@ struct OpenVINO_Provider : Provider {
454454
std::unique_ptr<IExecutionProvider>& ep) override {
455455
// Check if no devices are provided
456456
if (num_devices == 0) {
457-
return Status(common::ONNXRUNTIME, ORT_EP_FAIL, "No devices provided to CreateEp");
457+
return Status(common::ONNXRUNTIME, ORT_EP_FAIL, "No devices provided to CreateIExecutionProvider");
458458
}
459459

460-
// For provider options that we don't support anymore, give some guidance & examples
461-
// about how to make use of the option through load_config.
460+
// For provider options that we don't support directly but are still supported through load_config,
461+
// give some specific guidance & example about how to make use of the option through load_config.
462462
const std::vector<std::pair<std::string, std::string>> block_and_advise_entries = {
463463
{"cache_dir", "\"CACHE_DIR\": \"<filesystem_path>\""},
464464
{"precision", "\"INFERENCE_PRECISION_HINT\": \"F32\""},
465+
{"num_of_threads", "\"INFERENCE_NUM_THREADS\": \"1\""},
465466
{"num_streams", "\"NUM_STREAMS\": \"1\""},
466467
{"model_priority", "\"MODEL_PRIORITY\": \"LOW\""},
467468
{"enable_opencl_throttling", "\"GPU\": {\"PLUGIN_THROTTLE\": \"1\"}"},
@@ -480,7 +481,7 @@ struct OpenVINO_Provider : Provider {
480481

481482
// For the rest of the disallowed provider options, give a generic error message.
482483
const std::vector<std::string> blocked_provider_keys = {
483-
"device_type", "device_id", "device_luid", "context", "num_of_threads", "disable_dynamic_shapes"};
484+
"device_type", "device_id", "device_luid", "context", "disable_dynamic_shapes"};
484485

485486
for (const auto& key : blocked_provider_keys) {
486487
if (provider_options.find(key) != provider_options.end()) {

0 commit comments

Comments
 (0)