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