c = ga.get(g_c)
v = np.dot(a,b)
val = int(np.abs(np.sum(c-v))>0.0001)
val = ga.gop_add(val)
return val == 0
if __name__ == '__main__':
if nproc > MULTIPLIER**3:
if 0 == me:
print "You must use less than %s processors" % (MULTIPLIER**3+1)
else:
g_a = ga.create(ga.C_DBL, [N,N])
g_b = ga.create(ga.C_DBL, [N,N])
g_c = ga.create(ga.C_DBL, [N,N])
g_counter = ga.create(ga.C_INT, [1])
ga.zero(g_counter)
# put some fake data into input arrays A and B
if me == 0:
ga.put(g_a, np.random.random(N*N))
ga.put(g_b, np.random.random(N*N))
ga.sync()
if me == 0:
print "srumma...",
srumma(g_a, g_b, g_c, CHUNK_SIZE, MULTIPLIER, g_counter)
if me == 0:
print "done"
if me == 0:
print "verifying using ga.gemm...",
ok = verify_using_ga(g_a, g_b, g_c)
if me == 0:
if ok:
nprocs = ga.nnodes()
myrank = ga.nodeid()
g_pi = ga.create(ga.C_DBL, [1])
one_time = False
if len(sys.argv) == 2:
n = int(sys.argv[1])
one_time = True
while True:
if not one_time:
if myrank == 0:
n = get_n()
n = ga.brdcst(n)
else:
n = ga.brdcst(0)
if n == 0:
break
ga.zero(g_pi)
mypi = comp_pi(n, myrank, nprocs)
ga.acc(g_pi, mypi)
ga.sync()
if myrank == 0:
pi = ga.get(g_pi)[0]
prn_pi(pi, PI)
if one_time:
break
ga.destroy(g_pi)
