186 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
		
		
			
		
	
	
			186 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
|   | /* | ||
|  |  [auto_generated] | ||
|  |  boost/numeric/odeint/external/thrust/thrust_resize.hpp | ||
|  | 
 | ||
|  |  [begin_description] | ||
|  |  Enable resizing for thrusts device and host_vector. | ||
|  |  [end_description] | ||
|  | 
 | ||
|  |  Copyright 2010-2014 Mario Mulansky | ||
|  |  Copyright 2010-2011 Karsten Ahnert | ||
|  | 
 | ||
|  |  Distributed under the Boost Software License, Version 1.0. | ||
|  |  (See accompanying file LICENSE_1_0.txt or | ||
|  |  copy at http://www.boost.org/LICENSE_1_0.txt) | ||
|  |  */ | ||
|  | 
 | ||
|  | 
 | ||
|  | #ifndef BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_RESIZE_HPP_INCLUDED | ||
|  | #define BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_RESIZE_HPP_INCLUDED | ||
|  | 
 | ||
|  | #include <boost/range.hpp> | ||
|  | 
 | ||
|  | #include <thrust/device_vector.h> | ||
|  | #include <thrust/host_vector.h> | ||
|  | #include <thrust/distance.h> | ||
|  | 
 | ||
|  | #include <boost/numeric/odeint/util/resize.hpp> | ||
|  | #include <boost/numeric/odeint/util/same_size.hpp> | ||
|  | #include <boost/numeric/odeint/util/copy.hpp> | ||
|  | 
 | ||
|  | namespace boost { | ||
|  | namespace numeric { | ||
|  | namespace odeint { | ||
|  | 
 | ||
|  | // some macros that define the necessary utilities | ||
|  | 
 | ||
|  | #define ODEINT_THRUST_VECTOR_IS_RESIZEABLE( THRUST_VECTOR ) \ | ||
|  | template< class T , class A >                               \ | ||
|  | struct is_resizeable< THRUST_VECTOR<T,A> >                  \ | ||
|  | {                                                           \ | ||
|  |     struct type : public boost::true_type { };              \ | ||
|  |     const static bool value = type::value;                  \ | ||
|  | };                                                          \ | ||
|  | 
 | ||
|  | #define ODEINT_TRHUST_VECTOR_RESIZE_IMPL( THRUST_VECTOR )     \ | ||
|  | template< class T, class A >                                  \ | ||
|  | struct resize_impl< THRUST_VECTOR<T,A> , THRUST_VECTOR<T,A> > \ | ||
|  | {                                                             \ | ||
|  |     static void resize( THRUST_VECTOR<T,A> &x ,               \ | ||
|  |                         const THRUST_VECTOR<T,A> &y )         \ | ||
|  |     {                                                         \ | ||
|  |         x.resize( y.size() );                                 \ | ||
|  |     }                                                         \ | ||
|  | };                                                            \ | ||
|  | template< class T, class A, typename Range >                  \ | ||
|  | struct resize_impl< THRUST_VECTOR<T,A> , Range >              \ | ||
|  | {                                                             \ | ||
|  |     static void resize( THRUST_VECTOR<T,A> &x ,               \ | ||
|  |                         const Range &y )                      \ | ||
|  |     {                                                         \ | ||
|  |         x.resize( thrust::distance(boost::begin(y),           \ | ||
|  |                                    boost::end(y)));           \ | ||
|  |     }                                                         \ | ||
|  | };                                                            \ | ||
|  | 
 | ||
|  | 
 | ||
|  | #define ODEINT_THRUST_SAME_SIZE_IMPL( THRUST_VECTOR )            \ | ||
|  | template< class T , class A >                                    \ | ||
|  | struct same_size_impl< THRUST_VECTOR<T,A> , THRUST_VECTOR<T,A> > \ | ||
|  | {                                                                \ | ||
|  |     static bool same_size( const THRUST_VECTOR<T,A> &x ,         \ | ||
|  |                            const THRUST_VECTOR<T,A> &y )         \ | ||
|  |     {                                                            \ | ||
|  |         return x.size() == y.size();                             \ | ||
|  |     }                                                            \ | ||
|  | };                                                               \ | ||
|  | template< class T , class A, typename Range >                    \ | ||
|  | struct same_size_impl< THRUST_VECTOR<T,A> , Range >              \ | ||
|  | {                                                                \ | ||
|  |     static bool same_size( const THRUST_VECTOR<T,A> &x ,         \ | ||
|  |                            const Range &y )                      \ | ||
|  |     {                                                            \ | ||
|  |         return x.size() == thrust::distance(boost::begin(y),     \ | ||
|  |                                             boost::end(y));      \ | ||
|  |     }                                                            \ | ||
|  | };                                                               \ | ||
|  | 
 | ||
|  | 
 | ||
|  | #define ODEINT_THRUST_COPY_IMPL( THRUST_VECTOR )                        \ | ||
|  | template< class Container1 , class T , class A >                        \ | ||
|  | struct copy_impl< Container1 , THRUST_VECTOR<T,A> >                     \ | ||
|  | {                                                                       \ | ||
|  |     static void copy( const Container1 &from , THRUST_VECTOR<T,A> &to ) \ | ||
|  |     {                                                                   \ | ||
|  |         thrust::copy( boost::begin( from ) , boost::end( from ) ,       \ | ||
|  |                       boost::begin( to ) );                             \ | ||
|  |     }                                                                   \ | ||
|  | };                                                                      \ | ||
|  |                                                                         \ | ||
|  | template< class T , class A , class Container2 >                        \ | ||
|  | struct copy_impl< THRUST_VECTOR<T,A> , Container2 >                     \ | ||
|  | {                                                                       \ | ||
|  |     static void copy( const THRUST_VECTOR<T,A> &from , Container2 &to ) \ | ||
|  |     {                                                                   \ | ||
|  |         thrust::copy( boost::begin( from ) , boost::end( from ) ,       \ | ||
|  |                       boost::begin( to ) );                             \ | ||
|  |     }                                                                   \ | ||
|  | };                                                                      \ | ||
|  |                                                                         \ | ||
|  | template< class T , class A >                                           \ | ||
|  | struct copy_impl< THRUST_VECTOR<T,A> , THRUST_VECTOR<T,A> >             \ | ||
|  | {                                                                       \ | ||
|  |     static void copy( const THRUST_VECTOR<T,A> &from ,                  \ | ||
|  |                       THRUST_VECTOR<T,A> &to )                          \ | ||
|  |     {                                                                   \ | ||
|  |         thrust::copy( boost::begin( from ) , boost::end( from ) ,       \ | ||
|  |                       boost::begin( to ) );                             \ | ||
|  |     }                                                                   \ | ||
|  | };                                                                      \ | ||
|  | 
 | ||
|  | // add support for the standard thrust containers | ||
|  | 
 | ||
|  | ODEINT_THRUST_VECTOR_IS_RESIZEABLE( thrust::device_vector ) | ||
|  | ODEINT_TRHUST_VECTOR_RESIZE_IMPL( thrust::device_vector ) | ||
|  | ODEINT_THRUST_SAME_SIZE_IMPL( thrust::device_vector ) | ||
|  | ODEINT_THRUST_COPY_IMPL( thrust::device_vector ) | ||
|  | 
 | ||
|  | 
 | ||
|  | ODEINT_THRUST_VECTOR_IS_RESIZEABLE( thrust::host_vector ) | ||
|  | ODEINT_TRHUST_VECTOR_RESIZE_IMPL( thrust::host_vector ) | ||
|  | ODEINT_THRUST_SAME_SIZE_IMPL( thrust::host_vector ) | ||
|  | ODEINT_THRUST_COPY_IMPL( thrust::host_vector ) | ||
|  | 
 | ||
|  | 
 | ||
|  | } // odeint | ||
|  | } // numeric | ||
|  | } // boost | ||
|  | 
 | ||
