LAPACK  3.10.0
LAPACK: Linear Algebra PACKage

◆ LAPACKE_stpmqrt_work()

lapack_int LAPACKE_stpmqrt_work ( int  matrix_layout,
char  side,
char  trans,
lapack_int  m,
lapack_int  n,
lapack_int  k,
lapack_int  l,
lapack_int  nb,
const float *  v,
lapack_int  ldv,
const float *  t,
lapack_int  ldt,
float *  a,
lapack_int  lda,
float *  b,
lapack_int  ldb,
float *  work 
)

Definition at line 35 of file lapacke_stpmqrt_work.c.

41 {
42  lapack_int info = 0;
43  if( matrix_layout == LAPACK_COL_MAJOR ) {
44  /* Call LAPACK function and adjust info */
45  LAPACK_stpmqrt( &side, &trans, &m, &n, &k, &l, &nb, v, &ldv, t, &ldt, a,
46  &lda, b, &ldb, work, &info );
47  if( info < 0 ) {
48  info = info - 1;
49  }
50  } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
51  lapack_int lda_t = MAX(1,k);
52  lapack_int ldb_t = MAX(1,m);
53  lapack_int ldt_t = MAX(1,ldt);
54  lapack_int ldv_t = MAX(1,ldv);
55  float* v_t = NULL;
56  float* t_t = NULL;
57  float* a_t = NULL;
58  float* b_t = NULL;
59  /* Check leading dimension(s) */
60  if( lda < m ) {
61  info = -14;
62  LAPACKE_xerbla( "LAPACKE_stpmqrt_work", info );
63  return info;
64  }
65  if( ldb < n ) {
66  info = -16;
67  LAPACKE_xerbla( "LAPACKE_stpmqrt_work", info );
68  return info;
69  }
70  if( ldt < nb ) {
71  info = -12;
72  LAPACKE_xerbla( "LAPACKE_stpmqrt_work", info );
73  return info;
74  }
75  if( ldv < k ) {
76  info = -10;
77  LAPACKE_xerbla( "LAPACKE_stpmqrt_work", info );
78  return info;
79  }
80  /* Allocate memory for temporary array(s) */
81  v_t = (float*)LAPACKE_malloc( sizeof(float) * ldv_t * MAX(1,k) );
82  if( v_t == NULL ) {
84  goto exit_level_0;
85  }
86  t_t = (float*)LAPACKE_malloc( sizeof(float) * ldt_t * MAX(1,nb) );
87  if( t_t == NULL ) {
89  goto exit_level_1;
90  }
91  a_t = (float*)LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,m) );
92  if( a_t == NULL ) {
94  goto exit_level_2;
95  }
96  b_t = (float*)LAPACKE_malloc( sizeof(float) * ldb_t * MAX(1,n) );
97  if( b_t == NULL ) {
99  goto exit_level_3;
100  }
101  /* Transpose input matrices */
102  LAPACKE_sge_trans( matrix_layout, ldv, k, v, ldv, v_t, ldv_t );
103  LAPACKE_sge_trans( matrix_layout, ldt, nb, t, ldt, t_t, ldt_t );
104  LAPACKE_sge_trans( matrix_layout, k, m, a, lda, a_t, lda_t );
105  LAPACKE_sge_trans( matrix_layout, m, n, b, ldb, b_t, ldb_t );
106  /* Call LAPACK function and adjust info */
107  LAPACK_stpmqrt( &side, &trans, &m, &n, &k, &l, &nb, v_t, &ldv_t, t_t,
108  &ldt_t, a_t, &lda_t, b_t, &ldb_t, work, &info );
109  if( info < 0 ) {
110  info = info - 1;
111  }
112  /* Transpose output matrices */
113  LAPACKE_sge_trans( LAPACK_COL_MAJOR, k, m, a_t, lda_t, a, lda );
114  LAPACKE_sge_trans( LAPACK_COL_MAJOR, m, n, b_t, ldb_t, b, ldb );
115  /* Release memory and exit */
116  LAPACKE_free( b_t );
117 exit_level_3:
118  LAPACKE_free( a_t );
119 exit_level_2:
120  LAPACKE_free( t_t );
121 exit_level_1:
122  LAPACKE_free( v_t );
123 exit_level_0:
124  if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
125  LAPACKE_xerbla( "LAPACKE_stpmqrt_work", info );
126  }
127  } else {
128  info = -1;
129  LAPACKE_xerbla( "LAPACKE_stpmqrt_work", info );
130  }
131  return info;
132 }
#define lapack_int
Definition: lapack.h:83
#define LAPACK_stpmqrt(...)
Definition: lapack.h:20806
#define LAPACK_COL_MAJOR
Definition: lapacke.h:53
#define LAPACKE_free(p)
Definition: lapacke.h:46
#define LAPACK_ROW_MAJOR
Definition: lapacke.h:52
#define LAPACKE_malloc(size)
Definition: lapacke.h:43
#define LAPACK_TRANSPOSE_MEMORY_ERROR
Definition: lapacke.h:56
void LAPACKE_xerbla(const char *name, 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:46
Here is the call graph for this function:
Here is the caller graph for this function: