Actual source code: ex232.c
2: /* tests MatSeqSBAIJSetPreallocationCSR() and MatMPISBAIJSetPreallocationCSR() */
4: #include <petsc.h>
6: int main(int argc, char **args)
7: {
8: PetscInt ia[3] = {0, 2, 4};
9: PetscInt ja[4] = {0, 1, 0, 1};
10: PetscScalar c[16] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
11: Mat ssbaij, msbaij;
12: PetscBool v2 = PETSC_FALSE;
14: PetscFunctionBeginUser;
15: PetscCall(PetscInitialize(&argc, &args, (char *)0, (char *)0));
16: PetscCall(PetscOptionsGetBool(NULL, NULL, "-v2", &v2, NULL));
18: PetscCall(MatCreate(PETSC_COMM_SELF, &ssbaij));
19: PetscCall(MatCreate(PETSC_COMM_SELF, &msbaij));
20: if (!v2) {
21: PetscCall(MatSetType(ssbaij, MATSEQSBAIJ));
22: PetscCall(MatSetType(msbaij, MATMPISBAIJ));
23: }
24: PetscCall(MatSetBlockSize(ssbaij, 2));
25: PetscCall(MatSetSizes(ssbaij, 4, 4, 4, 4));
26: PetscCall(MatSetBlockSize(msbaij, 2));
27: PetscCall(MatSetSizes(msbaij, 4, 4, 4, 4));
28: if (v2) {
29: PetscCall(MatSetUp(ssbaij));
30: PetscCall(MatSetUp(msbaij));
31: PetscCall(MatSetType(ssbaij, MATSEQSBAIJ));
32: PetscCall(MatSetType(msbaij, MATMPISBAIJ));
33: }
34: PetscCall(MatSeqSBAIJSetPreallocationCSR(ssbaij, 2, ia, ja, c));
35: PetscCall(MatMPISBAIJSetPreallocationCSR(msbaij, 2, ia, ja, c));
36: PetscCall(MatView(ssbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF)));
37: PetscCall(MatView(msbaij, PETSC_VIEWER_STDOUT_(PETSC_COMM_SELF)));
38: PetscCall(MatDestroy(&ssbaij));
39: PetscCall(MatDestroy(&msbaij));
40: PetscCall(PetscFinalize());
41: return 0;
42: }
44: /*TEST
46: test:
47: filter: sed "s?\.??g"
49: test:
50: suffix: 2
51: filter: sed "s?\.??g"
52: args : -v2
53: output_file: output/ex232_1.out
55: TEST*/