1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
from mpi4py import rc
rc.initialize = False
from mpi4py import MPI
assert not MPI.Is_initialized()
assert not MPI.Is_finalized()
MPI.Init_thread()
assert MPI.Is_initialized()
assert not MPI.Is_finalized()
import sys
name, _ = MPI.get_vendor()
if name == 'MPICH':
assert MPI.Query_thread() == MPI.THREAD_MULTIPLE
if name == 'MPICH2' and sys.platform[:3] != 'win':
assert MPI.Query_thread() == MPI.THREAD_MULTIPLE
MPI.Finalize()
assert MPI.Is_initialized()
assert MPI.Is_finalized()
|
