lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix doc

This commit is contained in:
guoye 2020-06-05 16:33:49 +08:00
parent 6713e6bb89
commit 3f0545b98c
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
doc/source/index.rst
View File

@ -24,7 +24,7 @@
jittor.contrib jittor.contrib
jittor.dataset jittor.dataset
jittor.transform jittor.transform
jittor.compile_extern.mpi jittor.mpi
.. toctree:: .. toctree::

2
doc/source/jittor.compile_extern.mpi.md → doc/source/jittor.mpi.md
View File

@ -1,4 +1,4 @@
jittor.compile_extern.mpi jittor.mpi
===================== =====================
这里是Jittor的MPI模块的API文档，您可以通过`from jittor.compile_extern import mpi`来获取该模块。 这里是Jittor的MPI模块的API文档，您可以通过`from jittor.compile_extern import mpi`来获取该模块。

1
extern/mpi/inc/mpi_warper.h vendored
View File

@ -53,6 +53,7 @@ struct ArrayArgs;
Use jt.Module.mpi_param_broadcast(root=0) to broadcast all moudule parameters of this module in [root] MPI node to all MPI nodes. Use jt.Module.mpi_param_broadcast(root=0) to broadcast all moudule parameters of this module in [root] MPI node to all MPI nodes.
This operation has no gradient, and the input parameter type is numpy array.
*/ */
// @pyjt(broadcast) // @pyjt(broadcast)
void _mpi_broadcast(ArrayArgs&& args, int root); void _mpi_broadcast(ArrayArgs&& args, int root);