static int RunTest(int *iparam, double *dparam, real_Double_t *t_) { PLASMA_desc *L; int *piv; PASTE_CODE_IPARAM_LOCALS( iparam ); if ( M != N ) { fprintf(stderr, "This timing works only with M == N\n"); return -1; } /* Allocate Data */ PASTE_CODE_ALLOCATE_MATRIX( A, 1, PLASMA_Complex64_t, LDA, N ); PASTE_CODE_ALLOCATE_MATRIX( X, 1, PLASMA_Complex64_t, LDB, NRHS ); /* Initialiaze Data */ PLASMA_zplrnt( N, N, A, LDA, 51 ); PLASMA_zplrnt( N, NRHS, X, LDB, 5673 ); PLASMA_Alloc_Workspace_zgesv_incpiv(N, &L, &piv); /* Save A and b */ PASTE_CODE_ALLOCATE_COPY( Acpy, check, PLASMA_Complex64_t, A, LDA, N ); PASTE_CODE_ALLOCATE_COPY( B, check, PLASMA_Complex64_t, X, LDB, NRHS ); START_TIMING(); PLASMA_zgesv_incpiv( N, NRHS, A, LDA, L, piv, X, LDB ); STOP_TIMING(); /* Check the solution */ if (check) { dparam[IPARAM_RES] = z_check_solution(N, N, NRHS, Acpy, LDA, B, X, LDB, &(dparam[IPARAM_ANORM]), &(dparam[IPARAM_BNORM]), &(dparam[IPARAM_XNORM])); free(Acpy); free(B); } PLASMA_Dealloc_Handle_Tile( &L ); free( piv ); free( X ); free( A ); return 0; }
int testing_zgesv_incpiv(int argc, char **argv) { /* Check for valid arguments*/ if (argc != 4){ USAGE("GESV_INCPIV", "N LDA NRHS LDB", " - N : the size of the matrix\n" " - LDA : leading dimension of the matrix A\n" " - NRHS : number of RHS\n" " - LDB : leading dimension of the matrix B\n"); return -1; } int N = atoi(argv[0]); int LDA = atoi(argv[1]); int NRHS = atoi(argv[2]); int LDB = atoi(argv[3]); double eps; int info_solution; int i,j; int LDAxN = LDA*N; int LDBxNRHS = LDB*NRHS; PLASMA_Complex64_t *A1 = (PLASMA_Complex64_t *)malloc(LDA*N*(sizeof*A1)); PLASMA_Complex64_t *A2 = (PLASMA_Complex64_t *)malloc(LDA*N*(sizeof*A2)); PLASMA_Complex64_t *B1 = (PLASMA_Complex64_t *)malloc(LDB*NRHS*(sizeof*B1)); PLASMA_Complex64_t *B2 = (PLASMA_Complex64_t *)malloc(LDB*NRHS*(sizeof*B2)); PLASMA_Complex64_t *L; int *IPIV; /* Check if unable to allocate memory */ if ( (!A1) || (!A2)|| (!B1) || (!B2) ) { printf("Out of Memory \n "); return -2; } eps = BLAS_dfpinfo(blas_eps); /*---------------------------------------------------------- * TESTING ZGESV */ /* Initialize A1 and A2 Matrix */ LAPACKE_zlarnv_work(IONE, ISEED, LDAxN, A1); for ( i = 0; i < N; i++) for ( j = 0; j < N; j++) A2[LDA*j+i] = A1[LDA*j+i]; /* Initialize B1 and B2 */ LAPACKE_zlarnv_work(IONE, ISEED, LDBxNRHS, B1); for ( i = 0; i < N; i++) for ( j = 0; j < NRHS; j++) B2[LDB*j+i] = B1[LDB*j+i]; /* PLASMA ZGESV */ PLASMA_Alloc_Workspace_zgesv_incpiv(N, &L, &IPIV); PLASMA_zgesv_incpiv(N, NRHS, A2, LDA, L, IPIV, B2, LDB); printf("\n"); printf("------ TESTS FOR PLASMA INCPIV ZGESV ROUTINE ------- \n"); printf(" Size of the Matrix %d by %d\n", N, N); printf("\n"); printf(" The matrix A is randomly generated for each test.\n"); printf("============\n"); printf(" The relative machine precision (eps) is to be %e \n", eps); printf(" Computational tests pass if scaled residuals are less than 60.\n"); /* Check the factorization and the solution */ info_solution = check_solution(N, NRHS, A1, LDA, B1, B2, LDB, eps); if ((info_solution == 0)){ printf("***************************************************\n"); printf(" ---- TESTING INCPIV ZGESV ............... PASSED !\n"); printf("***************************************************\n"); } else{ printf("************************************************\n"); printf(" - TESTING INCPIV ZGESV ... FAILED !\n"); printf("************************************************\n"); } /*------------------------------------------------------------- * TESTING ZGETRF + ZGETRS */ /* Initialize A1 and A2 */ LAPACKE_zlarnv_work(IONE, ISEED, LDAxN, A1); for ( i = 0; i < N; i++) for ( j = 0; j < N; j++) A2[LDA*j+i] = A1[LDA*j+i]; /* Initialize B1 and B2 */ LAPACKE_zlarnv_work(IONE, ISEED, LDBxNRHS, B1); for ( i = 0; i < N; i++) for ( j = 0; j < NRHS; j++) B2[LDB*j+i] = B1[LDB*j+i]; /* Plasma routines */ PLASMA_zgetrf_incpiv(N, N, A2, LDA, L, IPIV); PLASMA_zgetrs_incpiv(PlasmaNoTrans, N, NRHS, A2, LDA, L, IPIV, B2, LDB); printf("\n"); printf("------ TESTS FOR PLASMA ZGETRF + ZGETRS ROUTINE ------- \n"); printf(" Size of the Matrix %d by %d\n", N, N); printf("\n"); printf(" The matrix A is randomly generated for each test.\n"); printf("============\n"); printf(" The relative machine precision (eps) is to be %e \n", eps); printf(" Computational tests pass if scaled residuals are less than 60.\n"); /* Check the solution */ info_solution = check_solution(N, NRHS, A1, LDA, B1, B2, LDB, eps); if ((info_solution == 0)){ printf("***************************************************\n"); printf(" ---- TESTING INCPIV ZGETRF + ZGETRS ..... PASSED !\n"); printf("***************************************************\n"); } else{ printf("***************************************************\n"); printf(" - TESTING INCPIV ZGETRF + ZGETRS ... FAILED !\n"); printf("***************************************************\n"); } /*------------------------------------------------------------- * TESTING ZGETRF + ZTRSMPL + ZTRSM */ /* Initialize A1 and A2 */ LAPACKE_zlarnv_work(IONE, ISEED, LDAxN, A1); for ( i = 0; i < N; i++) for ( j = 0; j < N; j++) A2[LDA*j+i] = A1[LDA*j+i]; /* Initialize B1 and B2 */ LAPACKE_zlarnv_work(IONE, ISEED, LDBxNRHS, B1); for ( i = 0; i < N; i++) for ( j = 0; j < NRHS; j++) B2[LDB*j+i] = B1[LDB*j+i]; /* PLASMA routines */ PLASMA_zgetrf_incpiv(N, N, A2, LDA, L, IPIV); PLASMA_ztrsmpl(N, NRHS, A2, LDA, L, IPIV, B2, LDB); PLASMA_ztrsm(PlasmaLeft, PlasmaUpper, PlasmaNoTrans, PlasmaNonUnit, N, NRHS, 1.0, A2, LDA, B2, LDB); printf("\n"); printf("------ TESTS FOR PLASMA INCPIV ZGETRF + ZTRSMPL + ZTRSM ROUTINE ------- \n"); printf(" Size of the Matrix %d by %d\n", N, N); printf("\n"); printf(" The matrix A is randomly generated for each test.\n"); printf("============\n"); printf(" The relative machine precision (eps) is to be %e \n", eps); printf(" Computational tests pass if scaled residuals are less than 60.\n"); /* Check the solution */ info_solution = check_solution(N, NRHS, A1, LDA, B1, B2, LDB, eps); if ((info_solution == 0)){ printf("***************************************************\n"); printf(" ---- TESTING INCPIV ZGETRF + ZTRSMPL + ZTRSM ... PASSED !\n"); printf("***************************************************\n"); } else{ printf("**************************************************\n"); printf(" - TESTING INCPIV ZGETRF + ZTRSMPL + ZTRSM ... FAILED !\n"); printf("**************************************************\n"); } free(A1); free(A2); free(B1); free(B2); free(IPIV); free(L); return 0; }
void PLASMA_ALLOC_WORKSPACE_ZGESV_INCPIV(int *N, PLASMA_Complex64_t **L, int **IPIV, int *INFO) { *INFO = PLASMA_Alloc_Workspace_zgesv_incpiv(*N, L, IPIV); }