1 static char help[] = "Tests the use of MatTranspose_Nest and MatMatMult_Nest_Dense\n"; 2 3 #include <petscmat.h> 4 5 PetscErrorCode TestInitialMatrix(void) 6 { 7 const PetscInt nr = 2,nc = 3,nk = 10; 8 PetscInt n,N,m,M; 9 const PetscInt arow[2*3] = { 2,2,2,3,3,3 }; 10 const PetscInt acol[2*3] = { 3,2,4,3,2,4 }; 11 Mat A,Atranspose,B,C; 12 Mat subs[2*3],**block; 13 Vec x,y,Ax,ATy; 14 PetscInt i,j; 15 PetscScalar dot1,dot2,zero = 0.0,one = 1.0; 16 PetscReal norm; 17 PetscScalar *valsB,*valsC; 18 PetscRandom rctx; 19 PetscErrorCode ierr; 20 21 PetscFunctionBegin; 22 ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rctx);CHKERRQ(ierr); 23 /* Force the random numbers to have imaginary part 0 so printed results are the same for --with-scalar-type=real or --with-scalar-type=complex */ 24 ierr = PetscRandomSetInterval(rctx,zero,one);CHKERRQ(ierr); 25 ierr = PetscRandomSetFromOptions(rctx);CHKERRQ(ierr); 26 for (i=0; i<(nr * nc); i++) { 27 ierr = MatCreateSeqDense(PETSC_COMM_WORLD,arow[i],acol[i],NULL,&subs[i]);CHKERRQ(ierr); 28 } 29 ierr = MatCreateNest(PETSC_COMM_WORLD,nr,NULL,nc,NULL,subs,&A);CHKERRQ(ierr); 30 ierr = MatCreateVecs(A, &x, NULL);CHKERRQ(ierr); 31 ierr = MatCreateVecs(A, NULL, &y);CHKERRQ(ierr); 32 ierr = VecDuplicate(x, &ATy);CHKERRQ(ierr); 33 ierr = VecDuplicate(y, &Ax);CHKERRQ(ierr); 34 ierr = MatSetRandom(A,rctx);CHKERRQ(ierr); 35 ierr = MatTranspose(A, MAT_INITIAL_MATRIX, &Atranspose);CHKERRQ(ierr); 36 37 ierr = MatView(A, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 38 ierr = MatNestGetSubMats(A, NULL, NULL, &block);CHKERRQ(ierr); 39 for (i=0; i<nr; i++) { 40 for (j=0; j<nc; j++) { 41 ierr = MatView(block[i][j], PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 42 } 43 } 44 45 ierr = MatView(Atranspose, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 46 ierr = MatNestGetSubMats(Atranspose, NULL, NULL, &block);CHKERRQ(ierr); 47 for (i=0; i<nc; i++) { 48 for (j=0; j<nr; j++) { 49 ierr = MatView(block[i][j], PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 50 } 51 } 52 53 /* Check <Ax, y> = <x, A^Ty> */ 54 for (i=0; i<10; i++) { 55 ierr = VecSetRandom(x,rctx);CHKERRQ(ierr); 56 ierr = VecSetRandom(y,rctx);CHKERRQ(ierr); 57 58 ierr = MatMult(A, x, Ax);CHKERRQ(ierr); 59 ierr = VecDot(Ax, y, &dot1);CHKERRQ(ierr); 60 ierr = MatMult(Atranspose, y, ATy);CHKERRQ(ierr); 61 ierr = VecDot(ATy, x, &dot2);CHKERRQ(ierr); 62 63 ierr = PetscPrintf(PETSC_COMM_WORLD, "<Ax, y> = %g\n", (double)PetscRealPart(dot1));CHKERRQ(ierr); 64 ierr = PetscPrintf(PETSC_COMM_WORLD, "<x, A^Ty> = %g\n",(double)PetscRealPart(dot2));CHKERRQ(ierr); 65 } 66 ierr = VecDestroy(&x);CHKERRQ(ierr); 67 ierr = VecDestroy(&y);CHKERRQ(ierr); 68 ierr = VecDestroy(&Ax);CHKERRQ(ierr); 69 70 ierr = MatCreateSeqDense(PETSC_COMM_WORLD,acol[0]+acol[nr]+acol[2*nr],nk,NULL,&B);CHKERRQ(ierr); 71 ierr = MatSetRandom(B,rctx);CHKERRQ(ierr); 72 ierr = MatMatMult(A,B,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&C);CHKERRQ(ierr); 73 ierr = MatMatMult(A,B,MAT_REUSE_MATRIX,PETSC_DEFAULT,&C);CHKERRQ(ierr); 74 ierr = MatGetSize(A,&M,&N);CHKERRQ(ierr); 75 ierr = MatGetLocalSize(A,&m,&n);CHKERRQ(ierr); 76 for (i=0; i<nk; i++) { 77 ierr = MatDenseGetColumn(B,i,&valsB);CHKERRQ(ierr); 78 ierr = VecCreateMPIWithArray(PETSC_COMM_WORLD,1,n,N,valsB,&x);CHKERRQ(ierr); 79 ierr = MatCreateVecs(A,NULL,&Ax);CHKERRQ(ierr); 80 ierr = MatMult(A,x,Ax);CHKERRQ(ierr); 81 ierr = VecDestroy(&x);CHKERRQ(ierr); 82 ierr = MatDenseGetColumn(C,i,&valsC);CHKERRQ(ierr); 83 ierr = VecCreateMPIWithArray(PETSC_COMM_WORLD,1,m,M,valsC,&y);CHKERRQ(ierr); 84 ierr = VecAXPY(y,-1.0,Ax);CHKERRQ(ierr); 85 ierr = VecDestroy(&Ax);CHKERRQ(ierr); 86 ierr = VecDestroy(&y);CHKERRQ(ierr); 87 ierr = MatDenseRestoreColumn(C,&valsC);CHKERRQ(ierr); 88 ierr = MatDenseRestoreColumn(B,&valsB);CHKERRQ(ierr); 89 } 90 ierr = MatNorm(C,NORM_INFINITY,&norm);CHKERRQ(ierr); 91 if (norm > PETSC_SMALL) SETERRQ1(PETSC_COMM_WORLD,PETSC_ERR_PLIB,"Error in MatMatMult(): %g\n",(double)norm); 92 ierr = MatDestroy(&C);CHKERRQ(ierr); 93 ierr = MatDestroy(&B);CHKERRQ(ierr); 94 95 for (i=0; i<(nr * nc); i++) { 96 ierr = MatDestroy(&subs[i]);CHKERRQ(ierr); 97 } 98 ierr = MatDestroy(&A);CHKERRQ(ierr); 99 ierr = MatDestroy(&Atranspose);CHKERRQ(ierr); 100 ierr = VecDestroy(&ATy);CHKERRQ(ierr); 101 ierr = PetscRandomDestroy(&rctx);CHKERRQ(ierr); 102 PetscFunctionReturn(0); 103 } 104 105 PetscErrorCode TestReuseMatrix(void) 106 { 107 const PetscInt n = 2; 108 Mat A; 109 Mat subs[2*2],**block; 110 PetscInt i,j; 111 PetscRandom rctx; 112 PetscErrorCode ierr; 113 PetscScalar zero = 0.0, one = 1.0; 114 115 PetscFunctionBegin; 116 ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rctx);CHKERRQ(ierr); 117 ierr = PetscRandomSetInterval(rctx,zero,one);CHKERRQ(ierr); 118 ierr = PetscRandomSetFromOptions(rctx);CHKERRQ(ierr); 119 for (i=0; i<(n * n); i++) { 120 ierr = MatCreateSeqDense(PETSC_COMM_WORLD,n,n,NULL,&subs[i]);CHKERRQ(ierr); 121 } 122 ierr = MatCreateNest(PETSC_COMM_WORLD,n,NULL,n,NULL,subs,&A);CHKERRQ(ierr); 123 ierr = MatSetRandom(A,rctx);CHKERRQ(ierr); 124 125 ierr = MatView(A, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 126 ierr = MatNestGetSubMats(A, NULL, NULL, &block);CHKERRQ(ierr); 127 for (i=0; i<n; i++) { 128 for (j=0; j<n; j++) { 129 ierr = MatView(block[i][j], PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 130 } 131 } 132 ierr = MatTranspose(A,MAT_INPLACE_MATRIX,&A);CHKERRQ(ierr); 133 ierr = MatView(A, PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 134 ierr = MatNestGetSubMats(A, NULL, NULL, &block);CHKERRQ(ierr); 135 for (i=0; i<n; i++) { 136 for (j=0; j<n; j++) { 137 ierr = MatView(block[i][j], PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 138 } 139 } 140 141 for (i=0; i<(n * n); i++) { 142 ierr = MatDestroy(&subs[i]);CHKERRQ(ierr); 143 } 144 ierr = MatDestroy(&A);CHKERRQ(ierr); 145 ierr = PetscRandomDestroy(&rctx);CHKERRQ(ierr); 146 PetscFunctionReturn(0); 147 } 148 149 int main(int argc,char **args) 150 { 151 PetscErrorCode ierr; 152 153 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 154 ierr = TestInitialMatrix();CHKERRQ(ierr); 155 ierr = TestReuseMatrix();CHKERRQ(ierr); 156 ierr = PetscFinalize(); 157 return ierr; 158 } 159 160 /*TEST 161 162 test: 163 args: -malloc_dump 164 165 TEST*/ 166