Skip to content

Commit f74ab61

Browse files
committed
fix parallel_state lint
Signed-off-by: Bill Nell <[email protected]>
1 parent c0fc027 commit f74ab61

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/distributed/parallel_state.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
"""
2424
import contextlib
2525
import gc
26-
import importlib
26+
import importlib.util
2727
import pickle
2828
import weakref
2929
from collections import namedtuple
@@ -925,7 +925,7 @@ def pplx_init(rank, world_size):
925925
nvshmem_get_unique_id, nvshmem_init)
926926
try:
927927
global PPLX_DID_INIT
928-
logger.debug("PPLX_INIT %s %d", rank, world_size)
928+
logger.info("PPLX_INIT rank=%d world=%d", rank, world_size)
929929
uid = nvshmem_get_unique_id(
930930
) if rank == 0 else nvshmem_alloc_empty_unique_id()
931931
uid_gpu = uid.cuda()
@@ -943,7 +943,7 @@ def pplx_finalize():
943943
global PPLX_DID_INIT
944944
if PPLX_DID_INIT:
945945
from pplx_kernels.nvshmem import nvshmem_finalize
946-
logger.debug("PPLX finalize")
946+
logger.info("PPLX finalize")
947947
nvshmem_finalize()
948948

949949

0 commit comments

Comments
 (0)