提交 f7b5eced 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

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

GitOrigin-RevId: 2f9603b08719fbe2653b17231ebc74d4ee06c6cf
上级 7a8183f4
......@@ -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 '
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册