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
|
# http://mvapich.cse.ohio-state.edu/benchmarks/
from mpi4py import MPI
def osu_barrier(
BENCHMARH="MPI Barrier Latency Test",
skip=1000,
loop=10000,
skip_large=10,
loop_large=100,
large_message_size=8192,
MAX_MSG_SIZE=1 << 20,
):
comm = MPI.COMM_WORLD
myid = comm.Get_rank()
numprocs = comm.Get_size()
if numprocs < 2:
if myid == 0:
errmsg = "This test requires at least two processes"
else:
errmsg = None
raise SystemExit(errmsg)
if myid == 0:
print(f"# {BENCHMARH}")
if myid == 0:
print(f"# {'Size [B]':<8s}{'Latency [us]':>20s}")
size = 0
_ = large_message_size # unused
_ = MAX_MSG_SIZE # unused
skip = skip_large
loop = loop_large
iterations = list(range(loop + skip))
#
comm.Barrier()
for i in iterations:
if i == skip:
t_start = MPI.Wtime()
comm.Barrier()
t_end = MPI.Wtime()
comm.Barrier()
#
if myid == 0:
latency = (t_end - t_start) * 1e6 / loop
print(f"{size:-10d}{latency:20.2f}")
if __name__ == "__main__":
osu_barrier()
|
