44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
# Copyright (C) 2006 Douglas Gregor <doug.gregor -at- gmail.com>.
|
|
|
|
# Use, modification and distribution is subject to the Boost Software
|
|
# License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
|
|
# http://www.boost.org/LICENSE_1_0.txt)
|
|
|
|
# Test basic communication.
|
|
|
|
from __future__ import print_function
|
|
import mpi
|
|
|
|
def ring_test(comm, value, kind, root):
|
|
next_peer = (comm.rank + 1) % comm.size;
|
|
prior_peer = (comm.rank + comm.size - 1) % comm.size;
|
|
|
|
if comm.rank == root:
|
|
print ("Passing %s around a ring from root %d..." % (kind, root)),
|
|
comm.send(next_peer, 0, value)
|
|
(other_value, stat) = comm.recv(return_status = True)
|
|
assert value == other_value
|
|
assert stat.source == prior_peer
|
|
assert stat.tag == 0
|
|
else:
|
|
msg = comm.probe()
|
|
other_value = comm.recv(msg.source, msg.tag)
|
|
assert value == other_value
|
|
comm.send(next_peer, 0, other_value)
|
|
|
|
comm.barrier()
|
|
if comm.rank == root:
|
|
print ("OK")
|
|
pass
|
|
|
|
if mpi.world.size < 2:
|
|
print ("ERROR: ring_test.py must be executed with more than one process")
|
|
mpi.world.abort(-1);
|
|
|
|
ring_test(mpi.world, 17, 'integers', 0)
|
|
ring_test(mpi.world, 17, 'integers', 1)
|
|
ring_test(mpi.world, 'Hello, World!', 'string', 0)
|
|
ring_test(mpi.world, 'Hello, World!', 'string', 1)
|
|
ring_test(mpi.world, ['Hello', 'MPI', 'Python', 'World'], 'list of strings', 0)
|
|
ring_test(mpi.world, ['Hello', 'MPI', 'Python', 'World'], 'list of strings', 1)
|