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: PetscInitialize(&argc, &argv, (char*)0,help);
15: MPI_Comm_size(PETSC_COMM_WORLD, &size);
18: MatCreate(PETSC_COMM_WORLD, &A);
19: MatSetType(A, MATMPIAIJ);
20: MatSetSizes(A, 1, 1, 2, 2);
21: MatMPIAIJSetPreallocationCSR(A, ia, ja, a);
23: PetscCalloc1(4 * lda,&data);
24: for (i = 0; i < 4; ++i) data[lda * i] = i * 1.0;
26: MatCreateDense(PETSC_COMM_WORLD, 1, PETSC_DECIDE, 2, 4, data, &B);
27: MatSetOptionsPrefix(B,"b_");
28: MatSetFromOptions(B);
29: MatDenseSetLDA(B, lda);
31: /* Test MatMatMult() */
32: MatMatMult(A, B, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &C);
33: MatMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C);
35: MatMatMultEqual(A,B,C,10,&flg);
38: /* Test user-provided mpidense matrix product */
39: MatDuplicate(C,MAT_COPY_VALUES,&C1);
40: MatMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C1);
41: MatMatMultEqual(A,B,C1,10,&flg);
44: MatDestroy(&C1);
45: MatDestroy(&C);
47: /* Test MatTransposeMatMult() */
48: MatTransposeMatMult(A, B, MAT_INITIAL_MATRIX, PETSC_DEFAULT, &C);
49: MatTransposeMatMult(A, B, MAT_REUSE_MATRIX, PETSC_DEFAULT, &C);
51: MatTransposeMatMultEqual(A,B,C,10,&flg);
53: MatDestroy(&C);
55: MatDestroy(&B);
56: MatDestroy(&A);
57: PetscFree(data);
59: PetscFinalize();
60: return 0;
61: }
63: /*TEST
65: test:
66: suffix: 1
67: nsize: 2
68: output_file: output/ex34.out
70: test:
71: suffix: 1_cuda
72: requires: cuda
73: nsize: 2
74: args: -b_mat_type mpidensecuda
75: output_file: output/ex34.out
76: TEST*/