33 lines
1.3 KiB
Python
33 lines
1.3 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 reduce() collective.
|
|
|
|
from __future__ import print_function
|
|
import mpi
|
|
from generators import *
|
|
|
|
def reduce_test(comm, generator, kind, op, op_kind, root):
|
|
if comm.rank == root:
|
|
print ("Reducing to %s of %s at root %d..." % (op_kind, kind, root)),
|
|
my_value = generator(comm.rank)
|
|
result = mpi.reduce(comm, my_value, op, root)
|
|
if comm.rank == root:
|
|
expected_result = generator(0);
|
|
for p in range(1, comm.size):
|
|
expected_result = op(expected_result, generator(p))
|
|
assert result == expected_result
|
|
print ("OK.")
|
|
else:
|
|
assert result == None
|
|
return
|
|
|
|
reduce_test(mpi.world, int_generator, "integers", lambda x,y:x + y, "sum", 0)
|
|
reduce_test(mpi.world, int_generator, "integers", lambda x,y:x * y, "product", 1)
|
|
reduce_test(mpi.world, int_generator, "integers", min, "minimum", 0)
|
|
reduce_test(mpi.world, string_generator, "strings", lambda x,y:x + y, "concatenation", 0)
|
|
reduce_test(mpi.world, string_list_generator, "list of strings", lambda x,y:x + y, "concatenation", 0)
|