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
|
// 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.dyn_rank_view;
#else
#include <Kokkos_Core.hpp>
#include <Kokkos_DynRankView.hpp>
#endif
#include <KokkosExp_InterOp.hpp>
// View
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<Kokkos::View<double*>>,
Kokkos::View<double*,
typename Kokkos::DefaultExecutionSpace::array_layout,
typename Kokkos::DefaultExecutionSpace::memory_space>>,
"Error! Unexpected python_view_type for: View");
// DynRankView
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<Kokkos::DynRankView<double>>,
Kokkos::DynRankView<
double, typename Kokkos::DefaultExecutionSpace::array_layout,
typename Kokkos::DefaultExecutionSpace::memory_space>>,
"Error! Unexpected python_view_type for: DynRankView");
// View + Execution Space
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<
Kokkos::View<double*, Kokkos::DefaultExecutionSpace>>,
Kokkos::View<double*,
typename Kokkos::DefaultExecutionSpace::array_layout,
typename Kokkos::DefaultExecutionSpace::memory_space>>,
"Error! Unexpected python_view_type for: View + Execution Space");
// DynRankView + Execution Space
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<
Kokkos::DynRankView<double, Kokkos::DefaultExecutionSpace>>,
Kokkos::DynRankView<
double, typename Kokkos::DefaultExecutionSpace::array_layout,
typename Kokkos::DefaultExecutionSpace::memory_space>>,
"Error! Unexpected python_view_type for: DynRankView + Execution Space");
// View + Memory space
static_assert(std::is_same_v<Kokkos::Experimental::python_view_type_t<
Kokkos::View<int64_t*, Kokkos::HostSpace>>,
Kokkos::View<int64_t*, Kokkos::LayoutRight,
Kokkos::HostSpace>>,
"Error! Unexpected python_view_type for: View + Memory space");
// DynRankView + Memory space
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<
Kokkos::DynRankView<int16_t, Kokkos::HostSpace>>,
Kokkos::DynRankView<int16_t, Kokkos::LayoutRight, Kokkos::HostSpace>>,
"Error! Unexpected python_view_type for: DynRankView + Memory space");
// View + Layout + Execution space
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<Kokkos::View<
int**, Kokkos::LayoutLeft, Kokkos::DefaultExecutionSpace>>,
Kokkos::View<int**, Kokkos::LayoutLeft,
typename Kokkos::DefaultExecutionSpace::memory_space>>,
"Error! Unexpected python_view_type for: View + Layout + Execution space");
// DynRankView + Layout + Execution space
static_assert(
std::is_same_v<Kokkos::Experimental::python_view_type_t<Kokkos::DynRankView<
int, Kokkos::LayoutLeft, Kokkos::DefaultExecutionSpace>>,
Kokkos::DynRankView<
int, Kokkos::LayoutLeft,
typename Kokkos::DefaultExecutionSpace::memory_space>>,
"Error! Unexpected python_view_type for: DynRankView + Layout + Execution "
"space");
// View + Layout + Memory Space
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<
Kokkos::View<uint32_t**, Kokkos::LayoutLeft, Kokkos::HostSpace>>,
Kokkos::View<uint32_t**, Kokkos::LayoutLeft, Kokkos::HostSpace>>,
"Error! Unexpected python_view_type for: View + Layout + Memory Space");
// DynRankView + Layout + Memory Space
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<Kokkos::DynRankView<
uint64_t, Kokkos::LayoutLeft, Kokkos::HostSpace>>,
Kokkos::DynRankView<uint64_t, Kokkos::LayoutLeft, Kokkos::HostSpace>>,
"Error! Unexpected python_view_type for: DynRankView + Layout + Memory "
"Space");
// View + Layout + Execution space + Memory Trait
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<Kokkos::View<
float***, Kokkos::LayoutLeft, Kokkos::DefaultHostExecutionSpace,
Kokkos::MemoryTraits<Kokkos::RandomAccess>>>,
Kokkos::View<float***, Kokkos::LayoutLeft,
typename Kokkos::DefaultHostExecutionSpace::memory_space,
Kokkos::MemoryTraits<Kokkos::RandomAccess>>>,
"Error! Unexpected python_view_type for: View + Layout + Execution space + "
"Memory Trait");
// DynRankView + Layout + Execution space + Memory trait
static_assert(
std::is_same_v<
Kokkos::Experimental::python_view_type_t<Kokkos::DynRankView<
float, Kokkos::LayoutLeft, Kokkos::DefaultHostExecutionSpace,
Kokkos::MemoryTraits<Kokkos::Atomic>>>,
Kokkos::DynRankView<
float, Kokkos::LayoutLeft,
typename Kokkos::DefaultHostExecutionSpace::memory_space,
Kokkos::MemoryTraits<Kokkos::Atomic>>>,
"Error! Unexpected python_view_type for: DynRankView + Layout + Execution "
"space + Memory trait");
|
