xref: /petsc/src/mat/impls/elemental/matelem.cxx (revision ec8cb81fa2e1a7169c626fcae18281d9b036689a)
1db31f6deSJed Brown #include <../src/mat/impls/elemental/matelemimpl.h> /*I "petscmat.h" I*/
2db31f6deSJed Brown 
35e9f5b67SHong Zhang /*
45e9f5b67SHong Zhang     The variable Petsc_Elemental_keyval is used to indicate an MPI attribute that
55e9f5b67SHong Zhang   is attached to a communicator, in this case the attribute is a Mat_Elemental_Grid
65e9f5b67SHong Zhang */
75e9f5b67SHong Zhang static PetscMPIInt Petsc_Elemental_keyval = MPI_KEYVAL_INVALID;
85e9f5b67SHong Zhang 
9db31f6deSJed Brown #undef __FUNCT__
10db31f6deSJed Brown #define __FUNCT__ "PetscElementalInitializePackage"
11db31f6deSJed Brown /*@C
12db31f6deSJed Brown    PetscElementalInitializePackage - Initialize Elemental package
13db31f6deSJed Brown 
14db31f6deSJed Brown    Logically Collective
15db31f6deSJed Brown 
16db31f6deSJed Brown    Level: developer
17db31f6deSJed Brown 
18db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalFinalizePackage()
19db31f6deSJed Brown @*/
20607a6623SBarry Smith PetscErrorCode PetscElementalInitializePackage(void)
21db31f6deSJed Brown {
22db31f6deSJed Brown   PetscErrorCode ierr;
23db31f6deSJed Brown 
24db31f6deSJed Brown   PetscFunctionBegin;
25db31f6deSJed Brown   if (elem::Initialized()) PetscFunctionReturn(0);
26db31f6deSJed Brown   { /* 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 */
27db31f6deSJed Brown     int zero = 0;
28db31f6deSJed Brown     char **nothing = 0;
29db31f6deSJed Brown     elem::Initialize(zero,nothing);
30db31f6deSJed Brown   }
31db31f6deSJed Brown   ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr);
32db31f6deSJed Brown   PetscFunctionReturn(0);
33db31f6deSJed Brown }
34db31f6deSJed Brown 
35db31f6deSJed Brown #undef __FUNCT__
36db31f6deSJed Brown #define __FUNCT__ "PetscElementalFinalizePackage"
37db31f6deSJed Brown /*@C
38db31f6deSJed Brown    PetscElementalFinalizePackage - Finalize Elemental package
39db31f6deSJed Brown 
40db31f6deSJed Brown    Logically Collective
41db31f6deSJed Brown 
42db31f6deSJed Brown    Level: developer
43db31f6deSJed Brown 
44db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalInitializePackage()
45db31f6deSJed Brown @*/
46db31f6deSJed Brown PetscErrorCode PetscElementalFinalizePackage(void)
47db31f6deSJed Brown {
48db31f6deSJed Brown 
49db31f6deSJed Brown   PetscFunctionBegin;
50db31f6deSJed Brown   elem::Finalize();
51db31f6deSJed Brown   PetscFunctionReturn(0);
52db31f6deSJed Brown }
53db31f6deSJed Brown 
54db31f6deSJed Brown #undef __FUNCT__
55db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental"
56db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer)
57db31f6deSJed Brown {
58db31f6deSJed Brown   PetscErrorCode ierr;
59db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
60db31f6deSJed Brown   PetscBool      iascii;
61db31f6deSJed Brown 
62db31f6deSJed Brown   PetscFunctionBegin;
63db31f6deSJed Brown   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
64db31f6deSJed Brown   if (iascii) {
65db31f6deSJed Brown     PetscViewerFormat format;
66db31f6deSJed Brown     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
67db31f6deSJed Brown     if (format == PETSC_VIEWER_ASCII_INFO) {
6879673f7bSHong Zhang       /* call elemental viewing function */
692d8adcc7SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr);
70ed667823SXuan Zhou       ierr = PetscViewerASCIIPrintf(viewer,"  allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr);
71ed667823SXuan Zhou       ierr = PetscViewerASCIIPrintf(viewer,"  grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr);
724fe7bbcaSHong Zhang       if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
7379673f7bSHong Zhang         /* call elemental viewing function */
74ce94432eSBarry Smith         ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"test matview_elemental 2\n");CHKERRQ(ierr);
754fe7bbcaSHong Zhang       }
7679673f7bSHong Zhang 
77db31f6deSJed Brown     } else if (format == PETSC_VIEWER_DEFAULT) {
78db31f6deSJed Brown       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
797a583510SJack Poulson       elem::Print( *a->emat, "Elemental matrix (cyclic ordering)" );
80db31f6deSJed Brown       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
81834d3fecSHong Zhang       if (A->factortype == MAT_FACTOR_NONE){
8261119200SXuan Zhou         Mat Adense;
83ce94432eSBarry Smith         ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
8461119200SXuan Zhou         ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr);
8561119200SXuan Zhou         ierr = MatView(Adense,viewer);CHKERRQ(ierr);
8661119200SXuan Zhou         ierr = MatDestroy(&Adense);CHKERRQ(ierr);
87834d3fecSHong Zhang       }
88ce94432eSBarry Smith     } else SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Format");
89d2daa67eSHong Zhang   } else {
905cb544a0SHong Zhang     /* convert to dense format and call MatView() */
9161119200SXuan Zhou     Mat Adense;
92ce94432eSBarry Smith     ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
9361119200SXuan Zhou     ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr);
9461119200SXuan Zhou     ierr = MatView(Adense,viewer);CHKERRQ(ierr);
9561119200SXuan Zhou     ierr = MatDestroy(&Adense);CHKERRQ(ierr);
96d2daa67eSHong Zhang   }
97db31f6deSJed Brown   PetscFunctionReturn(0);
98db31f6deSJed Brown }
99db31f6deSJed Brown 
100db31f6deSJed Brown #undef __FUNCT__
101180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental"
10215767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info)
103180a43e4SHong Zhang {
10415767789SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
10515767789SHong Zhang   PetscMPIInt    rank;
10615767789SHong Zhang 
107180a43e4SHong Zhang   PetscFunctionBegin;
108ce94432eSBarry Smith   MPI_Comm_rank(PetscObjectComm((PetscObject)A),&rank);
10915767789SHong Zhang 
11015767789SHong Zhang   /* if (!rank) printf("          .........MatGetInfo_Elemental ...\n"); */
1115cb544a0SHong Zhang   info->block_size     = 1.0;
11215767789SHong Zhang 
11315767789SHong Zhang   if (flag == MAT_LOCAL) {
11415767789SHong Zhang     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
11515767789SHong Zhang     info->nz_used        = info->nz_allocated;
11615767789SHong Zhang   } else if (flag == MAT_GLOBAL_MAX) {
117ce94432eSBarry Smith     //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr);
11815767789SHong Zhang     /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */
11915767789SHong Zhang     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet");
12015767789SHong Zhang   } else if (flag == MAT_GLOBAL_SUM) {
12115767789SHong Zhang     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet");
12215767789SHong Zhang     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
12315767789SHong Zhang     info->nz_used        = info->nz_allocated; /* assume Elemental does accurate allocation */
124ce94432eSBarry Smith     //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,PetscObjectComm((PetscObject)A));CHKERRQ(ierr);
12515767789SHong Zhang     //PetscPrintf(PETSC_COMM_SELF,"    ... [%d] locally allocated %g\n",rank,info->nz_allocated);
12615767789SHong Zhang   }
12715767789SHong Zhang 
12815767789SHong Zhang   info->nz_unneeded       = 0.0;
12915767789SHong Zhang   info->assemblies        = (double)A->num_ass;
13015767789SHong Zhang   info->mallocs           = 0;
13115767789SHong Zhang   info->memory            = ((PetscObject)A)->mem;
13215767789SHong Zhang   info->fill_ratio_given  = 0; /* determined by Elemental */
13315767789SHong Zhang   info->fill_ratio_needed = 0;
13415767789SHong Zhang   info->factor_mallocs    = 0;
135180a43e4SHong Zhang   PetscFunctionReturn(0);
136180a43e4SHong Zhang }
137180a43e4SHong Zhang 
138180a43e4SHong Zhang #undef __FUNCT__
139db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental"
140e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode)
141db31f6deSJed Brown {
142db31f6deSJed Brown   PetscErrorCode ierr;
143db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
144db31f6deSJed Brown   PetscMPIInt    rank;
145db31f6deSJed Brown   PetscInt       i,j,rrank,ridx,crank,cidx;
146db31f6deSJed Brown 
147db31f6deSJed Brown   PetscFunctionBegin;
148ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)A),&rank);CHKERRQ(ierr);
149db31f6deSJed Brown 
150db31f6deSJed Brown   const elem::Grid &grid = a->emat->Grid();
151db31f6deSJed Brown   for (i=0; i<nr; i++) {
152db31f6deSJed Brown     PetscInt erow,ecol,elrow,elcol;
153db31f6deSJed Brown     if (rows[i] < 0) continue;
154db31f6deSJed Brown     P2RO(A,0,rows[i],&rrank,&ridx);
155db31f6deSJed Brown     RO2E(A,0,rrank,ridx,&erow);
156ce94432eSBarry Smith     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation");
157db31f6deSJed Brown     for (j=0; j<nc; j++) {
158db31f6deSJed Brown       if (cols[j] < 0) continue;
159db31f6deSJed Brown       P2RO(A,1,cols[j],&crank,&cidx);
160db31f6deSJed Brown       RO2E(A,1,crank,cidx,&ecol);
161ce94432eSBarry Smith       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation");
162aae2c449SHong Zhang       if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */
163aae2c449SHong Zhang         if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported");
164aae2c449SHong Zhang         /* 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); */
165e6dea9dbSXuan Zhou         a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]);
166aae2c449SHong Zhang         a->interface->Axpy(1.0,*(a->esubmat),erow,ecol);
167aae2c449SHong Zhang         continue;
168ed36708cSHong Zhang       }
169db31f6deSJed Brown       elrow = erow / grid.MCSize();
170db31f6deSJed Brown       elcol = ecol / grid.MRSize();
171db31f6deSJed Brown       switch (imode) {
172e6dea9dbSXuan Zhou       case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break;
173e6dea9dbSXuan Zhou       case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break;
174ce94432eSBarry Smith       default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode);
175db31f6deSJed Brown       }
176db31f6deSJed Brown     }
177db31f6deSJed Brown   }
178db31f6deSJed Brown   PetscFunctionReturn(0);
179db31f6deSJed Brown }
180db31f6deSJed Brown 
181db31f6deSJed Brown #undef __FUNCT__
182db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental"
183db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y)
184db31f6deSJed Brown {
185db31f6deSJed Brown   Mat_Elemental         *a = (Mat_Elemental*)A->data;
186db31f6deSJed Brown   PetscErrorCode        ierr;
187e6dea9dbSXuan Zhou   const PetscElemScalar *x;
188e6dea9dbSXuan Zhou   PetscElemScalar       *y;
189df311e6cSXuan Zhou   PetscElemScalar       one = 1,zero = 0;
190db31f6deSJed Brown 
191db31f6deSJed Brown   PetscFunctionBegin;
192e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
193e6dea9dbSXuan Zhou   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
194db31f6deSJed Brown   { /* Scoping so that constructor is called before pointer is returned */
1950c18141cSBarry Smith     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ye;
1960c18141cSBarry Smith     xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n);
1970c18141cSBarry Smith     ye.Attach(A->rmap->N,1,*a->grid,0,0,y,A->rmap->n);
198db31f6deSJed Brown     elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye);
199db31f6deSJed Brown   }
200e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
201e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
202db31f6deSJed Brown   PetscFunctionReturn(0);
203db31f6deSJed Brown }
204db31f6deSJed Brown 
205db31f6deSJed Brown #undef __FUNCT__
2069426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental"
2079426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y)
2089426833fSXuan Zhou {
2099426833fSXuan Zhou   Mat_Elemental         *a = (Mat_Elemental*)A->data;
2109426833fSXuan Zhou   PetscErrorCode        ierr;
211df311e6cSXuan Zhou   const PetscElemScalar *x;
212df311e6cSXuan Zhou   PetscElemScalar       *y;
213e6dea9dbSXuan Zhou   PetscElemScalar       one = 1,zero = 0;
2149426833fSXuan Zhou 
2159426833fSXuan Zhou   PetscFunctionBegin;
216e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
217e6dea9dbSXuan Zhou   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
2189426833fSXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
2190c18141cSBarry Smith     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ye;
2200c18141cSBarry Smith     xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n);
2210c18141cSBarry Smith     ye.Attach(A->cmap->N,1,*a->grid,0,0,y,A->cmap->n);
2229426833fSXuan Zhou     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye);
2239426833fSXuan Zhou   }
224e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
225e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
2269426833fSXuan Zhou   PetscFunctionReturn(0);
2279426833fSXuan Zhou }
2289426833fSXuan Zhou 
2299426833fSXuan Zhou #undef __FUNCT__
230db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental"
231db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
232db31f6deSJed Brown {
233db31f6deSJed Brown   Mat_Elemental         *a = (Mat_Elemental*)A->data;
234db31f6deSJed Brown   PetscErrorCode        ierr;
235df311e6cSXuan Zhou   const PetscElemScalar *x;
236df311e6cSXuan Zhou   PetscElemScalar       *z;
237e6dea9dbSXuan Zhou   PetscElemScalar       one = 1;
238db31f6deSJed Brown 
239db31f6deSJed Brown   PetscFunctionBegin;
240db31f6deSJed Brown   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
241e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
242e6dea9dbSXuan Zhou   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
243db31f6deSJed Brown   { /* Scoping so that constructor is called before pointer is returned */
2440c18141cSBarry Smith     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ze;
2450c18141cSBarry Smith     xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n);
2460c18141cSBarry Smith     ze.Attach(A->rmap->N,1,*a->grid,0,0,z,A->rmap->n);
247db31f6deSJed Brown     elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze);
248db31f6deSJed Brown   }
249e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
250e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
251db31f6deSJed Brown   PetscFunctionReturn(0);
252db31f6deSJed Brown }
253db31f6deSJed Brown 
254db31f6deSJed Brown #undef __FUNCT__
255e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental"
256e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
257e883f9d5SXuan Zhou {
258e883f9d5SXuan Zhou   Mat_Elemental         *a = (Mat_Elemental*)A->data;
259e883f9d5SXuan Zhou   PetscErrorCode        ierr;
260df311e6cSXuan Zhou   const PetscElemScalar *x;
261df311e6cSXuan Zhou   PetscElemScalar       *z;
262e6dea9dbSXuan Zhou   PetscElemScalar       one = 1;
263e883f9d5SXuan Zhou 
264e883f9d5SXuan Zhou   PetscFunctionBegin;
265e883f9d5SXuan Zhou   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
266e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
267e6dea9dbSXuan Zhou   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
268e883f9d5SXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
2690c18141cSBarry Smith     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ze;
2700c18141cSBarry Smith     xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n);
2710c18141cSBarry Smith     ze.Attach(A->cmap->N,1,*a->grid,0,0,z,A->cmap->n);
272e883f9d5SXuan Zhou     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze);
273e883f9d5SXuan Zhou   }
274e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
275e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
276e883f9d5SXuan Zhou   PetscFunctionReturn(0);
277e883f9d5SXuan Zhou }
278e883f9d5SXuan Zhou 
279e883f9d5SXuan Zhou #undef __FUNCT__
2809a9e8502SHong Zhang #define __FUNCT__ "MatMatMultNumeric_Elemental"
2819a9e8502SHong Zhang static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C)
282c1d1b975SXuan Zhou {
283c1d1b975SXuan Zhou   Mat_Elemental    *a = (Mat_Elemental*)A->data;
284c1d1b975SXuan Zhou   Mat_Elemental    *b = (Mat_Elemental*)B->data;
2859a9e8502SHong Zhang   Mat_Elemental    *c = (Mat_Elemental*)C->data;
286e6dea9dbSXuan Zhou   PetscElemScalar  one = 1,zero = 0;
287c1d1b975SXuan Zhou 
288c1d1b975SXuan Zhou   PetscFunctionBegin;
289aae2c449SHong Zhang   { /* Scoping so that constructor is called before pointer is returned */
290c1d1b975SXuan Zhou     elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat);
291aae2c449SHong Zhang   }
2929a9e8502SHong Zhang   C->assembled = PETSC_TRUE;
2939a9e8502SHong Zhang   PetscFunctionReturn(0);
2949a9e8502SHong Zhang }
2959a9e8502SHong Zhang 
2969a9e8502SHong Zhang #undef __FUNCT__
2979a9e8502SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_Elemental"
2989a9e8502SHong Zhang static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
2999a9e8502SHong Zhang {
3009a9e8502SHong Zhang   PetscErrorCode ierr;
3019a9e8502SHong Zhang   Mat            Ce;
302ce94432eSBarry Smith   MPI_Comm       comm;
3039a9e8502SHong Zhang 
3049a9e8502SHong Zhang   PetscFunctionBegin;
305ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
3069a9e8502SHong Zhang   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
3079a9e8502SHong Zhang   ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
3089a9e8502SHong Zhang   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
3099a9e8502SHong Zhang   ierr = MatSetUp(Ce);CHKERRQ(ierr);
3109a9e8502SHong Zhang   *C = Ce;
3119a9e8502SHong Zhang   PetscFunctionReturn(0);
3129a9e8502SHong Zhang }
3139a9e8502SHong Zhang 
3149a9e8502SHong Zhang #undef __FUNCT__
3159a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental"
3169a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
3179a9e8502SHong Zhang {
3189a9e8502SHong Zhang   PetscErrorCode ierr;
3199a9e8502SHong Zhang 
3209a9e8502SHong Zhang   PetscFunctionBegin;
3219a9e8502SHong Zhang   if (scall == MAT_INITIAL_MATRIX){
3223ff4c91cSHong Zhang     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
3239a9e8502SHong Zhang     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
3243ff4c91cSHong Zhang     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
3259a9e8502SHong Zhang   }
3263ff4c91cSHong Zhang   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
3279a9e8502SHong Zhang   ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
3283ff4c91cSHong Zhang   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
329c1d1b975SXuan Zhou   PetscFunctionReturn(0);
330c1d1b975SXuan Zhou }
331c1d1b975SXuan Zhou 
332c1d1b975SXuan Zhou #undef __FUNCT__
333df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental"
334df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C)
335df311e6cSXuan Zhou {
336df311e6cSXuan Zhou   Mat_Elemental      *a = (Mat_Elemental*)A->data;
337df311e6cSXuan Zhou   Mat_Elemental      *b = (Mat_Elemental*)B->data;
338df311e6cSXuan Zhou   Mat_Elemental      *c = (Mat_Elemental*)C->data;
339e6dea9dbSXuan Zhou   PetscElemScalar    one = 1,zero = 0;
340df311e6cSXuan Zhou 
341df311e6cSXuan Zhou   PetscFunctionBegin;
342df311e6cSXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
343df311e6cSXuan Zhou     elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat);
344df311e6cSXuan Zhou   }
345df311e6cSXuan Zhou   C->assembled = PETSC_TRUE;
346df311e6cSXuan Zhou   PetscFunctionReturn(0);
347df311e6cSXuan Zhou }
348df311e6cSXuan Zhou 
349df311e6cSXuan Zhou #undef __FUNCT__
350df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental"
351df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
352df311e6cSXuan Zhou {
353df311e6cSXuan Zhou   PetscErrorCode ierr;
354df311e6cSXuan Zhou   Mat            Ce;
355ce94432eSBarry Smith   MPI_Comm       comm;
356df311e6cSXuan Zhou 
357df311e6cSXuan Zhou   PetscFunctionBegin;
358ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
359df311e6cSXuan Zhou   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
360df311e6cSXuan Zhou   ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
361df311e6cSXuan Zhou   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
362df311e6cSXuan Zhou   ierr = MatSetUp(Ce);CHKERRQ(ierr);
363df311e6cSXuan Zhou   *C = Ce;
364df311e6cSXuan Zhou   PetscFunctionReturn(0);
365df311e6cSXuan Zhou }
366df311e6cSXuan Zhou 
367df311e6cSXuan Zhou #undef __FUNCT__
368df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental"
369df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
370df311e6cSXuan Zhou {
371df311e6cSXuan Zhou   PetscErrorCode ierr;
372df311e6cSXuan Zhou 
373df311e6cSXuan Zhou   PetscFunctionBegin;
374df311e6cSXuan Zhou   if (scall == MAT_INITIAL_MATRIX){
375df311e6cSXuan Zhou     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
376df311e6cSXuan Zhou     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
377df311e6cSXuan Zhou     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
378df311e6cSXuan Zhou   }
379df311e6cSXuan Zhou   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
380df311e6cSXuan Zhou   ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
381df311e6cSXuan Zhou   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
382df311e6cSXuan Zhou   PetscFunctionReturn(0);
383df311e6cSXuan Zhou }
384df311e6cSXuan Zhou 
385df311e6cSXuan Zhou #undef __FUNCT__
38661119200SXuan Zhou #define __FUNCT__ "MatGetDiagonal_Elemental"
387a9d89745SXuan Zhou static PetscErrorCode MatGetDiagonal_Elemental(Mat A,Vec D)
38861119200SXuan Zhou {
389a9d89745SXuan Zhou   PetscInt        i,nrows,ncols,nD,rrank,ridx,crank,cidx;
390a9d89745SXuan Zhou   Mat_Elemental   *a = (Mat_Elemental*)A->data;
39161119200SXuan Zhou   PetscErrorCode  ierr;
392a9d89745SXuan Zhou   PetscElemScalar v;
393ce94432eSBarry Smith   MPI_Comm        comm;
39461119200SXuan Zhou 
39561119200SXuan Zhou   PetscFunctionBegin;
396ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
397a9d89745SXuan Zhou   ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr);
398a9d89745SXuan Zhou   nD = nrows>ncols ? ncols : nrows;
399a9d89745SXuan Zhou   for (i=0; i<nD; i++) {
400a9d89745SXuan Zhou     PetscInt erow,ecol;
401a9d89745SXuan Zhou     P2RO(A,0,i,&rrank,&ridx);
402a9d89745SXuan Zhou     RO2E(A,0,rrank,ridx,&erow);
403a9d89745SXuan Zhou     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
404a9d89745SXuan Zhou     P2RO(A,1,i,&crank,&cidx);
405a9d89745SXuan Zhou     RO2E(A,1,crank,cidx,&ecol);
406a9d89745SXuan Zhou     if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
407a9d89745SXuan Zhou     v = a->emat->Get(erow,ecol);
4087c8b904fSSatish Balay     ierr = VecSetValues(D,1,&i,(PetscScalar*)&v,INSERT_VALUES);CHKERRQ(ierr);
409ade3cc5eSXuan Zhou   }
410a9d89745SXuan Zhou   ierr = VecAssemblyBegin(D);CHKERRQ(ierr);
411a9d89745SXuan Zhou   ierr = VecAssemblyEnd(D);CHKERRQ(ierr);
41261119200SXuan Zhou   PetscFunctionReturn(0);
41361119200SXuan Zhou }
41461119200SXuan Zhou 
41561119200SXuan Zhou #undef __FUNCT__
416ade3cc5eSXuan Zhou #define __FUNCT__ "MatDiagonalScale_Elemental"
417ade3cc5eSXuan Zhou static PetscErrorCode MatDiagonalScale_Elemental(Mat X,Vec L,Vec R)
418ade3cc5eSXuan Zhou {
419ade3cc5eSXuan Zhou   Mat_Elemental         *x = (Mat_Elemental*)X->data;
420ade3cc5eSXuan Zhou   const PetscElemScalar *d;
421ade3cc5eSXuan Zhou   PetscErrorCode        ierr;
422ade3cc5eSXuan Zhou 
423ade3cc5eSXuan Zhou   PetscFunctionBegin;
4249065cd98SJed Brown   if (R) {
425ade3cc5eSXuan Zhou     ierr = VecGetArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr);
4260c18141cSBarry Smith     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> de;
4270c18141cSBarry Smith     de.LockedAttach(X->cmap->N,1,*x->grid,0,0,d,X->cmap->n);
428ade3cc5eSXuan Zhou     elem::DiagonalScale(elem::RIGHT,elem::NORMAL,de,*x->emat);
429ade3cc5eSXuan Zhou     ierr = VecRestoreArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr);
4309065cd98SJed Brown   }
4319065cd98SJed Brown   if (L) {
432ade3cc5eSXuan Zhou     ierr = VecGetArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr);
4330c18141cSBarry Smith     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> de;
4340c18141cSBarry Smith     de.LockedAttach(X->rmap->N,1,*x->grid,0,0,d,X->rmap->n);
435ade3cc5eSXuan Zhou     elem::DiagonalScale(elem::LEFT,elem::NORMAL,de,*x->emat);
436ade3cc5eSXuan Zhou     ierr = VecRestoreArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr);
437ade3cc5eSXuan Zhou   }
438ade3cc5eSXuan Zhou   PetscFunctionReturn(0);
439ade3cc5eSXuan Zhou }
440ade3cc5eSXuan Zhou 
441ade3cc5eSXuan Zhou #undef __FUNCT__
4424ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental"
443e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a)
44465b78793SXuan Zhou {
44565b78793SXuan Zhou   Mat_Elemental  *x = (Mat_Elemental*)X->data;
44665b78793SXuan Zhou 
44765b78793SXuan Zhou   PetscFunctionBegin;
4480c18141cSBarry Smith   elem::Scale((PetscElemScalar)a,*x->emat);
44965b78793SXuan Zhou   PetscFunctionReturn(0);
45065b78793SXuan Zhou }
45165b78793SXuan Zhou 
45265b78793SXuan Zhou #undef __FUNCT__
4534ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental"
454e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str)
455e09a3074SHong Zhang {
456e09a3074SHong Zhang   Mat_Elemental  *x = (Mat_Elemental*)X->data;
457e09a3074SHong Zhang   Mat_Elemental  *y = (Mat_Elemental*)Y->data;
45868446cf8SJed Brown   PetscErrorCode ierr;
459e09a3074SHong Zhang 
460e09a3074SHong Zhang   PetscFunctionBegin;
461e6dea9dbSXuan Zhou   elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat);
46221f6c9c4SJed Brown   ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
463e09a3074SHong Zhang   PetscFunctionReturn(0);
464e09a3074SHong Zhang }
465e09a3074SHong Zhang 
466ae844d54SHong Zhang #undef __FUNCT__
467d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental"
468d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str)
469d6223691SXuan Zhou {
470d6223691SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
471d6223691SXuan Zhou   Mat_Elemental *b=(Mat_Elemental*)B->data;
472d6223691SXuan Zhou 
473d6223691SXuan Zhou   PetscFunctionBegin;
474d6223691SXuan Zhou   elem::Copy(*a->emat,*b->emat);
475d6223691SXuan Zhou   PetscFunctionReturn(0);
476d6223691SXuan Zhou }
477d6223691SXuan Zhou 
478d6223691SXuan Zhou #undef __FUNCT__
479df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental"
480df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B)
481df311e6cSXuan Zhou {
482df311e6cSXuan Zhou   Mat            Be;
483ce94432eSBarry Smith   MPI_Comm       comm;
484df311e6cSXuan Zhou   Mat_Elemental  *a=(Mat_Elemental*)A->data;
485df311e6cSXuan Zhou   PetscErrorCode ierr;
486df311e6cSXuan Zhou 
487df311e6cSXuan Zhou   PetscFunctionBegin;
488ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
489df311e6cSXuan Zhou   ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
490df311e6cSXuan Zhou   ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
491df311e6cSXuan Zhou   ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
492df311e6cSXuan Zhou   ierr = MatSetUp(Be);CHKERRQ(ierr);
493df311e6cSXuan Zhou   *B = Be;
494df311e6cSXuan Zhou   if (op == MAT_COPY_VALUES) {
495df311e6cSXuan Zhou     Mat_Elemental *b=(Mat_Elemental*)Be->data;
496df311e6cSXuan Zhou     elem::Copy(*a->emat,*b->emat);
497df311e6cSXuan Zhou   }
498df311e6cSXuan Zhou   Be->assembled = PETSC_TRUE;
499df311e6cSXuan Zhou   PetscFunctionReturn(0);
500df311e6cSXuan Zhou }
501df311e6cSXuan Zhou 
502df311e6cSXuan Zhou #undef __FUNCT__
503d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental"
504d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
505d6223691SXuan Zhou {
506*ec8cb81fSBarry Smith   Mat            Be = *B;
5075262d616SXuan Zhou   PetscErrorCode ierr;
508ce94432eSBarry Smith   MPI_Comm       comm;
5094fe7bbcaSHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
510d6223691SXuan Zhou 
511d6223691SXuan Zhou   PetscFunctionBegin;
512ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
5135cb544a0SHong Zhang   /* Only out-of-place supported */
5145262d616SXuan Zhou   if (reuse == MAT_INITIAL_MATRIX){
5155262d616SXuan Zhou     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
5165262d616SXuan Zhou     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
5175262d616SXuan Zhou     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
5185262d616SXuan Zhou     ierr = MatSetUp(Be);CHKERRQ(ierr);
5195262d616SXuan Zhou     *B = Be;
5205262d616SXuan Zhou   }
5214fe7bbcaSHong Zhang   b = (Mat_Elemental*)Be->data;
5225262d616SXuan Zhou   elem::Transpose(*a->emat,*b->emat);
5235262d616SXuan Zhou   Be->assembled = PETSC_TRUE;
524d6223691SXuan Zhou   PetscFunctionReturn(0);
525d6223691SXuan Zhou }
526d6223691SXuan Zhou 
527d6223691SXuan Zhou #undef __FUNCT__
528dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental"
529dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A)
530dfcb0403SXuan Zhou {
531dfcb0403SXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
532dfcb0403SXuan Zhou 
533dfcb0403SXuan Zhou   PetscFunctionBegin;
534dfcb0403SXuan Zhou   elem::Conjugate(*a->emat);
535dfcb0403SXuan Zhou   PetscFunctionReturn(0);
536dfcb0403SXuan Zhou }
537dfcb0403SXuan Zhou 
538dfcb0403SXuan Zhou #undef __FUNCT__
5394a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental"
5404a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
5414a29722dSXuan Zhou {
542*ec8cb81fSBarry Smith   Mat            Be = *B;
5434a29722dSXuan Zhou   PetscErrorCode ierr;
544ce94432eSBarry Smith   MPI_Comm       comm;
5454a29722dSXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
5464a29722dSXuan Zhou 
5474a29722dSXuan Zhou   PetscFunctionBegin;
548ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
5495cb544a0SHong Zhang   /* Only out-of-place supported */
5504a29722dSXuan Zhou   if (reuse == MAT_INITIAL_MATRIX){
5514a29722dSXuan Zhou     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
5524a29722dSXuan Zhou     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
5534a29722dSXuan Zhou     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
5544a29722dSXuan Zhou     ierr = MatSetUp(Be);CHKERRQ(ierr);
5554a29722dSXuan Zhou     *B = Be;
5564a29722dSXuan Zhou   }
5574a29722dSXuan Zhou   b = (Mat_Elemental*)Be->data;
5584a29722dSXuan Zhou   elem::Adjoint(*a->emat,*b->emat);
5594a29722dSXuan Zhou   Be->assembled = PETSC_TRUE;
5604a29722dSXuan Zhou   PetscFunctionReturn(0);
5614a29722dSXuan Zhou }
5624a29722dSXuan Zhou 
5634a29722dSXuan Zhou #undef __FUNCT__
5641f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental"
5651f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X)
5661f881ff8SXuan Zhou {
5671f881ff8SXuan Zhou   Mat_Elemental     *a = (Mat_Elemental*)A->data;
5681f881ff8SXuan Zhou   PetscErrorCode    ierr;
569df311e6cSXuan Zhou   PetscElemScalar   *x;
5701f881ff8SXuan Zhou 
5711f881ff8SXuan Zhou   PetscFunctionBegin;
57245cf121fSXuan Zhou   ierr = VecCopy(B,X);CHKERRQ(ierr);
573e6dea9dbSXuan Zhou   ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
5740c18141cSBarry Smith   elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe;
5750c18141cSBarry Smith   xe.Attach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n);
5760c18141cSBarry Smith   elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer(xe);
577fc54b460SXuan Zhou   switch (A->factortype) {
578fc54b460SXuan Zhou   case MAT_FACTOR_LU:
5791f881ff8SXuan Zhou     if ((*a->pivot).AllocatedMemory()) {
580324645c7SJack Poulson       elem::lu::SolveAfter(elem::NORMAL,*a->emat,*a->pivot,xer);
58145cf121fSXuan Zhou       elem::Copy(xer,xe);
582fc54b460SXuan Zhou     } else {
583324645c7SJack Poulson       elem::lu::SolveAfter(elem::NORMAL,*a->emat,xer);
58445cf121fSXuan Zhou       elem::Copy(xer,xe);
5851f881ff8SXuan Zhou     }
586fc54b460SXuan Zhou     break;
587fc54b460SXuan Zhou   case MAT_FACTOR_CHOLESKY:
588324645c7SJack Poulson     elem::cholesky::SolveAfter(elem::UPPER,elem::NORMAL,*a->emat,xer);
589fc54b460SXuan Zhou     elem::Copy(xer,xe);
590fc54b460SXuan Zhou     break;
591fc54b460SXuan Zhou   default:
5924fe7bbcaSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
593fc54b460SXuan Zhou     break;
5941f881ff8SXuan Zhou   }
595e6dea9dbSXuan Zhou   ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
5961f881ff8SXuan Zhou   PetscFunctionReturn(0);
5971f881ff8SXuan Zhou }
5981f881ff8SXuan Zhou 
5991f881ff8SXuan Zhou #undef __FUNCT__
600df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental"
601df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X)
602df311e6cSXuan Zhou {
603df311e6cSXuan Zhou   PetscErrorCode    ierr;
604df311e6cSXuan Zhou 
605df311e6cSXuan Zhou   PetscFunctionBegin;
606df311e6cSXuan Zhou   ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr);
6073d7f40dbSXuan Zhou   ierr = VecAXPY(X,1,Y);CHKERRQ(ierr);
608df311e6cSXuan Zhou   PetscFunctionReturn(0);
609df311e6cSXuan Zhou }
610df311e6cSXuan Zhou 
611df311e6cSXuan Zhou #undef __FUNCT__
612ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental"
613ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X)
614ae844d54SHong Zhang {
6151f0e42cfSHong Zhang   Mat_Elemental *a=(Mat_Elemental*)A->data;
616d6223691SXuan Zhou   Mat_Elemental *b=(Mat_Elemental*)B->data;
6171f0e42cfSHong Zhang   Mat_Elemental *x=(Mat_Elemental*)X->data;
6181f0e42cfSHong Zhang 
619ae844d54SHong Zhang   PetscFunctionBegin;
620d6223691SXuan Zhou   elem::Copy(*b->emat,*x->emat);
621fc54b460SXuan Zhou   switch (A->factortype) {
622fc54b460SXuan Zhou   case MAT_FACTOR_LU:
623d6223691SXuan Zhou     if ((*a->pivot).AllocatedMemory()) {
624324645c7SJack Poulson       elem::lu::SolveAfter(elem::NORMAL,*a->emat,*a->pivot,*x->emat);
625fc54b460SXuan Zhou     } else {
626324645c7SJack Poulson       elem::lu::SolveAfter(elem::NORMAL,*a->emat,*x->emat);
627d6223691SXuan Zhou     }
628fc54b460SXuan Zhou     break;
629fc54b460SXuan Zhou   case MAT_FACTOR_CHOLESKY:
630324645c7SJack Poulson     elem::cholesky::SolveAfter(elem::UPPER,elem::NORMAL,*a->emat,*x->emat);
631fc54b460SXuan Zhou     break;
632fc54b460SXuan Zhou   default:
6334fe7bbcaSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
634fc54b460SXuan Zhou     break;
635fc54b460SXuan Zhou   }
636ae844d54SHong Zhang   PetscFunctionReturn(0);
637ae844d54SHong Zhang }
638ae844d54SHong Zhang 
639ae844d54SHong Zhang #undef __FUNCT__
640ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental"
641ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info)
642ae844d54SHong Zhang {
6437c920d81SXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
6447c920d81SXuan Zhou 
645ae844d54SHong Zhang   PetscFunctionBegin;
646d6223691SXuan Zhou   if (info->dtcol){
6477c920d81SXuan Zhou     elem::LU(*a->emat,*a->pivot);
6481293973dSHong Zhang   } else {
649d6223691SXuan Zhou     elem::LU(*a->emat);
650d6223691SXuan Zhou   }
6511293973dSHong Zhang   A->factortype = MAT_FACTOR_LU;
652834d3fecSHong Zhang   A->assembled  = PETSC_TRUE;
653ae844d54SHong Zhang   PetscFunctionReturn(0);
654ae844d54SHong Zhang }
655ae844d54SHong Zhang 
656d7c3f9d8SHong Zhang #undef __FUNCT__
657d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental"
658d7c3f9d8SHong Zhang static PetscErrorCode  MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
659d7c3f9d8SHong Zhang {
660d7c3f9d8SHong Zhang   PetscErrorCode ierr;
661d7c3f9d8SHong Zhang 
662d7c3f9d8SHong Zhang   PetscFunctionBegin;
663d7c3f9d8SHong Zhang   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
664d7c3f9d8SHong Zhang   ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr);
665d7c3f9d8SHong Zhang   PetscFunctionReturn(0);
666d7c3f9d8SHong Zhang }
667d7c3f9d8SHong Zhang 
668d7c3f9d8SHong Zhang #undef __FUNCT__
669d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental"
670d7c3f9d8SHong Zhang static PetscErrorCode  MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
671d7c3f9d8SHong Zhang {
672d7c3f9d8SHong Zhang   PetscFunctionBegin;
673d7c3f9d8SHong Zhang   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
674d7c3f9d8SHong Zhang   PetscFunctionReturn(0);
675d7c3f9d8SHong Zhang }
676d7c3f9d8SHong Zhang 
67745cf121fSXuan Zhou #undef __FUNCT__
67845cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental"
67945cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info)
68045cf121fSXuan Zhou {
68145cf121fSXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
682df311e6cSXuan Zhou   elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d;
68345cf121fSXuan Zhou 
68445cf121fSXuan Zhou   PetscFunctionBegin;
685c9fc186eSXuan Zhou   elem::Cholesky(elem::UPPER,*a->emat);
686fc54b460SXuan Zhou   A->factortype = MAT_FACTOR_CHOLESKY;
687834d3fecSHong Zhang   A->assembled  = PETSC_TRUE;
68845cf121fSXuan Zhou   PetscFunctionReturn(0);
68945cf121fSXuan Zhou }
69045cf121fSXuan Zhou 
69179673f7bSHong Zhang #undef __FUNCT__
69279673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental"
69379673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
69479673f7bSHong Zhang {
695cb76c1d8SXuan Zhou   PetscErrorCode ierr;
696cb76c1d8SXuan Zhou 
697cb76c1d8SXuan Zhou   PetscFunctionBegin;
698cb76c1d8SXuan Zhou   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
699cb76c1d8SXuan Zhou   ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr);
700cb76c1d8SXuan Zhou   PetscFunctionReturn(0);
70179673f7bSHong Zhang }
702cb76c1d8SXuan Zhou 
70379673f7bSHong Zhang #undef __FUNCT__
70479673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental"
70579673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info)
70679673f7bSHong Zhang {
70779673f7bSHong Zhang   PetscFunctionBegin;
70879673f7bSHong Zhang   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
70979673f7bSHong Zhang   PetscFunctionReturn(0);
71079673f7bSHong Zhang }
71179673f7bSHong Zhang 
7121293973dSHong Zhang #undef __FUNCT__
71315767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental"
71415767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type)
71515767789SHong Zhang {
71615767789SHong Zhang   PetscFunctionBegin;
71715767789SHong Zhang   *type = MATSOLVERELEMENTAL;
71815767789SHong Zhang   PetscFunctionReturn(0);
71915767789SHong Zhang }
72015767789SHong Zhang 
72115767789SHong Zhang #undef __FUNCT__
72215767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental"
72315767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F)
7241293973dSHong Zhang {
7251293973dSHong Zhang   Mat            B;
7261293973dSHong Zhang   PetscErrorCode ierr;
7271293973dSHong Zhang 
7281293973dSHong Zhang   PetscFunctionBegin;
7291293973dSHong Zhang   /* Create the factorization matrix */
730ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
7311293973dSHong Zhang   ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
7321293973dSHong Zhang   ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr);
7331293973dSHong Zhang   ierr = MatSetUp(B);CHKERRQ(ierr);
7341293973dSHong Zhang   B->factortype = ftype;
735bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr);
7361293973dSHong Zhang   *F            = B;
7371293973dSHong Zhang   PetscFunctionReturn(0);
7381293973dSHong Zhang }
7391293973dSHong Zhang 
7401f881ff8SXuan Zhou #undef __FUNCT__
7411f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental"
7421f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm)
7431f881ff8SXuan Zhou {
7441f881ff8SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
7451f881ff8SXuan Zhou 
7461f881ff8SXuan Zhou   PetscFunctionBegin;
7471f881ff8SXuan Zhou   switch (type){
7481f881ff8SXuan Zhou   case NORM_1:
749324645c7SJack Poulson     *nrm = elem::OneNorm(*a->emat);
7501f881ff8SXuan Zhou     break;
7511f881ff8SXuan Zhou   case NORM_FROBENIUS:
752324645c7SJack Poulson     *nrm = elem::FrobeniusNorm(*a->emat);
7531f881ff8SXuan Zhou     break;
7541f881ff8SXuan Zhou   case NORM_INFINITY:
755324645c7SJack Poulson     *nrm = elem::InfinityNorm(*a->emat);
7561f881ff8SXuan Zhou     break;
7571f881ff8SXuan Zhou   default:
7581f881ff8SXuan Zhou     printf("Error: unsupported norm type!\n");
7591f881ff8SXuan Zhou   }
7601f881ff8SXuan Zhou   PetscFunctionReturn(0);
7611f881ff8SXuan Zhou }
7621f881ff8SXuan Zhou 
7635262d616SXuan Zhou #undef __FUNCT__
7645262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental"
7655262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A)
7665262d616SXuan Zhou {
7675262d616SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
7685262d616SXuan Zhou 
7695262d616SXuan Zhou   PetscFunctionBegin;
7705262d616SXuan Zhou   elem::Zero(*a->emat);
7715262d616SXuan Zhou   PetscFunctionReturn(0);
7725262d616SXuan Zhou }
7735262d616SXuan Zhou 
774e09a3074SHong Zhang #undef __FUNCT__
775db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental"
776db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols)
777db31f6deSJed Brown {
778db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
779db31f6deSJed Brown   PetscErrorCode ierr;
780db31f6deSJed Brown   PetscInt       i,m,shift,stride,*idx;
781db31f6deSJed Brown 
782db31f6deSJed Brown   PetscFunctionBegin;
783db31f6deSJed Brown   if (rows) {
784db31f6deSJed Brown     m = a->emat->LocalHeight();
785db31f6deSJed Brown     shift = a->emat->ColShift();
786db31f6deSJed Brown     stride = a->emat->ColStride();
787785e854fSJed Brown     ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr);
788db31f6deSJed Brown     for (i=0; i<m; i++) {
789db31f6deSJed Brown       PetscInt rank,offset;
790db31f6deSJed Brown       E2RO(A,0,shift+i*stride,&rank,&offset);
791db31f6deSJed Brown       RO2P(A,0,rank,offset,&idx[i]);
792db31f6deSJed Brown     }
793db31f6deSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr);
794db31f6deSJed Brown   }
795db31f6deSJed Brown   if (cols) {
796db31f6deSJed Brown     m = a->emat->LocalWidth();
797db31f6deSJed Brown     shift = a->emat->RowShift();
798db31f6deSJed Brown     stride = a->emat->RowStride();
799785e854fSJed Brown     ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr);
800db31f6deSJed Brown     for (i=0; i<m; i++) {
801db31f6deSJed Brown       PetscInt rank,offset;
802db31f6deSJed Brown       E2RO(A,1,shift+i*stride,&rank,&offset);
803db31f6deSJed Brown       RO2P(A,1,rank,offset,&idx[i]);
804db31f6deSJed Brown     }
805db31f6deSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr);
806db31f6deSJed Brown   }
807db31f6deSJed Brown   PetscFunctionReturn(0);
808db31f6deSJed Brown }
809db31f6deSJed Brown 
810db31f6deSJed Brown #undef __FUNCT__
8112ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense"
81219fd82e9SBarry Smith static PetscErrorCode MatConvert_Elemental_Dense(Mat A,MatType newtype,MatReuse reuse,Mat *B)
813af295397SXuan Zhou {
8142ef0cf24SXuan Zhou   Mat                Bmpi;
815af295397SXuan Zhou   Mat_Elemental      *a = (Mat_Elemental*)A->data;
816ce94432eSBarry Smith   MPI_Comm           comm;
8172ef0cf24SXuan Zhou   PetscErrorCode     ierr;
8182ef0cf24SXuan Zhou   PetscInt           rrank,ridx,crank,cidx,nrows,ncols,i,j;
819df311e6cSXuan Zhou   PetscElemScalar    v;
820573b0fb4SBarry Smith   PetscBool          s1,s2,s3;
821af295397SXuan Zhou 
822af295397SXuan Zhou   PetscFunctionBegin;
823ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
824573b0fb4SBarry Smith   ierr = PetscStrcmp(newtype,MATDENSE,&s1);CHKERRQ(ierr);
825573b0fb4SBarry Smith   ierr = PetscStrcmp(newtype,MATSEQDENSE,&s2);CHKERRQ(ierr);
826573b0fb4SBarry Smith   ierr = PetscStrcmp(newtype,MATMPIDENSE,&s3);CHKERRQ(ierr);
827573b0fb4SBarry Smith   if (!s1 && !s2 && !s3) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE");
828af295397SXuan Zhou   ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr);
829af295397SXuan Zhou   ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
8302ef0cf24SXuan Zhou   ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr);
831af295397SXuan Zhou   ierr = MatSetUp(Bmpi);CHKERRQ(ierr);
8322ef0cf24SXuan Zhou   ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr);
8332ef0cf24SXuan Zhou   for (i=0; i<nrows; i++) {
8342ef0cf24SXuan Zhou     PetscInt erow,ecol;
8352ef0cf24SXuan Zhou     P2RO(A,0,i,&rrank,&ridx);
8362ef0cf24SXuan Zhou     RO2E(A,0,rrank,ridx,&erow);
8372ef0cf24SXuan Zhou     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
8382ef0cf24SXuan Zhou     for (j=0; j<ncols; j++) {
8392ef0cf24SXuan Zhou       P2RO(A,1,j,&crank,&cidx);
8402ef0cf24SXuan Zhou       RO2E(A,1,crank,cidx,&ecol);
8412ef0cf24SXuan Zhou       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
8422ef0cf24SXuan Zhou       v = a->emat->Get(erow,ecol);
843e6dea9dbSXuan Zhou       ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr);
8442ef0cf24SXuan Zhou     }
8452ef0cf24SXuan Zhou   }
846af295397SXuan Zhou   ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
847af295397SXuan Zhou   ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
848c4ad791aSXuan Zhou   if (reuse == MAT_REUSE_MATRIX) {
849c4ad791aSXuan Zhou     ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr);
850c4ad791aSXuan Zhou   } else {
851c4ad791aSXuan Zhou     *B = Bmpi;
852c4ad791aSXuan Zhou   }
853af295397SXuan Zhou   PetscFunctionReturn(0);
854af295397SXuan Zhou }
855af295397SXuan Zhou 
856af295397SXuan Zhou #undef __FUNCT__
857db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental"
858db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A)
859db31f6deSJed Brown {
860db31f6deSJed Brown   Mat_Elemental      *a = (Mat_Elemental*)A->data;
861db31f6deSJed Brown   PetscErrorCode     ierr;
8625e9f5b67SHong Zhang   Mat_Elemental_Grid *commgrid;
8635e9f5b67SHong Zhang   PetscBool          flg;
8645e9f5b67SHong Zhang   MPI_Comm           icomm;
865db31f6deSJed Brown 
866db31f6deSJed Brown   PetscFunctionBegin;
867c1ee1e62SHong Zhang   a->interface->Detach();
868aae2c449SHong Zhang   delete a->interface;
869aae2c449SHong Zhang   delete a->esubmat;
870db31f6deSJed Brown   delete a->emat;
8715e9f5b67SHong Zhang 
872ce94432eSBarry Smith   elem::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A));
8730c18141cSBarry Smith   ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr);
8745e9f5b67SHong Zhang   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
8755e9f5b67SHong Zhang   if (--commgrid->grid_refct == 0) {
8765e9f5b67SHong Zhang     delete commgrid->grid;
8775e9f5b67SHong Zhang     ierr = PetscFree(commgrid);CHKERRQ(ierr);
8785e9f5b67SHong Zhang   }
8795e9f5b67SHong Zhang   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
880bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",NULL);CHKERRQ(ierr);
881bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetFactor_petsc_C",NULL);CHKERRQ(ierr);
882bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSolverPackage_C",NULL);CHKERRQ(ierr);
883db31f6deSJed Brown   ierr = PetscFree(A->data);CHKERRQ(ierr);
884db31f6deSJed Brown   PetscFunctionReturn(0);
885db31f6deSJed Brown }
886db31f6deSJed Brown 
887db31f6deSJed Brown #undef __FUNCT__
888db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental"
889db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A)
890db31f6deSJed Brown {
891db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
892db31f6deSJed Brown   PetscErrorCode ierr;
893db31f6deSJed Brown   PetscMPIInt    rsize,csize;
894db31f6deSJed Brown 
895db31f6deSJed Brown   PetscFunctionBegin;
896db31f6deSJed Brown   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
897db31f6deSJed Brown   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
898db31f6deSJed Brown 
899efb79153SJack Poulson   a->emat->Resize(A->rmap->N,A->cmap->N);CHKERRQ(ierr);
900db31f6deSJed Brown   elem::Zero(*a->emat);
901db31f6deSJed Brown 
902db31f6deSJed Brown   ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr);
903db31f6deSJed Brown   ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr);
904ce94432eSBarry Smith   if (csize != rsize) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes");
905db31f6deSJed Brown   a->commsize = rsize;
906db31f6deSJed Brown   a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize;
907db31f6deSJed Brown   a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize;
908db31f6deSJed Brown   a->m[0]  = A->rmap->N / rsize + (a->mr[0] != rsize);
909db31f6deSJed Brown   a->m[1]  = A->cmap->N / csize + (a->mr[1] != csize);
910db31f6deSJed Brown   PetscFunctionReturn(0);
911db31f6deSJed Brown }
912db31f6deSJed Brown 
913aae2c449SHong Zhang #undef __FUNCT__
914aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental"
915aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type)
916aae2c449SHong Zhang {
917aae2c449SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
918aae2c449SHong Zhang 
919aae2c449SHong Zhang   PetscFunctionBegin;
920aae2c449SHong Zhang   a->interface->Detach();
921aae2c449SHong Zhang   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
922aae2c449SHong Zhang   PetscFunctionReturn(0);
923aae2c449SHong Zhang }
924aae2c449SHong Zhang 
925aae2c449SHong Zhang #undef __FUNCT__
926aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental"
927aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type)
928aae2c449SHong Zhang {
929aae2c449SHong Zhang   PetscFunctionBegin;
930aae2c449SHong Zhang   /* Currently does nothing */
931aae2c449SHong Zhang   PetscFunctionReturn(0);
932aae2c449SHong Zhang }
933aae2c449SHong Zhang 
93440d92e34SHong Zhang /* -------------------------------------------------------------------*/
93540d92e34SHong Zhang static struct _MatOps MatOps_Values = {
93640d92e34SHong Zhang        MatSetValues_Elemental,
93740d92e34SHong Zhang        0,
93840d92e34SHong Zhang        0,
93940d92e34SHong Zhang        MatMult_Elemental,
94040d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental,
9419426833fSXuan Zhou        MatMultTranspose_Elemental,
942e883f9d5SXuan Zhou        MatMultTransposeAdd_Elemental,
94340d92e34SHong Zhang        MatSolve_Elemental,
944df311e6cSXuan Zhou        MatSolveAdd_Elemental,
94540d92e34SHong Zhang        0, //MatSolveTranspose_Elemental,
94640d92e34SHong Zhang /*10*/ 0, //MatSolveTransposeAdd_Elemental,
94740d92e34SHong Zhang        MatLUFactor_Elemental,
94840d92e34SHong Zhang        MatCholeskyFactor_Elemental,
94940d92e34SHong Zhang        0,
95040d92e34SHong Zhang        MatTranspose_Elemental,
95140d92e34SHong Zhang /*15*/ MatGetInfo_Elemental,
95240d92e34SHong Zhang        0,
95361119200SXuan Zhou        MatGetDiagonal_Elemental,
954ade3cc5eSXuan Zhou        MatDiagonalScale_Elemental,
95540d92e34SHong Zhang        MatNorm_Elemental,
95640d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental,
95740d92e34SHong Zhang        MatAssemblyEnd_Elemental,
95840d92e34SHong Zhang        0, //MatSetOption_Elemental,
95940d92e34SHong Zhang        MatZeroEntries_Elemental,
96040d92e34SHong Zhang /*24*/ 0,
96140d92e34SHong Zhang        MatLUFactorSymbolic_Elemental,
96240d92e34SHong Zhang        MatLUFactorNumeric_Elemental,
96340d92e34SHong Zhang        MatCholeskyFactorSymbolic_Elemental,
96440d92e34SHong Zhang        MatCholeskyFactorNumeric_Elemental,
96540d92e34SHong Zhang /*29*/ MatSetUp_Elemental,
96640d92e34SHong Zhang        0,
96740d92e34SHong Zhang        0,
96840d92e34SHong Zhang        0,
96940d92e34SHong Zhang        0,
970df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental,
97140d92e34SHong Zhang        0,
97240d92e34SHong Zhang        0,
97340d92e34SHong Zhang        0,
97440d92e34SHong Zhang        0,
97540d92e34SHong Zhang /*39*/ MatAXPY_Elemental,
97640d92e34SHong Zhang        0,
97740d92e34SHong Zhang        0,
97840d92e34SHong Zhang        0,
97940d92e34SHong Zhang        MatCopy_Elemental,
98040d92e34SHong Zhang /*44*/ 0,
98140d92e34SHong Zhang        MatScale_Elemental,
98240d92e34SHong Zhang        0,
98340d92e34SHong Zhang        0,
98440d92e34SHong Zhang        0,
98540d92e34SHong Zhang /*49*/ 0,
98640d92e34SHong Zhang        0,
98740d92e34SHong Zhang        0,
98840d92e34SHong Zhang        0,
98940d92e34SHong Zhang        0,
99040d92e34SHong Zhang /*54*/ 0,
99140d92e34SHong Zhang        0,
99240d92e34SHong Zhang        0,
99340d92e34SHong Zhang        0,
99440d92e34SHong Zhang        0,
99540d92e34SHong Zhang /*59*/ 0,
99640d92e34SHong Zhang        MatDestroy_Elemental,
99740d92e34SHong Zhang        MatView_Elemental,
99840d92e34SHong Zhang        0,
99940d92e34SHong Zhang        0,
100040d92e34SHong Zhang /*64*/ 0,
100140d92e34SHong Zhang        0,
100240d92e34SHong Zhang        0,
100340d92e34SHong Zhang        0,
100440d92e34SHong Zhang        0,
100540d92e34SHong Zhang /*69*/ 0,
100640d92e34SHong Zhang        0,
10072ef0cf24SXuan Zhou        MatConvert_Elemental_Dense,
100840d92e34SHong Zhang        0,
100940d92e34SHong Zhang        0,
101040d92e34SHong Zhang /*74*/ 0,
101140d92e34SHong Zhang        0,
101240d92e34SHong Zhang        0,
101340d92e34SHong Zhang        0,
101440d92e34SHong Zhang        0,
101540d92e34SHong Zhang /*79*/ 0,
101640d92e34SHong Zhang        0,
101740d92e34SHong Zhang        0,
101840d92e34SHong Zhang        0,
101940d92e34SHong Zhang        0,
102040d92e34SHong Zhang /*84*/ 0,
102140d92e34SHong Zhang        0,
102240d92e34SHong Zhang        0,
102340d92e34SHong Zhang        0,
102440d92e34SHong Zhang        0,
102540d92e34SHong Zhang /*89*/ MatMatMult_Elemental,
102640d92e34SHong Zhang        MatMatMultSymbolic_Elemental,
102740d92e34SHong Zhang        MatMatMultNumeric_Elemental,
102840d92e34SHong Zhang        0,
102940d92e34SHong Zhang        0,
103040d92e34SHong Zhang /*94*/ 0,
1031df311e6cSXuan Zhou        MatMatTransposeMult_Elemental,
1032df311e6cSXuan Zhou        MatMatTransposeMultSymbolic_Elemental,
1033df311e6cSXuan Zhou        MatMatTransposeMultNumeric_Elemental,
103440d92e34SHong Zhang        0,
103540d92e34SHong Zhang /*99*/ 0,
103640d92e34SHong Zhang        0,
103740d92e34SHong Zhang        0,
1038dfcb0403SXuan Zhou        MatConjugate_Elemental,
103940d92e34SHong Zhang        0,
104040d92e34SHong Zhang /*104*/0,
104140d92e34SHong Zhang        0,
104240d92e34SHong Zhang        0,
104340d92e34SHong Zhang        0,
104440d92e34SHong Zhang        0,
104540d92e34SHong Zhang /*109*/MatMatSolve_Elemental,
104640d92e34SHong Zhang        0,
104740d92e34SHong Zhang        0,
104840d92e34SHong Zhang        0,
104940d92e34SHong Zhang        0,
105040d92e34SHong Zhang /*114*/0,
105140d92e34SHong Zhang        0,
105240d92e34SHong Zhang        0,
105340d92e34SHong Zhang        0,
105440d92e34SHong Zhang        0,
105540d92e34SHong Zhang /*119*/0,
10564a29722dSXuan Zhou        MatHermitianTranspose_Elemental,
105740d92e34SHong Zhang        0,
105840d92e34SHong Zhang        0,
105940d92e34SHong Zhang        0,
106040d92e34SHong Zhang /*124*/0,
106140d92e34SHong Zhang        0,
106240d92e34SHong Zhang        0,
106340d92e34SHong Zhang        0,
106440d92e34SHong Zhang        0,
106540d92e34SHong Zhang /*129*/0,
106640d92e34SHong Zhang        0,
106740d92e34SHong Zhang        0,
106840d92e34SHong Zhang        0,
106940d92e34SHong Zhang        0,
107040d92e34SHong Zhang /*134*/0,
107140d92e34SHong Zhang        0,
107240d92e34SHong Zhang        0,
107340d92e34SHong Zhang        0,
107440d92e34SHong Zhang        0
107540d92e34SHong Zhang };
107640d92e34SHong Zhang 
1077ed36708cSHong Zhang /*MC
1078ed36708cSHong Zhang    MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package
1079ed36708cSHong Zhang 
1080ed36708cSHong Zhang    Options Database Keys:
10815cc86fc1SJed Brown + -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions()
10825cc86fc1SJed Brown - -mat_elemental_grid_height - sets Grid Height for 2D cyclic ordering of internal matrix
1083ed36708cSHong Zhang 
1084ed36708cSHong Zhang   Level: beginner
1085ed36708cSHong Zhang 
10865cb544a0SHong Zhang .seealso: MATDENSE
1087ed36708cSHong Zhang M*/
10884a29722dSXuan Zhou 
1089db31f6deSJed Brown #undef __FUNCT__
1090db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental"
10918cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_Elemental(Mat A)
1092db31f6deSJed Brown {
1093db31f6deSJed Brown   Mat_Elemental      *a;
1094db31f6deSJed Brown   PetscErrorCode     ierr;
10955682a260SJack Poulson   PetscBool          flg,flg1;
10965e9f5b67SHong Zhang   Mat_Elemental_Grid *commgrid;
10975e9f5b67SHong Zhang   MPI_Comm           icomm;
10985682a260SJack Poulson   PetscInt           optv1;
1099db31f6deSJed Brown 
1100db31f6deSJed Brown   PetscFunctionBegin;
1101607a6623SBarry Smith   ierr = PetscElementalInitializePackage();CHKERRQ(ierr);
110240d92e34SHong Zhang   ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
110340d92e34SHong Zhang   A->insertmode = NOT_SET_VALUES;
1104db31f6deSJed Brown 
1105b00a9115SJed Brown   ierr = PetscNewLog(A,&a);CHKERRQ(ierr);
1106db31f6deSJed Brown   A->data = (void*)a;
1107db31f6deSJed Brown 
1108db31f6deSJed Brown   /* Set up the elemental matrix */
1109ce94432eSBarry Smith   elem::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A));
11105e9f5b67SHong Zhang 
11115e9f5b67SHong Zhang   /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */
11125e9f5b67SHong Zhang   if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) {
1113180a43e4SHong Zhang     ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0);
11145e9f5b67SHong Zhang   }
11150c18141cSBarry Smith   ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr);
11165e9f5b67SHong Zhang   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
11175e9f5b67SHong Zhang   if (!flg) {
1118b00a9115SJed Brown     ierr = PetscNewLog(A,&commgrid);CHKERRQ(ierr);
11195cb544a0SHong Zhang 
1120ce94432eSBarry Smith     ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr);
11215cb544a0SHong Zhang     /* displayed default grid sizes (CommSize,1) are set by us arbitrarily until elem::Grid() is called */
11220c18141cSBarry Smith     ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::Size(cxxcomm),&optv1,&flg1);CHKERRQ(ierr);
11235682a260SJack Poulson     if (flg1) {
11240c18141cSBarry Smith       if (elem::mpi::Size(cxxcomm) % optv1 != 0) {
11250c18141cSBarry Smith         SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Grid Height %D must evenly divide CommSize %D",optv1,(PetscInt)elem::mpi::Size(cxxcomm));
1126ed667823SXuan Zhou       }
11275682a260SJack Poulson       commgrid->grid = new elem::Grid(cxxcomm,optv1); /* use user-provided grid height */
11282ef0cf24SXuan Zhou     } else {
11292adf0be3SHong Zhang       commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */
1130ed667823SXuan Zhou     }
11315e9f5b67SHong Zhang     commgrid->grid_refct = 1;
11325e9f5b67SHong Zhang     ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr);
11335cb544a0SHong Zhang     PetscOptionsEnd();
11345e9f5b67SHong Zhang   } else {
11355e9f5b67SHong Zhang     commgrid->grid_refct++;
11365e9f5b67SHong Zhang   }
11375e9f5b67SHong Zhang   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
11385e9f5b67SHong Zhang   a->grid      = commgrid->grid;
1139df311e6cSXuan Zhou   a->emat      = new elem::DistMatrix<PetscElemScalar>(*a->grid);
1140df311e6cSXuan Zhou   a->esubmat   = new elem::Matrix<PetscElemScalar>(1,1);
1141df311e6cSXuan Zhou   a->interface = new elem::AxpyInterface<PetscElemScalar>;
11427c920d81SXuan Zhou   a->pivot     = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>;
1143db31f6deSJed Brown 
1144db31f6deSJed Brown   /* build cache for off array entries formed */
1145aae2c449SHong Zhang   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
1146bafd5131SHong Zhang 
1147bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",MatGetOwnershipIS_Elemental);CHKERRQ(ierr);
1148bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetFactor_elemental_C",MatGetFactor_elemental_elemental);CHKERRQ(ierr);
1149db31f6deSJed Brown 
1150db31f6deSJed Brown   ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr);
1151db31f6deSJed Brown   PetscFunctionReturn(0);
1152db31f6deSJed Brown }
11534a29722dSXuan Zhou 
1154