Homec4science
Diffusion lammps (lammm-devel) Browse

lammps/src/timer.cpp
632b105572bblammm-devel

Last Change
Raw File

timer.cpp
View Options

/* ----------------------------------------------------------------------
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
http://lammps.sandia.gov, Sandia National Laboratories
Steve Plimpton, sjplimp@sandia.gov
Copyright (2003) Sandia Corporation. Under the terms of Contract
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
certain rights in this software. This software is distributed under
the GNU General Public License.
See the README file in the top-level LAMMPS directory.
------------------------------------------------------------------------- */
#include "mpi.h"
#include "timer.h"
#include "memory.h"
#ifdef _WIN32
#include <windows.h>
#else
#include <sys/time.h>
#include <sys/resource.h>
#endif
using namespace LAMMPS_NS;
static double CPU_Time()
{
double rv = 0.0;
#ifdef _WIN32
// from MSD docs.
FILETIME ct,et,kt,ut;
union { FILETIME ft; uint64_t ui; } cpu;
if (GetProcessTimes(GetCurrentProcess(),&ct,&et,&kt,&ut)) {
cpu.ft = ut;
rv = cpu.ui * 0.0000001;
}
#else /* ! _WIN32 */
struct rusage ru;
if (getrusage(RUSAGE_SELF, &ru) == 0) {
rv = (double) ru.ru_utime.tv_sec;
rv += (double) ru.ru_utime.tv_usec * 0.000001;
}
#endif /* ! _WIN32 */
return rv;
}
/* ---------------------------------------------------------------------- */
Timer::Timer(LAMMPS *lmp) : Pointers(lmp)
{
memory->create(cpu_array,TIME_N,"timer:cpu_array");
memory->create(wall_array,TIME_N,"timer:wall_array");
}
/* ---------------------------------------------------------------------- */
Timer::~Timer()
{
memory->destroy(cpu_array);
memory->destroy(wall_array);
}
/* ---------------------------------------------------------------------- */
void Timer::init()
{
for (int i = 0; i < TIME_N; i++) cpu_array[i] = 0.0;
for (int i = 0; i < TIME_N; i++) wall_array[i] = 0.0;
}
/* ---------------------------------------------------------------------- */
void Timer::stamp()
{
// uncomment if want synchronized timing
// MPI_Barrier(world);
previous_cpu = CPU_Time();
previous_wall = MPI_Wtime();
}
/* ---------------------------------------------------------------------- */
void Timer::stamp(enum ttype which)
{
// uncomment if want synchronized timing
// MPI_Barrier(world);
double current_cpu = CPU_Time();
double current_wall = MPI_Wtime();
cpu_array[which] += current_cpu - previous_cpu;
wall_array[which] += current_wall - previous_wall;
previous_cpu = current_cpu;
previous_wall = current_wall;
}
/* ---------------------------------------------------------------------- */
void Timer::barrier_start(enum ttype which)
{
MPI_Barrier(world);
cpu_array[which] = CPU_Time();
wall_array[which] = MPI_Wtime();
}
/* ---------------------------------------------------------------------- */
void Timer::barrier_stop(enum ttype which)
{
MPI_Barrier(world);
double current_cpu = CPU_Time();
double current_wall = MPI_Wtime();
cpu_array[which] = current_cpu - cpu_array[which];
wall_array[which] = current_wall - wall_array[which];
}
/* ---------------------------------------------------------------------- */
double Timer::cpu(enum ttype which)
{
double current_cpu = CPU_Time();
return (current_cpu - cpu_array[which]);
}
/* ---------------------------------------------------------------------- */
double Timer::elapsed(enum ttype which)
{
double current_wall = MPI_Wtime();
return (current_wall - wall_array[which]);
}
/* ---------------------------------------------------------------------- */
void Timer::set_wall(enum ttype which, double newtime)
{
wall_array[which] = newtime;
}
c4science · Help