clock_check.hpp

Go to the documentation of this file.

#ifndef CLOCK_CHECK_HPP
#define CLOCK_CHECK_HPP
 
#include <chrono>
#include "my_mpi.hpp"
#include "timer_macro.hpp"
 
class ClockCheck{
 
    std::chrono::steady_clock::time_point start_time_;
    std::chrono::minutes duration_;
    int duration_int_;
    bool on;
 
    public:
 
    // Constructor takes the number of minutes
    ClockCheck(int minutes)
      : start_time_(std::chrono::steady_clock::now()),
        duration_int_(minutes),
        on(false)
    {
        if(minutes>0){
            duration_ = std::chrono::minutes(minutes);
            on = true;
        }
    }
 
    // Returns true if that many minutes have passed since initialization
    bool hasElapsed() const {
        if(!on) return false;
 
        GPTLstart("JOB_TIME_CHECK");
        auto now = std::chrono::steady_clock::now();
        auto time_elapsed = now - start_time_;
        bool time_has_elapsed = (time_elapsed > duration_);
 
        // Use time on root rank to ensure all ranks agree
        // If MPI_Bcast is expensive, could rely on local time until close to end.
        //if(time_remaining < 20){
#ifdef USE_MPI
        MPI_Bcast( &time_has_elapsed, 1, MPI_C_BOOL, 0, SML_COMM_WORLD );
#endif
        //}
        GPTLstop("JOB_TIME_CHECK");
 
        if(is_rank_zero() && time_has_elapsed) printf("\n\n*** Exiting XGC early due to command line walltime limit (max_walltime = %d minutes) ***\n\n\n", duration_int_);
 
        return time_has_elapsed;
    }
};
 
#endif