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

bugfix in msg passing

Browse Source
This commit is contained in:
tqchen 2014-11-30 17:42:18 -08:00
parent d8d648549f
commit 27f6f8ea9e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/engine_robust-inl.h
View File

@ -53,8 +53,10 @@ AllReduceRobust::MsgPassing(const NodeType &node_value,
// 2: recv message from parent // 2: recv message from parent
// 3: send message to childs // 3: send message to childs
int stage = 0; int stage = 0;
// if no childs, no need to reduce // if no childs, no need to, directly start passing message
if (nlink == static_cast<int>(parent_index != -1)) { if (nlink == static_cast<int>(parent_index != -1)) {
utils::Assert(parent_index == 0, "parent must be 0");
edge_out[parent_index] = func(node_value, edge_in, parent_index);
stage = 1; stage = 1;
} }
// while we have not passed the messages out // while we have not passed the messages out