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