xref: /petsc/src/mat/impls/dense/mpi/mpidense.c (revision 719d5645761d844e4357b7ee00a3296dffe0b787)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2be1d678aSKris Buschelman 
3ed3cc1f0SBarry Smith /*
4ed3cc1f0SBarry Smith    Basic functions for basic parallel dense matrices.
5ed3cc1f0SBarry Smith */
6ed3cc1f0SBarry Smith 
704fea9ffSBarry Smith 
889ac3759SBarry Smith #include "src/mat/impls/dense/mpi/mpidense.h"    /*I   "petscmat.h"  I*/
9eae6fb2eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
10eae6fb2eSBarry Smith static PetscMPIInt Plapack_nprows,Plapack_npcols,Plapack_ierror,Plapack_nb_alg;
11eae6fb2eSBarry Smith static MPI_Comm Plapack_comm_2d;
12eae6fb2eSBarry Smith #endif
138965ea79SLois Curfman McInnes 
14ba8c8a56SBarry Smith #undef __FUNCT__
15ab92ecdeSBarry Smith #define __FUNCT__ "MatDenseGetLocalMatrix"
16ab92ecdeSBarry Smith /*@
17ab92ecdeSBarry Smith 
18ab92ecdeSBarry Smith       MatDenseGetLocalMatrix - For a MATMPIDENSE or MATSEQDENSE matrix returns the sequential
19ab92ecdeSBarry Smith               matrix that represents the operator. For sequential matrices it returns itself.
20ab92ecdeSBarry Smith 
21ab92ecdeSBarry Smith     Input Parameter:
22ab92ecdeSBarry Smith .      A - the Seq or MPI dense matrix
23ab92ecdeSBarry Smith 
24ab92ecdeSBarry Smith     Output Parameter:
25ab92ecdeSBarry Smith .      B - the inner matrix
26ab92ecdeSBarry Smith 
278e6c10adSSatish Balay     Level: intermediate
288e6c10adSSatish Balay 
29ab92ecdeSBarry Smith @*/
30ab92ecdeSBarry Smith PetscErrorCode MatDenseGetLocalMatrix(Mat A,Mat *B)
31ab92ecdeSBarry Smith {
32ab92ecdeSBarry Smith   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
33ab92ecdeSBarry Smith   PetscErrorCode ierr;
34ab92ecdeSBarry Smith   PetscTruth     flg;
35ab92ecdeSBarry Smith 
36ab92ecdeSBarry Smith   PetscFunctionBegin;
37ab92ecdeSBarry Smith   ierr = PetscTypeCompare((PetscObject)A,MATMPIDENSE,&flg);CHKERRQ(ierr);
38ab92ecdeSBarry Smith   if (flg) {
39ab92ecdeSBarry Smith     *B = mat->A;
40ab92ecdeSBarry Smith   } else {
41ab92ecdeSBarry Smith     *B = A;
42ab92ecdeSBarry Smith   }
43ab92ecdeSBarry Smith   PetscFunctionReturn(0);
44ab92ecdeSBarry Smith }
45ab92ecdeSBarry Smith 
46ab92ecdeSBarry Smith #undef __FUNCT__
47ba8c8a56SBarry Smith #define __FUNCT__ "MatGetRow_MPIDense"
48ba8c8a56SBarry Smith PetscErrorCode MatGetRow_MPIDense(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
49ba8c8a56SBarry Smith {
50ba8c8a56SBarry Smith   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
51ba8c8a56SBarry Smith   PetscErrorCode ierr;
52d0f46423SBarry Smith   PetscInt       lrow,rstart = A->rmap->rstart,rend = A->rmap->rend;
53ba8c8a56SBarry Smith 
54ba8c8a56SBarry Smith   PetscFunctionBegin;
55ba8c8a56SBarry Smith   if (row < rstart || row >= rend) SETERRQ(PETSC_ERR_SUP,"only local rows")
56ba8c8a56SBarry Smith   lrow = row - rstart;
57ba8c8a56SBarry Smith   ierr = MatGetRow(mat->A,lrow,nz,(const PetscInt **)idx,(const PetscScalar **)v);CHKERRQ(ierr);
58ba8c8a56SBarry Smith   PetscFunctionReturn(0);
59ba8c8a56SBarry Smith }
60ba8c8a56SBarry Smith 
61ba8c8a56SBarry Smith #undef __FUNCT__
62ba8c8a56SBarry Smith #define __FUNCT__ "MatRestoreRow_MPIDense"
63ba8c8a56SBarry Smith PetscErrorCode MatRestoreRow_MPIDense(Mat mat,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
64ba8c8a56SBarry Smith {
65ba8c8a56SBarry Smith   PetscErrorCode ierr;
66ba8c8a56SBarry Smith 
67ba8c8a56SBarry Smith   PetscFunctionBegin;
68ba8c8a56SBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
69ba8c8a56SBarry Smith   if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);}
70ba8c8a56SBarry Smith   PetscFunctionReturn(0);
71ba8c8a56SBarry Smith }
72ba8c8a56SBarry Smith 
730de54da6SSatish Balay EXTERN_C_BEGIN
744a2ae208SSatish Balay #undef __FUNCT__
754a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonalBlock_MPIDense"
76be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatGetDiagonalBlock_MPIDense(Mat A,PetscTruth *iscopy,MatReuse reuse,Mat *B)
770de54da6SSatish Balay {
780de54da6SSatish Balay   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
796849ba73SBarry Smith   PetscErrorCode ierr;
80d0f46423SBarry Smith   PetscInt       m = A->rmap->n,rstart = A->rmap->rstart;
8187828ca2SBarry Smith   PetscScalar    *array;
820de54da6SSatish Balay   MPI_Comm       comm;
830de54da6SSatish Balay 
840de54da6SSatish Balay   PetscFunctionBegin;
85d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_SUP,"Only square matrices supported.");
860de54da6SSatish Balay 
870de54da6SSatish Balay   /* The reuse aspect is not implemented efficiently */
880de54da6SSatish Balay   if (reuse) { ierr = MatDestroy(*B);CHKERRQ(ierr);}
890de54da6SSatish Balay 
900de54da6SSatish Balay   ierr = PetscObjectGetComm((PetscObject)(mdn->A),&comm);CHKERRQ(ierr);
910de54da6SSatish Balay   ierr = MatGetArray(mdn->A,&array);CHKERRQ(ierr);
92f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,B);CHKERRQ(ierr);
93f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*B,m,m,m,m);CHKERRQ(ierr);
947adad957SLisandro Dalcin   ierr = MatSetType(*B,((PetscObject)mdn->A)->type_name);CHKERRQ(ierr);
955c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(*B,array+m*rstart);CHKERRQ(ierr);
960de54da6SSatish Balay   ierr = MatRestoreArray(mdn->A,&array);CHKERRQ(ierr);
970de54da6SSatish Balay   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
980de54da6SSatish Balay   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
990de54da6SSatish Balay 
1000de54da6SSatish Balay   *iscopy = PETSC_TRUE;
1010de54da6SSatish Balay   PetscFunctionReturn(0);
1020de54da6SSatish Balay }
1030de54da6SSatish Balay EXTERN_C_END
1040de54da6SSatish Balay 
1054a2ae208SSatish Balay #undef __FUNCT__
1064a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_MPIDense"
10713f74950SBarry Smith PetscErrorCode MatSetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1088965ea79SLois Curfman McInnes {
10939b7565bSBarry Smith   Mat_MPIDense   *A = (Mat_MPIDense*)mat->data;
110dfbe8321SBarry Smith   PetscErrorCode ierr;
111d0f46423SBarry Smith   PetscInt       i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row;
112273d9f13SBarry Smith   PetscTruth     roworiented = A->roworiented;
1138965ea79SLois Curfman McInnes 
1143a40ed3dSBarry Smith   PetscFunctionBegin;
1158965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
1165ef9f2a5SBarry Smith     if (idxm[i] < 0) continue;
117d0f46423SBarry Smith     if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large");
1188965ea79SLois Curfman McInnes     if (idxm[i] >= rstart && idxm[i] < rend) {
1198965ea79SLois Curfman McInnes       row = idxm[i] - rstart;
12039b7565bSBarry Smith       if (roworiented) {
12139b7565bSBarry Smith         ierr = MatSetValues(A->A,1,&row,n,idxn,v+i*n,addv);CHKERRQ(ierr);
1223a40ed3dSBarry Smith       } else {
1238965ea79SLois Curfman McInnes         for (j=0; j<n; j++) {
1245ef9f2a5SBarry Smith           if (idxn[j] < 0) continue;
125d0f46423SBarry Smith           if (idxn[j] >= mat->cmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large");
12639b7565bSBarry Smith           ierr = MatSetValues(A->A,1,&row,1,&idxn[j],v+i+j*m,addv);CHKERRQ(ierr);
12739b7565bSBarry Smith         }
1288965ea79SLois Curfman McInnes       }
1293a40ed3dSBarry Smith     } else {
1303782ba37SSatish Balay       if (!A->donotstash) {
13139b7565bSBarry Smith         if (roworiented) {
1328798bf22SSatish Balay           ierr = MatStashValuesRow_Private(&mat->stash,idxm[i],n,idxn,v+i*n);CHKERRQ(ierr);
133d36fbae8SSatish Balay         } else {
1348798bf22SSatish Balay           ierr = MatStashValuesCol_Private(&mat->stash,idxm[i],n,idxn,v+i,m);CHKERRQ(ierr);
13539b7565bSBarry Smith         }
136b49de8d1SLois Curfman McInnes       }
137b49de8d1SLois Curfman McInnes     }
1383782ba37SSatish Balay   }
1393a40ed3dSBarry Smith   PetscFunctionReturn(0);
140b49de8d1SLois Curfman McInnes }
141b49de8d1SLois Curfman McInnes 
1424a2ae208SSatish Balay #undef __FUNCT__
1434a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_MPIDense"
14413f74950SBarry Smith PetscErrorCode MatGetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
145b49de8d1SLois Curfman McInnes {
146b49de8d1SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
147dfbe8321SBarry Smith   PetscErrorCode ierr;
148d0f46423SBarry Smith   PetscInt       i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row;
149b49de8d1SLois Curfman McInnes 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
151b49de8d1SLois Curfman McInnes   for (i=0; i<m; i++) {
15297e567efSBarry Smith     if (idxm[i] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative row"); */
153d0f46423SBarry Smith     if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large");
154b49de8d1SLois Curfman McInnes     if (idxm[i] >= rstart && idxm[i] < rend) {
155b49de8d1SLois Curfman McInnes       row = idxm[i] - rstart;
156b49de8d1SLois Curfman McInnes       for (j=0; j<n; j++) {
15797e567efSBarry Smith         if (idxn[j] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative column"); */
158d0f46423SBarry Smith         if (idxn[j] >= mat->cmap->N) {
15929bbc08cSBarry Smith           SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large");
160a8c6a408SBarry Smith         }
161b49de8d1SLois Curfman McInnes         ierr = MatGetValues(mdn->A,1,&row,1,&idxn[j],v+i*n+j);CHKERRQ(ierr);
162b49de8d1SLois Curfman McInnes       }
163a8c6a408SBarry Smith     } else {
16429bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"Only local values currently supported");
1658965ea79SLois Curfman McInnes     }
1668965ea79SLois Curfman McInnes   }
1673a40ed3dSBarry Smith   PetscFunctionReturn(0);
1688965ea79SLois Curfman McInnes }
1698965ea79SLois Curfman McInnes 
1704a2ae208SSatish Balay #undef __FUNCT__
1714a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_MPIDense"
172dfbe8321SBarry Smith PetscErrorCode MatGetArray_MPIDense(Mat A,PetscScalar *array[])
173ff14e315SSatish Balay {
174ff14e315SSatish Balay   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
175dfbe8321SBarry Smith   PetscErrorCode ierr;
176ff14e315SSatish Balay 
1773a40ed3dSBarry Smith   PetscFunctionBegin;
178ff14e315SSatish Balay   ierr = MatGetArray(a->A,array);CHKERRQ(ierr);
1793a40ed3dSBarry Smith   PetscFunctionReturn(0);
180ff14e315SSatish Balay }
181ff14e315SSatish Balay 
1824a2ae208SSatish Balay #undef __FUNCT__
1834a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_MPIDense"
18413f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_MPIDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B)
185ca3fa75bSLois Curfman McInnes {
186ca3fa75bSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data,*newmatd;
187ca3fa75bSLois Curfman McInnes   Mat_SeqDense   *lmat = (Mat_SeqDense*)mat->A->data;
1886849ba73SBarry Smith   PetscErrorCode ierr;
18913f74950SBarry Smith   PetscInt       i,j,*irow,*icol,rstart,rend,nrows,ncols,nlrows,nlcols;
19087828ca2SBarry Smith   PetscScalar    *av,*bv,*v = lmat->v;
191ca3fa75bSLois Curfman McInnes   Mat            newmat;
192ca3fa75bSLois Curfman McInnes 
193ca3fa75bSLois Curfman McInnes   PetscFunctionBegin;
194ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
195ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
196b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
197b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
198ca3fa75bSLois Curfman McInnes 
199ca3fa75bSLois Curfman McInnes   /* No parallel redistribution currently supported! Should really check each index set
2007eba5e9cSLois Curfman McInnes      to comfirm that it is OK.  ... Currently supports only submatrix same partitioning as
2017eba5e9cSLois Curfman McInnes      original matrix! */
202ca3fa75bSLois Curfman McInnes 
203ca3fa75bSLois Curfman McInnes   ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr);
2047eba5e9cSLois Curfman McInnes   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
205ca3fa75bSLois Curfman McInnes 
206ca3fa75bSLois Curfman McInnes   /* Check submatrix call */
207ca3fa75bSLois Curfman McInnes   if (scall == MAT_REUSE_MATRIX) {
20829bbc08cSBarry Smith     /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */
2097eba5e9cSLois Curfman McInnes     /* Really need to test rows and column sizes! */
210ca3fa75bSLois Curfman McInnes     newmat = *B;
211ca3fa75bSLois Curfman McInnes   } else {
212ca3fa75bSLois Curfman McInnes     /* Create and fill new matrix */
2137adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
214f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,cs,PETSC_DECIDE,ncols);CHKERRQ(ierr);
2157adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
216878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
217ca3fa75bSLois Curfman McInnes   }
218ca3fa75bSLois Curfman McInnes 
219ca3fa75bSLois Curfman McInnes   /* Now extract the data pointers and do the copy, column at a time */
220ca3fa75bSLois Curfman McInnes   newmatd = (Mat_MPIDense*)newmat->data;
221ca3fa75bSLois Curfman McInnes   bv      = ((Mat_SeqDense *)newmatd->A->data)->v;
222ca3fa75bSLois Curfman McInnes 
223ca3fa75bSLois Curfman McInnes   for (i=0; i<ncols; i++) {
224ca3fa75bSLois Curfman McInnes     av = v + nlrows*icol[i];
225ca3fa75bSLois Curfman McInnes     for (j=0; j<nrows; j++) {
2267eba5e9cSLois Curfman McInnes       *bv++ = av[irow[j] - rstart];
227ca3fa75bSLois Curfman McInnes     }
228ca3fa75bSLois Curfman McInnes   }
229ca3fa75bSLois Curfman McInnes 
230ca3fa75bSLois Curfman McInnes   /* Assemble the matrices so that the correct flags are set */
231ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
232ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
233ca3fa75bSLois Curfman McInnes 
234ca3fa75bSLois Curfman McInnes   /* Free work space */
235ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
236ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
237ca3fa75bSLois Curfman McInnes   *B = newmat;
238ca3fa75bSLois Curfman McInnes   PetscFunctionReturn(0);
239ca3fa75bSLois Curfman McInnes }
240ca3fa75bSLois Curfman McInnes 
2414a2ae208SSatish Balay #undef __FUNCT__
2424a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense"
243dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[])
244ff14e315SSatish Balay {
2453a40ed3dSBarry Smith   PetscFunctionBegin;
2463a40ed3dSBarry Smith   PetscFunctionReturn(0);
247ff14e315SSatish Balay }
248ff14e315SSatish Balay 
2494a2ae208SSatish Balay #undef __FUNCT__
2504a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense"
251dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode)
2528965ea79SLois Curfman McInnes {
25339ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
2547adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)mat)->comm;
255dfbe8321SBarry Smith   PetscErrorCode ierr;
25613f74950SBarry Smith   PetscInt       nstash,reallocs;
2578965ea79SLois Curfman McInnes   InsertMode     addv;
2588965ea79SLois Curfman McInnes 
2593a40ed3dSBarry Smith   PetscFunctionBegin;
2608965ea79SLois Curfman McInnes   /* make sure all processors are either in INSERTMODE or ADDMODE */
261ca161407SBarry Smith   ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr);
2627056b6fcSBarry Smith   if (addv == (ADD_VALUES|INSERT_VALUES)) {
26329bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs");
2648965ea79SLois Curfman McInnes   }
265e0fa3b82SLois Curfman McInnes   mat->insertmode = addv; /* in case this processor had no cache */
2668965ea79SLois Curfman McInnes 
267d0f46423SBarry Smith   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr);
2688798bf22SSatish Balay   ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr);
269ae15b995SBarry Smith   ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr);
2703a40ed3dSBarry Smith   PetscFunctionReturn(0);
2718965ea79SLois Curfman McInnes }
2728965ea79SLois Curfman McInnes 
2734a2ae208SSatish Balay #undef __FUNCT__
2744a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense"
275dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode)
2768965ea79SLois Curfman McInnes {
27739ddd567SLois Curfman McInnes   Mat_MPIDense    *mdn=(Mat_MPIDense*)mat->data;
2786849ba73SBarry Smith   PetscErrorCode  ierr;
27913f74950SBarry Smith   PetscInt        i,*row,*col,flg,j,rstart,ncols;
28013f74950SBarry Smith   PetscMPIInt     n;
28187828ca2SBarry Smith   PetscScalar     *val;
282e0fa3b82SLois Curfman McInnes   InsertMode      addv=mat->insertmode;
2838965ea79SLois Curfman McInnes 
2843a40ed3dSBarry Smith   PetscFunctionBegin;
2858965ea79SLois Curfman McInnes   /*  wait on receives */
2867ef1d9bdSSatish Balay   while (1) {
2878798bf22SSatish Balay     ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr);
2887ef1d9bdSSatish Balay     if (!flg) break;
2898965ea79SLois Curfman McInnes 
2907ef1d9bdSSatish Balay     for (i=0; i<n;) {
2917ef1d9bdSSatish Balay       /* Now identify the consecutive vals belonging to the same row */
2927ef1d9bdSSatish Balay       for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; }
2937ef1d9bdSSatish Balay       if (j < n) ncols = j-i;
2947ef1d9bdSSatish Balay       else       ncols = n-i;
2957ef1d9bdSSatish Balay       /* Now assemble all these values with a single function call */
2967ef1d9bdSSatish Balay       ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr);
2977ef1d9bdSSatish Balay       i = j;
2988965ea79SLois Curfman McInnes     }
2997ef1d9bdSSatish Balay   }
3008798bf22SSatish Balay   ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr);
3018965ea79SLois Curfman McInnes 
30239ddd567SLois Curfman McInnes   ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr);
30339ddd567SLois Curfman McInnes   ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr);
3048965ea79SLois Curfman McInnes 
3056d4a8577SBarry Smith   if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) {
30639ddd567SLois Curfman McInnes     ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
3078965ea79SLois Curfman McInnes   }
3083a40ed3dSBarry Smith   PetscFunctionReturn(0);
3098965ea79SLois Curfman McInnes }
3108965ea79SLois Curfman McInnes 
3114a2ae208SSatish Balay #undef __FUNCT__
3124a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense"
313dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A)
3148965ea79SLois Curfman McInnes {
315dfbe8321SBarry Smith   PetscErrorCode ierr;
31639ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3173a40ed3dSBarry Smith 
3183a40ed3dSBarry Smith   PetscFunctionBegin;
3193a40ed3dSBarry Smith   ierr = MatZeroEntries(l->A);CHKERRQ(ierr);
3203a40ed3dSBarry Smith   PetscFunctionReturn(0);
3218965ea79SLois Curfman McInnes }
3228965ea79SLois Curfman McInnes 
3238965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the
3248965ea79SLois Curfman McInnes    matrix is square and the column and row owerships are identical.
3258965ea79SLois Curfman McInnes    This is a BUG. The only way to fix it seems to be to access
3263501a2bdSLois Curfman McInnes    mdn->A and mdn->B directly and not through the MatZeroRows()
3278965ea79SLois Curfman McInnes    routine.
3288965ea79SLois Curfman McInnes */
3294a2ae208SSatish Balay #undef __FUNCT__
3304a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense"
331f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
3328965ea79SLois Curfman McInnes {
33339ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3346849ba73SBarry Smith   PetscErrorCode ierr;
335d0f46423SBarry Smith   PetscInt       i,*owners = A->rmap->range;
33613f74950SBarry Smith   PetscInt       *nprocs,j,idx,nsends;
33713f74950SBarry Smith   PetscInt       nmax,*svalues,*starts,*owner,nrecvs;
3387adad957SLisandro Dalcin   PetscInt       *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source;
33913f74950SBarry Smith   PetscInt       *lens,*lrows,*values;
34013f74950SBarry Smith   PetscMPIInt    n,imdex,rank = l->rank,size = l->size;
3417adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)A)->comm;
3428965ea79SLois Curfman McInnes   MPI_Request    *send_waits,*recv_waits;
3438965ea79SLois Curfman McInnes   MPI_Status     recv_status,*send_status;
34435d8aa7fSBarry Smith   PetscTruth     found;
3458965ea79SLois Curfman McInnes 
3463a40ed3dSBarry Smith   PetscFunctionBegin;
3478965ea79SLois Curfman McInnes   /*  first count number of contributors to each processor */
34813f74950SBarry Smith   ierr  = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
34913f74950SBarry Smith   ierr  = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
35013f74950SBarry Smith   ierr  = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/
3518965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3528965ea79SLois Curfman McInnes     idx = rows[i];
35335d8aa7fSBarry Smith     found = PETSC_FALSE;
3548965ea79SLois Curfman McInnes     for (j=0; j<size; j++) {
3558965ea79SLois Curfman McInnes       if (idx >= owners[j] && idx < owners[j+1]) {
356c1dc657dSBarry Smith         nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break;
3578965ea79SLois Curfman McInnes       }
3588965ea79SLois Curfman McInnes     }
35929bbc08cSBarry Smith     if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range");
3608965ea79SLois Curfman McInnes   }
361c1dc657dSBarry Smith   nsends = 0;  for (i=0; i<size; i++) { nsends += nprocs[2*i+1];}
3628965ea79SLois Curfman McInnes 
3638965ea79SLois Curfman McInnes   /* inform other processors of number of messages and max length*/
364c1dc657dSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
3658965ea79SLois Curfman McInnes 
3668965ea79SLois Curfman McInnes   /* post receives:   */
36713f74950SBarry Smith   ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr);
368b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
3698965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
37013f74950SBarry Smith     ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr);
3718965ea79SLois Curfman McInnes   }
3728965ea79SLois Curfman McInnes 
3738965ea79SLois Curfman McInnes   /* do sends:
3748965ea79SLois Curfman McInnes       1) starts[i] gives the starting index in svalues for stuff going to
3758965ea79SLois Curfman McInnes          the ith processor
3768965ea79SLois Curfman McInnes   */
37713f74950SBarry Smith   ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr);
378b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
37913f74950SBarry Smith   ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
3808965ea79SLois Curfman McInnes   starts[0]  = 0;
381c1dc657dSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3828965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3838965ea79SLois Curfman McInnes     svalues[starts[owner[i]]++] = rows[i];
3848965ea79SLois Curfman McInnes   }
3858965ea79SLois Curfman McInnes 
3868965ea79SLois Curfman McInnes   starts[0] = 0;
387c1dc657dSBarry Smith   for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3888965ea79SLois Curfman McInnes   count = 0;
3898965ea79SLois Curfman McInnes   for (i=0; i<size; i++) {
390c1dc657dSBarry Smith     if (nprocs[2*i+1]) {
39113f74950SBarry Smith       ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr);
3928965ea79SLois Curfman McInnes     }
3938965ea79SLois Curfman McInnes   }
394606d414cSSatish Balay   ierr = PetscFree(starts);CHKERRQ(ierr);
3958965ea79SLois Curfman McInnes 
3968965ea79SLois Curfman McInnes   base = owners[rank];
3978965ea79SLois Curfman McInnes 
3988965ea79SLois Curfman McInnes   /*  wait on receives */
39913f74950SBarry Smith   ierr   = PetscMalloc(2*(nrecvs+1)*sizeof(PetscInt),&lens);CHKERRQ(ierr);
4008965ea79SLois Curfman McInnes   source = lens + nrecvs;
4018965ea79SLois Curfman McInnes   count  = nrecvs; slen = 0;
4028965ea79SLois Curfman McInnes   while (count) {
403ca161407SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
4048965ea79SLois Curfman McInnes     /* unpack receives into our local space */
40513f74950SBarry Smith     ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr);
4068965ea79SLois Curfman McInnes     source[imdex]  = recv_status.MPI_SOURCE;
4078965ea79SLois Curfman McInnes     lens[imdex]    = n;
4088965ea79SLois Curfman McInnes     slen += n;
4098965ea79SLois Curfman McInnes     count--;
4108965ea79SLois Curfman McInnes   }
411606d414cSSatish Balay   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
4128965ea79SLois Curfman McInnes 
4138965ea79SLois Curfman McInnes   /* move the data into the send scatter */
41413f74950SBarry Smith   ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr);
4158965ea79SLois Curfman McInnes   count = 0;
4168965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
4178965ea79SLois Curfman McInnes     values = rvalues + i*nmax;
4188965ea79SLois Curfman McInnes     for (j=0; j<lens[i]; j++) {
4198965ea79SLois Curfman McInnes       lrows[count++] = values[j] - base;
4208965ea79SLois Curfman McInnes     }
4218965ea79SLois Curfman McInnes   }
422606d414cSSatish Balay   ierr = PetscFree(rvalues);CHKERRQ(ierr);
423606d414cSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
424606d414cSSatish Balay   ierr = PetscFree(owner);CHKERRQ(ierr);
425606d414cSSatish Balay   ierr = PetscFree(nprocs);CHKERRQ(ierr);
4268965ea79SLois Curfman McInnes 
4278965ea79SLois Curfman McInnes   /* actually zap the local rows */
428f4df32b1SMatthew Knepley   ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr);
429606d414cSSatish Balay   ierr = PetscFree(lrows);CHKERRQ(ierr);
4308965ea79SLois Curfman McInnes 
4318965ea79SLois Curfman McInnes   /* wait on sends */
4328965ea79SLois Curfman McInnes   if (nsends) {
433b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
434ca161407SBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
435606d414cSSatish Balay     ierr = PetscFree(send_status);CHKERRQ(ierr);
4368965ea79SLois Curfman McInnes   }
437606d414cSSatish Balay   ierr = PetscFree(send_waits);CHKERRQ(ierr);
438606d414cSSatish Balay   ierr = PetscFree(svalues);CHKERRQ(ierr);
4398965ea79SLois Curfman McInnes 
4403a40ed3dSBarry Smith   PetscFunctionReturn(0);
4418965ea79SLois Curfman McInnes }
4428965ea79SLois Curfman McInnes 
4434a2ae208SSatish Balay #undef __FUNCT__
4444a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense"
445dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy)
4468965ea79SLois Curfman McInnes {
44739ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
448dfbe8321SBarry Smith   PetscErrorCode ierr;
449c456f294SBarry Smith 
4503a40ed3dSBarry Smith   PetscFunctionBegin;
451ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
452ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
45344cd7ae7SLois Curfman McInnes   ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr);
4543a40ed3dSBarry Smith   PetscFunctionReturn(0);
4558965ea79SLois Curfman McInnes }
4568965ea79SLois Curfman McInnes 
4574a2ae208SSatish Balay #undef __FUNCT__
4584a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense"
459dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz)
4608965ea79SLois Curfman McInnes {
46139ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
462dfbe8321SBarry Smith   PetscErrorCode ierr;
463c456f294SBarry Smith 
4643a40ed3dSBarry Smith   PetscFunctionBegin;
465ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
466ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
46744cd7ae7SLois Curfman McInnes   ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr);
4683a40ed3dSBarry Smith   PetscFunctionReturn(0);
4698965ea79SLois Curfman McInnes }
4708965ea79SLois Curfman McInnes 
4714a2ae208SSatish Balay #undef __FUNCT__
4724a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense"
473dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy)
474096963f5SLois Curfman McInnes {
475096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
476dfbe8321SBarry Smith   PetscErrorCode ierr;
47787828ca2SBarry Smith   PetscScalar    zero = 0.0;
478096963f5SLois Curfman McInnes 
4793a40ed3dSBarry Smith   PetscFunctionBegin;
4802dcb1b2aSMatthew Knepley   ierr = VecSet(yy,zero);CHKERRQ(ierr);
4817c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
482ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
483ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
4843a40ed3dSBarry Smith   PetscFunctionReturn(0);
485096963f5SLois Curfman McInnes }
486096963f5SLois Curfman McInnes 
4874a2ae208SSatish Balay #undef __FUNCT__
4884a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense"
489dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz)
490096963f5SLois Curfman McInnes {
491096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
492dfbe8321SBarry Smith   PetscErrorCode ierr;
493096963f5SLois Curfman McInnes 
4943a40ed3dSBarry Smith   PetscFunctionBegin;
4953501a2bdSLois Curfman McInnes   ierr = VecCopy(yy,zz);CHKERRQ(ierr);
4967c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
497ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
498ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
4993a40ed3dSBarry Smith   PetscFunctionReturn(0);
500096963f5SLois Curfman McInnes }
501096963f5SLois Curfman McInnes 
5024a2ae208SSatish Balay #undef __FUNCT__
5034a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense"
504dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v)
5058965ea79SLois Curfman McInnes {
50639ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
507096963f5SLois Curfman McInnes   Mat_SeqDense   *aloc = (Mat_SeqDense*)a->A->data;
508dfbe8321SBarry Smith   PetscErrorCode ierr;
509d0f46423SBarry Smith   PetscInt       len,i,n,m = A->rmap->n,radd;
51087828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
511ed3cc1f0SBarry Smith 
5123a40ed3dSBarry Smith   PetscFunctionBegin;
5132dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
5141ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
515096963f5SLois Curfman McInnes   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
516d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
517d0f46423SBarry Smith   len  = PetscMin(a->A->rmap->n,a->A->cmap->n);
518d0f46423SBarry Smith   radd = A->rmap->rstart*m;
51944cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
520096963f5SLois Curfman McInnes     x[i] = aloc->v[radd + i*m + i];
521096963f5SLois Curfman McInnes   }
5221ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
5233a40ed3dSBarry Smith   PetscFunctionReturn(0);
5248965ea79SLois Curfman McInnes }
5258965ea79SLois Curfman McInnes 
5264a2ae208SSatish Balay #undef __FUNCT__
5274a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense"
528dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat)
5298965ea79SLois Curfman McInnes {
5303501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
531dfbe8321SBarry Smith   PetscErrorCode ierr;
53201b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
53301b82886SBarry Smith   Mat_Plapack   *lu=(Mat_Plapack*)(mat->spptr);
53401b82886SBarry Smith #endif
535ed3cc1f0SBarry Smith 
5363a40ed3dSBarry Smith   PetscFunctionBegin;
53794d884c6SBarry Smith 
538aa482453SBarry Smith #if defined(PETSC_USE_LOG)
539d0f46423SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N);
5408965ea79SLois Curfman McInnes #endif
5418798bf22SSatish Balay   ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr);
5423501a2bdSLois Curfman McInnes   ierr = MatDestroy(mdn->A);CHKERRQ(ierr);
54305b42c5fSBarry Smith   if (mdn->lvec)   {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);}
54405b42c5fSBarry Smith   if (mdn->Mvctx)  {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);}
54501b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
5462fbe02b9SBarry Smith   if (lu) {
54762b4c0b3SBarry Smith     ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr);
54862b4c0b3SBarry Smith     ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr);
54962b4c0b3SBarry Smith     ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr);
55009d27a7eSBarry Smith 
5512fbe02b9SBarry Smith     if (lu->is_pla) {
55201b82886SBarry Smith       ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr);
55301b82886SBarry Smith       ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr);
55401b82886SBarry Smith       ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr);
555622d7880SLois Curfman McInnes     }
5562fbe02b9SBarry Smith   }
55701b82886SBarry Smith #endif
55801b82886SBarry Smith 
559606d414cSSatish Balay   ierr = PetscFree(mdn);CHKERRQ(ierr);
560dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
561901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr);
562901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
5634ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5644ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5654ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5663a40ed3dSBarry Smith   PetscFunctionReturn(0);
5678965ea79SLois Curfman McInnes }
56839ddd567SLois Curfman McInnes 
5694a2ae208SSatish Balay #undef __FUNCT__
5704a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary"
5716849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer)
5728965ea79SLois Curfman McInnes {
57339ddd567SLois Curfman McInnes   Mat_MPIDense      *mdn = (Mat_MPIDense*)mat->data;
574dfbe8321SBarry Smith   PetscErrorCode    ierr;
575aa05aa95SBarry Smith   PetscViewerFormat format;
576aa05aa95SBarry Smith   int               fd;
577d0f46423SBarry Smith   PetscInt          header[4],mmax,N = mat->cmap->N,i,j,m,k;
578aa05aa95SBarry Smith   PetscMPIInt       rank,tag  = ((PetscObject)viewer)->tag,size;
579578230a0SSatish Balay   PetscScalar       *work,*v,*vv;
580aa05aa95SBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)mdn->A->data;
581aa05aa95SBarry Smith   MPI_Status        status;
5827056b6fcSBarry Smith 
5833a40ed3dSBarry Smith   PetscFunctionBegin;
58439ddd567SLois Curfman McInnes   if (mdn->size == 1) {
58539ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
586aa05aa95SBarry Smith   } else {
587aa05aa95SBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
5887adad957SLisandro Dalcin     ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr);
5897adad957SLisandro Dalcin     ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr);
590aa05aa95SBarry Smith 
591aa05aa95SBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
592aa05aa95SBarry Smith     if (format == PETSC_VIEWER_BINARY_NATIVE) {
593aa05aa95SBarry Smith 
594aa05aa95SBarry Smith       if (!rank) {
595aa05aa95SBarry Smith         /* store the matrix as a dense matrix */
596aa05aa95SBarry Smith         header[0] = MAT_FILE_COOKIE;
597d0f46423SBarry Smith         header[1] = mat->rmap->N;
598aa05aa95SBarry Smith         header[2] = N;
599aa05aa95SBarry Smith         header[3] = MATRIX_BINARY_FORMAT_DENSE;
600aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
601aa05aa95SBarry Smith 
602aa05aa95SBarry Smith         /* get largest work array needed for transposing array */
603d0f46423SBarry Smith         mmax = mat->rmap->n;
604aa05aa95SBarry Smith         for (i=1; i<size; i++) {
605d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
6068965ea79SLois Curfman McInnes         }
607aa05aa95SBarry Smith         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr);
608aa05aa95SBarry Smith 
609aa05aa95SBarry Smith         /* write out local array, by rows */
610d0f46423SBarry Smith         m    = mat->rmap->n;
611aa05aa95SBarry Smith         v    = a->v;
612aa05aa95SBarry Smith         for (j=0; j<N; j++) {
613aa05aa95SBarry Smith           for (i=0; i<m; i++) {
614578230a0SSatish Balay             work[j + i*N] = *v++;
615aa05aa95SBarry Smith           }
616aa05aa95SBarry Smith         }
617aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
618aa05aa95SBarry Smith         /* get largest work array to receive messages from other processes, excludes process zero */
619aa05aa95SBarry Smith         mmax = 0;
620aa05aa95SBarry Smith         for (i=1; i<size; i++) {
621d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
622aa05aa95SBarry Smith         }
623578230a0SSatish Balay         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr);
624aa05aa95SBarry Smith         for(k = 1; k < size; k++) {
625f8009846SMatthew Knepley           v    = vv;
626d0f46423SBarry Smith           m    = mat->rmap->range[k+1] - mat->rmap->range[k];
6277adad957SLisandro Dalcin           ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr);
628aa05aa95SBarry Smith 
629aa05aa95SBarry Smith           for(j = 0; j < N; j++) {
630aa05aa95SBarry Smith             for(i = 0; i < m; i++) {
631578230a0SSatish Balay               work[j + i*N] = *v++;
632aa05aa95SBarry Smith             }
633aa05aa95SBarry Smith           }
634aa05aa95SBarry Smith           ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
635aa05aa95SBarry Smith         }
636aa05aa95SBarry Smith         ierr = PetscFree(work);CHKERRQ(ierr);
637578230a0SSatish Balay         ierr = PetscFree(vv);CHKERRQ(ierr);
638aa05aa95SBarry Smith       } else {
639d0f46423SBarry Smith         ierr = MPI_Send(a->v,mat->rmap->n*mat->cmap->N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr);
640aa05aa95SBarry Smith       }
6414aa34b0aSBarry Smith     } else {
642a542b6e8SBarry Smith       SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_BINARY_NATIVE");
643aa05aa95SBarry Smith     }
644aa05aa95SBarry Smith   }
6453a40ed3dSBarry Smith   PetscFunctionReturn(0);
6468965ea79SLois Curfman McInnes }
6478965ea79SLois Curfman McInnes 
6484a2ae208SSatish Balay #undef __FUNCT__
6494a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket"
6506849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer)
6518965ea79SLois Curfman McInnes {
65239ddd567SLois Curfman McInnes   Mat_MPIDense          *mdn = (Mat_MPIDense*)mat->data;
653dfbe8321SBarry Smith   PetscErrorCode        ierr;
65432dcc486SBarry Smith   PetscMPIInt           size = mdn->size,rank = mdn->rank;
655a313700dSBarry Smith   const PetscViewerType vtype;
65632077d6dSBarry Smith   PetscTruth            iascii,isdraw;
657b0a32e0cSBarry Smith   PetscViewer           sviewer;
658f3ef73ceSBarry Smith   PetscViewerFormat     format;
65901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
66001b82886SBarry Smith   Mat_Plapack           *lu=(Mat_Plapack*)(mat->spptr);
66101b82886SBarry Smith #endif
6628965ea79SLois Curfman McInnes 
6633a40ed3dSBarry Smith   PetscFunctionBegin;
66432077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
665fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
66632077d6dSBarry Smith   if (iascii) {
667b0a32e0cSBarry Smith     ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr);
668b0a32e0cSBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
669456192e2SBarry Smith     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
6704e220ebcSLois Curfman McInnes       MatInfo info;
671888f2ed8SSatish Balay       ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr);
672d0f46423SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap->n,
67377431f27SBarry Smith                    (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr);
674b0a32e0cSBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
6753501a2bdSLois Curfman McInnes       ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr);
67601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
67701b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr);
6786c995c7dSHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Processor mesh: nprows %d, npcols %d\n",lu->Plapack_nprows, lu->Plapack_npcols);CHKERRQ(ierr);
67901b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr);
6806c995c7dSHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Error checking: %d\n",lu->Plapack_ierror);CHKERRQ(ierr);
6816c995c7dSHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Algorithmic block size: %d\n",lu->Plapack_nb_alg);CHKERRQ(ierr);
68201b82886SBarry Smith #endif
6833a40ed3dSBarry Smith       PetscFunctionReturn(0);
684fb9695e5SSatish Balay     } else if (format == PETSC_VIEWER_ASCII_INFO) {
6853a40ed3dSBarry Smith       PetscFunctionReturn(0);
6868965ea79SLois Curfman McInnes     }
687f1af5d2fSBarry Smith   } else if (isdraw) {
688b0a32e0cSBarry Smith     PetscDraw  draw;
689f1af5d2fSBarry Smith     PetscTruth isnull;
690f1af5d2fSBarry Smith 
691b0a32e0cSBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
692b0a32e0cSBarry Smith     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
693f1af5d2fSBarry Smith     if (isnull) PetscFunctionReturn(0);
694f1af5d2fSBarry Smith   }
69577ed5343SBarry Smith 
6968965ea79SLois Curfman McInnes   if (size == 1) {
69739ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
6983a40ed3dSBarry Smith   } else {
6998965ea79SLois Curfman McInnes     /* assemble the entire matrix onto first processor. */
7008965ea79SLois Curfman McInnes     Mat         A;
701d0f46423SBarry Smith     PetscInt    M = mat->rmap->N,N = mat->cmap->N,m,row,i,nz;
702ba8c8a56SBarry Smith     PetscInt    *cols;
703ba8c8a56SBarry Smith     PetscScalar *vals;
7048965ea79SLois Curfman McInnes 
7057adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr);
7068965ea79SLois Curfman McInnes     if (!rank) {
707f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr);
7083a40ed3dSBarry Smith     } else {
709f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr);
7108965ea79SLois Curfman McInnes     }
7117adad957SLisandro Dalcin     /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */
712878740d9SKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
713878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL);
71452e6d16bSBarry Smith     ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr);
7158965ea79SLois Curfman McInnes 
71639ddd567SLois Curfman McInnes     /* Copy the matrix ... This isn't the most efficient means,
71739ddd567SLois Curfman McInnes        but it's quick for now */
71851022da4SBarry Smith     A->insertmode = INSERT_VALUES;
719d0f46423SBarry Smith     row = mat->rmap->rstart; m = mdn->A->rmap->n;
7208965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
721ba8c8a56SBarry Smith       ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
722ba8c8a56SBarry Smith       ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
723ba8c8a56SBarry Smith       ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
72439ddd567SLois Curfman McInnes       row++;
7258965ea79SLois Curfman McInnes     }
7268965ea79SLois Curfman McInnes 
7276d4a8577SBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7286d4a8577SBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
729b0a32e0cSBarry Smith     ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr);
730b9b97703SBarry Smith     if (!rank) {
7316831982aSBarry Smith       ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr);
7328965ea79SLois Curfman McInnes     }
733b0a32e0cSBarry Smith     ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr);
734b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
7358965ea79SLois Curfman McInnes     ierr = MatDestroy(A);CHKERRQ(ierr);
7368965ea79SLois Curfman McInnes   }
7373a40ed3dSBarry Smith   PetscFunctionReturn(0);
7388965ea79SLois Curfman McInnes }
7398965ea79SLois Curfman McInnes 
7404a2ae208SSatish Balay #undef __FUNCT__
7414a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense"
742dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer)
7438965ea79SLois Curfman McInnes {
744dfbe8321SBarry Smith   PetscErrorCode ierr;
74532077d6dSBarry Smith   PetscTruth     iascii,isbinary,isdraw,issocket;
7468965ea79SLois Curfman McInnes 
747433994e6SBarry Smith   PetscFunctionBegin;
7480f5bd95cSBarry Smith 
74932077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
750fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
751b0a32e0cSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr);
752fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
7530f5bd95cSBarry Smith 
75432077d6dSBarry Smith   if (iascii || issocket || isdraw) {
755f1af5d2fSBarry Smith     ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr);
7560f5bd95cSBarry Smith   } else if (isbinary) {
7573a40ed3dSBarry Smith     ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr);
7585cd90555SBarry Smith   } else {
759958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name);
7608965ea79SLois Curfman McInnes   }
7613a40ed3dSBarry Smith   PetscFunctionReturn(0);
7628965ea79SLois Curfman McInnes }
7638965ea79SLois Curfman McInnes 
7644a2ae208SSatish Balay #undef __FUNCT__
7654a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense"
766dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info)
7678965ea79SLois Curfman McInnes {
7683501a2bdSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
7693501a2bdSLois Curfman McInnes   Mat            mdn = mat->A;
770dfbe8321SBarry Smith   PetscErrorCode ierr;
771329f5518SBarry Smith   PetscReal      isend[5],irecv[5];
7728965ea79SLois Curfman McInnes 
7733a40ed3dSBarry Smith   PetscFunctionBegin;
774d0f46423SBarry Smith   info->rows_global    = (double)A->rmap->N;
775d0f46423SBarry Smith   info->columns_global = (double)A->cmap->N;
776d0f46423SBarry Smith   info->rows_local     = (double)A->rmap->n;
777d0f46423SBarry Smith   info->columns_local  = (double)A->cmap->N;
7784e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
7794e220ebcSLois Curfman McInnes   ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr);
7804e220ebcSLois Curfman McInnes   isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded;
7814e220ebcSLois Curfman McInnes   isend[3] = info->memory;  isend[4] = info->mallocs;
7828965ea79SLois Curfman McInnes   if (flag == MAT_LOCAL) {
7834e220ebcSLois Curfman McInnes     info->nz_used      = isend[0];
7844e220ebcSLois Curfman McInnes     info->nz_allocated = isend[1];
7854e220ebcSLois Curfman McInnes     info->nz_unneeded  = isend[2];
7864e220ebcSLois Curfman McInnes     info->memory       = isend[3];
7874e220ebcSLois Curfman McInnes     info->mallocs      = isend[4];
7888965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_MAX) {
7897adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
7904e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7914e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
7924e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
7934e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
7944e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
7958965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_SUM) {
7967adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
7974e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7984e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
7994e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
8004e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
8014e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
8028965ea79SLois Curfman McInnes   }
8034e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0; /* no parallel LU/ILU/Cholesky */
8044e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
8054e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
8063a40ed3dSBarry Smith   PetscFunctionReturn(0);
8078965ea79SLois Curfman McInnes }
8088965ea79SLois Curfman McInnes 
8094a2ae208SSatish Balay #undef __FUNCT__
8104a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense"
8114e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg)
8128965ea79SLois Curfman McInnes {
81339ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
814dfbe8321SBarry Smith   PetscErrorCode ierr;
8158965ea79SLois Curfman McInnes 
8163a40ed3dSBarry Smith   PetscFunctionBegin;
81712c028f9SKris Buschelman   switch (op) {
818512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
81912c028f9SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
82012c028f9SKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
8214e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82212c028f9SKris Buschelman     break;
82312c028f9SKris Buschelman   case MAT_ROW_ORIENTED:
8244e0d8c25SBarry Smith     a->roworiented = flg;
8254e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82612c028f9SKris Buschelman     break;
8274e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
82812c028f9SKris Buschelman   case MAT_USE_HASH_TABLE:
829290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
83012c028f9SKris Buschelman     break;
83112c028f9SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
8324e0d8c25SBarry Smith     a->donotstash = flg;
83312c028f9SKris Buschelman     break;
83477e54ba9SKris Buschelman   case MAT_SYMMETRIC:
83577e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
8369a4540c5SBarry Smith   case MAT_HERMITIAN:
8379a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
838600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
839290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
84077e54ba9SKris Buschelman     break;
84112c028f9SKris Buschelman   default:
842600fe468SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
8433a40ed3dSBarry Smith   }
8443a40ed3dSBarry Smith   PetscFunctionReturn(0);
8458965ea79SLois Curfman McInnes }
8468965ea79SLois Curfman McInnes 
8478965ea79SLois Curfman McInnes 
8484a2ae208SSatish Balay #undef __FUNCT__
8494a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense"
850dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr)
8515b2fa520SLois Curfman McInnes {
8525b2fa520SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8535b2fa520SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
85487828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
855dfbe8321SBarry Smith   PetscErrorCode ierr;
856d0f46423SBarry Smith   PetscInt       i,j,s2a,s3a,s2,s3,m=mdn->A->rmap->n,n=mdn->A->cmap->n;
8575b2fa520SLois Curfman McInnes 
8585b2fa520SLois Curfman McInnes   PetscFunctionBegin;
85972d926a5SLois Curfman McInnes   ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr);
8605b2fa520SLois Curfman McInnes   if (ll) {
86172d926a5SLois Curfman McInnes     ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr);
862175be7b4SMatthew Knepley     if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2);
8631ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
8645b2fa520SLois Curfman McInnes     for (i=0; i<m; i++) {
8655b2fa520SLois Curfman McInnes       x = l[i];
8665b2fa520SLois Curfman McInnes       v = mat->v + i;
8675b2fa520SLois Curfman McInnes       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
8685b2fa520SLois Curfman McInnes     }
8691ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
870efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8715b2fa520SLois Curfman McInnes   }
8725b2fa520SLois Curfman McInnes   if (rr) {
873175be7b4SMatthew Knepley     ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr);
874175be7b4SMatthew Knepley     if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3);
875ca9f406cSSatish Balay     ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
876ca9f406cSSatish Balay     ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
8771ebc52fbSHong Zhang     ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr);
8785b2fa520SLois Curfman McInnes     for (i=0; i<n; i++) {
8795b2fa520SLois Curfman McInnes       x = r[i];
8805b2fa520SLois Curfman McInnes       v = mat->v + i*m;
8815b2fa520SLois Curfman McInnes       for (j=0; j<m; j++) { (*v++) *= x;}
8825b2fa520SLois Curfman McInnes     }
8831ebc52fbSHong Zhang     ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr);
884efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8855b2fa520SLois Curfman McInnes   }
8865b2fa520SLois Curfman McInnes   PetscFunctionReturn(0);
8875b2fa520SLois Curfman McInnes }
8885b2fa520SLois Curfman McInnes 
8894a2ae208SSatish Balay #undef __FUNCT__
8904a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense"
891dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm)
892096963f5SLois Curfman McInnes {
8933501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8943501a2bdSLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
895dfbe8321SBarry Smith   PetscErrorCode ierr;
89613f74950SBarry Smith   PetscInt       i,j;
897329f5518SBarry Smith   PetscReal      sum = 0.0;
89887828ca2SBarry Smith   PetscScalar    *v = mat->v;
8993501a2bdSLois Curfman McInnes 
9003a40ed3dSBarry Smith   PetscFunctionBegin;
9013501a2bdSLois Curfman McInnes   if (mdn->size == 1) {
902064f8208SBarry Smith     ierr =  MatNorm(mdn->A,type,nrm);CHKERRQ(ierr);
9033501a2bdSLois Curfman McInnes   } else {
9043501a2bdSLois Curfman McInnes     if (type == NORM_FROBENIUS) {
905d0f46423SBarry Smith       for (i=0; i<mdn->A->cmap->n*mdn->A->rmap->n; i++) {
906aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
907329f5518SBarry Smith         sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
9083501a2bdSLois Curfman McInnes #else
9093501a2bdSLois Curfman McInnes         sum += (*v)*(*v); v++;
9103501a2bdSLois Curfman McInnes #endif
9113501a2bdSLois Curfman McInnes       }
9127adad957SLisandro Dalcin       ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
913064f8208SBarry Smith       *nrm = sqrt(*nrm);
914d0f46423SBarry Smith       ierr = PetscLogFlops(2*mdn->A->cmap->n*mdn->A->rmap->n);CHKERRQ(ierr);
9153a40ed3dSBarry Smith     } else if (type == NORM_1) {
916329f5518SBarry Smith       PetscReal *tmp,*tmp2;
917d0f46423SBarry Smith       ierr = PetscMalloc(2*A->cmap->N*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
918d0f46423SBarry Smith       tmp2 = tmp + A->cmap->N;
919d0f46423SBarry Smith       ierr = PetscMemzero(tmp,2*A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr);
920064f8208SBarry Smith       *nrm = 0.0;
9213501a2bdSLois Curfman McInnes       v = mat->v;
922d0f46423SBarry Smith       for (j=0; j<mdn->A->cmap->n; j++) {
923d0f46423SBarry Smith         for (i=0; i<mdn->A->rmap->n; i++) {
92467e560aaSBarry Smith           tmp[j] += PetscAbsScalar(*v);  v++;
9253501a2bdSLois Curfman McInnes         }
9263501a2bdSLois Curfman McInnes       }
927d0f46423SBarry Smith       ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
928d0f46423SBarry Smith       for (j=0; j<A->cmap->N; j++) {
929064f8208SBarry Smith         if (tmp2[j] > *nrm) *nrm = tmp2[j];
9303501a2bdSLois Curfman McInnes       }
931606d414cSSatish Balay       ierr = PetscFree(tmp);CHKERRQ(ierr);
932d0f46423SBarry Smith       ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
9333a40ed3dSBarry Smith     } else if (type == NORM_INFINITY) { /* max row norm */
934329f5518SBarry Smith       PetscReal ntemp;
9353501a2bdSLois Curfman McInnes       ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr);
9367adad957SLisandro Dalcin       ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
9373a40ed3dSBarry Smith     } else {
93829bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"No support for two norm");
9393501a2bdSLois Curfman McInnes     }
9403501a2bdSLois Curfman McInnes   }
9413a40ed3dSBarry Smith   PetscFunctionReturn(0);
9423501a2bdSLois Curfman McInnes }
9433501a2bdSLois Curfman McInnes 
9444a2ae208SSatish Balay #undef __FUNCT__
9454a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense"
946fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout)
9473501a2bdSLois Curfman McInnes {
9483501a2bdSLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
9493501a2bdSLois Curfman McInnes   Mat_SeqDense   *Aloc = (Mat_SeqDense*)a->A->data;
9503501a2bdSLois Curfman McInnes   Mat            B;
951d0f46423SBarry Smith   PetscInt       M = A->rmap->N,N = A->cmap->N,m,n,*rwork,rstart = A->rmap->rstart;
9526849ba73SBarry Smith   PetscErrorCode ierr;
95313f74950SBarry Smith   PetscInt       j,i;
95487828ca2SBarry Smith   PetscScalar    *v;
9553501a2bdSLois Curfman McInnes 
9563a40ed3dSBarry Smith   PetscFunctionBegin;
957e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place");
958fc4dec0aSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || A == *matout) {
9597adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
960d0f46423SBarry Smith     ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,N,M);CHKERRQ(ierr);
9617adad957SLisandro Dalcin     ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
962878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr);
963fc4dec0aSBarry Smith   } else {
964fc4dec0aSBarry Smith     B = *matout;
965fc4dec0aSBarry Smith   }
9663501a2bdSLois Curfman McInnes 
967d0f46423SBarry Smith   m = a->A->rmap->n; n = a->A->cmap->n; v = Aloc->v;
9681acff37aSSatish Balay   ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr);
9693501a2bdSLois Curfman McInnes   for (i=0; i<m; i++) rwork[i] = rstart + i;
9701acff37aSSatish Balay   for (j=0; j<n; j++) {
9713501a2bdSLois Curfman McInnes     ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr);
9723501a2bdSLois Curfman McInnes     v   += m;
9733501a2bdSLois Curfman McInnes   }
974606d414cSSatish Balay   ierr = PetscFree(rwork);CHKERRQ(ierr);
9756d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9766d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
977815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *matout != A) {
9783501a2bdSLois Curfman McInnes     *matout = B;
9793501a2bdSLois Curfman McInnes   } else {
980273d9f13SBarry Smith     ierr = MatHeaderCopy(A,B);CHKERRQ(ierr);
9813501a2bdSLois Curfman McInnes   }
9823a40ed3dSBarry Smith   PetscFunctionReturn(0);
983096963f5SLois Curfman McInnes }
984096963f5SLois Curfman McInnes 
985d9eff348SSatish Balay #include "petscblaslapack.h"
9864a2ae208SSatish Balay #undef __FUNCT__
9874a2ae208SSatish Balay #define __FUNCT__ "MatScale_MPIDense"
988f4df32b1SMatthew Knepley PetscErrorCode MatScale_MPIDense(Mat inA,PetscScalar alpha)
98944cd7ae7SLois Curfman McInnes {
99044cd7ae7SLois Curfman McInnes   Mat_MPIDense   *A = (Mat_MPIDense*)inA->data;
99144cd7ae7SLois Curfman McInnes   Mat_SeqDense   *a = (Mat_SeqDense*)A->A->data;
992f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
993efee365bSSatish Balay   PetscErrorCode ierr;
994d0f46423SBarry Smith   PetscBLASInt   one = 1,nz = PetscBLASIntCast(inA->rmap->n*inA->cmap->N);
99544cd7ae7SLois Curfman McInnes 
9963a40ed3dSBarry Smith   PetscFunctionBegin;
997f4df32b1SMatthew Knepley   BLASscal_(&nz,&oalpha,a->v,&one);
998efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
9993a40ed3dSBarry Smith   PetscFunctionReturn(0);
100044cd7ae7SLois Curfman McInnes }
100144cd7ae7SLois Curfman McInnes 
10026849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *);
10038965ea79SLois Curfman McInnes 
10044a2ae208SSatish Balay #undef __FUNCT__
10054a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_MPIDense"
1006dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_MPIDense(Mat A)
1007273d9f13SBarry Smith {
1008dfbe8321SBarry Smith   PetscErrorCode ierr;
1009273d9f13SBarry Smith 
1010273d9f13SBarry Smith   PetscFunctionBegin;
1011273d9f13SBarry Smith   ierr =  MatMPIDenseSetPreallocation(A,0);CHKERRQ(ierr);
1012273d9f13SBarry Smith   PetscFunctionReturn(0);
1013273d9f13SBarry Smith }
1014273d9f13SBarry Smith 
101501b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1016eae6fb2eSBarry Smith 
1017eae6fb2eSBarry Smith #undef __FUNCT__
1018eae6fb2eSBarry Smith #define __FUNCT__ "MatMPIDenseCopyToPlapack"
1019eae6fb2eSBarry Smith PetscErrorCode MatMPIDenseCopyToPlapack(Mat A,Mat F)
1020eae6fb2eSBarry Smith {
1021eae6fb2eSBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
1022eae6fb2eSBarry Smith   PetscErrorCode ierr;
1023d0f46423SBarry Smith   PetscInt       M=A->cmap->N,m=A->rmap->n,rstart;
1024eae6fb2eSBarry Smith   PetscScalar    *array;
1025eae6fb2eSBarry Smith   PetscReal      one = 1.0;
1026eae6fb2eSBarry Smith 
1027eae6fb2eSBarry Smith   PetscFunctionBegin;
10282fbe02b9SBarry Smith   /* Copy A into F->lu->A */
1029eae6fb2eSBarry Smith   ierr = PLA_Obj_set_to_zero(lu->A);CHKERRQ(ierr);
1030eae6fb2eSBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
1031eae6fb2eSBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
103279b0a62dSBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
1033eae6fb2eSBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
1034eae6fb2eSBarry Smith   ierr = PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);CHKERRQ(ierr);
1035eae6fb2eSBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
1036eae6fb2eSBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
1037eae6fb2eSBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
1038eae6fb2eSBarry Smith   lu->rstart = rstart;
1039eae6fb2eSBarry Smith   PetscFunctionReturn(0);
1040eae6fb2eSBarry Smith }
1041eae6fb2eSBarry Smith 
104201b82886SBarry Smith #undef __FUNCT__
10432fbe02b9SBarry Smith #define __FUNCT__ "MatMPIDenseCopyFromPlapack"
10442fbe02b9SBarry Smith PetscErrorCode MatMPIDenseCopyFromPlapack(Mat F,Mat A)
10452fbe02b9SBarry Smith {
10462fbe02b9SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
10472fbe02b9SBarry Smith   PetscErrorCode ierr;
1048d0f46423SBarry Smith   PetscInt       M=A->cmap->N,m=A->rmap->n,rstart;
10492fbe02b9SBarry Smith   PetscScalar    *array;
10502fbe02b9SBarry Smith   PetscReal      one = 1.0;
10512fbe02b9SBarry Smith 
10522fbe02b9SBarry Smith   PetscFunctionBegin;
10532fbe02b9SBarry Smith   /* Copy F into A->lu->A */
105479b0a62dSBarry Smith   ierr = MatZeroEntries(A);CHKERRQ(ierr);
10552fbe02b9SBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
10562fbe02b9SBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
10572fbe02b9SBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
10582fbe02b9SBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
10592fbe02b9SBarry Smith   ierr = PLA_API_axpy_global_to_matrix(m,M, &one,lu->A,rstart,0,(void *)array,m);CHKERRQ(ierr);
10602fbe02b9SBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
10612fbe02b9SBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
10622fbe02b9SBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
10632fbe02b9SBarry Smith   lu->rstart = rstart;
10642fbe02b9SBarry Smith   PetscFunctionReturn(0);
10652fbe02b9SBarry Smith }
10662fbe02b9SBarry Smith 
10672fbe02b9SBarry Smith #undef __FUNCT__
10682fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultNumeric_MPIDense_MPIDense"
10692fbe02b9SBarry Smith PetscErrorCode MatMatMultNumeric_MPIDense_MPIDense(Mat A,Mat B,Mat C)
10702fbe02b9SBarry Smith {
10712fbe02b9SBarry Smith   PetscErrorCode ierr;
10722fbe02b9SBarry Smith   Mat_Plapack    *luA = (Mat_Plapack*)A->spptr;
10732fbe02b9SBarry Smith   Mat_Plapack    *luB = (Mat_Plapack*)B->spptr;
10742fbe02b9SBarry Smith   Mat_Plapack    *luC = (Mat_Plapack*)C->spptr;
10752fbe02b9SBarry Smith   PLA_Obj        alpha = NULL,beta = NULL;
10762fbe02b9SBarry Smith 
10772fbe02b9SBarry Smith   PetscFunctionBegin;
10782fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(A,A);CHKERRQ(ierr);
10792fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(B,B);CHKERRQ(ierr);
10802fbe02b9SBarry Smith 
10816e6f9017SBarry Smith   /*
1082cb2480eaSBarry Smith   ierr = PLA_Global_show("A = ",luA->A,"%g ","");CHKERRQ(ierr);
1083cb2480eaSBarry Smith   ierr = PLA_Global_show("B = ",luB->A,"%g ","");CHKERRQ(ierr);
10846e6f9017SBarry Smith   */
1085cb2480eaSBarry Smith 
10862fbe02b9SBarry Smith   /* do the multiply in PLA  */
108779b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luA->A,NULL,NULL,&alpha);CHKERRQ(ierr);
108879b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luC->A,NULL,&beta,NULL);CHKERRQ(ierr);
108979b0a62dSBarry Smith   CHKMEMQ;
10902fbe02b9SBarry Smith 
1091f0e3846dSSatish Balay   ierr = PLA_Gemm(PLA_NO_TRANSPOSE,PLA_NO_TRANSPOSE,alpha,luA->A,luB->A,beta,luC->A); /* CHKERRQ(ierr); */
109279b0a62dSBarry Smith   CHKMEMQ;
10932fbe02b9SBarry Smith   ierr = PLA_Obj_free(&alpha);CHKERRQ(ierr);
10942fbe02b9SBarry Smith   ierr = PLA_Obj_free(&beta);CHKERRQ(ierr);
10952fbe02b9SBarry Smith 
10966e6f9017SBarry Smith   /*
1097cb2480eaSBarry Smith   ierr = PLA_Global_show("C = ",luC->A,"%g ","");CHKERRQ(ierr);
10986e6f9017SBarry Smith   */
10992fbe02b9SBarry Smith   ierr = MatMPIDenseCopyFromPlapack(C,C);CHKERRQ(ierr);
11002fbe02b9SBarry Smith   PetscFunctionReturn(0);
11012fbe02b9SBarry Smith }
11022fbe02b9SBarry Smith 
11032fbe02b9SBarry Smith #undef __FUNCT__
11042fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultSymbolic_MPIDense_MPIDense"
11052fbe02b9SBarry Smith PetscErrorCode MatMatMultSymbolic_MPIDense_MPIDense(Mat A,Mat B,PetscReal fill,Mat *C)
11062fbe02b9SBarry Smith {
11072fbe02b9SBarry Smith   PetscErrorCode ierr;
1108d0f46423SBarry Smith   PetscInt       m=A->rmap->n,n=B->cmap->n;
11092fbe02b9SBarry Smith   Mat            Cmat;
11102fbe02b9SBarry Smith 
11112fbe02b9SBarry Smith   PetscFunctionBegin;
1112d0f46423SBarry Smith   if (A->cmap->n != B->rmap->n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->cmap->n %d != B->rmap->n %d\n",A->cmap->n,B->rmap->n);
11136e6f9017SBarry Smith   SETERRQ(PETSC_ERR_LIB,"Due to aparent bugs in PLAPACK,this is not currently supported");
11142fbe02b9SBarry Smith   ierr = MatCreate(((PetscObject)B)->comm,&Cmat);CHKERRQ(ierr);
1115d0f46423SBarry Smith   ierr = MatSetSizes(Cmat,m,n,A->rmap->N,B->cmap->N);CHKERRQ(ierr);
11162fbe02b9SBarry Smith   ierr = MatSetType(Cmat,MATMPIDENSE);CHKERRQ(ierr);
11172fbe02b9SBarry Smith   ierr = MatAssemblyBegin(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11182fbe02b9SBarry Smith   ierr = MatAssemblyEnd(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1119ab235cb6SHong Zhang   //ierr = MatMPIDenseCreatePlapack(A);CHKERRQ(ierr);
1120ab235cb6SHong Zhang   //ierr = MatMPIDenseCreatePlapack(B);CHKERRQ(ierr);
1121ab235cb6SHong Zhang   //ierr = MatMPIDenseCreatePlapack(Cmat);CHKERRQ(ierr);
11222fbe02b9SBarry Smith   *C = Cmat;
11232fbe02b9SBarry Smith   PetscFunctionReturn(0);
11242fbe02b9SBarry Smith }
11252fbe02b9SBarry Smith 
11262fbe02b9SBarry Smith #undef __FUNCT__
11272fbe02b9SBarry Smith #define __FUNCT__ "MatMatMult_MPIDense_MPIDense"
11282fbe02b9SBarry Smith PetscErrorCode MatMatMult_MPIDense_MPIDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
11292fbe02b9SBarry Smith {
11302fbe02b9SBarry Smith   PetscErrorCode ierr;
11312fbe02b9SBarry Smith 
11322fbe02b9SBarry Smith   PetscFunctionBegin;
11332fbe02b9SBarry Smith   if (scall == MAT_INITIAL_MATRIX){
11342fbe02b9SBarry Smith     ierr = MatMatMultSymbolic_MPIDense_MPIDense(A,B,fill,C);CHKERRQ(ierr);
11352fbe02b9SBarry Smith   }
11362fbe02b9SBarry Smith   ierr = MatMatMultNumeric_MPIDense_MPIDense(A,B,*C);CHKERRQ(ierr);
11372fbe02b9SBarry Smith   PetscFunctionReturn(0);
11382fbe02b9SBarry Smith }
11392fbe02b9SBarry Smith 
1140ab235cb6SHong Zhang #undef __FUNCT__
11416c995c7dSHong Zhang #define __FUNCT__ "MatSolve_MPIDense"
11426c995c7dSHong Zhang PetscErrorCode MatSolve_MPIDense(Mat A,Vec b,Vec x)
11436c995c7dSHong Zhang {
11446c995c7dSHong Zhang   MPI_Comm       comm = ((PetscObject)A)->comm;
11456c995c7dSHong Zhang   Mat_Plapack    *lu = (Mat_Plapack*)A->spptr;
11466c995c7dSHong Zhang   PetscErrorCode ierr;
1147d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,i,j,*idx_pla,*idx_petsc,loc_m,loc_stride;
11486c995c7dSHong Zhang   PetscScalar    *array;
11496c995c7dSHong Zhang   PetscReal      one = 1.0;
11506c995c7dSHong Zhang   PetscMPIInt    size,rank,r_rank,r_nproc,c_rank,c_nproc;;
11516c995c7dSHong Zhang   PLA_Obj        v_pla = NULL;
11526c995c7dSHong Zhang   PetscScalar    *loc_buf;
11536c995c7dSHong Zhang   Vec            loc_x;
11546c995c7dSHong Zhang 
11556c995c7dSHong Zhang   PetscFunctionBegin;
11566c995c7dSHong Zhang   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
11576c995c7dSHong Zhang   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
11586c995c7dSHong Zhang 
11596c995c7dSHong Zhang   /* Create PLAPACK vector objects, then copy b into PLAPACK b */
11606c995c7dSHong Zhang   PLA_Mvector_create(lu->datatype,M,1,lu->templ,PLA_ALIGN_FIRST,&v_pla);
11616c995c7dSHong Zhang   PLA_Obj_set_to_zero(v_pla);
11626c995c7dSHong Zhang 
11636c995c7dSHong Zhang   /* Copy b into rhs_pla */
11646c995c7dSHong Zhang   PLA_API_begin();
11656c995c7dSHong Zhang   PLA_Obj_API_open(v_pla);
11666c995c7dSHong Zhang   ierr = VecGetArray(b,&array);CHKERRQ(ierr);
11676c995c7dSHong Zhang   PLA_API_axpy_vector_to_global(m,&one,(void *)array,1,v_pla,lu->rstart);
11686c995c7dSHong Zhang   ierr = VecRestoreArray(b,&array);CHKERRQ(ierr);
11696c995c7dSHong Zhang   PLA_Obj_API_close(v_pla);
11706c995c7dSHong Zhang   PLA_API_end();
11716c995c7dSHong Zhang 
11726c995c7dSHong Zhang   if (A->factor == MAT_FACTOR_LU){
11736c995c7dSHong Zhang     /* Apply the permutations to the right hand sides */
11746c995c7dSHong Zhang     PLA_Apply_pivots_to_rows (v_pla,lu->pivots);
11756c995c7dSHong Zhang 
11766c995c7dSHong Zhang     /* Solve L y = b, overwriting b with y */
11776c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_UNIT_DIAG,lu->A,v_pla );
11786c995c7dSHong Zhang 
11796c995c7dSHong Zhang     /* Solve U x = y (=b), overwriting b with x */
11806c995c7dSHong Zhang     PLA_Trsv( PLA_UPPER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla );
11816c995c7dSHong Zhang   } else { /*MAT_FACTOR_CHOLESKY */
11826c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla);
11836c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,(lu->datatype == MPI_DOUBLE ? PLA_TRANSPOSE : PLA_CONJUGATE_TRANSPOSE),
11846c995c7dSHong Zhang                                     PLA_NONUNIT_DIAG,lu->A,v_pla);
11856c995c7dSHong Zhang   }
11866c995c7dSHong Zhang 
11876c995c7dSHong Zhang   /* Copy PLAPACK x into Petsc vector x  */
11886c995c7dSHong Zhang   PLA_Obj_local_length(v_pla, &loc_m);
11896c995c7dSHong Zhang   PLA_Obj_local_buffer(v_pla, (void**)&loc_buf);
11906c995c7dSHong Zhang   PLA_Obj_local_stride(v_pla, &loc_stride);
11916c995c7dSHong Zhang   /*
11926c995c7dSHong Zhang     PetscPrintf(PETSC_COMM_SELF," [%d] b - local_m %d local_stride %d, loc_buf: %g %g, nb: %d\n",rank,loc_m,loc_stride,loc_buf[0],loc_buf[(loc_m-1)*loc_stride],lu->nb);
11936c995c7dSHong Zhang   */
11946c995c7dSHong Zhang   ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,loc_m*loc_stride,loc_buf,&loc_x);CHKERRQ(ierr);
11956c995c7dSHong Zhang   if (!lu->pla_solved){
11966c995c7dSHong Zhang 
11976c995c7dSHong Zhang     PLA_Temp_comm_row_info(lu->templ,&lu->comm_2d,&r_rank,&r_nproc);
11986c995c7dSHong Zhang     PLA_Temp_comm_col_info(lu->templ,&lu->comm_2d,&c_rank,&c_nproc);
11996c995c7dSHong Zhang     /* printf(" [%d] rank: %d %d, nproc: %d %d\n",rank,r_rank,c_rank,r_nproc,c_nproc); */
12006c995c7dSHong Zhang 
12016c995c7dSHong Zhang     /* Create IS and cts for VecScatterring */
12026c995c7dSHong Zhang     PLA_Obj_local_length(v_pla, &loc_m);
12036c995c7dSHong Zhang     PLA_Obj_local_stride(v_pla, &loc_stride);
12046c995c7dSHong Zhang     ierr = PetscMalloc((2*loc_m+1)*sizeof(PetscInt),&idx_pla);CHKERRQ(ierr);
12056c995c7dSHong Zhang     idx_petsc = idx_pla + loc_m;
12066c995c7dSHong Zhang 
12076c995c7dSHong Zhang     rstart = (r_rank*c_nproc+c_rank)*lu->nb;
12086c995c7dSHong Zhang     for (i=0; i<loc_m; i+=lu->nb){
12096c995c7dSHong Zhang       j = 0;
12106c995c7dSHong Zhang       while (j < lu->nb && i+j < loc_m){
12116c995c7dSHong Zhang         idx_petsc[i+j] = rstart + j; j++;
12126c995c7dSHong Zhang       }
12136c995c7dSHong Zhang       rstart += size*lu->nb;
12146c995c7dSHong Zhang     }
12156c995c7dSHong Zhang 
12166c995c7dSHong Zhang     for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride;
12176c995c7dSHong Zhang 
12186c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr);
12196c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr);
12206c995c7dSHong Zhang     ierr = PetscFree(idx_pla);CHKERRQ(ierr);
12216c995c7dSHong Zhang     ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr);
12226c995c7dSHong Zhang   }
12236c995c7dSHong Zhang   ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12246c995c7dSHong Zhang   ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12256c995c7dSHong Zhang 
12266c995c7dSHong Zhang   /* Free data */
12276c995c7dSHong Zhang   ierr = VecDestroy(loc_x);CHKERRQ(ierr);
12286c995c7dSHong Zhang   PLA_Obj_free(&v_pla);
12296c995c7dSHong Zhang 
12306c995c7dSHong Zhang   lu->pla_solved = PETSC_TRUE;
12316c995c7dSHong Zhang   PetscFunctionReturn(0);
12326c995c7dSHong Zhang }
12336c995c7dSHong Zhang 
12346c995c7dSHong Zhang #undef __FUNCT__
12356c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense"
1236*719d5645SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,MatFactorInfo *info)
12376c995c7dSHong Zhang {
1238*719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
12396c995c7dSHong Zhang   PetscErrorCode ierr;
1240d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12416c995c7dSHong Zhang   PetscInt       info_pla=0;
12426c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12436c995c7dSHong Zhang 
12446c995c7dSHong Zhang   PetscFunctionBegin;
12456c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12466c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12476c995c7dSHong Zhang     PLA_Obj_free (&lu->pivots);
12486c995c7dSHong Zhang   }
12496c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12506c995c7dSHong Zhang   lu->A = NULL; lu->pivots = NULL;
12516c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12526c995c7dSHong Zhang   PLA_Obj_set_to_zero(lu->A);
12536c995c7dSHong Zhang   PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);
12546c995c7dSHong Zhang 
12556c995c7dSHong Zhang   /* Copy A into lu->A */
12566c995c7dSHong Zhang   PLA_API_begin();
12576c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
12586c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
12596c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
12606c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
12616c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
12626c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
12636c995c7dSHong Zhang   PLA_API_end();
12646c995c7dSHong Zhang 
12656c995c7dSHong Zhang   /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */
12666c995c7dSHong Zhang   info_pla = PLA_LU(lu->A,lu->pivots);
12676c995c7dSHong Zhang   if (info_pla != 0)
12686c995c7dSHong Zhang     SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla);
12696c995c7dSHong Zhang 
12706c995c7dSHong Zhang   lu->CleanUpPlapack = PETSC_TRUE;
12716c995c7dSHong Zhang   lu->rstart         = rstart;
12726c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1273*719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1274*719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
12756c995c7dSHong Zhang   PetscFunctionReturn(0);
12766c995c7dSHong Zhang }
12776c995c7dSHong Zhang 
12786c995c7dSHong Zhang #undef __FUNCT__
12796c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense"
1280*719d5645SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,MatFactorInfo *info)
12816c995c7dSHong Zhang {
1282*719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)F->spptr;
12836c995c7dSHong Zhang   PetscErrorCode ierr;
1284d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12856c995c7dSHong Zhang   PetscInt       info_pla=0;
12866c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12876c995c7dSHong Zhang 
12886c995c7dSHong Zhang   PetscFunctionBegin;
12896c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12906c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12916c995c7dSHong Zhang   }
12926c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12936c995c7dSHong Zhang   lu->A      = NULL;
12946c995c7dSHong Zhang   lu->pivots = NULL;
12956c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12966c995c7dSHong Zhang 
12976c995c7dSHong Zhang   /* Copy A into lu->A */
12986c995c7dSHong Zhang   PLA_API_begin();
12996c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
13006c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
13016c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
13026c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
13036c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
13046c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
13056c995c7dSHong Zhang   PLA_API_end();
13066c995c7dSHong Zhang 
13076c995c7dSHong Zhang   /* Factor P A -> Chol */
13086c995c7dSHong Zhang   info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A);
13096c995c7dSHong Zhang   if (info_pla != 0)
13106c995c7dSHong Zhang     SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla);
13116c995c7dSHong Zhang 
13126c995c7dSHong Zhang   lu->CleanUpPlapack = PETSC_TRUE;
13136c995c7dSHong Zhang   lu->rstart         = rstart;
13146c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1315*719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1316*719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
13176c995c7dSHong Zhang   PetscFunctionReturn(0);
13186c995c7dSHong Zhang }
13196c995c7dSHong Zhang 
13206c995c7dSHong Zhang #undef __FUNCT__
1321ab235cb6SHong Zhang #define __FUNCT__ "MatFactorSymbolic_Plapack_Private"
1322ab235cb6SHong Zhang PetscErrorCode MatFactorSymbolic_Plapack_Private(Mat A,MatFactorInfo *info,Mat *F)
1323ab235cb6SHong Zhang {
1324ab235cb6SHong Zhang   Mat            B = *F;
1325ab235cb6SHong Zhang   Mat_Plapack    *lu;
1326ab235cb6SHong Zhang   PetscErrorCode ierr;
1327d0f46423SBarry Smith   PetscInt       M=A->rmap->N;
1328ab235cb6SHong Zhang   MPI_Comm       comm=((PetscObject)A)->comm,comm_2d;
1329ab235cb6SHong Zhang   PetscMPIInt    size;
1330ab235cb6SHong Zhang   PetscInt       ierror;
1331ab235cb6SHong Zhang 
1332ab235cb6SHong Zhang   PetscFunctionBegin;
1333ab235cb6SHong Zhang   lu = (Mat_Plapack*)(B->spptr);
1334ab235cb6SHong Zhang 
1335ab235cb6SHong Zhang   /* Set default Plapack parameters */
1336ab235cb6SHong Zhang   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1337ab235cb6SHong Zhang   lu->nprows = 1; lu->npcols = size;
1338ab235cb6SHong Zhang   ierror = 0;
1339ab235cb6SHong Zhang   lu->nb     = M/size;
1340ab235cb6SHong Zhang   if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */
1341ab235cb6SHong Zhang 
1342ab235cb6SHong Zhang   /* Set runtime options */
1343ab235cb6SHong Zhang   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr);
1344ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",lu->nprows,&lu->nprows,PETSC_NULL);CHKERRQ(ierr);
1345ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",lu->npcols,&lu->npcols,PETSC_NULL);CHKERRQ(ierr);
1346ab235cb6SHong Zhang 
1347ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr);
1348ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",ierror,&ierror,PETSC_NULL);CHKERRQ(ierr);
1349ab235cb6SHong Zhang   if (ierror){
1350ab235cb6SHong Zhang     PLA_Set_error_checking(ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );
1351ab235cb6SHong Zhang   } else {
1352ab235cb6SHong Zhang     PLA_Set_error_checking(ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );
1353ab235cb6SHong Zhang   }
1354ab235cb6SHong Zhang   lu->ierror = ierror;
1355ab235cb6SHong Zhang 
1356ab235cb6SHong Zhang   lu->nb_alg = 0;
1357ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",lu->nb_alg,&lu->nb_alg,PETSC_NULL);CHKERRQ(ierr);
1358ab235cb6SHong Zhang   if (lu->nb_alg){
1359ab235cb6SHong Zhang     pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,lu->nb_alg);
1360ab235cb6SHong Zhang   }
1361ab235cb6SHong Zhang   PetscOptionsEnd();
1362ab235cb6SHong Zhang 
1363ab235cb6SHong Zhang 
1364ab235cb6SHong Zhang   /* Create a 2D communicator */
1365ab235cb6SHong Zhang   PLA_Comm_1D_to_2D(comm,lu->nprows,lu->npcols,&comm_2d);
1366ab235cb6SHong Zhang   lu->comm_2d = comm_2d;
1367ab235cb6SHong Zhang 
1368ab235cb6SHong Zhang   /* Initialize PLAPACK */
1369ab235cb6SHong Zhang   PLA_Init(comm_2d);
1370ab235cb6SHong Zhang 
1371ab235cb6SHong Zhang   /* Create object distribution template */
1372ab235cb6SHong Zhang   lu->templ = NULL;
1373ab235cb6SHong Zhang   PLA_Temp_create(lu->nb, 0, &lu->templ);
1374ab235cb6SHong Zhang 
1375ab235cb6SHong Zhang   /* Use suggested nb_alg if it is not provided by user */
1376ab235cb6SHong Zhang   if (lu->nb_alg == 0){
1377ab235cb6SHong Zhang     PLA_Environ_nb_alg(PLA_OP_PAN_PAN,lu->templ,&lu->nb_alg);
1378ab235cb6SHong Zhang     pla_Environ_set_nb_alg(PLA_OP_ALL_ALG,lu->nb_alg);
1379ab235cb6SHong Zhang   }
1380ab235cb6SHong Zhang 
1381ab235cb6SHong Zhang   /* Set the datatype */
1382ab235cb6SHong Zhang #if defined(PETSC_USE_COMPLEX)
1383ab235cb6SHong Zhang   lu->datatype = MPI_DOUBLE_COMPLEX;
1384ab235cb6SHong Zhang #else
1385ab235cb6SHong Zhang   lu->datatype = MPI_DOUBLE;
1386ab235cb6SHong Zhang #endif
1387ab235cb6SHong Zhang 
1388ab235cb6SHong Zhang   lu->pla_solved     = PETSC_FALSE; /* MatSolve_Plapack() is called yet */
1389ab235cb6SHong Zhang   lu->mstruct        = DIFFERENT_NONZERO_PATTERN;
1390ab235cb6SHong Zhang   lu->CleanUpPlapack = PETSC_TRUE;
1391ab235cb6SHong Zhang   *F                 = B;
1392ab235cb6SHong Zhang   PetscFunctionReturn(0);
1393ab235cb6SHong Zhang }
1394ab235cb6SHong Zhang 
1395b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */
13962fbe02b9SBarry Smith #undef __FUNCT__
13976c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense"
1398*719d5645SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,MatFactorInfo *info)
139901b82886SBarry Smith {
140001b82886SBarry Smith   PetscErrorCode ierr;
1401b24902e0SBarry Smith   PetscTruth     issymmetric,set;
140201b82886SBarry Smith 
140301b82886SBarry Smith   PetscFunctionBegin;
1404b24902e0SBarry Smith   ierr = MatIsSymmetricKnown(A,&set,&issymmetric); CHKERRQ(ierr);
1405b24902e0SBarry Smith   if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()");
1406*719d5645SBarry Smith   ierr = MatFactorSymbolic_Plapack_Private(F,A,info);CHKERRQ(ierr);
1407*719d5645SBarry Smith   F->ops->choleskyfactornumeric  = MatCholeskyFactorNumeric_MPIDense;
1408b24902e0SBarry Smith   PetscFunctionReturn(0);
140901b82886SBarry Smith }
141001b82886SBarry Smith 
1411b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */
1412b24902e0SBarry Smith #undef __FUNCT__
14136c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense"
1414*719d5645SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,MatFactorInfo *info)
1415b24902e0SBarry Smith {
1416b24902e0SBarry Smith   PetscErrorCode ierr;
1417d0f46423SBarry Smith   PetscInt       M = A->rmap->N;
1418b24902e0SBarry Smith   Mat_Plapack    *lu;
141901b82886SBarry Smith 
1420b24902e0SBarry Smith   PetscFunctionBegin;
1421b24902e0SBarry Smith   ierr = MatFactorSymbolic_Plapack_Private(A,info,F);CHKERRQ(ierr);
1422*719d5645SBarry Smith   lu = (Mat_Plapack*)F->spptr;
1423b24902e0SBarry Smith   ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr);
1424*719d5645SBarry Smith   F->ops->lufactornumeric  = MatLUFactorNumeric_MPIDense;
142501b82886SBarry Smith   PetscFunctionReturn(0);
142601b82886SBarry Smith }
142701b82886SBarry Smith 
142801b82886SBarry Smith #undef __FUNCT__
1429b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_mpidense_plapack"
1430b24902e0SBarry Smith PetscErrorCode MatGetFactor_mpidense_plapack(Mat A,MatFactorType ftype,Mat *F)
143101b82886SBarry Smith {
143201b82886SBarry Smith   PetscErrorCode ierr;
14336c995c7dSHong Zhang   Mat_Plapack    *lu;
14346c995c7dSHong Zhang   PetscMPIInt    size;
14356c995c7dSHong Zhang   PetscInt       M;
143601b82886SBarry Smith 
143701b82886SBarry Smith   PetscFunctionBegin;
143801b82886SBarry Smith   /* Create the factorization matrix */
1439eae6fb2eSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr);
1440d0f46423SBarry Smith   ierr = MatSetSizes(*F,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
1441eae6fb2eSBarry Smith   ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr);
1442b24902e0SBarry Smith   ierr = PetscNewLog(A,Mat_Plapack,&lu);CHKERRQ(ierr);
1443b24902e0SBarry Smith   A->spptr = (void*)lu;
144401b82886SBarry Smith 
1445b24902e0SBarry Smith   lu = (Mat_Plapack*)(A->spptr);
144601b82886SBarry Smith 
1447b24902e0SBarry Smith   /* Set default Plapack parameters */
14486c995c7dSHong Zhang   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1449b24902e0SBarry Smith   lu->nb     = M/size;
1450b24902e0SBarry Smith   if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */
145101b82886SBarry Smith 
1452b24902e0SBarry Smith   /* Set runtime options */
1453b24902e0SBarry Smith   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr);
1454b24902e0SBarry Smith     ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr);
1455b24902e0SBarry Smith   PetscOptionsEnd();
145601b82886SBarry Smith 
1457b24902e0SBarry Smith   /* Create object distribution template */
1458b24902e0SBarry Smith   lu->templ = NULL;
1459b24902e0SBarry Smith   ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr);
1460b24902e0SBarry Smith 
1461b24902e0SBarry Smith   /* Set the datatype */
1462b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX)
1463b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE_COMPLEX;
1464b24902e0SBarry Smith #else
1465b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE;
1466b24902e0SBarry Smith #endif
1467b24902e0SBarry Smith 
1468d0f46423SBarry Smith   ierr = PLA_Matrix_create(lu->datatype,M,A->cmap->N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr);
1469b24902e0SBarry Smith 
1470b24902e0SBarry Smith 
1471b24902e0SBarry Smith   lu->pla_solved     = PETSC_FALSE; /* MatSolve_Plapack() is called yet */
1472b24902e0SBarry Smith 
1473b24902e0SBarry Smith   if (ftype == MAT_FACTOR_LU) {
1474b24902e0SBarry Smith     (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense;
1475b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY) {
14766c995c7dSHong Zhang     (*F)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense;
1477b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices");
1478*719d5645SBarry Smith   (*F)->factor = ftype;
147901b82886SBarry Smith   PetscFunctionReturn(0);
148001b82886SBarry Smith }
148101b82886SBarry Smith #endif
148201b82886SBarry Smith 
14838965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/
148409dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense,
148509dc0095SBarry Smith        MatGetRow_MPIDense,
148609dc0095SBarry Smith        MatRestoreRow_MPIDense,
148709dc0095SBarry Smith        MatMult_MPIDense,
148897304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense,
14897c922b88SBarry Smith        MatMultTranspose_MPIDense,
14907c922b88SBarry Smith        MatMultTransposeAdd_MPIDense,
14918965ea79SLois Curfman McInnes        0,
149209dc0095SBarry Smith        0,
149309dc0095SBarry Smith        0,
149497304618SKris Buschelman /*10*/ 0,
149509dc0095SBarry Smith        0,
149609dc0095SBarry Smith        0,
149709dc0095SBarry Smith        0,
149809dc0095SBarry Smith        MatTranspose_MPIDense,
149997304618SKris Buschelman /*15*/ MatGetInfo_MPIDense,
15006e4ee0c6SHong Zhang        MatEqual_MPIDense,
150109dc0095SBarry Smith        MatGetDiagonal_MPIDense,
15025b2fa520SLois Curfman McInnes        MatDiagonalScale_MPIDense,
150309dc0095SBarry Smith        MatNorm_MPIDense,
150497304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense,
150509dc0095SBarry Smith        MatAssemblyEnd_MPIDense,
150609dc0095SBarry Smith        0,
150709dc0095SBarry Smith        MatSetOption_MPIDense,
150809dc0095SBarry Smith        MatZeroEntries_MPIDense,
150997304618SKris Buschelman /*25*/ MatZeroRows_MPIDense,
1510919b68f7SBarry Smith        0,
151101b82886SBarry Smith        0,
151201b82886SBarry Smith        0,
151301b82886SBarry Smith        0,
151497304618SKris Buschelman /*30*/ MatSetUpPreallocation_MPIDense,
1515273d9f13SBarry Smith        0,
151609dc0095SBarry Smith        0,
151709dc0095SBarry Smith        MatGetArray_MPIDense,
151809dc0095SBarry Smith        MatRestoreArray_MPIDense,
151997304618SKris Buschelman /*35*/ MatDuplicate_MPIDense,
152009dc0095SBarry Smith        0,
152109dc0095SBarry Smith        0,
152209dc0095SBarry Smith        0,
152309dc0095SBarry Smith        0,
152497304618SKris Buschelman /*40*/ 0,
15252ce60cd0SSatish Balay        MatGetSubMatrices_MPIDense,
152609dc0095SBarry Smith        0,
152709dc0095SBarry Smith        MatGetValues_MPIDense,
152809dc0095SBarry Smith        0,
152997304618SKris Buschelman /*45*/ 0,
153009dc0095SBarry Smith        MatScale_MPIDense,
153109dc0095SBarry Smith        0,
153209dc0095SBarry Smith        0,
153309dc0095SBarry Smith        0,
1534521d7252SBarry Smith /*50*/ 0,
153509dc0095SBarry Smith        0,
153609dc0095SBarry Smith        0,
153709dc0095SBarry Smith        0,
153809dc0095SBarry Smith        0,
153997304618SKris Buschelman /*55*/ 0,
154009dc0095SBarry Smith        0,
154109dc0095SBarry Smith        0,
154209dc0095SBarry Smith        0,
154309dc0095SBarry Smith        0,
154497304618SKris Buschelman /*60*/ MatGetSubMatrix_MPIDense,
1545b9b97703SBarry Smith        MatDestroy_MPIDense,
1546b9b97703SBarry Smith        MatView_MPIDense,
1547357abbc8SBarry Smith        0,
154897304618SKris Buschelman        0,
154997304618SKris Buschelman /*65*/ 0,
155097304618SKris Buschelman        0,
155197304618SKris Buschelman        0,
155297304618SKris Buschelman        0,
155397304618SKris Buschelman        0,
155497304618SKris Buschelman /*70*/ 0,
155597304618SKris Buschelman        0,
155697304618SKris Buschelman        0,
155797304618SKris Buschelman        0,
155897304618SKris Buschelman        0,
155997304618SKris Buschelman /*75*/ 0,
156097304618SKris Buschelman        0,
156197304618SKris Buschelman        0,
156297304618SKris Buschelman        0,
156397304618SKris Buschelman        0,
156497304618SKris Buschelman /*80*/ 0,
156597304618SKris Buschelman        0,
156697304618SKris Buschelman        0,
156797304618SKris Buschelman        0,
1568865e5f61SKris Buschelman /*84*/ MatLoad_MPIDense,
1569865e5f61SKris Buschelman        0,
1570865e5f61SKris Buschelman        0,
1571865e5f61SKris Buschelman        0,
1572865e5f61SKris Buschelman        0,
1573865e5f61SKris Buschelman        0,
15742fbe02b9SBarry Smith /*90*/
15752fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
15762fbe02b9SBarry Smith        MatMatMult_MPIDense_MPIDense,
15774ae313f4SHong Zhang        MatMatMultSymbolic_MPIDense_MPIDense,
15782fbe02b9SBarry Smith        MatMatMultNumeric_MPIDense_MPIDense,
15792fbe02b9SBarry Smith #else
1580865e5f61SKris Buschelman        0,
1581865e5f61SKris Buschelman        0,
1582865e5f61SKris Buschelman        0,
15832fbe02b9SBarry Smith #endif
15842fbe02b9SBarry Smith        0,
1585865e5f61SKris Buschelman /*95*/ 0,
1586865e5f61SKris Buschelman        0,
1587865e5f61SKris Buschelman        0,
1588865e5f61SKris Buschelman        0};
15898965ea79SLois Curfman McInnes 
1590273d9f13SBarry Smith EXTERN_C_BEGIN
15914a2ae208SSatish Balay #undef __FUNCT__
1592a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense"
1593be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data)
1594a23d5eceSKris Buschelman {
1595a23d5eceSKris Buschelman   Mat_MPIDense   *a;
1596dfbe8321SBarry Smith   PetscErrorCode ierr;
1597a23d5eceSKris Buschelman 
1598a23d5eceSKris Buschelman   PetscFunctionBegin;
1599a23d5eceSKris Buschelman   mat->preallocated = PETSC_TRUE;
1600a23d5eceSKris Buschelman   /* Note:  For now, when data is specified above, this assumes the user correctly
1601a23d5eceSKris Buschelman    allocates the local dense storage space.  We should add error checking. */
1602a23d5eceSKris Buschelman 
1603a23d5eceSKris Buschelman   a    = (Mat_MPIDense*)mat->data;
1604f69a0ea3SMatthew Knepley   ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr);
1605d0f46423SBarry Smith   ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr);
16065c5985e7SKris Buschelman   ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr);
16075c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr);
160852e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
1609a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1610a23d5eceSKris Buschelman }
1611a23d5eceSKris Buschelman EXTERN_C_END
1612a23d5eceSKris Buschelman 
16130bad9183SKris Buschelman /*MC
1614fafad747SKris Buschelman    MATMPIDENSE - MATMPIDENSE = "mpidense" - A matrix type to be used for distributed dense matrices.
16150bad9183SKris Buschelman 
16160bad9183SKris Buschelman    Options Database Keys:
16170bad9183SKris Buschelman . -mat_type mpidense - sets the matrix type to "mpidense" during a call to MatSetFromOptions()
16180bad9183SKris Buschelman 
16190bad9183SKris Buschelman   Level: beginner
16200bad9183SKris Buschelman 
16217878bbefSBarry Smith   MATMPIDENSE matrices may use direct solvers (LU, Cholesky, and QR)
16227878bbefSBarry Smith   for parallel dense matrices via the external package PLAPACK, if PLAPACK is installed
16237878bbefSBarry Smith   (run config/configure.py with the option --download-plapack)
16247878bbefSBarry Smith 
16257878bbefSBarry Smith 
16267878bbefSBarry Smith   Options Database Keys:
16277878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition
16287878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition
16297878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector
16307878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size
16317878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag
16327878bbefSBarry Smith 
16337878bbefSBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE
16340bad9183SKris Buschelman M*/
16350bad9183SKris Buschelman 
1636a23d5eceSKris Buschelman EXTERN_C_BEGIN
1637a23d5eceSKris Buschelman #undef __FUNCT__
16384a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense"
1639be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat)
1640273d9f13SBarry Smith {
1641273d9f13SBarry Smith   Mat_MPIDense   *a;
1642dfbe8321SBarry Smith   PetscErrorCode ierr;
1643273d9f13SBarry Smith 
1644273d9f13SBarry Smith   PetscFunctionBegin;
164538f2d2fdSLisandro Dalcin   ierr              = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr);
1646b0a32e0cSBarry Smith   mat->data         = (void*)a;
1647273d9f13SBarry Smith   ierr              = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1648273d9f13SBarry Smith   mat->mapping      = 0;
1649273d9f13SBarry Smith 
1650273d9f13SBarry Smith   mat->insertmode = NOT_SET_VALUES;
16517adad957SLisandro Dalcin   ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr);
16527adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr);
1653273d9f13SBarry Smith 
1654d0f46423SBarry Smith   mat->rmap->bs = mat->cmap->bs = 1;
1655d0f46423SBarry Smith   ierr = PetscMapSetUp(mat->rmap);CHKERRQ(ierr);
1656d0f46423SBarry Smith   ierr = PetscMapSetUp(mat->cmap);CHKERRQ(ierr);
1657d0f46423SBarry Smith   a->nvec = mat->cmap->n;
1658273d9f13SBarry Smith 
1659273d9f13SBarry Smith   /* build cache for off array entries formed */
1660273d9f13SBarry Smith   a->donotstash = PETSC_FALSE;
16617adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr);
1662273d9f13SBarry Smith 
1663273d9f13SBarry Smith   /* stuff used for matrix vector multiply */
1664273d9f13SBarry Smith   a->lvec        = 0;
1665273d9f13SBarry Smith   a->Mvctx       = 0;
1666273d9f13SBarry Smith   a->roworiented = PETSC_TRUE;
1667273d9f13SBarry Smith 
1668273d9f13SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C",
1669273d9f13SBarry Smith                                      "MatGetDiagonalBlock_MPIDense",
1670273d9f13SBarry Smith                                      MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr);
1671a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C",
1672a23d5eceSKris Buschelman                                      "MatMPIDenseSetPreallocation_MPIDense",
1673a23d5eceSKris Buschelman                                      MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr);
16744ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C",
16754ae313f4SHong Zhang                                      "MatMatMult_MPIAIJ_MPIDense",
16764ae313f4SHong Zhang                                       MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr);
16774ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C",
16784ae313f4SHong Zhang                                      "MatMatMultSymbolic_MPIAIJ_MPIDense",
16794ae313f4SHong Zhang                                       MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr);
16804ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C",
16814ae313f4SHong Zhang                                      "MatMatMultNumeric_MPIAIJ_MPIDense",
16824ae313f4SHong Zhang                                       MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr);
16837878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1684b24902e0SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_plapack_C",
1685b24902e0SBarry Smith                                      "MatGetFactor_mpidense_plapack",
1686b24902e0SBarry Smith                                       MatGetFactor_mpidense_plapack);CHKERRQ(ierr);
1687db4efbfdSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1688db4efbfdSBarry Smith   ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr);
1689db4efbfdSBarry Smith #endif
1690db4efbfdSBarry Smith 
16917878bbefSBarry Smith #endif
169238aed534SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr);
169301b82886SBarry Smith 
1694273d9f13SBarry Smith   PetscFunctionReturn(0);
1695273d9f13SBarry Smith }
1696273d9f13SBarry Smith EXTERN_C_END
1697273d9f13SBarry Smith 
1698209238afSKris Buschelman /*MC
1699002d173eSKris Buschelman    MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices.
1700209238afSKris Buschelman 
1701209238afSKris Buschelman    This matrix type is identical to MATSEQDENSE when constructed with a single process communicator,
1702209238afSKris Buschelman    and MATMPIDENSE otherwise.
1703209238afSKris Buschelman 
1704209238afSKris Buschelman    Options Database Keys:
1705209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions()
1706209238afSKris Buschelman 
1707209238afSKris Buschelman   Level: beginner
1708209238afSKris Buschelman 
170901b82886SBarry Smith 
1710209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE
1711209238afSKris Buschelman M*/
1712209238afSKris Buschelman 
1713209238afSKris Buschelman EXTERN_C_BEGIN
1714209238afSKris Buschelman #undef __FUNCT__
1715209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense"
1716be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A)
1717dfbe8321SBarry Smith {
17186849ba73SBarry Smith   PetscErrorCode ierr;
171913f74950SBarry Smith   PetscMPIInt    size;
1720209238afSKris Buschelman 
1721209238afSKris Buschelman   PetscFunctionBegin;
17227adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1723209238afSKris Buschelman   if (size == 1) {
1724209238afSKris Buschelman     ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr);
1725209238afSKris Buschelman   } else {
1726209238afSKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
1727209238afSKris Buschelman   }
1728209238afSKris Buschelman   PetscFunctionReturn(0);
1729209238afSKris Buschelman }
1730209238afSKris Buschelman EXTERN_C_END
1731209238afSKris Buschelman 
17324a2ae208SSatish Balay #undef __FUNCT__
17334a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation"
1734273d9f13SBarry Smith /*@C
1735273d9f13SBarry Smith    MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries
1736273d9f13SBarry Smith 
1737273d9f13SBarry Smith    Not collective
1738273d9f13SBarry Smith 
1739273d9f13SBarry Smith    Input Parameters:
1740273d9f13SBarry Smith .  A - the matrix
1741273d9f13SBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1742273d9f13SBarry Smith    to control all matrix memory allocation.
1743273d9f13SBarry Smith 
1744273d9f13SBarry Smith    Notes:
1745273d9f13SBarry Smith    The dense format is fully compatible with standard Fortran 77
1746273d9f13SBarry Smith    storage by columns.
1747273d9f13SBarry Smith 
1748273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1749273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1750273d9f13SBarry Smith    set data=PETSC_NULL.
1751273d9f13SBarry Smith 
1752273d9f13SBarry Smith    Level: intermediate
1753273d9f13SBarry Smith 
1754273d9f13SBarry Smith .keywords: matrix,dense, parallel
1755273d9f13SBarry Smith 
1756273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
1757273d9f13SBarry Smith @*/
1758be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data)
1759273d9f13SBarry Smith {
1760dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar *);
1761273d9f13SBarry Smith 
1762273d9f13SBarry Smith   PetscFunctionBegin;
1763565567f0SKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1764a23d5eceSKris Buschelman   if (f) {
1765a23d5eceSKris Buschelman     ierr = (*f)(mat,data);CHKERRQ(ierr);
1766a23d5eceSKris Buschelman   }
1767273d9f13SBarry Smith   PetscFunctionReturn(0);
1768273d9f13SBarry Smith }
1769273d9f13SBarry Smith 
17704a2ae208SSatish Balay #undef __FUNCT__
17714a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense"
17728965ea79SLois Curfman McInnes /*@C
177339ddd567SLois Curfman McInnes    MatCreateMPIDense - Creates a sparse parallel matrix in dense format.
17748965ea79SLois Curfman McInnes 
1775db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1776db81eaa0SLois Curfman McInnes 
17778965ea79SLois Curfman McInnes    Input Parameters:
1778db81eaa0SLois Curfman McInnes +  comm - MPI communicator
17798965ea79SLois Curfman McInnes .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
1780db81eaa0SLois Curfman McInnes .  n - number of local columns (or PETSC_DECIDE to have calculated if N is given)
17818965ea79SLois Curfman McInnes .  M - number of global rows (or PETSC_DECIDE to have calculated if m is given)
1782db81eaa0SLois Curfman McInnes .  N - number of global columns (or PETSC_DECIDE to have calculated if n is given)
17837f5ff6fdSBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc
1784dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
17858965ea79SLois Curfman McInnes 
17868965ea79SLois Curfman McInnes    Output Parameter:
1787477f1c0bSLois Curfman McInnes .  A - the matrix
17888965ea79SLois Curfman McInnes 
1789b259b22eSLois Curfman McInnes    Notes:
179039ddd567SLois Curfman McInnes    The dense format is fully compatible with standard Fortran 77
179139ddd567SLois Curfman McInnes    storage by columns.
17928965ea79SLois Curfman McInnes 
179318f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
179418f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
17957f5ff6fdSBarry Smith    set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users).
179618f449edSLois Curfman McInnes 
17978965ea79SLois Curfman McInnes    The user MUST specify either the local or global matrix dimensions
17988965ea79SLois Curfman McInnes    (possibly both).
17998965ea79SLois Curfman McInnes 
1800027ccd11SLois Curfman McInnes    Level: intermediate
1801027ccd11SLois Curfman McInnes 
180239ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel
18038965ea79SLois Curfman McInnes 
180439ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
18058965ea79SLois Curfman McInnes @*/
1806be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A)
18078965ea79SLois Curfman McInnes {
18086849ba73SBarry Smith   PetscErrorCode ierr;
180913f74950SBarry Smith   PetscMPIInt    size;
18108965ea79SLois Curfman McInnes 
18113a40ed3dSBarry Smith   PetscFunctionBegin;
1812f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1813f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr);
1814273d9f13SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1815273d9f13SBarry Smith   if (size > 1) {
1816273d9f13SBarry Smith     ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr);
1817273d9f13SBarry Smith     ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1818273d9f13SBarry Smith   } else {
1819273d9f13SBarry Smith     ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1820273d9f13SBarry Smith     ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
18218c469469SLois Curfman McInnes   }
18223a40ed3dSBarry Smith   PetscFunctionReturn(0);
18238965ea79SLois Curfman McInnes }
18248965ea79SLois Curfman McInnes 
18254a2ae208SSatish Balay #undef __FUNCT__
18264a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense"
18276849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
18288965ea79SLois Curfman McInnes {
18298965ea79SLois Curfman McInnes   Mat            mat;
18303501a2bdSLois Curfman McInnes   Mat_MPIDense   *a,*oldmat = (Mat_MPIDense*)A->data;
1831dfbe8321SBarry Smith   PetscErrorCode ierr;
18328965ea79SLois Curfman McInnes 
18333a40ed3dSBarry Smith   PetscFunctionBegin;
18348965ea79SLois Curfman McInnes   *newmat       = 0;
18357adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr);
1836d0f46423SBarry Smith   ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
18377adad957SLisandro Dalcin   ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr);
1838834f8fabSBarry Smith   a                 = (Mat_MPIDense*)mat->data;
1839e04c1aa4SHong Zhang   ierr              = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
18403501a2bdSLois Curfman McInnes   mat->factor       = A->factor;
1841c456f294SBarry Smith   mat->assembled    = PETSC_TRUE;
1842273d9f13SBarry Smith   mat->preallocated = PETSC_TRUE;
18438965ea79SLois Curfman McInnes 
1844d0f46423SBarry Smith   mat->rmap->rstart     = A->rmap->rstart;
1845d0f46423SBarry Smith   mat->rmap->rend       = A->rmap->rend;
18468965ea79SLois Curfman McInnes   a->size              = oldmat->size;
18478965ea79SLois Curfman McInnes   a->rank              = oldmat->rank;
1848e0fa3b82SLois Curfman McInnes   mat->insertmode      = NOT_SET_VALUES;
1849b9b97703SBarry Smith   a->nvec              = oldmat->nvec;
18503782ba37SSatish Balay   a->donotstash        = oldmat->donotstash;
1851e04c1aa4SHong Zhang 
1852d0f46423SBarry Smith   ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
1853d0f46423SBarry Smith   ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
18547adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr);
18558965ea79SLois Curfman McInnes 
1856329f5518SBarry Smith   ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
18575609ef8eSBarry Smith   ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr);
185852e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
185901b82886SBarry Smith 
186001b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
186101b82886SBarry Smith   ierr = PetscMemcpy(mat->spptr,A->spptr,sizeof(Mat_Plapack));CHKERRQ(ierr);
186201b82886SBarry Smith #endif
18638965ea79SLois Curfman McInnes   *newmat = mat;
18643a40ed3dSBarry Smith   PetscFunctionReturn(0);
18658965ea79SLois Curfman McInnes }
18668965ea79SLois Curfman McInnes 
1867e090d566SSatish Balay #include "petscsys.h"
18688965ea79SLois Curfman McInnes 
18694a2ae208SSatish Balay #undef __FUNCT__
18704a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile"
1871a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat)
187290ace30eSBarry Smith {
18736849ba73SBarry Smith   PetscErrorCode ierr;
187432dcc486SBarry Smith   PetscMPIInt    rank,size;
187513f74950SBarry Smith   PetscInt       *rowners,i,m,nz,j;
187687828ca2SBarry Smith   PetscScalar    *array,*vals,*vals_ptr;
187790ace30eSBarry Smith   MPI_Status     status;
187890ace30eSBarry Smith 
18793a40ed3dSBarry Smith   PetscFunctionBegin;
1880d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
1881d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
188290ace30eSBarry Smith 
188390ace30eSBarry Smith   /* determine ownership of all rows */
188490ace30eSBarry Smith   m          = M/size + ((M % size) > rank);
188513f74950SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr);
188613f74950SBarry Smith   ierr       = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr);
188790ace30eSBarry Smith   rowners[0] = 0;
188890ace30eSBarry Smith   for (i=2; i<=size; i++) {
188990ace30eSBarry Smith     rowners[i] += rowners[i-1];
189090ace30eSBarry Smith   }
189190ace30eSBarry Smith 
1892f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1893f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1894be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1895878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
189690ace30eSBarry Smith   ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr);
189790ace30eSBarry Smith 
189890ace30eSBarry Smith   if (!rank) {
189987828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
190090ace30eSBarry Smith 
190190ace30eSBarry Smith     /* read in my part of the matrix numerical values  */
19020752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr);
190390ace30eSBarry Smith 
190490ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
190590ace30eSBarry Smith     vals_ptr = vals;
190690ace30eSBarry Smith     for (i=0; i<m; i++) {
190790ace30eSBarry Smith       for (j=0; j<N; j++) {
190890ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
190990ace30eSBarry Smith       }
191090ace30eSBarry Smith     }
191190ace30eSBarry Smith 
191290ace30eSBarry Smith     /* read in other processors and ship out */
191390ace30eSBarry Smith     for (i=1; i<size; i++) {
191490ace30eSBarry Smith       nz   = (rowners[i+1] - rowners[i])*N;
19150752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
19167adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr);
191790ace30eSBarry Smith     }
19183a40ed3dSBarry Smith   } else {
191990ace30eSBarry Smith     /* receive numeric values */
192087828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
192190ace30eSBarry Smith 
192290ace30eSBarry Smith     /* receive message of values*/
19237adad957SLisandro Dalcin     ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr);
192490ace30eSBarry Smith 
192590ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
192690ace30eSBarry Smith     vals_ptr = vals;
192790ace30eSBarry Smith     for (i=0; i<m; i++) {
192890ace30eSBarry Smith       for (j=0; j<N; j++) {
192990ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
193090ace30eSBarry Smith       }
193190ace30eSBarry Smith     }
193290ace30eSBarry Smith   }
1933606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
1934606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
19356d4a8577SBarry Smith   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19366d4a8577SBarry Smith   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19373a40ed3dSBarry Smith   PetscFunctionReturn(0);
193890ace30eSBarry Smith }
193990ace30eSBarry Smith 
19404a2ae208SSatish Balay #undef __FUNCT__
19414a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense"
1942a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat)
19438965ea79SLois Curfman McInnes {
19448965ea79SLois Curfman McInnes   Mat            A;
194587828ca2SBarry Smith   PetscScalar    *vals,*svals;
194619bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
19478965ea79SLois Curfman McInnes   MPI_Status     status;
194813f74950SBarry Smith   PetscMPIInt    rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz;
194913f74950SBarry Smith   PetscInt       header[4],*rowlengths = 0,M,N,*cols;
195013f74950SBarry Smith   PetscInt       *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols;
195113f74950SBarry Smith   PetscInt       i,nz,j,rstart,rend;
195213f74950SBarry Smith   int            fd;
19536849ba73SBarry Smith   PetscErrorCode ierr;
19548965ea79SLois Curfman McInnes 
19553a40ed3dSBarry Smith   PetscFunctionBegin;
1956d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1957d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
19588965ea79SLois Curfman McInnes   if (!rank) {
1959b0a32e0cSBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
19600752156aSBarry Smith     ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr);
1961552e946dSBarry Smith     if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object");
19628965ea79SLois Curfman McInnes   }
19638965ea79SLois Curfman McInnes 
196413f74950SBarry Smith   ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr);
196590ace30eSBarry Smith   M = header[1]; N = header[2]; nz = header[3];
196690ace30eSBarry Smith 
196790ace30eSBarry Smith   /*
196890ace30eSBarry Smith        Handle case where matrix is stored on disk as a dense matrix
196990ace30eSBarry Smith   */
197090ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) {
1971be5d1d56SKris Buschelman     ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr);
19723a40ed3dSBarry Smith     PetscFunctionReturn(0);
197390ace30eSBarry Smith   }
197490ace30eSBarry Smith 
19758965ea79SLois Curfman McInnes   /* determine ownership of all rows */
1976e44c0bd4SBarry Smith   m          = PetscMPIIntCast(M/size + ((M % size) > rank));
1977e44c0bd4SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr);
1978ca161407SBarry Smith   ierr       = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr);
19798965ea79SLois Curfman McInnes   rowners[0] = 0;
19808965ea79SLois Curfman McInnes   for (i=2; i<=size; i++) {
19818965ea79SLois Curfman McInnes     rowners[i] += rowners[i-1];
19828965ea79SLois Curfman McInnes   }
19838965ea79SLois Curfman McInnes   rstart = rowners[rank];
19848965ea79SLois Curfman McInnes   rend   = rowners[rank+1];
19858965ea79SLois Curfman McInnes 
19868965ea79SLois Curfman McInnes   /* distribute row lengths to all processors */
198713f74950SBarry Smith   ierr    = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr);
19888965ea79SLois Curfman McInnes   offlens = ourlens + (rend-rstart);
19898965ea79SLois Curfman McInnes   if (!rank) {
199013f74950SBarry Smith     ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
19910752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
199213f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr);
19938965ea79SLois Curfman McInnes     for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i];
19941466f1e1SBarry Smith     ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
1995606d414cSSatish Balay     ierr = PetscFree(sndcounts);CHKERRQ(ierr);
1996ca161407SBarry Smith   } else {
19971466f1e1SBarry Smith     ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
19988965ea79SLois Curfman McInnes   }
19998965ea79SLois Curfman McInnes 
20008965ea79SLois Curfman McInnes   if (!rank) {
20018965ea79SLois Curfman McInnes     /* calculate the number of nonzeros on each processor */
200213f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr);
200313f74950SBarry Smith     ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr);
20048965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
20058965ea79SLois Curfman McInnes       for (j=rowners[i]; j< rowners[i+1]; j++) {
20068965ea79SLois Curfman McInnes         procsnz[i] += rowlengths[j];
20078965ea79SLois Curfman McInnes       }
20088965ea79SLois Curfman McInnes     }
2009606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
20108965ea79SLois Curfman McInnes 
20118965ea79SLois Curfman McInnes     /* determine max buffer needed and allocate it */
20128965ea79SLois Curfman McInnes     maxnz = 0;
20138965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
20140452661fSBarry Smith       maxnz = PetscMax(maxnz,procsnz[i]);
20158965ea79SLois Curfman McInnes     }
201613f74950SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr);
20178965ea79SLois Curfman McInnes 
20188965ea79SLois Curfman McInnes     /* read in my part of the matrix column indices  */
20198965ea79SLois Curfman McInnes     nz = procsnz[0];
202013f74950SBarry Smith     ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
20210752156aSBarry Smith     ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr);
20228965ea79SLois Curfman McInnes 
20238965ea79SLois Curfman McInnes     /* read in every one elses and ship off */
20248965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
20258965ea79SLois Curfman McInnes       nz   = procsnz[i];
20260752156aSBarry Smith       ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
202713f74950SBarry Smith       ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr);
20288965ea79SLois Curfman McInnes     }
2029606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
20303a40ed3dSBarry Smith   } else {
20318965ea79SLois Curfman McInnes     /* determine buffer space needed for message */
20328965ea79SLois Curfman McInnes     nz = 0;
20338965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20348965ea79SLois Curfman McInnes       nz += ourlens[i];
20358965ea79SLois Curfman McInnes     }
203613f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
20378965ea79SLois Curfman McInnes 
20388965ea79SLois Curfman McInnes     /* receive message of column indices*/
203913f74950SBarry Smith     ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr);
204013f74950SBarry Smith     ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr);
204129bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
20428965ea79SLois Curfman McInnes   }
20438965ea79SLois Curfman McInnes 
20448965ea79SLois Curfman McInnes   /* loop over local rows, determining number of off diagonal entries */
204513f74950SBarry Smith   ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr);
20468965ea79SLois Curfman McInnes   jj = 0;
20478965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
20488965ea79SLois Curfman McInnes     for (j=0; j<ourlens[i]; j++) {
20498965ea79SLois Curfman McInnes       if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++;
20508965ea79SLois Curfman McInnes       jj++;
20518965ea79SLois Curfman McInnes     }
20528965ea79SLois Curfman McInnes   }
20538965ea79SLois Curfman McInnes 
20548965ea79SLois Curfman McInnes   /* create our matrix */
20558965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
20568965ea79SLois Curfman McInnes     ourlens[i] -= offlens[i];
20578965ea79SLois Curfman McInnes   }
2058f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
2059f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
2060be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
2061878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
20628965ea79SLois Curfman McInnes   A = *newmat;
20638965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
20648965ea79SLois Curfman McInnes     ourlens[i] += offlens[i];
20658965ea79SLois Curfman McInnes   }
20668965ea79SLois Curfman McInnes 
20678965ea79SLois Curfman McInnes   if (!rank) {
206887828ca2SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20698965ea79SLois Curfman McInnes 
20708965ea79SLois Curfman McInnes     /* read in my part of the matrix numerical values  */
20718965ea79SLois Curfman McInnes     nz = procsnz[0];
20720752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20738965ea79SLois Curfman McInnes 
20748965ea79SLois Curfman McInnes     /* insert into matrix */
20758965ea79SLois Curfman McInnes     jj      = rstart;
20768965ea79SLois Curfman McInnes     smycols = mycols;
20778965ea79SLois Curfman McInnes     svals   = vals;
20788965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20798965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
20808965ea79SLois Curfman McInnes       smycols += ourlens[i];
20818965ea79SLois Curfman McInnes       svals   += ourlens[i];
20828965ea79SLois Curfman McInnes       jj++;
20838965ea79SLois Curfman McInnes     }
20848965ea79SLois Curfman McInnes 
20858965ea79SLois Curfman McInnes     /* read in other processors and ship out */
20868965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
20878965ea79SLois Curfman McInnes       nz   = procsnz[i];
20880752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20897adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr);
20908965ea79SLois Curfman McInnes     }
2091606d414cSSatish Balay     ierr = PetscFree(procsnz);CHKERRQ(ierr);
20923a40ed3dSBarry Smith   } else {
20938965ea79SLois Curfman McInnes     /* receive numeric values */
209484d528b1SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20958965ea79SLois Curfman McInnes 
20968965ea79SLois Curfman McInnes     /* receive message of values*/
20977adad957SLisandro Dalcin     ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr);
2098ca161407SBarry Smith     ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr);
209929bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
21008965ea79SLois Curfman McInnes 
21018965ea79SLois Curfman McInnes     /* insert into matrix */
21028965ea79SLois Curfman McInnes     jj      = rstart;
21038965ea79SLois Curfman McInnes     smycols = mycols;
21048965ea79SLois Curfman McInnes     svals   = vals;
21058965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
21068965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
21078965ea79SLois Curfman McInnes       smycols += ourlens[i];
21088965ea79SLois Curfman McInnes       svals   += ourlens[i];
21098965ea79SLois Curfman McInnes       jj++;
21108965ea79SLois Curfman McInnes     }
21118965ea79SLois Curfman McInnes   }
2112606d414cSSatish Balay   ierr = PetscFree(ourlens);CHKERRQ(ierr);
2113606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
2114606d414cSSatish Balay   ierr = PetscFree(mycols);CHKERRQ(ierr);
2115606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
21168965ea79SLois Curfman McInnes 
21176d4a8577SBarry Smith   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
21186d4a8577SBarry Smith   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
21193a40ed3dSBarry Smith   PetscFunctionReturn(0);
21208965ea79SLois Curfman McInnes }
212190ace30eSBarry Smith 
21226e4ee0c6SHong Zhang #undef __FUNCT__
21236e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense"
21246e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag)
21256e4ee0c6SHong Zhang {
21266e4ee0c6SHong Zhang   Mat_MPIDense   *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data;
21276e4ee0c6SHong Zhang   Mat            a,b;
21286e4ee0c6SHong Zhang   PetscTruth     flg;
21296e4ee0c6SHong Zhang   PetscErrorCode ierr;
213090ace30eSBarry Smith 
21316e4ee0c6SHong Zhang   PetscFunctionBegin;
21326e4ee0c6SHong Zhang   a = matA->A;
21336e4ee0c6SHong Zhang   b = matB->A;
21346e4ee0c6SHong Zhang   ierr = MatEqual(a,b,&flg);CHKERRQ(ierr);
21357adad957SLisandro Dalcin   ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr);
21366e4ee0c6SHong Zhang   PetscFunctionReturn(0);
21376e4ee0c6SHong Zhang }
213890ace30eSBarry Smith 
213909d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
214009d27a7eSBarry Smith 
214109d27a7eSBarry Smith #undef __FUNCT__
214209d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage"
214309d27a7eSBarry Smith /*@C
214409d27a7eSBarry Smith   PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK.
214509d27a7eSBarry Smith   Level: developer
214609d27a7eSBarry Smith 
214709d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK
214809d27a7eSBarry Smith .seealso: PetscFinalize()
214909d27a7eSBarry Smith @*/
215009d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void)
215109d27a7eSBarry Smith {
215209d27a7eSBarry Smith   PetscErrorCode ierr;
215309d27a7eSBarry Smith 
215409d27a7eSBarry Smith   PetscFunctionBegin;
215509d27a7eSBarry Smith   ierr = PLA_Finalize();CHKERRQ(ierr);
215609d27a7eSBarry Smith   PetscFunctionReturn(0);
215709d27a7eSBarry Smith }
215809d27a7eSBarry Smith 
215909d27a7eSBarry Smith #undef __FUNCT__
216009d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage"
216109d27a7eSBarry Smith /*@C
216209d27a7eSBarry Smith   PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is
2163db4efbfdSBarry Smith   called from MatCreate_MPIDense() the first time an MPI dense matrix is called.
216409d27a7eSBarry Smith 
216509d27a7eSBarry Smith   Input Parameter:
2166db4efbfdSBarry Smith .   comm - the communicator the matrix lives on
216709d27a7eSBarry Smith 
216809d27a7eSBarry Smith   Level: developer
216909d27a7eSBarry Smith 
2170db4efbfdSBarry Smith    Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the
2171db4efbfdSBarry Smith   same communicator (because there is some global state that is initialized and used for all matrices). In addition if
2172db4efbfdSBarry Smith   PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators
2173db4efbfdSBarry Smith   cannot overlap.
2174db4efbfdSBarry Smith 
217509d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK
217609d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize()
217709d27a7eSBarry Smith @*/
2178db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm)
217909d27a7eSBarry Smith {
2180eae6fb2eSBarry Smith   PetscMPIInt    size;
218109d27a7eSBarry Smith   PetscErrorCode ierr;
218209d27a7eSBarry Smith 
218309d27a7eSBarry Smith   PetscFunctionBegin;
218409d27a7eSBarry Smith   if (!PLA_Initialized(PETSC_NULL)) {
218509d27a7eSBarry Smith 
218609d27a7eSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
218704fea9ffSBarry Smith     Plapack_nprows = 1;
218804fea9ffSBarry Smith     Plapack_npcols = size;
218909d27a7eSBarry Smith 
2190aeccfd6fSBarry Smith     ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr);
2191aeccfd6fSBarry Smith       ierr = PetscOptionsInt("-plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr);
2192aeccfd6fSBarry Smith       ierr = PetscOptionsInt("-plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr);
219379b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG)
219479b0a62dSBarry Smith       Plapack_ierror = 3;
219579b0a62dSBarry Smith #else
219679b0a62dSBarry Smith       Plapack_ierror = 0;
219779b0a62dSBarry Smith #endif
2198eae6fb2eSBarry Smith       ierr = PetscOptionsInt("-plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr);
2199eae6fb2eSBarry Smith       if (Plapack_ierror){
2200eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr);
2201aeccfd6fSBarry Smith       } else {
2202eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr);
2203aeccfd6fSBarry Smith       }
2204aeccfd6fSBarry Smith 
2205eae6fb2eSBarry Smith       Plapack_nb_alg = 0;
220679b0a62dSBarry Smith       ierr = PetscOptionsInt("-plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr);
2207eae6fb2eSBarry Smith       if (Plapack_nb_alg) {
2208eae6fb2eSBarry Smith         ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr);
2209aeccfd6fSBarry Smith       }
2210aeccfd6fSBarry Smith     PetscOptionsEnd();
2211aeccfd6fSBarry Smith 
221204fea9ffSBarry Smith     ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr);
221304fea9ffSBarry Smith     ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr);
221409d27a7eSBarry Smith     ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr);
221509d27a7eSBarry Smith   }
221609d27a7eSBarry Smith   PetscFunctionReturn(0);
221709d27a7eSBarry Smith }
221890ace30eSBarry Smith 
221990ace30eSBarry Smith 
222009d27a7eSBarry Smith #endif
2221