Skip to content

Commit 770275a

Browse files
authored
use logging.INFO as logging level for ipex launcher (#2675)
1 parent 594f859 commit 770275a

File tree

3 files changed

+11
-3
lines changed

3 files changed

+11
-3
lines changed

intel_extension_for_pytorch/cpu/launch/cpu_info.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,10 @@ def verbose(self, level, msg, warning_type=None):
216216
assert (
217217
level in logging_fn.keys()
218218
), f"Unrecognized logging level {level} is detected. Available levels are {logging_fn.keys()}."
219-
logging_fn[level](msg, _type=warning_type)
219+
if warning_type:
220+
logging_fn[level](msg, _type=warning_type)
221+
else:
222+
logging_fn[level](msg)
220223
else:
221224
print(msg)
222225

intel_extension_for_pytorch/cpu/launch/launch.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@
88
import intel_extension_for_pytorch.cpu.auto_ipex as auto_ipex
99
from .launcher_distributed import DistributedTrainingLauncher
1010
from .launcher_multi_instances import MultiInstancesLauncher
11-
from ...utils._logger import logger, WarningType, format_str
11+
from ...utils._logger import WarningType, format_str
1212

13+
logger = logging.getLogger("IPEX-launcher")
14+
logger.setLevel(logging.INFO)
1315
"""
1416
This is a script for launching PyTorch training and inference on Intel Xeon CPU with optimal configurations.
1517
Now, single instance inference/training, multi-instance inference/training and distributed training

intel_extension_for_pytorch/cpu/launch/launcher_base.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,10 @@ def verbose(self, level, msg, warning_type=None):
8585
assert (
8686
level in logging_fn.keys()
8787
), f"Unrecognized logging level {level} is detected. Available levels are {logging_fn.keys()}."
88-
logging_fn[level](msg, _type=warning_type)
88+
if warning_type:
89+
logging_fn[level](msg, _type=warning_type)
90+
else:
91+
logging_fn[level](msg)
8992
else:
9093
print(msg)
9194

0 commit comments

Comments
 (0)