[BugFix] Use consistent logger everywhere (#3738)

vllm/model_executor/parallel_utils/pynccl.py

@@ -21,7 +21,6 @@
 
 import ctypes
 import datetime
-import logging
 import os
 
 # ===================== import region =====================
@@ -29,7 +28,9 @@ import torch
 import torch.distributed as dist
 from torch.distributed import ReduceOp
 
-logger = logging.getLogger(__name__)
+from vllm.logger import init_logger
+
+logger = init_logger(__name__)
 
 so_file = os.environ.get("VLLM_NCCL_SO_PATH", "")
 

vllm/model_executor/parallel_utils/pynccl_utils.py

@@ -1,11 +1,12 @@
 import contextlib
-import logging
 from typing import Optional
 
 import torch
 from torch.distributed import ReduceOp
 
-logger = logging.getLogger(__name__)
+from vllm.logger import init_logger
+
+logger = init_logger(__name__)
 
 try:
     from vllm.model_executor.parallel_utils.pynccl import (NCCLCommunicator,