Skip to content

Commit

Permalink
use logging.INFO as logging level for ipex launcher (#2675)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhuhaozhe committed Mar 14, 2024
1 parent 594f859 commit 770275a
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 3 deletions.
5 changes: 4 additions & 1 deletion intel_extension_for_pytorch/cpu/launch/cpu_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down
4 changes: 3 additions & 1 deletion intel_extension_for_pytorch/cpu/launch/launch.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
5 changes: 4 additions & 1 deletion intel_extension_for_pytorch/cpu/launch/launcher_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit 770275a

Please sign in to comment.