Event.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_TIMER_EVENT_HPP
#define KOKKOS_UTILS_TIMER_EVENT_HPP
 
#include <chrono>
 
namespace Kokkos::utils::timer
{

template <typename T>
requires Kokkos::ExecutionSpace<T> || std::is_void_v<T>
struct Event
{
    using impl_clock_t = std::conditional_t<
        std::chrono::high_resolution_clock::is_steady,
        std::chrono::high_resolution_clock,
        std::chrono::steady_clock
    >;
 
    using impl_event_t = typename impl_clock_t::time_point;
 
    impl_event_t event;

    template <typename U = T>
    requires std::is_void_v<U>
    void record() {
        event = std::chrono::steady_clock::now();
    }
 
    template <typename Exec = T>
    requires Kokkos::ExecutionSpace<Exec>
    void record(const Exec&) {
        event = std::chrono::steady_clock::now();
    }

    template <typename Duration>
    Duration duration(const Event& other) const {
        return std::chrono::duration_cast<Duration>(other.event - event);
    }
};
 
} // namespace Kokkos::utils::timer
 
#ifdef KOKKOS_ENABLE_CUDA
#include "kokkos-utils/timer/Cuda/Event.hpp"
#endif // KOKKOS_ENABLE_CUDA
 
#ifdef KOKKOS_ENABLE_HIP
#include "kokkos-utils/timer/HIP/Event.hpp"
#endif // KOKKOS_ENABLE_HIP
 
#endif // KOKKOS_UTILS_TIMER_EVENT_HPP