52 #ifndef AMESOS2_KOKKOS_VIEW_COPY_ASSIGN_HPP
53 #define AMESOS2_KOKKOS_VIEW_COPY_ASSIGN_HPP
58 template<
class dst_t,
class src_t>
59 typename std::enable_if<static_cast<int>(dst_t::rank) == 1>::type
60 update_dst_size(dst_t & dst,
const src_t & src) {
61 if(dst.extent(0) != src.extent(0)) {
62 dst = dst_t(Kokkos::ViewAllocateWithoutInitializing(
"dst"),
67 template<
class dst_t,
class src_t>
68 typename std::enable_if<static_cast<int>(dst_t::rank) == 2>::type
69 update_dst_size(dst_t & dst,
const src_t & src) {
70 if(dst.extent(0) != src.extent(0) || dst.extent(1) != src.extent(1)) {
71 dst = dst_t(Kokkos::ViewAllocateWithoutInitializing(
"dst"),
72 src.extent(0), src.extent(1));
83 template<
class dst_t,
class src_t>
84 typename std::enable_if<std::is_same<
typename dst_t::value_type,
85 typename src_t::value_type>::value>::type
86 implement_copy_or_assign_same_mem_check_types(
87 [[maybe_unused]]
bool bInitialize,
105 template<
class dst_t,
class src_t>
106 void deep_copy_only(
bool bInitialize, dst_t & dst,
const src_t & src,
bool & bAssigned) {
107 update_dst_size(dst, src);
109 Kokkos::deep_copy(dst, src);
114 template<
class dst_t,
class src_t>
115 void deep_copy_only(dst_t & dst,
const src_t & src) {
117 deep_copy_only(
true, dst, src, bAssigned);
128 template<
class dst_t,
class src_t>
129 typename std::enable_if<!std::is_same<
typename dst_t::value_type,
130 typename src_t::value_type>::value>::type
131 implement_copy_or_assign_same_mem_check_types(
bool bInitialize, dst_t & dst,
const src_t & src,
bool & bAssigned) {
132 update_dst_size(dst, src);
134 Kokkos::deep_copy(dst, src);
143 template<
class dst_t,
class src_t>
144 typename std::enable_if<std::is_same<
typename dst_t::memory_space,
145 typename src_t::memory_space>::value>::type
146 deep_copy_or_assign_view(
bool bInitialize, dst_t & dst,
const src_t & src,
bool & bAssigned) {
147 implement_copy_or_assign_same_mem_check_types(bInitialize, dst, src, bAssigned);
152 template<
class dst_t,
class src_t>
153 typename std::enable_if<std::is_same<
typename dst_t::memory_space,
154 typename src_t::memory_space>::value>::type
155 deep_copy_or_assign_view(dst_t & dst,
const src_t & src) {
157 implement_copy_or_assign_same_mem_check_types(
true, dst, src, bAssigned);
160 template<
class dst_t,
class src_t>
161 typename std::enable_if<std::is_same<
typename dst_t::value_type,
162 typename src_t::value_type>::value>::type
163 implement_copy_or_assign_diff_mem_check_types(
bool bInitialize, dst_t & dst,
const src_t & src,
bool & bAssigned) {
164 update_dst_size(dst, src);
166 Kokkos::deep_copy(dst, src);
171 template<
class dst_t,
class src_t>
172 typename std::enable_if<static_cast<int>(dst_t::rank) == 1>::type
173 implement_copy_or_assign_diff_mem_diff_types_check_dim(dst_t & dst,
const src_t & src) {
174 Kokkos::View<typename dst_t::value_type*, typename src_t::execution_space>
175 intermediate(Kokkos::ViewAllocateWithoutInitializing(
"intermediate"), src.extent(0));
176 Kokkos::deep_copy(intermediate, src);
177 Kokkos::deep_copy(dst, intermediate);
180 template<
class dst_t,
class src_t>
181 typename std::enable_if<static_cast<int>(dst_t::rank) == 2>::type
182 implement_copy_or_assign_diff_mem_diff_types_check_dim(dst_t & dst,
const src_t & src) {
183 Kokkos::View<typename dst_t::value_type**, Kokkos::LayoutLeft, typename src_t::execution_space>
184 intermediate(Kokkos::ViewAllocateWithoutInitializing(
"intermediate"), src.extent(0), src.extent(1));
185 Kokkos::deep_copy(intermediate, src);
186 Kokkos::deep_copy(dst, intermediate);
189 template<
class dst_t,
class src_t>
190 typename std::enable_if<!std::is_same<
typename dst_t::value_type,
191 typename src_t::value_type>::value>::type
192 implement_copy_or_assign_diff_mem_check_types(
bool bInitialize, dst_t & dst,
const src_t & src,
bool & bAssigned) {
193 update_dst_size(dst, src);
198 implement_copy_or_assign_diff_mem_diff_types_check_dim(dst, src);
202 template<
class dst_t,
class src_t>
203 typename std::enable_if<!std::is_same<
typename dst_t::memory_space,
204 typename src_t::memory_space>::value>::type
205 deep_copy_or_assign_view(
bool bInitialize, dst_t & dst,
const src_t & src,
bool & bAssigned) {
206 implement_copy_or_assign_diff_mem_check_types(bInitialize, dst, src, bAssigned);
211 template<
class dst_t,
class src_t>
212 typename std::enable_if<!std::is_same<
typename dst_t::memory_space,
213 typename src_t::memory_space>::value>::type
214 deep_copy_or_assign_view(dst_t & dst,
const src_t & src) {
216 implement_copy_or_assign_diff_mem_check_types(
true, dst, src, bAssigned);
221 #endif // AMESOS2_KOKKOS_VIEW_COPY_ASSIGN_HPP