1*c4762a1bSJed Brown static char help[]= "Test MatSetRandom on MATMPIAIJ matrices\n\n"; 2*c4762a1bSJed Brown 3*c4762a1bSJed Brown /* 4*c4762a1bSJed Brown Adapted from an example Contributed-by: Jakub Kruzik <jakub.kruzik@vsb.cz> 5*c4762a1bSJed Brown */ 6*c4762a1bSJed Brown #include <petscmat.h> 7*c4762a1bSJed Brown int main(int argc,char **args) 8*c4762a1bSJed Brown { 9*c4762a1bSJed Brown Mat A[2]; 10*c4762a1bSJed Brown PetscErrorCode ierr; 11*c4762a1bSJed Brown PetscReal nrm,tol=10*PETSC_SMALL; 12*c4762a1bSJed Brown PetscRandom rctx; 13*c4762a1bSJed Brown 14*c4762a1bSJed Brown ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 15*c4762a1bSJed Brown ierr = PetscRandomCreate(PETSC_COMM_WORLD,&rctx);CHKERRQ(ierr); 16*c4762a1bSJed Brown 17*c4762a1bSJed Brown /* Call MatSetRandom on unassembled matrices */ 18*c4762a1bSJed Brown ierr = MatCreateAIJ(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,20,20,3,NULL,3,NULL,&A[0]);CHKERRQ(ierr); 19*c4762a1bSJed Brown ierr = MatCreateAIJ(PETSC_COMM_WORLD,PETSC_DECIDE,PETSC_DECIDE,20,20,3,NULL,3,NULL,&A[1]);CHKERRQ(ierr); 20*c4762a1bSJed Brown ierr = MatSetRandom(A[0],rctx);CHKERRQ(ierr); 21*c4762a1bSJed Brown ierr = MatSetRandom(A[1],rctx);CHKERRQ(ierr); 22*c4762a1bSJed Brown 23*c4762a1bSJed Brown ierr = MatAXPY(A[0],1.0,A[1],DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 24*c4762a1bSJed Brown ierr = MatAXPY(A[0],-1.0,A[0],SAME_NONZERO_PATTERN);CHKERRQ(ierr); 25*c4762a1bSJed Brown ierr = MatNorm(A[0],NORM_1,&nrm);CHKERRQ(ierr); 26*c4762a1bSJed Brown if (nrm > tol) {ierr = PetscPrintf(PETSC_COMM_WORLD,"Error: MatNorm(), norm1=: %g\n",(double)nrm);CHKERRQ(ierr);} 27*c4762a1bSJed Brown 28*c4762a1bSJed Brown /* Call MatSetRandom on assembled matrices */ 29*c4762a1bSJed Brown ierr = MatSetRandom(A[0],rctx);CHKERRQ(ierr); 30*c4762a1bSJed Brown ierr = MatSetRandom(A[1],rctx);CHKERRQ(ierr); 31*c4762a1bSJed Brown 32*c4762a1bSJed Brown ierr = MatAXPY(A[0],1.0,A[1],DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 33*c4762a1bSJed Brown ierr = MatAXPY(A[0],-1.0,A[0],SAME_NONZERO_PATTERN);CHKERRQ(ierr); 34*c4762a1bSJed Brown ierr = MatNorm(A[0],NORM_1,&nrm);CHKERRQ(ierr); 35*c4762a1bSJed Brown if (nrm > tol) {ierr = PetscPrintf(PETSC_COMM_WORLD,"Error: MatNorm(), norm1=: %g\n",(double)nrm);CHKERRQ(ierr);} 36*c4762a1bSJed Brown 37*c4762a1bSJed Brown ierr = MatDestroy(&A[0]);CHKERRQ(ierr); 38*c4762a1bSJed Brown ierr = MatDestroy(&A[1]);CHKERRQ(ierr); 39*c4762a1bSJed Brown ierr = PetscRandomDestroy(&rctx);CHKERRQ(ierr); 40*c4762a1bSJed Brown ierr = PetscFinalize(); 41*c4762a1bSJed Brown return ierr; 42*c4762a1bSJed Brown } 43*c4762a1bSJed Brown 44*c4762a1bSJed Brown /*TEST 45*c4762a1bSJed Brown test: 46*c4762a1bSJed Brown nsize: 3 47*c4762a1bSJed Brown TEST*/ 48