1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
import kokkos.core_impl;
#else
#include <Kokkos_Core.hpp>
#endif
namespace {
// Dummy policy for testing base class.
template <class... Args>
struct DummyPolicy : Kokkos::Impl::PolicyTraits<Args...> {
using execution_policy = DummyPolicy;
};
// Asserts that a policy constructor is semiregular.
// Semiregular is copyable and default initializable
// (regular requires equality comparable).
template <class Policy>
constexpr bool check_semiregular() {
static_assert(std::is_default_constructible_v<Policy>);
static_assert(std::is_copy_constructible_v<Policy>);
static_assert(std::is_move_constructible_v<Policy>);
static_assert(std::is_copy_assignable_v<Policy>);
static_assert(std::is_move_assignable_v<Policy>);
static_assert(std::is_destructible_v<Policy>);
return true;
}
static_assert(check_semiregular<DummyPolicy<>>());
static_assert(check_semiregular<Kokkos::RangePolicy<>>());
static_assert(check_semiregular<Kokkos::TeamPolicy<>>());
static_assert(check_semiregular<Kokkos::MDRangePolicy<Kokkos::Rank<2>>>());
// Assert that occupancy conversion and hints work properly.
template <class Policy>
void test_prefer_desired_occupancy() {
Policy policy;
using Kokkos::Experimental::DesiredOccupancy;
using Kokkos::Experimental::MaximizeOccupancy;
using Kokkos::Experimental::prefer;
using Kokkos::Experimental::WorkItemProperty;
static_assert(!Policy::experimental_contains_desired_occupancy);
// MaximizeOccupancy -> MaximizeOccupancy
auto const policy_still_no_occ = prefer(policy, MaximizeOccupancy{});
static_assert(
!decltype(policy_still_no_occ)::experimental_contains_desired_occupancy);
// MaximizeOccupancy -> DesiredOccupancy
auto const policy_with_occ =
prefer(policy_still_no_occ, DesiredOccupancy{33});
static_assert(
decltype(policy_with_occ)::experimental_contains_desired_occupancy);
EXPECT_EQ(policy_with_occ.impl_get_desired_occupancy().value(), 33);
// DesiredOccupancy -> DesiredOccupancy
auto const policy_change_occ = prefer(policy_with_occ, DesiredOccupancy{24});
static_assert(
decltype(policy_change_occ)::experimental_contains_desired_occupancy);
EXPECT_EQ(policy_change_occ.impl_get_desired_occupancy().value(), 24);
// DesiredOccupancy -> DesiredOccupancy w/ hint
auto policy_with_occ_and_hint = Kokkos::Experimental::require(
policy_change_occ,
Kokkos::Experimental::WorkItemProperty::HintLightWeight);
EXPECT_EQ(policy_with_occ_and_hint.impl_get_desired_occupancy().value(), 24);
// DesiredOccupancy -> MaximizeOccupancy
auto const policy_drop_occ =
prefer(policy_with_occ_and_hint, MaximizeOccupancy{});
static_assert(
!decltype(policy_drop_occ)::experimental_contains_desired_occupancy);
}
TEST(TEST_CATEGORY, execution_policy_occupancy_and_hint) {
test_prefer_desired_occupancy<DummyPolicy<>>();
test_prefer_desired_occupancy<Kokkos::RangePolicy<>>();
test_prefer_desired_occupancy<Kokkos::TeamPolicy<>>();
test_prefer_desired_occupancy<Kokkos::MDRangePolicy<Kokkos::Rank<2>>>();
}
// Check that the policy size does not increase if the user does not specify the
// occupancy (only pay for what you use).
// Disabling since EBO was not working on Windows with multiple configurations
#if !defined(_WIN32)
constexpr bool test_empty_base_optimization() {
DummyPolicy<> policy;
static_assert(sizeof(decltype(policy)) == 1);
using Kokkos::Experimental::DesiredOccupancy;
using Kokkos::Experimental::prefer;
static_assert(sizeof(decltype(prefer(policy, DesiredOccupancy{33}))) ==
sizeof(DesiredOccupancy));
return true;
}
static_assert(test_empty_base_optimization());
#endif
} // namespace
|