|  | // add support for thrust backend vectors, if available | ||
|  | 
 | ||
|  | #include <thrust/version.h> | ||
|  | 
 | ||
|  | #if THRUST_VERSION >= 100600 | ||
|  | 
 | ||
|  | #include <thrust/system/cpp/vector.h> | ||
|  | namespace boost { namespace numeric { namespace odeint { | ||
|  |     ODEINT_THRUST_VECTOR_IS_RESIZEABLE( thrust::cpp::vector ) | ||
|  |     ODEINT_TRHUST_VECTOR_RESIZE_IMPL( thrust::cpp::vector ) | ||
|  |     ODEINT_THRUST_SAME_SIZE_IMPL( thrust::cpp::vector ) | ||
|  |     ODEINT_THRUST_COPY_IMPL( thrust::cpp::vector ) | ||
|  | } } } | ||
|  | 
 | ||
|  | #ifdef _OPENMP | ||
|  | #include <thrust/system/omp/vector.h> | ||
|  | namespace boost { namespace numeric { namespace odeint { | ||
|  |     ODEINT_THRUST_VECTOR_IS_RESIZEABLE( thrust::omp::vector ) | ||
|  |     ODEINT_TRHUST_VECTOR_RESIZE_IMPL( thrust::omp::vector ) | ||
|  |     ODEINT_THRUST_SAME_SIZE_IMPL( thrust::omp::vector ) | ||
|  |     ODEINT_THRUST_COPY_IMPL( thrust::omp::vector ) | ||
|  | } } } | ||
|  | #endif // _OPENMP | ||
|  | 
 | ||
|  | #ifdef TBB_VERSION_MAJOR | ||
|  | #include <thrust/system/tbb/vector.h> | ||
|  | namespace boost { namespace numeric { namespace odeint { | ||
|  |     ODEINT_THRUST_VECTOR_IS_RESIZEABLE( thrust::tbb::vector ) | ||
|  |     ODEINT_TRHUST_VECTOR_RESIZE_IMPL( thrust::tbb::vector ) | ||
|  |     ODEINT_THRUST_SAME_SIZE_IMPL( thrust::tbb::vector ) | ||
|  |     ODEINT_THRUST_COPY_IMPL( thrust::tbb::vector ) | ||
|  | } } } | ||
|  | #endif // TBB_VERSION_MAJOR | ||
|  | 
 | ||
|  | #ifdef __CUDACC__ | ||
|  | #include <thrust/system/cuda/vector.h> | ||
|  | namespace boost { namespace numeric { namespace odeint { | ||
|  |     ODEINT_THRUST_VECTOR_IS_RESIZEABLE( thrust::cuda::vector ) | ||
|  |     ODEINT_TRHUST_VECTOR_RESIZE_IMPL( thrust::cuda::vector ) | ||
|  |     ODEINT_THRUST_SAME_SIZE_IMPL( thrust::cuda::vector ) | ||
|  |     ODEINT_THRUST_COPY_IMPL( thrust::cuda::vector ) | ||
|  | } } } | ||
|  | #endif // __CUDACC__ | ||
|  | 
 | ||
|  | #endif // THRUST_VERSION >= 100600 | ||
|  | 
 | ||
|  | #endif // BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_RESIZE_HPP_INCLUDED |