test_Events.cpp

Go to the documentation of this file.

#include "gmock/gmock.h"
#include "gtest/gtest.h"
 
#include "kokkos-utils/callbacks/Events.hpp"
 
#include "tests/callbacks/Helpers.hpp"

 
using execution_space = Kokkos::DefaultExecutionSpace;
 
namespace Kokkos::utils::tests::callbacks
{
 
using namespace Kokkos::utils::callbacks;

TEST(EventTest, BeginEvent)
{
    static_assert(BeginEvent<BeginParallelForEvent>);
    static_assert(BeginEvent<BeginParallelReduceEvent>);
    static_assert(BeginEvent<BeginParallelScanEvent>);
    static_assert(BeginEvent<BeginFenceEvent>);
 
    static_assert( ! BeginEvent<EndParallelForEvent>);
 
    static_assert( ! BeginEvent<PushRegionEvent>);
}

TEST(EventTest, EndEvent)
{
    static_assert(EndEvent<EndParallelForEvent>);
    static_assert(EndEvent<EndParallelReduceEvent>);
    static_assert(EndEvent<EndParallelScanEvent>);
    static_assert(EndEvent<EndFenceEvent>);
 
    static_assert( ! EndEvent<BeginParallelForEvent>);
 
    static_assert( ! EndEvent<PopRegionEvent>);
 
    static_assert( ! EndEvent<StopProfileSectionEvent>);
}

TEST(EventTest, DataEvent)
{
    static_assert(DataEvent<AllocateDataEvent>);
    static_assert(DataEvent<DeallocateDataEvent>);
 
    static_assert( ! DataEvent<PopRegionEvent>);
}

TEST(EventTest, ProfileSectionManipulationEvent)
{
    static_assert( ! ProfileSectionManipulationEvent<EndParallelForEvent>);
 
    static_assert(ProfileSectionManipulationEvent<StartProfileSectionEvent>);
    static_assert(ProfileSectionManipulationEvent<StopProfileSectionEvent>);
    static_assert(ProfileSectionManipulationEvent<DestroyProfileSectionEvent>);
 
    static_assert( ! ProfileSectionManipulationEvent<PushRegionEvent>);
    static_assert( ! ProfileSectionManipulationEvent<PopRegionEvent>);
}

TEST(EventTest, NamedEvent)
{
    static_assert(NamedEvent<BeginParallelForEvent>);
    static_assert(NamedEvent<BeginParallelReduceEvent>);
    static_assert(NamedEvent<BeginParallelScanEvent>);
    static_assert(NamedEvent<BeginFenceEvent>);
    static_assert(NamedEvent<CreateProfileSectionEvent>);
    static_assert(NamedEvent<PushRegionEvent>);
    static_assert(NamedEvent<ProfileEvent>);
 
    static_assert( ! NamedEvent<PopRegionEvent>);
}

TEST(EventTest, name)
{
#define CHECK_GET_NAME(__type__) \
    static_assert(get_name<__type__>() == #__type__);
 
    CHECK_GET_NAME(BeginParallelForEvent)
    CHECK_GET_NAME(BeginParallelReduceEvent)
    CHECK_GET_NAME(BeginParallelScanEvent)
    CHECK_GET_NAME(DestroyProfileSectionEvent)
}
 
template <Event>
struct EventTest;
 
template <BeginEvent EventType>
struct EventTest<EventType> : public ::testing::Test
{
    EventType event {.name = "my begin event", .dev_id = 1, .event_id = 2};
    std::string expt_descr = "{name = \"my begin event\", dev_id = 1, event_id = 2}";
};
 
template <EndEvent EventType>
struct EventTest<EventType> : public ::testing::Test
{
    EventType event {.event_id = 1};
    std::string expt_descr {"{event_id = 1}"};
};
 
template <DataEvent EventType>
struct EventTest<EventType> : public ::testing::Test
{
    EventType event {.alloc = {.kpsh = Kokkos::Profiling::make_space_handle("Host"), .name = "my data event", .ptr = reinterpret_cast<void*>(0x7ffdbc161a70), .size = 2}};
    std::string expt_descr {"{name = \"my data event\", space = \"Host\", ptr = 0x7ffdbc161a70, size = 2}"};
};
 
template <>
struct EventTest<BeginDeepCopyEvent> : public ::testing::Test
{
    BeginDeepCopyEvent event {
        .dst = {.kpsh = Kokkos::Profiling::make_space_handle("Cuda"), .name = "my destination", .ptr = reinterpret_cast<void*>(0x7ffdbc161a80), .size = 2},
        .src = {.kpsh = Kokkos::Profiling::make_space_handle("Host"), .name = "my source",      .ptr = reinterpret_cast<void*>(0x7ffdbc161a70), .size = 2}
    };
    std::string expt_descr {"{src = \"my source\" (Host, 0x7ffdbc161a70) -> dst = \"my destination\" (Cuda, 0x7ffdbc161a80) of size 2}"};
};
 
template <>
struct EventTest<EndDeepCopyEvent> : public ::testing::Test
{
    EndDeepCopyEvent event {};
    std::string expt_descr {"{}"};
};
 
template <>
struct EventTest<CreateProfileSectionEvent> : public ::testing::Test
{
    CreateProfileSectionEvent event {.name = "my profile section", .section_id = 1};
    std::string expt_descr {"{name = \"my profile section\", section_id = 1}"};
};
 
template <ProfileSectionManipulationEvent EventType>
struct EventTest<EventType> : public ::testing::Test
{
    EventType event {.section_id = 1};
    std::string expt_descr {"{section_id = 1}"};
};
 
template <>
struct EventTest<PushRegionEvent> : public ::testing::Test
{
    PushRegionEvent event {.name = "my push region event"};
    std::string expt_descr {"{name = \"my push region event\"}"};
};
 
template <>
struct EventTest<PopRegionEvent> : public ::testing::Test
{
    PopRegionEvent event {};
    std::string expt_descr {"{}"};
};
 
template <>
struct EventTest<ProfileEvent> : public ::testing::Test
{
    ProfileEvent event {.name = "my profile event"};
    std::string expt_descr {"{name = \"my profile event\"}"};
};
 
TYPED_TEST_SUITE(EventTest, EventTestTypes);

TYPED_TEST(EventTest, concept)
{
    static_assert(Event<TypeParam>);
}

TYPED_TEST(EventTest, traits)
{
    static_assert(std::is_aggregate_v<TypeParam>);
    static_assert(std::is_default_constructible_v<TypeParam>);
    static_assert(std::movable<TypeParam>);
}

TYPED_TEST(EventTest, description)
{
    ASSERT_THAT([&event = this->event]{ std::stringstream oss; oss << event; return oss.str(); }(), ::testing::StrEq(std::string(get_name<TypeParam>()) + ": " + this->expt_descr));
}

TYPED_TEST(EventTest, get_and_set_callback_from_and_in_eventset)
{
    using callback_fptr_t = decltype(get_callback_from_eventset<TypeParam>(std::declval<Kokkos::Tools::Experimental::EventSet>()));

    const auto context_callbacks = Kokkos::Tools::Experimental::get_callbacks();

    const auto context_callback = get_callback_from_eventset<TypeParam>(context_callbacks);
    static_assert(std::same_as<std::remove_const_t<decltype(context_callback)>, callback_fptr_t>);

    const auto callback_to_register = [] <typename... Args>(Args...) -> void {};
    static_assert(std::convertible_to<decltype(callback_to_register), callback_fptr_t>);

    get_callback_setter<TypeParam>()(callback_to_register);

    ASSERT_EQ(get_callback_from_eventset<TypeParam>(Kokkos::Tools::Experimental::get_callbacks()), callback_to_register);

    get_callback_setter<TypeParam>()(context_callback);
}
 
} // namespace Kokkos::utils::tests::callbacks