|
| 1 | +#include "mpi.h" |
| 2 | +#include <string> |
| 3 | +#include <iostream> |
| 4 | +#include <time.h> |
| 5 | +#include "Probes.h" |
| 6 | +#include "Variables.h" |
| 7 | +#include "Controller.h" |
| 8 | + |
| 9 | +using namespace std; |
| 10 | + |
| 11 | +double start_time, start_get, start_put, start_collective, start_barrier; |
| 12 | +static int n_procs, my_id; |
| 13 | + |
| 14 | +static SingleProbe *total_time_p; |
| 15 | + |
| 16 | +int MPI_Init_thread(int *argc, char ***argv, int required, int *provided) |
| 17 | +{ |
| 18 | + int provided_t, err = -1; |
| 19 | + bool async_on = false; |
| 20 | + |
| 21 | + AITuning_start("MPICH"); |
| 22 | + UserDefinedPerformanceVar *total_time_v = new UserDefinedPerformanceVar((char*)"total_time",(char*)"total_time_log.txt"); |
| 23 | + AITuning_addUserDefinedPerformanceVar(total_time_v); |
| 24 | + total_time_p = new SingleProbe((char*)"total_time_probe", total_time_v); |
| 25 | + |
| 26 | + err = PMPI_Init_thread(argc, argv, required, provided); |
| 27 | + |
| 28 | + MPI_Comm_size(MPI_COMM_WORLD, &n_procs); |
| 29 | + MPI_Comm_rank(MPI_COMM_WORLD, &my_id); |
| 30 | + |
| 31 | + start_time = MPI_Wtime(); |
| 32 | + |
| 33 | + return err; |
| 34 | +} |
| 35 | + |
| 36 | +int MPI_Finalize(void){ |
| 37 | + |
| 38 | + int err = -1; |
| 39 | + double end_time,average,total_time,elapsed_time; |
| 40 | + |
| 41 | + end_time = MPI_Wtime(); |
| 42 | + |
| 43 | + elapsed_time = end_time - start_time; |
| 44 | + |
| 45 | + MPI_Reduce(&elapsed_time, &total_time, 1, MPI_DOUBLE, MPI_SUM, 0, MPI_COMM_WORLD); |
| 46 | + |
| 47 | + total_time = total_time/n_procs; |
| 48 | + |
| 49 | + printf("Registring value in probe\n"); |
| 50 | + total_time_p->registerValue(total_time); |
| 51 | + |
| 52 | + err = PMPI_Finalize(); |
| 53 | + |
| 54 | + return err; |
| 55 | +} |
0 commit comments