typedefs.h

Go to the documentation of this file.

#ifndef _DG_TYPEDEFS_CUH_
#define _DG_TYPEDEFS_CUH_
#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include "sparseblockmat.h"
#include "sparsematrix.h"
 
namespace dg{
 
//vectors
template<class T>
using HVec_t  = thrust::host_vector<T>; 
using HVec  = thrust::host_vector<double>; 
using cHVec = thrust::host_vector<thrust::complex<double>>; 
using iHVec = thrust::host_vector<int>; 
using fHVec = thrust::host_vector<float>; 
 
using DVec  = thrust::device_vector<double>; 
using iDVec = thrust::device_vector<int>; 
using cDVec = thrust::device_vector<thrust::complex<double>>; 
using fDVec = thrust::device_vector<float>; 
 
//derivative matrices
template<class T>
using HMatrix_t = EllSparseBlockMat<T, thrust::host_vector>;
using HMatrix = EllSparseBlockMat<double, thrust::host_vector>; 
using fHMatrix = EllSparseBlockMat<float, thrust::host_vector>; 
using DMatrix = EllSparseBlockMat<double, thrust::device_vector>; 
using fDMatrix = EllSparseBlockMat<float, thrust::device_vector>; 
 
// Interpolation matrices
template<class real_type>
using IHMatrix_t = dg::SparseMatrix<int, real_type, thrust::host_vector>;
template<class real_type>
#if THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_CUDA
//Ell matrix can be almost 3x faster than csr for GPU
//However, sometimes matrices contain outlier rows that do not fit in ell
using IDMatrix_t = dg::SparseMatrix<int, real_type, thrust::device_vector>;
#else
// csr matrix can be much faster than ell for CPU (we have our own symv implementation!)
using IDMatrix_t = dg::SparseMatrix<int, real_type, thrust::device_vector>;
#endif
using IHMatrix = IHMatrix_t<double>;
using IDMatrix = IDMatrix_t<double>;
 
}//namespace dg
 
#ifdef MPI_VERSION
#include "mpi_vector.h"
#include "mpi_matrix.h"
 
namespace dg{
//using MPI_Vector<thrust::device_vector<double> >  MDVec; //!< MPI Device Vector s.a. dg::DVec
template<class T>
using MHVec_t   = dg::MPI_Vector<dg::HVec_t<T> >; 
using MHVec     = dg::MPI_Vector<dg::HVec >; 
using cMHVec    = dg::MPI_Vector<dg::cHVec >; 
using fMHVec    = dg::MPI_Vector<dg::fHVec >; 
using MDVec     = dg::MPI_Vector<dg::DVec >; 
using cMDVec    = dg::MPI_Vector<dg::cDVec >; 
using fMDVec    = dg::MPI_Vector<dg::fDVec >; 
 
// TODO These may be private
template< class T>
using CooMat_t  = dg::CooSparseBlockMat<T, thrust::host_vector>;
using CooMat    = dg::CooSparseBlockMat<double, thrust::host_vector>;
using fCooMat   = dg::CooSparseBlockMat<float, thrust::host_vector>;
using DCooMat   = dg::CooSparseBlockMat<double, thrust::device_vector>;
using fDCooMat  = dg::CooSparseBlockMat<float, thrust::device_vector>;
 
template<class T>
using MHMatrix_t  = dg::MPISparseBlockMat<thrust::host_vector, dg::HMatrix_t<T>, dg::CooMat_t<T>>; 
using MHMatrix    = dg::MPISparseBlockMat<thrust::host_vector, dg::HMatrix, dg::CooMat>; 
using fMHMatrix   = dg::MPISparseBlockMat<thrust::host_vector, dg::fHMatrix, dg::fCooMat>; 
using MDMatrix    = dg::MPISparseBlockMat<thrust::device_vector, dg::DMatrix, dg::DCooMat>; 
using fMDMatrix   = dg::MPISparseBlockMat<thrust::device_vector, dg::fDMatrix, dg::fDCooMat>; 
 
template<class real_type>
using MIHMatrix_t = MPIDistMat< thrust::host_vector, IHMatrix_t<real_type> >;
template<class real_type>
using MIDMatrix_t = MPIDistMat< thrust::device_vector, IDMatrix_t<real_type> >;
using MIHMatrix = MIHMatrix_t<double>;
using MIDMatrix = MIDMatrix_t<double>;
 
 
}//namespace dg
#endif //MPI_VERSION
 
//MPI-independent definitions
namespace dg{
//vectors
namespace x{
#ifdef MPI_VERSION
using HVec  = MHVec;
using cHVec = cMHVec;
using fHVec = fMHVec;
 
using DVec  = MDVec;
using cDVec = cMDVec;
using fDVec = fMDVec;
 
//derivative matrices
using HMatrix = MHMatrix;
using fHMatrix = fMHMatrix;
using DMatrix = MDMatrix;
using fDMatrix = fMDMatrix;
//interpolation matrices
using IHMatrix = MIHMatrix;
using IDMatrix = MIDMatrix;
#else
using HVec  = HVec;
using cHVec = cHVec;
using fHVec = fHVec;
 
using DVec  = DVec;
using cDVec = cDVec;
using fDVec = fDVec;
 
//derivative matrices
using HMatrix = HMatrix;
using fHMatrix = fHMatrix;
using DMatrix = DMatrix;
using fDMatrix = fDMatrix;
//interpolation matrices
using IHMatrix = IHMatrix;
using IDMatrix = IDMatrix;
#endif //MPI_VERSION
}//namespace x
}//namespace dg
 
#ifdef MPI_VERSION
#define DG_RANK0 if(rank==0)
#else //MPI_VERSION
#define DG_RANK0
#endif //MPI_VERSION
 
#endif//_DG_TYPEDEFS_CUH_