xref: /petsc/src/mat/impls/elemental/matelem.cxx (revision 124af5d2312c3ea7f8e3d0fcd276a230a3dee0c3)
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;
25e8146892SSatish Balay   if (El::Initialized()) PetscFunctionReturn(0);
26e05a9d59SSatish Balay   El::Initialize();   /* called by the 1st call of MatCreate_Elemental */
27db31f6deSJed Brown   ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr);
28db31f6deSJed Brown   PetscFunctionReturn(0);
29db31f6deSJed Brown }
30db31f6deSJed Brown 
31db31f6deSJed Brown #undef __FUNCT__
32db31f6deSJed Brown #define __FUNCT__ "PetscElementalFinalizePackage"
33db31f6deSJed Brown /*@C
34db31f6deSJed Brown    PetscElementalFinalizePackage - Finalize Elemental package
35db31f6deSJed Brown 
36db31f6deSJed Brown    Logically Collective
37db31f6deSJed Brown 
38db31f6deSJed Brown    Level: developer
39db31f6deSJed Brown 
40db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalInitializePackage()
41db31f6deSJed Brown @*/
42db31f6deSJed Brown PetscErrorCode PetscElementalFinalizePackage(void)
43db31f6deSJed Brown {
44db31f6deSJed Brown   PetscFunctionBegin;
45e8146892SSatish Balay   El::Finalize();  /* called by PetscFinalize() */
46db31f6deSJed Brown   PetscFunctionReturn(0);
47db31f6deSJed Brown }
48db31f6deSJed Brown 
49db31f6deSJed Brown #undef __FUNCT__
50db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental"
51db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer)
52db31f6deSJed Brown {
53db31f6deSJed Brown   PetscErrorCode ierr;
54db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
55db31f6deSJed Brown   PetscBool      iascii;
56db31f6deSJed Brown 
57db31f6deSJed Brown   PetscFunctionBegin;
58db31f6deSJed Brown   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
59db31f6deSJed Brown   if (iascii) {
60db31f6deSJed Brown     PetscViewerFormat format;
61db31f6deSJed Brown     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
62db31f6deSJed Brown     if (format == PETSC_VIEWER_ASCII_INFO) {
6379673f7bSHong Zhang       /* call elemental viewing function */
642d8adcc7SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr);
65ed667823SXuan Zhou       ierr = PetscViewerASCIIPrintf(viewer,"  allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr);
66ed667823SXuan Zhou       ierr = PetscViewerASCIIPrintf(viewer,"  grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr);
674fe7bbcaSHong Zhang       if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
6879673f7bSHong Zhang         /* call elemental viewing function */
69ce94432eSBarry Smith         ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"test matview_elemental 2\n");CHKERRQ(ierr);
704fe7bbcaSHong Zhang       }
7179673f7bSHong Zhang 
72db31f6deSJed Brown     } else if (format == PETSC_VIEWER_DEFAULT) {
73db31f6deSJed Brown       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
74e8146892SSatish Balay       El::Print( *a->emat, "Elemental matrix (cyclic ordering)" );
75db31f6deSJed Brown       ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
76834d3fecSHong Zhang       if (A->factortype == MAT_FACTOR_NONE){
7761119200SXuan Zhou         Mat Adense;
78ce94432eSBarry Smith         ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
7961119200SXuan Zhou         ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr);
8061119200SXuan Zhou         ierr = MatView(Adense,viewer);CHKERRQ(ierr);
8161119200SXuan Zhou         ierr = MatDestroy(&Adense);CHKERRQ(ierr);
82834d3fecSHong Zhang       }
83ce94432eSBarry Smith     } else SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Format");
84d2daa67eSHong Zhang   } else {
855cb544a0SHong Zhang     /* convert to dense format and call MatView() */
8661119200SXuan Zhou     Mat Adense;
87ce94432eSBarry Smith     ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr);
8861119200SXuan Zhou     ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr);
8961119200SXuan Zhou     ierr = MatView(Adense,viewer);CHKERRQ(ierr);
9061119200SXuan Zhou     ierr = MatDestroy(&Adense);CHKERRQ(ierr);
91d2daa67eSHong Zhang   }
92db31f6deSJed Brown   PetscFunctionReturn(0);
93db31f6deSJed Brown }
94db31f6deSJed Brown 
95db31f6deSJed Brown #undef __FUNCT__
96180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental"
9715767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info)
98180a43e4SHong Zhang {
9915767789SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
10015767789SHong Zhang 
101180a43e4SHong Zhang   PetscFunctionBegin;
1025cb544a0SHong Zhang   info->block_size = 1.0;
10315767789SHong Zhang 
10415767789SHong Zhang   if (flag == MAT_LOCAL) {
10515767789SHong Zhang     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
10615767789SHong Zhang     info->nz_used        = info->nz_allocated;
10715767789SHong Zhang   } else if (flag == MAT_GLOBAL_MAX) {
108b2566f29SBarry Smith     //ierr = MPIU_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr);
10915767789SHong Zhang     /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */
11015767789SHong Zhang     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet");
11115767789SHong Zhang   } else if (flag == MAT_GLOBAL_SUM) {
11215767789SHong Zhang     //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet");
11315767789SHong Zhang     info->nz_allocated   = (double)(*a->emat).AllocatedMemory(); /* locally allocated */
11415767789SHong Zhang     info->nz_used        = info->nz_allocated; /* assume Elemental does accurate allocation */
115b2566f29SBarry Smith     //ierr = MPIU_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,PetscObjectComm((PetscObject)A));CHKERRQ(ierr);
11615767789SHong Zhang     //PetscPrintf(PETSC_COMM_SELF,"    ... [%d] locally allocated %g\n",rank,info->nz_allocated);
11715767789SHong Zhang   }
11815767789SHong Zhang 
11915767789SHong Zhang   info->nz_unneeded       = 0.0;
12015767789SHong Zhang   info->assemblies        = (double)A->num_ass;
12115767789SHong Zhang   info->mallocs           = 0;
12215767789SHong Zhang   info->memory            = ((PetscObject)A)->mem;
12315767789SHong Zhang   info->fill_ratio_given  = 0; /* determined by Elemental */
12415767789SHong Zhang   info->fill_ratio_needed = 0;
12515767789SHong Zhang   info->factor_mallocs    = 0;
126180a43e4SHong Zhang   PetscFunctionReturn(0);
127180a43e4SHong Zhang }
128180a43e4SHong Zhang 
129180a43e4SHong Zhang #undef __FUNCT__
13032d7a744SHong Zhang #define __FUNCT__ "MatSetOption_Elemental"
13132d7a744SHong Zhang PetscErrorCode MatSetOption_Elemental(Mat A,MatOption op,PetscBool flg)
13232d7a744SHong Zhang {
13332d7a744SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
13432d7a744SHong Zhang 
13532d7a744SHong Zhang   PetscFunctionBegin;
13632d7a744SHong Zhang   switch (op) {
13732d7a744SHong Zhang   case MAT_NEW_NONZERO_LOCATIONS:
13832d7a744SHong Zhang   case MAT_NEW_NONZERO_LOCATION_ERR:
13932d7a744SHong Zhang   case MAT_NEW_NONZERO_ALLOCATION_ERR:
14032d7a744SHong Zhang   case MAT_ROW_ORIENTED:
14132d7a744SHong Zhang     a->roworiented = flg;
14232d7a744SHong Zhang     break;
1435a8bcd62SHong Zhang   case MAT_SYMMETRIC:
1445a8bcd62SHong Zhang     break;
14532d7a744SHong Zhang   default:
14632d7a744SHong Zhang     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
14732d7a744SHong Zhang   }
14832d7a744SHong Zhang   PetscFunctionReturn(0);
14932d7a744SHong Zhang }
15032d7a744SHong Zhang 
15132d7a744SHong Zhang #undef __FUNCT__
152db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental"
153e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode)
154db31f6deSJed Brown {
155db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
156fbbcb730SJack Poulson   PetscInt       i,j,rrank,ridx,crank,cidx,erow,ecol,numQueues=0;
157db31f6deSJed Brown 
158db31f6deSJed Brown   PetscFunctionBegin;
159fbbcb730SJack Poulson   // TODO: Initialize matrix to all zeros?
160fbbcb730SJack Poulson 
161fbbcb730SJack Poulson   // Count the number of queues from this process
16232d7a744SHong Zhang   if (a->roworiented) {
163db31f6deSJed Brown     for (i=0; i<nr; i++) {
164db31f6deSJed Brown       if (rows[i] < 0) continue;
165db31f6deSJed Brown       P2RO(A,0,rows[i],&rrank,&ridx);
166db31f6deSJed Brown       RO2E(A,0,rrank,ridx,&erow);
167ce94432eSBarry Smith       if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation");
168db31f6deSJed Brown       for (j=0; j<nc; j++) {
169db31f6deSJed Brown         if (cols[j] < 0) continue;
170db31f6deSJed Brown         P2RO(A,1,cols[j],&crank,&cidx);
171db31f6deSJed Brown         RO2E(A,1,crank,cidx,&ecol);
172ce94432eSBarry Smith         if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation");
173fbbcb730SJack Poulson         if (!a->emat->IsLocal(erow,ecol) ){ /* off-proc entry */
174fbbcb730SJack Poulson           /* printf("Will later remotely update (%d,%d)\n",erow,ecol); */
175aae2c449SHong Zhang           if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported");
176fbbcb730SJack Poulson           ++numQueues;
177aae2c449SHong Zhang           continue;
178ed36708cSHong Zhang         }
179fbbcb730SJack Poulson         /* printf("Locally updating (%d,%d)\n",erow,ecol); */
180db31f6deSJed Brown         switch (imode) {
181fbbcb730SJack Poulson         case INSERT_VALUES: a->emat->Set(erow,ecol,(PetscElemScalar)vals[i*nc+j]); break;
182fbbcb730SJack Poulson         case ADD_VALUES: a->emat->Update(erow,ecol,(PetscElemScalar)vals[i*nc+j]); break;
183ce94432eSBarry Smith         default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode);
184db31f6deSJed Brown         }
185db31f6deSJed Brown       }
186db31f6deSJed Brown     }
187fbbcb730SJack Poulson 
188fbbcb730SJack Poulson     /* printf("numQueues=%d\n",numQueues); */
189fbbcb730SJack Poulson     a->emat->Reserve( numQueues );
190fbbcb730SJack Poulson     for (i=0; i<nr; i++) {
191fbbcb730SJack Poulson       if (rows[i] < 0) continue;
192fbbcb730SJack Poulson       P2RO(A,0,rows[i],&rrank,&ridx);
193fbbcb730SJack Poulson       RO2E(A,0,rrank,ridx,&erow);
194fbbcb730SJack Poulson       for (j=0; j<nc; j++) {
195fbbcb730SJack Poulson         if (cols[j] < 0) continue;
196fbbcb730SJack Poulson         P2RO(A,1,cols[j],&crank,&cidx);
197fbbcb730SJack Poulson         RO2E(A,1,crank,cidx,&ecol);
198fbbcb730SJack Poulson         if ( !a->emat->IsLocal(erow,ecol) ) { /*off-proc entry*/
199fbbcb730SJack Poulson           /* printf("Queueing remotely update of (%d,%d)\n",erow,ecol); */
200fbbcb730SJack Poulson           a->emat->QueueUpdate( erow, ecol, vals[i*nc+j] );
201fbbcb730SJack Poulson         }
202fbbcb730SJack Poulson       }
203fbbcb730SJack Poulson     }
20432d7a744SHong Zhang   } else { /* columnoriented */
20532d7a744SHong Zhang     for (j=0; j<nc; j++) {
20632d7a744SHong Zhang       if (cols[j] < 0) continue;
20732d7a744SHong Zhang       P2RO(A,1,cols[j],&crank,&cidx);
20832d7a744SHong Zhang       RO2E(A,1,crank,cidx,&ecol);
20932d7a744SHong Zhang       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation");
21032d7a744SHong Zhang       for (i=0; i<nr; i++) {
21132d7a744SHong Zhang         if (rows[i] < 0) continue;
21232d7a744SHong Zhang         P2RO(A,0,rows[i],&rrank,&ridx);
21332d7a744SHong Zhang         RO2E(A,0,rrank,ridx,&erow);
21432d7a744SHong Zhang         if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation");
21532d7a744SHong Zhang         if (!a->emat->IsLocal(erow,ecol) ){ /* off-proc entry */
21632d7a744SHong Zhang           /* printf("Will later remotely update (%d,%d)\n",erow,ecol); */
21732d7a744SHong Zhang           if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported");
21832d7a744SHong Zhang           ++numQueues;
21932d7a744SHong Zhang           continue;
22032d7a744SHong Zhang         }
22132d7a744SHong Zhang         /* printf("Locally updating (%d,%d)\n",erow,ecol); */
22232d7a744SHong Zhang         switch (imode) {
22332d7a744SHong Zhang         case INSERT_VALUES: a->emat->Set(erow,ecol,(PetscElemScalar)vals[i+j*nr]); break;
22432d7a744SHong Zhang         case ADD_VALUES: a->emat->Update(erow,ecol,(PetscElemScalar)vals[i+j*nr]); break;
22532d7a744SHong Zhang         default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode);
22632d7a744SHong Zhang         }
22732d7a744SHong Zhang       }
22832d7a744SHong Zhang     }
22932d7a744SHong Zhang 
23032d7a744SHong Zhang     /* printf("numQueues=%d\n",numQueues); */
23132d7a744SHong Zhang     a->emat->Reserve( numQueues );
23232d7a744SHong Zhang     for (j=0; j<nc; j++) {
23332d7a744SHong Zhang       if (cols[j] < 0) continue;
23432d7a744SHong Zhang       P2RO(A,1,cols[j],&crank,&cidx);
23532d7a744SHong Zhang       RO2E(A,1,crank,cidx,&ecol);
23632d7a744SHong Zhang 
23732d7a744SHong Zhang       for (i=0; i<nr; i++) {
23832d7a744SHong Zhang         if (rows[i] < 0) continue;
23932d7a744SHong Zhang         P2RO(A,0,rows[i],&rrank,&ridx);
24032d7a744SHong Zhang         RO2E(A,0,rrank,ridx,&erow);
24132d7a744SHong Zhang         if ( !a->emat->IsLocal(erow,ecol) ) { /*off-proc entry*/
24232d7a744SHong Zhang           /* printf("Queueing remotely update of (%d,%d)\n",erow,ecol); */
24332d7a744SHong Zhang           a->emat->QueueUpdate( erow, ecol, vals[i+j*nr] );
24432d7a744SHong Zhang         }
24532d7a744SHong Zhang       }
24632d7a744SHong Zhang     }
24732d7a744SHong Zhang   }
248db31f6deSJed Brown   PetscFunctionReturn(0);
249db31f6deSJed Brown }
250db31f6deSJed Brown 
251db31f6deSJed Brown #undef __FUNCT__
252db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental"
253db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y)
254db31f6deSJed Brown {
255db31f6deSJed Brown   Mat_Elemental         *a = (Mat_Elemental*)A->data;
256db31f6deSJed Brown   PetscErrorCode        ierr;
257e6dea9dbSXuan Zhou   const PetscElemScalar *x;
258e6dea9dbSXuan Zhou   PetscElemScalar       *y;
259df311e6cSXuan Zhou   PetscElemScalar       one = 1,zero = 0;
260db31f6deSJed Brown 
261db31f6deSJed Brown   PetscFunctionBegin;
262e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
263e6dea9dbSXuan Zhou   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
264db31f6deSJed Brown   { /* Scoping so that constructor is called before pointer is returned */
265e8146892SSatish Balay     El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ye;
2660c18141cSBarry Smith     xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n);
2670c18141cSBarry Smith     ye.Attach(A->rmap->N,1,*a->grid,0,0,y,A->rmap->n);
268e8146892SSatish Balay     El::Gemv(El::NORMAL,one,*a->emat,xe,zero,ye);
269db31f6deSJed Brown   }
270e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
271e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
272db31f6deSJed Brown   PetscFunctionReturn(0);
273db31f6deSJed Brown }
274db31f6deSJed Brown 
275db31f6deSJed Brown #undef __FUNCT__
2769426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental"
2779426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y)
2789426833fSXuan Zhou {
2799426833fSXuan Zhou   Mat_Elemental         *a = (Mat_Elemental*)A->data;
2809426833fSXuan Zhou   PetscErrorCode        ierr;
281df311e6cSXuan Zhou   const PetscElemScalar *x;
282df311e6cSXuan Zhou   PetscElemScalar       *y;
283e6dea9dbSXuan Zhou   PetscElemScalar       one = 1,zero = 0;
2849426833fSXuan Zhou 
2859426833fSXuan Zhou   PetscFunctionBegin;
286e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
287e6dea9dbSXuan Zhou   ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
2889426833fSXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
289e8146892SSatish Balay     El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ye;
2900c18141cSBarry Smith     xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n);
2910c18141cSBarry Smith     ye.Attach(A->cmap->N,1,*a->grid,0,0,y,A->cmap->n);
292e8146892SSatish Balay     El::Gemv(El::TRANSPOSE,one,*a->emat,xe,zero,ye);
2939426833fSXuan Zhou   }
294e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
295e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr);
2969426833fSXuan Zhou   PetscFunctionReturn(0);
2979426833fSXuan Zhou }
2989426833fSXuan Zhou 
2999426833fSXuan Zhou #undef __FUNCT__
300db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental"
301db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
302db31f6deSJed Brown {
303db31f6deSJed Brown   Mat_Elemental         *a = (Mat_Elemental*)A->data;
304db31f6deSJed Brown   PetscErrorCode        ierr;
305df311e6cSXuan Zhou   const PetscElemScalar *x;
306df311e6cSXuan Zhou   PetscElemScalar       *z;
307e6dea9dbSXuan Zhou   PetscElemScalar       one = 1;
308db31f6deSJed Brown 
309db31f6deSJed Brown   PetscFunctionBegin;
310db31f6deSJed Brown   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
311e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
312e6dea9dbSXuan Zhou   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
313db31f6deSJed Brown   { /* Scoping so that constructor is called before pointer is returned */
314e8146892SSatish Balay     El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ze;
3150c18141cSBarry Smith     xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n);
3160c18141cSBarry Smith     ze.Attach(A->rmap->N,1,*a->grid,0,0,z,A->rmap->n);
317e8146892SSatish Balay     El::Gemv(El::NORMAL,one,*a->emat,xe,one,ze);
318db31f6deSJed Brown   }
319e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
320e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
321db31f6deSJed Brown   PetscFunctionReturn(0);
322db31f6deSJed Brown }
323db31f6deSJed Brown 
324db31f6deSJed Brown #undef __FUNCT__
325e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental"
326e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z)
327e883f9d5SXuan Zhou {
328e883f9d5SXuan Zhou   Mat_Elemental         *a = (Mat_Elemental*)A->data;
329e883f9d5SXuan Zhou   PetscErrorCode        ierr;
330df311e6cSXuan Zhou   const PetscElemScalar *x;
331df311e6cSXuan Zhou   PetscElemScalar       *z;
332e6dea9dbSXuan Zhou   PetscElemScalar       one = 1;
333e883f9d5SXuan Zhou 
334e883f9d5SXuan Zhou   PetscFunctionBegin;
335e883f9d5SXuan Zhou   if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);}
336e6dea9dbSXuan Zhou   ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
337e6dea9dbSXuan Zhou   ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
338e883f9d5SXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
339e8146892SSatish Balay     El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ze;
3400c18141cSBarry Smith     xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n);
3410c18141cSBarry Smith     ze.Attach(A->cmap->N,1,*a->grid,0,0,z,A->cmap->n);
342e8146892SSatish Balay     El::Gemv(El::TRANSPOSE,one,*a->emat,xe,one,ze);
343e883f9d5SXuan Zhou   }
344e6dea9dbSXuan Zhou   ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr);
345e6dea9dbSXuan Zhou   ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr);
346e883f9d5SXuan Zhou   PetscFunctionReturn(0);
347e883f9d5SXuan Zhou }
348e883f9d5SXuan Zhou 
349e883f9d5SXuan Zhou #undef __FUNCT__
3509a9e8502SHong Zhang #define __FUNCT__ "MatMatMultNumeric_Elemental"
3519a9e8502SHong Zhang static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C)
352c1d1b975SXuan Zhou {
353c1d1b975SXuan Zhou   Mat_Elemental    *a = (Mat_Elemental*)A->data;
354c1d1b975SXuan Zhou   Mat_Elemental    *b = (Mat_Elemental*)B->data;
3559a9e8502SHong Zhang   Mat_Elemental    *c = (Mat_Elemental*)C->data;
356e6dea9dbSXuan Zhou   PetscElemScalar  one = 1,zero = 0;
357c1d1b975SXuan Zhou 
358c1d1b975SXuan Zhou   PetscFunctionBegin;
359aae2c449SHong Zhang   { /* Scoping so that constructor is called before pointer is returned */
360e8146892SSatish Balay     El::Gemm(El::NORMAL,El::NORMAL,one,*a->emat,*b->emat,zero,*c->emat);
361aae2c449SHong Zhang   }
3629a9e8502SHong Zhang   C->assembled = PETSC_TRUE;
3639a9e8502SHong Zhang   PetscFunctionReturn(0);
3649a9e8502SHong Zhang }
3659a9e8502SHong Zhang 
3669a9e8502SHong Zhang #undef __FUNCT__
3679a9e8502SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_Elemental"
3689a9e8502SHong Zhang static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
3699a9e8502SHong Zhang {
3709a9e8502SHong Zhang   PetscErrorCode ierr;
3719a9e8502SHong Zhang   Mat            Ce;
372ce94432eSBarry Smith   MPI_Comm       comm;
3739a9e8502SHong Zhang 
3749a9e8502SHong Zhang   PetscFunctionBegin;
375ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
3769a9e8502SHong Zhang   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
3779a9e8502SHong Zhang   ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
3789a9e8502SHong Zhang   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
3799a9e8502SHong Zhang   ierr = MatSetUp(Ce);CHKERRQ(ierr);
3809a9e8502SHong Zhang   *C = Ce;
3819a9e8502SHong Zhang   PetscFunctionReturn(0);
3829a9e8502SHong Zhang }
3839a9e8502SHong Zhang 
3849a9e8502SHong Zhang #undef __FUNCT__
3859a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental"
3869a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
3879a9e8502SHong Zhang {
3889a9e8502SHong Zhang   PetscErrorCode ierr;
3899a9e8502SHong Zhang 
3909a9e8502SHong Zhang   PetscFunctionBegin;
3919a9e8502SHong Zhang   if (scall == MAT_INITIAL_MATRIX){
3923ff4c91cSHong Zhang     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
3939a9e8502SHong Zhang     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
3943ff4c91cSHong Zhang     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
3959a9e8502SHong Zhang   }
3963ff4c91cSHong Zhang   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
3979a9e8502SHong Zhang   ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
3983ff4c91cSHong Zhang   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
399c1d1b975SXuan Zhou   PetscFunctionReturn(0);
400c1d1b975SXuan Zhou }
401c1d1b975SXuan Zhou 
402c1d1b975SXuan Zhou #undef __FUNCT__
403df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental"
404df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C)
405df311e6cSXuan Zhou {
406df311e6cSXuan Zhou   Mat_Elemental      *a = (Mat_Elemental*)A->data;
407df311e6cSXuan Zhou   Mat_Elemental      *b = (Mat_Elemental*)B->data;
408df311e6cSXuan Zhou   Mat_Elemental      *c = (Mat_Elemental*)C->data;
409e6dea9dbSXuan Zhou   PetscElemScalar    one = 1,zero = 0;
410df311e6cSXuan Zhou 
411df311e6cSXuan Zhou   PetscFunctionBegin;
412df311e6cSXuan Zhou   { /* Scoping so that constructor is called before pointer is returned */
413e8146892SSatish Balay     El::Gemm(El::NORMAL,El::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat);
414df311e6cSXuan Zhou   }
415df311e6cSXuan Zhou   C->assembled = PETSC_TRUE;
416df311e6cSXuan Zhou   PetscFunctionReturn(0);
417df311e6cSXuan Zhou }
418df311e6cSXuan Zhou 
419df311e6cSXuan Zhou #undef __FUNCT__
420df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental"
421df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C)
422df311e6cSXuan Zhou {
423df311e6cSXuan Zhou   PetscErrorCode ierr;
424df311e6cSXuan Zhou   Mat            Ce;
425ce94432eSBarry Smith   MPI_Comm       comm;
426df311e6cSXuan Zhou 
427df311e6cSXuan Zhou   PetscFunctionBegin;
428ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
429df311e6cSXuan Zhou   ierr = MatCreate(comm,&Ce);CHKERRQ(ierr);
430df311e6cSXuan Zhou   ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
431df311e6cSXuan Zhou   ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr);
432df311e6cSXuan Zhou   ierr = MatSetUp(Ce);CHKERRQ(ierr);
433df311e6cSXuan Zhou   *C = Ce;
434df311e6cSXuan Zhou   PetscFunctionReturn(0);
435df311e6cSXuan Zhou }
436df311e6cSXuan Zhou 
437df311e6cSXuan Zhou #undef __FUNCT__
438df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental"
439df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
440df311e6cSXuan Zhou {
441df311e6cSXuan Zhou   PetscErrorCode ierr;
442df311e6cSXuan Zhou 
443df311e6cSXuan Zhou   PetscFunctionBegin;
444df311e6cSXuan Zhou   if (scall == MAT_INITIAL_MATRIX){
445df311e6cSXuan Zhou     ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
446df311e6cSXuan Zhou     ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr);
447df311e6cSXuan Zhou     ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr);
448df311e6cSXuan Zhou   }
449df311e6cSXuan Zhou   ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
450df311e6cSXuan Zhou   ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr);
451df311e6cSXuan Zhou   ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr);
452df311e6cSXuan Zhou   PetscFunctionReturn(0);
453df311e6cSXuan Zhou }
454df311e6cSXuan Zhou 
455df311e6cSXuan Zhou #undef __FUNCT__
45661119200SXuan Zhou #define __FUNCT__ "MatGetDiagonal_Elemental"
457a9d89745SXuan Zhou static PetscErrorCode MatGetDiagonal_Elemental(Mat A,Vec D)
45861119200SXuan Zhou {
459a9d89745SXuan Zhou   PetscInt        i,nrows,ncols,nD,rrank,ridx,crank,cidx;
460a9d89745SXuan Zhou   Mat_Elemental   *a = (Mat_Elemental*)A->data;
46161119200SXuan Zhou   PetscErrorCode  ierr;
462a9d89745SXuan Zhou   PetscElemScalar v;
463ce94432eSBarry Smith   MPI_Comm        comm;
46461119200SXuan Zhou 
46561119200SXuan Zhou   PetscFunctionBegin;
466ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
467a9d89745SXuan Zhou   ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr);
468a9d89745SXuan Zhou   nD = nrows>ncols ? ncols : nrows;
469a9d89745SXuan Zhou   for (i=0; i<nD; i++) {
470a9d89745SXuan Zhou     PetscInt erow,ecol;
471a9d89745SXuan Zhou     P2RO(A,0,i,&rrank,&ridx);
472a9d89745SXuan Zhou     RO2E(A,0,rrank,ridx,&erow);
473a9d89745SXuan Zhou     if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
474a9d89745SXuan Zhou     P2RO(A,1,i,&crank,&cidx);
475a9d89745SXuan Zhou     RO2E(A,1,crank,cidx,&ecol);
476a9d89745SXuan Zhou     if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
477a9d89745SXuan Zhou     v = a->emat->Get(erow,ecol);
4787c8b904fSSatish Balay     ierr = VecSetValues(D,1,&i,(PetscScalar*)&v,INSERT_VALUES);CHKERRQ(ierr);
479ade3cc5eSXuan Zhou   }
480a9d89745SXuan Zhou   ierr = VecAssemblyBegin(D);CHKERRQ(ierr);
481a9d89745SXuan Zhou   ierr = VecAssemblyEnd(D);CHKERRQ(ierr);
48261119200SXuan Zhou   PetscFunctionReturn(0);
48361119200SXuan Zhou }
48461119200SXuan Zhou 
48561119200SXuan Zhou #undef __FUNCT__
486ade3cc5eSXuan Zhou #define __FUNCT__ "MatDiagonalScale_Elemental"
487ade3cc5eSXuan Zhou static PetscErrorCode MatDiagonalScale_Elemental(Mat X,Vec L,Vec R)
488ade3cc5eSXuan Zhou {
489ade3cc5eSXuan Zhou   Mat_Elemental         *x = (Mat_Elemental*)X->data;
490ade3cc5eSXuan Zhou   const PetscElemScalar *d;
491ade3cc5eSXuan Zhou   PetscErrorCode        ierr;
492ade3cc5eSXuan Zhou 
493ade3cc5eSXuan Zhou   PetscFunctionBegin;
4949065cd98SJed Brown   if (R) {
495ade3cc5eSXuan Zhou     ierr = VecGetArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr);
496e8146892SSatish Balay     El::DistMatrix<PetscElemScalar,El::VC,El::STAR> de;
4970c18141cSBarry Smith     de.LockedAttach(X->cmap->N,1,*x->grid,0,0,d,X->cmap->n);
498e8146892SSatish Balay     El::DiagonalScale(El::RIGHT,El::NORMAL,de,*x->emat);
499ade3cc5eSXuan Zhou     ierr = VecRestoreArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr);
5009065cd98SJed Brown   }
5019065cd98SJed Brown   if (L) {
502ade3cc5eSXuan Zhou     ierr = VecGetArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr);
503e8146892SSatish Balay     El::DistMatrix<PetscElemScalar,El::VC,El::STAR> de;
5040c18141cSBarry Smith     de.LockedAttach(X->rmap->N,1,*x->grid,0,0,d,X->rmap->n);
505e8146892SSatish Balay     El::DiagonalScale(El::LEFT,El::NORMAL,de,*x->emat);
506ade3cc5eSXuan Zhou     ierr = VecRestoreArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr);
507ade3cc5eSXuan Zhou   }
508ade3cc5eSXuan Zhou   PetscFunctionReturn(0);
509ade3cc5eSXuan Zhou }
510ade3cc5eSXuan Zhou 
511ade3cc5eSXuan Zhou #undef __FUNCT__
5124ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental"
513e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a)
51465b78793SXuan Zhou {
51565b78793SXuan Zhou   Mat_Elemental  *x = (Mat_Elemental*)X->data;
51665b78793SXuan Zhou 
51765b78793SXuan Zhou   PetscFunctionBegin;
518e8146892SSatish Balay   El::Scale((PetscElemScalar)a,*x->emat);
51965b78793SXuan Zhou   PetscFunctionReturn(0);
52065b78793SXuan Zhou }
52165b78793SXuan Zhou 
522f82baa17SHong Zhang /*
523f82baa17SHong Zhang   MatAXPY - Computes Y = a*X + Y.
524f82baa17SHong Zhang */
52565b78793SXuan Zhou #undef __FUNCT__
5264ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental"
527e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str)
528e09a3074SHong Zhang {
529e09a3074SHong Zhang   Mat_Elemental  *x = (Mat_Elemental*)X->data;
530e09a3074SHong Zhang   Mat_Elemental  *y = (Mat_Elemental*)Y->data;
53168446cf8SJed Brown   PetscErrorCode ierr;
532e09a3074SHong Zhang 
533e09a3074SHong Zhang   PetscFunctionBegin;
534e8146892SSatish Balay   El::Axpy((PetscElemScalar)a,*x->emat,*y->emat);
53521f6c9c4SJed Brown   ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
536e09a3074SHong Zhang   PetscFunctionReturn(0);
537e09a3074SHong Zhang }
538e09a3074SHong Zhang 
539ae844d54SHong Zhang #undef __FUNCT__
540d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental"
541d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str)
542d6223691SXuan Zhou {
543d6223691SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
544d6223691SXuan Zhou   Mat_Elemental *b=(Mat_Elemental*)B->data;
545d6223691SXuan Zhou 
546d6223691SXuan Zhou   PetscFunctionBegin;
547e8146892SSatish Balay   El::Copy(*a->emat,*b->emat);
548d6223691SXuan Zhou   PetscFunctionReturn(0);
549d6223691SXuan Zhou }
550d6223691SXuan Zhou 
551d6223691SXuan Zhou #undef __FUNCT__
552df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental"
553df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B)
554df311e6cSXuan Zhou {
555df311e6cSXuan Zhou   Mat            Be;
556ce94432eSBarry Smith   MPI_Comm       comm;
557df311e6cSXuan Zhou   Mat_Elemental  *a=(Mat_Elemental*)A->data;
558df311e6cSXuan Zhou   PetscErrorCode ierr;
559df311e6cSXuan Zhou 
560df311e6cSXuan Zhou   PetscFunctionBegin;
561ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
562df311e6cSXuan Zhou   ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
563df311e6cSXuan Zhou   ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
564df311e6cSXuan Zhou   ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
565df311e6cSXuan Zhou   ierr = MatSetUp(Be);CHKERRQ(ierr);
566df311e6cSXuan Zhou   *B = Be;
567df311e6cSXuan Zhou   if (op == MAT_COPY_VALUES) {
568df311e6cSXuan Zhou     Mat_Elemental *b=(Mat_Elemental*)Be->data;
569e8146892SSatish Balay     El::Copy(*a->emat,*b->emat);
570df311e6cSXuan Zhou   }
571df311e6cSXuan Zhou   Be->assembled = PETSC_TRUE;
572df311e6cSXuan Zhou   PetscFunctionReturn(0);
573df311e6cSXuan Zhou }
574df311e6cSXuan Zhou 
575df311e6cSXuan Zhou #undef __FUNCT__
576d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental"
577d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
578d6223691SXuan Zhou {
579ec8cb81fSBarry Smith   Mat            Be = *B;
5805262d616SXuan Zhou   PetscErrorCode ierr;
581ce94432eSBarry Smith   MPI_Comm       comm;
5824fe7bbcaSHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
583d6223691SXuan Zhou 
584d6223691SXuan Zhou   PetscFunctionBegin;
585ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
5865cb544a0SHong Zhang   /* Only out-of-place supported */
5875262d616SXuan Zhou   if (reuse == MAT_INITIAL_MATRIX){
5885262d616SXuan Zhou     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
5895262d616SXuan Zhou     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
5905262d616SXuan Zhou     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
5915262d616SXuan Zhou     ierr = MatSetUp(Be);CHKERRQ(ierr);
5925262d616SXuan Zhou     *B = Be;
5935262d616SXuan Zhou   }
5944fe7bbcaSHong Zhang   b = (Mat_Elemental*)Be->data;
595e8146892SSatish Balay   El::Transpose(*a->emat,*b->emat);
5965262d616SXuan Zhou   Be->assembled = PETSC_TRUE;
597d6223691SXuan Zhou   PetscFunctionReturn(0);
598d6223691SXuan Zhou }
599d6223691SXuan Zhou 
600d6223691SXuan Zhou #undef __FUNCT__
601dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental"
602dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A)
603dfcb0403SXuan Zhou {
604dfcb0403SXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
605dfcb0403SXuan Zhou 
606dfcb0403SXuan Zhou   PetscFunctionBegin;
607e8146892SSatish Balay   El::Conjugate(*a->emat);
608dfcb0403SXuan Zhou   PetscFunctionReturn(0);
609dfcb0403SXuan Zhou }
610dfcb0403SXuan Zhou 
611dfcb0403SXuan Zhou #undef __FUNCT__
6124a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental"
6134a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B)
6144a29722dSXuan Zhou {
615ec8cb81fSBarry Smith   Mat            Be = *B;
6164a29722dSXuan Zhou   PetscErrorCode ierr;
617ce94432eSBarry Smith   MPI_Comm       comm;
6184a29722dSXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data, *b;
6194a29722dSXuan Zhou 
6204a29722dSXuan Zhou   PetscFunctionBegin;
621ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
6225cb544a0SHong Zhang   /* Only out-of-place supported */
6234a29722dSXuan Zhou   if (reuse == MAT_INITIAL_MATRIX){
6244a29722dSXuan Zhou     ierr = MatCreate(comm,&Be);CHKERRQ(ierr);
6254a29722dSXuan Zhou     ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
6264a29722dSXuan Zhou     ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr);
6274a29722dSXuan Zhou     ierr = MatSetUp(Be);CHKERRQ(ierr);
6284a29722dSXuan Zhou     *B = Be;
6294a29722dSXuan Zhou   }
6304a29722dSXuan Zhou   b = (Mat_Elemental*)Be->data;
631e8146892SSatish Balay   El::Adjoint(*a->emat,*b->emat);
6324a29722dSXuan Zhou   Be->assembled = PETSC_TRUE;
6334a29722dSXuan Zhou   PetscFunctionReturn(0);
6344a29722dSXuan Zhou }
6354a29722dSXuan Zhou 
6364a29722dSXuan Zhou #undef __FUNCT__
6371f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental"
6381f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X)
6391f881ff8SXuan Zhou {
6401f881ff8SXuan Zhou   Mat_Elemental     *a = (Mat_Elemental*)A->data;
6411f881ff8SXuan Zhou   PetscErrorCode    ierr;
642df311e6cSXuan Zhou   PetscElemScalar   *x;
6431f881ff8SXuan Zhou 
6441f881ff8SXuan Zhou   PetscFunctionBegin;
64545cf121fSXuan Zhou   ierr = VecCopy(B,X);CHKERRQ(ierr);
646e6dea9dbSXuan Zhou   ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
647e8146892SSatish Balay   El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe;
6480c18141cSBarry Smith   xe.Attach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n);
649e8146892SSatish Balay   El::DistMatrix<PetscElemScalar,El::MC,El::MR> xer(xe);
650fc54b460SXuan Zhou   switch (A->factortype) {
651fc54b460SXuan Zhou   case MAT_FACTOR_LU:
6521f881ff8SXuan Zhou     if ((*a->pivot).AllocatedMemory()) {
653e8146892SSatish Balay       El::lu::SolveAfter(El::NORMAL,*a->emat,*a->pivot,xer);
654e8146892SSatish Balay       El::Copy(xer,xe);
655fc54b460SXuan Zhou     } else {
656e8146892SSatish Balay       El::lu::SolveAfter(El::NORMAL,*a->emat,xer);
657e8146892SSatish Balay       El::Copy(xer,xe);
6581f881ff8SXuan Zhou     }
659fc54b460SXuan Zhou     break;
660fc54b460SXuan Zhou   case MAT_FACTOR_CHOLESKY:
661e8146892SSatish Balay     El::cholesky::SolveAfter(El::UPPER,El::NORMAL,*a->emat,xer);
662e8146892SSatish Balay     El::Copy(xer,xe);
663fc54b460SXuan Zhou     break;
664fc54b460SXuan Zhou   default:
6654fe7bbcaSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
666fc54b460SXuan Zhou     break;
6671f881ff8SXuan Zhou   }
668e6dea9dbSXuan Zhou   ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr);
6691f881ff8SXuan Zhou   PetscFunctionReturn(0);
6701f881ff8SXuan Zhou }
6711f881ff8SXuan Zhou 
6721f881ff8SXuan Zhou #undef __FUNCT__
673df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental"
674df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X)
675df311e6cSXuan Zhou {
676df311e6cSXuan Zhou   PetscErrorCode    ierr;
677df311e6cSXuan Zhou 
678df311e6cSXuan Zhou   PetscFunctionBegin;
679df311e6cSXuan Zhou   ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr);
6803d7f40dbSXuan Zhou   ierr = VecAXPY(X,1,Y);CHKERRQ(ierr);
681df311e6cSXuan Zhou   PetscFunctionReturn(0);
682df311e6cSXuan Zhou }
683df311e6cSXuan Zhou 
684df311e6cSXuan Zhou #undef __FUNCT__
685ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental"
686ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X)
687ae844d54SHong Zhang {
6881f0e42cfSHong Zhang   Mat_Elemental *a=(Mat_Elemental*)A->data;
689d6223691SXuan Zhou   Mat_Elemental *b=(Mat_Elemental*)B->data;
6901f0e42cfSHong Zhang   Mat_Elemental *x=(Mat_Elemental*)X->data;
6911f0e42cfSHong Zhang 
692ae844d54SHong Zhang   PetscFunctionBegin;
693e8146892SSatish Balay   El::Copy(*b->emat,*x->emat);
694fc54b460SXuan Zhou   switch (A->factortype) {
695fc54b460SXuan Zhou   case MAT_FACTOR_LU:
696d6223691SXuan Zhou     if ((*a->pivot).AllocatedMemory()) {
697e8146892SSatish Balay       El::lu::SolveAfter(El::NORMAL,*a->emat,*a->pivot,*x->emat);
698fc54b460SXuan Zhou     } else {
699e8146892SSatish Balay       El::lu::SolveAfter(El::NORMAL,*a->emat,*x->emat);
700d6223691SXuan Zhou     }
701fc54b460SXuan Zhou     break;
702fc54b460SXuan Zhou   case MAT_FACTOR_CHOLESKY:
703e8146892SSatish Balay     El::cholesky::SolveAfter(El::UPPER,El::NORMAL,*a->emat,*x->emat);
704fc54b460SXuan Zhou     break;
705fc54b460SXuan Zhou   default:
7064fe7bbcaSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType");
707fc54b460SXuan Zhou     break;
708fc54b460SXuan Zhou   }
709ae844d54SHong Zhang   PetscFunctionReturn(0);
710ae844d54SHong Zhang }
711ae844d54SHong Zhang 
712ae844d54SHong Zhang #undef __FUNCT__
713ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental"
714ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info)
715ae844d54SHong Zhang {
7167c920d81SXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
717*124af5d2SSatish Balay   PetscErrorCode ierr;
7187c920d81SXuan Zhou 
719ae844d54SHong Zhang   PetscFunctionBegin;
720d6223691SXuan Zhou   if (info->dtcol){
721e8146892SSatish Balay     El::LU(*a->emat,*a->pivot);
7221293973dSHong Zhang   } else {
723e8146892SSatish Balay     El::LU(*a->emat);
724d6223691SXuan Zhou   }
7251293973dSHong Zhang   A->factortype = MAT_FACTOR_LU;
726834d3fecSHong Zhang   A->assembled  = PETSC_TRUE;
727f6224b95SHong Zhang 
728f6224b95SHong Zhang   ierr = PetscFree(A->solvertype);CHKERRQ(ierr);
729f6224b95SHong Zhang   ierr = PetscStrallocpy(MATSOLVERELEMENTAL,&A->solvertype);CHKERRQ(ierr);
730ae844d54SHong Zhang   PetscFunctionReturn(0);
731ae844d54SHong Zhang }
732ae844d54SHong Zhang 
733d7c3f9d8SHong Zhang #undef __FUNCT__
734d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental"
735d7c3f9d8SHong Zhang static PetscErrorCode  MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
736d7c3f9d8SHong Zhang {
737d7c3f9d8SHong Zhang   PetscErrorCode ierr;
738d7c3f9d8SHong Zhang 
739d7c3f9d8SHong Zhang   PetscFunctionBegin;
740d7c3f9d8SHong Zhang   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
741d7c3f9d8SHong Zhang   ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr);
742d7c3f9d8SHong Zhang   PetscFunctionReturn(0);
743d7c3f9d8SHong Zhang }
744d7c3f9d8SHong Zhang 
745d7c3f9d8SHong Zhang #undef __FUNCT__
746d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental"
747d7c3f9d8SHong Zhang static PetscErrorCode  MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
748d7c3f9d8SHong Zhang {
749d7c3f9d8SHong Zhang   PetscFunctionBegin;
750d7c3f9d8SHong Zhang   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
751d7c3f9d8SHong Zhang   PetscFunctionReturn(0);
752d7c3f9d8SHong Zhang }
753d7c3f9d8SHong Zhang 
75445cf121fSXuan Zhou #undef __FUNCT__
75545cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental"
75645cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info)
75745cf121fSXuan Zhou {
75845cf121fSXuan Zhou   Mat_Elemental  *a = (Mat_Elemental*)A->data;
759e8146892SSatish Balay   El::DistMatrix<PetscElemScalar,El::MC,El::STAR> d;
760*124af5d2SSatish Balay   PetscErrorCode ierr;
76145cf121fSXuan Zhou 
76245cf121fSXuan Zhou   PetscFunctionBegin;
763e8146892SSatish Balay   El::Cholesky(El::UPPER,*a->emat);
764fc54b460SXuan Zhou   A->factortype = MAT_FACTOR_CHOLESKY;
765834d3fecSHong Zhang   A->assembled  = PETSC_TRUE;
766f6224b95SHong Zhang 
767f6224b95SHong Zhang   ierr = PetscFree(A->solvertype);CHKERRQ(ierr);
768f6224b95SHong Zhang   ierr = PetscStrallocpy(MATSOLVERELEMENTAL,&A->solvertype);CHKERRQ(ierr);
76945cf121fSXuan Zhou   PetscFunctionReturn(0);
77045cf121fSXuan Zhou }
77145cf121fSXuan Zhou 
77279673f7bSHong Zhang #undef __FUNCT__
77379673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental"
77479673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info)
77579673f7bSHong Zhang {
776cb76c1d8SXuan Zhou   PetscErrorCode ierr;
777cb76c1d8SXuan Zhou 
778cb76c1d8SXuan Zhou   PetscFunctionBegin;
779cb76c1d8SXuan Zhou   ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
780cb76c1d8SXuan Zhou   ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr);
781cb76c1d8SXuan Zhou   PetscFunctionReturn(0);
78279673f7bSHong Zhang }
783cb76c1d8SXuan Zhou 
78479673f7bSHong Zhang #undef __FUNCT__
78579673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental"
78679673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info)
78779673f7bSHong Zhang {
78879673f7bSHong Zhang   PetscFunctionBegin;
78979673f7bSHong Zhang   /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */
79079673f7bSHong Zhang   PetscFunctionReturn(0);
79179673f7bSHong Zhang }
79279673f7bSHong Zhang 
7931293973dSHong Zhang #undef __FUNCT__
79415767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental"
79515767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type)
79615767789SHong Zhang {
79715767789SHong Zhang   PetscFunctionBegin;
79815767789SHong Zhang   *type = MATSOLVERELEMENTAL;
79915767789SHong Zhang   PetscFunctionReturn(0);
80015767789SHong Zhang }
80115767789SHong Zhang 
80215767789SHong Zhang #undef __FUNCT__
80315767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental"
80415767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F)
8051293973dSHong Zhang {
8061293973dSHong Zhang   Mat            B;
8071293973dSHong Zhang   PetscErrorCode ierr;
8081293973dSHong Zhang 
8091293973dSHong Zhang   PetscFunctionBegin;
8101293973dSHong Zhang   /* Create the factorization matrix */
811ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
8121293973dSHong Zhang   ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
8131293973dSHong Zhang   ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr);
8141293973dSHong Zhang   ierr = MatSetUp(B);CHKERRQ(ierr);
8151293973dSHong Zhang   B->factortype = ftype;
81600c67f3bSHong Zhang   ierr = PetscFree(B->solvertype);CHKERRQ(ierr);
81700c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERELEMENTAL,&B->solvertype);CHKERRQ(ierr);
81800c67f3bSHong Zhang 
819bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr);
8201293973dSHong Zhang   *F            = B;
8211293973dSHong Zhang   PetscFunctionReturn(0);
8221293973dSHong Zhang }
8231293973dSHong Zhang 
8241f881ff8SXuan Zhou #undef __FUNCT__
82542c9c57cSBarry Smith #define __FUNCT__ "MatSolverPackageRegister_Elemental"
82629b38603SBarry Smith PETSC_EXTERN PetscErrorCode MatSolverPackageRegister_Elemental(void)
82742c9c57cSBarry Smith {
82818713533SBarry Smith   PetscErrorCode ierr;
82918713533SBarry Smith 
83042c9c57cSBarry Smith   PetscFunctionBegin;
83142c9c57cSBarry Smith   ierr = MatSolverPackageRegister(MATSOLVERELEMENTAL,MATELEMENTAL,        MAT_FACTOR_LU,MatGetFactor_elemental_elemental);CHKERRQ(ierr);
83242c9c57cSBarry Smith   ierr = MatSolverPackageRegister(MATSOLVERELEMENTAL,MATELEMENTAL,        MAT_FACTOR_CHOLESKY,MatGetFactor_elemental_elemental);CHKERRQ(ierr);
83342c9c57cSBarry Smith   PetscFunctionReturn(0);
83442c9c57cSBarry Smith }
83542c9c57cSBarry Smith 
83642c9c57cSBarry Smith #undef __FUNCT__
8371f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental"
8381f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm)
8391f881ff8SXuan Zhou {
8401f881ff8SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
8411f881ff8SXuan Zhou 
8421f881ff8SXuan Zhou   PetscFunctionBegin;
8431f881ff8SXuan Zhou   switch (type){
8441f881ff8SXuan Zhou   case NORM_1:
845e8146892SSatish Balay     *nrm = El::OneNorm(*a->emat);
8461f881ff8SXuan Zhou     break;
8471f881ff8SXuan Zhou   case NORM_FROBENIUS:
848e8146892SSatish Balay     *nrm = El::FrobeniusNorm(*a->emat);
8491f881ff8SXuan Zhou     break;
8501f881ff8SXuan Zhou   case NORM_INFINITY:
851e8146892SSatish Balay     *nrm = El::InfinityNorm(*a->emat);
8521f881ff8SXuan Zhou     break;
8531f881ff8SXuan Zhou   default:
8541f881ff8SXuan Zhou     printf("Error: unsupported norm type!\n");
8551f881ff8SXuan Zhou   }
8561f881ff8SXuan Zhou   PetscFunctionReturn(0);
8571f881ff8SXuan Zhou }
8581f881ff8SXuan Zhou 
8595262d616SXuan Zhou #undef __FUNCT__
8605262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental"
8615262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A)
8625262d616SXuan Zhou {
8635262d616SXuan Zhou   Mat_Elemental *a=(Mat_Elemental*)A->data;
8645262d616SXuan Zhou 
8655262d616SXuan Zhou   PetscFunctionBegin;
866e8146892SSatish Balay   El::Zero(*a->emat);
8675262d616SXuan Zhou   PetscFunctionReturn(0);
8685262d616SXuan Zhou }
8695262d616SXuan Zhou 
870e09a3074SHong Zhang #undef __FUNCT__
871db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental"
872db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols)
873db31f6deSJed Brown {
874db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
875db31f6deSJed Brown   PetscErrorCode ierr;
876db31f6deSJed Brown   PetscInt       i,m,shift,stride,*idx;
877db31f6deSJed Brown 
878db31f6deSJed Brown   PetscFunctionBegin;
879db31f6deSJed Brown   if (rows) {
880db31f6deSJed Brown     m = a->emat->LocalHeight();
881db31f6deSJed Brown     shift = a->emat->ColShift();
882db31f6deSJed Brown     stride = a->emat->ColStride();
883785e854fSJed Brown     ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr);
884db31f6deSJed Brown     for (i=0; i<m; i++) {
885db31f6deSJed Brown       PetscInt rank,offset;
886db31f6deSJed Brown       E2RO(A,0,shift+i*stride,&rank,&offset);
887db31f6deSJed Brown       RO2P(A,0,rank,offset,&idx[i]);
888db31f6deSJed Brown     }
889db31f6deSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr);
890db31f6deSJed Brown   }
891db31f6deSJed Brown   if (cols) {
892db31f6deSJed Brown     m = a->emat->LocalWidth();
893db31f6deSJed Brown     shift = a->emat->RowShift();
894db31f6deSJed Brown     stride = a->emat->RowStride();
895785e854fSJed Brown     ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr);
896db31f6deSJed Brown     for (i=0; i<m; i++) {
897db31f6deSJed Brown       PetscInt rank,offset;
898db31f6deSJed Brown       E2RO(A,1,shift+i*stride,&rank,&offset);
899db31f6deSJed Brown       RO2P(A,1,rank,offset,&idx[i]);
900db31f6deSJed Brown     }
901db31f6deSJed Brown     ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr);
902db31f6deSJed Brown   }
903db31f6deSJed Brown   PetscFunctionReturn(0);
904db31f6deSJed Brown }
905db31f6deSJed Brown 
906db31f6deSJed Brown #undef __FUNCT__
9072ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense"
90819fd82e9SBarry Smith static PetscErrorCode MatConvert_Elemental_Dense(Mat A,MatType newtype,MatReuse reuse,Mat *B)
909af295397SXuan Zhou {
9102ef0cf24SXuan Zhou   Mat                Bmpi;
911af295397SXuan Zhou   Mat_Elemental      *a = (Mat_Elemental*)A->data;
912ce94432eSBarry Smith   MPI_Comm           comm;
9132ef0cf24SXuan Zhou   PetscErrorCode     ierr;
914fa9e26c8SHong Zhang   IS                 isrows,iscols;
915fa9e26c8SHong Zhang   PetscInt           rrank,ridx,crank,cidx,nrows,ncols,i,j,erow,ecol,elrow,elcol;
916fa9e26c8SHong Zhang   const PetscInt     *rows,*cols;
917df311e6cSXuan Zhou   PetscElemScalar    v;
918fa9e26c8SHong Zhang   const El::Grid     &grid = a->emat->Grid();
919af295397SXuan Zhou 
920af295397SXuan Zhou   PetscFunctionBegin;
921ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
922a000e53bSHong Zhang 
923de0a22f0SHong Zhang   if (reuse == MAT_REUSE_MATRIX) {
924de0a22f0SHong Zhang     Bmpi = *B;
925de0a22f0SHong Zhang   } else {
926af295397SXuan Zhou     ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr);
927af295397SXuan Zhou     ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr);
9282ef0cf24SXuan Zhou     ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr);
929af295397SXuan Zhou     ierr = MatSetUp(Bmpi);CHKERRQ(ierr);
930de0a22f0SHong Zhang   }
931fa9e26c8SHong Zhang 
932fa9e26c8SHong Zhang   /* Get local entries of A */
933fa9e26c8SHong Zhang   ierr = MatGetOwnershipIS(A,&isrows,&iscols);CHKERRQ(ierr);
934fa9e26c8SHong Zhang   ierr = ISGetLocalSize(isrows,&nrows);CHKERRQ(ierr);
935fa9e26c8SHong Zhang   ierr = ISGetIndices(isrows,&rows);CHKERRQ(ierr);
936fa9e26c8SHong Zhang   ierr = ISGetLocalSize(iscols,&ncols);CHKERRQ(ierr);
937fa9e26c8SHong Zhang   ierr = ISGetIndices(iscols,&cols);CHKERRQ(ierr);
938fa9e26c8SHong Zhang 
93957299f2fSHong Zhang   if (a->roworiented) {
9402ef0cf24SXuan Zhou     for (i=0; i<nrows; i++) {
941fa9e26c8SHong Zhang       P2RO(A,0,rows[i],&rrank,&ridx); /* convert indices between PETSc <-> (Rank,Offset) <-> Elemental */
9422ef0cf24SXuan Zhou       RO2E(A,0,rrank,ridx,&erow);
9432ef0cf24SXuan Zhou       if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
9442ef0cf24SXuan Zhou       for (j=0; j<ncols; j++) {
945fa9e26c8SHong Zhang         P2RO(A,1,cols[j],&crank,&cidx);
9462ef0cf24SXuan Zhou         RO2E(A,1,crank,cidx,&ecol);
9472ef0cf24SXuan Zhou         if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
948fa9e26c8SHong Zhang 
949fa9e26c8SHong Zhang         elrow = erow / grid.MCSize(); /* Elemental local row index */
950fa9e26c8SHong Zhang         elcol = ecol / grid.MRSize(); /* Elemental local column index */
951fa9e26c8SHong Zhang         v = a->emat->GetLocal(elrow,elcol);
952fa9e26c8SHong Zhang         ierr = MatSetValues(Bmpi,1,&rows[i],1,&cols[j],(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr);
9532ef0cf24SXuan Zhou       }
9542ef0cf24SXuan Zhou     }
95557299f2fSHong Zhang   } else { /* column-oriented */
95657299f2fSHong Zhang     for (j=0; j<ncols; j++) {
95757299f2fSHong Zhang       P2RO(A,1,cols[j],&crank,&cidx);
95857299f2fSHong Zhang       RO2E(A,1,crank,cidx,&ecol);
95957299f2fSHong Zhang       if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation");
96057299f2fSHong Zhang       for (i=0; i<nrows; i++) {
96157299f2fSHong Zhang         P2RO(A,0,rows[i],&rrank,&ridx); /* convert indices between PETSc <-> (Rank,Offset) <-> Elemental */
96257299f2fSHong Zhang         RO2E(A,0,rrank,ridx,&erow);
96357299f2fSHong Zhang         if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation");
96457299f2fSHong Zhang 
96557299f2fSHong Zhang         elrow = erow / grid.MCSize(); /* Elemental local row index */
96657299f2fSHong Zhang         elcol = ecol / grid.MRSize(); /* Elemental local column index */
96757299f2fSHong Zhang         v = a->emat->GetLocal(elrow,elcol);
96857299f2fSHong Zhang         ierr = MatSetValues(Bmpi,1,&rows[i],1,&cols[j],(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr);
96957299f2fSHong Zhang       }
97057299f2fSHong Zhang     }
97157299f2fSHong Zhang   }
972af295397SXuan Zhou   ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
973af295397SXuan Zhou   ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
974511c6705SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
97528be2f97SBarry Smith     ierr = MatHeaderReplace(A,&Bmpi);CHKERRQ(ierr);
976c4ad791aSXuan Zhou   } else {
977c4ad791aSXuan Zhou     *B = Bmpi;
978c4ad791aSXuan Zhou   }
979fa9e26c8SHong Zhang   ierr = ISDestroy(&isrows);CHKERRQ(ierr);
980fa9e26c8SHong Zhang   ierr = ISDestroy(&iscols);CHKERRQ(ierr);
981af295397SXuan Zhou   PetscFunctionReturn(0);
982af295397SXuan Zhou }
983af295397SXuan Zhou 
984af295397SXuan Zhou #undef __FUNCT__
985af8000cdSHong Zhang #define __FUNCT__ "MatConvert_SeqAIJ_Elemental"
986cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
987af8000cdSHong Zhang {
988af8000cdSHong Zhang   Mat               mat_elemental;
989af8000cdSHong Zhang   PetscErrorCode    ierr;
990af8000cdSHong Zhang   PetscInt          M=A->rmap->N,N=A->cmap->N,row,ncols;
991af8000cdSHong Zhang   const PetscInt    *cols;
992af8000cdSHong Zhang   const PetscScalar *vals;
993af8000cdSHong Zhang 
994af8000cdSHong Zhang   PetscFunctionBegin;
995af8000cdSHong Zhang   ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr);
996af8000cdSHong Zhang   ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr);
997af8000cdSHong Zhang   ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr);
998af8000cdSHong Zhang   ierr = MatSetUp(mat_elemental);CHKERRQ(ierr);
999af8000cdSHong Zhang   for (row=0; row<M; row++) {
1000af8000cdSHong Zhang     ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
1001af8000cdSHong Zhang     /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */
1002af8000cdSHong Zhang     ierr = MatSetValues(mat_elemental,1,&row,ncols,cols,vals,ADD_VALUES);CHKERRQ(ierr);
1003af8000cdSHong Zhang     ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
1004af8000cdSHong Zhang   }
1005af8000cdSHong Zhang   ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1006af8000cdSHong Zhang   ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1007af8000cdSHong Zhang 
1008511c6705SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
100928be2f97SBarry Smith     ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr);
1010af8000cdSHong Zhang   } else {
1011af8000cdSHong Zhang     *newmat = mat_elemental;
1012af8000cdSHong Zhang   }
1013af8000cdSHong Zhang   PetscFunctionReturn(0);
1014af8000cdSHong Zhang }
1015af8000cdSHong Zhang 
1016af8000cdSHong Zhang #undef __FUNCT__
10175d7652ecSHong Zhang #define __FUNCT__ "MatConvert_MPIAIJ_Elemental"
1018cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_MPIAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
10195d7652ecSHong Zhang {
10205d7652ecSHong Zhang   Mat               mat_elemental;
10215d7652ecSHong Zhang   PetscErrorCode    ierr;
10225d7652ecSHong Zhang   PetscInt          row,ncols,rstart=A->rmap->rstart,rend=A->rmap->rend,j;
10235d7652ecSHong Zhang   const PetscInt    *cols;
10245d7652ecSHong Zhang   const PetscScalar *vals;
10255d7652ecSHong Zhang 
10265d7652ecSHong Zhang   PetscFunctionBegin;
10275d7652ecSHong Zhang   ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr);
10285d7652ecSHong Zhang   ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
10295d7652ecSHong Zhang   ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr);
10305d7652ecSHong Zhang   ierr = MatSetUp(mat_elemental);CHKERRQ(ierr);
10315d7652ecSHong Zhang   for (row=rstart; row<rend; row++) {
10325d7652ecSHong Zhang     ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
10335d7652ecSHong Zhang     for (j=0; j<ncols; j++) {
10345d7652ecSHong Zhang       /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */
10355d7652ecSHong Zhang       ierr = MatSetValues(mat_elemental,1,&row,1,&cols[j],&vals[j],ADD_VALUES);CHKERRQ(ierr);
10365d7652ecSHong Zhang     }
10375d7652ecSHong Zhang     ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
10385d7652ecSHong Zhang   }
10395d7652ecSHong Zhang   ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10405d7652ecSHong Zhang   ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10415d7652ecSHong Zhang 
1042511c6705SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
104328be2f97SBarry Smith     ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr);
10445d7652ecSHong Zhang   } else {
10455d7652ecSHong Zhang     *newmat = mat_elemental;
10465d7652ecSHong Zhang   }
10475d7652ecSHong Zhang   PetscFunctionReturn(0);
10485d7652ecSHong Zhang }
10495d7652ecSHong Zhang 
10505d7652ecSHong Zhang #undef __FUNCT__
10516214f412SHong Zhang #define __FUNCT__ "MatConvert_SeqSBAIJ_Elemental"
1052cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
10536214f412SHong Zhang {
10546214f412SHong Zhang   Mat               mat_elemental;
10556214f412SHong Zhang   PetscErrorCode    ierr;
10566214f412SHong Zhang   PetscInt          M=A->rmap->N,N=A->cmap->N,row,ncols,j;
10576214f412SHong Zhang   const PetscInt    *cols;
10586214f412SHong Zhang   const PetscScalar *vals;
10596214f412SHong Zhang 
10606214f412SHong Zhang   PetscFunctionBegin;
10616214f412SHong Zhang   ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr);
10626214f412SHong Zhang   ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr);
10636214f412SHong Zhang   ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr);
10646214f412SHong Zhang   ierr = MatSetUp(mat_elemental);CHKERRQ(ierr);
10656214f412SHong Zhang   ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
10666214f412SHong Zhang   for (row=0; row<M; row++) {
10676214f412SHong Zhang     ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
10686214f412SHong Zhang     /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */
10696214f412SHong Zhang     ierr = MatSetValues(mat_elemental,1,&row,ncols,cols,vals,ADD_VALUES);CHKERRQ(ierr);
10706214f412SHong Zhang     for (j=0; j<ncols; j++) { /* lower triangular part */
10716214f412SHong Zhang       if (cols[j] == row) continue;
10726214f412SHong Zhang       ierr = MatSetValues(mat_elemental,1,&cols[j],1,&row,&vals[j],ADD_VALUES);CHKERRQ(ierr);
10736214f412SHong Zhang     }
10746214f412SHong Zhang     ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
10756214f412SHong Zhang   }
10766214f412SHong Zhang   ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
10776214f412SHong Zhang   ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10786214f412SHong Zhang   ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10796214f412SHong Zhang 
1080511c6705SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
108128be2f97SBarry Smith     ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr);
10826214f412SHong Zhang   } else {
10836214f412SHong Zhang     *newmat = mat_elemental;
10846214f412SHong Zhang   }
10856214f412SHong Zhang   PetscFunctionReturn(0);
10866214f412SHong Zhang }
10876214f412SHong Zhang 
10886214f412SHong Zhang #undef __FUNCT__
10896214f412SHong Zhang #define __FUNCT__ "MatConvert_MPISBAIJ_Elemental"
1090cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_MPISBAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
10916214f412SHong Zhang {
10926214f412SHong Zhang   Mat               mat_elemental;
10936214f412SHong Zhang   PetscErrorCode    ierr;
10946214f412SHong Zhang   PetscInt          M=A->rmap->N,N=A->cmap->N,row,ncols,j,rstart=A->rmap->rstart,rend=A->rmap->rend;
10956214f412SHong Zhang   const PetscInt    *cols;
10966214f412SHong Zhang   const PetscScalar *vals;
10976214f412SHong Zhang 
10986214f412SHong Zhang   PetscFunctionBegin;
10996214f412SHong Zhang   ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr);
11006214f412SHong Zhang   ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr);
11016214f412SHong Zhang   ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr);
11026214f412SHong Zhang   ierr = MatSetUp(mat_elemental);CHKERRQ(ierr);
11036214f412SHong Zhang   ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11046214f412SHong Zhang   for (row=rstart; row<rend; row++) {
11056214f412SHong Zhang     ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
11066214f412SHong Zhang     /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */
11076214f412SHong Zhang     ierr = MatSetValues(mat_elemental,1,&row,ncols,cols,vals,ADD_VALUES);CHKERRQ(ierr);
11086214f412SHong Zhang     for (j=0; j<ncols; j++) { /* lower triangular part */
11096214f412SHong Zhang       if (cols[j] == row) continue;
11106214f412SHong Zhang       ierr = MatSetValues(mat_elemental,1,&cols[j],1,&row,&vals[j],ADD_VALUES);CHKERRQ(ierr);
11116214f412SHong Zhang     }
11126214f412SHong Zhang     ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr);
11136214f412SHong Zhang   }
11146214f412SHong Zhang   ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11156214f412SHong Zhang   ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11166214f412SHong Zhang   ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11176214f412SHong Zhang 
1118511c6705SHong Zhang   if (reuse == MAT_INPLACE_MATRIX) {
111928be2f97SBarry Smith     ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr);
11206214f412SHong Zhang   } else {
11216214f412SHong Zhang     *newmat = mat_elemental;
11226214f412SHong Zhang   }
11236214f412SHong Zhang   PetscFunctionReturn(0);
11246214f412SHong Zhang }
11256214f412SHong Zhang 
11266214f412SHong Zhang #undef __FUNCT__
1127db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental"
1128db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A)
1129db31f6deSJed Brown {
1130db31f6deSJed Brown   Mat_Elemental      *a = (Mat_Elemental*)A->data;
1131db31f6deSJed Brown   PetscErrorCode     ierr;
11325e9f5b67SHong Zhang   Mat_Elemental_Grid *commgrid;
11335e9f5b67SHong Zhang   PetscBool          flg;
11345e9f5b67SHong Zhang   MPI_Comm           icomm;
1135db31f6deSJed Brown 
1136db31f6deSJed Brown   PetscFunctionBegin;
1137db31f6deSJed Brown   delete a->emat;
1138da0640a4SHong Zhang   delete a->pivot;
11395e9f5b67SHong Zhang 
1140e8146892SSatish Balay   El::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A));
11410c18141cSBarry Smith   ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr);
11425e9f5b67SHong Zhang   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
11435e9f5b67SHong Zhang   if (--commgrid->grid_refct == 0) {
11445e9f5b67SHong Zhang     delete commgrid->grid;
11455e9f5b67SHong Zhang     ierr = PetscFree(commgrid);CHKERRQ(ierr);
1146a000e53bSHong Zhang     ierr = MPI_Keyval_free(&Petsc_Elemental_keyval);CHKERRQ(ierr);
11475e9f5b67SHong Zhang   }
11485e9f5b67SHong Zhang   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
1149bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",NULL);CHKERRQ(ierr);
1150bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSolverPackage_C",NULL);CHKERRQ(ierr);
1151d5560a0bSSatish Balay   ierr = PetscObjectComposeFunction((PetscObject)A,"MatElementalHermitianGenDefEig_C",NULL);CHKERRQ(ierr);
1152db31f6deSJed Brown   ierr = PetscFree(A->data);CHKERRQ(ierr);
1153db31f6deSJed Brown   PetscFunctionReturn(0);
1154db31f6deSJed Brown }
1155db31f6deSJed Brown 
1156db31f6deSJed Brown #undef __FUNCT__
1157db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental"
1158db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A)
1159db31f6deSJed Brown {
1160db31f6deSJed Brown   Mat_Elemental  *a = (Mat_Elemental*)A->data;
1161db31f6deSJed Brown   PetscErrorCode ierr;
1162db31f6deSJed Brown   PetscMPIInt    rsize,csize;
1163db31f6deSJed Brown 
1164db31f6deSJed Brown   PetscFunctionBegin;
1165db31f6deSJed Brown   ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr);
1166db31f6deSJed Brown   ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr);
1167db31f6deSJed Brown 
1168efb79153SJack Poulson   a->emat->Resize(A->rmap->N,A->cmap->N);CHKERRQ(ierr);
1169e8146892SSatish Balay   El::Zero(*a->emat);
1170db31f6deSJed Brown 
1171db31f6deSJed Brown   ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr);
1172db31f6deSJed Brown   ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr);
1173ce94432eSBarry Smith   if (csize != rsize) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes");
1174db31f6deSJed Brown   a->commsize = rsize;
1175db31f6deSJed Brown   a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize;
1176db31f6deSJed Brown   a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize;
1177db31f6deSJed Brown   a->m[0]  = A->rmap->N / rsize + (a->mr[0] != rsize);
1178db31f6deSJed Brown   a->m[1]  = A->cmap->N / csize + (a->mr[1] != csize);
1179db31f6deSJed Brown   PetscFunctionReturn(0);
1180db31f6deSJed Brown }
1181db31f6deSJed Brown 
1182aae2c449SHong Zhang #undef __FUNCT__
1183aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental"
1184aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type)
1185aae2c449SHong Zhang {
1186aae2c449SHong Zhang   Mat_Elemental  *a = (Mat_Elemental*)A->data;
1187aae2c449SHong Zhang 
1188aae2c449SHong Zhang   PetscFunctionBegin;
1189fbbcb730SJack Poulson   /* printf("Calling ProcessQueues\n"); */
1190fbbcb730SJack Poulson   a->emat->ProcessQueues();
1191fbbcb730SJack Poulson   /* printf("Finished ProcessQueues\n"); */
1192aae2c449SHong Zhang   PetscFunctionReturn(0);
1193aae2c449SHong Zhang }
1194aae2c449SHong Zhang 
1195aae2c449SHong Zhang #undef __FUNCT__
1196aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental"
1197aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type)
1198aae2c449SHong Zhang {
1199aae2c449SHong Zhang   PetscFunctionBegin;
1200aae2c449SHong Zhang   /* Currently does nothing */
1201aae2c449SHong Zhang   PetscFunctionReturn(0);
1202aae2c449SHong Zhang }
1203aae2c449SHong Zhang 
12047b30e0f1SHong Zhang #undef __FUNCT__
12057b30e0f1SHong Zhang #define __FUNCT__ "MatLoad_Elemental"
12067b30e0f1SHong Zhang PetscErrorCode MatLoad_Elemental(Mat newMat, PetscViewer viewer)
12077b30e0f1SHong Zhang {
12087b30e0f1SHong Zhang   PetscErrorCode ierr;
12097b30e0f1SHong Zhang   Mat            Adense,Ae;
12107b30e0f1SHong Zhang   MPI_Comm       comm;
12117b30e0f1SHong Zhang 
12127b30e0f1SHong Zhang   PetscFunctionBegin;
12137b30e0f1SHong Zhang   ierr = PetscObjectGetComm((PetscObject)newMat,&comm);CHKERRQ(ierr);
12147b30e0f1SHong Zhang   ierr = MatCreate(comm,&Adense);CHKERRQ(ierr);
12157b30e0f1SHong Zhang   ierr = MatSetType(Adense,MATDENSE);CHKERRQ(ierr);
12167b30e0f1SHong Zhang   ierr = MatLoad(Adense,viewer);CHKERRQ(ierr);
12177b30e0f1SHong Zhang   ierr = MatConvert(Adense, MATELEMENTAL, MAT_INITIAL_MATRIX,&Ae);CHKERRQ(ierr);
12187b30e0f1SHong Zhang   ierr = MatDestroy(&Adense);CHKERRQ(ierr);
121928be2f97SBarry Smith   ierr = MatHeaderReplace(newMat,&Ae);CHKERRQ(ierr);
12207b30e0f1SHong Zhang   PetscFunctionReturn(0);
12217b30e0f1SHong Zhang }
12227b30e0f1SHong Zhang 
12231d08bef3SHong Zhang #undef __FUNCT__
1224d5560a0bSSatish Balay #define __FUNCT__ "MatElementalHermitianGenDefEig_Elemental"
12252ef15aa2SHong Zhang PetscErrorCode MatElementalHermitianGenDefEig_Elemental(El::Pencil eigtype,El::UpperOrLower uplo,Mat A,Mat B,Mat *evals,Mat *evec,El::SortType sort,El::HermitianEigSubset<PetscElemScalar> subset,const El::HermitianEigCtrl<PetscElemScalar> ctrl)
12261d08bef3SHong Zhang {
12275dd12673SHong Zhang   PetscErrorCode ierr;
1228874e4880SHong Zhang   Mat_Elemental  *a=(Mat_Elemental*)A->data,*b=(Mat_Elemental*)B->data,*x;
1229874e4880SHong Zhang   MPI_Comm       comm;
1230874e4880SHong Zhang   Mat            EVAL;
12312ef15aa2SHong Zhang   Mat_Elemental  *e;
1232874e4880SHong Zhang 
12331d08bef3SHong Zhang   PetscFunctionBegin;
1234874e4880SHong Zhang   /* Compute eigenvalues and eigenvectors */
1235d5560a0bSSatish Balay   El::DistMatrix<PetscElemScalar,El::VR,El::STAR> w( *a->grid ); /* holding eigenvalues */
1236d5560a0bSSatish Balay   El::DistMatrix<PetscElemScalar>                 X( *a->grid ); /* holding eigenvectors */
1237935dd83bSHong Zhang   El::HermitianGenDefEig(eigtype,uplo,*a->emat,*b->emat,w,X,sort,subset,ctrl);
1238d5560a0bSSatish Balay   /* El::Print(w, "Eigenvalues"); */
12395dd12673SHong Zhang 
1240874e4880SHong Zhang   /* Wrap w and X into PETSc's MATMATELEMENTAL matrices */
1241d59fe0e1SHong Zhang   ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr);
1242d59fe0e1SHong Zhang   ierr = MatCreate(comm,evec);CHKERRQ(ierr);
1243d59fe0e1SHong Zhang   ierr = MatSetSizes(*evec,PETSC_DECIDE,PETSC_DECIDE,X.Height(),X.Width());CHKERRQ(ierr);
1244d59fe0e1SHong Zhang   ierr = MatSetType(*evec,MATELEMENTAL);CHKERRQ(ierr);
1245d59fe0e1SHong Zhang   ierr = MatSetFromOptions(*evec);CHKERRQ(ierr);
1246d59fe0e1SHong Zhang   ierr = MatSetUp(*evec);CHKERRQ(ierr);
1247d59fe0e1SHong Zhang   ierr = MatAssemblyBegin(*evec,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1248d59fe0e1SHong Zhang   ierr = MatAssemblyEnd(*evec,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1249d59fe0e1SHong Zhang 
1250874e4880SHong Zhang   x = (Mat_Elemental*)(*evec)->data;
1251d59fe0e1SHong Zhang   //delete x->emat; //-- memory leak???
1252d59fe0e1SHong Zhang   *x->emat = X;
1253d59fe0e1SHong Zhang 
1254874e4880SHong Zhang   ierr = MatCreate(comm,&EVAL);CHKERRQ(ierr);
1255874e4880SHong Zhang   ierr = MatSetSizes(EVAL,PETSC_DECIDE,PETSC_DECIDE,w.Height(),w.Width());CHKERRQ(ierr);
1256874e4880SHong Zhang   ierr = MatSetType(EVAL,MATELEMENTAL);CHKERRQ(ierr);
1257874e4880SHong Zhang   ierr = MatSetFromOptions(EVAL);CHKERRQ(ierr);
1258874e4880SHong Zhang   ierr = MatSetUp(EVAL);CHKERRQ(ierr);
1259874e4880SHong Zhang   ierr = MatAssemblyBegin(EVAL,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1260874e4880SHong Zhang   ierr = MatAssemblyEnd(EVAL,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
12612ef15aa2SHong Zhang   e         = (Mat_Elemental*)EVAL->data;
1262874e4880SHong Zhang   *e->emat = w; //-- memory leak???
1263874e4880SHong Zhang   *evals   = EVAL;
1264874e4880SHong Zhang 
1265874e4880SHong Zhang #if defined(MV)
1266f82baa17SHong Zhang   /* Test correctness norm = || - A*X + B*X*w || */
1267d59fe0e1SHong Zhang   {
1268f82baa17SHong Zhang     PetscElemScalar alpha,beta;
1269d5560a0bSSatish Balay     El::DistMatrix<PetscElemScalar> Y(*a->grid); //tmp matrix
1270f82baa17SHong Zhang     alpha = 1.0; beta=0.0;
1271d5560a0bSSatish Balay     El::Gemm(El::NORMAL,El::NORMAL,alpha,*b->emat,X,beta,Y); //Y = B*X
1272d5560a0bSSatish Balay     El::DiagonalScale(El::RIGHT,El::NORMAL, w, Y); //Y = Y*w
1273f82baa17SHong Zhang     alpha = -1.0; beta=1.0;
1274d5560a0bSSatish Balay     El::Gemm(El::NORMAL,El::NORMAL,alpha,*a->emat,X,beta,Y); //Y = - A*X + B*X*w
1275f82baa17SHong Zhang 
1276d5560a0bSSatish Balay     PetscElemScalar norm = El::FrobeniusNorm(Y);
1277d59fe0e1SHong Zhang     if ((*a->grid).Rank()==0) printf("  norm (- A*X + B*X*w) = %g\n",norm);
1278d59fe0e1SHong Zhang   }
1279f82baa17SHong Zhang 
1280874e4880SHong Zhang   {
1281874e4880SHong Zhang     PetscMPIInt rank;
1282874e4880SHong Zhang     ierr = MPI_Comm_rank(comm,&rank);
1283874e4880SHong Zhang     printf("w: [%d] [%d, %d %d] %d; X: %d %d\n",rank,w.DistRank(),w.ColRank(),w.RowRank(),w.LocalHeight(),X.LocalHeight(),X.LocalWidth());
12845dd12673SHong Zhang   }
1285874e4880SHong Zhang #endif
12861d08bef3SHong Zhang   PetscFunctionReturn(0);
12871d08bef3SHong Zhang }
12881d08bef3SHong Zhang 
12891d08bef3SHong Zhang #undef __FUNCT__
1290d5560a0bSSatish Balay #define __FUNCT__ "MatElementalHermitianGenDefEig"
12911d08bef3SHong Zhang /*@
12922ef15aa2SHong Zhang   MatElementalHermitianGenDefEig - Compute the set of eigenvalues of the Hermitian-definite matrix pencil determined by the subset structure
12931d08bef3SHong Zhang 
12941d08bef3SHong Zhang    Logically Collective on Mat
12951d08bef3SHong Zhang 
12961d08bef3SHong Zhang    Level: beginner
12971d08bef3SHong Zhang 
129896a0c994SBarry Smith    References:
129996a0c994SBarry Smith .      Elemental Users' Guide
130096a0c994SBarry Smith 
13011d08bef3SHong Zhang @*/
13022ef15aa2SHong Zhang PetscErrorCode MatElementalHermitianGenDefEig(El::Pencil type,El::UpperOrLower uplo,Mat A,Mat B,Mat *evals,Mat *evec,El::SortType sort,El::HermitianEigSubset<PetscElemScalar> subset,const El::HermitianEigCtrl<PetscElemScalar> ctrl)
13031d08bef3SHong Zhang {
13041d08bef3SHong Zhang   PetscErrorCode ierr;
13051d08bef3SHong Zhang 
13061d08bef3SHong Zhang   PetscFunctionBegin;
1307163d334eSBarry Smith   ierr = PetscUseMethod(A,"MatElementalHermitianGenDefEig_C",(El::Pencil,El::UpperOrLower,Mat,Mat,Mat*,Mat*,El::SortType,El::HermitianEigSubset<PetscElemScalar>,const El::HermitianEigCtrl<PetscElemScalar>),(type,uplo,A,B,evals,evec,sort,subset,ctrl));CHKERRQ(ierr);
13081d08bef3SHong Zhang   PetscFunctionReturn(0);
13091d08bef3SHong Zhang }
13101d08bef3SHong Zhang 
131140d92e34SHong Zhang /* -------------------------------------------------------------------*/
131240d92e34SHong Zhang static struct _MatOps MatOps_Values = {
131340d92e34SHong Zhang        MatSetValues_Elemental,
131440d92e34SHong Zhang        0,
131540d92e34SHong Zhang        0,
131640d92e34SHong Zhang        MatMult_Elemental,
131740d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental,
13189426833fSXuan Zhou        MatMultTranspose_Elemental,
1319e883f9d5SXuan Zhou        MatMultTransposeAdd_Elemental,
132040d92e34SHong Zhang        MatSolve_Elemental,
1321df311e6cSXuan Zhou        MatSolveAdd_Elemental,
13222ef15aa2SHong Zhang        0,
13232ef15aa2SHong Zhang /*10*/ 0,
132440d92e34SHong Zhang        MatLUFactor_Elemental,
132540d92e34SHong Zhang        MatCholeskyFactor_Elemental,
132640d92e34SHong Zhang        0,
132740d92e34SHong Zhang        MatTranspose_Elemental,
132840d92e34SHong Zhang /*15*/ MatGetInfo_Elemental,
132940d92e34SHong Zhang        0,
133061119200SXuan Zhou        MatGetDiagonal_Elemental,
1331ade3cc5eSXuan Zhou        MatDiagonalScale_Elemental,
133240d92e34SHong Zhang        MatNorm_Elemental,
133340d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental,
133440d92e34SHong Zhang        MatAssemblyEnd_Elemental,
133532d7a744SHong Zhang        MatSetOption_Elemental,
133640d92e34SHong Zhang        MatZeroEntries_Elemental,
133740d92e34SHong Zhang /*24*/ 0,
133840d92e34SHong Zhang        MatLUFactorSymbolic_Elemental,
133940d92e34SHong Zhang        MatLUFactorNumeric_Elemental,
134040d92e34SHong Zhang        MatCholeskyFactorSymbolic_Elemental,
134140d92e34SHong Zhang        MatCholeskyFactorNumeric_Elemental,
134240d92e34SHong Zhang /*29*/ MatSetUp_Elemental,
134340d92e34SHong Zhang        0,
134440d92e34SHong Zhang        0,
134540d92e34SHong Zhang        0,
134640d92e34SHong Zhang        0,
1347df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental,
134840d92e34SHong Zhang        0,
134940d92e34SHong Zhang        0,
135040d92e34SHong Zhang        0,
135140d92e34SHong Zhang        0,
135240d92e34SHong Zhang /*39*/ MatAXPY_Elemental,
135340d92e34SHong Zhang        0,
135440d92e34SHong Zhang        0,
135540d92e34SHong Zhang        0,
135640d92e34SHong Zhang        MatCopy_Elemental,
135740d92e34SHong Zhang /*44*/ 0,
135840d92e34SHong Zhang        MatScale_Elemental,
13597d68702bSBarry Smith        MatShift_Basic,
136040d92e34SHong Zhang        0,
136140d92e34SHong Zhang        0,
136240d92e34SHong Zhang /*49*/ 0,
136340d92e34SHong Zhang        0,
136440d92e34SHong Zhang        0,
136540d92e34SHong Zhang        0,
136640d92e34SHong Zhang        0,
136740d92e34SHong Zhang /*54*/ 0,
136840d92e34SHong Zhang        0,
136940d92e34SHong Zhang        0,
137040d92e34SHong Zhang        0,
137140d92e34SHong Zhang        0,
137240d92e34SHong Zhang /*59*/ 0,
137340d92e34SHong Zhang        MatDestroy_Elemental,
137440d92e34SHong Zhang        MatView_Elemental,
137540d92e34SHong Zhang        0,
137640d92e34SHong Zhang        0,
137740d92e34SHong Zhang /*64*/ 0,
137840d92e34SHong Zhang        0,
137940d92e34SHong Zhang        0,
138040d92e34SHong Zhang        0,
138140d92e34SHong Zhang        0,
138240d92e34SHong Zhang /*69*/ 0,
138340d92e34SHong Zhang        0,
13842ef0cf24SXuan Zhou        MatConvert_Elemental_Dense,
138540d92e34SHong Zhang        0,
138640d92e34SHong Zhang        0,
138740d92e34SHong Zhang /*74*/ 0,
138840d92e34SHong Zhang        0,
138940d92e34SHong Zhang        0,
139040d92e34SHong Zhang        0,
139140d92e34SHong Zhang        0,
139240d92e34SHong Zhang /*79*/ 0,
139340d92e34SHong Zhang        0,
139440d92e34SHong Zhang        0,
139540d92e34SHong Zhang        0,
13967b30e0f1SHong Zhang        MatLoad_Elemental,
139740d92e34SHong Zhang /*84*/ 0,
139840d92e34SHong Zhang        0,
139940d92e34SHong Zhang        0,
140040d92e34SHong Zhang        0,
140140d92e34SHong Zhang        0,
140240d92e34SHong Zhang /*89*/ MatMatMult_Elemental,
140340d92e34SHong Zhang        MatMatMultSymbolic_Elemental,
140440d92e34SHong Zhang        MatMatMultNumeric_Elemental,
140540d92e34SHong Zhang        0,
140640d92e34SHong Zhang        0,
140740d92e34SHong Zhang /*94*/ 0,
1408df311e6cSXuan Zhou        MatMatTransposeMult_Elemental,
1409df311e6cSXuan Zhou        MatMatTransposeMultSymbolic_Elemental,
1410df311e6cSXuan Zhou        MatMatTransposeMultNumeric_Elemental,
141140d92e34SHong Zhang        0,
141240d92e34SHong Zhang /*99*/ 0,
141340d92e34SHong Zhang        0,
141440d92e34SHong Zhang        0,
1415dfcb0403SXuan Zhou        MatConjugate_Elemental,
141640d92e34SHong Zhang        0,
141740d92e34SHong Zhang /*104*/0,
141840d92e34SHong Zhang        0,
141940d92e34SHong Zhang        0,
142040d92e34SHong Zhang        0,
142140d92e34SHong Zhang        0,
142240d92e34SHong Zhang /*109*/MatMatSolve_Elemental,
142340d92e34SHong Zhang        0,
142440d92e34SHong Zhang        0,
142540d92e34SHong Zhang        0,
142640d92e34SHong Zhang        0,
142740d92e34SHong Zhang /*114*/0,
142840d92e34SHong Zhang        0,
142940d92e34SHong Zhang        0,
143040d92e34SHong Zhang        0,
143140d92e34SHong Zhang        0,
143240d92e34SHong Zhang /*119*/0,
14334a29722dSXuan Zhou        MatHermitianTranspose_Elemental,
143440d92e34SHong Zhang        0,
143540d92e34SHong Zhang        0,
143640d92e34SHong Zhang        0,
143740d92e34SHong Zhang /*124*/0,
143840d92e34SHong Zhang        0,
143940d92e34SHong Zhang        0,
144040d92e34SHong Zhang        0,
144140d92e34SHong Zhang        0,
144240d92e34SHong Zhang /*129*/0,
144340d92e34SHong Zhang        0,
144440d92e34SHong Zhang        0,
144540d92e34SHong Zhang        0,
144640d92e34SHong Zhang        0,
144740d92e34SHong Zhang /*134*/0,
144840d92e34SHong Zhang        0,
144940d92e34SHong Zhang        0,
145040d92e34SHong Zhang        0,
145140d92e34SHong Zhang        0
145240d92e34SHong Zhang };
145340d92e34SHong Zhang 
1454ed36708cSHong Zhang /*MC
1455ed36708cSHong Zhang    MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package
1456ed36708cSHong Zhang 
1457c2b89b5dSBarry Smith   Use ./configure --download-elemental to install PETSc to use Elemental
1458c2b89b5dSBarry Smith 
1459c2b89b5dSBarry Smith   Use -pc_type lu -pc_factor_mat_solver_package elemental to us this direct solver
1460c2b89b5dSBarry Smith 
1461ed36708cSHong Zhang    Options Database Keys:
14625cc86fc1SJed Brown + -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions()
14635cc86fc1SJed Brown - -mat_elemental_grid_height - sets Grid Height for 2D cyclic ordering of internal matrix
1464ed36708cSHong Zhang 
1465ed36708cSHong Zhang   Level: beginner
1466ed36708cSHong Zhang 
14675cb544a0SHong Zhang .seealso: MATDENSE
1468ed36708cSHong Zhang M*/
14694a29722dSXuan Zhou 
1470db31f6deSJed Brown #undef __FUNCT__
1471db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental"
14728cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_Elemental(Mat A)
1473db31f6deSJed Brown {
1474db31f6deSJed Brown   Mat_Elemental      *a;
1475db31f6deSJed Brown   PetscErrorCode     ierr;
14765682a260SJack Poulson   PetscBool          flg,flg1;
14775e9f5b67SHong Zhang   Mat_Elemental_Grid *commgrid;
14785e9f5b67SHong Zhang   MPI_Comm           icomm;
14795682a260SJack Poulson   PetscInt           optv1;
1480db31f6deSJed Brown 
1481db31f6deSJed Brown   PetscFunctionBegin;
1482607a6623SBarry Smith   ierr = PetscElementalInitializePackage();CHKERRQ(ierr);
148340d92e34SHong Zhang   ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
148440d92e34SHong Zhang   A->insertmode = NOT_SET_VALUES;
1485db31f6deSJed Brown 
1486b00a9115SJed Brown   ierr = PetscNewLog(A,&a);CHKERRQ(ierr);
1487db31f6deSJed Brown   A->data = (void*)a;
1488db31f6deSJed Brown 
1489db31f6deSJed Brown   /* Set up the elemental matrix */
1490e8146892SSatish Balay   El::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A));
14915e9f5b67SHong Zhang 
14925e9f5b67SHong Zhang   /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */
14935e9f5b67SHong Zhang   if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) {
1494180a43e4SHong Zhang     ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0);
1495a000e53bSHong Zhang     /* ierr = MPI_Comm_create_Keyval(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); -- new version? */
14965e9f5b67SHong Zhang   }
14970c18141cSBarry Smith   ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr);
14985e9f5b67SHong Zhang   ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr);
14995e9f5b67SHong Zhang   if (!flg) {
1500b00a9115SJed Brown     ierr = PetscNewLog(A,&commgrid);CHKERRQ(ierr);
15015cb544a0SHong Zhang 
1502ce94432eSBarry Smith     ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr);
1503e8146892SSatish Balay     /* displayed default grid sizes (CommSize,1) are set by us arbitrarily until El::Grid() is called */
1504e8146892SSatish Balay     ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",El::mpi::Size(cxxcomm),&optv1,&flg1);CHKERRQ(ierr);
15055682a260SJack Poulson     if (flg1) {
1506e8146892SSatish Balay       if (El::mpi::Size(cxxcomm) % optv1 != 0) {
1507e8146892SSatish Balay         SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Grid Height %D must evenly divide CommSize %D",optv1,(PetscInt)El::mpi::Size(cxxcomm));
1508ed667823SXuan Zhou       }
1509e8146892SSatish Balay       commgrid->grid = new El::Grid(cxxcomm,optv1); /* use user-provided grid height */
15102ef0cf24SXuan Zhou     } else {
1511e8146892SSatish Balay       commgrid->grid = new El::Grid(cxxcomm); /* use Elemental default grid sizes */
1512da0640a4SHong Zhang       /* printf("new commgrid->grid = %p\n",commgrid->grid);  -- memory leak revealed by valgrind? */
1513ed667823SXuan Zhou     }
15145e9f5b67SHong Zhang     commgrid->grid_refct = 1;
15155e9f5b67SHong Zhang     ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr);
15165cb544a0SHong Zhang     PetscOptionsEnd();
15175e9f5b67SHong Zhang   } else {
15185e9f5b67SHong Zhang     commgrid->grid_refct++;
15195e9f5b67SHong Zhang   }
15205e9f5b67SHong Zhang   ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr);
15215e9f5b67SHong Zhang   a->grid      = commgrid->grid;
1522e8146892SSatish Balay   a->emat      = new El::DistMatrix<PetscElemScalar>(*a->grid);
1523dc1b4460SJack Poulson   a->pivot     = new El::DistMatrix<PetscInt,El::VC,El::STAR>(*a->grid);
152432d7a744SHong Zhang   a->roworiented = PETSC_TRUE;
1525db31f6deSJed Brown 
1526bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",MatGetOwnershipIS_Elemental);CHKERRQ(ierr);
1527d5560a0bSSatish Balay   ierr = PetscObjectComposeFunction((PetscObject)A,"MatElementalHermitianGenDefEig_C",MatElementalHermitianGenDefEig_Elemental);CHKERRQ(ierr);
1528db31f6deSJed Brown 
1529db31f6deSJed Brown   ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr);
1530db31f6deSJed Brown   PetscFunctionReturn(0);
1531db31f6deSJed Brown }
1532