Blaze 3.9
ormrq.h
Go to the documentation of this file.
1//=================================================================================================
33//=================================================================================================
34
35#ifndef _BLAZE_MATH_LAPACK_ORMRQ_H_
36#define _BLAZE_MATH_LAPACK_ORMRQ_H_
37
38
39//*************************************************************************************************
40// Includes
41//*************************************************************************************************
42
43#include <memory>
44#include <blaze/math/Aliases.h>
56#include <blaze/util/Assert.h>
58#include <blaze/util/mpl/Xor.h>
60
61
62namespace blaze {
63
64//=================================================================================================
65//
66// LAPACK FUNCTIONS TO MULTIPLY Q FROM A RQ DECOMPOSITION WITH A MATRIX (ORMRQ)
67//
68//=================================================================================================
69
70//*************************************************************************************************
73template< typename MT1, bool SO1, typename MT2, bool SO2 >
74void ormrq( DenseMatrix<MT1,SO1>& C, const DenseMatrix<MT2,SO2>& A,
75 char side, char trans, const ElementType_t<MT2>* tau );
77//*************************************************************************************************
78
79
80//*************************************************************************************************
145template< typename MT1, bool SO1, typename MT2, bool SO2 >
147 char side, char trans, const ElementType_t<MT2>* tau )
148{
155
162
163 using ET = ElementType_t<MT1>;
164
165 if( (*A).rows() < (*A).columns() ) {
166 BLAZE_THROW_INVALID_ARGUMENT( "Invalid size of Q matrix" );
167 }
168
169 if( side != 'L' && side != 'R' ) {
170 BLAZE_THROW_INVALID_ARGUMENT( "Invalid side argument provided" );
171 }
172
173 if( trans != 'N' && trans != 'T' ) {
174 BLAZE_THROW_INVALID_ARGUMENT( "Invalid trans argument provided" );
175 }
176
177 blas_int_t m ( numeric_cast<blas_int_t>( SO1 ? (*C).rows() : (*C).columns() ) );
178 blas_int_t n ( numeric_cast<blas_int_t>( SO1 ? (*C).columns() : (*C).rows() ) );
179 blas_int_t k ( numeric_cast<blas_int_t>( min( (*A).rows(), (*A).columns() ) ) );
180 blas_int_t lda ( numeric_cast<blas_int_t>( (*A).spacing() ) );
181 blas_int_t ldc ( numeric_cast<blas_int_t>( (*C).spacing() ) );
182 blas_int_t info( 0 );
183
184 if( m == 0 || n == 0 || k == 0 ) {
185 return;
186 }
187
188 if( IsRowMajorMatrix_v<MT1> ) {
189 ( side == 'L' )?( side = 'R' ):( side = 'L' );
190 }
191
192 if( IsRowMajorMatrix_v<MT1> ^ IsRowMajorMatrix_v<MT2> ) {
193 ( trans == 'N' )?( trans = 'T' ):( trans = 'N' );
194 }
195
196 blas_int_t lwork( k*ldc );
197 const std::unique_ptr<ET[]> work( new ET[lwork] );
198 const size_t offset( (*A).rows() - (*A).columns() );
199
200 if( SO2 ) {
201 ormrq( side, trans, m, n, k, (*A).data()+offset, lda, tau, (*C).data(), ldc, work.get(), lwork, &info );
202 }
203 else {
204 ormql( side, trans, m, n, k, (*A).data(offset), lda, tau, (*C).data(), ldc, work.get(), lwork, &info );
205 }
206
207 BLAZE_INTERNAL_ASSERT( info == 0, "Invalid argument for Q multiplication" );
208}
209//*************************************************************************************************
210
211} // namespace blaze
212
213#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.
Header file for the IsRowMajorMatrix type trait.
Constraint on the data type.
Cast operators for numeric types.
Header file for the Xor_t alias template.
Header file for the CLAPACK ormql wrapper functions.
Header file for the CLAPACK ormrq wrapper functions.
Base class for dense matrices.
Definition: DenseMatrix.h:82
Constraint on the data type.
Header file for the DenseMatrix base class.
#define BLAZE_CONSTRAINT_MUST_BE_BUILTIN_TYPE(T)
Constraint on the data type.
Definition: Builtin.h:60
decltype(auto) min(const DenseMatrix< MT1, SO1 > &lhs, const DenseMatrix< MT2, SO2 > &rhs)
Computes the componentwise minimum of the dense matrices lhs and rhs.
Definition: DMatDMatMapExpr.h:1339
decltype(auto) trans(const DenseMatrix< MT, SO > &dm)
Calculation of the transpose of the given dense matrix.
Definition: DMatTransExpr.h:766
void ormql(char side, char trans, blas_int_t m, blas_int_t n, blas_int_t k, const float *A, blas_int_t lda, const float *tau, float *C, blas_int_t ldc, float *work, blas_int_t lwork, blas_int_t *info)
LAPACK kernel for the multiplication of the single precision Q from a QL decomposition with another m...
Definition: ormql.h:145
void ormrq(DenseMatrix< MT1, SO1 > &C, const DenseMatrix< MT2, SO2 > &A, char side, char trans, const ElementType_t< MT2 > *tau)
LAPACK kernel for the multiplication of the double precision Q from a RQ decomposition with another m...
Definition: ormrq.h:146
#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
#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.