trsv.hpp
Go to the documentation of this file.
1 /*!
2  *
3  *
4  * \brief Triangular solve kernel for vector expressions.
5  *
6  * \author O. Krause
7  * \date 2012
8  *
9  *
10  * \par Copyright 1995-2015 Shark Development Team
11  *
12  * <BR><HR>
13  * This file is part of Shark.
14  * <http://image.diku.dk/shark/>
15  *
16  * Shark is free software: you can redistribute it and/or modify
17  * it under the terms of the GNU Lesser General Public License as published
18  * by the Free Software Foundation, either version 3 of the License, or
19  * (at your option) any later version.
20  *
21  * Shark is distributed in the hope that it will be useful,
22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24  * GNU Lesser General Public License for more details.
25  *
26  * You should have received a copy of the GNU Lesser General Public License
27  * along with Shark. If not, see <http://www.gnu.org/licenses/>.
28  *
29  */
30 
31 #ifndef REMORA_KERNELS_TRSV_HPP
32 #define REMORA_KERNELS_TRSV_HPP
33 
34 #include <type_traits>
35 #ifdef REMORA_USE_CBLAS
36 #include "cblas/trsv.hpp"
37 #else
38 // if no bindings are included, we have to provide the default has_optimized_gemv
39 // otherwise the binding will take care of this
40 namespace remora {namespace bindings{
41 template<class M1, class M2>
42 struct has_optimized_trsv
43 : public std::false_type{};
44 }}
45 #endif
46 
47 #include "default/trsv.hpp"
48 
49 namespace remora{namespace kernels{
50 
51 ///\brief Implements the TRiangular Solver for Vectors.
52 ///
53 /// It solves Systems of the form Ax = b where A is a square lower or upper triangular matrix.
54 /// It can optionally assume that the diagonal is 1 and won't access the diagonal elements.
55 template <class Triangular, class Side, typename MatA, typename V>
56 void trsv(
57  matrix_expression<MatA, cpu_tag> const &A,
58  vector_expression<V, cpu_tag> &b
59 ){
60  REMORA_SIZE_CHECK(A().size1() == A().size2());
61  REMORA_SIZE_CHECK(A().size1() == b().size());
62 
63  bindings::trsv<Triangular, Side>(A,b,typename bindings::has_optimized_trsv<MatA, V>::type());
64 }
65 
66 }}
67 
68 #ifdef REMORA_USE_CLBLAST
69 #include "clBlast/trsv.hpp"
70 #elif REMORA_USE_GPU
71 #include "gpu/trsv.hpp"
72 #endif
73 
74 #endif