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