From 8307019380b63b54d3653e19439c2867fbd681ef Mon Sep 17 00:00:00 2001 From: Haiqi Pan Date: Wed, 1 Mar 2023 05:31:04 -0800 Subject: [PATCH] fix benchmark_app python to support YES and NO values for -pin parameter (#15963) * support YES and NO for -pin * add if property_name == 'AFFINITY' --- tools/benchmark_tool/openvino/tools/benchmark/main.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tools/benchmark_tool/openvino/tools/benchmark/main.py b/tools/benchmark_tool/openvino/tools/benchmark/main.py index 6919082c023..6eed765c57a 100644 --- a/tools/benchmark_tool/openvino/tools/benchmark/main.py +++ b/tools/benchmark_tool/openvino/tools/benchmark/main.py @@ -276,6 +276,11 @@ def main(): return def set_nthreads_pin(property_name, property_value): + if property_name == "AFFINITY": + if property_value == "YES": + property_value = "CORE" + elif property_value == "NO": + property_value = "NONE" if property_name in supported_properties or device_name == AUTO_DEVICE_NAME: # create nthreads/pin primary property for HW device or AUTO if -d is AUTO directly. config[device][property_name] = property_value