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
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
#include <gtest/gtest.h>
#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#include <Kokkos_Core.hpp>
#endif
#include "include/ToolTestingUtilities.hpp"
TEST(kokkosp, create_mirror_no_init) {
using namespace Kokkos::Test::Tools;
listen_tool_events(Config::DisableAll(), Config::EnableKernels());
Kokkos::View<int*, Kokkos::DefaultExecutionSpace> device_view("device view",
10);
Kokkos::View<int*, Kokkos::HostSpace> host_view("host view", 10);
auto success = validate_absence(
[&]() {
auto mirror_device =
Kokkos::create_mirror(Kokkos::WithoutInitializing, device_view);
auto mirror_host = Kokkos::create_mirror(
Kokkos::WithoutInitializing, Kokkos::DefaultHostExecutionSpace{},
host_view);
auto mirror_device_view = Kokkos::create_mirror_view(
Kokkos::WithoutInitializing, device_view);
auto mirror_host_view = Kokkos::create_mirror_view(
Kokkos::WithoutInitializing, Kokkos::DefaultHostExecutionSpace{},
host_view);
},
[&](BeginParallelForEvent) {
return MatchDiagnostic{true, {"Found begin event"}};
},
[&](EndParallelForEvent) {
return MatchDiagnostic{true, {"Found end event"}};
});
ASSERT_TRUE(success);
}
TEST(kokkosp, create_mirror_no_init_view_ctor) {
using namespace Kokkos::Test::Tools;
listen_tool_events(Config::DisableAll(), Config::EnableKernels());
Kokkos::View<int*, Kokkos::DefaultExecutionSpace> device_view("device view",
10);
Kokkos::View<int*, Kokkos::HostSpace> host_view("host view", 10);
auto success = validate_absence(
[&]() {
auto mirror_device = Kokkos::create_mirror(
Kokkos::view_alloc(Kokkos::HostSpace{},
Kokkos::WithoutInitializing),
device_view);
auto mirror_host = Kokkos::create_mirror(
Kokkos::view_alloc(Kokkos::HostSpace{}, Kokkos::WithoutInitializing,
Kokkos::DefaultHostExecutionSpace{}),
host_view);
auto mirror_device_view = Kokkos::create_mirror_view(
Kokkos::view_alloc(Kokkos::HostSpace{},
Kokkos::WithoutInitializing),
device_view);
auto mirror_host_view = Kokkos::create_mirror_view(
Kokkos::view_alloc(Kokkos::HostSpace{}, Kokkos::WithoutInitializing,
Kokkos::DefaultHostExecutionSpace{}),
host_view);
mirror_host_view = Kokkos::create_mirror_view(
Kokkos::view_alloc(Kokkos::WithoutInitializing), host_view);
},
[&](BeginParallelForEvent) {
return MatchDiagnostic{true, {"Found begin event"}};
},
[&](EndParallelForEvent) {
return MatchDiagnostic{true, {"Found end event"}};
});
ASSERT_TRUE(success);
}
TEST(kokkosp, create_mirror_view_and_copy) {
#ifdef KOKKOS_ENABLE_OPENMPTARGET // FIXME_OPENMPTARGET
if (std::is_same<Kokkos::DefaultExecutionSpace,
Kokkos::Experimental::OpenMPTarget>::value)
GTEST_SKIP() << "skipping since the OpenMPTarget has unexpected fences";
#endif
#ifdef KOKKOS_ENABLE_CUDA
if (std::is_same_v<Kokkos::DefaultExecutionSpace::memory_space,
Kokkos::CudaUVMSpace>)
GTEST_SKIP()
<< "skipping since the CudaUVMSpace requires additional fences";
#endif
using namespace Kokkos::Test::Tools;
listen_tool_events(Config::DisableAll(), Config::EnableKernels(),
Config::EnableFences());
Kokkos::View<int*, Kokkos::DefaultExecutionSpace> device_view;
Kokkos::View<int*, Kokkos::HostSpace> host_view("host view", 10);
auto success = validate_absence(
[&]() {
auto mirror_device = Kokkos::create_mirror_view_and_copy(
Kokkos::view_alloc(
Kokkos::DefaultExecutionSpace{},
typename Kokkos::DefaultExecutionSpace::memory_space{}),
host_view);
// Avoid fences for deallocation when mirror_device goes out of scope.
device_view = mirror_device;
},
[&](BeginParallelForEvent) {
return MatchDiagnostic{true, {"Found parallel_for event"}};
},
[&](BeginFenceEvent) {
return MatchDiagnostic{true, {"Found fence event"}};
});
ASSERT_TRUE(success);
}
|
