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; 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(); int nproc = rabit::GetWorldSize();
double tsum = tdiff; double tsum = tdiff;
rabit::Allreduce<op::Sum>(&tsum, 1); rabit::Allreduce<op::Sum>(&tsum, 1);
@ -62,6 +62,11 @@ inline void PrintStats(const char *name, double tdiff) {
double tstd = sqrt(tsqr / nproc); double tstd = sqrt(tsqr / nproc);
if (rabit::GetRank() == 0) { if (rabit::GetRank() == 0) {
utils::LogPrintf("%s: mean=%g, std=%g sec\n", name, tavg, tstd); 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; tot_tdiff = utils::GetTime() - tstart;
// use allreduce to get the sum and std of time // use allreduce to get the sum and std of time
PrintStats("max_tdiff", max_tdiff); PrintStats("max_tdiff", max_tdiff, n, nrep, sizeof(float));
PrintStats("sum_tdiff", sum_tdiff); PrintStats("sum_tdiff", sum_tdiff, n, nrep, sizeof(float));
PrintStats("bcast_tdiff", bcast_tdiff); PrintStats("bcast_tdiff", bcast_tdiff, n, nrep, sizeof(char));
PrintStats("tot_tdiff", tot_tdiff); PrintStats("tot_tdiff", tot_tdiff, 0, nrep, sizeof(float));
rabit::Finalize(); rabit::Finalize();
return 0; return 0;
} }