3 boost/numeric/odeint/external/viennacl_operations.hpp
9 Copyright 2012 Denis Demidov
10 Copyright 2012 Karsten Ahnert
11 Copyright 2012 Mario Mulansky
13 Distributed under the Boost Software License, Version 1.0.
14 (See accompanying file LICENSE_1_0.txt or
15 copy at http://www.boost.org/LICENSE_1_0.txt)
19 #ifndef BOOST_NUMERIC_ODEINT_EXTERNAL_VIENNACL_VIENNACL_OPERATIONS_HPP_INCLUDED
20 #define BOOST_NUMERIC_ODEINT_EXTERNAL_VIENNACL_VIENNACL_OPERATIONS_HPP_INCLUDED
22 #include <viennacl/vector.hpp>
24 #ifdef VIENNACL_WITH_OPENCL
25 # include <viennacl/generator/custom_operation.hpp>
33 #ifdef VIENNACL_WITH_OPENCL
34 struct viennacl_operations
37 template< class Fac1 = double , class Fac2 = Fac1 >
43 scale_sum2( Fac1 alpha1 , Fac2 alpha2 )
44 : m_alpha1( alpha1 ) , m_alpha2( alpha2 )
47 template< class T1 , class T2 , class T3 >
48 void operator()( viennacl::vector<T1> &v1 ,
49 const viennacl::vector<T2> &v2 ,
50 const viennacl::vector<T3> &v3
53 using namespace viennacl;
55 static generator::symbolic_vector <0, T1> sym_v1;
56 static generator::symbolic_vector <1, T2> sym_v2;
57 static generator::symbolic_vector <2, T3> sym_v3;
58 static generator::cpu_symbolic_scalar<3, Fac1> sym_a1;
59 static generator::cpu_symbolic_scalar<4, Fac2> sym_a2;
61 static generator::custom_operation op(
62 sym_v1 = sym_a1 * sym_v2
67 ocl::enqueue( op(v1, v2, v3, m_alpha1, m_alpha2) );
70 typedef void result_type;
74 template< class Fac1 = double , class Fac2 = Fac1 , class Fac3 = Fac2 >
81 scale_sum3( Fac1 alpha1 , Fac2 alpha2 , Fac3 alpha3 )
82 : m_alpha1( alpha1 ) , m_alpha2( alpha2 ) , m_alpha3( alpha3 )
85 template< class T1 , class T2 , class T3 , class T4 >
86 void operator()( viennacl::vector<T1> &v1 ,
87 const viennacl::vector<T2> &v2 ,
88 const viennacl::vector<T3> &v3 ,
89 const viennacl::vector<T4> &v4
92 using namespace viennacl;
94 static generator::symbolic_vector <0, T1> sym_v1;
95 static generator::symbolic_vector <1, T2> sym_v2;
96 static generator::symbolic_vector <2, T3> sym_v3;
97 static generator::symbolic_vector <3, T4> sym_v4;
98 static generator::cpu_symbolic_scalar<4, Fac1> sym_a1;
99 static generator::cpu_symbolic_scalar<5, Fac2> sym_a2;
100 static generator::cpu_symbolic_scalar<6, Fac3> sym_a3;
102 static generator::custom_operation op(
103 sym_v1 = sym_a1 * sym_v2
109 ocl::enqueue( op(v1, v2, v3, v4, m_alpha1, m_alpha2, m_alpha3) );
112 typedef void result_type;
116 template< class Fac1 = double , class Fac2 = Fac1 , class Fac3 = Fac2 , class Fac4 = Fac3 >
124 scale_sum4( Fac1 alpha1 , Fac2 alpha2 , Fac3 alpha3 , Fac4 alpha4 )
125 : m_alpha1( alpha1 ) , m_alpha2( alpha2 ) , m_alpha3( alpha3 ) , m_alpha4( alpha4 ) { }
127 template< class T1 , class T2 , class T3 , class T4 , class T5 >
128 void operator()( viennacl::vector<T1> &v1 ,
129 const viennacl::vector<T2> &v2 ,
130 const viennacl::vector<T3> &v3 ,
131 const viennacl::vector<T4> &v4 ,
132 const viennacl::vector<T5> &v5
135 using namespace viennacl;
137 static generator::symbolic_vector <0, T1> sym_v1;
138 static generator::symbolic_vector <1, T2> sym_v2;
139 static generator::symbolic_vector <2, T3> sym_v3;
140 static generator::symbolic_vector <3, T4> sym_v4;
141 static generator::symbolic_vector <4, T5> sym_v5;
142 static generator::cpu_symbolic_scalar<5, Fac1> sym_a1;
143 static generator::cpu_symbolic_scalar<6, Fac2> sym_a2;
144 static generator::cpu_symbolic_scalar<7, Fac3> sym_a3;
145 static generator::cpu_symbolic_scalar<8, Fac4> sym_a4;
147 static generator::custom_operation op(
148 sym_v1 = sym_a1 * sym_v2
155 ocl::enqueue( op(v1, v2, v3, v4, v5,
156 m_alpha1, m_alpha2, m_alpha3, m_alpha4) );
159 typedef void result_type;
163 template< class Fac1 = double , class Fac2 = Fac1 , class Fac3 = Fac2 , class Fac4 = Fac3 , class Fac5 = Fac4 >
172 scale_sum5( Fac1 alpha1 , Fac2 alpha2 , Fac3 alpha3 , Fac4 alpha4 , Fac5 alpha5 )
173 : m_alpha1( alpha1 ) , m_alpha2( alpha2 ) , m_alpha3( alpha3 ) , m_alpha4( alpha4 ) , m_alpha5( alpha5 ) { }
175 template< class T1 , class T2 , class T3 , class T4 , class T5 , class T6 >
176 void operator()( viennacl::vector<T1> &v1 ,
177 const viennacl::vector<T2> &v2 ,
178 const viennacl::vector<T3> &v3 ,
179 const viennacl::vector<T4> &v4 ,
180 const viennacl::vector<T5> &v5 ,
181 const viennacl::vector<T6> &v6
184 using namespace viennacl;
186 static generator::symbolic_vector < 0, T1> sym_v1;
187 static generator::symbolic_vector < 1, T2> sym_v2;
188 static generator::symbolic_vector < 2, T3> sym_v3;
189 static generator::symbolic_vector < 3, T4> sym_v4;
190 static generator::symbolic_vector < 4, T5> sym_v5;
191 static generator::symbolic_vector < 5, T6> sym_v6;
192 static generator::cpu_symbolic_scalar< 6, Fac1> sym_a1;
193 static generator::cpu_symbolic_scalar< 7, Fac2> sym_a2;
194 static generator::cpu_symbolic_scalar< 8, Fac3> sym_a3;
195 static generator::cpu_symbolic_scalar< 9, Fac4> sym_a4;
196 static generator::cpu_symbolic_scalar<10, Fac5> sym_a5;
198 static generator::custom_operation op(
199 sym_v1 = sym_a1 * sym_v2
207 ocl::enqueue( op(v1, v2, v3, v4, v5, v6,
208 m_alpha1, m_alpha2, m_alpha3, m_alpha4, m_alpha5) );
211 typedef void result_type;
217 struct viennacl_operations : public default_operations {};
226 #endif // BOOST_NUMERIC_ODEINT_EXTERNAL_VIENNACL_VIENNACL_OPERATIONS_HPP_INCLUDED