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