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
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
#include <TestDynViewAPI.hpp>
namespace Test {
TEST(TEST_CATEGORY, dyn_rank_view_api_operator_rank12345) {
TestDynViewAPI<double, TEST_EXECSPACE>::run_operator_test_rank12345();
}
template <typename SharedMemorySpace>
void test_dyn_rank_view_resize() {
int n = 1000;
Kokkos::DynRankView<double, SharedMemorySpace> device_view("device view", n);
// Make sure we don't deallocate memory in Kokkos::resize
auto device_view_copy = device_view;
Kokkos::resize(device_view, 2 * n);
// Loop in reverse to increase likelihood of missing fence detection assuming
// that resize copies values in order.
for (int i = 2 * n - 1; i >= 0; --i) device_view(i) = i + 1;
Kokkos::fence();
// Check that Kokkos::resize completed before setting the values on the host
// manually (possibly because of missing fences).
for (int i = 0; i < 2 * n; ++i) ASSERT_EQ(device_view(i), i + 1);
}
template <typename SharedMemorySpace>
void test_dyn_rank_view_realloc() {
int n = 1000;
Kokkos::DynRankView<double, SharedMemorySpace> device_view("device view", n);
// Make sure we don't deallocate memory in Kokkos::realloc
auto device_view_copy = device_view;
Kokkos::realloc(device_view, 2 * n);
// Loop in reverse to increase likelihood of missing fence detection assuming
// that realloc sets values in order.
for (int i = 2 * n - 1; i >= 0; --i) device_view(i) = i + 1;
Kokkos::fence();
// Check that Kokkos::realloc completed before setting the values on the host
// manually (possibly because of missing fences).
for (int i = 0; i < 2 * n; ++i) ASSERT_EQ(device_view(i), i + 1);
}
#ifdef KOKKOS_HAS_SHARED_SPACE
TEST(TEST_CATEGORY, dyn_rank_view_check_fence_resize_realloc) {
if constexpr (std::is_same_v<TEST_EXECSPACE, Kokkos::DefaultExecutionSpace>) {
test_dyn_rank_view_resize<Kokkos::SharedSpace>();
test_dyn_rank_view_realloc<Kokkos::SharedSpace>();
} else {
GTEST_SKIP() << "skipping since not default execution space";
}
}
#endif
} // namespace Test
|
