Blaze 3.9
potrs.h
Go to the documentation of this file.
1//=================================================================================================
33//=================================================================================================
34
35#ifndef _BLAZE_MATH_LAPACK_POTRS_H_
36#define _BLAZE_MATH_LAPACK_POTRS_H_
37
38
39//*************************************************************************************************
40// Includes
41//*************************************************************************************************
42
43#include <blaze/math/Aliases.h>
55#include <blaze/util/Assert.h>
58
59
60namespace blaze {
61
62//=================================================================================================
63//
64// LAPACK LLH-BASED SUBSTITUTION FUNCTIONS (POTRS)
65//
66//=================================================================================================
67
68//*************************************************************************************************
71template< typename MT, bool SO, typename VT, bool TF >
72void potrs( const DenseMatrix<MT,SO>& A, DenseVector<VT,TF>& b, char uplo );
73
74template< typename MT1, bool SO1, typename MT2, bool SO2 >
75void potrs( const DenseMatrix<MT1,SO1>& A, DenseMatrix<MT2,SO2>& B, char uplo );
77//*************************************************************************************************
78
79
80//*************************************************************************************************
162template< typename MT // Type of the system matrix
163 , bool SO // Storage order of the system matrix
164 , typename VT // Type of the right-hand side vector
165 , bool TF > // Transpose flag of the right-hand side vector
166inline void potrs( const DenseMatrix<MT,SO>& A, DenseVector<VT,TF>& b, char uplo )
167{
173
178
179 if( !isSquare( *A ) ) {
180 BLAZE_THROW_INVALID_ARGUMENT( "Invalid non-square matrix provided" );
181 }
182
183 if( (*b).size() != (*A).rows() ) {
184 BLAZE_THROW_INVALID_ARGUMENT( "Invalid right-hand side vector provided" );
185 }
186
187 if( uplo != 'L' && uplo != 'U' ) {
188 BLAZE_THROW_INVALID_ARGUMENT( "Invalid uplo argument provided" );
189 }
190
191 blas_int_t n ( numeric_cast<blas_int_t>( (*A).rows() ) );
192 blas_int_t nrhs( 1 );
193 blas_int_t lda ( numeric_cast<blas_int_t>( (*A).spacing() ) );
194 blas_int_t ldb ( numeric_cast<blas_int_t>( (*b).size() ) );
195 blas_int_t info( 0 );
196
197 if( n == 0 ) {
198 return;
199 }
200
201 if( IsRowMajorMatrix_v<MT> ) {
202 ( uplo == 'L' )?( uplo = 'U' ):( uplo = 'L' );
203 }
204
205 potrs( uplo, n, nrhs, (*A).data(), lda, (*b).data(), ldb, &info );
206
207 BLAZE_INTERNAL_ASSERT( info == 0, "Invalid function argument" );
208}
209//*************************************************************************************************
210
211
212//*************************************************************************************************
294template< typename MT1 // Type of the system matrix
295 , bool SO1 // Storage order of the system matrix
296 , typename MT2 // Type of the right-hand side matrix
297 , bool SO2 > // Storage order of the right-hand side matrix
298inline void potrs( const DenseMatrix<MT1,SO1>& A, DenseMatrix<MT2,SO2>& B, char uplo )
299{
305
311
312 if( !isSquare( *A ) ) {
313 BLAZE_THROW_INVALID_ARGUMENT( "Invalid non-square matrix provided" );
314 }
315
316 if( uplo != 'L' && uplo != 'U' ) {
317 BLAZE_THROW_INVALID_ARGUMENT( "Invalid uplo argument provided" );
318 }
319
320 blas_int_t n ( numeric_cast<blas_int_t>( (*A).rows() ) );
321 blas_int_t mrhs( numeric_cast<blas_int_t>( SO2 ? (*B).rows() : (*B).columns() ) );
322 blas_int_t nrhs( numeric_cast<blas_int_t>( SO2 ? (*B).columns() : (*B).rows() ) );
323 blas_int_t lda ( numeric_cast<blas_int_t>( (*A).spacing() ) );
324 blas_int_t ldb ( numeric_cast<blas_int_t>( (*B).spacing() ) );
325 blas_int_t info( 0 );
326
327 if( n != mrhs ) {
328 BLAZE_THROW_INVALID_ARGUMENT( "Invalid right-hand side matrix provided" );
329 }
330
331 if( n == 0 ) {
332 return;
333 }
334
335 if( IsRowMajorMatrix_v<MT1> ) {
336 ( uplo == 'L' )?( uplo = 'U' ):( uplo = 'L' );
337 }
338
339 potrs( uplo, n, nrhs, (*A).data(), lda, (*B).data(), ldb, &info );
340
341 BLAZE_INTERNAL_ASSERT( info == 0, "Invalid function argument" );
342}
343//*************************************************************************************************
344
345} // namespace blaze
346
347#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.
Header file for the IsRowMajorMatrix type trait.
Constraint on the data type.
Cast operators for numeric types.
Data type constraint.
Header file for the CLAPACK potrs 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
void potrs(const DenseMatrix< MT1, SO1 > &A, DenseMatrix< MT2, SO2 > &B, char uplo)
LAPACK kernel for the substitution step of solving a positive definite linear system of equations ( )...
Definition: potrs.h:298
#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.