1
import mpi4py.MPI # initialize Message Passing Interface
2
import ga # initialize Global Arrays
9
def print_distribution(g_a):
10
for i in range(ga.nnodes()):
11
lo,hi = ga.distribution(g_a, i)
12
print "P=%s lo=%s hi=%s" % (i,lo,hi)
14
# create some irregular arrays
17
if nproc < np.prod(block):
18
raise ValueError, "ERROR: fewer procs than requested blocks"
19
g_a = ga.create_irreg(ga.C_DBL, [8,10], block, map, "Array A")
21
ga.error("Could not create global array A",g_a)
22
g_b = ga.create(ga.C_INT, (2,3,4,5,6))
25
print_distribution(g_a)
26
print_distribution(g_b)