switch to info to print ipex and torch-ccl version number

Lightning-AI · Feb 23, 2023 · fb64ac4 · fb64ac4
1 parent 37490aa
commit fb64ac4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/lightning/fabric/accelerators/xpu.py b/src/lightning/fabric/accelerators/xpu.py
@@ -15,15 +15,15 @@
 from typing import Dict, List, Optional, Union
 
 import torch
-from lightning_utilities.core.rank_zero import rank_zero_info, rank_zero_warn
+from lightning_utilities.core.rank_zero import rank_zero_info
 
 from lightning.fabric.accelerators.accelerator import Accelerator
 from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_1_12, _TORCH_GREATER_EQUAL_1_13
 
 try:
  import intel_extension_for_pytorch as ipex
 
- rank_zero_warn(f"You are using Intel® Extension for PyTorch* {ipex.__version__}")
+ rank_zero_info(f"Using Intel® Extension for PyTorch* {ipex.__version__}")
 except ImportError:
  pass
 

diff --git a/src/lightning/fabric/utilities/distributed.py b/src/lightning/fabric/utilities/distributed.py
@@ -12,13 +12,13 @@
 
 from lightning.fabric.plugins.environments.cluster_environment import ClusterEnvironment
 from lightning.fabric.utilities.imports import _TORCH_GREATER_EQUAL_1_12
-from lightning.fabric.utilities.rank_zero import rank_zero_info, rank_zero_warn
+from lightning.fabric.utilities.rank_zero import rank_zero_info
 from lightning.fabric.utilities.types import ReduceOp
 
 try:
  import oneccl_bindings_for_pytorch as torch_ccl
 
- rank_zero_warn(f"You are using Intel® oneCCL Bindings for PyTorch* {torch_ccl.__version__}")
+ rank_zero_info(f"Using Intel® oneCCL Bindings for PyTorch* {torch_ccl.__version__}")
 except ImportError:
  pass