DenseMatrix/Morpheus_Copy_Impl.hpp

 
#ifndef MORPHEUS_DENSEMATRIX_COPY_IMPL_HPP
#define MORPHEUS_DENSEMATRIX_COPY_IMPL_HPP
 
#include <Morpheus_FormatTags.hpp>
 
#include <impl/Morpheus_Utils.hpp>
#include <Kokkos_Core.hpp>
 
namespace Morpheus {
namespace Impl {
 
template <typename SourceType, typename DestinationType>
void copy(const SourceType& src, DestinationType& dst,
          typename std::enable_if_t<
              Morpheus::is_dense_matrix_format_container_v<SourceType> &&
              Morpheus::is_dense_matrix_format_container_v<DestinationType>>* =
              nullptr) {
  MORPHEUS_ASSERT(
      (dst.nrows() == src.nrows()) && (dst.ncols() == src.ncols()),
      "Destination matrix must have equal shape to the source matrix");
 
  // Kokkos has src and dst the other way round
  Kokkos::deep_copy(dst.view(), src.const_view());
}
 
}  // namespace Impl
}  // namespace Morpheus
 
#endif  // MORPHEUS_DENSEMATRIX_COPY_IMPL_HPP