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