1 2 static char help[] = "Tests MatCopy() and MatStore/RetrieveValues().\n\n"; 3 4 #include <petscmat.h> 5 6 int main(int argc,char **args) 7 { 8 Mat C,A; 9 PetscInt i, n = 10,midx[3],bs=1; 10 PetscErrorCode ierr; 11 PetscScalar v[3]; 12 PetscBool flg,isAIJ; 13 MatType type; 14 PetscMPIInt size; 15 16 ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr; 17 CHKERRMPI(MPI_Comm_size(PETSC_COMM_WORLD,&size)); 18 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL)); 19 CHKERRQ(PetscOptionsGetInt(NULL,NULL,"-mat_block_size",&bs,NULL)); 20 21 CHKERRQ(MatCreate(PETSC_COMM_WORLD,&C)); 22 CHKERRQ(MatSetSizes(C,PETSC_DECIDE,PETSC_DECIDE,n,n)); 23 CHKERRQ(MatSetType(C,MATAIJ)); 24 CHKERRQ(MatSetFromOptions(C)); 25 CHKERRQ(PetscObjectSetName((PetscObject)C,"initial")); 26 27 CHKERRQ(MatGetType(C,&type)); 28 if (size == 1) { 29 CHKERRQ(PetscObjectTypeCompare((PetscObject)C,MATSEQAIJ,&isAIJ)); 30 } else { 31 CHKERRQ(PetscObjectTypeCompare((PetscObject)C,MATMPIAIJ,&isAIJ)); 32 } 33 CHKERRQ(MatSeqAIJSetPreallocation(C,3,NULL)); 34 CHKERRQ(MatMPIAIJSetPreallocation(C,3,NULL,3,NULL)); 35 CHKERRQ(MatSeqBAIJSetPreallocation(C,bs,3,NULL)); 36 CHKERRQ(MatMPIBAIJSetPreallocation(C,bs,3,NULL,3,NULL)); 37 CHKERRQ(MatSeqSBAIJSetPreallocation(C,bs,3,NULL)); 38 CHKERRQ(MatMPISBAIJSetPreallocation(C,bs,3,NULL,3,NULL)); 39 40 v[0] = -1.; v[1] = 2.; v[2] = -1.; 41 for (i=1; i<n-1; i++) { 42 midx[2] = i-1; midx[1] = i; midx[0] = i+1; 43 CHKERRQ(MatSetValues(C,1,&i,3,midx,v,INSERT_VALUES)); 44 } 45 i = 0; midx[0] = 0; midx[1] = 1; 46 v[0] = 2.0; v[1] = -1.; 47 CHKERRQ(MatSetValues(C,1,&i,2,midx,v,INSERT_VALUES)); 48 i = n-1; midx[0] = n-2; midx[1] = n-1; 49 v[0] = -1.0; v[1] = 2.; 50 CHKERRQ(MatSetValues(C,1,&i,2,midx,v,INSERT_VALUES)); 51 52 CHKERRQ(MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY)); 53 CHKERRQ(MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY)); 54 CHKERRQ(MatView(C,NULL)); 55 CHKERRQ(MatViewFromOptions(C,NULL,"-view")); 56 57 /* test matduplicate */ 58 CHKERRQ(MatDuplicate(C,MAT_COPY_VALUES,&A)); 59 CHKERRQ(PetscObjectSetName((PetscObject)A,"duplicate_copy")); 60 CHKERRQ(MatViewFromOptions(A,NULL,"-view")); 61 CHKERRQ(MatEqual(A,C,&flg)); 62 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_SUP,"MatDuplicate(C,MAT_COPY_VALUES,): Matrices are NOT equal"); 63 CHKERRQ(MatDestroy(&A)); 64 65 /* test matrices with different nonzero patterns - Note: A is created with different nonzero pattern of C! */ 66 CHKERRQ(MatCreate(PETSC_COMM_WORLD,&A)); 67 CHKERRQ(MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,n,n)); 68 CHKERRQ(MatSetFromOptions(A)); 69 CHKERRQ(MatSetUp(A)); 70 CHKERRQ(MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY)); 71 CHKERRQ(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY)); 72 73 CHKERRQ(MatCopy(C,A,DIFFERENT_NONZERO_PATTERN)); 74 CHKERRQ(PetscObjectSetName((PetscObject)A,"copy_diffnnz")); 75 CHKERRQ(MatViewFromOptions(A,NULL,"-view")); 76 CHKERRQ(MatEqual(A,C,&flg)); 77 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_SUP,"MatCopy(C,A,DIFFERENT_NONZERO_PATTERN): Matrices are NOT equal"); 78 79 /* test matrices with same nonzero pattern */ 80 CHKERRQ(MatDestroy(&A)); 81 CHKERRQ(MatDuplicate(C,MAT_DO_NOT_COPY_VALUES,&A)); 82 CHKERRQ(MatCopy(C,A,SAME_NONZERO_PATTERN)); 83 CHKERRQ(PetscObjectSetName((PetscObject)A,"copy_samennz")); 84 CHKERRQ(MatViewFromOptions(A,NULL,"-view")); 85 CHKERRQ(MatEqual(A,C,&flg)); 86 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_SUP,"MatCopy(C,A,SAME_NONZERO_PATTERN): Matrices are NOT equal"); 87 88 /* test subset nonzero pattern */ 89 CHKERRQ(MatCopy(C,A,SUBSET_NONZERO_PATTERN)); 90 CHKERRQ(PetscObjectSetName((PetscObject)A,"copy_subnnz")); 91 CHKERRQ(MatViewFromOptions(A,NULL,"-view")); 92 CHKERRQ(MatEqual(A,C,&flg)); 93 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_SUP,"MatCopy(C,A,SUBSET_NONZERO_PATTERN): Matrices are NOT equal"); 94 95 /* Test MatCopy on a matrix obtained after MatConvert from AIJ 96 see https://lists.mcs.anl.gov/pipermail/petsc-dev/2019-April/024289.html */ 97 CHKERRQ(MatHasCongruentLayouts(C,&flg)); 98 if (flg) { 99 Mat Cs,Cse; 100 MatType Ctype,Cstype; 101 102 CHKERRQ(MatGetType(C,&Ctype)); 103 CHKERRQ(MatTranspose(C,MAT_INITIAL_MATRIX,&Cs)); 104 CHKERRQ(MatAXPY(Cs,1.0,C,DIFFERENT_NONZERO_PATTERN)); 105 CHKERRQ(MatConvert(Cs,MATAIJ,MAT_INPLACE_MATRIX,&Cs)); 106 CHKERRQ(MatSetOption(Cs,MAT_SYMMETRIC,PETSC_TRUE)); 107 CHKERRQ(MatGetType(Cs,&Cstype)); 108 109 CHKERRQ(PetscObjectSetName((PetscObject)Cs,"symm_initial")); 110 CHKERRQ(MatViewFromOptions(Cs,NULL,"-view")); 111 112 CHKERRQ(MatConvert(Cs,Ctype,MAT_INITIAL_MATRIX,&Cse)); 113 CHKERRQ(PetscObjectSetName((PetscObject)Cse,"symm_conv_init")); 114 CHKERRQ(MatViewFromOptions(Cse,NULL,"-view")); 115 CHKERRQ(MatMultEqual(Cs,Cse,5,&flg)); 116 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatConvert MAT_INITIAL_MATRIX %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 117 118 CHKERRQ(MatConvert(Cs,Ctype,MAT_REUSE_MATRIX,&Cse)); 119 CHKERRQ(PetscObjectSetName((PetscObject)Cse,"symm_conv_reuse")); 120 CHKERRQ(MatViewFromOptions(Cse,NULL,"-view")); 121 CHKERRQ(MatMultEqual(Cs,Cse,5,&flg)); 122 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatConvert MAT_REUSE_MATRIX %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 123 124 CHKERRQ(MatCopy(Cs,Cse,SAME_NONZERO_PATTERN)); 125 CHKERRQ(PetscObjectSetName((PetscObject)Cse,"symm_conv_copy_samennz")); 126 CHKERRQ(MatViewFromOptions(Cse,NULL,"-view")); 127 CHKERRQ(MatMultEqual(Cs,Cse,5,&flg)); 128 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatCopy(...SAME_NONZERO_PATTERN) %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 129 130 CHKERRQ(MatCopy(Cs,Cse,SUBSET_NONZERO_PATTERN)); 131 CHKERRQ(PetscObjectSetName((PetscObject)Cse,"symm_conv_copy_subnnz")); 132 CHKERRQ(MatViewFromOptions(Cse,NULL,"-view")); 133 CHKERRQ(MatMultEqual(Cs,Cse,5,&flg)); 134 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatCopy(...SUBSET_NONZERO_PATTERN) %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 135 136 CHKERRQ(MatCopy(Cs,Cse,DIFFERENT_NONZERO_PATTERN)); 137 CHKERRQ(PetscObjectSetName((PetscObject)Cse,"symm_conv_copy_diffnnz")); 138 CHKERRQ(MatViewFromOptions(Cse,NULL,"-view")); 139 CHKERRQ(MatMultEqual(Cs,Cse,5,&flg)); 140 PetscCheck(flg,PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatCopy(...DIFFERENT_NONZERO_PATTERN) %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 141 142 CHKERRQ(MatDestroy(&Cse)); 143 CHKERRQ(MatDestroy(&Cs)); 144 } 145 146 /* test MatStore/RetrieveValues() */ 147 if (isAIJ) { 148 CHKERRQ(MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE)); 149 CHKERRQ(MatStoreValues(A)); 150 CHKERRQ(MatZeroEntries(A)); 151 CHKERRQ(MatRetrieveValues(A)); 152 } 153 154 CHKERRQ(MatDestroy(&C)); 155 CHKERRQ(MatDestroy(&A)); 156 ierr = PetscFinalize(); 157 return ierr; 158 } 159 160 /*TEST 161 162 testset: 163 nsize: {{1 2}separate output} 164 args: -view ::ascii_info -mat_type {{aij baij sbaij mpiaij mpibaij mpisbaij}separate output} -mat_block_size {{1 2}separate output} 165 166 TEST*/ 167