1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
#!/usr/bin/env 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 broadcast() collective.
from __future__ import print_function
import mpi
def broadcast_test(comm, value, kind, root):
if comm.rank == 0:
print ("Broadcasting %s from root %d..." % (kind, root)),
got_value = None
got_value = mpi.broadcast(comm, value, root)
if comm.rank == 0:
print ("OK.")
return
broadcast_test(mpi.world, 17, 'integer', 0)
broadcast_test(mpi.world, 'Hello, World!', 'string', 0)
broadcast_test(mpi.world, ['Hello', 'MPI', 'Python', 'World'],
'list of strings', 0)
if mpi.world.size > 1:
broadcast_test(mpi.world, 17, 'integer', 1)
broadcast_test(mpi.world, 'Hello, World!', 'string', 1)
broadcast_test(mpi.world, ['Hello', 'MPI', 'Python', 'World'],
'list of strings', 1)
|
