From ee12b800075c5b65884d94f2027c6a40972512ed Mon Sep 17 00:00:00 2001 From: lzhengning Date: Thu, 23 Jun 2022 11:53:21 +0800 Subject: [PATCH] polish mpi document --- doc/source/jittor.mpi.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doc/source/jittor.mpi.md b/doc/source/jittor.mpi.md index 4274c1f7..d0892cf9 100644 --- a/doc/source/jittor.mpi.md +++ b/doc/source/jittor.mpi.md @@ -134,8 +134,8 @@ def val(epoch): 目前MPI开放接口如下： * `jt.in_mpi`: 当计图不在MPI环境下时，`jt.mpi == False`， 用户可以用这个判断是否在mpi环境下。 -* `jt.world_rank`: 获取当前进程总数量，如果没有用mpi，则为1。 -* `jt.rank`: 获取当前进程的编号，区间为`0 ～ jt.world_rank-1`， 如果没有用mpi，则为0。 +* `jt.world_size`: 获取当前进程总数量，如果没有用mpi，则为1。 +* `jt.rank`: 获取当前进程的编号，区间为`0 ～ jt.world_size-1`， 如果没有用mpi，则为0。 * `jt.mpi`: 计图的MPI模块。 * `jt.Module.mpi_param_broadcast(root=0)`: 将模块的参数从root节点广播给其他节点。 * `jt.mpi.mpi_reduce(x, op='add', root=0)`: 将所有节点的变量x使用算子op，reduce到root节点。如果op是'add'或者'sum'，该接口会把所有变量求和，如果op是'mean'，该接口会取均值。