lapack_int LAPACKE_zupmtr( int matrix_order, char side, char uplo, char trans, lapack_int m, lapack_int n, const lapack_complex_double* ap, const lapack_complex_double* tau, lapack_complex_double* c, lapack_int ldc ) { lapack_int info = 0; /* Additional scalars declarations for work arrays */ lapack_int lwork; lapack_complex_double* work = NULL; if( matrix_order != LAPACK_COL_MAJOR && matrix_order != LAPACK_ROW_MAJOR ) { LAPACKE_xerbla( "LAPACKE_zupmtr", -1 ); return -1; } #ifndef LAPACK_DISABLE_NAN_CHECK /* Optionally check input matrices for NaNs */ lapack_int r = LAPACKE_lsame( side, 'l' ) ? m : n; if( LAPACKE_zpp_nancheck( r, ap ) ) { return -7; } if( LAPACKE_zge_nancheck( matrix_order, m, n, c, ldc ) ) { return -9; } if( LAPACKE_z_nancheck( m-1, tau, 1 ) ) { return -8; } #endif /* Additional scalars initializations for work arrays */ if( LAPACKE_lsame( side, 'l' ) ) { lwork = MAX(1,n); } else if( LAPACKE_lsame( side, 'r' ) ) { lwork = MAX(1,m); } else { lwork = 1; /* Any value */ } /* Allocate memory for working array(s) */ work = (lapack_complex_double*) LAPACKE_malloc( sizeof(lapack_complex_double) * lwork ); if( work == NULL ) { info = LAPACK_WORK_MEMORY_ERROR; goto exit_level_0; } /* Call middle-level interface */ info = LAPACKE_zupmtr_work( matrix_order, side, uplo, trans, m, n, ap, tau, c, ldc, work ); /* Release memory and exit */ LAPACKE_free( work ); exit_level_0: if( info == LAPACK_WORK_MEMORY_ERROR ) { LAPACKE_xerbla( "LAPACKE_zupmtr", info ); } return info; }
int main(void) { /* Local scalars */ char side, side_i; char uplo, uplo_i; char trans, trans_i; lapack_int m, m_i; lapack_int n, n_i; lapack_int ldc, ldc_i; lapack_int ldc_r; lapack_int info, info_i; /* Declare scalars */ lapack_int lwork; lapack_int i; int failed; /* Local arrays */ lapack_complex_double *ap = NULL, *ap_i = NULL; lapack_complex_double *tau = NULL, *tau_i = NULL; lapack_complex_double *c = NULL, *c_i = NULL; lapack_complex_double *work = NULL, *work_i = NULL; lapack_complex_double *c_save = NULL; lapack_complex_double *ap_r = NULL; lapack_complex_double *c_r = NULL; /* Iniitialize the scalar parameters */ init_scalars_zupmtr( &side, &uplo, &trans, &m, &n, &ldc ); lwork = MAX(m,n); ldc_r = n+2; side_i = side; uplo_i = uplo; trans_i = trans; m_i = m; n_i = n; ldc_i = ldc; /* Allocate memory for the LAPACK routine arrays */ ap = (lapack_complex_double *) LAPACKE_malloc( ((m*(m+1)/2)) * sizeof(lapack_complex_double) ); tau = (lapack_complex_double *) LAPACKE_malloc( (m-1) * sizeof(lapack_complex_double) ); c = (lapack_complex_double *) LAPACKE_malloc( ldc*n * sizeof(lapack_complex_double) ); work = (lapack_complex_double *) LAPACKE_malloc( lwork * sizeof(lapack_complex_double) ); /* Allocate memory for the C interface function arrays */ ap_i = (lapack_complex_double *) LAPACKE_malloc( ((m*(m+1)/2)) * sizeof(lapack_complex_double) ); tau_i = (lapack_complex_double *) LAPACKE_malloc( (m-1) * sizeof(lapack_complex_double) ); c_i = (lapack_complex_double *) LAPACKE_malloc( ldc*n * sizeof(lapack_complex_double) ); work_i = (lapack_complex_double *) LAPACKE_malloc( lwork * sizeof(lapack_complex_double) ); /* Allocate memory for the backup arrays */ c_save = (lapack_complex_double *) LAPACKE_malloc( ldc*n * sizeof(lapack_complex_double) ); /* Allocate memory for the row-major arrays */ ap_r = (lapack_complex_double *) LAPACKE_malloc( m*(m+1)/2 * sizeof(lapack_complex_double) ); c_r = (lapack_complex_double *) LAPACKE_malloc( m*(n+2) * sizeof(lapack_complex_double) ); /* Initialize input arrays */ init_ap( (m*(m+1)/2), ap ); init_tau( (m-1), tau ); init_c( ldc*n, c ); init_work( lwork, work ); /* Backup the ouptut arrays */ for( i = 0; i < ldc*n; i++ ) { c_save[i] = c[i]; } /* Call the LAPACK routine */ zupmtr_( &side, &uplo, &trans, &m, &n, ap, tau, c, &ldc, work, &info ); /* Initialize input data, call the column-major middle-level * interface to LAPACK routine and check the results */ for( i = 0; i < (m*(m+1)/2); i++ ) { ap_i[i] = ap[i]; } for( i = 0; i < (m-1); i++ ) { tau_i[i] = tau[i]; } for( i = 0; i < ldc*n; i++ ) { c_i[i] = c_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } info_i = LAPACKE_zupmtr_work( LAPACK_COL_MAJOR, side_i, uplo_i, trans_i, m_i, n_i, ap_i, tau_i, c_i, ldc_i, work_i ); failed = compare_zupmtr( c, c_i, info, info_i, ldc, n ); if( failed == 0 ) { printf( "PASSED: column-major middle-level interface to zupmtr\n" ); } else { printf( "FAILED: column-major middle-level interface to zupmtr\n" ); } /* Initialize input data, call the column-major high-level * interface to LAPACK routine and check the results */ for( i = 0; i < (m*(m+1)/2); i++ ) { ap_i[i] = ap[i]; } for( i = 0; i < (m-1); i++ ) { tau_i[i] = tau[i]; } for( i = 0; i < ldc*n; i++ ) { c_i[i] = c_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } info_i = LAPACKE_zupmtr( LAPACK_COL_MAJOR, side_i, uplo_i, trans_i, m_i, n_i, ap_i, tau_i, c_i, ldc_i ); failed = compare_zupmtr( c, c_i, info, info_i, ldc, n ); if( failed == 0 ) { printf( "PASSED: column-major high-level interface to zupmtr\n" ); } else { printf( "FAILED: column-major high-level interface to zupmtr\n" ); } /* Initialize input data, call the row-major middle-level * interface to LAPACK routine and check the results */ for( i = 0; i < (m*(m+1)/2); i++ ) { ap_i[i] = ap[i]; } for( i = 0; i < (m-1); i++ ) { tau_i[i] = tau[i]; } for( i = 0; i < ldc*n; i++ ) { c_i[i] = c_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } LAPACKE_zpp_trans( LAPACK_COL_MAJOR, uplo, m, ap_i, ap_r ); LAPACKE_zge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 ); info_i = LAPACKE_zupmtr_work( LAPACK_ROW_MAJOR, side_i, uplo_i, trans_i, m_i, n_i, ap_r, tau_i, c_r, ldc_r, work_i ); LAPACKE_zge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc ); failed = compare_zupmtr( c, c_i, info, info_i, ldc, n ); if( failed == 0 ) { printf( "PASSED: row-major middle-level interface to zupmtr\n" ); } else { printf( "FAILED: row-major middle-level interface to zupmtr\n" ); } /* Initialize input data, call the row-major high-level * interface to LAPACK routine and check the results */ for( i = 0; i < (m*(m+1)/2); i++ ) { ap_i[i] = ap[i]; } for( i = 0; i < (m-1); i++ ) { tau_i[i] = tau[i]; } for( i = 0; i < ldc*n; i++ ) { c_i[i] = c_save[i]; } for( i = 0; i < lwork; i++ ) { work_i[i] = work[i]; } /* Init row_major arrays */ LAPACKE_zpp_trans( LAPACK_COL_MAJOR, uplo, m, ap_i, ap_r ); LAPACKE_zge_trans( LAPACK_COL_MAJOR, m, n, c_i, ldc, c_r, n+2 ); info_i = LAPACKE_zupmtr( LAPACK_ROW_MAJOR, side_i, uplo_i, trans_i, m_i, n_i, ap_r, tau_i, c_r, ldc_r ); LAPACKE_zge_trans( LAPACK_ROW_MAJOR, m, n, c_r, n+2, c_i, ldc ); failed = compare_zupmtr( c, c_i, info, info_i, ldc, n ); if( failed == 0 ) { printf( "PASSED: row-major high-level interface to zupmtr\n" ); } else { printf( "FAILED: row-major high-level interface to zupmtr\n" ); } /* Release memory */ if( ap != NULL ) { LAPACKE_free( ap ); } if( ap_i != NULL ) { LAPACKE_free( ap_i ); } if( ap_r != NULL ) { LAPACKE_free( ap_r ); } if( tau != NULL ) { LAPACKE_free( tau ); } if( tau_i != NULL ) { LAPACKE_free( tau_i ); } if( c != NULL ) { LAPACKE_free( c ); } if( c_i != NULL ) { LAPACKE_free( c_i ); } if( c_r != NULL ) { LAPACKE_free( c_r ); } if( c_save != NULL ) { LAPACKE_free( c_save ); } if( work != NULL ) { LAPACKE_free( work ); } if( work_i != NULL ) { LAPACKE_free( work_i ); } return 0; }