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
|
from array import array
from math import pi as PI
from sys import argv
from mpi4py import MPI
cmd = "./cpi-worker-py.exe"
if len(argv) > 1:
cmd = argv[1]
print(f"{argv[0]} -> {cmd}")
worker = MPI.COMM_SELF.Spawn(cmd, None, 5)
n = array("i", [100])
worker.Bcast([n, MPI.INT], root=MPI.ROOT)
pi = array("d", [0.0])
worker.Reduce(
sendbuf=None,
recvbuf=[pi, MPI.DOUBLE],
op=MPI.SUM,
root=MPI.ROOT,
)
pi = pi[0]
worker.Disconnect()
print(f"pi: {pi:.16f}, error: {abs(PI - pi):.16f}")
|
