00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034 #include "lapacke.h"
00035 #include "lapacke_utils.h"
00036
00037 lapack_int LAPACKE_sgbbrd_work( int matrix_order, char vect, lapack_int m,
00038 lapack_int n, lapack_int ncc, lapack_int kl,
00039 lapack_int ku, float* ab, lapack_int ldab,
00040 float* d, float* e, float* q, lapack_int ldq,
00041 float* pt, lapack_int ldpt, float* c,
00042 lapack_int ldc, float* work )
00043 {
00044 lapack_int info = 0;
00045 if( matrix_order == LAPACK_COL_MAJOR ) {
00046
00047 LAPACK_sgbbrd( &vect, &m, &n, &ncc, &kl, &ku, ab, &ldab, d, e, q, &ldq,
00048 pt, &ldpt, c, &ldc, work, &info );
00049 if( info < 0 ) {
00050 info = info - 1;
00051 }
00052 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00053 lapack_int ldab_t = MAX(1,kl+ku+1);
00054 lapack_int ldc_t = MAX(1,m);
00055 lapack_int ldpt_t = MAX(1,n);
00056 lapack_int ldq_t = MAX(1,m);
00057 float* ab_t = NULL;
00058 float* q_t = NULL;
00059 float* pt_t = NULL;
00060 float* c_t = NULL;
00061
00062 if( ldab < n ) {
00063 info = -9;
00064 LAPACKE_xerbla( "LAPACKE_sgbbrd_work", info );
00065 return info;
00066 }
00067 if( ldc < ncc ) {
00068 info = -17;
00069 LAPACKE_xerbla( "LAPACKE_sgbbrd_work", info );
00070 return info;
00071 }
00072 if( ldpt < n ) {
00073 info = -15;
00074 LAPACKE_xerbla( "LAPACKE_sgbbrd_work", info );
00075 return info;
00076 }
00077 if( ldq < m ) {
00078 info = -13;
00079 LAPACKE_xerbla( "LAPACKE_sgbbrd_work", info );
00080 return info;
00081 }
00082
00083 ab_t = (float*)LAPACKE_malloc( sizeof(float) * ldab_t * MAX(1,n) );
00084 if( ab_t == NULL ) {
00085 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00086 goto exit_level_0;
00087 }
00088 if( LAPACKE_lsame( vect, 'b' ) || LAPACKE_lsame( vect, 'q' ) ) {
00089 q_t = (float*)LAPACKE_malloc( sizeof(float) * ldq_t * MAX(1,m) );
00090 if( q_t == NULL ) {
00091 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00092 goto exit_level_1;
00093 }
00094 }
00095 if( LAPACKE_lsame( vect, 'b' ) || LAPACKE_lsame( vect, 'p' ) ) {
00096 pt_t = (float*)LAPACKE_malloc( sizeof(float) * ldpt_t * MAX(1,n) );
00097 if( pt_t == NULL ) {
00098 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00099 goto exit_level_2;
00100 }
00101 }
00102 if( ncc != 0 ) {
00103 c_t = (float*)LAPACKE_malloc( sizeof(float) * ldc_t * MAX(1,ncc) );
00104 if( c_t == NULL ) {
00105 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00106 goto exit_level_3;
00107 }
00108 }
00109
00110 LAPACKE_sgb_trans( matrix_order, m, n, kl, ku, ab, ldab, ab_t, ldab_t );
00111 if( ncc != 0 ) {
00112 LAPACKE_sge_trans( matrix_order, m, ncc, c, ldc, c_t, ldc_t );
00113 }
00114
00115 LAPACK_sgbbrd( &vect, &m, &n, &ncc, &kl, &ku, ab_t, &ldab_t, d, e, q_t,
00116 &ldq_t, pt_t, &ldpt_t, c_t, &ldc_t, work, &info );
00117 if( info < 0 ) {
00118 info = info - 1;
00119 }
00120
00121 LAPACKE_sgb_trans( LAPACK_COL_MAJOR, m, n, kl, ku, ab_t, ldab_t, ab,
00122 ldab );
00123 if( LAPACKE_lsame( vect, 'b' ) || LAPACKE_lsame( vect, 'q' ) ) {
00124 LAPACKE_sge_trans( LAPACK_COL_MAJOR, m, m, q_t, ldq_t, q, ldq );
00125 }
00126 if( LAPACKE_lsame( vect, 'b' ) || LAPACKE_lsame( vect, 'p' ) ) {
00127 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, pt_t, ldpt_t, pt, ldpt );
00128 }
00129 if( ncc != 0 ) {
00130 LAPACKE_sge_trans( LAPACK_COL_MAJOR, m, ncc, c_t, ldc_t, c, ldc );
00131 }
00132
00133 if( ncc != 0 ) {
00134 LAPACKE_free( c_t );
00135 }
00136 exit_level_3:
00137 if( LAPACKE_lsame( vect, 'b' ) || LAPACKE_lsame( vect, 'p' ) ) {
00138 LAPACKE_free( pt_t );
00139 }
00140 exit_level_2:
00141 if( LAPACKE_lsame( vect, 'b' ) || LAPACKE_lsame( vect, 'q' ) ) {
00142 LAPACKE_free( q_t );
00143 }
00144 exit_level_1:
00145 LAPACKE_free( ab_t );
00146 exit_level_0:
00147 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00148 LAPACKE_xerbla( "LAPACKE_sgbbrd_work", info );
00149 }
00150 } else {
00151 info = -1;
00152 LAPACKE_xerbla( "LAPACKE_sgbbrd_work", info );
00153 }
00154 return info;
00155 }