xref: /petsc/src/mat/impls/elemental/matelem.cxx (revision 2adf0be3efa174207a96fd669ae5b85ebe85da0d)
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    Input Arguments:
17db31f6deSJed Brown .  path - the dynamic library path or PETSC_NULL
18db31f6deSJed Brown 
19db31f6deSJed Brown    Level: developer
20db31f6deSJed Brown 
21db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalFinalizePackage()
22db31f6deSJed Brown @*/
23db31f6deSJed Brown PetscErrorCode PetscElementalInitializePackage(const char *path)
24db31f6deSJed Brown {
25db31f6deSJed Brown   PetscErrorCode ierr;
26db31f6deSJed Brown 
27db31f6deSJed Brown   PetscFunctionBegin;
28db31f6deSJed Brown   if (elem::Initialized()) PetscFunctionReturn(0);
29db31f6deSJed 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 */
30db31f6deSJed Brown     int zero = 0;
31db31f6deSJed Brown     char **nothing = 0;
32db31f6deSJed Brown     elem::Initialize(zero,nothing);
33db31f6deSJed Brown   }
34db31f6deSJed Brown   ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr);
35db31f6deSJed Brown   PetscFunctionReturn(0);
36db31f6deSJed Brown }
37db31f6deSJed Brown 
38db31f6deSJed Brown #undef __FUNCT__
39db31f6deSJed Brown #define __FUNCT__ "PetscElementalFinalizePackage"
40db31f6deSJed Brown /*@C
41db31f6deSJed Brown    PetscElementalFinalizePackage - Finalize Elemental package
42db31f6deSJed Brown 
43db31f6deSJed Brown    Logically Collective
44db31f6deSJed Brown 
45db31f6deSJed Brown    Level: developer
46db31f6deSJed Brown 
47db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalInitializePackage()
48db31f6deSJed Brown @*/
49db31f6deSJed Brown PetscErrorCode PetscElementalFinalizePackage(void)
50db31f6deSJed Brown {
51db31f6deSJed Brown 
52db31f6deSJed Brown   PetscFunctionBegin;
53db31f6deSJed Brown   elem::Finalize();
54db31f6deSJed Brown   PetscFunctionReturn(0);
55db31f6deSJed Brown }
56db31f6deSJed Brown 
57db31f6deSJed Brown #undef __FUNCT__
58db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental"
59db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer)
60db31f6deSJed Brown {
61db31f6deSJed Brown   PetscErrorCode ierr;
62db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
63db31f6deSJed Brown   PetscBool      iascii;
64db31f6deSJed Brown 
65db31f6deSJed Brown   PetscFunctionBegin;
66db31f6deSJed Brown   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
67db31f6deSJed Brown   if (iascii) {
68db31f6deSJed Brown     PetscViewerFormat format;
69db31f6deSJed Brown     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
70db31f6deSJed Brown     if (format == PETSC_VIEWER_ASCII_INFO) {
7179673f7bSHong Zhang       /* call elemental viewing function */
722d8adcc7SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr);
73ed667823SXuan Zhou       ierr = PetscViewerASCIIPrintf(viewer,"  allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr);
74ed667823SXuan Zhou       ierr = PetscViewerASCIIPrintf(viewer,"  grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr);
754fe7bbcaSHong Zhang       if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
7679673f7bSHong Zhang         /* call elemental viewing function */
7780622e0cSXuan Zhou         ierr = PetscPrintf(((PetscObject)viewer)->comm,"test matview_elemental 2\n");CHKERRQ(ierr);
784fe7bbcaSHong Zhang       }
7979673f7bSHong Zhang 
80db31f6deSJed Brown     } else if (format == PETSC_VIEWER_DEFAULT) {
81db31f6deSJed Brown       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
82db31f6deSJed Brown       ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr);
83db31f6deSJed Brown       a->emat->Print("Elemental matrix (cyclic ordering)");
84db31f6deSJed Brown       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
85834d3fecSHong Zhang       if (A->factortype == MAT_FACTOR_NONE){
86d2daa67eSHong Zhang         Mat Aaij;
870022473aSHong Zhang         ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
880022473aSHong Zhang         ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr);
89d2daa67eSHong Zhang         ierr = MatView(Aaij,viewer);CHKERRQ(ierr);
900022473aSHong Zhang         ierr = MatDestroy(&Aaij);CHKERRQ(ierr);
91834d3fecSHong Zhang       }
92db31f6deSJed Brown     } else SETERRQ(((PetscObject)viewer)->comm,PETSC_ERR_SUP,"Format");
93d2daa67eSHong Zhang   } else {
94d2daa67eSHong Zhang     /* convert to aij/mpidense format and call MatView() */
95d2daa67eSHong Zhang     Mat Aaij;
96d2daa67eSHong Zhang     ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
97d2daa67eSHong Zhang     ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr);
98d2daa67eSHong Zhang     ierr = MatView(Aaij,viewer);CHKERRQ(ierr);
99d2daa67eSHong Zhang     ierr = MatDestroy(&Aaij);CHKERRQ(ierr);
100d2daa67eSHong Zhang   }
101db31f6deSJed Brown   PetscFunctionReturn(0);
102db31f6deSJed Brown }
103db31f6deSJed Brown 
104db31f6deSJed Brown #undef __FUNCT__
105180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental"
10615767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info)
107180a43e4SHong Zhang {
10815767789SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
10915767789SHong Zhang   PetscMPIInt    rank;
11015767789SHong Zhang 
111180a43e4SHong Zhang   PetscFunctionBegin;
11215767789SHong Zhang   MPI_Comm_rank(((PetscObject)A)->comm,&rank);
11315767789SHong Zhang 
11415767789SHong Zhang   /* if (!rank) printf("          .........MatGetInfo_Elemental ...\n"); */
11515767789SHong Zhang   info->block_size     = 1.0; /* ? */
11615767789SHong Zhang 
11715767789SHong Zhang   if (flag == MAT_LOCAL) {
11815767789SHong Zhang     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
11915767789SHong Zhang     info->nz_used        = info->nz_allocated;
12015767789SHong Zhang   } else if (flag == MAT_GLOBAL_MAX) {
12115767789SHong Zhang     //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,((PetscObject)matin)->comm);CHKERRQ(ierr);
12215767789SHong Zhang     /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */
12315767789SHong Zhang     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet");
12415767789SHong Zhang   } else if (flag == MAT_GLOBAL_SUM) {
12515767789SHong Zhang     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet");
12615767789SHong Zhang     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
12715767789SHong Zhang     info->nz_used        = info->nz_allocated; /* assume Elemental does accurate allocation */
12815767789SHong Zhang     //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
12915767789SHong Zhang     //PetscPrintf(PETSC_COMM_SELF,"    ... [%d] locally allocated %g\n",rank,info->nz_allocated);
13015767789SHong Zhang   }
13115767789SHong Zhang 
13215767789SHong Zhang   info->nz_unneeded       = 0.0;
13315767789SHong Zhang   info->assemblies        = (double)A->num_ass;
13415767789SHong Zhang   info->mallocs           = 0;
13515767789SHong Zhang   info->memory            = ((PetscObject)A)->mem;
13615767789SHong Zhang   info->fill_ratio_given  = 0; /* determined by Elemental */
13715767789SHong Zhang   info->fill_ratio_needed = 0;
13815767789SHong Zhang   info->factor_mallocs    = 0;
139180a43e4SHong Zhang   PetscFunctionReturn(0);
140180a43e4SHong Zhang }
141180a43e4SHong Zhang 
142180a43e4SHong Zhang #undef __FUNCT__
143db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental"
144e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode)
145db31f6deSJed Brown {
146db31f6deSJed Brown   PetscErrorCode ierr;
147db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
148db31f6deSJed Brown   PetscMPIInt    rank;
149db31f6deSJed Brown   PetscInt       i,j,rrank,ridx,crank,cidx;
150db31f6deSJed Brown 
151db31f6deSJed Brown   PetscFunctionBegin;
152db31f6deSJed Brown   ierr = MPI_Comm_rank(((PetscObject)A)->comm,&rank);CHKERRQ(ierr);
153db31f6deSJed Brown 
154db31f6deSJed Brown   const elem::Grid &grid = a->emat->Grid();
155db31f6deSJed Brown   for (i=0; i<nr; i++) {
156db31f6deSJed Brown     PetscInt erow,ecol,elrow,elcol;
157db31f6deSJed Brown     if (rows[i] < 0) continue;
158db31f6deSJed Brown     P2RO(A,0,rows[i],&rrank,&ridx);
159db31f6deSJed Brown     RO2E(A,0,rrank,ridx,&erow);
160db31f6deSJed Brown     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect row translation");
161db31f6deSJed Brown     for (j=0; j<nc; j++) {
162db31f6deSJed Brown       if (cols[j] < 0) continue;
163db31f6deSJed Brown       P2RO(A,1,cols[j],&crank,&cidx);
164db31f6deSJed Brown       RO2E(A,1,crank,cidx,&ecol);
165db31f6deSJed Brown       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect col translation");
166aae2c449SHong Zhang       if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */
167aae2c449SHong Zhang         if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported");
168aae2c449SHong 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); */
169e6dea9dbSXuan Zhou         a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]);
170aae2c449SHong Zhang         a->interface->Axpy(1.0,*(a->esubmat),erow,ecol);
171aae2c449SHong Zhang         continue;
172ed36708cSHong Zhang       }
173db31f6deSJed Brown       elrow = erow / grid.MCSize();
174db31f6deSJed Brown       elcol = ecol / grid.MRSize();
175db31f6deSJed Brown       switch (imode) {
176e6dea9dbSXuan Zhou       case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break;
177e6dea9dbSXuan Zhou       case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break;
178db31f6deSJed Brown       default: SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode);
179db31f6deSJed Brown       }
180db31f6deSJed Brown     }
181db31f6deSJed Brown   }
182db31f6deSJed Brown   PetscFunctionReturn(0);
183db31f6deSJed Brown }
184db31f6deSJed Brown 
185db31f6deSJed Brown #undef __FUNCT__
186db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental"
187db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y)
188db31f6deSJed Brown {
189db31f6deSJed Brown   Mat_Elemental         *a = (Mat_Elemental*)A->data;
190db31f6deSJed Brown   PetscErrorCode        ierr;
191e6dea9dbSXuan Zhou   const PetscElemScalar *x;
192e6dea9dbSXuan Zhou   PetscElemScalar       *y;
193df311e6cSXuan Zhou   PetscElemScalar       one = 1,zero = 0;
194db31f6deSJed Brown 
195db31f6deSJed Brown   PetscFunctionBegin;
196e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
197e6dea9dbSXuan Zhou   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
198db31f6deSJed Brown   { /* Scoping so that constructor is called before pointer is returned */
199df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid);
200df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,y,A->rmap->n,*a->grid);
201db31f6deSJed Brown     elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye);
202db31f6deSJed Brown   }
203e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
204e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
205db31f6deSJed Brown   PetscFunctionReturn(0);
206db31f6deSJed Brown }
207db31f6deSJed Brown 
208db31f6deSJed Brown #undef __FUNCT__
2099426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental"
2109426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y)
2119426833fSXuan Zhou {
2129426833fSXuan Zhou   Mat_Elemental         *a = (Mat_Elemental*)A->data;
2139426833fSXuan Zhou   PetscErrorCode        ierr;
214df311e6cSXuan Zhou   const PetscElemScalar *x;
215df311e6cSXuan Zhou   PetscElemScalar       *y;
216e6dea9dbSXuan Zhou   PetscElemScalar       one = 1,zero = 0;
2179426833fSXuan Zhou 
2189426833fSXuan Zhou   PetscFunctionBegin;
219e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
220e6dea9dbSXuan Zhou   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
2219426833fSXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
222df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
223df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,y,A->cmap->n,*a->grid);
2249426833fSXuan Zhou     elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye);
2259426833fSXuan Zhou   }
226e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
227e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
2289426833fSXuan Zhou   PetscFunctionReturn(0);
2299426833fSXuan Zhou }
2309426833fSXuan Zhou 
2319426833fSXuan Zhou #undef __FUNCT__
232db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental"
233db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
234db31f6deSJed Brown {
235db31f6deSJed Brown   Mat_Elemental         *a = (Mat_Elemental*)A->data;
236db31f6deSJed Brown   PetscErrorCode        ierr;
237df311e6cSXuan Zhou   const PetscElemScalar *x;
238df311e6cSXuan Zhou   PetscElemScalar       *z;
239e6dea9dbSXuan Zhou   PetscElemScalar       one = 1;
240db31f6deSJed Brown 
241db31f6deSJed Brown   PetscFunctionBegin;
242db31f6deSJed Brown   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
243e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
244e6dea9dbSXuan Zhou   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
245db31f6deSJed Brown   { /* Scoping so that constructor is called before pointer is returned */
246df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid);
247df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,z,A->rmap->n,*a->grid);
248db31f6deSJed Brown     elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze);
249db31f6deSJed Brown   }
250e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
251e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
252db31f6deSJed Brown   PetscFunctionReturn(0);
253db31f6deSJed Brown }
254db31f6deSJed Brown 
255db31f6deSJed Brown #undef __FUNCT__
256e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental"
257e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
258e883f9d5SXuan Zhou {
259e883f9d5SXuan Zhou   Mat_Elemental         *a = (Mat_Elemental*)A->data;
260e883f9d5SXuan Zhou   PetscErrorCode        ierr;
261df311e6cSXuan Zhou   const PetscElemScalar *x;
262df311e6cSXuan Zhou   PetscElemScalar       *z;
263e6dea9dbSXuan Zhou   PetscElemScalar       one = 1;
264e883f9d5SXuan Zhou 
265e883f9d5SXuan Zhou   PetscFunctionBegin;
266e883f9d5SXuan Zhou   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
267e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
268e6dea9dbSXuan Zhou   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
269e883f9d5SXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
270df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
271df311e6cSXuan Zhou     elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,z,A->cmap->n,*a->grid);
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;
3029a9e8502SHong Zhang   MPI_Comm       comm=((PetscObject)A)->comm;
3039a9e8502SHong Zhang 
3049a9e8502SHong Zhang   PetscFunctionBegin;
3059a9e8502SHong Zhang   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
3069a9e8502SHong Zhang   ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
3079a9e8502SHong Zhang   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
3089a9e8502SHong Zhang   ierr = MatSetUp(Ce);CHKERRQ(ierr);
3099a9e8502SHong Zhang   *C = Ce;
3109a9e8502SHong Zhang   PetscFunctionReturn(0);
3119a9e8502SHong Zhang }
3129a9e8502SHong Zhang 
3139a9e8502SHong Zhang #undef __FUNCT__
3149a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental"
3159a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
3169a9e8502SHong Zhang {
3179a9e8502SHong Zhang   PetscErrorCode ierr;
3189a9e8502SHong Zhang 
3199a9e8502SHong Zhang   PetscFunctionBegin;
3209a9e8502SHong Zhang   if (scall == MAT_INITIAL_MATRIX){
3219a9e8502SHong Zhang     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
3229a9e8502SHong Zhang     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
3239a9e8502SHong Zhang     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
3249a9e8502SHong Zhang   }
3259a9e8502SHong Zhang   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
3269a9e8502SHong Zhang   ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
3279a9e8502SHong Zhang   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
328c1d1b975SXuan Zhou   PetscFunctionReturn(0);
329c1d1b975SXuan Zhou }
330c1d1b975SXuan Zhou 
331c1d1b975SXuan Zhou #undef __FUNCT__
332df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental"
333df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C)
334df311e6cSXuan Zhou {
335df311e6cSXuan Zhou   Mat_Elemental      *a = (Mat_Elemental*)A->data;
336df311e6cSXuan Zhou   Mat_Elemental      *b = (Mat_Elemental*)B->data;
337df311e6cSXuan Zhou   Mat_Elemental      *c = (Mat_Elemental*)C->data;
338e6dea9dbSXuan Zhou   PetscElemScalar    one = 1,zero = 0;
339df311e6cSXuan Zhou 
340df311e6cSXuan Zhou   PetscFunctionBegin;
341df311e6cSXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
342df311e6cSXuan Zhou     elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat);
343df311e6cSXuan Zhou   }
344df311e6cSXuan Zhou   C->assembled = PETSC_TRUE;
345df311e6cSXuan Zhou   PetscFunctionReturn(0);
346df311e6cSXuan Zhou }
347df311e6cSXuan Zhou 
348df311e6cSXuan Zhou #undef __FUNCT__
349df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental"
350df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
351df311e6cSXuan Zhou {
352df311e6cSXuan Zhou   PetscErrorCode ierr;
353df311e6cSXuan Zhou   Mat            Ce;
354df311e6cSXuan Zhou   MPI_Comm       comm=((PetscObject)A)->comm;
355df311e6cSXuan Zhou 
356df311e6cSXuan Zhou   PetscFunctionBegin;
357df311e6cSXuan Zhou   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
358df311e6cSXuan Zhou   ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
359df311e6cSXuan Zhou   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
360df311e6cSXuan Zhou   ierr = MatSetUp(Ce);CHKERRQ(ierr);
361df311e6cSXuan Zhou   *C = Ce;
362df311e6cSXuan Zhou   PetscFunctionReturn(0);
363df311e6cSXuan Zhou }
364df311e6cSXuan Zhou 
365df311e6cSXuan Zhou #undef __FUNCT__
366df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental"
367df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
368df311e6cSXuan Zhou {
369df311e6cSXuan Zhou   PetscErrorCode ierr;
370df311e6cSXuan Zhou 
371df311e6cSXuan Zhou   PetscFunctionBegin;
372df311e6cSXuan Zhou   if (scall == MAT_INITIAL_MATRIX){
373df311e6cSXuan Zhou     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
374df311e6cSXuan Zhou     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
375df311e6cSXuan Zhou     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
376df311e6cSXuan Zhou   }
377df311e6cSXuan Zhou   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
378df311e6cSXuan Zhou   ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
379df311e6cSXuan Zhou   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
380df311e6cSXuan Zhou   PetscFunctionReturn(0);
381df311e6cSXuan Zhou }
382df311e6cSXuan Zhou 
383df311e6cSXuan Zhou #undef __FUNCT__
3844ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental"
385e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a)
38665b78793SXuan Zhou {
38765b78793SXuan Zhou   Mat_Elemental  *x = (Mat_Elemental*)X->data;
38865b78793SXuan Zhou 
38965b78793SXuan Zhou   PetscFunctionBegin;
390e6dea9dbSXuan Zhou   elem::Scal((PetscElemScalar)a,*x->emat);
39165b78793SXuan Zhou   PetscFunctionReturn(0);
39265b78793SXuan Zhou }
39365b78793SXuan Zhou 
39465b78793SXuan Zhou #undef __FUNCT__
3954ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental"
396e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str)
397e09a3074SHong Zhang {
398e09a3074SHong Zhang   Mat_Elemental  *x = (Mat_Elemental*)X->data;
399e09a3074SHong Zhang   Mat_Elemental  *y = (Mat_Elemental*)Y->data;
400e09a3074SHong Zhang 
401e09a3074SHong Zhang   PetscFunctionBegin;
402e6dea9dbSXuan Zhou   elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat);
403e09a3074SHong Zhang   PetscFunctionReturn(0);
404e09a3074SHong Zhang }
405e09a3074SHong Zhang 
406ae844d54SHong Zhang #undef __FUNCT__
407d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental"
408d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str)
409d6223691SXuan Zhou {
410d6223691SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
411d6223691SXuan Zhou   Mat_Elemental *b=(Mat_Elemental*)B->data;
412d6223691SXuan Zhou 
413d6223691SXuan Zhou   PetscFunctionBegin;
414d6223691SXuan Zhou   elem::Copy(*a->emat,*b->emat);
415d6223691SXuan Zhou   PetscFunctionReturn(0);
416d6223691SXuan Zhou }
417d6223691SXuan Zhou 
418d6223691SXuan Zhou #undef __FUNCT__
419df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental"
420df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B)
421df311e6cSXuan Zhou {
422df311e6cSXuan Zhou   Mat            Be;
423df311e6cSXuan Zhou   MPI_Comm       comm=((PetscObject)A)->comm;
424df311e6cSXuan Zhou   Mat_Elemental  *a=(Mat_Elemental*)A->data;
425df311e6cSXuan Zhou   PetscErrorCode ierr;
426df311e6cSXuan Zhou 
427df311e6cSXuan Zhou   PetscFunctionBegin;
428df311e6cSXuan Zhou   ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
429df311e6cSXuan Zhou   ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
430df311e6cSXuan Zhou   ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
431df311e6cSXuan Zhou   ierr = MatSetUp(Be);CHKERRQ(ierr);
432df311e6cSXuan Zhou   *B = Be;
433df311e6cSXuan Zhou   if (op == MAT_COPY_VALUES) {
434df311e6cSXuan Zhou     Mat_Elemental *b=(Mat_Elemental*)Be->data;
435df311e6cSXuan Zhou     elem::Copy(*a->emat,*b->emat);
436df311e6cSXuan Zhou   }
437df311e6cSXuan Zhou   Be->assembled = PETSC_TRUE;
438df311e6cSXuan Zhou   PetscFunctionReturn(0);
439df311e6cSXuan Zhou }
440df311e6cSXuan Zhou 
441df311e6cSXuan Zhou #undef __FUNCT__
442d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental"
443d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
444d6223691SXuan Zhou {
4453512f328SXuan Zhou   /* Only out-of-place supported */
4465262d616SXuan Zhou   Mat            Be;
4475262d616SXuan Zhou   PetscErrorCode ierr;
4485262d616SXuan Zhou   MPI_Comm       comm=((PetscObject)A)->comm;
4494fe7bbcaSHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
450d6223691SXuan Zhou 
451d6223691SXuan Zhou   PetscFunctionBegin;
4525262d616SXuan Zhou   if (reuse == MAT_INITIAL_MATRIX){
4535262d616SXuan Zhou     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
4545262d616SXuan Zhou     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
4555262d616SXuan Zhou     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
4565262d616SXuan Zhou     ierr = MatSetUp(Be);CHKERRQ(ierr);
4575262d616SXuan Zhou     *B = Be;
4585262d616SXuan Zhou   }
4594fe7bbcaSHong Zhang   b = (Mat_Elemental*)Be->data;
4605262d616SXuan Zhou   elem::Transpose(*a->emat,*b->emat);
4615262d616SXuan Zhou   Be->assembled = PETSC_TRUE;
462d6223691SXuan Zhou   PetscFunctionReturn(0);
463d6223691SXuan Zhou }
464d6223691SXuan Zhou 
465d6223691SXuan Zhou #undef __FUNCT__
466dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental"
467dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A)
468dfcb0403SXuan Zhou {
469dfcb0403SXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
470dfcb0403SXuan Zhou 
471dfcb0403SXuan Zhou   PetscFunctionBegin;
472dfcb0403SXuan Zhou   elem::Conjugate(*a->emat);
473dfcb0403SXuan Zhou   PetscFunctionReturn(0);
474dfcb0403SXuan Zhou }
475dfcb0403SXuan Zhou 
476dfcb0403SXuan Zhou #undef __FUNCT__
4774a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental"
4784a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
4794a29722dSXuan Zhou {
4804a29722dSXuan Zhou   /* Only out-of-place supported */
4814a29722dSXuan Zhou   Mat            Be;
4824a29722dSXuan Zhou   PetscErrorCode ierr;
4834a29722dSXuan Zhou   MPI_Comm       comm=((PetscObject)A)->comm;
4844a29722dSXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
4854a29722dSXuan Zhou 
4864a29722dSXuan Zhou   PetscFunctionBegin;
4874a29722dSXuan Zhou   if (reuse == MAT_INITIAL_MATRIX){
4884a29722dSXuan Zhou     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
4894a29722dSXuan Zhou     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
4904a29722dSXuan Zhou     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
4914a29722dSXuan Zhou     ierr = MatSetUp(Be);CHKERRQ(ierr);
4924a29722dSXuan Zhou     *B = Be;
4934a29722dSXuan Zhou   }
4944a29722dSXuan Zhou   b = (Mat_Elemental*)Be->data;
4954a29722dSXuan Zhou   elem::Adjoint(*a->emat,*b->emat);
4964a29722dSXuan Zhou   Be->assembled = PETSC_TRUE;
4974a29722dSXuan Zhou   PetscFunctionReturn(0);
4984a29722dSXuan Zhou }
4994a29722dSXuan Zhou 
5004a29722dSXuan Zhou #undef __FUNCT__
5011f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental"
5021f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X)
5031f881ff8SXuan Zhou {
5041f881ff8SXuan Zhou   Mat_Elemental     *a = (Mat_Elemental*)A->data;
5051f881ff8SXuan Zhou   PetscErrorCode    ierr;
506df311e6cSXuan Zhou   PetscElemScalar   *x;
5071f881ff8SXuan Zhou 
5081f881ff8SXuan Zhou   PetscFunctionBegin;
50945cf121fSXuan Zhou   ierr = VecCopy(B,X);CHKERRQ(ierr);
510e6dea9dbSXuan Zhou   ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
511df311e6cSXuan Zhou   elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid);
512df311e6cSXuan Zhou   elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe;
513fc54b460SXuan Zhou   switch (A->factortype) {
514fc54b460SXuan Zhou   case MAT_FACTOR_LU:
5151f881ff8SXuan Zhou     if ((*a->pivot).AllocatedMemory()) {
5164b28e8e8SXuan Zhou       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer);
51745cf121fSXuan Zhou       elem::Copy(xer,xe);
518fc54b460SXuan Zhou     } else {
5194b28e8e8SXuan Zhou       elem::SolveAfterLU(elem::NORMAL,*a->emat,xer);
52045cf121fSXuan Zhou       elem::Copy(xer,xe);
5211f881ff8SXuan Zhou     }
522fc54b460SXuan Zhou     break;
523fc54b460SXuan Zhou   case MAT_FACTOR_CHOLESKY:
524fc54b460SXuan Zhou     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer);
525fc54b460SXuan Zhou     elem::Copy(xer,xe);
526fc54b460SXuan Zhou     break;
527fc54b460SXuan Zhou   default:
5284fe7bbcaSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
529fc54b460SXuan Zhou     break;
5301f881ff8SXuan Zhou   }
531e6dea9dbSXuan Zhou   ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
5321f881ff8SXuan Zhou   PetscFunctionReturn(0);
5331f881ff8SXuan Zhou }
5341f881ff8SXuan Zhou 
5351f881ff8SXuan Zhou #undef __FUNCT__
536df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental"
537df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X)
538df311e6cSXuan Zhou {
539df311e6cSXuan Zhou   PetscErrorCode    ierr;
540df311e6cSXuan Zhou 
541df311e6cSXuan Zhou   PetscFunctionBegin;
542df311e6cSXuan Zhou   ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr);
5433d7f40dbSXuan Zhou   ierr = VecAXPY(X,1,Y);CHKERRQ(ierr);
544df311e6cSXuan Zhou   PetscFunctionReturn(0);
545df311e6cSXuan Zhou }
546df311e6cSXuan Zhou 
547df311e6cSXuan Zhou #undef __FUNCT__
548ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental"
549ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X)
550ae844d54SHong Zhang {
5511f0e42cfSHong Zhang   Mat_Elemental *a=(Mat_Elemental*)A->data;
552d6223691SXuan Zhou   Mat_Elemental *b=(Mat_Elemental*)B->data;
5531f0e42cfSHong Zhang   Mat_Elemental *x=(Mat_Elemental*)X->data;
5541f0e42cfSHong Zhang 
555ae844d54SHong Zhang   PetscFunctionBegin;
556d6223691SXuan Zhou   elem::Copy(*b->emat,*x->emat);
557fc54b460SXuan Zhou   switch (A->factortype) {
558fc54b460SXuan Zhou   case MAT_FACTOR_LU:
559d6223691SXuan Zhou     if ((*a->pivot).AllocatedMemory()) {
5604b28e8e8SXuan Zhou       elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat);
561fc54b460SXuan Zhou     } else {
5624b28e8e8SXuan Zhou       elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat);
563d6223691SXuan Zhou     }
564fc54b460SXuan Zhou     break;
565fc54b460SXuan Zhou   case MAT_FACTOR_CHOLESKY:
566fc54b460SXuan Zhou     elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat);
567fc54b460SXuan Zhou     break;
568fc54b460SXuan Zhou   default:
5694fe7bbcaSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
570fc54b460SXuan Zhou     break;
571fc54b460SXuan Zhou   }
572ae844d54SHong Zhang   PetscFunctionReturn(0);
573ae844d54SHong Zhang }
574ae844d54SHong Zhang 
575ae844d54SHong Zhang #undef __FUNCT__
576ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental"
577ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info)
578ae844d54SHong Zhang {
5797c920d81SXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
5807c920d81SXuan Zhou 
581ae844d54SHong Zhang   PetscFunctionBegin;
582d6223691SXuan Zhou   if (info->dtcol){
5837c920d81SXuan Zhou     elem::LU(*a->emat,*a->pivot);
5841293973dSHong Zhang   } else {
585d6223691SXuan Zhou     elem::LU(*a->emat);
586d6223691SXuan Zhou   }
5871293973dSHong Zhang   A->factortype = MAT_FACTOR_LU;
588834d3fecSHong Zhang   A->assembled  = PETSC_TRUE;
589ae844d54SHong Zhang   PetscFunctionReturn(0);
590ae844d54SHong Zhang }
591ae844d54SHong Zhang 
592d7c3f9d8SHong Zhang #undef __FUNCT__
593d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental"
594d7c3f9d8SHong Zhang static PetscErrorCode  MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
595d7c3f9d8SHong Zhang {
596d7c3f9d8SHong Zhang   PetscErrorCode ierr;
597d7c3f9d8SHong Zhang 
598d7c3f9d8SHong Zhang   PetscFunctionBegin;
599d7c3f9d8SHong Zhang   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
600d7c3f9d8SHong Zhang   ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr);
601d7c3f9d8SHong Zhang   PetscFunctionReturn(0);
602d7c3f9d8SHong Zhang }
603d7c3f9d8SHong Zhang 
604d7c3f9d8SHong Zhang #undef __FUNCT__
605d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental"
606d7c3f9d8SHong Zhang static PetscErrorCode  MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
607d7c3f9d8SHong Zhang {
608d7c3f9d8SHong Zhang   PetscFunctionBegin;
609d7c3f9d8SHong Zhang   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
610d7c3f9d8SHong Zhang   PetscFunctionReturn(0);
611d7c3f9d8SHong Zhang }
612d7c3f9d8SHong Zhang 
61345cf121fSXuan Zhou #undef __FUNCT__
61445cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental"
61545cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info)
61645cf121fSXuan Zhou {
61745cf121fSXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
618df311e6cSXuan Zhou   elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d;
61945cf121fSXuan Zhou 
62045cf121fSXuan Zhou   PetscFunctionBegin;
621c9fc186eSXuan Zhou   elem::Cholesky(elem::UPPER,*a->emat);
622fc54b460SXuan Zhou   A->factortype = MAT_FACTOR_CHOLESKY;
623834d3fecSHong Zhang   A->assembled  = PETSC_TRUE;
62445cf121fSXuan Zhou   PetscFunctionReturn(0);
62545cf121fSXuan Zhou }
62645cf121fSXuan Zhou 
62779673f7bSHong Zhang #undef __FUNCT__
62879673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental"
62979673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
63079673f7bSHong Zhang {
631cb76c1d8SXuan Zhou   PetscErrorCode ierr;
632cb76c1d8SXuan Zhou 
633cb76c1d8SXuan Zhou   PetscFunctionBegin;
634cb76c1d8SXuan Zhou   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
635cb76c1d8SXuan Zhou   ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr);
636cb76c1d8SXuan Zhou   PetscFunctionReturn(0);
63779673f7bSHong Zhang }
638cb76c1d8SXuan Zhou 
63979673f7bSHong Zhang #undef __FUNCT__
64079673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental"
64179673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info)
64279673f7bSHong Zhang {
64379673f7bSHong Zhang   PetscFunctionBegin;
64479673f7bSHong Zhang   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
64579673f7bSHong Zhang   PetscFunctionReturn(0);
64679673f7bSHong Zhang }
64779673f7bSHong Zhang 
64815767789SHong Zhang EXTERN_C_BEGIN
6491293973dSHong Zhang #undef __FUNCT__
65015767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental"
65115767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type)
65215767789SHong Zhang {
65315767789SHong Zhang   PetscFunctionBegin;
65415767789SHong Zhang   *type = MATSOLVERELEMENTAL;
65515767789SHong Zhang   PetscFunctionReturn(0);
65615767789SHong Zhang }
65715767789SHong Zhang EXTERN_C_END
65815767789SHong Zhang 
65915767789SHong Zhang EXTERN_C_BEGIN
66015767789SHong Zhang #undef __FUNCT__
66115767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental"
66215767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F)
6631293973dSHong Zhang {
6641293973dSHong Zhang   Mat            B;
6651293973dSHong Zhang   PetscErrorCode ierr;
6661293973dSHong Zhang 
6671293973dSHong Zhang   PetscFunctionBegin;
6681293973dSHong Zhang   /* Create the factorization matrix */
6691293973dSHong Zhang   ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
6701293973dSHong Zhang   ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
6711293973dSHong Zhang   ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr);
6721293973dSHong Zhang   ierr = MatSetUp(B);CHKERRQ(ierr);
6731293973dSHong Zhang   B->factortype = ftype;
67415767789SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_elemental_elemental",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr);
6751293973dSHong Zhang   *F            = B;
6761293973dSHong Zhang   PetscFunctionReturn(0);
6771293973dSHong Zhang }
67815767789SHong Zhang EXTERN_C_END
6791293973dSHong Zhang 
6801f881ff8SXuan Zhou #undef __FUNCT__
6811f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental"
6821f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm)
6831f881ff8SXuan Zhou {
6841f881ff8SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
6851f881ff8SXuan Zhou 
6861f881ff8SXuan Zhou   PetscFunctionBegin;
6871f881ff8SXuan Zhou   switch (type){
6881f881ff8SXuan Zhou   case NORM_1:
6891f881ff8SXuan Zhou     *nrm = elem::Norm(*a->emat,elem::ONE_NORM);
6901f881ff8SXuan Zhou     break;
6911f881ff8SXuan Zhou   case NORM_FROBENIUS:
6921f881ff8SXuan Zhou     *nrm = elem::Norm(*a->emat,elem::FROBENIUS_NORM);
6931f881ff8SXuan Zhou     break;
6941f881ff8SXuan Zhou   case NORM_INFINITY:
6951f881ff8SXuan Zhou     *nrm = elem::Norm(*a->emat,elem::INFINITY_NORM);
6961f881ff8SXuan Zhou     break;
6971f881ff8SXuan Zhou   default:
6981f881ff8SXuan Zhou     printf("Error: unsupported norm type!\n");
6991f881ff8SXuan Zhou   }
7001f881ff8SXuan Zhou   PetscFunctionReturn(0);
7011f881ff8SXuan Zhou }
7021f881ff8SXuan Zhou 
7035262d616SXuan Zhou #undef __FUNCT__
7045262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental"
7055262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A)
7065262d616SXuan Zhou {
7075262d616SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
7085262d616SXuan Zhou 
7095262d616SXuan Zhou   PetscFunctionBegin;
7105262d616SXuan Zhou   elem::Zero(*a->emat);
7115262d616SXuan Zhou   PetscFunctionReturn(0);
7125262d616SXuan Zhou }
7135262d616SXuan Zhou 
7141293973dSHong Zhang EXTERN_C_BEGIN
715e09a3074SHong Zhang #undef __FUNCT__
716db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental"
717db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols)
718db31f6deSJed Brown {
719db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
720db31f6deSJed Brown   PetscErrorCode ierr;
721db31f6deSJed Brown   PetscInt       i,m,shift,stride,*idx;
722db31f6deSJed Brown 
723db31f6deSJed Brown   PetscFunctionBegin;
724db31f6deSJed Brown   if (rows) {
725db31f6deSJed Brown     m = a->emat->LocalHeight();
726db31f6deSJed Brown     shift = a->emat->ColShift();
727db31f6deSJed Brown     stride = a->emat->ColStride();
728db31f6deSJed Brown     ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr);
729db31f6deSJed Brown     for (i=0; i<m; i++) {
730db31f6deSJed Brown       PetscInt rank,offset;
731db31f6deSJed Brown       E2RO(A,0,shift+i*stride,&rank,&offset);
732db31f6deSJed Brown       RO2P(A,0,rank,offset,&idx[i]);
733db31f6deSJed Brown     }
734db31f6deSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr);
735db31f6deSJed Brown   }
736db31f6deSJed Brown   if (cols) {
737db31f6deSJed Brown     m = a->emat->LocalWidth();
738db31f6deSJed Brown     shift = a->emat->RowShift();
739db31f6deSJed Brown     stride = a->emat->RowStride();
740db31f6deSJed Brown     ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr);
741db31f6deSJed Brown     for (i=0; i<m; i++) {
742db31f6deSJed Brown       PetscInt rank,offset;
743db31f6deSJed Brown       E2RO(A,1,shift+i*stride,&rank,&offset);
744db31f6deSJed Brown       RO2P(A,1,rank,offset,&idx[i]);
745db31f6deSJed Brown     }
746db31f6deSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr);
747db31f6deSJed Brown   }
748db31f6deSJed Brown   PetscFunctionReturn(0);
749db31f6deSJed Brown }
7501293973dSHong Zhang EXTERN_C_END
751db31f6deSJed Brown 
752db31f6deSJed Brown #undef __FUNCT__
7532ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense"
7542ef0cf24SXuan Zhou static PetscErrorCode MatConvert_Elemental_Dense(Mat A,const MatType newtype,MatReuse reuse,Mat *B)
755af295397SXuan Zhou {
7562ef0cf24SXuan Zhou   Mat                Bmpi;
757af295397SXuan Zhou   Mat_Elemental      *a = (Mat_Elemental*)A->data;
758af295397SXuan Zhou   MPI_Comm           comm=((PetscObject)A)->comm;
7592ef0cf24SXuan Zhou   PetscErrorCode     ierr;
7602ef0cf24SXuan Zhou   PetscInt           rrank,ridx,crank,cidx,nrows,ncols,i,j;
761df311e6cSXuan Zhou   PetscElemScalar    v;
762af295397SXuan Zhou 
763af295397SXuan Zhou   PetscFunctionBegin;
764c4ad791aSXuan Zhou   if (strcmp(newtype,MATDENSE) && strcmp(newtype,MATSEQDENSE) && strcmp(newtype,MATMPIDENSE)) {
765c4ad791aSXuan Zhou     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE");
766c4ad791aSXuan Zhou   }
767af295397SXuan Zhou   ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr);
768af295397SXuan Zhou   ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
7692ef0cf24SXuan Zhou   ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr);
770af295397SXuan Zhou   ierr = MatSetUp(Bmpi);CHKERRQ(ierr);
7712ef0cf24SXuan Zhou   ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr);
7722ef0cf24SXuan Zhou   for (i=0; i<nrows; i++) {
7732ef0cf24SXuan Zhou     PetscInt erow,ecol;
7742ef0cf24SXuan Zhou     P2RO(A,0,i,&rrank,&ridx);
7752ef0cf24SXuan Zhou     RO2E(A,0,rrank,ridx,&erow);
7762ef0cf24SXuan Zhou     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
7772ef0cf24SXuan Zhou     for (j=0; j<ncols; j++) {
7782ef0cf24SXuan Zhou       P2RO(A,1,j,&crank,&cidx);
7792ef0cf24SXuan Zhou       RO2E(A,1,crank,cidx,&ecol);
7802ef0cf24SXuan Zhou       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
7812ef0cf24SXuan Zhou       v = a->emat->Get(erow,ecol);
782e6dea9dbSXuan Zhou       ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr);
7832ef0cf24SXuan Zhou     }
7842ef0cf24SXuan Zhou   }
785af295397SXuan Zhou   ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
786af295397SXuan Zhou   ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
787c4ad791aSXuan Zhou   if (reuse == MAT_REUSE_MATRIX) {
788c4ad791aSXuan Zhou     ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr);
789c4ad791aSXuan Zhou   } else {
790c4ad791aSXuan Zhou     *B = Bmpi;
791c4ad791aSXuan Zhou   }
792af295397SXuan Zhou   PetscFunctionReturn(0);
793af295397SXuan Zhou }
794af295397SXuan Zhou 
795af295397SXuan Zhou #undef __FUNCT__
796db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental"
797db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A)
798db31f6deSJed Brown {
799db31f6deSJed Brown   Mat_Elemental      *a = (Mat_Elemental*)A->data;
800db31f6deSJed Brown   PetscErrorCode     ierr;
8015e9f5b67SHong Zhang   Mat_Elemental_Grid *commgrid;
8025e9f5b67SHong Zhang   PetscBool          flg;
8035e9f5b67SHong Zhang   MPI_Comm           icomm;
804db31f6deSJed Brown 
805db31f6deSJed Brown   PetscFunctionBegin;
806c1ee1e62SHong Zhang   a->interface->Detach();
807aae2c449SHong Zhang   delete a->interface;
808aae2c449SHong Zhang   delete a->esubmat;
809db31f6deSJed Brown   delete a->emat;
8105e9f5b67SHong Zhang 
811180a43e4SHong Zhang   elem::mpi::Comm cxxcomm(((PetscObject)A)->comm);
8125e9f5b67SHong Zhang   ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr);
8135e9f5b67SHong Zhang   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
8145e9f5b67SHong Zhang   if (--commgrid->grid_refct == 0) {
8155e9f5b67SHong Zhang     delete commgrid->grid;
8165e9f5b67SHong Zhang     ierr = PetscFree(commgrid);CHKERRQ(ierr);
8175e9f5b67SHong Zhang   }
8185e9f5b67SHong Zhang   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
819db31f6deSJed Brown   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","",PETSC_NULL);CHKERRQ(ierr);
8201293973dSHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_petsc_C","",PETSC_NULL);CHKERRQ(ierr);
82115767789SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatFactorGetSolverPackage_C","",PETSC_NULL);CHKERRQ(ierr);
822db31f6deSJed Brown   ierr = PetscFree(A->data);CHKERRQ(ierr);
823db31f6deSJed Brown   PetscFunctionReturn(0);
824db31f6deSJed Brown }
825db31f6deSJed Brown 
826db31f6deSJed Brown #undef __FUNCT__
827db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental"
828db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A)
829db31f6deSJed Brown {
830db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
831db31f6deSJed Brown   PetscErrorCode ierr;
832db31f6deSJed Brown   PetscMPIInt    rsize,csize;
833db31f6deSJed Brown 
834db31f6deSJed Brown   PetscFunctionBegin;
835db31f6deSJed Brown   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
836db31f6deSJed Brown   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
837db31f6deSJed Brown 
838db31f6deSJed Brown   a->emat->ResizeTo(A->rmap->N,A->cmap->N);CHKERRQ(ierr);
839db31f6deSJed Brown   elem::Zero(*a->emat);
840db31f6deSJed Brown 
841db31f6deSJed Brown   ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr);
842db31f6deSJed Brown   ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr);
843db31f6deSJed Brown   if (csize != rsize) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes");
844db31f6deSJed Brown   a->commsize = rsize;
845db31f6deSJed Brown   a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize;
846db31f6deSJed Brown   a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize;
847db31f6deSJed Brown   a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize);
848db31f6deSJed Brown   a->m[1] = A->cmap->N / csize + (a->mr[1] != csize);
849db31f6deSJed Brown   PetscFunctionReturn(0);
850db31f6deSJed Brown }
851db31f6deSJed Brown 
852aae2c449SHong Zhang #undef __FUNCT__
853aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental"
854aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type)
855aae2c449SHong Zhang {
856aae2c449SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
857aae2c449SHong Zhang 
858aae2c449SHong Zhang   PetscFunctionBegin;
859aae2c449SHong Zhang   a->interface->Detach();
860aae2c449SHong Zhang   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
861aae2c449SHong Zhang   PetscFunctionReturn(0);
862aae2c449SHong Zhang }
863aae2c449SHong Zhang 
864aae2c449SHong Zhang #undef __FUNCT__
865aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental"
866aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type)
867aae2c449SHong Zhang {
868aae2c449SHong Zhang   PetscFunctionBegin;
869aae2c449SHong Zhang   /* Currently does nothing */
870aae2c449SHong Zhang   PetscFunctionReturn(0);
871aae2c449SHong Zhang }
872aae2c449SHong Zhang 
87340d92e34SHong Zhang /* -------------------------------------------------------------------*/
87440d92e34SHong Zhang static struct _MatOps MatOps_Values = {
87540d92e34SHong Zhang        MatSetValues_Elemental,
87640d92e34SHong Zhang        0,
87740d92e34SHong Zhang        0,
87840d92e34SHong Zhang        MatMult_Elemental,
87940d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental,
8809426833fSXuan Zhou        MatMultTranspose_Elemental,
881e883f9d5SXuan Zhou        MatMultTransposeAdd_Elemental,
88240d92e34SHong Zhang        MatSolve_Elemental,
883df311e6cSXuan Zhou        MatSolveAdd_Elemental,
88440d92e34SHong Zhang        0, //MatSolveTranspose_Elemental,
88540d92e34SHong Zhang /*10*/ 0, //MatSolveTransposeAdd_Elemental,
88640d92e34SHong Zhang        MatLUFactor_Elemental,
88740d92e34SHong Zhang        MatCholeskyFactor_Elemental,
88840d92e34SHong Zhang        0,
88940d92e34SHong Zhang        MatTranspose_Elemental,
89040d92e34SHong Zhang /*15*/ MatGetInfo_Elemental,
89140d92e34SHong Zhang        0,
892*2adf0be3SHong Zhang        0, //MatGetDiagonal_Elemental,?
893*2adf0be3SHong Zhang        0, //MatDiagonalScale_Elemental,?
89440d92e34SHong Zhang        MatNorm_Elemental,
89540d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental,
89640d92e34SHong Zhang        MatAssemblyEnd_Elemental,
89740d92e34SHong Zhang        0, //MatSetOption_Elemental,
89840d92e34SHong Zhang        MatZeroEntries_Elemental,
89940d92e34SHong Zhang /*24*/ 0,
90040d92e34SHong Zhang        MatLUFactorSymbolic_Elemental,
90140d92e34SHong Zhang        MatLUFactorNumeric_Elemental,
90240d92e34SHong Zhang        MatCholeskyFactorSymbolic_Elemental,
90340d92e34SHong Zhang        MatCholeskyFactorNumeric_Elemental,
90440d92e34SHong Zhang /*29*/ MatSetUp_Elemental,
90540d92e34SHong Zhang        0,
90640d92e34SHong Zhang        0,
90740d92e34SHong Zhang        0,
90840d92e34SHong Zhang        0,
909df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental,
91040d92e34SHong Zhang        0,
91140d92e34SHong Zhang        0,
91240d92e34SHong Zhang        0,
91340d92e34SHong Zhang        0,
91440d92e34SHong Zhang /*39*/ MatAXPY_Elemental,
91540d92e34SHong Zhang        0,
91640d92e34SHong Zhang        0,
91740d92e34SHong Zhang        0,
91840d92e34SHong Zhang        MatCopy_Elemental,
91940d92e34SHong Zhang /*44*/ 0,
92040d92e34SHong Zhang        MatScale_Elemental,
92140d92e34SHong Zhang        0,
92240d92e34SHong Zhang        0,
92340d92e34SHong Zhang        0,
92440d92e34SHong Zhang /*49*/ 0,
92540d92e34SHong Zhang        0,
92640d92e34SHong Zhang        0,
92740d92e34SHong Zhang        0,
92840d92e34SHong Zhang        0,
92940d92e34SHong Zhang /*54*/ 0,
93040d92e34SHong Zhang        0,
93140d92e34SHong Zhang        0,
93240d92e34SHong Zhang        0,
93340d92e34SHong Zhang        0,
93440d92e34SHong Zhang /*59*/ 0,
93540d92e34SHong Zhang        MatDestroy_Elemental,
93640d92e34SHong Zhang        MatView_Elemental,
93740d92e34SHong Zhang        0,
93840d92e34SHong Zhang        0,
93940d92e34SHong Zhang /*64*/ 0,
94040d92e34SHong Zhang        0,
94140d92e34SHong Zhang        0,
94240d92e34SHong Zhang        0,
94340d92e34SHong Zhang        0,
94440d92e34SHong Zhang /*69*/ 0,
94540d92e34SHong Zhang        0,
9462ef0cf24SXuan Zhou        MatConvert_Elemental_Dense,
94740d92e34SHong Zhang        0,
94840d92e34SHong Zhang        0,
94940d92e34SHong Zhang /*74*/ 0,
95040d92e34SHong Zhang        0,
95140d92e34SHong Zhang        0,
95240d92e34SHong Zhang        0,
95340d92e34SHong Zhang        0,
95440d92e34SHong Zhang /*79*/ 0,
95540d92e34SHong Zhang        0,
95640d92e34SHong Zhang        0,
95740d92e34SHong Zhang        0,
95840d92e34SHong Zhang        0,
95940d92e34SHong Zhang /*84*/ 0,
96040d92e34SHong Zhang        0,
96140d92e34SHong Zhang        0,
96240d92e34SHong Zhang        0,
96340d92e34SHong Zhang        0,
96440d92e34SHong Zhang /*89*/ MatMatMult_Elemental,
96540d92e34SHong Zhang        MatMatMultSymbolic_Elemental,
96640d92e34SHong Zhang        MatMatMultNumeric_Elemental,
96740d92e34SHong Zhang        0,
96840d92e34SHong Zhang        0,
96940d92e34SHong Zhang /*94*/ 0,
970df311e6cSXuan Zhou        MatMatTransposeMult_Elemental,
971df311e6cSXuan Zhou        MatMatTransposeMultSymbolic_Elemental,
972df311e6cSXuan Zhou        MatMatTransposeMultNumeric_Elemental,
97340d92e34SHong Zhang        0,
97440d92e34SHong Zhang /*99*/ 0,
97540d92e34SHong Zhang        0,
97640d92e34SHong Zhang        0,
977dfcb0403SXuan Zhou        MatConjugate_Elemental,
97840d92e34SHong Zhang        0,
97940d92e34SHong Zhang /*104*/0,
98040d92e34SHong Zhang        0,
98140d92e34SHong Zhang        0,
98240d92e34SHong Zhang        0,
98340d92e34SHong Zhang        0,
98440d92e34SHong Zhang /*109*/MatMatSolve_Elemental,
98540d92e34SHong Zhang        0,
98640d92e34SHong Zhang        0,
98740d92e34SHong Zhang        0,
98840d92e34SHong Zhang        0,
98940d92e34SHong Zhang /*114*/0,
99040d92e34SHong Zhang        0,
99140d92e34SHong Zhang        0,
99240d92e34SHong Zhang        0,
99340d92e34SHong Zhang        0,
99440d92e34SHong Zhang /*119*/0,
9954a29722dSXuan Zhou        MatHermitianTranspose_Elemental,
99640d92e34SHong Zhang        0,
99740d92e34SHong Zhang        0,
99840d92e34SHong Zhang        0,
99940d92e34SHong Zhang /*124*/0,
100040d92e34SHong Zhang        0,
100140d92e34SHong Zhang        0,
100240d92e34SHong Zhang        0,
100340d92e34SHong Zhang        0,
100440d92e34SHong Zhang /*129*/0,
100540d92e34SHong Zhang        0,
100640d92e34SHong Zhang        0,
100740d92e34SHong Zhang        0,
100840d92e34SHong Zhang        0,
100940d92e34SHong Zhang /*134*/0,
101040d92e34SHong Zhang        0,
101140d92e34SHong Zhang        0,
101240d92e34SHong Zhang        0,
101340d92e34SHong Zhang        0
101440d92e34SHong Zhang };
101540d92e34SHong Zhang 
1016ed36708cSHong Zhang /*MC
1017ed36708cSHong Zhang    MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package
1018ed36708cSHong Zhang 
1019ed36708cSHong Zhang    Options Database Keys:
1020ed36708cSHong Zhang . -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions()
1021ed36708cSHong Zhang 
1022ed36708cSHong Zhang   Level: beginner
1023ed36708cSHong Zhang 
1024ed36708cSHong Zhang .seealso: MATDENSE,MatCreateElemental()
1025ed36708cSHong Zhang M*/
10264a29722dSXuan Zhou 
1027db31f6deSJed Brown #undef __FUNCT__
1028db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental"
1029db31f6deSJed Brown PETSC_EXTERN_C PetscErrorCode MatCreate_Elemental(Mat A)
1030db31f6deSJed Brown {
1031db31f6deSJed Brown   Mat_Elemental      *a;
1032db31f6deSJed Brown   PetscErrorCode     ierr;
1033ed667823SXuan Zhou   PetscBool          flg,flg1,flg2;
10345e9f5b67SHong Zhang   Mat_Elemental_Grid *commgrid;
10355e9f5b67SHong Zhang   MPI_Comm           icomm;
1036ed667823SXuan Zhou   PetscInt           optv1,optv2;
1037db31f6deSJed Brown 
1038db31f6deSJed Brown   PetscFunctionBegin;
1039db31f6deSJed Brown   ierr = PetscElementalInitializePackage(PETSC_NULL);CHKERRQ(ierr);
104040d92e34SHong Zhang   ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
104140d92e34SHong Zhang   A->insertmode = NOT_SET_VALUES;
1042db31f6deSJed Brown 
1043db31f6deSJed Brown   ierr = PetscNewLog(A,Mat_Elemental,&a);CHKERRQ(ierr);
1044db31f6deSJed Brown   A->data = (void*)a;
1045db31f6deSJed Brown 
1046db31f6deSJed Brown   /* Set up the elemental matrix */
1047db31f6deSJed Brown   elem::mpi::Comm cxxcomm(((PetscObject)A)->comm);
1048ed667823SXuan Zhou   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr);
10495e9f5b67SHong Zhang 
10505e9f5b67SHong Zhang   /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */
10515e9f5b67SHong Zhang   if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) {
1052180a43e4SHong Zhang     ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0);
10535e9f5b67SHong Zhang   }
10545e9f5b67SHong Zhang   ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr);
10555e9f5b67SHong Zhang   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
10565e9f5b67SHong Zhang   if (!flg) {
10575e9f5b67SHong Zhang     ierr = PetscNewLog(A,Mat_Elemental_Grid,&commgrid);CHKERRQ(ierr);
10582ef0cf24SXuan Zhou     ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr);
10592ef0cf24SXuan Zhou     ierr = PetscOptionsInt("-mat_elemental_grid_width","Grid Width","None",1,&optv2,&flg2);CHKERRQ(ierr);
1060ed667823SXuan Zhou     if (flg1 || flg2) {
1061ed667823SXuan Zhou       if (optv1*optv2 != elem::mpi::CommSize(cxxcomm)) {
10622ef0cf24SXuan Zhou         SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Grid Height times Grid Width must equal CommSize");
1063ed667823SXuan Zhou       }
1064*2adf0be3SHong Zhang       commgrid->grid = new elem::Grid(cxxcomm,optv1,optv2); /* use user-provided grid sizes */
10652ef0cf24SXuan Zhou     } else {
1066*2adf0be3SHong Zhang       commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */
1067ed667823SXuan Zhou     }
10685e9f5b67SHong Zhang     commgrid->grid_refct = 1;
10695e9f5b67SHong Zhang     ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr);
10705e9f5b67SHong Zhang   } else {
10715e9f5b67SHong Zhang     commgrid->grid_refct++;
10725e9f5b67SHong Zhang   }
10735e9f5b67SHong Zhang   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
10745e9f5b67SHong Zhang   a->grid      = commgrid->grid;
1075df311e6cSXuan Zhou   a->emat      = new elem::DistMatrix<PetscElemScalar>(*a->grid);
1076df311e6cSXuan Zhou   a->esubmat   = new elem::Matrix<PetscElemScalar>(1,1);
1077df311e6cSXuan Zhou   a->interface = new elem::AxpyInterface<PetscElemScalar>;
10787c920d81SXuan Zhou   a->pivot     = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>;
1079db31f6deSJed Brown 
1080db31f6deSJed Brown   /* build cache for off array entries formed */
1081aae2c449SHong Zhang   a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat));
1082bafd5131SHong Zhang 
1083db31f6deSJed Brown   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","MatGetOwnershipIS_Elemental",MatGetOwnershipIS_Elemental);CHKERRQ(ierr);
108415767789SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_elemental_C","MatGetFactor_elemental_elemental",MatGetFactor_elemental_elemental);CHKERRQ(ierr);
1085db31f6deSJed Brown 
1086db31f6deSJed Brown   ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr);
1087ed667823SXuan Zhou   PetscOptionsEnd();
1088db31f6deSJed Brown   PetscFunctionReturn(0);
1089db31f6deSJed Brown }
10904a29722dSXuan Zhou 
1091