123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118 |
- #ifndef BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_OPERATIONS_DISPATCHER_HPP_DEFINED
- #define BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_OPERATIONS_DISPATCHER_HPP_DEFINED
- #include <thrust/host_vector.h>
- #include <thrust/device_vector.h>
- #include <boost/numeric/odeint/external/thrust/thrust_operations.hpp>
- #include <boost/numeric/odeint/algebra/operations_dispatcher.hpp>
- namespace boost {
- namespace numeric {
- namespace odeint {
- template< class T , class A >
- struct operations_dispatcher< thrust::host_vector< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- template< class T , class A >
- struct operations_dispatcher< thrust::device_vector< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- }
- }
- }
- #include <thrust/version.h>
- #if THRUST_VERSION >= 101000
- #include <thrust/detail/vector_base.h>
- namespace boost { namespace numeric { namespace odeint {
- template< class T , class A >
- struct operations_dispatcher< thrust::detail::vector_base< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- } } }
- #elif THRUST_VERSION >= 100600
- #include <thrust/system/cpp/vector.h>
- namespace boost { namespace numeric { namespace odeint {
- template< class T , class A >
- struct operations_dispatcher< thrust::cpp::vector< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- } } }
- #ifdef _OPENMP
- #include <thrust/system/omp/vector.h>
- namespace boost { namespace numeric { namespace odeint {
- template< class T , class A >
- struct operations_dispatcher< thrust::omp::vector< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- } } }
- #endif
- #ifdef TBB_VERSION_MAJOR
- #include <thrust/system/tbb/vector.h>
- namespace boost { namespace numeric { namespace odeint {
- template< class T , class A >
- struct operations_dispatcher< thrust::tbb::vector< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- } } }
- #endif
- #ifdef __CUDACC__
- #include <thrust/system/cuda/vector.h>
- namespace boost { namespace numeric { namespace odeint {
- template< class T , class A >
- struct operations_dispatcher< thrust::cuda::vector< T , A > >
- {
- typedef thrust_operations operations_type;
- };
- } } }
- #endif
- #endif
- #endif
|