trtrs.h
Go to the documentation of this file.
1 //=================================================================================================
33 //=================================================================================================
34 
35 #ifndef _BLAZE_MATH_LAPACK_TRTRS_H_
36 #define _BLAZE_MATH_LAPACK_TRTRS_H_
37 
38 
39 //*************************************************************************************************
40 // Includes
41 //*************************************************************************************************
42 
43 #include <blaze/math/Aliases.h>
48 #include <blaze/math/Exception.h>
53 #include <blaze/util/Assert.h>
55 #include <blaze/util/NumericCast.h>
56 
57 
58 namespace blaze {
59 
60 //=================================================================================================
61 //
62 // LAPACK TRIANGULAR SUBSTITUTION FUNCTIONS (TRTRS)
63 //
64 //=================================================================================================
65 
66 //*************************************************************************************************
69 template< typename MT, bool SO, typename VT, bool TF >
70 inline void trtrs( const DenseMatrix<MT,SO>& A, DenseVector<VT,TF>& b,
71  char uplo, char trans, char diag );
72 
73 template< typename MT1, bool SO1, typename MT2, bool SO2 >
74 inline void trtrs( const DenseMatrix<MT1,SO1>& A, DenseMatrix<MT2,SO2>& B,
75  char uplo, char trans, char diag );
77 //*************************************************************************************************
78 
79 
80 //*************************************************************************************************
165 template< typename MT // Type of the system matrix
166  , bool SO // Storage order of the system matrix
167  , typename VT // Type of the right-hand side vector
168  , bool TF > // Transpose flag of the right-hand side vector
169 inline void trtrs( const DenseMatrix<MT,SO>& A, DenseVector<VT,TF>& b, char uplo, char trans, char diag )
170 {
175 
176  if( !isSquare( ~A ) ) {
177  BLAZE_THROW_INVALID_ARGUMENT( "Invalid non-square matrix provided" );
178  }
179 
180  if( (~b).size() != (~A).rows() ) {
181  BLAZE_THROW_INVALID_ARGUMENT( "Invalid right-hand side vector provided" );
182  }
183 
184  if( uplo != 'L' && uplo != 'U' ) {
185  BLAZE_THROW_INVALID_ARGUMENT( "Invalid uplo argument provided" );
186  }
187 
188  if( trans != 'N' && trans != 'T' && trans != 'C' ) {
189  BLAZE_THROW_INVALID_ARGUMENT( "Invalid trans argument provided" );
190  }
191 
192  if( diag != 'U' && diag != 'N' ) {
193  BLAZE_THROW_INVALID_ARGUMENT( "Invalid diag argument provided" );
194  }
195 
196  int n ( numeric_cast<int>( (~A).rows() ) );
197  int nrhs( 1 );
198  int lda ( numeric_cast<int>( (~A).spacing() ) );
199  int ldb ( numeric_cast<int>( (~b).size() ) );
200  int info( 0 );
201 
202  if( n == 0 ) {
203  return;
204  }
205 
207  ( uplo == 'L' )?( uplo = 'U' ):( uplo = 'L' );
208  }
209 
210  trtrs( uplo, trans, diag, n, nrhs, (~A).data(), lda, (~b).data(), ldb, &info );
211 
212  BLAZE_INTERNAL_ASSERT( info == 0, "Invalid function argument" );
213 }
214 //*************************************************************************************************
215 
216 
217 //*************************************************************************************************
306 template< typename MT1 // Type of the system matrix
307  , bool SO1 // Storage order of the system matrix
308  , typename MT2 // Type of the right-hand side matrix
309  , bool SO2 > // Storage order of the right-hand side matrix
311  char uplo, char trans, char diag )
312 {
321 
322  if( !isSquare( ~A ) ) {
323  BLAZE_THROW_INVALID_ARGUMENT( "Invalid non-square matrix provided" );
324  }
325 
326  if( uplo != 'L' && uplo != 'U' ) {
327  BLAZE_THROW_INVALID_ARGUMENT( "Invalid uplo argument provided" );
328  }
329 
330  if( trans != 'N' && trans != 'T' && trans != 'C' ) {
331  BLAZE_THROW_INVALID_ARGUMENT( "Invalid trans argument provided" );
332  }
333 
334  if( diag != 'U' && diag != 'N' ) {
335  BLAZE_THROW_INVALID_ARGUMENT( "Invalid diag argument provided" );
336  }
337 
338  int n ( numeric_cast<int>( (~A).rows() ) );
339  int mrhs( numeric_cast<int>( SO2 ? (~B).rows() : (~B).columns() ) );
340  int nrhs( numeric_cast<int>( SO2 ? (~B).columns() : (~B).rows() ) );
341  int lda ( numeric_cast<int>( (~A).spacing() ) );
342  int ldb ( numeric_cast<int>( (~B).spacing() ) );
343  int info( 0 );
344 
345  if( n != mrhs ) {
346  BLAZE_THROW_INVALID_ARGUMENT( "Invalid right-hand side matrix provided" );
347  }
348 
349  if( n == 0 ) {
350  return;
351  }
352 
354  ( uplo == 'L' )?( uplo = 'U' ):( uplo = 'L' );
355  }
356 
357  trtrs( uplo, trans, diag, n, nrhs, (~A).data(), lda, (~B).data(), ldb, &info );
358 
359  BLAZE_INTERNAL_ASSERT( info == 0, "Invalid function argument" );
360 }
361 //*************************************************************************************************
362 
363 } // namespace blaze
364 
365 #endif
#define BLAZE_THROW_INVALID_ARGUMENT(MESSAGE)
Macro for the emission of a std::invalid_argument exception.This macro encapsulates the default way o...
Definition: Exception.h:235
Constraint on the data type.
Header file for auxiliary alias declarations.
Data type constraint.
#define BLAZE_CONSTRAINT_MUST_HAVE_MUTABLE_DATA_ACCESS(T)
Constraint on the data type.In case the given data type T does not provide low-level data access to m...
Definition: MutableDataAccess.h:61
BLAZE_ALWAYS_INLINE size_t size(const Vector< VT, TF > &vector) noexcept
Returns the current size/dimension of the vector.
Definition: Vector.h:265
#define BLAZE_CONSTRAINT_MUST_NOT_BE_COMPUTATION_TYPE(T)
Constraint on the data type.In case the given data type T is a computational expression (i...
Definition: Computation.h:81
Header file for the DenseVector base class.
Constraint on the data type.
Cast operators for numeric types.
#define BLAZE_CONSTRAINT_MUST_NOT_BE_ADAPTOR_TYPE(T)
Constraint on the data type.In case the given data type T is an adaptor type (as for instance LowerMa...
Definition: Adaptor.h:81
Base class for dense matrices.The DenseMatrix class is a base class for all dense matrix classes...
Definition: DenseMatrix.h:78
BLAZE_ALWAYS_INLINE size_t spacing(const DenseMatrix< MT, SO > &dm) noexcept
Returns the spacing between the beginning of two rows/columns.
Definition: DenseMatrix.h:110
Constraint on the data type.
Namespace of the Blaze C++ math library.
Definition: Blaze.h:57
Compile time check for row-major matrix types.This type trait tests whether or not the given template...
Definition: IsRowMajorMatrix.h:110
Header file for the DenseMatrix base class.
typename T::ElementType ElementType_
Alias declaration for nested ElementType type definitions.The ElementType_ alias declaration provides...
Definition: Aliases.h:163
Base class for N-dimensional dense vectors.The DenseVector class is a base class for all arbitrarily ...
Definition: DenseVector.h:70
#define BLAZE_CONSTRAINT_MUST_BE_SAME_TYPE(A, B)
Data type constraint.In case the two types A and B are not the same (ignoring all cv-qualifiers of bo...
Definition: SameType.h:71
BLAZE_ALWAYS_INLINE size_t columns(const Matrix< MT, SO > &matrix) noexcept
Returns the current number of columns of the matrix.
Definition: Matrix.h:340
void trtrs(char uplo, char trans, char diag, int n, int nrhs, const float *A, int lda, float *B, int ldb, int *info)
LAPACK kernel for the substitution step of solving a triangular single precision linear system of equ...
Definition: trtrs.h:137
Header file for the exception macros of the math module.
Constraint on the data type.
Header file for run time assertion macros.
#define BLAZE_CONSTRAINT_MUST_BE_BLAS_COMPATIBLE_TYPE(T)
Constraint on the data type.In case the given data type T is not a BLAS compatible data type (i...
Definition: BLASCompatible.h:61
BLAZE_ALWAYS_INLINE size_t rows(const Matrix< MT, SO > &matrix) noexcept
Returns the current number of rows of the matrix.
Definition: Matrix.h:324
decltype(auto) trans(const DenseMatrix< MT, SO > &dm)
Calculation of the transpose of the given dense matrix.
Definition: DMatTransExpr.h:790
Header file for the IsRowMajorMatrix type trait.
BLAZE_ALWAYS_INLINE bool isSquare(const Matrix< MT, SO > &matrix) noexcept
Checks if the given matrix is a square matrix.
Definition: Matrix.h:742
Header file for the CLAPACK trtrs wrapper functions.
#define BLAZE_INTERNAL_ASSERT(expr, msg)
Run time assertion macro for internal checks.In case of an invalid run time expression, the program execution is terminated. The BLAZE_INTERNAL_ASSERT macro can be disabled by setting the BLAZE_USER_ASSERTION flag to zero or by defining NDEBUG during the compilation.
Definition: Assert.h:101