123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217 |
- /*
- [auto_generated]
- boost/numeric/odeint/external/thrust/thrust_algebra.hpp
- [begin_description]
- An algebra for thrusts device_vectors.
- [end_description]
- Copyright 2010-2013 Mario Mulansky
- Copyright 2010-2011 Karsten Ahnert
- Copyright 2013 Kyle Lutz
- 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_ALGEBRA_HPP_INCLUDED
- #define BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_ALGEBRA_HPP_INCLUDED
- #include <thrust/device_vector.h>
- #include <thrust/for_each.h>
- #include <thrust/iterator/zip_iterator.h>
- #include <boost/range.hpp>
- namespace boost {
- namespace numeric {
- namespace odeint {
- namespace detail {
- // to use in thrust::reduce
- template< class Value >
- struct maximum
- {
- template< class Fac1 , class Fac2 >
- __host__ __device__
- Value operator()( const Fac1 t1 , const Fac2 t2 ) const
- {
- return ( abs( t1 ) < abs( t2 ) ) ? t2 : t1 ;
- }
- typedef Value result_type;
- };
- }
- /** ToDO extend until for_each14 for rk78 */
- /*
- * The const versions are needed for boost.range to work, i.e.
- * it allows you to do
- * for_each1( make_pair( vec1.begin() , vec1.begin() + 10 ) , op );
- */
- struct thrust_algebra
- {
- template< class StateType , class Operation >
- static void for_each1( StateType &s , Operation op )
- {
- thrust::for_each( boost::begin(s) , boost::end(s) , op );
- }
- template< class StateType1 , class StateType2 , class Operation >
- static void for_each2( StateType1 &s1 , StateType2 &s2 , Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ) ) ,
- op);
- }
- template< class StateType1 , class StateType2 , class StateType3 , class Operation >
- static void for_each3( StateType1 &s1 , StateType2 &s2 , StateType3 &s3 , Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ,
- boost::begin(s3) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ,
- boost::end(s3) ) ) ,
- op);
- }
- template< class StateType1 , class StateType2 , class StateType3 , class StateType4 ,
- class Operation >
- static void for_each4( StateType1 &s1 , StateType2 &s2 , StateType3 &s3 , StateType4 &s4 ,
- Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ,
- boost::begin(s3) ,
- boost::begin(s4) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ,
- boost::end(s3) ,
- boost::end(s4) ) ) ,
- op);
- }
- template< class StateType1 , class StateType2 , class StateType3 ,
- class StateType4 , class StateType5 ,class Operation >
- static void for_each5( StateType1 &s1 , StateType2 &s2 , StateType3 &s3 , StateType4 &s4 ,
- StateType5 &s5 , Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ,
- boost::begin(s3) ,
- boost::begin(s4) ,
- boost::begin(s5) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ,
- boost::end(s3) ,
- boost::end(s4) ,
- boost::end(s5) ) ) ,
- op);
- }
- template< class StateType1 , class StateType2 , class StateType3 ,
- class StateType4 , class StateType5 , class StateType6 , class Operation >
- static void for_each6( StateType1 &s1 , StateType2 &s2 , StateType3 &s3 , StateType4 &s4 ,
- StateType5 &s5 , StateType6 &s6 , Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ,
- boost::begin(s3) ,
- boost::begin(s4) ,
- boost::begin(s5) ,
- boost::begin(s6) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ,
- boost::end(s3) ,
- boost::end(s4) ,
- boost::end(s5) ,
- boost::end(s6) ) ) ,
- op);
- }
- template< class StateType1 , class StateType2 , class StateType3 , class StateType4 ,
- class StateType5 , class StateType6 , class StateType7 , class Operation >
- static void for_each7( StateType1 &s1 , StateType2 &s2 , StateType3 &s3 , StateType4 &s4 ,
- StateType5 &s5 , StateType6 &s6 , StateType7 &s7 , Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ,
- boost::begin(s3) ,
- boost::begin(s4) ,
- boost::begin(s5) ,
- boost::begin(s6) ,
- boost::begin(s7) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ,
- boost::end(s3) ,
- boost::end(s4) ,
- boost::end(s5) ,
- boost::end(s6) ,
- boost::end(s7) ) ) ,
- op);
- }
- template< class StateType1 , class StateType2 , class StateType3 , class StateType4 ,
- class StateType5 , class StateType6 , class StateType7 , class StateType8 , class Operation >
- static void for_each8( StateType1 &s1 , StateType2 &s2 , StateType3 &s3 , StateType4 &s4 ,
- StateType5 &s5 , StateType6 &s6 , StateType7 &s7 , StateType8 &s8 , Operation op )
- {
- thrust::for_each(
- thrust::make_zip_iterator( thrust::make_tuple( boost::begin(s1) ,
- boost::begin(s2) ,
- boost::begin(s3) ,
- boost::begin(s4) ,
- boost::begin(s5) ,
- boost::begin(s6) ,
- boost::begin(s7) ,
- boost::begin(s8) ) ) ,
- thrust::make_zip_iterator( thrust::make_tuple( boost::end(s1) ,
- boost::end(s2) ,
- boost::end(s3) ,
- boost::end(s4) ,
- boost::end(s5) ,
- boost::end(s6) ,
- boost::end(s7) ,
- boost::end(s8) ) ) ,
- op);
- }
- template< class S >
- static typename S::value_type norm_inf( const S &s )
- {
- typedef typename S::value_type value_type;
- return thrust::reduce( boost::begin( s ) , boost::end( s ) ,
- static_cast<value_type>(0) ,
- detail::maximum<value_type>() );
- }
- };
- } // odeint
- } // numeric
- } // boost
- #endif // BOOST_NUMERIC_ODEINT_EXTERNAL_THRUST_THRUST_ALGEBRA_HPP_INCLUDED
|