Blaze 3.9
getrs.h
Go to the documentation of this file.
1//=================================================================================================
33//=================================================================================================
34
35#ifndef _BLAZE_MATH_LAPACK_GETRS_H_
36#define _BLAZE_MATH_LAPACK_GETRS_H_
37
38
39//*************************************************************************************************
40// Includes
41//*************************************************************************************************
42
43#include <blaze/math/Aliases.h>
54#include <blaze/util/Assert.h>
57
58
59namespace blaze {
60
61//=================================================================================================
62//
63// LAPACK LU-BASED SUBSTITUTION FUNCTIONS (GETRS)
64//
65//=================================================================================================
66
67//*************************************************************************************************
70template< typename MT, bool SO, typename VT, bool TF >
71void getrs( const DenseMatrix<MT,SO>& A, DenseVector<VT,TF>& b, char trans,
72 const blas_int_t* ipiv );
73
74template< typename MT1, bool SO1, typename MT2, bool SO2 >
75void getrs( const DenseMatrix<MT1,SO1>& A, DenseMatrix<MT2,SO2>& B, char trans,
76 const blas_int_t* ipiv );
78//*************************************************************************************************
79
80
81//*************************************************************************************************
168template< typename MT // Type of the system matrix
169 , bool SO // Storage order of the system matrix
170 , typename VT // Type of the right-hand side vector
171 , bool TF > // Transpose flag of the right-hand side vector
172inline void getrs( const DenseMatrix<MT,SO>& A, DenseVector<VT,TF>& b, char trans,
173 const blas_int_t* ipiv )
174{
180
185
186 if( !isSquare( *A ) ) {
187 BLAZE_THROW_INVALID_ARGUMENT( "Invalid non-square matrix provided" );
188 }
189
190 if( (*b).size() != (*A).rows() ) {
191 BLAZE_THROW_INVALID_ARGUMENT( "Invalid right-hand side vector provided" );
192 }
193
194 if( trans != 'N' && trans != 'T' && trans != 'C' ) {
195 BLAZE_THROW_INVALID_ARGUMENT( "Invalid trans argument provided" );
196 }
197
198 blas_int_t n ( numeric_cast<blas_int_t>( (*A).rows() ) );
199 blas_int_t nrhs( 1 );
200 blas_int_t lda ( numeric_cast<blas_int_t>( (*A).spacing() ) );
201 blas_int_t ldb ( numeric_cast<blas_int_t>( (*b).size() ) );
202 blas_int_t info( 0 );
203
204 if( n == 0 ) {
205 return;
206 }
207
208 getrs( trans, n, nrhs, (*A).data(), lda, ipiv, (*b).data(), ldb, &info );
209
210 BLAZE_INTERNAL_ASSERT( info == 0, "Invalid function argument" );
211}
212//*************************************************************************************************
213
214
215//*************************************************************************************************
306template< 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 trans, const blas_int_t* ipiv )
312{
318
324
325 if( !isSquare( *A ) ) {
326 BLAZE_THROW_INVALID_ARGUMENT( "Invalid non-square matrix provided" );
327 }
328
329 if( trans != 'N' && trans != 'T' && trans != 'C' ) {
330 BLAZE_THROW_INVALID_ARGUMENT( "Invalid trans argument provided" );
331 }
332
333 blas_int_t n ( numeric_cast<blas_int_t>( (*A).rows() ) );
334 blas_int_t mrhs( numeric_cast<blas_int_t>( SO2 ? (*B).rows() : (*B).columns() ) );
335 blas_int_t nrhs( numeric_cast<blas_int_t>( SO2 ? (*B).columns() : (*B).rows() ) );
336 blas_int_t lda ( numeric_cast<blas_int_t>( (*A).spacing() ) );
337 blas_int_t ldb ( numeric_cast<blas_int_t>( (*B).spacing() ) );
338 blas_int_t info( 0 );
339
340 if( n != mrhs ) {
341 BLAZE_THROW_INVALID_ARGUMENT( "Invalid right-hand side matrix provided" );
342 }
343
344 if( n == 0 ) {
345 return;
346 }
347
348 getrs( trans, n, nrhs, (*A).data(), lda, ipiv, (*B).data(), ldb, &info );
349
350 BLAZE_INTERNAL_ASSERT( info == 0, "Invalid function argument" );
351}
352//*************************************************************************************************
353
354} // namespace blaze
355
356#endif
Constraint on the data type.
Header file for auxiliary alias declarations.
typename T::ElementType ElementType_t
Alias declaration for nested ElementType type definitions.
Definition: Aliases.h:190
Header file for run time assertion macros.
Constraint on the data type.
Constraint on the data type.
Constraint on the data type.
Constraint on the data type.
Cast operators for numeric types.
Data type constraint.
Header file for the CLAPACK getrs wrapper functions.
Base class for dense matrices.
Definition: DenseMatrix.h:82
Base class for N-dimensional dense vectors.
Definition: DenseVector.h:77
Constraint on the data type.
Header file for the DenseMatrix base class.
Header file for the DenseVector base class.
#define BLAZE_CONSTRAINT_MUST_BE_SAME_TYPE(A, B)
Data type constraint.
Definition: SameType.h:71
decltype(auto) trans(const DenseMatrix< MT, SO > &dm)
Calculation of the transpose of the given dense matrix.
Definition: DMatTransExpr.h:766
void getrs(const DenseMatrix< MT1, SO1 > &A, DenseMatrix< MT2, SO2 > &B, char trans, const blas_int_t *ipiv)
LAPACK kernel for the substitution step of solving a general linear system of equations ( ).
Definition: getrs.h:310
#define BLAZE_CONSTRAINT_MUST_BE_BLAS_COMPATIBLE_TYPE(T)
Constraint on the data type.
Definition: BLASCompatible.h:61
#define BLAZE_CONSTRAINT_MUST_BE_CONTIGUOUS_TYPE(T)
Constraint on the data type.
Definition: Contiguous.h:61
#define BLAZE_CONSTRAINT_MUST_NOT_BE_COMPUTATION_TYPE(T)
Constraint on the data type.
Definition: Computation.h:81
#define BLAZE_CONSTRAINT_MUST_NOT_BE_ADAPTOR_TYPE(T)
Constraint on the data type.
Definition: Adaptor.h:81
#define BLAZE_CONSTRAINT_MUST_HAVE_MUTABLE_DATA_ACCESS(T)
Constraint on the data type.
Definition: MutableDataAccess.h:61
#define BLAZE_CONSTRAINT_MUST_HAVE_CONST_DATA_ACCESS(T)
Constraint on the data type.
Definition: ConstDataAccess.h:61
int32_t blas_int_t
Signed integer type used in the BLAS/LAPACK wrapper functions.
Definition: Types.h:64
bool isSquare(const Matrix< MT, SO > &matrix) noexcept
Checks if the given matrix is a square matrix.
Definition: Matrix.h:1383
#define BLAZE_INTERNAL_ASSERT(expr, msg)
Run time assertion macro for internal checks.
Definition: Assert.h:101
#define BLAZE_THROW_INVALID_ARGUMENT(MESSAGE)
Macro for the emission of a std::invalid_argument exception.
Definition: Exception.h:235
Header file for the exception macros of the math module.