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

7
test/speed_test.cpp
View File

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