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
|
// 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 <gtest/gtest.h>
template <class ViewT, class Exts>
void test_layout_single_rank_impl(ViewT a, Exts exts, int stride) {
// Avoid warning for pointless comparison of unsigned with 0
if constexpr (ViewT::rank() > 0) {
for (unsigned r = 0; r < ViewT::rank(); r++) {
ASSERT_EQ(a.extent(r), exts.extent(r));
ASSERT_EQ(a.layout().dimension[r], exts.extent(r));
}
}
if constexpr (ViewT::rank() > 1) {
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
// In Legacy View the stride wasn't propagated correctly
ASSERT_EQ(a.layout().stride, stride);
int actual_stride =
std::is_same_v<typename ViewT::array_layout, Kokkos::LayoutLeft>
? a.stride(1)
: a.stride(ViewT::rank() - 2);
#else
// In Legacy View the padded dimension was always the leftmost
int actual_stride =
std::is_same_v<typename ViewT::array_layout, Kokkos::LayoutLeft>
? a.stride(1)
: a.stride(0);
#endif
ASSERT_EQ(actual_stride, stride);
}
{
ViewT b("B", a.layout());
ASSERT_EQ(a.layout(), b.layout());
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
ASSERT_EQ(a.mapping(), b.mapping());
#endif
}
{
ViewT b(a.data(), a.layout());
ASSERT_EQ(a.layout(), b.layout());
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
ASSERT_EQ(a.mapping(), b.mapping());
#endif
}
{
ViewT b(Kokkos::view_alloc("B"), a.layout());
ASSERT_EQ(a.layout(), b.layout());
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
ASSERT_EQ(a.mapping(), b.mapping());
#endif
}
{
ViewT b(Kokkos::view_wrap(a.data()), a.layout());
ASSERT_EQ(a.layout(), b.layout());
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
ASSERT_EQ(a.mapping(), b.mapping());
#endif
}
}
template <class ViewT, std::integral... Sizes>
void test_layout_single_rank(Sizes... sizes) {
#ifdef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
using extents_type =
typename decltype(std::declval<ViewT>().to_mdspan())::extents_type;
#else
using extents_type = typename ViewT::extents_type;
using mapping_type = typename ViewT::mapping_type;
#endif
extents_type exts{sizes...};
int padded_extent = 1;
if (ViewT::rank() > 0) {
padded_extent =
std::is_same_v<typename ViewT::array_layout, Kokkos::LayoutLeft>
?
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
exts.extent(0)
: exts.extent(ViewT::rank() - 1);
#else
exts.extent(0)
: ((sizes * ... * 1)) / exts.extent(0);
#endif
}
// Test with no padding
test_layout_single_rank_impl(ViewT("A", sizes...), exts, padded_extent);
// using array_layout because I need LayoutLeft/Right, not
// layout_left/right_padded
using Layout = typename ViewT::array_layout;
Layout l;
// Avoid warning for pointless comparison with 0
if constexpr (ViewT::rank() > 0)
for (size_t r = 0; r < ViewT::rank(); r++) l.dimension[r] = exts.extent(r);
test_layout_single_rank_impl(ViewT("A", l), exts, padded_extent);
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
test_layout_single_rank_impl(ViewT("A", mapping_type(exts)), exts,
padded_extent);
#endif
// Test with padding
padded_extent += 3;
l.stride = padded_extent;
test_layout_single_rank_impl(ViewT("A", l), exts, padded_extent);
#ifndef KOKKOS_ENABLE_IMPL_VIEW_LEGACY
test_layout_single_rank_impl(ViewT("A", mapping_type(exts, padded_extent)),
exts, padded_extent);
#endif
}
template <class Layout>
void test_layout() {
test_layout_single_rank<Kokkos::View<double, Layout, TEST_EXECSPACE>>();
test_layout_single_rank<Kokkos::View<double*, Layout, TEST_EXECSPACE>>(7);
test_layout_single_rank<Kokkos::View<double[7], Layout, TEST_EXECSPACE>>(7);
test_layout_single_rank<Kokkos::View<double**, Layout, TEST_EXECSPACE>>(7,
11);
test_layout_single_rank<Kokkos::View<double* [11], Layout, TEST_EXECSPACE>>(
7, 11);
test_layout_single_rank<Kokkos::View<double*******, Layout, TEST_EXECSPACE>>(
7, 11, 13, 17, 19, 2, 3);
test_layout_single_rank<
Kokkos::View<double***** [2][3], Layout, TEST_EXECSPACE>>(7, 11, 13, 17,
19, 2, 3);
}
TEST(TEST_CATEGORY, view_legacy_layout_member) {
test_layout<Kokkos::LayoutLeft>();
test_layout<Kokkos::LayoutRight>();
}
|
