[GitHub] threeleafzerg commented on a change in pull request #10696: [MXNET-366]Extend MXNet Distributed Training by AllReduce

[GitBox <gi...@apache.org>]

threeleafzerg commented on a change in pull request #10696: [MXNET-366]Extend MXNet Distributed Training by AllReduce
URL: https://github.com/apache/incubator-mxnet/pull/10696#discussion_r195619148
 
 

 ##########
 File path: include/mxnet/c_api.h
 ##########
 @@ -1877,6 +1877,76 @@ MXNET_DLL int MXKVStorePullEx(KVStoreHandle handle,
                               NDArrayHandle* vals,
                               int priority);
 
+/*!
+ * \brief aggregate and sum up a list of (key, value) pairs from from all nodes, the result is stored
+ *        in out_vals. It has the same syntax as allreduce.
+ * \param handle handle to the kvstore
+ * \param num the number of key-value pairs
+ * \param keys the list of keys
+ * \param in_vals the list of values to be aggregated
+ * \param out_vals the list of values to store the result
+ * \param priority the priority of the action
+ * \return 0 when success, -1 when failure happens
+ */
+MXNET_DLL int MXKVStorePushPull(KVStoreHandle handle,
+                                mx_uint num,
 
 Review comment:
   OK, I will add them.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services