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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
// 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
#include <Kokkos_TypeInfo.hpp>
#include <gtest/gtest.h>
namespace Test {
namespace Impl {
template <class ViewTypeDst, class ViewTypeSrc>
struct TestAssignability {
using mapping_type =
Kokkos::Impl::ViewMapping<typename ViewTypeDst::traits,
typename ViewTypeSrc::traits
#ifdef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
,
typename ViewTypeDst::specialize
#else
,
void
#endif
>;
template <class MappingType>
static void try_assign(
ViewTypeDst& dst, ViewTypeSrc& src,
std::enable_if_t<MappingType::is_assignable>* = nullptr) {
dst = src;
}
template <class MappingType>
static void try_assign(
ViewTypeDst&, ViewTypeSrc&,
std::enable_if_t<!MappingType::is_assignable>* = nullptr) {
FAIL() << "TestAssignability::try_assign: Unexpected call path";
}
template <class... Dimensions>
static void test(bool always, bool sometimes, Dimensions... dims) {
ViewTypeDst dst;
ViewTypeSrc src("SRC", dims...);
bool is_always_assignable =
Kokkos::is_always_assignable<ViewTypeDst, ViewTypeSrc>::value;
bool is_assignable = Kokkos::is_assignable(dst, src);
if (sometimes) {
try_assign<mapping_type>(dst, src);
}
ASSERT_EQ(always, is_always_assignable)
<< Kokkos::Impl::TypeInfo<ViewTypeSrc>::name() << " to "
<< Kokkos::Impl::TypeInfo<ViewTypeDst>::name();
ASSERT_EQ(sometimes, is_assignable)
<< Kokkos::Impl::TypeInfo<ViewTypeSrc>::name() << " to "
<< Kokkos::Impl::TypeInfo<ViewTypeDst>::name();
}
};
} // namespace Impl
TEST(TEST_CATEGORY, view_is_assignable) {
using namespace Kokkos;
using h_exec = typename DefaultHostExecutionSpace::memory_space;
using d_exec = typename TEST_EXECSPACE::memory_space;
using left = LayoutLeft;
using right = LayoutRight;
using stride = LayoutStride;
// Static/Dynamic Extents
Impl::TestAssignability<View<int*, left, d_exec>,
View<int*, left, d_exec>>::test(true, true, 10);
Impl::TestAssignability<View<int[10], left, d_exec>,
View<int*, left, d_exec>>::test(false, true, 10);
Impl::TestAssignability<View<int[5], left, d_exec>,
View<int*, left, d_exec>>::test(false, false, 10);
Impl::TestAssignability<View<int*, left, d_exec>,
View<int[10], left, d_exec>>::test(true, true);
Impl::TestAssignability<View<int[10], left, d_exec>,
View<int[10], left, d_exec>>::test(true, true);
Impl::TestAssignability<View<int[5], left, d_exec>,
View<int[10], left, d_exec>>::test(false, false);
Impl::TestAssignability<View<int**, left, d_exec>,
View<int**, left, d_exec>>::test(true, true, 10, 10);
Impl::TestAssignability<View<int* [10], left, d_exec>,
View<int**, left, d_exec>>::test(false, true, 10, 10);
Impl::TestAssignability<View<int* [5], left, d_exec>,
View<int**, left, d_exec>>::test(false, false, 10,
10);
Impl::TestAssignability<View<int**, left, d_exec>,
View<int* [10], left, d_exec>>::test(true, true, 10);
Impl::TestAssignability<View<int* [10], left, d_exec>,
View<int* [10], left, d_exec>>::test(true, true, 10);
Impl::TestAssignability<View<int* [5], left, d_exec>,
View<int* [10], left, d_exec>>::test(false, false,
10);
// Mismatch value_type
Impl::TestAssignability<View<int*, left, d_exec>,
View<double*, left, d_exec>>::test(false, false, 10);
// Layout assignment
Impl::TestAssignability<View<int, left, d_exec>,
View<int, right, d_exec>>::test(true, true);
Impl::TestAssignability<View<int*, left, d_exec>,
View<int*, right, d_exec>>::test(true, true, 10);
Impl::TestAssignability<View<int[5], left, d_exec>,
View<int*, right, d_exec>>::test(false, false, 10);
Impl::TestAssignability<View<int[10], left, d_exec>,
View<int*, right, d_exec>>::test(false, true, 10);
Impl::TestAssignability<View<int*, left, d_exec>,
View<int[5], right, d_exec>>::test(true, true);
Impl::TestAssignability<View<int[5], left, d_exec>,
View<int[10], right, d_exec>>::test(false, false);
// This could be made possible (due to the degenerate nature of the views) but
// we do not allow this yet
// TestAssignability<View<int**,left,d_exec>,View<int**,right,d_exec>>::test(false,true,10,1);
Impl::TestAssignability<View<int**, left, d_exec>,
View<int**, right, d_exec>>::test(false, false, 10,
2);
Impl::TestAssignability<View<int**, stride, d_exec>,
View<int**, right, d_exec>>::test(true, true, 10, 2);
Impl::TestAssignability<View<int**, stride, d_exec>,
View<int**, left, d_exec>>::test(true, true, 10, 2);
// Space Assignment
bool expected = Kokkos::Impl::MemorySpaceAccess<d_exec, h_exec>::assignable;
Impl::TestAssignability<View<int*, left, d_exec>,
View<int*, left, h_exec>>::test(expected, expected,
10);
expected = Kokkos::Impl::MemorySpaceAccess<h_exec, d_exec>::assignable;
Impl::TestAssignability<View<int*, left, h_exec>,
View<int*, left, d_exec>>::test(expected, expected,
10);
// reference type and const-qualified types
using SomeViewType = View<int*, left, d_exec>;
static_assert(is_always_assignable_v<SomeViewType, SomeViewType>);
static_assert(is_always_assignable_v<SomeViewType, SomeViewType&>);
static_assert(is_always_assignable_v<SomeViewType, SomeViewType const>);
static_assert(is_always_assignable_v<SomeViewType, SomeViewType const&>);
static_assert(is_always_assignable_v<SomeViewType&, SomeViewType>);
static_assert(is_always_assignable_v<SomeViewType&, SomeViewType&>);
static_assert(is_always_assignable_v<SomeViewType&, SomeViewType const>);
static_assert(is_always_assignable_v<SomeViewType&, SomeViewType const&>);
}
} // namespace Test
|
