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_cgelsy( int matrix_order, lapack_int m, lapack_int n,
00038 lapack_int nrhs, lapack_complex_float* a,
00039 lapack_int lda, lapack_complex_float* b,
00040 lapack_int ldb, lapack_int* jpvt, float rcond,
00041 lapack_int* rank )
00042 {
00043 lapack_int info = 0;
00044 lapack_int lwork = -1;
00045 float* rwork = NULL;
00046 lapack_complex_float* work = NULL;
00047 lapack_complex_float work_query;
00048 if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) {
00049 LAPACKE_xerbla( "LAPACKE_cgelsy", -1 );
00050 return -1;
00051 }
00052 #ifndef LAPACK_DISABLE_NAN_CHECK
00053
00054 if( LAPACKE_cge_nancheck( matrix_order, m, n, a, lda ) ) {
00055 return -5;
00056 }
00057 if( LAPACKE_cge_nancheck( matrix_order, MAX(m,n), nrhs, b, ldb ) ) {
00058 return -7;
00059 }
00060 if( LAPACKE_s_nancheck( 1, &rcond, 1 ) ) {
00061 return -10;
00062 }
00063 #endif
00064
00065 rwork = (float*)LAPACKE_malloc( sizeof(float) * MAX(1,2*n) );
00066 if( rwork == NULL ) {
00067 info = LAPACK_WORK_MEMORY_ERROR;
00068 goto exit_level_0;
00069 }
00070
00071 info = LAPACKE_cgelsy_work( matrix_order, m, n, nrhs, a, lda, b, ldb, jpvt,
00072 rcond, rank, &work_query, lwork, rwork );
00073 if( info != 0 ) {
00074 goto exit_level_1;
00075 }
00076 lwork = LAPACK_C2INT( work_query );
00077
00078 work = (lapack_complex_float*)
00079 LAPACKE_malloc( sizeof(lapack_complex_float) * lwork );
00080 if( work == NULL ) {
00081 info = LAPACK_WORK_MEMORY_ERROR;
00082 goto exit_level_1;
00083 }
00084
00085 info = LAPACKE_cgelsy_work( matrix_order, m, n, nrhs, a, lda, b, ldb, jpvt,
00086 rcond, rank, work, lwork, rwork );
00087
00088 LAPACKE_free( work );
00089 exit_level_1:
00090 LAPACKE_free( rwork );
00091 exit_level_0:
00092 if( info == LAPACK_WORK_MEMORY_ERROR ) {
00093 LAPACKE_xerbla( "LAPACKE_cgelsy", info );
00094 }
00095 return info;
00096 }