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

ok

Browse Source
This commit is contained in:
tqchen 2014-12-06 11:20:52 -08:00
parent 0e012cb05e
commit 1519f74f3c
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
test/speed_test.cpp
View File

@ -51,7 +51,7 @@ inline void TestBcast(size_t n, int root) {
bcast_tdiff += utils::GetTime() - tstart;
}
inline void PrintStats(const char *name, double tdiff) {
inline void PrintStats(const char *name, double tdiff, int n, int nrep, size_t size) {
int nproc = rabit::GetWorldSize();
double tsum = tdiff;
rabit::Allreduce<op::Sum>(&tsum, 1);
@ -62,6 +62,11 @@ inline void PrintStats(const char *name, double tdiff) {
double tstd = sqrt(tsqr / nproc);
if (rabit::GetRank() == 0) {
utils::LogPrintf("%s: mean=%g, std=%g sec\n", name, tavg, tstd);
double ndata = n;
ndata *= nrep * size;
if (n != 0) {
utils::LogPrintf("%s-speed: %g MB/sec\n", name, (ndata / tavg) / 1024 / 1024 );
}
}
}
@ -87,10 +92,10 @@ int main(int argc, char *argv[]) {
}
tot_tdiff = utils::GetTime() - tstart;
// use allreduce to get the sum and std of time
PrintStats("max_tdiff", max_tdiff);
PrintStats("sum_tdiff", sum_tdiff);
PrintStats("bcast_tdiff", bcast_tdiff);
PrintStats("tot_tdiff", tot_tdiff);
PrintStats("max_tdiff", max_tdiff, n, nrep, sizeof(float));
PrintStats("sum_tdiff", sum_tdiff, n, nrep, sizeof(float));
PrintStats("bcast_tdiff", bcast_tdiff, n, nrep, sizeof(char));
PrintStats("tot_tdiff", tot_tdiff, 0, nrep, sizeof(float));
rabit::Finalize();
return 0;
}