xref: /petsc/src/mat/impls/elemental/matelem.cxx (revision 2adf0be3efa174207a96fd669ae5b85ebe85da0d)
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 #undef __FUNCT__
58 #define __FUNCT__ "MatView_Elemental"
59 static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer)
60 {
61   PetscErrorCode ierr;
62   Mat_Elemental  *a = (Mat_Elemental*)A->data;
63   PetscBool      iascii;
64 
65   PetscFunctionBegin;
66   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
67   if (iascii) {
68     PetscViewerFormat format;
69     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
70     if (format == PETSC_VIEWER_ASCII_INFO) {
71       /* call elemental viewing function */
72       ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr);
73       ierr = PetscViewerASCIIPrintf(viewer,"  allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr);
74       ierr = PetscViewerASCIIPrintf(viewer,"  grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr);
75       if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
76         /* call elemental viewing function */
77         ierr = PetscPrintf(((PetscObject)viewer)->comm,"test matview_elemental 2\n");CHKERRQ(ierr);
78       }
79 
80     } else if (format == PETSC_VIEWER_DEFAULT) {
81       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
82       ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr);
83       a->emat->Print("Elemental matrix (cyclic ordering)");
84       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
85       if (A->factortype == MAT_FACTOR_NONE){
86         Mat Aaij;
87         ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
88         ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr);
89         ierr = MatView(Aaij,viewer);CHKERRQ(ierr);
90         ierr = MatDestroy(&Aaij);CHKERRQ(ierr);
91       }
92     } else SETERRQ(((PetscObject)viewer)->comm,PETSC_ERR_SUP,"Format");
93   } else {
94     /* convert to aij/mpidense format and call MatView() */
95     Mat Aaij;
96     ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
97     ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr);
98     ierr = MatView(Aaij,viewer);CHKERRQ(ierr);
99     ierr = MatDestroy(&Aaij);CHKERRQ(ierr);
100   }
101   PetscFunctionReturn(0);
102 }
103 
104 #undef __FUNCT__
105 #define __FUNCT__ "MatGetInfo_Elemental"
106 static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info)
107 {
108   Mat_Elemental  *a = (Mat_Elemental*)A->data;
109   PetscMPIInt    rank;
110 
111   PetscFunctionBegin;
112   MPI_Comm_rank(((PetscObject)A)->comm,&rank);
113 
114   /* if (!rank) printf("          .........MatGetInfo_Elemental ...\n"); */
115   info->block_size     = 1.0; /* ? */
116 
117   if (flag == MAT_LOCAL) {
118     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
119     info->nz_used        = info->nz_allocated;
120   } else if (flag == MAT_GLOBAL_MAX) {
121     //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,((PetscObject)matin)->comm);CHKERRQ(ierr);
122     /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */
123     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet");
124   } else if (flag == MAT_GLOBAL_SUM) {
125     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet");
126     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
127     info->nz_used        = info->nz_allocated; /* assume Elemental does accurate allocation */
128     //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
129     //PetscPrintf(PETSC_COMM_SELF,"    ... [%d] locally allocated %g\n",rank,info->nz_allocated);
130   }
131 
132   info->nz_unneeded       = 0.0;
133   info->assemblies        = (double)A->num_ass;
134   info->mallocs           = 0;
135   info->memory            = ((PetscObject)A)->mem;
136   info->fill_ratio_given  = 0; /* determined by Elemental */
137   info->fill_ratio_needed = 0;
138   info->factor_mallocs    = 0;
139   PetscFunctionReturn(0);
140 }
141 
142 #undef __FUNCT__
143 #define __FUNCT__ "MatSetValues_Elemental"
144 static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode)
145 {
146   PetscErrorCode ierr;
147   Mat_Elemental  *a = (Mat_Elemental*)A->data;
148   PetscMPIInt    rank;
149   PetscInt       i,j,rrank,ridx,crank,cidx;
150 
151   PetscFunctionBegin;
152   ierr = MPI_Comm_rank(((PetscObject)A)->comm,&rank);CHKERRQ(ierr);
153 
154   const elem::Grid &grid = a->emat->Grid();
155   for (i=0; i<nr; i++) {
156     PetscInt erow,ecol,elrow,elcol;
157     if (rows[i] < 0) continue;
158     P2RO(A,0,rows[i],&rrank,&ridx);
159     RO2E(A,0,rrank,ridx,&erow);
160     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect row translation");
161     for (j=0; j<nc; j++) {
162       if (cols[j] < 0) continue;
163       P2RO(A,1,cols[j],&crank,&cidx);
164       RO2E(A,1,crank,cidx,&ecol);
165       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect col translation");
166       if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */
167         if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported");
168         /* 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); */
169         a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]);
170         a->interface->Axpy(1.0,*(a->esubmat),erow,ecol);
171         continue;
172       }
173       elrow = erow / grid.MCSize();
174       elcol = ecol / grid.MRSize();
175       switch (imode) {
176       case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break;
177       case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break;
178       default: SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode);
179       }
180     }
181   }
182   PetscFunctionReturn(0);
183 }
184 
185 #undef __FUNCT__
186 #define __FUNCT__ "MatMult_Elemental"
187 static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y)
188 {
189   Mat_Elemental         *a = (Mat_Elemental*)A->data;
190   PetscErrorCode        ierr;
191   const PetscElemScalar *x;
192   PetscElemScalar       *y;
193   PetscElemScalar       one = 1,zero = 0;
194 
195   PetscFunctionBegin;
196   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
197   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
198   { /* Scoping so that constructor is called before pointer is returned */
199     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid);
200     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,y,A->rmap->n,*a->grid);
201     elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye);
202   }
203   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
204   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
205   PetscFunctionReturn(0);
206 }
207 
208 #undef __FUNCT__
209 #define __FUNCT__ "MatMultTranspose_Elemental"
210 static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y)
211 {
212   Mat_Elemental         *a = (Mat_Elemental*)A->data;
213   PetscErrorCode        ierr;
214   const PetscElemScalar *x;
215   PetscElemScalar       *y;
216   PetscElemScalar       one = 1,zero = 0;
217 
218   PetscFunctionBegin;
219   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
220   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
221   { /* Scoping so that constructor is called before pointer is returned */
222     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
223     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,y,A->cmap->n,*a->grid);
224     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye);
225   }
226   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
227   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
228   PetscFunctionReturn(0);
229 }
230 
231 #undef __FUNCT__
232 #define __FUNCT__ "MatMultAdd_Elemental"
233 static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
234 {
235   Mat_Elemental         *a = (Mat_Elemental*)A->data;
236   PetscErrorCode        ierr;
237   const PetscElemScalar *x;
238   PetscElemScalar       *z;
239   PetscElemScalar       one = 1;
240 
241   PetscFunctionBegin;
242   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
243   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
244   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
245   { /* Scoping so that constructor is called before pointer is returned */
246     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid);
247     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,z,A->rmap->n,*a->grid);
248     elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze);
249   }
250   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
251   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
252   PetscFunctionReturn(0);
253 }
254 
255 #undef __FUNCT__
256 #define __FUNCT__ "MatMultTransposeAdd_Elemental"
257 static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
258 {
259   Mat_Elemental         *a = (Mat_Elemental*)A->data;
260   PetscErrorCode        ierr;
261   const PetscElemScalar *x;
262   PetscElemScalar       *z;
263   PetscElemScalar       one = 1;
264 
265   PetscFunctionBegin;
266   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
267   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
268   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
269   { /* Scoping so that constructor is called before pointer is returned */
270     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
271     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,z,A->cmap->n,*a->grid);
272     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze);
273   }
274   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
275   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
276   PetscFunctionReturn(0);
277 }
278 
279 #undef __FUNCT__
280 #define __FUNCT__ "MatMatMultNumeric_Elemental"
281 static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C)
282 {
283   Mat_Elemental    *a = (Mat_Elemental*)A->data;
284   Mat_Elemental    *b = (Mat_Elemental*)B->data;
285   Mat_Elemental    *c = (Mat_Elemental*)C->data;
286   PetscElemScalar  one = 1,zero = 0;
287 
288   PetscFunctionBegin;
289   { /* Scoping so that constructor is called before pointer is returned */
290     elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat);
291   }
292   C->assembled = PETSC_TRUE;
293   PetscFunctionReturn(0);
294 }
295 
296 #undef __FUNCT__
297 #define __FUNCT__ "MatMatMultSymbolic_Elemental"
298 static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
299 {
300   PetscErrorCode ierr;
301   Mat            Ce;
302   MPI_Comm       comm=((PetscObject)A)->comm;
303 
304   PetscFunctionBegin;
305   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
306   ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
307   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
308   ierr = MatSetUp(Ce);CHKERRQ(ierr);
309   *C = Ce;
310   PetscFunctionReturn(0);
311 }
312 
313 #undef __FUNCT__
314 #define __FUNCT__ "MatMatMult_Elemental"
315 static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
316 {
317   PetscErrorCode ierr;
318 
319   PetscFunctionBegin;
320   if (scall == MAT_INITIAL_MATRIX){
321     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
322     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
323     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
324   }
325   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
326   ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
327   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
328   PetscFunctionReturn(0);
329 }
330 
331 #undef __FUNCT__
332 #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental"
333 static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C)
334 {
335   Mat_Elemental      *a = (Mat_Elemental*)A->data;
336   Mat_Elemental      *b = (Mat_Elemental*)B->data;
337   Mat_Elemental      *c = (Mat_Elemental*)C->data;
338   PetscElemScalar    one = 1,zero = 0;
339 
340   PetscFunctionBegin;
341   { /* Scoping so that constructor is called before pointer is returned */
342     elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat);
343   }
344   C->assembled = PETSC_TRUE;
345   PetscFunctionReturn(0);
346 }
347 
348 #undef __FUNCT__
349 #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental"
350 static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
351 {
352   PetscErrorCode ierr;
353   Mat            Ce;
354   MPI_Comm       comm=((PetscObject)A)->comm;
355 
356   PetscFunctionBegin;
357   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
358   ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
359   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
360   ierr = MatSetUp(Ce);CHKERRQ(ierr);
361   *C = Ce;
362   PetscFunctionReturn(0);
363 }
364 
365 #undef __FUNCT__
366 #define __FUNCT__ "MatMatTransposeMult_Elemental"
367 static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
368 {
369   PetscErrorCode ierr;
370 
371   PetscFunctionBegin;
372   if (scall == MAT_INITIAL_MATRIX){
373     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
374     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
375     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
376   }
377   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
378   ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
379   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
380   PetscFunctionReturn(0);
381 }
382 
383 #undef __FUNCT__
384 #define __FUNCT__ "MatScale_Elemental"
385 static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a)
386 {
387   Mat_Elemental  *x = (Mat_Elemental*)X->data;
388 
389   PetscFunctionBegin;
390   elem::Scal((PetscElemScalar)a,*x->emat);
391   PetscFunctionReturn(0);
392 }
393 
394 #undef __FUNCT__
395 #define __FUNCT__ "MatAXPY_Elemental"
396 static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str)
397 {
398   Mat_Elemental  *x = (Mat_Elemental*)X->data;
399   Mat_Elemental  *y = (Mat_Elemental*)Y->data;
400 
401   PetscFunctionBegin;
402   elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat);
403   PetscFunctionReturn(0);
404 }
405 
406 #undef __FUNCT__
407 #define __FUNCT__ "MatCopy_Elemental"
408 static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str)
409 {
410   Mat_Elemental *a=(Mat_Elemental*)A->data;
411   Mat_Elemental *b=(Mat_Elemental*)B->data;
412 
413   PetscFunctionBegin;
414   elem::Copy(*a->emat,*b->emat);
415   PetscFunctionReturn(0);
416 }
417 
418 #undef __FUNCT__
419 #define __FUNCT__ "MatDuplicate_Elemental"
420 static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B)
421 {
422   Mat            Be;
423   MPI_Comm       comm=((PetscObject)A)->comm;
424   Mat_Elemental  *a=(Mat_Elemental*)A->data;
425   PetscErrorCode ierr;
426 
427   PetscFunctionBegin;
428   ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
429   ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
430   ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
431   ierr = MatSetUp(Be);CHKERRQ(ierr);
432   *B = Be;
433   if (op == MAT_COPY_VALUES) {
434     Mat_Elemental *b=(Mat_Elemental*)Be->data;
435     elem::Copy(*a->emat,*b->emat);
436   }
437   Be->assembled = PETSC_TRUE;
438   PetscFunctionReturn(0);
439 }
440 
441 #undef __FUNCT__
442 #define __FUNCT__ "MatTranspose_Elemental"
443 static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
444 {
445   /* Only out-of-place supported */
446   Mat            Be;
447   PetscErrorCode ierr;
448   MPI_Comm       comm=((PetscObject)A)->comm;
449   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
450 
451   PetscFunctionBegin;
452   if (reuse == MAT_INITIAL_MATRIX){
453     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
454     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
455     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
456     ierr = MatSetUp(Be);CHKERRQ(ierr);
457     *B = Be;
458   }
459   b = (Mat_Elemental*)Be->data;
460   elem::Transpose(*a->emat,*b->emat);
461   Be->assembled = PETSC_TRUE;
462   PetscFunctionReturn(0);
463 }
464 
465 #undef __FUNCT__
466 #define __FUNCT__ "MatConjugate_Elemental"
467 static PetscErrorCode MatConjugate_Elemental(Mat A)
468 {
469   Mat_Elemental  *a = (Mat_Elemental*)A->data;
470 
471   PetscFunctionBegin;
472   elem::Conjugate(*a->emat);
473   PetscFunctionReturn(0);
474 }
475 
476 #undef __FUNCT__
477 #define __FUNCT__ "MatHermitianTranspose_Elemental"
478 static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
479 {
480   /* Only out-of-place supported */
481   Mat            Be;
482   PetscErrorCode ierr;
483   MPI_Comm       comm=((PetscObject)A)->comm;
484   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
485 
486   PetscFunctionBegin;
487   if (reuse == MAT_INITIAL_MATRIX){
488     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
489     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
490     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
491     ierr = MatSetUp(Be);CHKERRQ(ierr);
492     *B = Be;
493   }
494   b = (Mat_Elemental*)Be->data;
495   elem::Adjoint(*a->emat,*b->emat);
496   Be->assembled = PETSC_TRUE;
497   PetscFunctionReturn(0);
498 }
499 
500 #undef __FUNCT__
501 #define __FUNCT__ "MatSolve_Elemental"
502 static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X)
503 {
504   Mat_Elemental     *a = (Mat_Elemental*)A->data;
505   PetscErrorCode    ierr;
506   PetscElemScalar   *x;
507 
508   PetscFunctionBegin;
509   ierr = VecCopy(B,X);CHKERRQ(ierr);
510   ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
511   elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
512   elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe;
513   switch (A->factortype) {
514   case MAT_FACTOR_LU:
515     if ((*a->pivot).AllocatedMemory()) {
516       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer);
517       elem::Copy(xer,xe);
518     } else {
519       elem::SolveAfterLU(elem::NORMAL,*a->emat,xer);
520       elem::Copy(xer,xe);
521     }
522     break;
523   case MAT_FACTOR_CHOLESKY:
524     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer);
525     elem::Copy(xer,xe);
526     break;
527   default:
528     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
529     break;
530   }
531   ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
532   PetscFunctionReturn(0);
533 }
534 
535 #undef __FUNCT__
536 #define __FUNCT__ "MatSolveAdd_Elemental"
537 static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X)
538 {
539   PetscErrorCode    ierr;
540 
541   PetscFunctionBegin;
542   ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr);
543   ierr = VecAXPY(X,1,Y);CHKERRQ(ierr);
544   PetscFunctionReturn(0);
545 }
546 
547 #undef __FUNCT__
548 #define __FUNCT__ "MatMatSolve_Elemental"
549 static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X)
550 {
551   Mat_Elemental *a=(Mat_Elemental*)A->data;
552   Mat_Elemental *b=(Mat_Elemental*)B->data;
553   Mat_Elemental *x=(Mat_Elemental*)X->data;
554 
555   PetscFunctionBegin;
556   elem::Copy(*b->emat,*x->emat);
557   switch (A->factortype) {
558   case MAT_FACTOR_LU:
559     if ((*a->pivot).AllocatedMemory()) {
560       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat);
561     } else {
562       elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat);
563     }
564     break;
565   case MAT_FACTOR_CHOLESKY:
566     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat);
567     break;
568   default:
569     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
570     break;
571   }
572   PetscFunctionReturn(0);
573 }
574 
575 #undef __FUNCT__
576 #define __FUNCT__ "MatLUFactor_Elemental"
577 static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info)
578 {
579   Mat_Elemental  *a = (Mat_Elemental*)A->data;
580 
581   PetscFunctionBegin;
582   if (info->dtcol){
583     elem::LU(*a->emat,*a->pivot);
584   } else {
585     elem::LU(*a->emat);
586   }
587   A->factortype = MAT_FACTOR_LU;
588   A->assembled  = PETSC_TRUE;
589   PetscFunctionReturn(0);
590 }
591 
592 #undef __FUNCT__
593 #define __FUNCT__ "MatLUFactorNumeric_Elemental"
594 static PetscErrorCode  MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
595 {
596   PetscErrorCode ierr;
597 
598   PetscFunctionBegin;
599   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
600   ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr);
601   PetscFunctionReturn(0);
602 }
603 
604 #undef __FUNCT__
605 #define __FUNCT__ "MatLUFactorSymbolic_Elemental"
606 static PetscErrorCode  MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
607 {
608   PetscFunctionBegin;
609   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
610   PetscFunctionReturn(0);
611 }
612 
613 #undef __FUNCT__
614 #define __FUNCT__ "MatCholeskyFactor_Elemental"
615 static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info)
616 {
617   Mat_Elemental  *a = (Mat_Elemental*)A->data;
618   elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d;
619 
620   PetscFunctionBegin;
621   elem::Cholesky(elem::UPPER,*a->emat);
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,(PetscScalar *)&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, //MatGetDiagonal_Elemental,?
893        0, //MatDiagonalScale_Elemental,?
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        MatConjugate_Elemental,
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        MatHermitianTranspose_Elemental,
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 
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); /* use user-provided grid sizes */
1065     } else {
1066       commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */
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 
1091