static char help[] = "Tests external direct solvers. Simplified from ex125.c\n\ Example: mpiexec -n ./ex130 -f -mat_solver_type 1 -mat_superlu_equil \n\n"; #include int main(int argc, char **args) { Mat A, F; Vec u, x, b; PetscMPIInt rank, size; PetscInt m, n, nfact, ipack = 0; PetscReal norm, tol = 1.e-12, Anorm; IS perm, iperm; MatFactorInfo info; PetscBool flg, testMatSolve = PETSC_TRUE; PetscViewer fd; /* viewer */ char file[PETSC_MAX_PATH_LEN]; /* input file name */ PetscFunctionBeginUser; PetscCall(PetscInitialize(&argc, &args, NULL, help)); PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank)); PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size)); /* Determine file from which we read the matrix A */ PetscCall(PetscOptionsGetString(NULL, NULL, "-f", file, sizeof(file), &flg)); PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must indicate binary file with the -f option"); /* Load matrix A */ PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, file, FILE_MODE_READ, &fd)); PetscCall(MatCreate(PETSC_COMM_WORLD, &A)); PetscCall(MatLoad(A, fd)); PetscCall(VecCreate(PETSC_COMM_WORLD, &b)); PetscCall(VecLoad(b, fd)); PetscCall(PetscViewerDestroy(&fd)); PetscCall(MatGetLocalSize(A, &m, &n)); PetscCheck(m == n, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "This example is not intended for rectangular matrices (%d, %d)", m, n); PetscCall(MatNorm(A, NORM_INFINITY, &Anorm)); /* Create vectors */ PetscCall(VecDuplicate(b, &x)); PetscCall(VecDuplicate(x, &u)); /* save the true solution */ /* Test LU Factorization */ PetscCall(MatGetOrdering(A, MATORDERINGNATURAL, &perm, &iperm)); PetscCall(PetscOptionsGetInt(NULL, NULL, "-mat_solver_type", &ipack, NULL)); switch (ipack) { case 1: #if defined(PETSC_HAVE_SUPERLU) if (rank == 0) printf(" SUPERLU LU:\n"); PetscCall(MatGetFactor(A, MATSOLVERSUPERLU, MAT_FACTOR_LU, &F)); break; #endif case 2: #if defined(PETSC_HAVE_MUMPS) if (rank == 0) printf(" MUMPS LU:\n"); PetscCall(MatGetFactor(A, MATSOLVERMUMPS, MAT_FACTOR_LU, &F)); { /* test mumps options */ PetscInt icntl_7 = 5; PetscCall(MatMumpsSetIcntl(F, 7, icntl_7)); } break; #endif default: if (rank == 0) printf(" PETSC LU:\n"); PetscCall(MatGetFactor(A, MATSOLVERPETSC, MAT_FACTOR_LU, &F)); } info.fill = 5.0; PetscCall(MatLUFactorSymbolic(F, A, perm, iperm, &info)); for (nfact = 0; nfact < 1; nfact++) { if (rank == 0) printf(" %d-the LU numfactorization \n", nfact); PetscCall(MatLUFactorNumeric(F, A, &info)); /* Test MatSolve() */ if (testMatSolve) { PetscCall(MatSolve(F, b, x)); /* Check the residual */ PetscCall(MatMult(A, x, u)); PetscCall(VecAXPY(u, -1.0, b)); PetscCall(VecNorm(u, NORM_INFINITY, &norm)); if (norm > tol) { if (rank == 0) PetscCall(PetscPrintf(PETSC_COMM_SELF, "MatSolve: rel residual %g/%g = %g, LU numfact %d\n", norm, Anorm, norm / Anorm, nfact)); } } } /* Free data structures */ PetscCall(MatDestroy(&A)); PetscCall(MatDestroy(&F)); PetscCall(ISDestroy(&perm)); PetscCall(ISDestroy(&iperm)); PetscCall(VecDestroy(&x)); PetscCall(VecDestroy(&b)); PetscCall(VecDestroy(&u)); PetscCall(PetscFinalize()); return 0; }