xref: /petsc/src/mat/impls/elemental/matelem.cxx (revision 23d4d1a5ea4abd8afa816ae028e8f894b7d8ac8f)
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 Adense;
87         ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
88         ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr);
89         ierr = MatView(Adense,viewer);CHKERRQ(ierr);
90         ierr = MatDestroy(&Adense);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 Adense;
96     ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
97     ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr);
98     ierr = MatView(Adense,viewer);CHKERRQ(ierr);
99     ierr = MatDestroy(&Adense);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__ "MatGetDiagonal_Elemental"
385 static PetscErrorCode MatGetDiagonal_Elemental(Mat X,Vec D)
386 {
387   Mat_Elemental   *x = (Mat_Elemental*)X->data;
388   PetscElemScalar *d;
389   PetscErrorCode  ierr;
390 
391   PetscFunctionBegin;
392   ierr = VecGetArray(D,(PetscScalar **)&d);CHKERRQ(ierr);
393   elem::DistMatrix<PetscElemScalar,elem::MD,elem::STAR> de((X->rmap->N > X->cmap->N)?X->cmap->N:X->rmap->N,1,0,d,(X->rmap->N > X->cmap->N)?X->cmap->n:X->rmap->n,*x->grid);
394   x->emat->GetDiagonal(de,0);
395   ierr = VecRestoreArray(D,(PetscScalar **)&d);CHKERRQ(ierr);
396   PetscFunctionReturn(0);
397 }
398 
399 #undef __FUNCT__
400 #define __FUNCT__ "MatScale_Elemental"
401 static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a)
402 {
403   Mat_Elemental  *x = (Mat_Elemental*)X->data;
404 
405   PetscFunctionBegin;
406   elem::Scal((PetscElemScalar)a,*x->emat);
407   PetscFunctionReturn(0);
408 }
409 
410 #undef __FUNCT__
411 #define __FUNCT__ "MatAXPY_Elemental"
412 static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str)
413 {
414   Mat_Elemental  *x = (Mat_Elemental*)X->data;
415   Mat_Elemental  *y = (Mat_Elemental*)Y->data;
416 
417   PetscFunctionBegin;
418   elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat);
419   PetscFunctionReturn(0);
420 }
421 
422 #undef __FUNCT__
423 #define __FUNCT__ "MatCopy_Elemental"
424 static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str)
425 {
426   Mat_Elemental *a=(Mat_Elemental*)A->data;
427   Mat_Elemental *b=(Mat_Elemental*)B->data;
428 
429   PetscFunctionBegin;
430   elem::Copy(*a->emat,*b->emat);
431   PetscFunctionReturn(0);
432 }
433 
434 #undef __FUNCT__
435 #define __FUNCT__ "MatDuplicate_Elemental"
436 static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B)
437 {
438   Mat            Be;
439   MPI_Comm       comm=((PetscObject)A)->comm;
440   Mat_Elemental  *a=(Mat_Elemental*)A->data;
441   PetscErrorCode ierr;
442 
443   PetscFunctionBegin;
444   ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
445   ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
446   ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
447   ierr = MatSetUp(Be);CHKERRQ(ierr);
448   *B = Be;
449   if (op == MAT_COPY_VALUES) {
450     Mat_Elemental *b=(Mat_Elemental*)Be->data;
451     elem::Copy(*a->emat,*b->emat);
452   }
453   Be->assembled = PETSC_TRUE;
454   PetscFunctionReturn(0);
455 }
456 
457 #undef __FUNCT__
458 #define __FUNCT__ "MatTranspose_Elemental"
459 static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
460 {
461   /* Only out-of-place supported */
462   Mat            Be;
463   PetscErrorCode ierr;
464   MPI_Comm       comm=((PetscObject)A)->comm;
465   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
466 
467   PetscFunctionBegin;
468   if (reuse == MAT_INITIAL_MATRIX){
469     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
470     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
471     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
472     ierr = MatSetUp(Be);CHKERRQ(ierr);
473     *B = Be;
474   }
475   b = (Mat_Elemental*)Be->data;
476   elem::Transpose(*a->emat,*b->emat);
477   Be->assembled = PETSC_TRUE;
478   PetscFunctionReturn(0);
479 }
480 
481 #undef __FUNCT__
482 #define __FUNCT__ "MatConjugate_Elemental"
483 static PetscErrorCode MatConjugate_Elemental(Mat A)
484 {
485   Mat_Elemental  *a = (Mat_Elemental*)A->data;
486 
487   PetscFunctionBegin;
488   elem::Conjugate(*a->emat);
489   PetscFunctionReturn(0);
490 }
491 
492 #undef __FUNCT__
493 #define __FUNCT__ "MatHermitianTranspose_Elemental"
494 static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
495 {
496   /* Only out-of-place supported */
497   Mat            Be;
498   PetscErrorCode ierr;
499   MPI_Comm       comm=((PetscObject)A)->comm;
500   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
501 
502   PetscFunctionBegin;
503   if (reuse == MAT_INITIAL_MATRIX){
504     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
505     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
506     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
507     ierr = MatSetUp(Be);CHKERRQ(ierr);
508     *B = Be;
509   }
510   b = (Mat_Elemental*)Be->data;
511   elem::Adjoint(*a->emat,*b->emat);
512   Be->assembled = PETSC_TRUE;
513   PetscFunctionReturn(0);
514 }
515 
516 #undef __FUNCT__
517 #define __FUNCT__ "MatSolve_Elemental"
518 static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X)
519 {
520   Mat_Elemental     *a = (Mat_Elemental*)A->data;
521   PetscErrorCode    ierr;
522   PetscElemScalar   *x;
523 
524   PetscFunctionBegin;
525   ierr = VecCopy(B,X);CHKERRQ(ierr);
526   ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
527   elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
528   elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe;
529   switch (A->factortype) {
530   case MAT_FACTOR_LU:
531     if ((*a->pivot).AllocatedMemory()) {
532       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer);
533       elem::Copy(xer,xe);
534     } else {
535       elem::SolveAfterLU(elem::NORMAL,*a->emat,xer);
536       elem::Copy(xer,xe);
537     }
538     break;
539   case MAT_FACTOR_CHOLESKY:
540     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer);
541     elem::Copy(xer,xe);
542     break;
543   default:
544     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
545     break;
546   }
547   ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
548   PetscFunctionReturn(0);
549 }
550 
551 #undef __FUNCT__
552 #define __FUNCT__ "MatSolveAdd_Elemental"
553 static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X)
554 {
555   PetscErrorCode    ierr;
556 
557   PetscFunctionBegin;
558   ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr);
559   ierr = VecAXPY(X,1,Y);CHKERRQ(ierr);
560   PetscFunctionReturn(0);
561 }
562 
563 #undef __FUNCT__
564 #define __FUNCT__ "MatMatSolve_Elemental"
565 static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X)
566 {
567   Mat_Elemental *a=(Mat_Elemental*)A->data;
568   Mat_Elemental *b=(Mat_Elemental*)B->data;
569   Mat_Elemental *x=(Mat_Elemental*)X->data;
570 
571   PetscFunctionBegin;
572   elem::Copy(*b->emat,*x->emat);
573   switch (A->factortype) {
574   case MAT_FACTOR_LU:
575     if ((*a->pivot).AllocatedMemory()) {
576       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat);
577     } else {
578       elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat);
579     }
580     break;
581   case MAT_FACTOR_CHOLESKY:
582     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat);
583     break;
584   default:
585     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
586     break;
587   }
588   PetscFunctionReturn(0);
589 }
590 
591 #undef __FUNCT__
592 #define __FUNCT__ "MatLUFactor_Elemental"
593 static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info)
594 {
595   Mat_Elemental  *a = (Mat_Elemental*)A->data;
596 
597   PetscFunctionBegin;
598   if (info->dtcol){
599     elem::LU(*a->emat,*a->pivot);
600   } else {
601     elem::LU(*a->emat);
602   }
603   A->factortype = MAT_FACTOR_LU;
604   A->assembled  = PETSC_TRUE;
605   PetscFunctionReturn(0);
606 }
607 
608 #undef __FUNCT__
609 #define __FUNCT__ "MatLUFactorNumeric_Elemental"
610 static PetscErrorCode  MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
611 {
612   PetscErrorCode ierr;
613 
614   PetscFunctionBegin;
615   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
616   ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr);
617   PetscFunctionReturn(0);
618 }
619 
620 #undef __FUNCT__
621 #define __FUNCT__ "MatLUFactorSymbolic_Elemental"
622 static PetscErrorCode  MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
623 {
624   PetscFunctionBegin;
625   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
626   PetscFunctionReturn(0);
627 }
628 
629 #undef __FUNCT__
630 #define __FUNCT__ "MatCholeskyFactor_Elemental"
631 static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info)
632 {
633   Mat_Elemental  *a = (Mat_Elemental*)A->data;
634   elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d;
635 
636   PetscFunctionBegin;
637   elem::Cholesky(elem::UPPER,*a->emat);
638   A->factortype = MAT_FACTOR_CHOLESKY;
639   A->assembled  = PETSC_TRUE;
640   PetscFunctionReturn(0);
641 }
642 
643 #undef __FUNCT__
644 #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental"
645 static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
646 {
647   PetscErrorCode ierr;
648 
649   PetscFunctionBegin;
650   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
651   ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr);
652   PetscFunctionReturn(0);
653 }
654 
655 #undef __FUNCT__
656 #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental"
657 static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info)
658 {
659   PetscFunctionBegin;
660   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
661   PetscFunctionReturn(0);
662 }
663 
664 EXTERN_C_BEGIN
665 #undef __FUNCT__
666 #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental"
667 PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type)
668 {
669   PetscFunctionBegin;
670   *type = MATSOLVERELEMENTAL;
671   PetscFunctionReturn(0);
672 }
673 EXTERN_C_END
674 
675 EXTERN_C_BEGIN
676 #undef __FUNCT__
677 #define __FUNCT__ "MatGetFactor_elemental_elemental"
678 static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F)
679 {
680   Mat            B;
681   PetscErrorCode ierr;
682 
683   PetscFunctionBegin;
684   /* Create the factorization matrix */
685   ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
686   ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
687   ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr);
688   ierr = MatSetUp(B);CHKERRQ(ierr);
689   B->factortype = ftype;
690   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_elemental_elemental",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr);
691   *F            = B;
692   PetscFunctionReturn(0);
693 }
694 EXTERN_C_END
695 
696 #undef __FUNCT__
697 #define __FUNCT__ "MatNorm_Elemental"
698 static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm)
699 {
700   Mat_Elemental *a=(Mat_Elemental*)A->data;
701 
702   PetscFunctionBegin;
703   switch (type){
704   case NORM_1:
705     *nrm = elem::Norm(*a->emat,elem::ONE_NORM);
706     break;
707   case NORM_FROBENIUS:
708     *nrm = elem::Norm(*a->emat,elem::FROBENIUS_NORM);
709     break;
710   case NORM_INFINITY:
711     *nrm = elem::Norm(*a->emat,elem::INFINITY_NORM);
712     break;
713   default:
714     printf("Error: unsupported norm type!\n");
715   }
716   PetscFunctionReturn(0);
717 }
718 
719 #undef __FUNCT__
720 #define __FUNCT__ "MatZeroEntries_Elemental"
721 static PetscErrorCode MatZeroEntries_Elemental(Mat A)
722 {
723   Mat_Elemental *a=(Mat_Elemental*)A->data;
724 
725   PetscFunctionBegin;
726   elem::Zero(*a->emat);
727   PetscFunctionReturn(0);
728 }
729 
730 EXTERN_C_BEGIN
731 #undef __FUNCT__
732 #define __FUNCT__ "MatGetOwnershipIS_Elemental"
733 static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols)
734 {
735   Mat_Elemental  *a = (Mat_Elemental*)A->data;
736   PetscErrorCode ierr;
737   PetscInt       i,m,shift,stride,*idx;
738 
739   PetscFunctionBegin;
740   if (rows) {
741     m = a->emat->LocalHeight();
742     shift = a->emat->ColShift();
743     stride = a->emat->ColStride();
744     ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr);
745     for (i=0; i<m; i++) {
746       PetscInt rank,offset;
747       E2RO(A,0,shift+i*stride,&rank,&offset);
748       RO2P(A,0,rank,offset,&idx[i]);
749     }
750     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr);
751   }
752   if (cols) {
753     m = a->emat->LocalWidth();
754     shift = a->emat->RowShift();
755     stride = a->emat->RowStride();
756     ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr);
757     for (i=0; i<m; i++) {
758       PetscInt rank,offset;
759       E2RO(A,1,shift+i*stride,&rank,&offset);
760       RO2P(A,1,rank,offset,&idx[i]);
761     }
762     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr);
763   }
764   PetscFunctionReturn(0);
765 }
766 EXTERN_C_END
767 
768 #undef __FUNCT__
769 #define __FUNCT__ "MatConvert_Elemental_Dense"
770 static PetscErrorCode MatConvert_Elemental_Dense(Mat A,const MatType newtype,MatReuse reuse,Mat *B)
771 {
772   Mat                Bmpi;
773   Mat_Elemental      *a = (Mat_Elemental*)A->data;
774   MPI_Comm           comm=((PetscObject)A)->comm;
775   PetscErrorCode     ierr;
776   PetscInt           rrank,ridx,crank,cidx,nrows,ncols,i,j;
777   PetscElemScalar    v;
778 
779   PetscFunctionBegin;
780   if (strcmp(newtype,MATDENSE) && strcmp(newtype,MATSEQDENSE) && strcmp(newtype,MATMPIDENSE)) {
781     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE");
782   }
783   ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr);
784   ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
785   ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr);
786   ierr = MatSetUp(Bmpi);CHKERRQ(ierr);
787   ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr);
788   for (i=0; i<nrows; i++) {
789     PetscInt erow,ecol;
790     P2RO(A,0,i,&rrank,&ridx);
791     RO2E(A,0,rrank,ridx,&erow);
792     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
793     for (j=0; j<ncols; j++) {
794       P2RO(A,1,j,&crank,&cidx);
795       RO2E(A,1,crank,cidx,&ecol);
796       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
797       v = a->emat->Get(erow,ecol);
798       ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr);
799     }
800   }
801   ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
802   ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
803   if (reuse == MAT_REUSE_MATRIX) {
804     ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr);
805   } else {
806     *B = Bmpi;
807   }
808   PetscFunctionReturn(0);
809 }
810 
811 #undef __FUNCT__
812 #define __FUNCT__ "MatDestroy_Elemental"
813 static PetscErrorCode MatDestroy_Elemental(Mat A)
814 {
815   Mat_Elemental      *a = (Mat_Elemental*)A->data;
816   PetscErrorCode     ierr;
817   Mat_Elemental_Grid *commgrid;
818   PetscBool          flg;
819   MPI_Comm           icomm;
820 
821   PetscFunctionBegin;
822   a->interface->Detach();
823   delete a->interface;
824   delete a->esubmat;
825   delete a->emat;
826 
827   elem::mpi::Comm cxxcomm(((PetscObject)A)->comm);
828   ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr);
829   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
830   if (--commgrid->grid_refct == 0) {
831     delete commgrid->grid;
832     ierr = PetscFree(commgrid);CHKERRQ(ierr);
833   }
834   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
835   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","",PETSC_NULL);CHKERRQ(ierr);
836   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_petsc_C","",PETSC_NULL);CHKERRQ(ierr);
837   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatFactorGetSolverPackage_C","",PETSC_NULL);CHKERRQ(ierr);
838   ierr = PetscFree(A->data);CHKERRQ(ierr);
839   PetscFunctionReturn(0);
840 }
841 
842 #undef __FUNCT__
843 #define __FUNCT__ "MatSetUp_Elemental"
844 PetscErrorCode MatSetUp_Elemental(Mat A)
845 {
846   Mat_Elemental  *a = (Mat_Elemental*)A->data;
847   PetscErrorCode ierr;
848   PetscMPIInt    rsize,csize;
849 
850   PetscFunctionBegin;
851   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
852   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
853 
854   a->emat->ResizeTo(A->rmap->N,A->cmap->N);CHKERRQ(ierr);
855   elem::Zero(*a->emat);
856 
857   ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr);
858   ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr);
859   if (csize != rsize) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes");
860   a->commsize = rsize;
861   a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize;
862   a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize;
863   a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize);
864   a->m[1] = A->cmap->N / csize + (a->mr[1] != csize);
865   PetscFunctionReturn(0);
866 }
867 
868 #undef __FUNCT__
869 #define __FUNCT__ "MatAssemblyBegin_Elemental"
870 PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type)
871 {
872   Mat_Elemental  *a = (Mat_Elemental*)A->data;
873 
874   PetscFunctionBegin;
875   a->interface->Detach();
876   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
877   PetscFunctionReturn(0);
878 }
879 
880 #undef __FUNCT__
881 #define __FUNCT__ "MatAssemblyEnd_Elemental"
882 PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type)
883 {
884   PetscFunctionBegin;
885   /* Currently does nothing */
886   PetscFunctionReturn(0);
887 }
888 
889 /* -------------------------------------------------------------------*/
890 static struct _MatOps MatOps_Values = {
891        MatSetValues_Elemental,
892        0,
893        0,
894        MatMult_Elemental,
895 /* 4*/ MatMultAdd_Elemental,
896        MatMultTranspose_Elemental,
897        MatMultTransposeAdd_Elemental,
898        MatSolve_Elemental,
899        MatSolveAdd_Elemental,
900        0, //MatSolveTranspose_Elemental,
901 /*10*/ 0, //MatSolveTransposeAdd_Elemental,
902        MatLUFactor_Elemental,
903        MatCholeskyFactor_Elemental,
904        0,
905        MatTranspose_Elemental,
906 /*15*/ MatGetInfo_Elemental,
907        0,
908        MatGetDiagonal_Elemental,
909        0, //MatDiagonalScale_Elemental,?
910        MatNorm_Elemental,
911 /*20*/ MatAssemblyBegin_Elemental,
912        MatAssemblyEnd_Elemental,
913        0, //MatSetOption_Elemental,
914        MatZeroEntries_Elemental,
915 /*24*/ 0,
916        MatLUFactorSymbolic_Elemental,
917        MatLUFactorNumeric_Elemental,
918        MatCholeskyFactorSymbolic_Elemental,
919        MatCholeskyFactorNumeric_Elemental,
920 /*29*/ MatSetUp_Elemental,
921        0,
922        0,
923        0,
924        0,
925 /*34*/ MatDuplicate_Elemental,
926        0,
927        0,
928        0,
929        0,
930 /*39*/ MatAXPY_Elemental,
931        0,
932        0,
933        0,
934        MatCopy_Elemental,
935 /*44*/ 0,
936        MatScale_Elemental,
937        0,
938        0,
939        0,
940 /*49*/ 0,
941        0,
942        0,
943        0,
944        0,
945 /*54*/ 0,
946        0,
947        0,
948        0,
949        0,
950 /*59*/ 0,
951        MatDestroy_Elemental,
952        MatView_Elemental,
953        0,
954        0,
955 /*64*/ 0,
956        0,
957        0,
958        0,
959        0,
960 /*69*/ 0,
961        0,
962        MatConvert_Elemental_Dense,
963        0,
964        0,
965 /*74*/ 0,
966        0,
967        0,
968        0,
969        0,
970 /*79*/ 0,
971        0,
972        0,
973        0,
974        0,
975 /*84*/ 0,
976        0,
977        0,
978        0,
979        0,
980 /*89*/ MatMatMult_Elemental,
981        MatMatMultSymbolic_Elemental,
982        MatMatMultNumeric_Elemental,
983        0,
984        0,
985 /*94*/ 0,
986        MatMatTransposeMult_Elemental,
987        MatMatTransposeMultSymbolic_Elemental,
988        MatMatTransposeMultNumeric_Elemental,
989        0,
990 /*99*/ 0,
991        0,
992        0,
993        MatConjugate_Elemental,
994        0,
995 /*104*/0,
996        0,
997        0,
998        0,
999        0,
1000 /*109*/MatMatSolve_Elemental,
1001        0,
1002        0,
1003        0,
1004        0,
1005 /*114*/0,
1006        0,
1007        0,
1008        0,
1009        0,
1010 /*119*/0,
1011        MatHermitianTranspose_Elemental,
1012        0,
1013        0,
1014        0,
1015 /*124*/0,
1016        0,
1017        0,
1018        0,
1019        0,
1020 /*129*/0,
1021        0,
1022        0,
1023        0,
1024        0,
1025 /*134*/0,
1026        0,
1027        0,
1028        0,
1029        0
1030 };
1031 
1032 /*MC
1033    MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package
1034 
1035    Options Database Keys:
1036 . -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions()
1037 
1038   Level: beginner
1039 
1040 .seealso: MATDENSE,MatCreateElemental()
1041 M*/
1042 
1043 #undef __FUNCT__
1044 #define __FUNCT__ "MatCreate_Elemental"
1045 PETSC_EXTERN_C PetscErrorCode MatCreate_Elemental(Mat A)
1046 {
1047   Mat_Elemental      *a;
1048   PetscErrorCode     ierr;
1049   PetscBool          flg,flg1,flg2;
1050   Mat_Elemental_Grid *commgrid;
1051   MPI_Comm           icomm;
1052   PetscInt           optv1,optv2;
1053 
1054   PetscFunctionBegin;
1055   ierr = PetscElementalInitializePackage(PETSC_NULL);CHKERRQ(ierr);
1056   ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1057   A->insertmode = NOT_SET_VALUES;
1058 
1059   ierr = PetscNewLog(A,Mat_Elemental,&a);CHKERRQ(ierr);
1060   A->data = (void*)a;
1061 
1062   /* Set up the elemental matrix */
1063   elem::mpi::Comm cxxcomm(((PetscObject)A)->comm);
1064   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr);
1065 
1066   /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */
1067   if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) {
1068     ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0);
1069   }
1070   ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr);
1071   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
1072   if (!flg) {
1073     ierr = PetscNewLog(A,Mat_Elemental_Grid,&commgrid);CHKERRQ(ierr);
1074     ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr);
1075     ierr = PetscOptionsInt("-mat_elemental_grid_width","Grid Width","None",1,&optv2,&flg2);CHKERRQ(ierr);
1076     if (flg1 || flg2) {
1077       if (optv1*optv2 != elem::mpi::CommSize(cxxcomm)) {
1078         SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Grid Height times Grid Width must equal CommSize");
1079       }
1080       commgrid->grid = new elem::Grid(cxxcomm,optv1,optv2); /* use user-provided grid sizes */
1081     } else {
1082       commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */
1083     }
1084     commgrid->grid_refct = 1;
1085     ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr);
1086   } else {
1087     commgrid->grid_refct++;
1088   }
1089   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
1090   a->grid      = commgrid->grid;
1091   a->emat      = new elem::DistMatrix<PetscElemScalar>(*a->grid);
1092   a->esubmat   = new elem::Matrix<PetscElemScalar>(1,1);
1093   a->interface = new elem::AxpyInterface<PetscElemScalar>;
1094   a->pivot     = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>;
1095 
1096   /* build cache for off array entries formed */
1097   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
1098 
1099   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","MatGetOwnershipIS_Elemental",MatGetOwnershipIS_Elemental);CHKERRQ(ierr);
1100   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_elemental_C","MatGetFactor_elemental_elemental",MatGetFactor_elemental_elemental);CHKERRQ(ierr);
1101 
1102   ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr);
1103   PetscOptionsEnd();
1104   PetscFunctionReturn(0);
1105 }
1106 
1107