这里是Jittor的MPI模块的API文档,您可以通过from jittor.compile_extern import mpi来获取该模块。
from jittor.compile_extern import mpi
.. automodule:: jittor_mpi_core :members: :undoc-members: .. automodule:: jittor_mpi_core.ops :members: :undoc-members: