gbtrf.cc (flens/lapack/interface/src/gbtrf.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

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

namespace flens { namespace lapack {

extern "C" {

//-- dgbtrf --------------------------------------------------------------------
void
LAPACK_DECL(dgbtrf)(const INTEGER    *M,
                    const INTEGER    *N,
                    const INTEGER    *KL,
                    const INTEGER    *KU,
                    DOUBLE           *AB,
                    const INTEGER    *LDAB,
                    INTEGER          *IPIV,
                    INTEGER          *INFO)
{
    using std::min;

//
//  Test the input parameters so that we pass LAPACK error checks
//
    INTEGER KV = *KU + *KL;

    *INFO = 0;
    if (*M<0) {
        *INFO = -1;
    } else if (*N<0) {
        *INFO = -2;
    } else if (*KL<0) {
        *INFO = -3;
    } else if (*KU<0) {
        *INFO = -4;
    } else if (*LDAB<*KL+KV+1) {
        *INFO = -6;
    }
    if (*INFO!=0) {
        *INFO = -(*INFO);
        LAPACK_ERROR("DGBTRF", INFO);
        *INFO = -(*INFO);
        return;
    }
//
//  Call FLENS implementation
//
    INTEGER MN = min(*M,*N);

    DGbMatrixView       A     = DBSView(*M, *N, *KL, KV, AB, *LDAB);
    IDenseVectorView    piv   = IArrayView(MN, IPIV, 1);

    *INFO = trf(A, piv);
}

} // extern "C"

} } // namespace lapack, flens