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_shgeqz( int matrix_order, char job, char compq, char compz,
00038 lapack_int n, lapack_int ilo, lapack_int ihi,
00039 float* h, lapack_int ldh, float* t, lapack_int ldt,
00040 float* alphar, float* alphai, float* beta, float* q,
00041 lapack_int ldq, float* z, lapack_int ldz )
00042 {
00043 lapack_int info = 0;
00044 lapack_int lwork = -1;
00045 float* work = NULL;
00046 float work_query;
00047 if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) {
00048 LAPACKE_xerbla( "LAPACKE_shgeqz", -1 );
00049 return -1;
00050 }
00051 #ifndef LAPACK_DISABLE_NAN_CHECK
00052
00053 if( LAPACKE_sge_nancheck( matrix_order, n, n, h, ldh ) ) {
00054 return -8;
00055 }
00056 if( LAPACKE_lsame( compq, 'i' ) || LAPACKE_lsame( compq, 'v' ) ) {
00057 if( LAPACKE_sge_nancheck( matrix_order, n, n, q, ldq ) ) {
00058 return -15;
00059 }
00060 }
00061 if( LAPACKE_sge_nancheck( matrix_order, n, n, t, ldt ) ) {
00062 return -10;
00063 }
00064 if( LAPACKE_lsame( compz, 'i' ) || LAPACKE_lsame( compz, 'v' ) ) {
00065 if( LAPACKE_sge_nancheck( matrix_order, n, n, z, ldz ) ) {
00066 return -17;
00067 }
00068 }
00069 #endif
00070
00071 info = LAPACKE_shgeqz_work( matrix_order, job, compq, compz, n, ilo, ihi, h,
00072 ldh, t, ldt, alphar, alphai, beta, q, ldq, z,
00073 ldz, &work_query, lwork );
00074 if( info != 0 ) {
00075 goto exit_level_0;
00076 }
00077 lwork = (lapack_int)work_query;
00078
00079 work = (float*)LAPACKE_malloc( sizeof(float) * lwork );
00080 if( work == NULL ) {
00081 info = LAPACK_WORK_MEMORY_ERROR;
00082 goto exit_level_0;
00083 }
00084
00085 info = LAPACKE_shgeqz_work( matrix_order, job, compq, compz, n, ilo, ihi, h,
00086 ldh, t, ldt, alphar, alphai, beta, q, ldq, z,
00087 ldz, work, lwork );
00088
00089 LAPACKE_free( work );
00090 exit_level_0:
00091 if( info == LAPACK_WORK_MEMORY_ERROR ) {
00092 LAPACKE_xerbla( "LAPACKE_shgeqz", info );
00093 }
00094 return info;
00095 }