BeginEndTimerListener.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_CALLBACKS_BEGINENDTIMERLISTENER_HPP
#define KOKKOS_UTILS_CALLBACKS_BEGINENDTIMERLISTENER_HPP
 
#include "kokkos-utils/callbacks/ConjunctionMatcher.hpp"
#include "kokkos-utils/callbacks/EnqueuedEventTimer.hpp"
#include "kokkos-utils/callbacks/EnqueuedEventWithLaunchTimer.hpp"
#include "kokkos-utils/callbacks/EventBeginEndIdMatcher.hpp"
#include "kokkos-utils/callbacks/EventQueueMatcher.hpp"
#include "kokkos-utils/callbacks/TimerListener.hpp"
 
namespace Kokkos::utils::callbacks
{

template <Matcher MatcherType, Event BeginEventType, Kokkos::ExecutionSpace Exec, typename TimerType = EnqueuedEventTimer<Exec>>
requires MatcherFor<MatcherType, BeginEventType>
struct BeginEndTimerListener : public TimerListener<EventBeginEndIdMatcher<BeginEventType, ConjunctionMatcher<EventQueueMatcher<Exec>, MatcherType>>, TimerType>
{
    using queue_matcher_t = EventQueueMatcher<Exec>;
    using matcher_t       = EventBeginEndIdMatcher<BeginEventType, ConjunctionMatcher<queue_matcher_t, MatcherType>>;
    using timer_t         = TimerType;
    using base_t          = TimerListener<matcher_t, timer_t>;
 
    template <typename T>
    BeginEndTimerListener(T&& matcher_, const Exec& exec) : base_t{
        matcher_t{.matcher = ConjunctionMatcher{queue_matcher_t{.exec = exec}, MatcherType{std::forward<T>(matcher_)}}},
        timer_t{.exec = exec}
    } {}
};

template <MatcherFor<BeginParallelForEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using ParallelForTimerListener = BeginEndTimerListener<MatcherType, BeginParallelForEvent, Exec>;

template <MatcherFor<BeginParallelReduceEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using ParallelReduceTimerListener = BeginEndTimerListener<MatcherType, BeginParallelReduceEvent, Exec>;

template <MatcherFor<BeginParallelScanEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using ParallelScanTimerListener = BeginEndTimerListener<MatcherType, BeginParallelScanEvent, Exec>;

template <MatcherFor<BeginDeepCopyEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using DeepCopyTimerListener = BeginEndTimerListener<MatcherType, BeginDeepCopyEvent, Exec>;

template <MatcherFor<BeginParallelForEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using ParallelForWithLaunchTimerListener = BeginEndTimerListener<MatcherType, BeginParallelForEvent, Exec, EnqueuedEventWithLaunchTimer<Exec>>;

template <MatcherFor<BeginParallelReduceEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using ParallelWithLaunchReduceTimerListener = BeginEndTimerListener<MatcherType, BeginParallelReduceEvent, Exec, EnqueuedEventWithLaunchTimer<Exec>>;

template <MatcherFor<BeginParallelScanEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using ParallelWithLaunchScanTimerListener = BeginEndTimerListener<MatcherType, BeginParallelScanEvent, Exec, EnqueuedEventWithLaunchTimer<Exec>>;

template <MatcherFor<BeginDeepCopyEvent> MatcherType, Kokkos::ExecutionSpace Exec>
using DeepCopyWithLaunchTimerListener = BeginEndTimerListener<MatcherType, BeginDeepCopyEvent, Exec, EnqueuedEventWithLaunchTimer<Exec>>;
 
} // namespace Kokkos::utils::callbacks
 
#endif // KOKKOS_UTILS_CALLBACKS_BEGINENDTIMERLISTENER_HPP