Actual source code: ex34.c
1: static char help[] = "Test MatMatMult() and MatTransposeMatMult() for MPIAIJ and MPIDENSE matrices. \n\
2: Sequential part of mpidense matrix allows changes made by MatDenseSetLDA(). \n\n";
4: #include <petsc.h>
6: int main(int argc, char **argv)
7: {
8: Mat A, B, C, C1;
9: PetscMPIInt size;
10: PetscInt i, ia[2] = {0, 2}, ja[2] = {0, 1}, lda = 4;
11: PetscScalar a[2] = {1.0, 1.0}, *data;
12: PetscBool flg;
14: PetscFunctionBeginUser;
15: PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
16: PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
17: PetscCheck(size == 2, PETSC_COMM_WORLD, PETSC_ERR_WRONG_MPI_SIZE, "Must use 2 processors");
19: PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
20: PetscCall(MatSetType(A, MATMPIAIJ));
21: PetscCall(MatSetSizes(A, 1, 1, 2, 2));
22: PetscCall(MatMPIAIJSetPreallocationCSR(A, ia, ja, a));
24: PetscCall(PetscCalloc1(4 * lda, &data));
25: for (i = 0; i < 4; ++i) data[lda * i] = i * 1.0;
27: PetscCall(MatCreateDense(PETSC_COMM_WORLD, 1, PETSC_DECIDE, 2, 4, data, &B));
28: PetscCall(MatSetOptionsPrefix(B, "b_"));
29: PetscCall(MatSetFromOptions(B));
30: PetscCall(MatDenseSetLDA(B, lda));
32: /* Test MatMatMult() */
33: PetscCall(MatMatMult(A, B, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &C));
34: PetscCall(MatMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C));
36: PetscCall(MatMatMultEqual(A, B, C, 10, &flg));
37: PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_NOTSAMETYPE, "Error in MatMatMult() for C");
39: /* Test user-provided mpidense matrix product */
40: PetscCall(MatDuplicate(C, MAT_COPY_VALUES, &C1));
41: PetscCall(MatMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C1));
42: PetscCall(MatMatMultEqual(A, B, C1, 10, &flg));
43: PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_NOTSAMETYPE, "Error in MatMatMult() for C1");
45: PetscCall(MatDestroy(&C1));
46: PetscCall(MatDestroy(&C));
48: /* Test MatTransposeMatMult() */
49: PetscCall(MatTransposeMatMult(A, B, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &C));
50: PetscCall(MatTransposeMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C));
52: PetscCall(MatTransposeMatMultEqual(A, B, C, 10, &flg));
53: PetscCheck(flg, PETSC_COMM_SELF, PETSC_ERR_ARG_NOTSAMETYPE, "Error in MatTransposeMatMult()");
54: PetscCall(MatDestroy(&C));
56: PetscCall(MatDestroy(&B));
57: PetscCall(MatDestroy(&A));
58: PetscCall(PetscFree(data));
60: PetscCall(PetscFinalize());
61: return 0;
62: }
64: /*TEST
66: test:
67: suffix: 1
68: nsize: 2
69: output_file: output/ex34.out
71: test:
72: suffix: 1_cuda
73: requires: cuda
74: nsize: 2
75: args: -b_mat_type mpidensecuda
76: output_file: output/ex34.out
77: TEST*/