EventInProfileSectionMatcher.hpp

Go to the documentation of this file.

#ifndef KOKKOS_UTILS_CALLBACKS_EVENTINPROFILESECTIONMATCHER_HPP
#define KOKKOS_UTILS_CALLBACKS_EVENTINPROFILESECTIONMATCHER_HPP
 
#include "Kokkos_NumericTraits.hpp"
 
#include "kokkos-utils/callbacks/Events.hpp"
#include "kokkos-utils/callbacks/Matcher.hpp"
 
namespace Kokkos::utils::callbacks
{

template <typename MatcherType> requires MatcherFor<MatcherType, CreateProfileSectionEvent>
struct EventInProfileSectionMatcher
{
    static constexpr uint32_t invalid_section_id = Kokkos::Experimental::finite_max_v<uint32_t>;
 
    bool operator()(const CreateProfileSectionEvent& event)
    {
        if (matcher(event))
        {
            if (section_id != invalid_section_id) Kokkos::abort("EventInProfileSectionMatcher cannot match a create event twice.");
            section_id = event.section_id;
        }
        return false;
    }
 
    bool operator()(const StartProfileSectionEvent& event)
    {
        if (event.section_id == section_id) recording = true;
        return false;
    }
 
    bool operator()(const StopProfileSectionEvent& event)
    {
        if (event.section_id == section_id) recording = false;
        return false;
    }
 
    template <Event EventType>
    bool operator()(const EventType& /* event */) const {
        return recording;
    }
 
    MatcherType matcher {};
    uint32_t section_id = invalid_section_id;
    bool recording = false;
};
 
} // namespace Kokkos::utils::callbacks
 
#endif // KOKKOS_UTILS_CALLBACKS_EVENTINPROFILESECTIONMATCHER_HPP