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