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 CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 19 PetscCheckFalse(size != 1,PETSC_COMM_WORLD,PETSC_ERR_SUP,"This is a uniprocessor example only!"); 20 21 /* Create MAIJ matrix, P */ 22 CHKERRQ(MatCreate(PETSC_COMM_SELF,&pA)); 23 CHKERRQ(MatSetSizes(pA,3,3,3,3)); 24 CHKERRQ(MatSetType(pA,MATSEQAIJ)); 25 CHKERRQ(MatSetUp(pA)); 26 CHKERRQ(MatSetOption(pA,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE)); 27 CHKERRQ(MatSetValues(pA,3,pij,3,pij,pa,ADD_VALUES)); 28 CHKERRQ(MatAssemblyBegin(pA,MAT_FINAL_ASSEMBLY)); 29 CHKERRQ(MatAssemblyEnd(pA,MAT_FINAL_ASSEMBLY)); 30 CHKERRQ(MatCreateMAIJ(pA,3,&P)); 31 CHKERRQ(MatDestroy(&pA)); 32 33 /* Create AIJ equivalent matrix, aijP, for comparison testing */ 34 CHKERRQ(MatConvert(P,MATSEQAIJ,MAT_INITIAL_MATRIX,&aijP)); 35 36 /* Create AIJ matrix A */ 37 CHKERRQ(MatCreate(PETSC_COMM_SELF,&A)); 38 CHKERRQ(MatSetSizes(A,9,9,9,9)); 39 CHKERRQ(MatSetType(A,MATSEQAIJ)); 40 CHKERRQ(MatSetUp(A)); 41 CHKERRQ(MatSetOption(A,MAT_IGNORE_ZERO_ENTRIES,PETSC_TRUE)); 42 CHKERRQ(MatSetValues(A,3,aij[0],3,aij[0],pa,ADD_VALUES)); 43 CHKERRQ(MatSetValues(A,3,aij[1],3,aij[1],pa,ADD_VALUES)); 44 CHKERRQ(MatSetValues(A,3,aij[2],3,aij[2],pa,ADD_VALUES)); 45 for (i=0; i<9; i++) { 46 CHKERRQ(MatSetValue(A,i,i,one,ADD_VALUES)); 47 } 48 CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 49 CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 50 51 /* Perform PtAP_SeqAIJ_SeqAIJ for comparison testing */ 52 CHKERRQ(MatPtAP(A,aijP,MAT_INITIAL_MATRIX,1.,&C)); 53 54 /* Perform PtAP_SeqAIJ_SeqMAIJ */ 55 /* Developer API */ 56 CHKERRQ(MatProductCreate(A,P,NULL,&mC)); 57 CHKERRQ(MatProductSetType(mC,MATPRODUCT_PtAP)); 58 CHKERRQ(MatProductSetAlgorithm(mC,"default")); 59 CHKERRQ(MatProductSetFill(mC,PETSC_DEFAULT)); 60 CHKERRQ(MatProductSetFromOptions(mC)); 61 CHKERRQ(MatProductSymbolic(mC)); 62 CHKERRQ(MatProductNumeric(mC)); 63 CHKERRQ(MatProductNumeric(mC)); 64 65 /* Check mC = C */ 66 CHKERRQ(MatEqual(C,mC,&flg)); 67 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"MatProduct C != mC"); 68 CHKERRQ(MatDestroy(&mC)); 69 70 /* User API */ 71 CHKERRQ(MatPtAP(A,P,MAT_INITIAL_MATRIX,1.,&mC)); 72 CHKERRQ(MatPtAP(A,P,MAT_REUSE_MATRIX,1.,&mC)); 73 74 /* Check mC = C */ 75 CHKERRQ(MatEqual(C,mC,&flg)); 76 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"MatPtAP C != mC"); 77 CHKERRQ(MatDestroy(&mC)); 78 79 /* Cleanup */ 80 CHKERRQ(MatDestroy(&P)); 81 CHKERRQ(MatDestroy(&aijP)); 82 CHKERRQ(MatDestroy(&A)); 83 CHKERRQ(MatDestroy(&C)); 84 ierr = PetscFinalize(); 85 return ierr; 86 } 87 88 /*TEST 89 90 test: 91 output_file: output/ex101.out 92 93 TEST*/ 94