Replaced calls to omp_get_wtime with a wrapper function get_wtime that is implemented either by mpi_wtime, omp_get_wtime or a custom implementation. The choice depends on active compiler flags.