inline void
Syr
( UpperOrLower uplo, T alpha, const Matrix<T>& x, Matrix<T>& A,
bool conjugate=false )
{
#ifndef RELEASE
PushCallStack("Syr");
if( A.Height() != A.Width() )
throw std::logic_error("A must be square");
if( x.Width() != 1 && x.Height() != 1 )
throw std::logic_error("x must be a vector");
const int xLength = ( x.Width()==1 ? x.Height() : x.Width() );
if( xLength != A.Height() )
throw std::logic_error("x must conform with A");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
const int m = A.Height();
const int incx = ( x.Width()==1 ? 1 : x.LDim() );
if( conjugate )
{
blas::Her
( uploChar, m, alpha, x.LockedBuffer(), incx, A.Buffer(), A.LDim() );
}
else
{
blas::Syr
( uploChar, m, alpha, x.LockedBuffer(), incx, A.Buffer(), A.LDim() );
}
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
Hemv
( UpperOrLower uplo,
T alpha, const Matrix<T>& A, const Matrix<T>& x, T beta, Matrix<T>& y )
{
#ifndef RELEASE
PushCallStack("Hemv");
if( A.Height() != A.Width() )
throw std::logic_error("A must be square");
if( ( x.Height() != 1 && x.Width() != 1 ) ||
( y.Height() != 1 && y.Width() != 1 ) )
throw std::logic_error("x and y must be vectors");
const int xLength = ( x.Width()==1 ? x.Height() : x.Width() );
const int yLength = ( y.Width()==1 ? y.Height() : y.Width() );
if( A.Height() != xLength || A.Height() != yLength )
throw std::logic_error("A must conform with x and y");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
const int m = A.Height();
const int incx = ( x.Width()==1 ? 1 : x.LDim() );
const int incy = ( y.Width()==1 ? 1 : y.LDim() );
blas::Hemv
( uploChar, m,
alpha, A.LockedBuffer(), A.LDim(), x.LockedBuffer(), incx,
beta, y.Buffer(), incy );
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
Trmv
( UpperOrLower uplo, Orientation orientation, UnitOrNonUnit diag,
const Matrix<T>& A, Matrix<T>& x )
{
#ifndef RELEASE
PushCallStack("Trmv");
if( x.Height() != 1 && x.Width() != 1 )
throw std::logic_error("x must be a vector");
if( A.Height() != A.Width() )
throw std::logic_error("A must be square");
const int xLength = ( x.Width()==1 ? x.Height() : x.Width() );
if( xLength != A.Height() )
throw std::logic_error("x must conform with A");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
const char transChar = OrientationToChar( orientation );
const char diagChar = UnitOrNonUnitToChar( diag );
const int m = A.Height();
const int incx = ( x.Width()==1 ? 1 : x.LDim() );
blas::Trmv
( uploChar, transChar, diagChar, m,
A.LockedBuffer(), A.LDim(), x.Buffer(), incx );
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
Trmm
( LeftOrRight side, UpperOrLower uplo,
Orientation orientation, UnitOrNonUnit diag,
T alpha, const Matrix<T>& A, Matrix<T>& B )
{
#ifndef RELEASE
CallStackEntry entry("Trmm");
if( A.Height() != A.Width() )
LogicError("Triangular matrix must be square");
if( side == LEFT )
{
if( A.Height() != B.Height() )
LogicError("Nonconformal Trmm");
}
else
{
if( A.Height() != B.Width() )
LogicError("Nonconformal Trmm");
}
#endif
const char sideChar = LeftOrRightToChar( side );
const char uploChar = UpperOrLowerToChar( uplo );
const char transChar = OrientationToChar( orientation );
const char diagChar = UnitOrNonUnitToChar( diag );
blas::Trmm
( sideChar, uploChar, transChar, diagChar, B.Height(), B.Width(),
alpha, A.LockedBuffer(), A.LDim(), B.Buffer(), B.LDim() );
}
inline void
Symm
( LeftOrRight side, UpperOrLower uplo,
T alpha, const Matrix<T>& A, const Matrix<T>& B, T beta, Matrix<T>& C,
bool conjugate=false )
{
#ifndef RELEASE
CallStackEntry entry("Symm");
#endif
const char sideChar = LeftOrRightToChar( side );
const char uploChar = UpperOrLowerToChar( uplo );
if( conjugate )
{
blas::Hemm
( sideChar, uploChar, C.Height(), C.Width(),
alpha, A.LockedBuffer(), A.LDim(),
B.LockedBuffer(), B.LDim(),
beta, C.Buffer(), C.LDim() );
}
else
{
blas::Symm
( sideChar, uploChar, C.Height(), C.Width(),
alpha, A.LockedBuffer(), A.LDim(),
B.LockedBuffer(), B.LDim(),
beta, C.Buffer(), C.LDim() );
}
}
inline void
Syrk
( UpperOrLower uplo, Orientation orientation,
T alpha, const Matrix<T>& A, T beta, Matrix<T>& C )
{
#ifndef RELEASE
PushCallStack("Syrk");
if( orientation == NORMAL )
{
if( A.Height() != C.Height() || A.Height() != C.Width() )
throw std::logic_error("Nonconformal Syrk");
}
else if( orientation == TRANSPOSE )
{
if( A.Width() != C.Height() || A.Width() != C.Width() )
throw std::logic_error("Nonconformal Syrk");
}
else
throw std::logic_error
("Syrk only accepts NORMAL and TRANSPOSE options");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
const char transChar = OrientationToChar( orientation );
const int k = ( orientation == NORMAL ? A.Width() : A.Height() );
blas::Syrk
( uploChar, transChar, C.Height(), k,
alpha, A.LockedBuffer(), A.LDim(),
beta, C.Buffer(), C.LDim() );
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
Cholesky( UpperOrLower uplo, Matrix<F>& A )
{
#ifndef RELEASE
PushCallStack("Cholesky");
if( A.Height() != A.Width() )
throw std::logic_error("A must be square");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
lapack::Cholesky( uploChar, A.Height(), A.Buffer(), A.LDim() );
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
TriangularInverse
( UpperOrLower uplo, UnitOrNonUnit diag, Matrix<F>& A )
{
#ifndef RELEASE
PushCallStack("TriangularInverse");
if( A.Height() != A.Width() )
throw std::logic_error("A must be square");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
const char diagChar = UnitOrNonUnitToChar( diag );
lapack::TriangularInverse
( uploChar, diagChar, A.Height(), A.Buffer(), A.LDim() );
#ifndef RELEASE
PopCallStack();
#endif
}
void TrrkMKL
( UpperOrLower uplo,
Orientation orientA, Orientation orientB,
T alpha, const Matrix<T>& A, const Matrix<T>& B,
T beta, Matrix<T>& C )
{
EL_DEBUG_CSE
const char uploChar = UpperOrLowerToChar( uplo );
const char orientAChar = OrientationToChar( orientA );
const char orientBChar = OrientationToChar( orientB );
const auto n = C.Height();
const auto k = orientA == NORMAL ? A.Width() : A.Height();
mkl::Trrk
( uploChar, orientAChar, orientBChar,
n, k,
alpha, A.LockedBuffer(), A.LDim(),
B.LockedBuffer(), B.LDim(),
beta, C.Buffer(), C.LDim() );
}
inline void
Hemm
( LeftOrRight side, UpperOrLower uplo,
T alpha, const Matrix<T>& A, const Matrix<T>& B, T beta, Matrix<T>& C )
{
#ifndef RELEASE
PushCallStack("Hemm");
#endif
const char sideChar = LeftOrRightToChar( side );
const char uploChar = UpperOrLowerToChar( uplo );
blas::Hemm
( sideChar, uploChar, C.Height(), C.Width(),
alpha, A.LockedBuffer(), A.LDim(),
B.LockedBuffer(), B.LDim(),
beta, C.Buffer(), C.LDim() );
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
Trsm
( LeftOrRight side, UpperOrLower uplo,
Orientation orientation, UnitOrNonUnit diag,
F alpha, const Matrix<F>& A, Matrix<F>& B,
bool checkIfSingular )
{
#ifndef RELEASE
PushCallStack("Trsm");
if( A.Height() != A.Width() )
throw std::logic_error("Triangular matrix must be square");
if( side == LEFT )
{
if( A.Height() != B.Height() )
throw std::logic_error("Nonconformal Trsm");
}
else
{
if( A.Height() != B.Width() )
throw std::logic_error("Nonconformal Trsm");
}
#endif
const char sideChar = LeftOrRightToChar( side );
const char uploChar = UpperOrLowerToChar( uplo );
const char transChar = OrientationToChar( orientation );
const char diagChar = UnitOrNonUnitToChar( diag );
if( checkIfSingular && diag != UNIT )
{
const int n = A.Height();
for( int j=0; j<n; ++j )
if( A.Get(j,j) == F(0) )
throw SingularMatrixException();
}
blas::Trsm
( sideChar, uploChar, transChar, diagChar, B.Height(), B.Width(),
alpha, A.LockedBuffer(), A.LDim(), B.Buffer(), B.LDim() );
#ifndef RELEASE
PopCallStack();
#endif
}
inline void
Syr2
( UpperOrLower uplo,
T alpha, const Matrix<T>& x, const Matrix<T>& y, Matrix<T>& A,
bool conjugate=false )
{
#ifndef RELEASE
CallStackEntry entry("Syr2");
if( A.Height() != A.Width() )
LogicError("A must be square");
if( (x.Width() != 1 && x.Height() != 1) ||
(y.Width() != 1 && y.Height() != 1) )
LogicError("x and y must be vectors");
const Int xLength = ( x.Width()==1 ? x.Height() : x.Width() );
const Int yLength = ( y.Width()==1 ? y.Height() : y.Width() );
if( xLength != A.Height() || yLength != A.Height() )
LogicError("x and y must conform with A");
#endif
const char uploChar = UpperOrLowerToChar( uplo );
const Int m = A.Height();
const Int incx = ( x.Width()==1 ? 1 : x.LDim() );
const Int incy = ( y.Width()==1 ? 1 : y.LDim() );
if( conjugate )
{
blas::Her2
( uploChar, m,
alpha, x.LockedBuffer(), incx, y.LockedBuffer(), incy,
A.Buffer(), A.LDim() );
}
else
{
blas::Syr2
( uploChar, m,
alpha, x.LockedBuffer(), incx, y.LockedBuffer(), incy,
A.Buffer(), A.LDim() );
}
}
