Home Browse Files 
      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

#ifndef HPC_CUDA_COPY_H
#define HPC_CUDA_COPY_H 1

#include <cassert>
#include <hpc/cuda/check.h>
#include <hpc/cuda/gematrix.h>
#include <hpc/cuda/densevector.h>
#include <hpc/matvec/gematrix.h>
#include <hpc/matvec/densevector.h>

namespace hpc { namespace cuda {

template<typename T, typename IndexX, typename IndexY>
void
copy(const hpc::matvec::DenseVector<T, IndexX>& x, DenseVector<T, IndexY>& y) {
   assert(x.length == y.length);
   CHECK_CUDA(cudaMemcpy, y.data, x.data, x.length * sizeof(T),
      cudaMemcpyHostToDevice);
}

template<typename T, typename IndexX, typename IndexY>
void
copy(const DenseVector<T, IndexX>& x, hpc::matvec::DenseVector<T, IndexY>& y) {
   assert(x.length == y.length);
   CHECK_CUDA(cudaMemcpy, y.data, x.data, x.length * sizeof(T),
      cudaMemcpyDeviceToHost);
}

template<typename T, typename IndexA, typename IndexB>
void
copy(const hpc::matvec::GeMatrix<T, IndexA>& A, GeMatrix<T, IndexB>& B) {
   assert(A.numRows == B.numRows && A.numCols == B.numCols &&
      A.incRow == B.incRow);
   CHECK_CUDA(cudaMemcpy, B.data, A.data, A.numRows * A.numCols * sizeof(T),
      cudaMemcpyHostToDevice);
}

template<typename T, typename IndexA, typename IndexB>
void
copy(const GeMatrix<T, IndexA>& A, hpc::matvec::GeMatrix<T, IndexB>& B) {
   assert(A.numRows == B.numRows && A.numCols == B.numCols &&
      A.incRow == B.incRow);
   CHECK_CUDA(cudaMemcpy, B.data, A.data, A.numRows * A.numCols * sizeof(T),
      cudaMemcpyDeviceToHost);
}

} } // namespaces cuda, hpc

#endif // HPC_CUDA_COPY_H





































































































Home