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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
from neuron import h
pc = h.ParallelContext()
# each subworld has 3 ranks if nhost_world is multiple of subsize
# (if not, last will have nworld%subsize ranks)
subsize = 3
pc.subworlds(subsize)
ibbs = pc.id_world() // subsize
nbbs = pc.nhost_world() // subsize
x = 0 if pc.nhost_world() % subsize == 0 else 1
if pc.id_world() == 0:
print("id_world nhost_world id_bbs nhost_bbs ibbs nbbs id nhost")
pc.barrier()
print(
"%5d %9d %8d %8d %7d %3d %5d %4d"
% (
pc.id_world(),
pc.nhost_world(),
pc.id_bbs(),
pc.nhost_bbs(),
ibbs,
nbbs + x,
pc.id(),
pc.nhost(),
)
)
assert pc.nhost() == (subsize if ibbs < nbbs else (pc.nhost_world() - subsize * nbbs))
assert pc.id() == pc.id_world() % subsize
# id_bbs and nhost_bbs for non-zero id not -1.
assert pc.nhost_bbs() == ((nbbs + x) if pc.id() == 0 else -1)
assert pc.id_bbs() == (ibbs if pc.id() == 0 else -1)
def f(arg):
print("f(%s) id_world=%d id=%d" % (str(arg), pc.id_world(), pc.id()))
return (arg, pc.id_world(), pc.id())
x = 0
def gcntxt(arg):
global x
x = int(arg)
# print("gcntxt(%s) id_world=%d" % (str(int(arg)), pc.id_world()))
def gtest(arg):
if x != int(arg):
h.execerror("gtest %d != %d" % (x, int(arg)))
pass
# print("gtest x=%d id_world=%d" % (x, pc.id_world()))
pc.runworker()
for i in range(5):
pc.submit(f, "submit %d" % i)
hs = h.ref("")
while pc.working():
pc.upkstr(hs)
print("working returned %s (submit passed %s)" % (str(pc.pyret()), hs[0]))
pc.context(gcntxt, 6)
for i in range(3 * pc.nhost_world()):
pc.submit(gtest, 6)
x = 6
while pc.working():
pass
pc.done()
h.quit()
|
