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 <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#include <Kokkos_Core.hpp>
#endif
#include <cstddef>
namespace {
struct Foo {
KOKKOS_FUNCTION Foo& operator=(double val_) {
val = val_;
return *this;
}
float val;
KOKKOS_FUNCTION bool operator==(double val_) {
return val_ == static_cast<double>(val);
}
};
template <class D1, class D2, class... Extents>
void test_deep_copy_assignable_types(Extents... exts) {
Kokkos::View<D1, TEST_EXECSPACE> a("A", exts...);
Kokkos::deep_copy(a, 1.5);
Kokkos::View<D2, TEST_EXECSPACE> b("B", exts...);
Kokkos::deep_copy(b, a);
auto h_b = Kokkos::create_mirror_view(b);
Kokkos::deep_copy(h_b, b);
ASSERT_TRUE(h_b((exts - 1)...) == 1.5);
// Check that
Kokkos::deep_copy(TEST_EXECSPACE(), a, 2.5);
Kokkos::deep_copy(TEST_EXECSPACE(), b, a);
Kokkos::deep_copy(TEST_EXECSPACE(), h_b, b);
TEST_EXECSPACE().fence();
ASSERT_TRUE(h_b((exts - 1)...) == 2.5);
#ifdef KOKKOS_HAS_SHARED_SPACE
Kokkos::View<D1, Kokkos::SharedSpace> s("S", exts...);
Kokkos::deep_copy(s, 1.5);
Kokkos::deep_copy(b, s);
Kokkos::deep_copy(h_b, b);
ASSERT_TRUE(h_b((exts - 1)...) == 1.5);
#endif
#ifdef KOKKOS_HAS_SHARED_HOST_PINNED_SPACE
Kokkos::View<D1, Kokkos::SharedHostPinnedSpace> sp("S", exts...);
Kokkos::deep_copy(sp, 2.5);
Kokkos::deep_copy(b, sp);
Kokkos::deep_copy(h_b, b);
ASSERT_TRUE(h_b((exts - 1)...) == 2.5);
#endif
}
} // namespace
|
