LAPACK 3.11.0
LAPACK: Linear Algebra PACKage
Loading...
Searching...
No Matches

◆ LAPACKE_stprfb_work()

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

Definition at line 35 of file lapacke_stprfb_work.c.

42{
43 lapack_int info = 0;
44 if( matrix_layout == LAPACK_COL_MAJOR ) {
45 /* Call LAPACK function and adjust info */
46 LAPACK_stprfb( &side, &trans, &direct, &storev, &m, &n, &k, &l, v, &ldv,
47 t, &ldt, a, &lda, b, &ldb, work, &ldwork );
48 if( info < 0 ) {
49 info = info - 1;
50 }
51 } else if( matrix_layout == LAPACK_ROW_MAJOR ) {
52 lapack_int lda_t = MAX(1,k);
53 lapack_int ldb_t = MAX(1,m);
54 lapack_int ldt_t = MAX(1,ldt);
55 lapack_int ldv_t = MAX(1,ldv);
56 float* v_t = NULL;
57 float* t_t = NULL;
58 float* a_t = NULL;
59 float* b_t = NULL;
60 /* Check leading dimension(s) */
61 if( lda < m ) {
62 info = -15;
63 LAPACKE_xerbla( "LAPACKE_stprfb_work", info );
64 return info;
65 }
66 if( ldb < n ) {
67 info = -17;
68 LAPACKE_xerbla( "LAPACKE_stprfb_work", info );
69 return info;
70 }
71 if( ldt < k ) {
72 info = -13;
73 LAPACKE_xerbla( "LAPACKE_stprfb_work", info );
74 return info;
75 }
76 if( ldv < k ) {
77 info = -11;
78 LAPACKE_xerbla( "LAPACKE_stprfb_work", info );
79 return info;
80 }
81 /* Allocate memory for temporary array(s) */
82 v_t = (float*)LAPACKE_malloc( sizeof(float) * ldv_t * MAX(1,k) );
83 if( v_t == NULL ) {
85 goto exit_level_0;
86 }
87 t_t = (float*)LAPACKE_malloc( sizeof(float) * ldt_t * MAX(1,k) );
88 if( t_t == NULL ) {
90 goto exit_level_1;
91 }
92 a_t = (float*)LAPACKE_malloc( sizeof(float) * lda_t * MAX(1,m) );
93 if( a_t == NULL ) {
95 goto exit_level_2;
96 }
97 b_t = (float*)LAPACKE_malloc( sizeof(float) * ldb_t * MAX(1,n) );
98 if( b_t == NULL ) {
100 goto exit_level_3;
101 }
102 /* Transpose input matrices */
103 LAPACKE_sge_trans( matrix_layout, ldv, k, v, ldv, v_t, ldv_t );
104 LAPACKE_sge_trans( matrix_layout, ldt, k, t, ldt, t_t, ldt_t );
105 LAPACKE_sge_trans( matrix_layout, k, m, a, lda, a_t, lda_t );
106 LAPACKE_sge_trans( matrix_layout, m, n, b, ldb, b_t, ldb_t );
107 /* Call LAPACK function and adjust info */
108 LAPACK_stprfb( &side, &trans, &direct, &storev, &m, &n, &k, &l, v_t,
109 &ldv_t, t_t, &ldt_t, a_t, &lda_t, b_t, &ldb_t, work,
110 &ldwork );
111 info = 0; /* LAPACK call is ok! */
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 );
117exit_level_3:
118 LAPACKE_free( a_t );
119exit_level_2:
120 LAPACKE_free( t_t );
121exit_level_1:
122 LAPACKE_free( v_t );
123exit_level_0:
124 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
125 LAPACKE_xerbla( "LAPACKE_stprfb_work", info );
126 }
127 } else {
128 info = -1;
129 LAPACKE_xerbla( "LAPACKE_stprfb_work", info );
130 }
131 return info;
132}
#define LAPACK_stprfb(...)
Definition: lapack.h:20967
#define lapack_int
Definition: lapack.h:87
#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: