Actual source code: ex176.c
2: static char help[] = "Tests MatCreateMPIAIJWithArrays() abd MatUpdateMPIAIJWithArray()\n";
4: #include <petscmat.h>
6: /*
7: * This is an extremely simple example to test MatUpdateMPIAIJWithArrays()
8: *
9: * A =
11: 1 2 0 3 0 0
12: 0 4 5 0 0 6
13: 7 0 8 0 9 0
14: 0 10 11 12 0 13
15: 0 14 15 0 0 16
16: 17 0 0 0 0 18
17: *
18: * */
20: int main(int argc, char **argv)
21: {
22: Mat A, B;
23: PetscInt i[3][3] = {
24: {0, 3, 6},
25: {0, 3, 7},
26: {0, 3, 5}
27: };
28: PetscInt j[3][7] = {
29: {0, 1, 3, 1, 2, 5, -1},
30: {0, 2, 4, 1, 2, 3, 5 },
31: {1, 2, 5, 0, 5, -1, -1}
32: };
33: PetscScalar a[3][7] = {
34: {1, 2, 3, 4, 5, 6, -1},
35: {7, 8, 9, 10, 11, 12, 13},
36: {14, 15, 16, 17, 18, -1, -1}
37: };
38: PetscScalar anew[3][7] = {
39: {10, 20, 30, 40, 50, 60, -1 },
40: {70, 80, 90, 100, 110, 120, 130},
41: {140, 150, 160, 170, 180, -1, -1 }
42: };
43: MPI_Comm comm;
44: PetscMPIInt rank;
45: PetscBool equal = PETSC_FALSE;
48: PetscInitialize(&argc, &argv, NULL, help);
49: comm = PETSC_COMM_WORLD;
50: MPI_Comm_rank(comm, &rank);
51: MatCreateMPIAIJWithArrays(comm, 2, PETSC_DETERMINE, PETSC_DETERMINE, 6, i[rank], j[rank], a[rank], &B);
53: MatCreateMPIAIJWithArrays(comm, 2, PETSC_DETERMINE, PETSC_DETERMINE, 6, i[rank], j[rank], a[rank], &A);
54: MatSetFromOptions(A); /* might change A's type */
56: MatEqual(A, B, &equal);
59: MatUpdateMPIAIJWithArray(A, anew[rank]);
60: MatUpdateMPIAIJWithArray(B, anew[rank]);
61: MatEqual(A, B, &equal);
64: MatUpdateMPIAIJWithArray(A, a[rank]);
65: MatUpdateMPIAIJWithArray(B, a[rank]);
66: MatEqual(A, B, &equal);
69: MatDestroy(&A);
70: MatDestroy(&B);
71: PetscFinalize();
72: return 0;
73: }
75: /*TEST
76: testset:
77: nsize: {{1 3}}
78: output_file: output/empty.out
80: test:
81: suffix: aij
83: test:
84: requires: cuda
85: suffix: cuda
86: # since the matrices are created with MatCreateMPIxxx(), users are allowed to pass 'mpiaijcusparse' even with one rank
87: args: -mat_type {{aijcusparse mpiaijcusparse}}
89: test:
90: requires: kokkos_kernels
91: suffix: kok
92: args: -mat_type aijkokkos
93: TEST*/