我正在使用 mpi4py 并行化我的代码，我有这样的功能:
def do_something(some_integer: int, comm) -> None:
...
其中 comm 参数用于 mpi 通信器，如下所示:
from mpi4py import MPI
comm = MPI.COMM_WORLD
我想对我的函数使用显式类型，但我不清楚 comm 参数的类型应该是什么，即 comm: 之后应该是什么？
最佳答案
因此MPI documentation该类型应该是:
mpi4py.MPI.Intracomm
所以尝试一下
def do_something(some_integer: int, comm: mpi4py.MPI.Intracomm) -> None:
关于python - python 中的 mpi4py 通信器是什么类型？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/66748172/