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