From 3f7551e06e78c533a4b61d7bc872785c6f9be05a Mon Sep 17 00:00:00 2001 From: guozhong wang Date: Sat, 13 Jan 2024 00:04:46 +0800 Subject: [PATCH] Add try for get_property (#510) * Add try for get_property * refine code --------- Co-authored-by: fishbell --- optimum/intel/openvino/utils.py | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/optimum/intel/openvino/utils.py b/optimum/intel/openvino/utils.py index 9548c750bf..46442d4d22 100644 --- a/optimum/intel/openvino/utils.py +++ b/optimum/intel/openvino/utils.py @@ -133,12 +133,15 @@ def _print_compiled_model_properties(compiled_model): skip_keys = {"SUPPORTED_METRICS", "SUPPORTED_CONFIG_KEYS", supported_properties} keys = set(compiled_model.get_property(supported_properties)) - skip_keys for k in keys: - value = compiled_model.get_property(k) - if k == properties.device.properties(): - for device_key in value.keys(): - logger.info(f" {device_key}:") - for k2, value2 in value.get(device_key).items(): - if k2 not in skip_keys: - logger.info(f" {k2}: {value2}") - else: - logger.info(f" {k}: {value}") + try: + value = compiled_model.get_property(k) + if k == properties.device.properties(): + for device_key in value.keys(): + logger.info(f" {device_key}:") + for k2, value2 in value.get(device_key).items(): + if k2 not in skip_keys: + logger.info(f" {k2}: {value2}") + else: + logger.info(f" {k}: {value}") + except Exception: + logger.error(f"[error] Get property of '{k}' failed")