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 2323"""
2424import contextlib
2525import gc
26- import importlib
26+ import importlib . util
2727import pickle
2828import weakref
2929from 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
You can’t perform that action at this time.
0 commit comments