Timer.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_TIMER_TIMER_HPP
#define KOKKOS_UTILS_TIMER_TIMER_HPP
 
#include "kokkos-utils/timer/Duration.hpp"
#include "kokkos-utils/timer/Event.hpp"
 
namespace Kokkos::utils::timer
{
template <Kokkos::utils::concepts::ExecutionSpace Exec>
class Timer
{
public:
    using event_t = Event<Exec>;
 
public:
    void reset(const Exec& exec)
    {
        tick.record(exec);
        started = true;
    }

    void stop(const Exec& exec)
    {
        tock.record(exec);
        stopped = true;
    }

    bool is_valid() const { return started && stopped; }

    template <typename Duration = milliseconds>
    Duration duration() { return tick.template duration<Duration>(tock); }
 
private:
    event_t tick {},         tock {};
    bool    started = false, stopped = false;
};
 
} // namespace Kokkos::utils::timer
 
#endif // KOKKOS_UTILS_TIMER_TIMER_HPP