in mpi_utils.py [0:0]
def mpi_dtype(dtype):
return {
"float32": MPI.FLOAT,
"float64": MPI.DOUBLE,
"int8": MPI.CHAR,
"uint8": MPI.UNSIGNED_CHAR,
"int16": MPI.SHORT,
"uint16": MPI.UNSIGNED_SHORT,
"int32": MPI.INT,
"uint32": MPI.UNSIGNED,
"int64": MPI.LONG,
"uint64": MPI.UNSIGNED_LONG,
}[dtype]