diff options
| -rw-r--r-- | training/mpi_batch_optimize.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/training/mpi_batch_optimize.cc b/training/mpi_batch_optimize.cc index 5a6bf301..39a8af7d 100644 --- a/training/mpi_batch_optimize.cc +++ b/training/mpi_batch_optimize.cc @@ -335,6 +335,7 @@ int main(int argc, char** argv) { while (!converged) { observer.Reset(); #ifdef HAVE_MPI + mpi::timer timer; world.barrier(); #endif if (rank == 0) { @@ -404,6 +405,7 @@ int main(int argc, char** argv) { #ifdef HAVE_MPI mpi::broadcast(world, &lambdas[0], lambdas.size(), 0); mpi::broadcast(world, cint, 0); + if (rank == 0) { cerr << " ELAPSED TIME THIS ITERATION=" << timer.elapsed() << endl; } #endif converged = cint; } |