Browse Source

refactor(mgb/opr-mm): set False as default value of local_grad

GitOrigin-RevId: 2f9603b087
tags/v1.0.0-rc1
Megvii Engine Team Xinran Xu 4 years ago
parent
commit
f7b5eced23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/opr-mm/impl/collective_comm.oprdecl

+ 1
- 1
src/opr-mm/impl/collective_comm.oprdecl View File

@@ -8,7 +8,6 @@ decl_raw_opr(
'operation to which this operator belongs.', 'int'),
Doc('is_root', 'whether this node is root node', 'bool'),
Doc('rank', 'rank of this node, if is -1, generate one', 'int'),
Doc('local_grad', 'whether use local grad', 'bool'),
Doc('server_addr', 'rpc server ip address'),
Doc('port', 'server rpc listening port'),
Doc('param', 'The only component of *param* is *mode*, which refers to '
@@ -20,6 +19,7 @@ decl_raw_opr(
'any).', ':class:`~megbrain.opr_param_defs.DType`', 'None'),
Doc('backend', 'Backend for collective communication, nccl or ucx',
'str', '\'nccl\''),
Doc('local_grad', 'whether use local grad', 'bool', 'False'),
Doc('output_buffer', 'The external dev buffer reserving output result',
':class:`.SharedND`', 'None'),
Doc('disable', 'If true, the execution will return directly and the output '


Loading…
Cancel
Save