static int
RunTest(int *iparam, double *dparam, real_Double_t *t_)
{
PLASMA_desc *T;
PASTE_CODE_IPARAM_LOCALS( iparam );
if ( M != N && check ) {
fprintf(stderr, "Check cannot be perfomed with M != N\n");
check = 0;
}
/* Allocate Data */
PASTE_CODE_ALLOCATE_MATRIX( A, 1, PLASMA_Complex64_t, LDA, N );
/* Initialize Data */
PLASMA_zplrnt(M, N, A, LDA, 3456);
/* Allocate Workspace */
PLASMA_Alloc_Workspace_zgels(M, N, &T);
/* Save AT in lapack layout for check */
PASTE_CODE_ALLOCATE_COPY( Acpy, check, PLASMA_Complex64_t, A, LDA, N );
START_TIMING();
PLASMA_zgeqrf( M, N, A, LDA, T );
STOP_TIMING();
/* Check the solution */
if ( check )
{
PASTE_CODE_ALLOCATE_MATRIX( X, 1, PLASMA_Complex64_t, LDB, NRHS );
PLASMA_zplrnt( N, NRHS, X, LDB, 5673 );
PASTE_CODE_ALLOCATE_COPY( B, 1, PLASMA_Complex64_t, X, LDB, NRHS );
PLASMA_zgeqrs(M, N, NRHS, A, LDA, T, X, LDB);
dparam[IPARAM_RES] = z_check_solution(M, N, NRHS, Acpy, LDA, B, X, LDB,
&(dparam[IPARAM_ANORM]),
&(dparam[IPARAM_BNORM]),
&(dparam[IPARAM_XNORM]));
free( Acpy );
free( B );
free( X );
}
/* Free Workspace */
PLASMA_Dealloc_Handle_Tile( &T );
free( A );
return 0;
}
int P_zgeqrf(
int M,
int N,
void *A,
void *T
) {
int info;
#if CHECK_VERSION_BEQ(2,4,5)
info = PLASMA_zgeqrf(M, N, A, M, T);
#else
PLASMA_desc *descT;
int NB, IB;
int MT, NT;
/* Get autotuned or set tile size; T matrix allocated with R */
PLASMA_Alloc_Workspace_zgeqrf(1, 1, &descT);
PLASMA_Get(PLASMA_TILE_SIZE, &NB);
PLASMA_Get(PLASMA_INNER_BLOCK_SIZE, &IB);
PLASMA_Dealloc_Handle_Tile(&descT);
MT = (M%NB==0) ? (M/NB) : (M/NB+1);
NT = (N%NB==0) ? (N/NB) : (N/NB+1);
// possibly allocate space for descT in R and keep it in qr object instead
info = PLASMA_Desc_Create(&descT, T, PlasmaComplexDouble,
IB, NB, IB*NB, MT*IB, NT*NB, 0, 0, MT*IB, NT*NB);
info = PLASMA_zgeqrf(M, N, A, M, descT);
PLASMA_Desc_Destroy(&descT);
#endif
return(info);
}
void PLASMA_ZGEQRF(int *M, int *N, PLASMA_Complex64_t *A, int *LDA, PLASMA_Complex64_t **T, int *INFO)
{ *INFO = PLASMA_zgeqrf(*M, *N, A, *LDA, *T); }
int main ()
{
int cores = 2;
int M = 15;
int N = 10;
int LDA = 15;
int K = min(M, N);
int info;
int info_ortho, info_factorization;
int i,j;
int LDAxN = LDA*N;
PLASMA_Complex64_t *A1 = (PLASMA_Complex64_t *)malloc(LDA*N*sizeof(PLASMA_Complex64_t));
PLASMA_Complex64_t *A2 = (PLASMA_Complex64_t *)malloc(LDA*N*sizeof(PLASMA_Complex64_t));
PLASMA_Complex64_t *Q = (PLASMA_Complex64_t *)malloc(LDA*N*sizeof(PLASMA_Complex64_t));
PLASMA_desc *T;
/* Check if unable to allocate memory */
if ((!A1)||(!A2)||(!Q)){
printf("Out of Memory \n ");
return EXIT_SUCCESS;
}
/* Plasma Initialization */
PLASMA_Init(cores);
printf("-- PLASMA is initialized to run on %d cores. \n",cores);
/* Allocate T */
PLASMA_Alloc_Workspace_zgeqrf(M, N, &T);
/* Initialize A1 and A2 */
LAPACKE_zlarnv_work(IONE, ISEED, LDAxN, A1);
for (i = 0; i < M; i++)
for (j = 0; j < N; j++)
A2[LDA*j+i] = A1[LDA*j+i] ;
/* Factorization QR of the matrix A2 */
info = PLASMA_zgeqrf(M, N, A2, LDA, T);
/* Building the economy-size Q */
memset((void*)Q, 0, LDA*N*sizeof(PLASMA_Complex64_t));
for (i = 0; i < K; i++)
Q[LDA*i+i] = 1.0;
PLASMA_zungqr(M, N, K, A2, LDA, T, Q, LDA);
/* Check the orthogonality, factorization and the solution */
info_ortho = check_orthogonality(M, N, LDA, Q);
info_factorization = check_factorization(M, N, A1, A2, LDA, Q);
printf("--- info %d %d %d \n",info_factorization,info_ortho,info);
if ((info_ortho != 0)|(info_factorization != 0)|(info != 0))
printf("-- Error in ZGEQRF example ! \n");
else
printf("-- Run of ZGEQRF example successful ! \n");
free(A1); free(A2); free(Q); free(T);
PLASMA_Finalize();
return EXIT_SUCCESS;
}
