ksp/tests/ex16f.F90

20   PetscErrorCode ierr
37   PetscCallA(PetscInitialize(ierr))
40   PetscCallA(PetscOptionsGetString(PETSC_NULL_OPTIONS, PETSC_NULL_CHARACTER, '-f', f, flg, ierr))
41   PetscCallA(PetscViewerBinaryOpen(PETSC_COMM_WORLD, f, FILE_MODE_READ, fd, ierr))
43   PetscCallA(MatCreate(PETSC_COMM_WORLD, A, ierr))
44   PetscCallA(MatSetType(A, MATSEQAIJ, ierr))
45   PetscCallA(MatLoad(A, fd, ierr))
47   PetscCallA(VecCreate(PETSC_COMM_WORLD, b, ierr))
48   PetscCallA(VecLoad(b, fd, ierr))
49   PetscCallA(PetscViewerDestroy(fd, ierr))
52   PetscCallA(VecDuplicate(b, x, ierr))
53   PetscCallA(VecDuplicate(b, u, ierr))
56   PetscCallA(KSPCreate(PETSC_COMM_WORLD, ksp1, ierr))
57   PetscCallA(KSPSetOptionsPrefix(ksp1, 'a', ierr))
58   PetscCallA(KSPAppendOptionsPrefix(ksp1, '_', ierr))
59   PetscCallA(KSPSetOperators(ksp1, A, A, ierr))
60   PetscCallA(KSPSetFromOptions(ksp1, ierr))
61   PetscCallA(KSPSolve(ksp1, b, x, ierr))
64   PetscCallA(MatMult(A, x, u, ierr))
65   PetscCallA(VecAXPY(u, none, b, ierr))
66   PetscCallA(VecNorm(u, NORM_2, norm, ierr))
67   PetscCallA(KSPGetIterationNumber(ksp1, its, ierr))
73   PetscCallA(ISCreateStride(PETSC_COMM_SELF, ifive, izero, ione, isrow, ierr))
74   PetscCallA(MatZeroRowsIS(A, isrow, five, PETSC_NULL_VEC, PETSC_NULL_VEC, ierr))
77   PetscCallA(KSPCreate(PETSC_COMM_WORLD, ksp2, ierr))
78   PetscCallA(KSPSetOptionsPrefix(ksp2, 'b', ierr))
79   PetscCallA(KSPAppendOptionsPrefix(ksp2, '_', ierr))
80   PetscCallA(KSPSetOperators(ksp2, A, A, ierr))
81   PetscCallA(KSPSetFromOptions(ksp2, ierr))
82   PetscCallA(KSPSolve(ksp2, b, x, ierr))
85   PetscCallA(MatMult(A, x, u, ierr))
86   PetscCallA(VecAXPY(u, none, b, ierr))
87   PetscCallA(VecNorm(u, NORM_2, norm, ierr))
88   PetscCallA(KSPGetIterationNumber(ksp2, its, ierr))
92   PetscCallA(KSPDestroy(ksp1, ierr))
93   PetscCallA(KSPDestroy(ksp2, ierr))
94   PetscCallA(VecDestroy(b, ierr))
95   PetscCallA(VecDestroy(x, ierr))
96   PetscCallA(VecDestroy(u, ierr))
97   PetscCallA(MatDestroy(A, ierr))
98   PetscCallA(ISDestroy(isrow, ierr))
100   PetscCallA(PetscFinalize(ierr))