1 
2 static char help[] = "Tests MatCopy() and MatStore/RetrieveValues().\n\n";
3 
4 #include <petscmat.h>
5 
main(int argc,char ** args)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