Actual source code: ex52.c
2: static char help[] = "Tests various routines in MatMPIBAIJ format.\n";
4: #include <petscmat.h>
6: int main(int argc,char **args)
7: {
8: Mat A;
9: PetscInt m=2,bs=1,M,row,col,start,end,i,j,k;
11: PetscMPIInt rank,size;
12: PetscScalar data=100;
13: PetscBool flg;
15: PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr;
16: MPI_Comm_rank(PETSC_COMM_WORLD,&rank);
17: MPI_Comm_size(PETSC_COMM_WORLD,&size);
19: /* Test MatSetValues() and MatGetValues() */
20: PetscOptionsGetInt(NULL,NULL,"-mat_block_size",&bs,NULL);
21: PetscOptionsGetInt(NULL,NULL,"-mat_size",&m,NULL);
23: M = m*bs*size;
24: MatCreateBAIJ(PETSC_COMM_WORLD,bs,PETSC_DECIDE,PETSC_DECIDE,M,M,PETSC_DECIDE,NULL,PETSC_DECIDE,NULL,&A);
26: MatGetOwnershipRange(A,&start,&end);
27: PetscOptionsHasName(NULL,NULL,"-column_oriented",&flg);
28: if (flg) {
29: MatSetOption(A,MAT_ROW_ORIENTED,PETSC_FALSE);
30: }
32: /* inproc assembly */
33: for (row=start; row<end; row++) {
34: for (col=start; col<end; col++,data+=1) {
35: MatSetValues(A,1,&row,1,&col,&data,INSERT_VALUES);
36: }
37: }
38: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
39: MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);
41: /* offproc assembly */
42: data = 5.0;
43: row = (M+start-1)%M;
44: for (col=0; col<M; col++) {
45: MatSetValues(A,1,&row,1,&col,&data,ADD_VALUES);
46: }
47: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
48: MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);
50: /* Test MatSetValuesBlocked() */
51: PetscOptionsHasName(NULL,NULL,"-test_setvaluesblocked",&flg);
52: if (flg) {
53: PetscScalar *bval;
54: row /= bs;
55: col = start/bs;
56: PetscMalloc1(bs*bs,&bval);
57: k = 1;
58: /* row oriented - default */
59: for (i=0; i<bs; i++) {
60: for (j=0; j<bs; j++) {
61: bval[i*bs+j] = (PetscScalar)k; k++;
62: }
63: }
64: MatSetValuesBlocked(A,1,&row,1,&col,bval,INSERT_VALUES);
65: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
66: MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);
67: PetscFree(bval);
68: }
70: MatView(A,PETSC_VIEWER_STDOUT_WORLD);
71: MatDestroy(&A);
72: PetscFinalize();
73: return ierr;
74: }
76: /*TEST
78: test:
79: suffix: 1
80: nsize: 3
81: args: -mat_block_size 2 -test_setvaluesblocked
83: test:
84: suffix: 2
85: nsize: 3
86: args: -mat_block_size 2 -test_setvaluesblocked -column_oriented
88: test:
89: suffix: 3
90: nsize: 3
91: args: -mat_block_size 1 -test_setvaluesblocked
93: test:
94: suffix: 4
95: nsize: 3
96: args: -mat_block_size 1 -test_setvaluesblocked -column_oriented
98: TEST*/