LAPACK  3.7.1
LAPACK: Linear Algebra PACKage

◆ LAPACKE_sgbsvx_work()

lapack_int LAPACKE_sgbsvx_work ( int  matrix_layout,
char  fact,
char  trans,
lapack_int  n,
lapack_int  kl,
lapack_int  ku,
lapack_int  nrhs,
float *  ab,
lapack_int  ldab,
float *  afb,
lapack_int  ldafb,
lapack_int ipiv,
char *  equed,
float *  r,
float *  c,
float *  b,
lapack_int  ldb,
float *  x,
lapack_int  ldx,
float *  rcond,
float *  ferr,
float *  berr,
float *  work,
lapack_int iwork 
)

Definition at line 36 of file lapacke_sgbsvx_work.c.

44 {
45  lapack_int info = 0;
46  if( matrix_layout == LAPACK_COL_MAJOR ) {
47  /* Call LAPACK function and adjust info */
48  LAPACK_sgbsvx( &fact, &trans, &n, &kl, &ku, &nrhs, ab, &ldab, afb,
49  &ldafb, ipiv, equed, r, c, b, &ldb, x, &ldx, rcond, ferr,
50  berr, work, iwork, &info );
51  if( info < 0 ) {
52  info = info - 1;
53  }
54  } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
55  lapack_int ldab_t = MAX(1,kl+ku+1);
56  lapack_int ldafb_t = MAX(1,2*kl+ku+1);
57  lapack_int ldb_t = MAX(1,n);
58  lapack_int ldx_t = MAX(1,n);
59  float* ab_t = NULL;
60  float* afb_t = NULL;
61  float* b_t = NULL;
62  float* x_t = NULL;
63  /* Check leading dimension(s) */
64  if( ldab < n ) {
65  info = -9;
66  LAPACKE_xerbla( "LAPACKE_sgbsvx_work", info );
67  return info;
68  }
69  if( ldafb < n ) {
70  info = -11;
71  LAPACKE_xerbla( "LAPACKE_sgbsvx_work", info );
72  return info;
73  }
74  if( ldb < nrhs ) {
75  info = -17;
76  LAPACKE_xerbla( "LAPACKE_sgbsvx_work", info );
77  return info;
78  }
79  if( ldx < nrhs ) {
80  info = -19;
81  LAPACKE_xerbla( "LAPACKE_sgbsvx_work", info );
82  return info;
83  }
84  /* Allocate memory for temporary array(s) */
85  ab_t = (float*)LAPACKE_malloc( sizeof(float) * ldab_t * MAX(1,n) );
86  if( ab_t == NULL ) {
88  goto exit_level_0;
89  }
90  afb_t = (float*)LAPACKE_malloc( sizeof(float) * ldafb_t * MAX(1,n) );
91  if( afb_t == NULL ) {
93  goto exit_level_1;
94  }
95  b_t = (float*)LAPACKE_malloc( sizeof(float) * ldb_t * MAX(1,nrhs) );
96  if( b_t == NULL ) {
98  goto exit_level_2;
99  }
100  x_t = (float*)LAPACKE_malloc( sizeof(float) * ldx_t * MAX(1,nrhs) );
101  if( x_t == NULL ) {
103  goto exit_level_3;
104  }
105  /* Transpose input matrices */
106  LAPACKE_sgb_trans( matrix_layout, n, n, kl, ku, ab, ldab, ab_t, ldab_t );
107  if( LAPACKE_lsame( fact, 'f' ) ) {
108  LAPACKE_sgb_trans( matrix_layout, n, n, kl, kl+ku, afb, ldafb, afb_t,
109  ldafb_t );
110  }
111  LAPACKE_sge_trans( matrix_layout, n, nrhs, b, ldb, b_t, ldb_t );
112  /* Call LAPACK function and adjust info */
113  LAPACK_sgbsvx( &fact, &trans, &n, &kl, &ku, &nrhs, ab_t, &ldab_t, afb_t,
114  &ldafb_t, ipiv, equed, r, c, b_t, &ldb_t, x_t, &ldx_t,
115  rcond, ferr, berr, work, iwork, &info );
116  if( info < 0 ) {
117  info = info - 1;
118  }
119  /* Transpose output matrices */
120  if( LAPACKE_lsame( fact, 'e' ) && ( LAPACKE_lsame( *equed, 'b' ) ||
121  LAPACKE_lsame( *equed, 'c' ) || LAPACKE_lsame( *equed, 'r' ) ) ) {
122  LAPACKE_sgb_trans( LAPACK_COL_MAJOR, n, n, kl, ku, ab_t, ldab_t, ab,
123  ldab );
124  }
125  if( LAPACKE_lsame( fact, 'e' ) || LAPACKE_lsame( fact, 'n' ) ) {
126  LAPACKE_sgb_trans( LAPACK_COL_MAJOR, n, n, kl, kl+ku, afb_t,
127  ldafb_t, afb, ldafb );
128  }
129  if( LAPACKE_lsame( fact, 'f' ) && ( LAPACKE_lsame( *equed, 'b' ) ||
130  LAPACKE_lsame( *equed, 'c' ) || LAPACKE_lsame( *equed, 'r' ) ) ) {
131  LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, nrhs, b_t, ldb_t, b, ldb );
132  }
133  LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, nrhs, x_t, ldx_t, x, ldx );
134  /* Release memory and exit */
135  LAPACKE_free( x_t );
136 exit_level_3:
137  LAPACKE_free( b_t );
138 exit_level_2:
139  LAPACKE_free( afb_t );
140 exit_level_1:
141  LAPACKE_free( ab_t );
142 exit_level_0:
143  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
144  LAPACKE_xerbla( "LAPACKE_sgbsvx_work", info );
145  }
146  } else {
147  info = -1;
148  LAPACKE_xerbla( "LAPACKE_sgbsvx_work", info );
149  }
150  return info;
151 }
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:119
#define LAPACKE_free(p)
Definition: lapacke.h:113
void LAPACK_sgbsvx(char *fact, char *trans, lapack_int *n, lapack_int *kl, lapack_int *ku, lapack_int *nrhs, float *ab, lapack_int *ldab, float *afb, lapack_int *ldafb, lapack_int *ipiv, char *equed, float *r, float *c, float *b, lapack_int *ldb, float *x, lapack_int *ldx, float *rcond, float *ferr, float *berr, float *work, lapack_int *iwork, lapack_int *info)
void LAPACKE_sge_trans(int matrix_layout, lapack_int m, lapack_int n, const float *in, lapack_int ldin, float *out, lapack_int ldout)
#define MAX(x, y)
Definition: lapacke_utils.h:47
#define LAPACKE_malloc(size)
Definition: lapacke.h:110
lapack_logical LAPACKE_lsame(char ca, char cb)
Definition: lapacke_lsame.c:36
#define LAPACK_COL_MAJOR
Definition: lapacke.h:120
void LAPACKE_xerbla(const char *name, lapack_int info)
void LAPACKE_sgb_trans(int matrix_layout, lapack_int m, lapack_int n, lapack_int kl, lapack_int ku, const float *in, lapack_int ldin, float *out, lapack_int ldout)
#define lapack_int
Definition: lapacke.h:47
#define LAPACK_TRANSPOSE_MEMORY_ERROR
Definition: lapacke.h:123
Here is the call graph for this function:
Here is the caller graph for this function: