Skip to content

Commit

Permalink
add prints
Browse files Browse the repository at this point in the history
  • Loading branch information
Sebastian-Belkner committed Jul 2, 2023
1 parent d2fce98 commit 83e3c33
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions delensalot/core/mpi.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,9 @@ def enable():


def disable():

global barrier, send, receive, bcast, ANY_SOURCE, name, rank, size, finalize, disabled
print('disabling mpi')
barrier = lambda: -1
send = lambda _, dest: 0
receive = lambda _, source: 0
Expand All @@ -88,7 +90,9 @@ def disable():
log.info('mpi.py : diabled, rank %s in %s' % (rank, size))

def init():

global barrier, send, receive, bcast, ANY_SOURCE, name, rank, size, finalize, disabled
print('enabling mpi')
from mpi4py import MPI
rank = MPI.COMM_WORLD.Get_rank()
size = MPI.COMM_WORLD.Get_size()
Expand Down

0 comments on commit 83e3c33

Please sign in to comment.