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 <typename T>
requires Kokkos::ExecutionSpace<T> || std::is_void_v<T>
class Timer
{
public:
    using event_t = Event<T>;
 
public:
    bool is_valid() const { return tick.has_value() && tock.has_value(); }

    template <typename Duration = milliseconds>
    Duration duration() { return tick->template duration<Duration>(*tock); }

    template <typename U = T>
    requires std::is_void_v<U>
    void start()
    {
        if(! tick.has_value()) tick.emplace();
        tick->record();
    }
 
    template <typename Exec = T>
    requires Kokkos::ExecutionSpace<Exec>
    void start(const Exec& exec)
    {
        if(! tick.has_value()) tick.emplace();
        tick->record(exec);
    }

    template <typename U = T>
    requires std::is_void_v<U>
    void stop()
    {
        if(! tock.has_value()) tock.emplace();
        tock->record();
    }
 
    template <typename Exec = T>
    requires Kokkos::ExecutionSpace<Exec>
    void stop(const Exec& exec)
    {
        if(! tock.has_value()) tock.emplace();
        tock->record(exec);
    }
 
private:
    std::optional<event_t> tick = std::nullopt, tock = std::nullopt;
};
 
} // namespace Kokkos::utils::timer
 
#endif // KOKKOS_UTILS_TIMER_TIMER_HPP