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

Merge branch 'master' of https://github.com/tqchen/allreduce

Browse Source
This commit is contained in:
nachocano 2014-12-06 11:59:22 -08:00
commit 771891491c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
test/speed_test.cpp
View File

@ -43,11 +43,12 @@ inline void TestBcast(size_t n, int root) {
s[i] = char(i % 126 + 1); s[i] = char(i % 126 + 1);
} }
std::string res; std::string res;
res.resize(n);
if (root == rank) { if (root == rank) {
res = s; res = s;
} }
double tstart = utils::GetTime(); double tstart = utils::GetTime();
rabit::Broadcast(&res, root); rabit::Broadcast(&res[0], res.length(), root);
bcast_tdiff += utils::GetTime() - tstart; bcast_tdiff += utils::GetTime() - tstart;
} }