From f7b5eced23ac12342f9841cf61f72d5ea652452c Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 13 Aug 2020 11:18:39 +0800 Subject: [PATCH] refactor(mgb/opr-mm): set False as default value of local_grad GitOrigin-RevId: 2f9603b08719fbe2653b17231ebc74d4ee06c6cf --- src/opr-mm/impl/collective_comm.oprdecl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/opr-mm/impl/collective_comm.oprdecl b/src/opr-mm/impl/collective_comm.oprdecl index a5fc48b8..fbf96963 100644 --- a/src/opr-mm/impl/collective_comm.oprdecl +++ b/src/opr-mm/impl/collective_comm.oprdecl @@ -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 '