Actual source code: ex101.c
1: static char help[] = "Testing PtAP for SeqMAIJ matrix, P, with SeqAIJ matrix, A.\n\n";
3: #include <petscmat.h>
5: int main(int argc, char **argv)
6: {
7: Mat pA, P, aijP;
8: PetscScalar pa[] = {1., -1., 0., 0., 1., -1., 0., 0., 1.};
9: PetscInt i, pij[] = {0, 1, 2};
10: PetscInt aij[3][3] = {
11: {0, 1, 2},
12: {3, 4, 5},
13: {6, 7, 8}
14: };
15: Mat A, mC, C;
16: PetscScalar one = 1.;
17: PetscMPIInt size;
18: PetscBool flg;
19: MatProductAlgorithm alg;
21: PetscFunctionBeginUser;
22: PetscCall(PetscInitialize(&argc, &argv, NULL, help));
23: PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
24: PetscCheck(size == 1, PETSC_COMM_WORLD, PETSC_ERR_WRONG_MPI_SIZE, "This is a uniprocessor example only!");
26: /* Create MAIJ matrix, P */
27: PetscCall(MatCreate(PETSC_COMM_SELF, &pA));
28: PetscCall(MatSetSizes(pA, 3, 3, 3, 3));
29: PetscCall(MatSetType(pA, MATSEQAIJ));
30: PetscCall(MatSetUp(pA));
31: PetscCall(MatSetOption(pA, MAT_IGNORE_ZERO_ENTRIES, PETSC_TRUE));
32: PetscCall(MatSetValues(pA, 3, pij, 3, pij, pa, ADD_VALUES));
33: PetscCall(MatAssemblyBegin(pA, MAT_FINAL_ASSEMBLY));
34: PetscCall(MatAssemblyEnd(pA, MAT_FINAL_ASSEMBLY));
35: PetscCall(MatCreateMAIJ(pA, 3, &P));
36: PetscCall(MatDestroy(&pA));
38: /* Create AIJ equivalent matrix, aijP, for comparison testing */
39: PetscCall(MatConvert(P, MATSEQAIJ, MAT_INITIAL_MATRIX, &aijP));
41: /* Create AIJ matrix A */
42: PetscCall(MatCreate(PETSC_COMM_SELF, &A));
43: PetscCall(MatSetSizes(A, 9, 9, 9, 9));
44: PetscCall(MatSetType(A, MATSEQAIJ));
45: PetscCall(MatSetUp(A));
46: PetscCall(MatSetOption(A, MAT_IGNORE_ZERO_ENTRIES, PETSC_TRUE));
47: PetscCall(MatSetValues(A, 3, aij[0], 3, aij[0], pa, ADD_VALUES));
48: PetscCall(MatSetValues(A, 3, aij[1], 3, aij[1], pa, ADD_VALUES));
49: PetscCall(MatSetValues(A, 3, aij[2], 3, aij[2], pa, ADD_VALUES));
50: for (i = 0; i < 9; i++) PetscCall(MatSetValue(A, i, i, one, ADD_VALUES));
51: PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
52: PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
54: /* Perform PtAP_SeqAIJ_SeqAIJ for comparison testing */
55: PetscCall(MatPtAP(A, aijP, MAT_INITIAL_MATRIX, 1., &C));
57: /* Perform PtAP_SeqAIJ_SeqMAIJ */
58: /* Developer API */
59: PetscCall(MatProductCreate(A, P, NULL, &mC));
60: PetscCall(MatProductSetType(mC, MATPRODUCT_PtAP));
61: PetscCall(MatProductSetAlgorithm(mC, MATPRODUCTALGORITHMDEFAULT));
62: PetscCall(MatProductSetFill(mC, PETSC_DEFAULT));
63: PetscCall(MatProductSetFromOptions(mC));
64: PetscCall(MatProductSymbolic(mC));
65: PetscCall(MatProductNumeric(mC));
66: PetscCall(MatProductNumeric(mC));
67: PetscCall(MatProductGetAlgorithm(mC, &alg));
68: PetscCall(PetscPrintf(PETSC_COMM_SELF, "MatProduct algorithm: %s\n", alg));
70: /* Check mC = C */
71: PetscCall(MatEqual(C, mC, &flg));
72: PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "MatProduct C != mC");
73: PetscCall(MatDestroy(&mC));
75: /* User API */
76: PetscCall(MatPtAP(A, P, MAT_INITIAL_MATRIX, 1., &mC));
77: PetscCall(MatPtAP(A, P, MAT_REUSE_MATRIX, 1., &mC));
79: /* Check mC = C */
80: PetscCall(MatEqual(C, mC, &flg));
81: PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "MatPtAP C != mC");
82: PetscCall(MatDestroy(&mC));
84: /* Cleanup */
85: PetscCall(MatDestroy(&P));
86: PetscCall(MatDestroy(&aijP));
87: PetscCall(MatDestroy(&A));
88: PetscCall(MatDestroy(&C));
89: PetscCall(PetscFinalize());
90: return 0;
91: }
93: /*TEST
95: test:
96: output_file: output/ex101.out
98: TEST*/