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_zunmhr_work( int matrix_order, char side, char trans,
00038 lapack_int m, lapack_int n, lapack_int ilo,
00039 lapack_int ihi, const lapack_complex_double* a,
00040 lapack_int lda,
00041 const lapack_complex_double* tau,
00042 lapack_complex_double* c, lapack_int ldc,
00043 lapack_complex_double* work, lapack_int lwork )
00044 {
00045 lapack_int info = 0;
00046 if( matrix_order == LAPACK_COL_MAJOR ) {
00047
00048 LAPACK_zunmhr( &side, &trans, &m, &n, &ilo, &ihi, a, &lda, tau, c, &ldc,
00049 work, &lwork, &info );
00050 if( info < 0 ) {
00051 info = info - 1;
00052 }
00053 } else if( matrix_order == LAPACK_ROW_MAJOR ) {
00054 lapack_int r = LAPACKE_lsame( side, 'l' ) ? m : n;
00055 lapack_int lda_t = MAX(1,r);
00056 lapack_int ldc_t = MAX(1,m);
00057 lapack_complex_double* a_t = NULL;
00058 lapack_complex_double* c_t = NULL;
00059
00060 if( lda < r ) {
00061 info = -9;
00062 LAPACKE_xerbla( "LAPACKE_zunmhr_work", info );
00063 return info;
00064 }
00065 if( ldc < n ) {
00066 info = -12;
00067 LAPACKE_xerbla( "LAPACKE_zunmhr_work", info );
00068 return info;
00069 }
00070
00071 if( lwork == -1 ) {
00072 LAPACK_zunmhr( &side, &trans, &m, &n, &ilo, &ihi, a, &lda_t, tau, c,
00073 &ldc_t, work, &lwork, &info );
00074 return (info < 0) ? (info - 1) : info;
00075 }
00076
00077 a_t = (lapack_complex_double*)
00078 LAPACKE_malloc( sizeof(lapack_complex_double) * lda_t * MAX(1,r) );
00079 if( a_t == NULL ) {
00080 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00081 goto exit_level_0;
00082 }
00083 c_t = (lapack_complex_double*)
00084 LAPACKE_malloc( sizeof(lapack_complex_double) * ldc_t * MAX(1,n) );
00085 if( c_t == NULL ) {
00086 info = LAPACK_TRANSPOSE_MEMORY_ERROR;
00087 goto exit_level_1;
00088 }
00089
00090 LAPACKE_zge_trans( matrix_order, r, r, a, lda, a_t, lda_t );
00091 LAPACKE_zge_trans( matrix_order, m, n, c, ldc, c_t, ldc_t );
00092
00093 LAPACK_zunmhr( &side, &trans, &m, &n, &ilo, &ihi, a_t, &lda_t, tau, c_t,
00094 &ldc_t, work, &lwork, &info );
00095 if( info < 0 ) {
00096 info = info - 1;
00097 }
00098
00099 LAPACKE_zge_trans( LAPACK_COL_MAJOR, m, n, c_t, ldc_t, c, ldc );
00100
00101 LAPACKE_free( c_t );
00102 exit_level_1:
00103 LAPACKE_free( a_t );
00104 exit_level_0:
00105 if( info == LAPACK_TRANSPOSE_MEMORY_ERROR ) {
00106 LAPACKE_xerbla( "LAPACKE_zunmhr_work", info );
00107 }
00108 } else {
00109 info = -1;
00110 LAPACKE_xerbla( "LAPACKE_zunmhr_work", info );
00111 }
00112 return info;
00113 }