From 770275a755ea0445675720a3f6f14e77c491fceb Mon Sep 17 00:00:00 2001 From: zhuhaozhe Date: Thu, 14 Mar 2024 21:02:38 +0800 Subject: [PATCH] use logging.INFO as logging level for ipex launcher (#2675) --- intel_extension_for_pytorch/cpu/launch/cpu_info.py | 5 ++++- intel_extension_for_pytorch/cpu/launch/launch.py | 4 +++- intel_extension_for_pytorch/cpu/launch/launcher_base.py | 5 ++++- 3 files changed, 11 insertions(+), 3 deletions(-) diff --git a/intel_extension_for_pytorch/cpu/launch/cpu_info.py b/intel_extension_for_pytorch/cpu/launch/cpu_info.py index b90011bee..162096710 100644 --- a/intel_extension_for_pytorch/cpu/launch/cpu_info.py +++ b/intel_extension_for_pytorch/cpu/launch/cpu_info.py @@ -216,7 +216,10 @@ def verbose(self, level, msg, warning_type=None): assert ( level in logging_fn.keys() ), f"Unrecognized logging level {level} is detected. Available levels are {logging_fn.keys()}." - logging_fn[level](msg, _type=warning_type) + if warning_type: + logging_fn[level](msg, _type=warning_type) + else: + logging_fn[level](msg) else: print(msg) diff --git a/intel_extension_for_pytorch/cpu/launch/launch.py b/intel_extension_for_pytorch/cpu/launch/launch.py index c23a5264b..ce6c17ae3 100644 --- a/intel_extension_for_pytorch/cpu/launch/launch.py +++ b/intel_extension_for_pytorch/cpu/launch/launch.py @@ -8,8 +8,10 @@ import intel_extension_for_pytorch.cpu.auto_ipex as auto_ipex from .launcher_distributed import DistributedTrainingLauncher from .launcher_multi_instances import MultiInstancesLauncher -from ...utils._logger import logger, WarningType, format_str +from ...utils._logger import WarningType, format_str +logger = logging.getLogger("IPEX-launcher") +logger.setLevel(logging.INFO) """ This is a script for launching PyTorch training and inference on Intel Xeon CPU with optimal configurations. Now, single instance inference/training, multi-instance inference/training and distributed training diff --git a/intel_extension_for_pytorch/cpu/launch/launcher_base.py b/intel_extension_for_pytorch/cpu/launch/launcher_base.py index 0b06cafef..baaa4d1f7 100644 --- a/intel_extension_for_pytorch/cpu/launch/launcher_base.py +++ b/intel_extension_for_pytorch/cpu/launch/launcher_base.py @@ -85,7 +85,10 @@ def verbose(self, level, msg, warning_type=None): assert ( level in logging_fn.keys() ), f"Unrecognized logging level {level} is detected. Available levels are {logging_fn.keys()}." - logging_fn[level](msg, _type=warning_type) + if warning_type: + logging_fn[level](msg, _type=warning_type) + else: + logging_fn[level](msg) else: print(msg)