1 2 static char help[] ="Tests MatCreateMPIAIJWithArrays() abd MatUpdateMPIAIJWithArrays()\n"; 3 4 #include <petscmat.h> 5 6 /* 7 * This is an extremely simple example to test MatUpdateMPIAIJWithArrays() 8 * 9 * A = 10 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 * */ 19 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; 29 30 PetscFunctionBeginUser; 31 PetscCall(PetscInitialize(&argc,&argv,NULL,help)); 32 comm = PETSC_COMM_WORLD; 33 PetscCallMPI(MPI_Comm_rank(comm,&rank)); 34 PetscCallMPI(MPI_Comm_size(comm,&size)); 35 PetscCheck(size == 3,comm,PETSC_ERR_WRONG_MPI_SIZE,"You have to use three MPI processes to run this example "); 36 PetscCall(MatCreateMPIAIJWithArrays(comm,2,2,PETSC_DETERMINE,PETSC_DETERMINE,i[rank],j[rank],a[rank],&A)); 37 PetscCall(MatView(A,NULL)); 38 PetscCall(MatUpdateMPIAIJWithArrays(A,2,2,PETSC_DETERMINE,PETSC_DETERMINE,i[rank],j[rank],anew[rank])); 39 PetscCall(MatView(A,NULL)); 40 PetscCall(MatUpdateMPIAIJWithArrays(A,2,2,PETSC_DETERMINE,PETSC_DETERMINE,i[rank],j[rank],a[rank])); 41 PetscCall(MatView(A,NULL)); 42 PetscCall(MatUpdateMPIAIJWithArray(A,anew[rank])); 43 PetscCall(MatView(A,NULL)); 44 PetscCall(MatDestroy(&A)); 45 PetscCall(PetscFinalize()); 46 return 0; 47 } 48 49 /*TEST 50 test: 51 nsize: 3 52 53 TEST*/ 54