refine log (#707)

Co-authored-by: baishihao <baishihao@sensetime.com>

Author: shihaobai

lightllm/common/vllm_kernel/_custom_ops.py

@@ -16,6 +16,8 @@
         logger.error("vllm or lightllm_kernel is not installed, you can't use custom ops")
 
 try:
+    import os
+
     ENABLE_CUSTOM_GATHER = os.getenv("ENABLE_CUSTOM_GATHER", "False").upper() in ["ON", "TRUE", "1"]
     if ENABLE_CUSTOM_GATHER:
         from lightllm.common.vllm_kernel._ops import init_custom_gather_ar

lightllm/distributed/communication_op.py

@@ -36,17 +36,13 @@
 try:
     HAS_VLLM = True
     from .custom_all_reduce import CustomAllreduce
-
-    logger.info("using custom allreduce")
 except:
     HAS_VLLM = False
     logger.info("vllm or lightllm_kernel is not installed, you can't use custom allreduce")
 
 try:
     HAS_LIGHTLLM_KERNEL = True
     from .custom_all_gather import CustomAllgather
-
-    logger.info("using custom allgather")
 except:
     HAS_LIGHTLLM_KERNEL = False
     logger.info("lightllm_kernel is not installed, you can't use custom allgather")