44 #ifndef KOKKOS_CORE_CONCEPTS_HPP 45 #define KOKKOS_CORE_CONCEPTS_HPP 47 #include <type_traits> 50 #include <Kokkos_Core_fwd.hpp> 65 static_assert( std::is_same<T,Static>::value
66 || std::is_same<T,Dynamic>::value
67 ,
"Kokkos: Invalid Schedule<> type." 69 using schedule_type = Schedule ;
77 static_assert(std::is_integral<T>::value,
"Kokkos: Invalid IndexType<>.");
78 using index_type = IndexType ;
86 template<
unsigned int maxT = 0
87 ,
unsigned int minB = 0
93 static unsigned int constexpr maxTperB {maxT};
94 static unsigned int constexpr minBperSM {minB};
104 #define KOKKOS_IMPL_IS_CONCEPT( CONCEPT ) \ 105 template< typename T > struct is_ ## CONCEPT { \ 107 template< typename , typename = std::true_type > struct have : std::false_type {}; \ 108 template< typename U > struct have<U,typename std::is_same<U,typename U:: CONCEPT >::type> : std::true_type {}; \ 110 enum { value = is_ ## CONCEPT::template have<T>::value }; \ 115 KOKKOS_IMPL_IS_CONCEPT( memory_space )
116 KOKKOS_IMPL_IS_CONCEPT( memory_traits )
117 KOKKOS_IMPL_IS_CONCEPT( execution_space )
118 KOKKOS_IMPL_IS_CONCEPT( execution_policy )
119 KOKKOS_IMPL_IS_CONCEPT( array_layout )
120 KOKKOS_IMPL_IS_CONCEPT( reducer )
126 using Kokkos::is_memory_space ;
127 using Kokkos::is_memory_traits ;
128 using Kokkos::is_execution_space ;
129 using Kokkos::is_execution_policy ;
130 using Kokkos::is_array_layout ;
134 KOKKOS_IMPL_IS_CONCEPT( iteration_pattern )
135 KOKKOS_IMPL_IS_CONCEPT( schedule_type )
136 KOKKOS_IMPL_IS_CONCEPT( index_type )
137 KOKKOS_IMPL_IS_CONCEPT( launch_bounds )
141 #undef KOKKOS_IMPL_IS_CONCEPT 149 template<
class ExecutionSpace ,
class MemorySpace >
151 static_assert( Kokkos::is_execution_space<ExecutionSpace>::value
152 ,
"Execution space is not valid" );
153 static_assert( Kokkos::is_memory_space<MemorySpace>::value
154 ,
"Memory space is not valid" );
155 typedef ExecutionSpace execution_space;
156 typedef MemorySpace memory_space;
157 typedef Device<execution_space,memory_space> device_type;
161 template<
typename T >
165 template<
typename ,
typename =
void >
166 struct exe : std::false_type {
typedef void space ; };
168 template<
typename ,
typename =
void >
169 struct mem : std::false_type {
typedef void space ; };
171 template<
typename ,
typename =
void >
172 struct dev : std::false_type {
typedef void space ; };
174 template<
typename U >
175 struct exe<U,typename std::conditional<true,void,typename U::execution_space>::type>
176 : std::is_same<U,typename U::execution_space>::type
177 {
typedef typename U::execution_space space ; };
179 template<
typename U >
180 struct mem<U,typename std::conditional<true,void,typename U::memory_space>::type>
181 : std::is_same<U,typename U::memory_space>::type
182 {
typedef typename U::memory_space space ; };
184 template<
typename U >
185 struct dev<U,typename std::conditional<true,void,typename U::device_type>::type>
186 : std::is_same<U,typename U::device_type>::type
187 {
typedef typename U::device_type space ; };
189 typedef typename is_space::template exe<T> is_exe ;
190 typedef typename is_space::template mem<T> is_mem ;
191 typedef typename is_space::template dev<T> is_dev ;
195 enum { value = is_exe::value || is_mem::value || is_dev::value };
197 typedef typename is_exe::space execution_space ;
198 typedef typename is_mem::space memory_space ;
203 typedef typename std::conditional
204 < std::is_same< memory_space , Kokkos::HostSpace >::value
205 #if defined( KOKKOS_ENABLE_CUDA ) 206 || std::is_same< memory_space , Kokkos::CudaUVMSpace >::value
207 || std::is_same< memory_space , Kokkos::CudaHostPinnedSpace >::value
211 >::type host_memory_space ;
213 #if defined( KOKKOS_ENABLE_CUDA ) 214 typedef typename std::conditional
215 < std::is_same< execution_space , Kokkos::Cuda >::value
216 , Kokkos::DefaultHostExecutionSpace , execution_space
217 >::type host_execution_space ;
219 #if defined( KOKKOS_ENABLE_OPENMPTARGET ) 220 typedef typename std::conditional
221 < std::is_same< execution_space , Kokkos::Experimental::OpenMPTarget >::value
222 , Kokkos::DefaultHostExecutionSpace , execution_space
223 >::type host_execution_space ;
225 typedef execution_space host_execution_space ;
229 typedef typename std::conditional
230 < std::is_same< execution_space , host_execution_space >::value &&
231 std::is_same< memory_space , host_memory_space >::value
232 , T , Kokkos::Device< host_execution_space , host_memory_space >
233 >::type host_mirror_space ;
240 using Kokkos::is_space ;
256 template<
typename DstMemorySpace ,
typename SrcMemorySpace >
259 static_assert( Kokkos::is_memory_space< DstMemorySpace >::value &&
260 Kokkos::is_memory_space< SrcMemorySpace >::value
261 ,
"template arguments must be memory spaces" );
270 enum { assignable = std::is_same<DstMemorySpace,SrcMemorySpace>::value };
275 enum { accessible = assignable };
280 enum { deepcopy = assignable };
306 template<
typename AccessSpace ,
typename MemorySpace >
310 static_assert( Kokkos::is_space< AccessSpace >::value
311 ,
"template argument #1 must be a Kokkos space" );
313 static_assert( Kokkos::is_memory_space< MemorySpace >::value
314 ,
"template argument #2 must be a Kokkos memory space" );
319 <
typename AccessSpace::execution_space::memory_space
320 ,
typename AccessSpace::memory_space
322 ,
"template argument #1 is an invalid space" );
325 <
typename AccessSpace::execution_space::memory_space , MemorySpace >
329 <
typename AccessSpace::memory_space , MemorySpace >
339 enum { accessible = exe_access::accessible };
347 is_memory_space< AccessSpace >::value && mem_access::assignable };
350 enum { deepcopy = mem_access::deepcopy };
356 typedef typename std::conditional
357 < std::is_same<typename AccessSpace::memory_space,MemorySpace>::value ||
358 ! exe_access::accessible
360 , Kokkos::Device< typename AccessSpace::execution_space , MemorySpace >
375 #endif // KOKKOS_CORE_CONCEPTS_HPP Can AccessSpace access MemorySpace ?
Memory management for host memory.
Specify Launch Bounds for CUDA execution.
Access relationship between DstMemorySpace and SrcMemorySpace.