1 2 static char help[] = "Tests MatGetRowMax(), MatGetRowMin(), MatGetRowMaxAbs()\n"; 3 4 #include <petscmat.h> 5 6 #define M 5 7 #define N 6 8 9 int main(int argc,char **args) 10 { 11 Mat A; 12 Vec min,max,maxabs,e; 13 PetscInt m,n,j,col,cols[N]; 14 PetscInt imin[M],imax[M],imaxabs[M],indices[N],row,testcase=0; 15 PetscScalar values[N]; 16 PetscErrorCode ierr; 17 MatType type; 18 PetscMPIInt size,rank; 19 PetscBool doTest=PETSC_TRUE; 20 PetscReal enorm; 21 22 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 23 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 24 ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRQ(ierr); 25 ierr = PetscOptionsGetInt(NULL,NULL,"-testcase",&testcase,NULL);CHKERRQ(ierr); 26 27 ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 28 if (testcase == 1) { /* proc[0] holds entire A and other processes have no entry */ 29 if (!rank) { 30 ierr = MatSetSizes(A,M,N,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 31 } else { 32 ierr = MatSetSizes(A,0,0,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 33 } 34 testcase = 0; 35 } else { 36 ierr = MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); 37 } 38 ierr = MatSetFromOptions(A);CHKERRQ(ierr); 39 ierr = MatSetUp(A);CHKERRQ(ierr); 40 41 if (!rank) { /* proc[0] sets matrix A */ 42 for (j=0; j<N; j++) indices[j] = j; 43 switch (testcase) { 44 case 1: /* see testcast 0 */ 45 break; 46 case 2: 47 row = 0; 48 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; values[3] = -4.0; values[4] = 0.0; values[5] = 1.0; 49 ierr = MatSetValues(A,1,&row,N,indices,values,INSERT_VALUES);CHKERRQ(ierr); 50 row = 2; 51 indices[0] = 0; indices[1] = 3; indices[2] = 5; 52 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; 53 ierr = MatSetValues(A,1,&row,3,indices,values,INSERT_VALUES);CHKERRQ(ierr); 54 row = 3; 55 indices[0] = 0; indices[1] = 1; indices[2] = 4; 56 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; 57 ierr = MatSetValues(A,1,&row,3,indices,values,INSERT_VALUES);CHKERRQ(ierr); 58 row = 4; 59 indices[0] = 0; indices[1] = 1; indices[2] = 2; 60 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; 61 ierr = MatSetValues(A,1,&row,3,indices,values,INSERT_VALUES);CHKERRQ(ierr); 62 break; 63 case 3: 64 row = 0; 65 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; 66 ierr = MatSetValues(A,1,&row,3,indices+1,values,INSERT_VALUES);CHKERRQ(ierr); 67 row = 1; 68 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; 69 ierr = MatSetValues(A,1,&row,3,indices,values,INSERT_VALUES);CHKERRQ(ierr); 70 row = 2; 71 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; 72 ierr = MatSetValues(A,1,&row,3,indices,values,INSERT_VALUES);CHKERRQ(ierr); 73 row = 3; 74 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; values[3] = -1.0; 75 ierr = MatSetValues(A,1,&row,4,indices,values,INSERT_VALUES);CHKERRQ(ierr); 76 row = 4; 77 values[0] = -2.0; values[1] = -2.0; values[2] = -2.0; values[3] = -1.0; 78 ierr = MatSetValues(A,1,&row,4,indices,values,INSERT_VALUES);CHKERRQ(ierr); 79 break; 80 81 default: 82 row = 0; 83 values[0] = -1.0; values[1] = 0.0; values[2] = 1.0; values[3] = 3.0; values[4] = 4.0; values[5] = -5.0; 84 ierr = MatSetValues(A,1,&row,N,indices,values,INSERT_VALUES);CHKERRQ(ierr); 85 row = 1; 86 ierr = MatSetValues(A,1,&row,3,indices,values,INSERT_VALUES);CHKERRQ(ierr); 87 row = 3; 88 ierr = MatSetValues(A,1,&row,1,indices+4,values+4,INSERT_VALUES);CHKERRQ(ierr); 89 row = 4; 90 ierr = MatSetValues(A,1,&row,2,indices+4,values+4,INSERT_VALUES);CHKERRQ(ierr); 91 } 92 } 93 ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 94 ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 95 ierr = MatView(A,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 96 97 ierr = MatGetLocalSize(A, &m,&n);CHKERRQ(ierr); 98 ierr = VecCreate(PETSC_COMM_WORLD,&min);CHKERRQ(ierr); 99 ierr = VecSetSizes(min,m,PETSC_DECIDE);CHKERRQ(ierr); 100 ierr = VecSetFromOptions(min);CHKERRQ(ierr); 101 ierr = VecDuplicate(min,&max);CHKERRQ(ierr); 102 ierr = VecDuplicate(min,&maxabs);CHKERRQ(ierr); 103 ierr = VecDuplicate(min,&e);CHKERRQ(ierr); 104 105 /* Test MatGetRowMax() */ 106 ierr = PetscPrintf(PETSC_COMM_WORLD,"\n Row Maximums\n");CHKERRQ(ierr); 107 ierr = MatGetRowMax(A,max,NULL);CHKERRQ(ierr); 108 ierr = MatGetRowMax(A,max,imax);CHKERRQ(ierr); 109 ierr = VecView(max,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 110 ierr = VecGetLocalSize(max,&n);CHKERRQ(ierr); 111 ierr = PetscIntView(n,imax,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 112 113 /* Test MatGetRowMin() */ 114 ierr = MatScale(A,-1.0);CHKERRQ(ierr); 115 /* ierr = MatView(A,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); */ 116 ierr = PetscPrintf(PETSC_COMM_WORLD,"\n Row Minimums\n");CHKERRQ(ierr); 117 ierr = MatGetRowMin(A,min,NULL);CHKERRQ(ierr); 118 ierr = MatGetRowMin(A,min,imin);CHKERRQ(ierr); 119 120 ierr = VecWAXPY(e,1.0,max,min);CHKERRQ(ierr); /* e = max + min */ 121 ierr = VecNorm(e,NORM_INFINITY,&enorm);CHKERRQ(ierr); 122 if (enorm > PETSC_MACHINE_EPSILON) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_PLIB,"max+min > PETSC_MACHINE_EPSILON "); 123 for (j = 0; j < n; j++) if (imin[j] != imax[j]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"imin != imax"); 124 125 /* Test MatGetRowMaxAbs() */ 126 ierr = PetscPrintf(PETSC_COMM_WORLD,"\n Row Maximum Absolute\n");CHKERRQ(ierr); 127 ierr = MatGetRowMaxAbs(A,maxabs,NULL);CHKERRQ(ierr); 128 ierr = MatGetRowMaxAbs(A,maxabs,imaxabs);CHKERRQ(ierr); 129 ierr = VecView(maxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 130 ierr = PetscIntView(n,imaxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 131 exit(1); 132 133 /* Test MatGetRowMin, MatGetRowMax and MatGetRowMaxAbs */ 134 if (size == 1) { 135 ierr = MatGetRowMin(A,min,imin);CHKERRQ(ierr); 136 ierr = MatGetRowMax(A,max,imax);CHKERRQ(ierr); 137 ierr = MatGetRowMaxAbs(A,maxabs,imaxabs);CHKERRQ(ierr); 138 139 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Minimums\n");CHKERRQ(ierr); 140 ierr = VecView(min,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 141 ierr = PetscIntView(5,imin,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 142 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Maximums\n");CHKERRQ(ierr); 143 ierr = VecView(max,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 144 ierr = PetscIntView(5,imax,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 145 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Maximum Absolute Values\n");CHKERRQ(ierr); 146 ierr = VecView(maxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 147 ierr = PetscIntView(5,imaxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 148 149 } else { 150 ierr = MatGetType(A,&type);CHKERRQ(ierr); 151 ierr = PetscPrintf(PETSC_COMM_WORLD,"\nMatrix type: %s\n",type);CHKERRQ(ierr); 152 /* AIJ */ 153 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIAIJ,&doTest);CHKERRQ(ierr); 154 if (doTest) { 155 ierr = MatGetRowMaxAbs(A,maxabs,NULL);CHKERRQ(ierr); 156 ierr = MatGetRowMaxAbs(A,maxabs,imaxabs);CHKERRQ(ierr); 157 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Maximum Absolute Values:\n");CHKERRQ(ierr); 158 ierr = VecView(maxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 159 } 160 /* BAIJ */ 161 ierr = PetscObjectTypeCompare((PetscObject)A,MATMPIBAIJ,&doTest);CHKERRQ(ierr); 162 if (doTest) { 163 ierr = MatGetRowMaxAbs(A,maxabs,NULL);CHKERRQ(ierr); 164 ierr = MatGetRowMaxAbs(A,maxabs,imaxabs);CHKERRQ(ierr); 165 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Maximum Absolute Values:\n");CHKERRQ(ierr); 166 ierr = VecView(maxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 167 } 168 } 169 170 if (size == 1) { 171 ierr = MatConvert(A,MATDENSE,MAT_INPLACE_MATRIX,&A);CHKERRQ(ierr); 172 173 ierr = MatGetRowMin(A,min,imin);CHKERRQ(ierr); 174 ierr = MatGetRowMax(A,max,imax);CHKERRQ(ierr); 175 ierr = MatGetRowMaxAbs(A,maxabs,imaxabs);CHKERRQ(ierr); 176 177 ierr = MatView(A,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 178 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Minimums\n");CHKERRQ(ierr); 179 ierr = VecView(min,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 180 ierr = PetscIntView(5,imin,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 181 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Maximums\n");CHKERRQ(ierr); 182 ierr = VecView(max,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 183 ierr = PetscIntView(5,imax,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 184 ierr = PetscPrintf(PETSC_COMM_WORLD,"Row Maximum Absolute Values\n");CHKERRQ(ierr); 185 ierr = VecView(maxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 186 ierr = PetscIntView(5,imaxabs,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 187 } 188 189 ierr = VecDestroy(&min);CHKERRQ(ierr); 190 ierr = VecDestroy(&max);CHKERRQ(ierr); 191 ierr = VecDestroy(&maxabs);CHKERRQ(ierr); 192 ierr = VecDestroy(&e);CHKERRQ(ierr); 193 ierr = MatDestroy(&A);CHKERRQ(ierr); 194 ierr = PetscFinalize(); 195 return ierr; 196 } 197 198 199 200 /*TEST 201 202 test: 203 output_file: output/ex114.out 204 205 test: 206 suffix: 2 207 nsize: 2 208 209 test: 210 suffix: 3 211 nsize: 2 212 args: -mat_type baij 213 214 TEST*/ 215