RegionTimerListener.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_CALLBACKS_REGIONTIMERLISTENER_HPP
#define KOKKOS_UTILS_CALLBACKS_REGIONTIMERLISTENER_HPP
 
#include "kokkos-utils/callbacks/EventRegionMatcher.hpp"
#include "kokkos-utils/callbacks/EventTypeMatcher.hpp"
#include "kokkos-utils/callbacks/TimerListener.hpp"
#include "kokkos-utils/timer/Timer.hpp"
 
namespace Kokkos::utils::callbacks
{

template <MatcherFor<PushRegionEvent> MatcherType, typename TimerType = Kokkos::utils::timer::Timer<void>>
struct RegionTimerListener : public TimerListener<EventTypeMatcher<EventRegionMatcher<MatcherType, true>, PushRegionEvent, PopRegionEvent>, TimerType>
{
    using region_matcher_t = EventRegionMatcher<MatcherType, true>;
    using matcher_t        = EventTypeMatcher<region_matcher_t, PushRegionEvent, PopRegionEvent>;
    using timer_t          = TimerType;
    using base_t           = TimerListener<matcher_t, timer_t>;
 
    template <typename T, typename U = TimerType>
    RegionTimerListener(T&& matcher_, U&& timer_ = TimerType{}) : base_t{
        matcher_t{matcher_t{.matcher = {.matcher = {std::forward<T>(matcher_)}}}},
        timer_t{std::forward<U>(timer_)}
    } {}
};
 
} // namespace Kokkos::utils::callbacks
 
#endif // KOKKOS_UTILS_CALLBACKS_REGIONTIMERLISTENER_HPP