Actual source code: ex122.c
1: static char help[] = "Test MatMatMult() for AIJ and Dense matrices.\n\n";
3: #include <petscmat.h>
5: int main(int argc, char **argv)
6: {
7: Mat A, B, C;
8: PetscInt M = 10, N = 5;
9: PetscRandom r;
10: PetscBool equal = PETSC_FALSE;
11: PetscReal fill = 1.0;
12: PetscInt nza, am, an;
15: PetscInitialize(&argc, &argv, (char *)0, help);
16: PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL);
17: PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL);
18: PetscOptionsGetReal(NULL, NULL, "-fill", &fill, NULL);
20: PetscRandomCreate(PETSC_COMM_WORLD, &r);
21: PetscRandomSetFromOptions(r);
23: /* create a aij matrix A */
24: MatCreate(PETSC_COMM_WORLD, &A);
25: MatSetSizes(A, PETSC_DECIDE, PETSC_DECIDE, N, M);
26: MatSetType(A, MATAIJ);
27: nza = (PetscInt)(.3 * M); /* num of nozeros in each row of A */
28: MatSeqAIJSetPreallocation(A, nza, NULL);
29: MatMPIAIJSetPreallocation(A, nza, NULL, nza, NULL);
30: MatSetRandom(A, r);
32: /* create a dense matrix B */
33: MatGetLocalSize(A, &am, &an);
34: MatCreate(PETSC_COMM_WORLD, &B);
35: MatSetSizes(B, PETSC_DECIDE, am, N, PETSC_DECIDE);
36: MatSetType(B, MATDENSE);
37: MatSeqDenseSetPreallocation(B, NULL);
38: MatMPIDenseSetPreallocation(B, NULL);
39: MatSetRandom(B, r);
40: PetscRandomDestroy(&r);
42: /* Test MatMatMult() */
43: MatMatMult(B, A, MAT_INITIAL_MATRIX, fill, &C);
44: MatMatMult(B, A, MAT_REUSE_MATRIX, fill, &C);
45: MatMatMultEqual(B, A, C, 10, &equal);
48: MatDestroy(&C);
49: MatDestroy(&B);
50: MatDestroy(&A);
51: PetscFinalize();
52: return 0;
53: }
55: /*TEST
57: test:
58: output_file: output/ex122.out
60: TEST*/