Fix bugs in SMPE.

[simgrid.git] / src / smpi / smpi_bench.c

diff --git a/src/smpi/smpi_bench.c b/src/smpi/smpi_bench.c
index bec96df..1ff5814 100644 (file)
--- a/src/smpi/smpi_bench.c
+++ b/src/smpi/smpi_bench.c
@@ -32,24 +32,22 @@ static void smpi_execute(double duration) {
   }
 }
 
-void smpi_bench_begin(MPI_Comm comm, const char* mpi_call) {
+void smpi_bench_begin(int rank, const char* mpi_call) {
   double now = SIMIX_get_clock();
-  int rank = comm != MPI_COMM_NULL ? smpi_comm_rank(comm) : -1;
 
-  if(mpi_call && xbt_cfg_get_int(_surf_cfg_set, "SMPE")) {
+  if(mpi_call && rank > 0 && xbt_cfg_get_int(_surf_cfg_set, "SMPE")) {
     INFO3("SMPE: ts=%f rank=%d type=end et=%s", now, rank, mpi_call);
   }
   xbt_os_timer_start(smpi_process_timer());
 }
 
-void smpi_bench_end(MPI_Comm comm, const char* mpi_call) {
+void smpi_bench_end(int rank, const char* mpi_call) {
   xbt_os_timer_t timer = smpi_process_timer();
   double now = SIMIX_get_clock();
-  int rank = comm != MPI_COMM_NULL ? smpi_comm_rank(comm) : -1;
 
   xbt_os_timer_stop(timer);
   smpi_execute(xbt_os_timer_elapsed(timer));
-  if(mpi_call && xbt_cfg_get_int(_surf_cfg_set, "SMPE")) {
+  if(mpi_call && rank > 0 && xbt_cfg_get_int(_surf_cfg_set, "SMPE")) {
     INFO3("SMPE: ts=%f rank=%d type=begin et=%s", now, rank, mpi_call);
   }
 }