Actual source code: ex176.c
2: static char help[] ="Tests MatCreateMPIAIJWithArrays() abd MatUpdateMPIAIJWithArrays()\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;
23: PetscInt i[3][3] = {{0, 3, 6},{0, 3, 7},{0, 3, 5}};
24: PetscInt j[3][7] = {{0, 1, 3, 1, 2, 5, -1},{0, 2, 4, 1, 2, 3, 5},{1, 2, 5, 0, 5, -1, -1}};
25: PetscScalar a[3][7] = {{1, 2, 3, 4, 5, 6, -1}, {7, 8, 9, 10, 11, 12, 13},{14, 15, 16, 17, 18, -1, -1}};
26: PetscScalar anew[3][7] = {{10, 20, 30, 40, 50, 60, -1}, {70, 80, 90, 100, 110, 120, 130},{140, 150, 160, 170, 180, -1, -1}};
27: MPI_Comm comm;
28: PetscMPIInt rank,size;
30: PetscInitialize(&argc,&argv,NULL,help);
31: comm = PETSC_COMM_WORLD;
32: MPI_Comm_rank(comm,&rank);
33: MPI_Comm_size(comm,&size);
35: MatCreateMPIAIJWithArrays(comm,2,2,PETSC_DETERMINE,PETSC_DETERMINE,i[rank],j[rank],a[rank],&A);
36: MatView(A,NULL);
37: MatUpdateMPIAIJWithArrays(A,2,2,PETSC_DETERMINE,PETSC_DETERMINE,i[rank],j[rank],anew[rank]);
38: MatView(A,NULL);
39: MatUpdateMPIAIJWithArrays(A,2,2,PETSC_DETERMINE,PETSC_DETERMINE,i[rank],j[rank],a[rank]);
40: MatView(A,NULL);
41: MatDestroy(&A);
42: PetscFinalize();
43: return 0;
44: }
46: /*TEST
47: test:
48: nsize: 3
50: TEST*/