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