python 中的 mpi4py 通信器是什么类型的？

Question

我正在使用 mpi4py 来并行化我的代码，我有这种 function：

def do_something(some_integer: int, comm) -> None:
    ...

其中comm参数适用于 mpi 通信器，如下所示：

from mpi4py import MPI
comm = MPI.COMM_WORLD

我想为我的函数使用显式类型，但我不清楚 comm 参数的类型应该是什么，即comm:之后应该是什么？

Answer 1

因此，该类型的MPI 文档应为：

mpi4py.MPI.Intracomm

所以尝试一下

def do_something(some_integer: int, comm: mpi4py.MPI.Intracomm) -> None: