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 ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr); 18 ierr = PetscOptionsGetInt(NULL,NULL,"-n",&n,NULL);CHKERRQ(ierr); 19 ierr = PetscOptionsGetInt(NULL,NULL,"-mat_block_size",&bs,NULL);CHKERRQ(ierr); 20 21 ierr = MatCreate(PETSC_COMM_WORLD,&C);CHKERRQ(ierr); 22 ierr = MatSetSizes(C,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr); 23 ierr = MatSetType(C,MATAIJ);CHKERRQ(ierr); 24 ierr = MatSetFromOptions(C);CHKERRQ(ierr); 25 ierr = PetscObjectSetName((PetscObject)C,"initial");CHKERRQ(ierr); 26 27 ierr = MatGetType(C,&type);CHKERRQ(ierr); 28 if (size == 1) { 29 ierr = PetscObjectTypeCompare((PetscObject)C,MATSEQAIJ,&isAIJ);CHKERRQ(ierr); 30 } else { 31 ierr = PetscObjectTypeCompare((PetscObject)C,MATMPIAIJ,&isAIJ);CHKERRQ(ierr); 32 } 33 ierr = MatSeqAIJSetPreallocation(C,3,NULL);CHKERRQ(ierr); 34 ierr = MatMPIAIJSetPreallocation(C,3,NULL,3,NULL);CHKERRQ(ierr); 35 ierr = MatSeqBAIJSetPreallocation(C,bs,3,NULL);CHKERRQ(ierr); 36 ierr = MatMPIBAIJSetPreallocation(C,bs,3,NULL,3,NULL);CHKERRQ(ierr); 37 ierr = MatSeqSBAIJSetPreallocation(C,bs,3,NULL);CHKERRQ(ierr); 38 ierr = MatMPISBAIJSetPreallocation(C,bs,3,NULL,3,NULL);CHKERRQ(ierr); 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 ierr = MatSetValues(C,1,&i,3,midx,v,INSERT_VALUES);CHKERRQ(ierr); 44 } 45 i = 0; midx[0] = 0; midx[1] = 1; 46 v[0] = 2.0; v[1] = -1.; 47 ierr = MatSetValues(C,1,&i,2,midx,v,INSERT_VALUES);CHKERRQ(ierr); 48 i = n-1; midx[0] = n-2; midx[1] = n-1; 49 v[0] = -1.0; v[1] = 2.; 50 ierr = MatSetValues(C,1,&i,2,midx,v,INSERT_VALUES);CHKERRQ(ierr); 51 52 ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 53 ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 54 ierr = MatView(C,NULL);CHKERRQ(ierr); 55 ierr = MatViewFromOptions(C,NULL,"-view");CHKERRQ(ierr); 56 57 /* test matduplicate */ 58 ierr = MatDuplicate(C,MAT_COPY_VALUES,&A);CHKERRQ(ierr); 59 ierr = PetscObjectSetName((PetscObject)A,"duplicate_copy");CHKERRQ(ierr); 60 ierr = MatViewFromOptions(A,NULL,"-view");CHKERRQ(ierr); 61 ierr = MatEqual(A,C,&flg);CHKERRQ(ierr); 62 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatDuplicate(C,MAT_COPY_VALUES,): Matrices are NOT equal"); 63 ierr = MatDestroy(&A);CHKERRQ(ierr); 64 65 /* test matrices with different nonzero patterns - Note: A is created with different nonzero pattern of C! */ 66 ierr = MatCreate(PETSC_COMM_WORLD,&A);CHKERRQ(ierr); 67 ierr = MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,n,n);CHKERRQ(ierr); 68 ierr = MatSetFromOptions(A);CHKERRQ(ierr); 69 ierr = MatSetUp(A);CHKERRQ(ierr); 70 ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 71 ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 72 73 ierr = MatCopy(C,A,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 74 ierr = PetscObjectSetName((PetscObject)A,"copy_diffnnz");CHKERRQ(ierr); 75 ierr = MatViewFromOptions(A,NULL,"-view");CHKERRQ(ierr); 76 ierr = MatEqual(A,C,&flg);CHKERRQ(ierr); 77 if (!flg) SETERRQ(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 ierr = MatDestroy(&A);CHKERRQ(ierr); 81 ierr = MatDuplicate(C,MAT_DO_NOT_COPY_VALUES,&A);CHKERRQ(ierr); 82 ierr = MatCopy(C,A,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 83 ierr = PetscObjectSetName((PetscObject)A,"copy_samennz");CHKERRQ(ierr); 84 ierr = MatViewFromOptions(A,NULL,"-view");CHKERRQ(ierr); 85 ierr = MatEqual(A,C,&flg);CHKERRQ(ierr); 86 if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"MatCopy(C,A,SAME_NONZERO_PATTERN): Matrices are NOT equal"); 87 88 /* test subset nonzero pattern */ 89 ierr = MatCopy(C,A,SUBSET_NONZERO_PATTERN);CHKERRQ(ierr); 90 ierr = PetscObjectSetName((PetscObject)A,"copy_subnnz");CHKERRQ(ierr); 91 ierr = MatViewFromOptions(A,NULL,"-view");CHKERRQ(ierr); 92 ierr = MatEqual(A,C,&flg);CHKERRQ(ierr); 93 if (!flg) SETERRQ(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 ierr = MatHasCongruentLayouts(C,&flg);CHKERRQ(ierr); 98 if (flg) { 99 Mat Cs,Cse; 100 MatType Ctype,Cstype; 101 102 ierr = MatGetType(C,&Ctype);CHKERRQ(ierr); 103 ierr = MatTranspose(C,MAT_INITIAL_MATRIX,&Cs);CHKERRQ(ierr); 104 ierr = MatAXPY(Cs,1.0,C,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 105 ierr = MatConvert(Cs,MATAIJ,MAT_INPLACE_MATRIX,&Cs);CHKERRQ(ierr); 106 ierr = MatSetOption(Cs,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr); 107 ierr = MatGetType(Cs,&Cstype);CHKERRQ(ierr); 108 109 ierr = PetscObjectSetName((PetscObject)Cs,"symm_initial");CHKERRQ(ierr); 110 ierr = MatViewFromOptions(Cs,NULL,"-view");CHKERRQ(ierr); 111 112 ierr = MatConvert(Cs,Ctype,MAT_INITIAL_MATRIX,&Cse);CHKERRQ(ierr); 113 ierr = PetscObjectSetName((PetscObject)Cse,"symm_conv_init");CHKERRQ(ierr); 114 ierr = MatViewFromOptions(Cse,NULL,"-view");CHKERRQ(ierr); 115 ierr = MatMultEqual(Cs,Cse,5,&flg);CHKERRQ(ierr); 116 if (!flg) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatConvert MAT_INITIAL_MATRIX %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 117 118 ierr = MatConvert(Cs,Ctype,MAT_REUSE_MATRIX,&Cse);CHKERRQ(ierr); 119 ierr = PetscObjectSetName((PetscObject)Cse,"symm_conv_reuse");CHKERRQ(ierr); 120 ierr = MatViewFromOptions(Cse,NULL,"-view");CHKERRQ(ierr); 121 ierr = MatMultEqual(Cs,Cse,5,&flg);CHKERRQ(ierr); 122 if (!flg) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatConvert MAT_REUSE_MATRIX %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 123 124 ierr = MatCopy(Cs,Cse,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 125 ierr = PetscObjectSetName((PetscObject)Cse,"symm_conv_copy_samennz");CHKERRQ(ierr); 126 ierr = MatViewFromOptions(Cse,NULL,"-view");CHKERRQ(ierr); 127 ierr = MatMultEqual(Cs,Cse,5,&flg);CHKERRQ(ierr); 128 if (!flg) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatCopy(...SAME_NONZERO_PATTERN) %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 129 130 ierr = MatCopy(Cs,Cse,SUBSET_NONZERO_PATTERN);CHKERRQ(ierr); 131 ierr = PetscObjectSetName((PetscObject)Cse,"symm_conv_copy_subnnz");CHKERRQ(ierr); 132 ierr = MatViewFromOptions(Cse,NULL,"-view");CHKERRQ(ierr); 133 ierr = MatMultEqual(Cs,Cse,5,&flg);CHKERRQ(ierr); 134 if (!flg) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatCopy(...SUBSET_NONZERO_PATTERN) %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 135 136 ierr = MatCopy(Cs,Cse,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 137 ierr = PetscObjectSetName((PetscObject)Cse,"symm_conv_copy_diffnnz");CHKERRQ(ierr); 138 ierr = MatViewFromOptions(Cse,NULL,"-view");CHKERRQ(ierr); 139 ierr = MatMultEqual(Cs,Cse,5,&flg);CHKERRQ(ierr); 140 if (!flg) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"MatCopy(...DIFFERENT_NONZERO_PATTERN) %s -> %s: Matrices are NOT multequal",Ctype,Cstype); 141 142 ierr = MatDestroy(&Cse);CHKERRQ(ierr); 143 ierr = MatDestroy(&Cs);CHKERRQ(ierr); 144 } 145 146 /* test MatStore/RetrieveValues() */ 147 if (isAIJ) { 148 ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);CHKERRQ(ierr); 149 ierr = MatStoreValues(A);CHKERRQ(ierr); 150 ierr = MatZeroEntries(A);CHKERRQ(ierr); 151 ierr = MatRetrieveValues(A);CHKERRQ(ierr); 152 } 153 154 ierr = MatDestroy(&C);CHKERRQ(ierr); 155 ierr = MatDestroy(&A);CHKERRQ(ierr); 156 ierr = PetscFinalize(); 157 return ierr; 158 } 159 160 161 162 163 /*TEST 164 165 testset: 166 nsize: {{1 2}separate output} 167 args: -view ::ascii_info -mat_type {{aij baij sbaij mpiaij mpibaij mpisbaij}separate output} -mat_block_size {{1 2}separate output} 168 169 TEST*/ 170