lapack_int LAPACKE_cgehrd( int matrix_layout, lapack_int n, lapack_int ilo, lapack_int ihi, lapack_complex_float* a, lapack_int lda, lapack_complex_float* tau ) { lapack_int info = 0; lapack_int lwork = -1; lapack_complex_float* work = NULL; lapack_complex_float work_query; if( matrix_layout != LAPACK_COL_MAJOR && matrix_layout != LAPACK_ROW_MAJOR ) { LAPACKE_xerbla( "LAPACKE_cgehrd", -1 ); return -1; } #ifndef LAPACK_DISABLE_NAN_CHECK /* Optionally check input matrices for NaNs */ if( LAPACKE_cge_nancheck( matrix_layout, n, n, a, lda ) ) { return -5; } #endif /* Query optimal working array(s) size */ info = LAPACKE_cgehrd_work( matrix_layout, n, ilo, ihi, a, lda, tau, &work_query, lwork ); if( info != 0 ) { goto exit_level_0; } lwork = LAPACK_C2INT( work_query ); /* Allocate memory for work arrays */ work = (lapack_complex_float*) LAPACKE_malloc( sizeof(lapack_complex_float) * lwork ); if( work == NULL ) { info = LAPACK_WORK_MEMORY_ERROR; goto exit_level_0; } /* Call middle-level interface */ info = LAPACKE_cgehrd_work( matrix_layout, n, ilo, ihi, a, lda, tau, work, lwork ); /* Release memory and exit */ LAPACKE_free( work ); exit_level_0: if( info == LAPACK_WORK_MEMORY_ERROR ) { LAPACKE_xerbla( "LAPACKE_cgehrd", info ); } return info; }
int main(void) { /* Local scalars */ lapack_int n, n_i; lapack_int ilo, ilo_i; lapack_int ihi, ihi_i; lapack_int lda, lda_i; lapack_int lda_r; lapack_int lwork, lwork_i; lapack_int info, info_i; lapack_int i; int failed; /* Local arrays */ lapack_complex_float *a = NULL, *a_i = NULL; lapack_complex_float *tau = NULL, *tau_i = NULL; lapack_complex_float *work = NULL, *work_i = NULL; lapack_complex_float *a_save = NULL; lapack_complex_float *tau_save = NULL; lapack_complex_float *a_r = NULL; /* Iniitialize the scalar parameters */ init_scalars_cgehrd( &n, &ilo, &ihi, &lda, &lwork ); lda_r = n+2; n_i = n; ilo_i = ilo; ihi_i = ihi; lda_i = lda; lwork_i = lwork; /* Allocate memory for the LAPACK routine arrays */ a = (lapack_complex_float *) LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) ); tau = (lapack_complex_float *) LAPACKE_malloc( (n-1) * sizeof(lapack_complex_float) ); work = (lapack_complex_float *) LAPACKE_malloc( lwork * sizeof(lapack_complex_float) ); /* Allocate memory for the C interface function arrays */ a_i = (lapack_complex_float *) LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) ); tau_i = (lapack_complex_float *) LAPACKE_malloc( (n-1) * sizeof(lapack_complex_float) ); work_i = (lapack_complex_float *) LAPACKE_malloc( lwork * sizeof(lapack_complex_float) ); /* Allocate memory for the backup arrays */ a_save = (lapack_complex_float *) LAPACKE_malloc( lda*n * sizeof(lapack_complex_float) ); tau_save = (lapack_complex_float *) LAPACKE_malloc( (n-1) * sizeof(lapack_complex_float) ); /* Allocate memory for the row-major arrays */ a_r = (lapack_complex_float *) LAPACKE_malloc( n*(n+2) * sizeof(lapack_complex_float) ); /* Initialize input arrays */ init_a( lda*n, a ); init_tau( (n-1), tau ); init_work( lwork, work ); /* Backup the ouptut arrays */ for( i = 0; i < lda*n; i++ ) { a_save[i] = a[i]; } for( i = 0; i < (n-1); i++ ) { tau_save[i] = tau[i]; } /* Call the LAPACK routine */ cgehrd_( &n, &ilo, &ihi, a, &lda, tau, work, &lwork, &info ); /* Initialize input data, call the column-major middle-level * interface to LAPACK routine and check the results */ for( i = 0; i < lda*n; i++ ) { a_i[i] = a_save[i]; } for( i = 0; i < (n-1); i++ ) { tau_i[i] = tau_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } info_i = LAPACKE_cgehrd_work( LAPACK_COL_MAJOR, n_i, ilo_i, ihi_i, a_i, lda_i, tau_i, work_i, lwork_i ); failed = compare_cgehrd( a, a_i, tau, tau_i, info, info_i, lda, n ); if( failed == 0 ) { printf( "PASSED: column-major middle-level interface to cgehrd\n" ); } else { printf( "FAILED: column-major middle-level interface to cgehrd\n" ); } /* Initialize input data, call the column-major high-level * interface to LAPACK routine and check the results */ for( i = 0; i < lda*n; i++ ) { a_i[i] = a_save[i]; } for( i = 0; i < (n-1); i++ ) { tau_i[i] = tau_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } info_i = LAPACKE_cgehrd( LAPACK_COL_MAJOR, n_i, ilo_i, ihi_i, a_i, lda_i, tau_i ); failed = compare_cgehrd( a, a_i, tau, tau_i, info, info_i, lda, n ); if( failed == 0 ) { printf( "PASSED: column-major high-level interface to cgehrd\n" ); } else { printf( "FAILED: column-major high-level interface to cgehrd\n" ); } /* Initialize input data, call the row-major middle-level * interface to LAPACK routine and check the results */ for( i = 0; i < lda*n; i++ ) { a_i[i] = a_save[i]; } for( i = 0; i < (n-1); i++ ) { tau_i[i] = tau_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 ); info_i = LAPACKE_cgehrd_work( LAPACK_ROW_MAJOR, n_i, ilo_i, ihi_i, a_r, lda_r, tau_i, work_i, lwork_i ); LAPACKE_cge_trans( LAPACK_ROW_MAJOR, n, n, a_r, n+2, a_i, lda ); failed = compare_cgehrd( a, a_i, tau, tau_i, info, info_i, lda, n ); if( failed == 0 ) { printf( "PASSED: row-major middle-level interface to cgehrd\n" ); } else { printf( "FAILED: row-major middle-level interface to cgehrd\n" ); } /* Initialize input data, call the row-major high-level * interface to LAPACK routine and check the results */ for( i = 0; i < lda*n; i++ ) { a_i[i] = a_save[i]; } for( i = 0; i < (n-1); i++ ) { tau_i[i] = tau_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } /* Init row_major arrays */ LAPACKE_cge_trans( LAPACK_COL_MAJOR, n, n, a_i, lda, a_r, n+2 ); info_i = LAPACKE_cgehrd( LAPACK_ROW_MAJOR, n_i, ilo_i, ihi_i, a_r, lda_r, tau_i ); LAPACKE_cge_trans( LAPACK_ROW_MAJOR, n, n, a_r, n+2, a_i, lda ); failed = compare_cgehrd( a, a_i, tau, tau_i, info, info_i, lda, n ); if( failed == 0 ) { printf( "PASSED: row-major high-level interface to cgehrd\n" ); } else { printf( "FAILED: row-major high-level interface to cgehrd\n" ); } /* Release memory */ if( a != NULL ) { LAPACKE_free( a ); } if( a_i != NULL ) { LAPACKE_free( a_i ); } if( a_r != NULL ) { LAPACKE_free( a_r ); } if( a_save != NULL ) { LAPACKE_free( a_save ); } if( tau != NULL ) { LAPACKE_free( tau ); } if( tau_i != NULL ) { LAPACKE_free( tau_i ); } if( tau_save != NULL ) { LAPACKE_free( tau_save ); } if( work != NULL ) { LAPACKE_free( work ); } if( work_i != NULL ) { LAPACKE_free( work_i ); } return 0; }