DenseVector/Morpheus_Copy_Impl.hpp

 
#ifndef MORPHEUS_DENSEVECTOR_COPY_IMPL_HPP
#define MORPHEUS_DENSEVECTOR_COPY_IMPL_HPP
 
#include <Morpheus_FormatTags.hpp>
 
#include <impl/Morpheus_Utils.hpp>
#include <impl/DenseVector/Serial/Morpheus_Copy_Impl.hpp>
#include <impl/DenseVector/OpenMP/Morpheus_Copy_Impl.hpp>
#include <impl/DenseVector/Cuda/Morpheus_Copy_Impl.hpp>
#include <impl/DenseVector/HIP/Morpheus_Copy_Impl.hpp>
 
#include <Kokkos_Core.hpp>
 
#include <utility>  // std::pair
 
namespace Morpheus {
namespace Impl {
 
template <typename SourceType, typename DestinationType>
void copy(const SourceType& src, DestinationType& dst,
          typename std::enable_if_t<
              Morpheus::is_dense_vector_format_container_v<SourceType> &&
              Morpheus::is_dense_vector_format_container_v<DestinationType>>* =
              nullptr) {
  MORPHEUS_ASSERT(
      dst.size() == src.size(),
      "Destination vector must be of equal size to the source vector");
 
  // Kokkos has src and dst the other way round
  Kokkos::deep_copy(dst.view(), src.const_view());
}
 
template <typename SourceType, typename DestinationType>
void copy(const SourceType& src, DestinationType& dst,
          const typename SourceType::size_type src_begin,
          const typename SourceType::size_type src_end,
          const typename DestinationType::size_type dst_begin,
          const typename DestinationType::size_type dst_end,
          typename std::enable_if_t<
              Morpheus::is_dense_vector_format_container_v<SourceType> &&
              Morpheus::is_dense_vector_format_container_v<DestinationType>>* =
              nullptr) {
  auto src_sub =
      Kokkos::subview(src.const_view(), std::make_pair(src_begin, src_end));
  auto dst_sub =
      Kokkos::subview(dst.view(), std::make_pair(dst_begin, dst_end));
 
  // Kokkos has src and dst the other way round
  Kokkos::deep_copy(dst_sub, src_sub);
}
 
}  // namespace Impl
}  // namespace Morpheus
 
#endif  // MORPHEUS_DENSEVECTOR_COPY_IMPL_HPP