Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add requirments

Browse Source
This commit is contained in:
tqchen
2015-07-23 22:22:52 -07:00
parent 744f9015bb
commit 270a49ee75
4 changed files with 183 additions and 144 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/rabit.h
View File

@@ -138,7 +138,7 @@ inline void Broadcast(std::string *sendrecv_data, int root);
*/
template<typename OP, typename DType>
inline void Allreduce(DType *sendrecvbuf, size_t count,
void (*prepare_fun)(void *arg) = NULL,
void (*prepare_fun)(void *) = NULL,
void *prepare_arg = NULL);
// C++11 support for lambda prepare function
#if DMLC_USE_CXX11
@@ -262,7 +262,7 @@ class Reducer {
* \param prepare_arg argument used to pass into the lazy preprocessing function
*/
inline void Allreduce(DType *sendrecvbuf, size_t count,
void (*prepare_fun)(void *arg) = NULL,
void (*prepare_fun)(void *) = NULL,
void *prepare_arg = NULL);
#if DMLC_USE_CXX11
/*!
@@ -306,7 +306,7 @@ class SerializeReducer {
*/
inline void Allreduce(DType *sendrecvobj,
size_t max_nbyte, size_t count,
void (*prepare_fun)(void *arg) = NULL,
void (*prepare_fun)(void *) = NULL,
void *prepare_arg = NULL);
// C++11 support for lambda prepare function
#if DMLC_USE_CXX11