SequenceOfRegionTimerListener.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_CALLBACKS_SEQUENCEOFREGIONTIMERLISTENER_HPP
#define KOKKOS_UTILS_CALLBACKS_SEQUENCEOFREGIONTIMERLISTENER_HPP
 
#include "kokkos-utils/callbacks/RegionTimerListener.hpp"
 
namespace Kokkos::utils::callbacks
{
template <Matcher MatcherType>
struct SequenceOfRegionTimerListener
{
public:
    using region_timer_t = RegionTimerListener<MatcherType>;
    using matcher_t      = typename region_timer_t::matcher_t;
    using timer_t        = typename region_timer_t::timer_t;
 
    template <typename... Ms>
    explicit SequenceOfRegionTimerListener(Ms&&... matchers_)
        : matchers{std::forward<Ms>(matchers_)...},
          timers(matchers.size())
    {}
 
    void reset() { current = 0; }

    bool all_matched() const {
        return current == timers.size();
    }

    void operator()(const PushRegionEvent& event)
    {
        if(all_matched()) return;
 
        const auto matched = matchers.at(current).operator()(event);
 
        if(matched)
            timers.at(current).start();
    }

    void operator()(const PopRegionEvent& event)
    {
        if(all_matched()) return;
 
        const auto matched = matchers.at(current).operator()(event);
 
        if(matched)
        {
            timers.at(current).stop();
            ++current;
        }
    }
 
public:
    std::vector<matcher_t> matchers;
    std::vector<timer_t>   timers;
 
private:
    size_t current = 0;
};
 
} // namespace Kokkos::utils::callbacks
 
#endif // KOKKOS_UTILS_CALLBACKS_SEQUENCEOFREGIONTIMERLISTENER_HPP