xref: /petsc/src/mat/impls/elemental/matelem.cxx (revision 1576778905152c0ce298ae0d34e7b8a4436c6ffd)
1 #include <../src/mat/impls/elemental/matelemimpl.h> /*I "petscmat.h" I*/
2 
3 /*
4     The variable Petsc_Elemental_keyval is used to indicate an MPI attribute that
5   is attached to a communicator, in this case the attribute is a Mat_Elemental_Grid
6 */
7 static PetscMPIInt Petsc_Elemental_keyval = MPI_KEYVAL_INVALID;
8 
9 #undef __FUNCT__
10 #define __FUNCT__ "PetscElementalInitializePackage"
11 /*@C
12    PetscElementalInitializePackage - Initialize Elemental package
13 
14    Logically Collective
15 
16    Input Arguments:
17 .  path - the dynamic library path or PETSC_NULL
18 
19    Level: developer
20 
21 .seealso: MATELEMENTAL, PetscElementalFinalizePackage()
22 @*/
23 PetscErrorCode PetscElementalInitializePackage(const char *path)
24 {
25   PetscErrorCode ierr;
26 
27   PetscFunctionBegin;
28   if (elem::Initialized()) PetscFunctionReturn(0);
29   { /* We have already initialized MPI, so this song and dance is just to pass these variables (which won't be used by Elemental) through the interface that needs references */
30     int zero = 0;
31     char **nothing = 0;
32     elem::Initialize(zero,nothing);
33   }
34   ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr);
35   PetscFunctionReturn(0);
36 }
37 
38 #undef __FUNCT__
39 #define __FUNCT__ "PetscElementalFinalizePackage"
40 /*@C
41    PetscElementalFinalizePackage - Finalize Elemental package
42 
43    Logically Collective
44 
45    Level: developer
46 
47 .seealso: MATELEMENTAL, PetscElementalInitializePackage()
48 @*/
49 PetscErrorCode PetscElementalFinalizePackage(void)
50 {
51 
52   PetscFunctionBegin;
53   elem::Finalize();
54   PetscFunctionReturn(0);
55 }
56 
57 /* Sets Elemental options from the options database */
58 #undef __FUNCT__
59 #define __FUNCT__ "PetscSetElementalFromOptions"
60 PetscErrorCode PetscSetElementalFromOptions(Mat A)
61 {
62   //Mat_Elemental  *a = (Mat_Elemental*)A->data;
63   PetscErrorCode ierr;
64   //PetscInt       optv1,optv2;
65   //PetscBool      flg;
66 
67   PetscFunctionBegin;
68   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr);
69   //ierr = PetscOptionsInt("-mat_elemental_grid_height","Gird Height","None",(*a->emat).Grid().Height(),&optv1,&flg);CHKERRQ(ierr);
70   //ierr = PetscOptionsInt("-mat_elemental_grid_width","Gird Width","None",(*a->emat).Grid().Width(),&optv2,&flg);CHKERRQ(ierr);
71   //if (flg) (*a->emat).Grid().ResizeTo(optv1,optv2);
72   // ierr = PetscOptionsInt("-mat_mumps_icntl_2","ICNTL(2): output stream for diagnostic printing, statistics, and warning","None",mumps->id.ICNTL(2),&icntl,&flg);CHKERRQ(ierr);
73   // if (flg) mumps->id.ICNTL(2) = icntl;
74   // ierr = PetscOptionsInt("-mat_mumps_icntl_3","ICNTL(3): output stream for global information, collected on the host","None",mumps->id.ICNTL(3),&icntl,&flg);CHKERRQ(ierr);
75   // if (flg) mumps->id.ICNTL(3) = icntl;
76 
77 
78   PetscOptionsEnd();
79   PetscFunctionReturn(0);
80 }
81 
82 #undef __FUNCT__
83 #define __FUNCT__ "MatView_Elemental"
84 static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer)
85 {
86   PetscErrorCode ierr;
87   Mat_Elemental  *a = (Mat_Elemental*)A->data;
88   PetscBool      iascii;
89 
90   PetscFunctionBegin;
91   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
92   if (iascii) {
93     PetscViewerFormat format;
94     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
95     if (format == PETSC_VIEWER_ASCII_INFO) {
96       /* call elemental viewing function */
97       ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr);
98       ierr = PetscViewerASCIIPrintf(viewer,"  allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr);
99       ierr = PetscViewerASCIIPrintf(viewer,"  grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr);
100       if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
101         /* call elemental viewing function */
102         ierr = PetscPrintf(((PetscObject)viewer)->comm,"test matview_elemental 2\n");CHKERRQ(ierr);
103       }
104 
105     } else if (format == PETSC_VIEWER_DEFAULT) {
106       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
107       ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr);
108       a->emat->Print("Elemental matrix (cyclic ordering)");
109       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
110       if (A->factortype == MAT_FACTOR_NONE){
111         Mat Aaij;
112         ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
113         ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr);
114         ierr = MatView(Aaij,viewer);CHKERRQ(ierr);
115         ierr = MatDestroy(&Aaij);CHKERRQ(ierr);
116       }
117     } else SETERRQ(((PetscObject)viewer)->comm,PETSC_ERR_SUP,"Format");
118   } else {
119     /* convert to aij/mpidense format and call MatView() */
120     Mat Aaij;
121     ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
122     ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr);
123     ierr = MatView(Aaij,viewer);CHKERRQ(ierr);
124     ierr = MatDestroy(&Aaij);CHKERRQ(ierr);
125   }
126   PetscFunctionReturn(0);
127 }
128 
129 #undef __FUNCT__
130 #define __FUNCT__ "MatGetInfo_Elemental"
131 static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info)
132 {
133   Mat_Elemental  *a = (Mat_Elemental*)A->data;
134   PetscMPIInt    rank;
135 
136   PetscFunctionBegin;
137   MPI_Comm_rank(((PetscObject)A)->comm,&rank);
138 
139   /* if (!rank) printf("          .........MatGetInfo_Elemental ...\n"); */
140   info->block_size     = 1.0; /* ? */
141 
142   if (flag == MAT_LOCAL) {
143     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
144     info->nz_used        = info->nz_allocated;
145   } else if (flag == MAT_GLOBAL_MAX) {
146     //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,((PetscObject)matin)->comm);CHKERRQ(ierr);
147     /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */
148     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet");
149   } else if (flag == MAT_GLOBAL_SUM) {
150     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet");
151     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
152     info->nz_used        = info->nz_allocated; /* assume Elemental does accurate allocation */
153     //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
154     //PetscPrintf(PETSC_COMM_SELF,"    ... [%d] locally allocated %g\n",rank,info->nz_allocated);
155   }
156 
157   info->nz_unneeded       = 0.0;
158   info->assemblies        = (double)A->num_ass;
159   info->mallocs           = 0;
160   info->memory            = ((PetscObject)A)->mem;
161   info->fill_ratio_given  = 0; /* determined by Elemental */
162   info->fill_ratio_needed = 0;
163   info->factor_mallocs    = 0;
164   PetscFunctionReturn(0);
165 }
166 
167 #undef __FUNCT__
168 #define __FUNCT__ "MatSetValues_Elemental"
169 static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscElemScalar *vals,InsertMode imode)
170 {
171   PetscErrorCode ierr;
172   Mat_Elemental  *a = (Mat_Elemental*)A->data;
173   PetscMPIInt    rank;
174   PetscInt       i,j,rrank,ridx,crank,cidx;
175 
176   PetscFunctionBegin;
177   ierr = MPI_Comm_rank(((PetscObject)A)->comm,&rank);CHKERRQ(ierr);
178 
179   const elem::Grid &grid = a->emat->Grid();
180   for (i=0; i<nr; i++) {
181     PetscInt erow,ecol,elrow,elcol;
182     if (rows[i] < 0) continue;
183     P2RO(A,0,rows[i],&rrank,&ridx);
184     RO2E(A,0,rrank,ridx,&erow);
185     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect row translation");
186     for (j=0; j<nc; j++) {
187       if (cols[j] < 0) continue;
188       P2RO(A,1,cols[j],&crank,&cidx);
189       RO2E(A,1,crank,cidx,&ecol);
190       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect col translation");
191       if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */
192         if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported");
193         /* PetscPrintf(PETSC_COMM_SELF,"[%D] add off-proc entry (%D,%D, %g) (%D %D)\n",rank,rows[i],cols[j],*(vals+i*nc),erow,ecol); */
194         a->esubmat->Set(0,0, vals[i*nc+j]);
195         a->interface->Axpy(1.0,*(a->esubmat),erow,ecol);
196         continue;
197       }
198       elrow = erow / grid.MCSize();
199       elcol = ecol / grid.MRSize();
200       switch (imode) {
201       case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,vals[i*nc+j]); break;
202       case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,vals[i*nc+j]); break;
203       default: SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode);
204       }
205     }
206   }
207   PetscFunctionReturn(0);
208 }
209 
210 #undef __FUNCT__
211 #define __FUNCT__ "MatMult_Elemental"
212 static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y)
213 {
214   Mat_Elemental     *a = (Mat_Elemental*)A->data;
215   PetscErrorCode    ierr;
216   const PetscScalar *x;
217   PetscScalar       *y;
218   PetscElemScalar       one = 1,zero = 0;
219 
220   PetscFunctionBegin;
221   ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
222   ierr = VecGetArray(Y,&y);CHKERRQ(ierr);
223   { /* Scoping so that constructor is called before pointer is returned */
224     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid);
225     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,y,A->rmap->n,*a->grid);
226     elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye);
227   }
228   ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
229   ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr);
230   PetscFunctionReturn(0);
231 }
232 
233 #undef __FUNCT__
234 #define __FUNCT__ "MatMultTranspose_Elemental"
235 static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y)
236 {
237   Mat_Elemental     *a = (Mat_Elemental*)A->data;
238   PetscErrorCode    ierr;
239   const PetscElemScalar *x;
240   PetscElemScalar       *y;
241   PetscScalar       one = 1,zero = 0;
242 
243   PetscFunctionBegin;
244   ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
245   ierr = VecGetArray(Y,&y);CHKERRQ(ierr);
246   { /* Scoping so that constructor is called before pointer is returned */
247     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
248     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,y,A->cmap->n,*a->grid);
249     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye);
250   }
251   ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
252   ierr = VecRestoreArray(Y,&y);CHKERRQ(ierr);
253   PetscFunctionReturn(0);
254 }
255 
256 #undef __FUNCT__
257 #define __FUNCT__ "MatMultAdd_Elemental"
258 static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
259 {
260   Mat_Elemental     *a = (Mat_Elemental*)A->data;
261   PetscErrorCode    ierr;
262   const PetscElemScalar *x;
263   PetscElemScalar       *z;
264   PetscScalar       one = 1;
265 
266   PetscFunctionBegin;
267   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
268   ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
269   ierr = VecGetArray(Z,&z);CHKERRQ(ierr);
270   { /* Scoping so that constructor is called before pointer is returned */
271     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid);
272     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,z,A->rmap->n,*a->grid);
273     elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze);
274   }
275   ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
276   ierr = VecRestoreArray(Z,&z);CHKERRQ(ierr);
277   PetscFunctionReturn(0);
278 }
279 
280 #undef __FUNCT__
281 #define __FUNCT__ "MatMultTransposeAdd_Elemental"
282 static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
283 {
284   Mat_Elemental     *a = (Mat_Elemental*)A->data;
285   PetscErrorCode    ierr;
286   const PetscElemScalar *x;
287   PetscElemScalar       *z;
288   PetscScalar       one = 1;
289 
290   PetscFunctionBegin;
291   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
292   ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
293   ierr = VecGetArray(Z,&z);CHKERRQ(ierr);
294   { /* Scoping so that constructor is called before pointer is returned */
295     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
296     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,z,A->cmap->n,*a->grid);
297     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze);
298   }
299   ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
300   ierr = VecRestoreArray(Z,&z);CHKERRQ(ierr);
301   PetscFunctionReturn(0);
302 }
303 
304 #undef __FUNCT__
305 #define __FUNCT__ "MatMatMultNumeric_Elemental"
306 static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C)
307 {
308   Mat_Elemental  *a = (Mat_Elemental*)A->data;
309   Mat_Elemental  *b = (Mat_Elemental*)B->data;
310   Mat_Elemental  *c = (Mat_Elemental*)C->data;
311   PetscScalar    one = 1,zero = 0;
312 
313   PetscFunctionBegin;
314   { /* Scoping so that constructor is called before pointer is returned */
315     elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat);
316   }
317   C->assembled = PETSC_TRUE;
318   PetscFunctionReturn(0);
319 }
320 
321 #undef __FUNCT__
322 #define __FUNCT__ "MatMatMultSymbolic_Elemental"
323 static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
324 {
325   PetscErrorCode ierr;
326   Mat            Ce;
327   MPI_Comm       comm=((PetscObject)A)->comm;
328 
329   PetscFunctionBegin;
330   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
331   ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
332   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
333   ierr = MatSetUp(Ce);CHKERRQ(ierr);
334   *C = Ce;
335   PetscFunctionReturn(0);
336 }
337 
338 #undef __FUNCT__
339 #define __FUNCT__ "MatMatMult_Elemental"
340 static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
341 {
342   PetscErrorCode ierr;
343 
344   PetscFunctionBegin;
345   if (scall == MAT_INITIAL_MATRIX){
346     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
347     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
348     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
349   }
350   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
351   ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
352   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
353   PetscFunctionReturn(0);
354 }
355 
356 #undef __FUNCT__
357 #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental"
358 static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C)
359 {
360   Mat_Elemental  *a = (Mat_Elemental*)A->data;
361   Mat_Elemental  *b = (Mat_Elemental*)B->data;
362   Mat_Elemental  *c = (Mat_Elemental*)C->data;
363   PetscScalar    one = 1,zero = 0;
364 
365   PetscFunctionBegin;
366   { /* Scoping so that constructor is called before pointer is returned */
367     elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat);
368   }
369   C->assembled = PETSC_TRUE;
370   PetscFunctionReturn(0);
371 }
372 
373 #undef __FUNCT__
374 #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental"
375 static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
376 {
377   PetscErrorCode ierr;
378   Mat            Ce;
379   MPI_Comm       comm=((PetscObject)A)->comm;
380 
381   PetscFunctionBegin;
382   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
383   ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
384   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
385   ierr = MatSetUp(Ce);CHKERRQ(ierr);
386   *C = Ce;
387   PetscFunctionReturn(0);
388 }
389 
390 #undef __FUNCT__
391 #define __FUNCT__ "MatMatTransposeMult_Elemental"
392 static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
393 {
394   PetscErrorCode ierr;
395 
396   PetscFunctionBegin;
397   if (scall == MAT_INITIAL_MATRIX){
398     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
399     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
400     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
401   }
402   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
403   ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
404   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
405   PetscFunctionReturn(0);
406 }
407 
408 #undef __FUNCT__
409 #define __FUNCT__ "MatScale_Elemental"
410 static PetscErrorCode MatScale_Elemental(Mat X,PetscElemScalar a)
411 {
412   Mat_Elemental  *x = (Mat_Elemental*)X->data;
413 
414   PetscFunctionBegin;
415   elem::Scal(a,*x->emat);
416   PetscFunctionReturn(0);
417 }
418 
419 #undef __FUNCT__
420 #define __FUNCT__ "MatAXPY_Elemental"
421 static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscElemScalar a,Mat X,MatStructure str)
422 {
423   Mat_Elemental  *x = (Mat_Elemental*)X->data;
424   Mat_Elemental  *y = (Mat_Elemental*)Y->data;
425 
426   PetscFunctionBegin;
427   elem::Axpy(a,*x->emat,*y->emat);
428   PetscFunctionReturn(0);
429 }
430 
431 #undef __FUNCT__
432 #define __FUNCT__ "MatCopy_Elemental"
433 static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str)
434 {
435   Mat_Elemental *a=(Mat_Elemental*)A->data;
436   Mat_Elemental *b=(Mat_Elemental*)B->data;
437 
438   PetscFunctionBegin;
439   elem::Copy(*a->emat,*b->emat);
440   PetscFunctionReturn(0);
441 }
442 
443 #undef __FUNCT__
444 #define __FUNCT__ "MatDuplicate_Elemental"
445 static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B)
446 {
447   Mat            Be;
448   MPI_Comm       comm=((PetscObject)A)->comm;
449   Mat_Elemental  *a=(Mat_Elemental*)A->data;
450   PetscErrorCode ierr;
451 
452   PetscFunctionBegin;
453   ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
454   ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
455   ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
456   ierr = MatSetUp(Be);CHKERRQ(ierr);
457   *B = Be;
458   if (op == MAT_COPY_VALUES) {
459     Mat_Elemental *b=(Mat_Elemental*)Be->data;
460     elem::Copy(*a->emat,*b->emat);
461   }
462   Be->assembled = PETSC_TRUE;
463   PetscFunctionReturn(0);
464 }
465 
466 #undef __FUNCT__
467 #define __FUNCT__ "MatTranspose_Elemental"
468 static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
469 {
470   /* Only out-of-place supported */
471   Mat            Be;
472   PetscErrorCode ierr;
473   MPI_Comm       comm=((PetscObject)A)->comm;
474   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
475 
476   PetscFunctionBegin;
477   if (reuse == MAT_INITIAL_MATRIX){
478     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
479     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
480     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
481     ierr = MatSetUp(Be);CHKERRQ(ierr);
482     *B = Be;
483   }
484   b = (Mat_Elemental*)Be->data;
485   elem::Transpose(*a->emat,*b->emat);
486   Be->assembled = PETSC_TRUE;
487   PetscFunctionReturn(0);
488 }
489 
490 #undef __FUNCT__
491 #define __FUNCT__ "MatSolve_Elemental"
492 static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X)
493 {
494   Mat_Elemental     *a = (Mat_Elemental*)A->data;
495   PetscErrorCode    ierr;
496   PetscElemScalar       *x;
497 
498   PetscFunctionBegin;
499   ierr = VecCopy(B,X);CHKERRQ(ierr);
500   ierr = VecGetArray(X,&x);CHKERRQ(ierr);
501   elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
502   elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe;
503   switch (A->factortype) {
504   case MAT_FACTOR_LU:
505     if ((*a->pivot).AllocatedMemory()) {
506       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer);
507       elem::Copy(xer,xe);
508     } else {
509       elem::SolveAfterLU(elem::NORMAL,*a->emat,xer);
510       elem::Copy(xer,xe);
511     }
512     break;
513   case MAT_FACTOR_CHOLESKY:
514     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer);
515     elem::Copy(xer,xe);
516     break;
517   default:
518     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
519     break;
520   }
521   ierr = VecRestoreArray(X,&x);CHKERRQ(ierr);
522   PetscFunctionReturn(0);
523 }
524 
525 #undef __FUNCT__
526 #define __FUNCT__ "MatSolveAdd_Elemental"
527 static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X)
528 {
529   PetscErrorCode    ierr;
530 
531   PetscFunctionBegin;
532   ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr);
533   ierr = VecAXPY(X,1,Y);CHKERRQ(ierr);
534   PetscFunctionReturn(0);
535 }
536 
537 #undef __FUNCT__
538 #define __FUNCT__ "MatMatSolve_Elemental"
539 static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X)
540 {
541   Mat_Elemental *a=(Mat_Elemental*)A->data;
542   Mat_Elemental *b=(Mat_Elemental*)B->data;
543   Mat_Elemental *x=(Mat_Elemental*)X->data;
544 
545   PetscFunctionBegin;
546   elem::Copy(*b->emat,*x->emat);
547   switch (A->factortype) {
548   case MAT_FACTOR_LU:
549     if ((*a->pivot).AllocatedMemory()) {
550       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat);
551     } else {
552       elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat);
553     }
554     break;
555   case MAT_FACTOR_CHOLESKY:
556     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat);
557     break;
558   default:
559     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
560     break;
561   }
562   PetscFunctionReturn(0);
563 }
564 
565 #undef __FUNCT__
566 #define __FUNCT__ "MatLUFactor_Elemental"
567 static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info)
568 {
569   Mat_Elemental  *a = (Mat_Elemental*)A->data;
570 
571   PetscFunctionBegin;
572   if (info->dtcol){
573     elem::LU(*a->emat,*a->pivot);
574   } else {
575     elem::LU(*a->emat);
576   }
577   A->factortype = MAT_FACTOR_LU;
578   A->assembled  = PETSC_TRUE;
579   PetscFunctionReturn(0);
580 }
581 
582 #undef __FUNCT__
583 #define __FUNCT__ "MatLUFactorNumeric_Elemental"
584 static PetscErrorCode  MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
585 {
586   PetscErrorCode ierr;
587 
588   PetscFunctionBegin;
589   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
590   ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr);
591   PetscFunctionReturn(0);
592 }
593 
594 #undef __FUNCT__
595 #define __FUNCT__ "MatLUFactorSymbolic_Elemental"
596 static PetscErrorCode  MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
597 {
598   PetscFunctionBegin;
599   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
600   PetscFunctionReturn(0);
601 }
602 
603 #undef __FUNCT__
604 #define __FUNCT__ "MatCholeskyFactor_Elemental"
605 static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info)
606 {
607   Mat_Elemental  *a = (Mat_Elemental*)A->data;
608   elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d;
609 
610   PetscFunctionBegin;
611   elem::Cholesky(elem::UPPER,*a->emat);
612   // if (info->dtcol){
613   //   /* A = U^T * U for SPD Matrix A */
614   //   printf("Cholesky Factorization for SPD Matrices...\n");
615   //   elem::Cholesky(elem::UPPER,*a->emat);
616   // } else {
617   //   /* A = U^T * D * U * for Symmetric Matrix A */
618   //   printf("LDL^T Factorization for Symmetric Matrices.\n");
619   //   printf("This routine does not pivot. Use with caution.\n");
620   //   elem::LDLT(*a->emat,d);
621   // }
622   A->factortype = MAT_FACTOR_CHOLESKY;
623   A->assembled  = PETSC_TRUE;
624   PetscFunctionReturn(0);
625 }
626 
627 #undef __FUNCT__
628 #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental"
629 static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
630 {
631   PetscErrorCode ierr;
632 
633   PetscFunctionBegin;
634   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
635   ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr);
636   PetscFunctionReturn(0);
637 }
638 
639 #undef __FUNCT__
640 #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental"
641 static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info)
642 {
643   PetscFunctionBegin;
644   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
645   PetscFunctionReturn(0);
646 }
647 
648 EXTERN_C_BEGIN
649 #undef __FUNCT__
650 #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental"
651 PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type)
652 {
653   PetscFunctionBegin;
654   *type = MATSOLVERELEMENTAL;
655   PetscFunctionReturn(0);
656 }
657 EXTERN_C_END
658 
659 EXTERN_C_BEGIN
660 #undef __FUNCT__
661 #define __FUNCT__ "MatGetFactor_elemental_elemental"
662 static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F)
663 {
664   Mat            B;
665   PetscErrorCode ierr;
666 
667   PetscFunctionBegin;
668   /* Create the factorization matrix */
669   ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
670   ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
671   ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr);
672   ierr = MatSetUp(B);CHKERRQ(ierr);
673   B->factortype = ftype;
674   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_elemental_elemental",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr);
675   *F            = B;
676   PetscFunctionReturn(0);
677 }
678 EXTERN_C_END
679 
680 #undef __FUNCT__
681 #define __FUNCT__ "MatNorm_Elemental"
682 static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm)
683 {
684   Mat_Elemental *a=(Mat_Elemental*)A->data;
685 
686   PetscFunctionBegin;
687   switch (type){
688   case NORM_1:
689     *nrm = elem::Norm(*a->emat,elem::ONE_NORM);
690     break;
691   case NORM_FROBENIUS:
692     *nrm = elem::Norm(*a->emat,elem::FROBENIUS_NORM);
693     break;
694   case NORM_INFINITY:
695     *nrm = elem::Norm(*a->emat,elem::INFINITY_NORM);
696     break;
697   default:
698     printf("Error: unsupported norm type!\n");
699   }
700   PetscFunctionReturn(0);
701 }
702 
703 #undef __FUNCT__
704 #define __FUNCT__ "MatZeroEntries_Elemental"
705 static PetscErrorCode MatZeroEntries_Elemental(Mat A)
706 {
707   Mat_Elemental *a=(Mat_Elemental*)A->data;
708 
709   PetscFunctionBegin;
710   elem::Zero(*a->emat);
711   PetscFunctionReturn(0);
712 }
713 
714 EXTERN_C_BEGIN
715 #undef __FUNCT__
716 #define __FUNCT__ "MatGetOwnershipIS_Elemental"
717 static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols)
718 {
719   Mat_Elemental  *a = (Mat_Elemental*)A->data;
720   PetscErrorCode ierr;
721   PetscInt       i,m,shift,stride,*idx;
722 
723   PetscFunctionBegin;
724   if (rows) {
725     m = a->emat->LocalHeight();
726     shift = a->emat->ColShift();
727     stride = a->emat->ColStride();
728     ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr);
729     for (i=0; i<m; i++) {
730       PetscInt rank,offset;
731       E2RO(A,0,shift+i*stride,&rank,&offset);
732       RO2P(A,0,rank,offset,&idx[i]);
733     }
734     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr);
735   }
736   if (cols) {
737     m = a->emat->LocalWidth();
738     shift = a->emat->RowShift();
739     stride = a->emat->RowStride();
740     ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr);
741     for (i=0; i<m; i++) {
742       PetscInt rank,offset;
743       E2RO(A,1,shift+i*stride,&rank,&offset);
744       RO2P(A,1,rank,offset,&idx[i]);
745     }
746     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr);
747   }
748   PetscFunctionReturn(0);
749 }
750 EXTERN_C_END
751 
752 #undef __FUNCT__
753 #define __FUNCT__ "MatConvert_Elemental_Dense"
754 static PetscErrorCode MatConvert_Elemental_Dense(Mat A,const MatType newtype,MatReuse reuse,Mat *B)
755 {
756   Mat                Bmpi;
757   Mat_Elemental      *a = (Mat_Elemental*)A->data;
758   MPI_Comm           comm=((PetscObject)A)->comm;
759   PetscErrorCode     ierr;
760   PetscInt           rrank,ridx,crank,cidx,nrows,ncols,i,j;
761   PetscElemScalar        v;
762 
763   PetscFunctionBegin;
764   if (strcmp(newtype,MATDENSE) && strcmp(newtype,MATSEQDENSE) && strcmp(newtype,MATMPIDENSE)) {
765     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE");
766   }
767   ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr);
768   ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
769   ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr);
770   ierr = MatSetUp(Bmpi);CHKERRQ(ierr);
771   ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr);
772   for (i=0; i<nrows; i++) {
773     PetscInt erow,ecol;
774     P2RO(A,0,i,&rrank,&ridx);
775     RO2E(A,0,rrank,ridx,&erow);
776     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
777     for (j=0; j<ncols; j++) {
778       P2RO(A,1,j,&crank,&cidx);
779       RO2E(A,1,crank,cidx,&ecol);
780       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
781       v = a->emat->Get(erow,ecol);
782       ierr = MatSetValues(Bmpi,1,&i,1,&j,&v,INSERT_VALUES);CHKERRQ(ierr);
783     }
784   }
785   ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
786   ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
787   if (reuse == MAT_REUSE_MATRIX) {
788     ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr);
789   } else {
790     *B = Bmpi;
791   }
792   PetscFunctionReturn(0);
793 }
794 
795 #undef __FUNCT__
796 #define __FUNCT__ "MatDestroy_Elemental"
797 static PetscErrorCode MatDestroy_Elemental(Mat A)
798 {
799   Mat_Elemental      *a = (Mat_Elemental*)A->data;
800   PetscErrorCode     ierr;
801   Mat_Elemental_Grid *commgrid;
802   PetscBool          flg;
803   MPI_Comm           icomm;
804 
805   PetscFunctionBegin;
806   a->interface->Detach();
807   delete a->interface;
808   delete a->esubmat;
809   delete a->emat;
810 
811   elem::mpi::Comm cxxcomm(((PetscObject)A)->comm);
812   ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr);
813   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
814   if (--commgrid->grid_refct == 0) {
815     delete commgrid->grid;
816     ierr = PetscFree(commgrid);CHKERRQ(ierr);
817   }
818   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
819   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","",PETSC_NULL);CHKERRQ(ierr);
820   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_petsc_C","",PETSC_NULL);CHKERRQ(ierr);
821   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatFactorGetSolverPackage_C","",PETSC_NULL);CHKERRQ(ierr);
822   ierr = PetscFree(A->data);CHKERRQ(ierr);
823   PetscFunctionReturn(0);
824 }
825 
826 #undef __FUNCT__
827 #define __FUNCT__ "MatSetUp_Elemental"
828 PetscErrorCode MatSetUp_Elemental(Mat A)
829 {
830   Mat_Elemental  *a = (Mat_Elemental*)A->data;
831   PetscErrorCode ierr;
832   PetscMPIInt    rsize,csize;
833 
834   PetscFunctionBegin;
835   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
836   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
837 
838   a->emat->ResizeTo(A->rmap->N,A->cmap->N);CHKERRQ(ierr);
839   elem::Zero(*a->emat);
840 
841   ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr);
842   ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr);
843   if (csize != rsize) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes");
844   a->commsize = rsize;
845   a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize;
846   a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize;
847   a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize);
848   a->m[1] = A->cmap->N / csize + (a->mr[1] != csize);
849   PetscFunctionReturn(0);
850 }
851 
852 #undef __FUNCT__
853 #define __FUNCT__ "MatAssemblyBegin_Elemental"
854 PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type)
855 {
856   Mat_Elemental  *a = (Mat_Elemental*)A->data;
857 
858   PetscFunctionBegin;
859   a->interface->Detach();
860   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
861   PetscFunctionReturn(0);
862 }
863 
864 #undef __FUNCT__
865 #define __FUNCT__ "MatAssemblyEnd_Elemental"
866 PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type)
867 {
868   PetscFunctionBegin;
869   /* Currently does nothing */
870   PetscFunctionReturn(0);
871 }
872 
873 /* -------------------------------------------------------------------*/
874 static struct _MatOps MatOps_Values = {
875        MatSetValues_Elemental,
876        0,
877        0,
878        MatMult_Elemental,
879 /* 4*/ MatMultAdd_Elemental,
880        MatMultTranspose_Elemental,
881        MatMultTransposeAdd_Elemental,
882        MatSolve_Elemental,
883        MatSolveAdd_Elemental,
884        0, //MatSolveTranspose_Elemental,
885 /*10*/ 0, //MatSolveTransposeAdd_Elemental,
886        MatLUFactor_Elemental,
887        MatCholeskyFactor_Elemental,
888        0,
889        MatTranspose_Elemental,
890 /*15*/ MatGetInfo_Elemental,
891        0,
892        0,
893        0,
894        MatNorm_Elemental,
895 /*20*/ MatAssemblyBegin_Elemental,
896        MatAssemblyEnd_Elemental,
897        0, //MatSetOption_Elemental,
898        MatZeroEntries_Elemental,
899 /*24*/ 0,
900        MatLUFactorSymbolic_Elemental,
901        MatLUFactorNumeric_Elemental,
902        MatCholeskyFactorSymbolic_Elemental,
903        MatCholeskyFactorNumeric_Elemental,
904 /*29*/ MatSetUp_Elemental,
905        0,
906        0,
907        0,
908        0,
909 /*34*/ MatDuplicate_Elemental,
910        0,
911        0,
912        0,
913        0,
914 /*39*/ MatAXPY_Elemental,
915        0,
916        0,
917        0,
918        MatCopy_Elemental,
919 /*44*/ 0,
920        MatScale_Elemental,
921        0,
922        0,
923        0,
924 /*49*/ 0,
925        0,
926        0,
927        0,
928        0,
929 /*54*/ 0,
930        0,
931        0,
932        0,
933        0,
934 /*59*/ 0,
935        MatDestroy_Elemental,
936        MatView_Elemental,
937        0,
938        0,
939 /*64*/ 0,
940        0,
941        0,
942        0,
943        0,
944 /*69*/ 0,
945        0,
946        MatConvert_Elemental_Dense,
947        0,
948        0,
949 /*74*/ 0,
950        0,
951        0,
952        0,
953        0,
954 /*79*/ 0,
955        0,
956        0,
957        0,
958        0,
959 /*84*/ 0,
960        0,
961        0,
962        0,
963        0,
964 /*89*/ MatMatMult_Elemental,
965        MatMatMultSymbolic_Elemental,
966        MatMatMultNumeric_Elemental,
967        0,
968        0,
969 /*94*/ 0,
970        MatMatTransposeMult_Elemental,
971        MatMatTransposeMultSymbolic_Elemental,
972        MatMatTransposeMultNumeric_Elemental,
973        0,
974 /*99*/ 0,
975        0,
976        0,
977        0,
978        0,
979 /*104*/0,
980        0,
981        0,
982        0,
983        0,
984 /*109*/MatMatSolve_Elemental,
985        0,
986        0,
987        0,
988        0,
989 /*114*/0,
990        0,
991        0,
992        0,
993        0,
994 /*119*/0,
995        0,
996        0,
997        0,
998        0,
999 /*124*/0,
1000        0,
1001        0,
1002        0,
1003        0,
1004 /*129*/0,
1005        0,
1006        0,
1007        0,
1008        0,
1009 /*134*/0,
1010        0,
1011        0,
1012        0,
1013        0
1014 };
1015 
1016 /*MC
1017    MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package
1018 
1019    Options Database Keys:
1020 . -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions()
1021 
1022   Level: beginner
1023 
1024 .seealso: MATDENSE,MatCreateElemental()
1025 M*/
1026 EXTERN_C_BEGIN
1027 #undef __FUNCT__
1028 #define __FUNCT__ "MatCreate_Elemental"
1029 PETSC_EXTERN_C PetscErrorCode MatCreate_Elemental(Mat A)
1030 {
1031   Mat_Elemental      *a;
1032   PetscErrorCode     ierr;
1033   PetscBool          flg,flg1,flg2;
1034   Mat_Elemental_Grid *commgrid;
1035   MPI_Comm           icomm;
1036   PetscInt           optv1,optv2;
1037 
1038   PetscFunctionBegin;
1039   ierr = PetscElementalInitializePackage(PETSC_NULL);CHKERRQ(ierr);
1040   ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1041   A->insertmode = NOT_SET_VALUES;
1042 
1043   ierr = PetscNewLog(A,Mat_Elemental,&a);CHKERRQ(ierr);
1044   A->data = (void*)a;
1045 
1046   /* Set up the elemental matrix */
1047   elem::mpi::Comm cxxcomm(((PetscObject)A)->comm);
1048   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr);
1049 
1050   /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */
1051   if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) {
1052     ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0);
1053   }
1054   ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr);
1055   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
1056   if (!flg) {
1057     ierr = PetscNewLog(A,Mat_Elemental_Grid,&commgrid);CHKERRQ(ierr);
1058     ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr);
1059     ierr = PetscOptionsInt("-mat_elemental_grid_width","Grid Width","None",1,&optv2,&flg2);CHKERRQ(ierr);
1060     if (flg1 || flg2) {
1061       if (optv1*optv2 != elem::mpi::CommSize(cxxcomm)) {
1062         SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Grid Height times Grid Width must equal CommSize");
1063       }
1064       commgrid->grid = new elem::Grid(cxxcomm,optv1,optv2);
1065     } else {
1066       commgrid->grid = new elem::Grid(cxxcomm);
1067     }
1068     commgrid->grid_refct = 1;
1069     ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr);
1070   } else {
1071     commgrid->grid_refct++;
1072   }
1073   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
1074   a->grid      = commgrid->grid;
1075   a->emat      = new elem::DistMatrix<PetscElemScalar>(*a->grid);
1076   a->esubmat   = new elem::Matrix<PetscElemScalar>(1,1);
1077   a->interface = new elem::AxpyInterface<PetscElemScalar>;
1078   a->pivot     = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>;
1079 
1080   /* build cache for off array entries formed */
1081   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
1082 
1083   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","MatGetOwnershipIS_Elemental",MatGetOwnershipIS_Elemental);CHKERRQ(ierr);
1084   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_elemental_C","MatGetFactor_elemental_elemental",MatGetFactor_elemental_elemental);CHKERRQ(ierr);
1085 
1086   ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr);
1087   PetscOptionsEnd();
1088   PetscFunctionReturn(0);
1089 }
1090 EXTERN_C_END
1091