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

Question

我正在使用 mpi4py 来并行化我的代码，我有这样的功能：

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

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

from mpi4py import MPI
comm = MPI.COMM_WORLD

我想为我的函数使用显式类型，但我不清楚 comm 参数的类型应该是什么，即 [=14= 之后应该是什么类型] ?

Answer 1

因此 MPI documentation 类型应该是：

mpi4py.MPI.Intracomm

所以试试

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

What type for a mpi4py communicator in python?