Event.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_TIMER_EVENT_HPP
#define KOKKOS_UTILS_TIMER_EVENT_HPP
 
#include <chrono>
 
#include "kokkos-utils/concepts/ExecutionSpace.hpp"
 
namespace Kokkos::utils::timer
{

template <Kokkos::utils::concepts::ExecutionSpace Exec>
struct Event
{
    using impl_event_t = std::chrono::steady_clock::time_point;
 
    impl_event_t event;

    void record(const Exec& /* 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