getri.cc (flens/lapack/interface/src/getri.cc)

       1
       2
       3
       4
       5
       6
       7
       8
       9
      10
      11
      12
      13
      14
      15
      16
      17
      18
      19
      20
      21
      22
      23
      24
      25
      26
      27
      28
      29
      30
      31
      32
      33
      34
      35
      36
      37
      38
      39
      40
      41
      42
      43
      44
      45
      46
      47
      48
      49
      50
      51
      52
      53
      54
      55
      56
      57
      58
      59
      60
      61
      62
      63
      64
      65
      66
      67
      68
      69
      70
      71
      72
      73
      74
      75
      76
      77
      78
      79
      80
      81
      82
      83
      84
      85
      86
      87
      88
      89
      90
      91
      92
      93
      94
      95
      96
      97
      98
      99
     100
     101
     102
     103

#include <flens/lapack/interface/include/config.h>

namespace flens { namespace lapack {

extern "C" {

//-- dgetri --------------------------------------------------------------------
void
LAPACK_DECL(dgetri)(const INTEGER    *N,
                    DOUBLE           *A,
                    const INTEGER    *LDA,
                    const INTEGER    *IPIV,
                    DOUBLE           *WORK,
                    const INTEGER    *LWORK,
                    INTEGER          *INFO)
{
    using std::max;
    using std::min;
//
//  Test the input parameters so that we pass LAPACK error checks
//
    *INFO = 0;
    bool lQuery = (*LWORK==-1);

    if (*N<0) {
        *INFO = 1;
    } else if (*LDA<max(INTEGER(1),*N)) {
        *INFO = 3;
    } else if (*LWORK<max(INTEGER(1),*N) && !lQuery) {
        *INFO = 6;
    }

    if (*INFO!=0) {
        LAPACK_ERROR("DGETRI", INFO);
        *INFO = -(*INFO);
        return;
    }

//
//  Setup FLENS matrix/vector types
//
    DGeMatrixView          _A      = DFSView(*N, *N, A, *LDA);
    IConstDenseVectorView  _IPIV   = IConstArrayView(*N, IPIV, 1);
    DDenseVectorView       _WORK   = DArrayView(*LWORK, WORK, 1);

//
//  Call FLENS implementation
//
    tri(_A, _IPIV, _WORK);
}

//-- zgetri --------------------------------------------------------------------
void
LAPACK_DECL(zgetri)(const INTEGER    *N,
                    DOUBLE_COMPLEX   *A,
                    const INTEGER    *LDA,
                    const INTEGER    *IPIV,
                    DOUBLE_COMPLEX   *WORK,
                    const INTEGER    *LWORK,
                    INTEGER          *INFO)
{
    using std::max;
    using std::min;
//
//  Test the input parameters so that we pass LAPACK error checks
//
    *INFO = 0;
    bool lQuery = (*LWORK==-1);

    if (*N<0) {
        *INFO = 1;
    } else if (*LDA<max(INTEGER(1),*N)) {
        *INFO = 3;
    } else if (*LWORK<max(INTEGER(1),*N) && !lQuery) {
        *INFO = 6;
    }

    if (*INFO!=0) {
        LAPACK_ERROR("ZGETRI", INFO);
        *INFO = -(*INFO);
        return;
    }

//
//  Setup FLENS matrix/vector types
//
    auto zA     = reinterpret_cast<CXX_DOUBLE_COMPLEX *>(A);
    auto zWORK  = reinterpret_cast<CXX_DOUBLE_COMPLEX *>(WORK);

    ZGeMatrixView          _A      = ZFSView(*N, *N, zA, *LDA);
    IConstDenseVectorView  _IPIV   = IConstArrayView(*N, IPIV, 1);
    ZDenseVectorView       _WORK   = ZArrayView(*LWORK, zWORK, 1);

//
//  Call FLENS implementation
//
    tri(_A, _IPIV, _WORK);
}

} // extern "C"

} } // namespace lapack, flens