def is_mpi_env():
"""
Test if current environment is MPI or not
"""
try:
import mpi4py
except ImportError:
return False
try:
import mpi4py.MPI
except ImportError:
return False
if mpi4py.MPI.COMM_WORLD.size == 1 and mpi4py.MPI.COMM_WORLD.rank == 0:
return False
return True
评论列表
文章目录