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_ssbtrd_work( int matrix_order, char vect, char uplo,
00038 lapack_int n, lapack_int kd, float* ab,
00039 lapack_int ldab, float* d, float* e, float* q,
00040 lapack_int ldq, float* work )
00041 {
00042 lapack_int info = 0;
00043 if( matrix_order == LAPACK_COL_MAJOR ) {
00044
00045 LAPACK_ssbtrd( &vect, &uplo, &n, &kd, ab, &ldab, d, e, q, &ldq, work,
00046 &info );
00047 if( info < 0 ) {
00048 info = info - 1;
00049 }
00050 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00051 lapack_int ldab_t = MAX(1,kd+1);
00052 lapack_int ldq_t = MAX(1,n);
00053 float* ab_t = NULL;
00054 float* q_t = NULL;
00055
00056 if( ldab < n ) {
00057 info = -7;
00058 LAPACKE_xerbla( "LAPACKE_ssbtrd_work", info );
00059 return info;
00060 }
00061 if( ldq < n ) {
00062 info = -11;
00063 LAPACKE_xerbla( "LAPACKE_ssbtrd_work", info );
00064 return info;
00065 }
00066
00067 ab_t = (float*)LAPACKE_malloc( sizeof(float) * ldab_t * MAX(1,n) );
00068 if( ab_t == NULL ) {
00069 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00070 goto exit_level_0;
00071 }
00072 if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
00073 q_t = (float*)LAPACKE_malloc( sizeof(float) * ldq_t * MAX(1,n) );
00074 if( q_t == NULL ) {
00075 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00076 goto exit_level_1;
00077 }
00078 }
00079
00080 LAPACKE_ssb_trans( matrix_order, uplo, n, kd, ab, ldab, ab_t, ldab_t );
00081 if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
00082 LAPACKE_sge_trans( matrix_order, n, n, q, ldq, q_t, ldq_t );
00083 }
00084
00085 LAPACK_ssbtrd( &vect, &uplo, &n, &kd, ab_t, &ldab_t, d, e, q_t, &ldq_t,
00086 work, &info );
00087 if( info < 0 ) {
00088 info = info - 1;
00089 }
00090
00091 LAPACKE_ssb_trans( LAPACK_COL_MAJOR, uplo, n, kd, ab_t, ldab_t, ab,
00092 ldab );
00093 if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
00094 LAPACKE_sge_trans( LAPACK_COL_MAJOR, n, n, q_t, ldq_t, q, ldq );
00095 }
00096
00097 if( LAPACKE_lsame( vect, 'u' ) || LAPACKE_lsame( vect, 'v' ) ) {
00098 LAPACKE_free( q_t );
00099 }
00100 exit_level_1:
00101 LAPACKE_free( ab_t );
00102 exit_level_0:
00103 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00104 LAPACKE_xerbla( "LAPACKE_ssbtrd_work", info );
00105 }
00106 } else {
00107 info = -1;
00108 LAPACKE_xerbla( "LAPACKE_ssbtrd_work", info );
00109 }
00110 return info;
00111 }