File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 23
23
"""
24
24
import contextlib
25
25
import gc
26
- import importlib
26
+ import importlib . util
27
27
import pickle
28
28
import weakref
29
29
from collections import namedtuple
@@ -925,7 +925,7 @@ def pplx_init(rank, world_size):
925
925
nvshmem_get_unique_id , nvshmem_init )
926
926
try :
927
927
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 )
929
929
uid = nvshmem_get_unique_id (
930
930
) if rank == 0 else nvshmem_alloc_empty_unique_id ()
931
931
uid_gpu = uid .cuda ()
@@ -943,7 +943,7 @@ def pplx_finalize():
943
943
global PPLX_DID_INIT
944
944
if PPLX_DID_INIT :
945
945
from pplx_kernels .nvshmem import nvshmem_finalize
946
- logger .debug ("PPLX finalize" )
946
+ logger .info ("PPLX finalize" )
947
947
nvshmem_finalize ()
948
948
949
949
You can’t perform that action at this time.
0 commit comments