1 static char help[] = "Testing PtAP for SeqMAIJ matrix, P, with SeqAIJ matrix, A.\n\n"; 2 3 #include <petscmat.h> 4 5 int main(int argc,char **argv) 6 { 7 Mat pA,P,aijP; 8 PetscScalar pa[]={1.,-1.,0.,0.,1.,-1.,0.,0.,1.}; 9 PetscInt i,pij[]={0,1,2}; 10 PetscInt aij[3][3]={{0,1,2},{3,4,5},{6,7,8}}; 11 Mat A,mC,C; 12 PetscScalar one=1.; 13 PetscErrorCode ierr; 14 PetscMPIInt size; 15 PetscBool flg; 16 17 ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 18 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 19 if (size != 1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"This is a uniprocessor example only!"); 20 21 /* Create MAIJ matrix, P */ 22 ierr = MatCreate(PETSC_COMM_SELF,&pA);CHKERRQ(ierr); 23 ierr = MatSetSizes(pA,3,3,3,3);CHKERRQ(ierr); 24 ierr = MatSetType(pA,MATSEQAIJ);CHKERRQ(ierr); 25 ierr = MatSetUp(pA);CHKERRQ(ierr); 26 ierr = MatSetOption(pA,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE);CHKERRQ(ierr); 27 ierr = MatSetValues(pA,3,pij,3,pij,pa,ADD_VALUES);CHKERRQ(ierr); 28 ierr = MatAssemblyBegin(pA,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 29 ierr = MatAssemblyEnd(pA,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 30 ierr = MatCreateMAIJ(pA,3,&P);CHKERRQ(ierr); 31 ierr = MatDestroy(&pA);CHKERRQ(ierr); 32 33 /* Create AIJ equivalent matrix, aijP, for comparison testing */ 34 ierr = MatConvert(P,MATSEQAIJ,MAT_INITIAL_MATRIX,&aijP);CHKERRQ(ierr); 35 36 /* Create AIJ matrix A */ 37 ierr = MatCreate(PETSC_COMM_SELF,&A);CHKERRQ(ierr); 38 ierr = MatSetSizes(A,9,9,9,9);CHKERRQ(ierr); 39 ierr = MatSetType(A,MATSEQAIJ);CHKERRQ(ierr); 40 ierr = MatSetUp(A);CHKERRQ(ierr); 41 ierr = MatSetOption(A,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE);CHKERRQ(ierr); 42 ierr = MatSetValues(A,3,aij[0],3,aij[0],pa,ADD_VALUES);CHKERRQ(ierr); 43 ierr = MatSetValues(A,3,aij[1],3,aij[1],pa,ADD_VALUES);CHKERRQ(ierr); 44 ierr = MatSetValues(A,3,aij[2],3,aij[2],pa,ADD_VALUES);CHKERRQ(ierr); 45 for (i=0; i<9; i++) { 46 ierr = MatSetValue(A,i,i,one,ADD_VALUES);CHKERRQ(ierr); 47 } 48 ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 49 ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 50 51 /* Perform PtAP_SeqAIJ_SeqAIJ for comparison testing */ 52 ierr = MatPtAP(A,aijP,MAT_INITIAL_MATRIX,1.,&C);CHKERRQ(ierr); 53 54 /* Perform PtAP_SeqAIJ_SeqMAIJ */ 55 /* Developer API */ 56 ierr = MatProductCreate(A,P,NULL,&mC);CHKERRQ(ierr); 57 ierr = MatProductSetType(mC,MATPRODUCT_PtAP);CHKERRQ(ierr); 58 ierr = MatProductSetAlgorithm(mC,"default");CHKERRQ(ierr); 59 ierr = MatProductSetFill(mC,PETSC_DEFAULT);CHKERRQ(ierr); 60 ierr = MatProductSetFromOptions(mC);CHKERRQ(ierr); 61 ierr = MatProductSymbolic(mC);CHKERRQ(ierr); 62 ierr = MatProductNumeric(mC);CHKERRQ(ierr); 63 ierr = MatProductNumeric(mC);CHKERRQ(ierr); 64 65 /* Check mC = C */ 66 ierr = MatEqual(C,mC,&flg);CHKERRQ(ierr); 67 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"MatProduct C != mC"); 68 ierr = MatDestroy(&mC);CHKERRQ(ierr); 69 70 /* User API */ 71 ierr = MatPtAP(A,P,MAT_INITIAL_MATRIX,1.,&mC);CHKERRQ(ierr); 72 ierr = MatPtAP(A,P,MAT_REUSE_MATRIX,1.,&mC);CHKERRQ(ierr); 73 74 /* Check mC = C */ 75 ierr = MatEqual(C,mC,&flg);CHKERRQ(ierr); 76 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"MatPtAP C != mC"); 77 ierr = MatDestroy(&mC);CHKERRQ(ierr); 78 79 /* Cleanup */ 80 ierr = MatDestroy(&P);CHKERRQ(ierr); 81 ierr = MatDestroy(&aijP);CHKERRQ(ierr); 82 ierr = MatDestroy(&A);CHKERRQ(ierr); 83 ierr = MatDestroy(&C);CHKERRQ(ierr); 84 ierr = PetscFinalize(); 85 return ierr; 86 } 87 88 /*TEST 89 90 test: 91 output_file: output/ex101.out 92 93 TEST*/ 94