tensor_traits_thrust.h

Go to the documentation of this file.

#pragma once
 
#include <thrust/host_vector.h>
#include <thrust/device_vector.h>
#include <thrust/complex.h>
#include "vector_categories.h"
#include "tensor_traits.h"
#include "tensor_traits_scalar.h"
 
namespace dg
{
 
template<class T>
struct TensorTraits<thrust::host_vector<T>> //, std::enable_if_t< dg::is_scalar<T>::value>>
{
    using value_type        = T;
    using tensor_category   = ThrustVectorTag;
    using execution_policy  = SerialTag;
};
//template<class T>
//struct TensorTraits<thrust::host_vector<T>,
//    std::enable_if_t< !dg::is_scalar<T>::value>>
//{
//    using value_type        = get_value_type<T>;
//    using tensor_category   = RecursiveVectorTag;
//    using execution_policy  = get_execution_policy<T>;
//};
 
template<class T>
struct TensorTraits<thrust::device_vector<T> >
{
    using value_type        = T;
    using tensor_category   = ThrustVectorTag;
#if THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_CUDA
    using execution_policy  = CudaTag ;
#elif THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_OMP
    using execution_policy  = OmpTag ;
#else
    using execution_policy  = SerialTag ;
#endif
};
 
template<class T>
struct TensorTraits<thrust::complex<T>>
{
    using value_type        = thrust::complex<T>;
    using tensor_category   = ComplexTag;
    using execution_policy  = AnyPolicyTag;
};
//thrust::cpp is an alias for thrust::system::cpp
//tag is a class deriving from thrust::execution_policy<tag>
//raw pointers have the std::random_access_iterator_tag as iterator_category in thrust
template<class Tag>
struct ThrustTag { };
template <>
struct ThrustTag<SerialTag>
{
    using thrust_tag = thrust::cpp::tag;
};
#if THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_CUDA
template <>
struct ThrustTag<CudaTag>
{
    using thrust_tag = thrust::cuda::tag;
};
#elif THRUST_DEVICE_SYSTEM==THRUST_DEVICE_SYSTEM_OMP
template <>
struct ThrustTag<OmpTag>
{
    using thrust_tag = thrust::omp::tag;
};
#endif
template<class Vector>
using get_thrust_tag = typename ThrustTag<get_execution_policy<Vector>>::thrust_tag;
 
} //namespace dg