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