xref: /petsc/src/mat/impls/dense/mpi/mpidense.c (revision 5d00a290c924ab1b2c83000ba3a0ae1df9cca2e4)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2be1d678aSKris Buschelman 
3ed3cc1f0SBarry Smith /*
4ed3cc1f0SBarry Smith    Basic functions for basic parallel dense matrices.
5ed3cc1f0SBarry Smith */
6ed3cc1f0SBarry Smith 
704fea9ffSBarry Smith 
87c4f633dSBarry 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;
1895d0c19d7SBarry Smith   PetscInt       i,j,rstart,rend,nrows,ncols,nlrows,nlcols;
1905d0c19d7SBarry Smith   const PetscInt *irow,*icol;
19187828ca2SBarry Smith   PetscScalar    *av,*bv,*v = lmat->v;
192ca3fa75bSLois Curfman McInnes   Mat            newmat;
193ca3fa75bSLois Curfman McInnes 
194ca3fa75bSLois Curfman McInnes   PetscFunctionBegin;
195ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
196ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
197b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
198b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
199ca3fa75bSLois Curfman McInnes 
200ca3fa75bSLois Curfman McInnes   /* No parallel redistribution currently supported! Should really check each index set
2017eba5e9cSLois Curfman McInnes      to comfirm that it is OK.  ... Currently supports only submatrix same partitioning as
2027eba5e9cSLois Curfman McInnes      original matrix! */
203ca3fa75bSLois Curfman McInnes 
204ca3fa75bSLois Curfman McInnes   ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr);
2057eba5e9cSLois Curfman McInnes   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
206ca3fa75bSLois Curfman McInnes 
207ca3fa75bSLois Curfman McInnes   /* Check submatrix call */
208ca3fa75bSLois Curfman McInnes   if (scall == MAT_REUSE_MATRIX) {
20929bbc08cSBarry Smith     /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */
2107eba5e9cSLois Curfman McInnes     /* Really need to test rows and column sizes! */
211ca3fa75bSLois Curfman McInnes     newmat = *B;
212ca3fa75bSLois Curfman McInnes   } else {
213ca3fa75bSLois Curfman McInnes     /* Create and fill new matrix */
2147adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
215f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,cs,PETSC_DECIDE,ncols);CHKERRQ(ierr);
2167adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
217878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
218ca3fa75bSLois Curfman McInnes   }
219ca3fa75bSLois Curfman McInnes 
220ca3fa75bSLois Curfman McInnes   /* Now extract the data pointers and do the copy, column at a time */
221ca3fa75bSLois Curfman McInnes   newmatd = (Mat_MPIDense*)newmat->data;
222ca3fa75bSLois Curfman McInnes   bv      = ((Mat_SeqDense *)newmatd->A->data)->v;
223ca3fa75bSLois Curfman McInnes 
224ca3fa75bSLois Curfman McInnes   for (i=0; i<ncols; i++) {
225ca3fa75bSLois Curfman McInnes     av = v + nlrows*icol[i];
226ca3fa75bSLois Curfman McInnes     for (j=0; j<nrows; j++) {
2277eba5e9cSLois Curfman McInnes       *bv++ = av[irow[j] - rstart];
228ca3fa75bSLois Curfman McInnes     }
229ca3fa75bSLois Curfman McInnes   }
230ca3fa75bSLois Curfman McInnes 
231ca3fa75bSLois Curfman McInnes   /* Assemble the matrices so that the correct flags are set */
232ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
233ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
234ca3fa75bSLois Curfman McInnes 
235ca3fa75bSLois Curfman McInnes   /* Free work space */
236ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
237ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
238ca3fa75bSLois Curfman McInnes   *B = newmat;
239ca3fa75bSLois Curfman McInnes   PetscFunctionReturn(0);
240ca3fa75bSLois Curfman McInnes }
241ca3fa75bSLois Curfman McInnes 
2424a2ae208SSatish Balay #undef __FUNCT__
2434a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense"
244dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[])
245ff14e315SSatish Balay {
2463a40ed3dSBarry Smith   PetscFunctionBegin;
2473a40ed3dSBarry Smith   PetscFunctionReturn(0);
248ff14e315SSatish Balay }
249ff14e315SSatish Balay 
2504a2ae208SSatish Balay #undef __FUNCT__
2514a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense"
252dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode)
2538965ea79SLois Curfman McInnes {
25439ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
2557adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)mat)->comm;
256dfbe8321SBarry Smith   PetscErrorCode ierr;
25713f74950SBarry Smith   PetscInt       nstash,reallocs;
2588965ea79SLois Curfman McInnes   InsertMode     addv;
2598965ea79SLois Curfman McInnes 
2603a40ed3dSBarry Smith   PetscFunctionBegin;
2618965ea79SLois Curfman McInnes   /* make sure all processors are either in INSERTMODE or ADDMODE */
262ca161407SBarry Smith   ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr);
2637056b6fcSBarry Smith   if (addv == (ADD_VALUES|INSERT_VALUES)) {
26429bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs");
2658965ea79SLois Curfman McInnes   }
266e0fa3b82SLois Curfman McInnes   mat->insertmode = addv; /* in case this processor had no cache */
2678965ea79SLois Curfman McInnes 
268d0f46423SBarry Smith   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr);
2698798bf22SSatish Balay   ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr);
270ae15b995SBarry Smith   ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr);
2713a40ed3dSBarry Smith   PetscFunctionReturn(0);
2728965ea79SLois Curfman McInnes }
2738965ea79SLois Curfman McInnes 
2744a2ae208SSatish Balay #undef __FUNCT__
2754a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense"
276dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode)
2778965ea79SLois Curfman McInnes {
27839ddd567SLois Curfman McInnes   Mat_MPIDense    *mdn=(Mat_MPIDense*)mat->data;
2796849ba73SBarry Smith   PetscErrorCode  ierr;
28013f74950SBarry Smith   PetscInt        i,*row,*col,flg,j,rstart,ncols;
28113f74950SBarry Smith   PetscMPIInt     n;
28287828ca2SBarry Smith   PetscScalar     *val;
283e0fa3b82SLois Curfman McInnes   InsertMode      addv=mat->insertmode;
2848965ea79SLois Curfman McInnes 
2853a40ed3dSBarry Smith   PetscFunctionBegin;
2868965ea79SLois Curfman McInnes   /*  wait on receives */
2877ef1d9bdSSatish Balay   while (1) {
2888798bf22SSatish Balay     ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr);
2897ef1d9bdSSatish Balay     if (!flg) break;
2908965ea79SLois Curfman McInnes 
2917ef1d9bdSSatish Balay     for (i=0; i<n;) {
2927ef1d9bdSSatish Balay       /* Now identify the consecutive vals belonging to the same row */
2937ef1d9bdSSatish Balay       for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; }
2947ef1d9bdSSatish Balay       if (j < n) ncols = j-i;
2957ef1d9bdSSatish Balay       else       ncols = n-i;
2967ef1d9bdSSatish Balay       /* Now assemble all these values with a single function call */
2977ef1d9bdSSatish Balay       ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr);
2987ef1d9bdSSatish Balay       i = j;
2998965ea79SLois Curfman McInnes     }
3007ef1d9bdSSatish Balay   }
3018798bf22SSatish Balay   ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr);
3028965ea79SLois Curfman McInnes 
30339ddd567SLois Curfman McInnes   ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr);
30439ddd567SLois Curfman McInnes   ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr);
3058965ea79SLois Curfman McInnes 
3066d4a8577SBarry Smith   if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) {
30739ddd567SLois Curfman McInnes     ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
3088965ea79SLois Curfman McInnes   }
3093a40ed3dSBarry Smith   PetscFunctionReturn(0);
3108965ea79SLois Curfman McInnes }
3118965ea79SLois Curfman McInnes 
3124a2ae208SSatish Balay #undef __FUNCT__
3134a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense"
314dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A)
3158965ea79SLois Curfman McInnes {
316dfbe8321SBarry Smith   PetscErrorCode ierr;
31739ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3183a40ed3dSBarry Smith 
3193a40ed3dSBarry Smith   PetscFunctionBegin;
3203a40ed3dSBarry Smith   ierr = MatZeroEntries(l->A);CHKERRQ(ierr);
3213a40ed3dSBarry Smith   PetscFunctionReturn(0);
3228965ea79SLois Curfman McInnes }
3238965ea79SLois Curfman McInnes 
3248965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the
3258965ea79SLois Curfman McInnes    matrix is square and the column and row owerships are identical.
3268965ea79SLois Curfman McInnes    This is a BUG. The only way to fix it seems to be to access
3273501a2bdSLois Curfman McInnes    mdn->A and mdn->B directly and not through the MatZeroRows()
3288965ea79SLois Curfman McInnes    routine.
3298965ea79SLois Curfman McInnes */
3304a2ae208SSatish Balay #undef __FUNCT__
3314a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense"
332f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
3338965ea79SLois Curfman McInnes {
33439ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3356849ba73SBarry Smith   PetscErrorCode ierr;
336d0f46423SBarry Smith   PetscInt       i,*owners = A->rmap->range;
33713f74950SBarry Smith   PetscInt       *nprocs,j,idx,nsends;
33813f74950SBarry Smith   PetscInt       nmax,*svalues,*starts,*owner,nrecvs;
3397adad957SLisandro Dalcin   PetscInt       *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source;
34013f74950SBarry Smith   PetscInt       *lens,*lrows,*values;
34113f74950SBarry Smith   PetscMPIInt    n,imdex,rank = l->rank,size = l->size;
3427adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)A)->comm;
3438965ea79SLois Curfman McInnes   MPI_Request    *send_waits,*recv_waits;
3448965ea79SLois Curfman McInnes   MPI_Status     recv_status,*send_status;
34535d8aa7fSBarry Smith   PetscTruth     found;
3468965ea79SLois Curfman McInnes 
3473a40ed3dSBarry Smith   PetscFunctionBegin;
3488965ea79SLois Curfman McInnes   /*  first count number of contributors to each processor */
34913f74950SBarry Smith   ierr  = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
35013f74950SBarry Smith   ierr  = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
35113f74950SBarry Smith   ierr  = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/
3528965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3538965ea79SLois Curfman McInnes     idx = rows[i];
35435d8aa7fSBarry Smith     found = PETSC_FALSE;
3558965ea79SLois Curfman McInnes     for (j=0; j<size; j++) {
3568965ea79SLois Curfman McInnes       if (idx >= owners[j] && idx < owners[j+1]) {
357c1dc657dSBarry Smith         nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break;
3588965ea79SLois Curfman McInnes       }
3598965ea79SLois Curfman McInnes     }
36029bbc08cSBarry Smith     if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range");
3618965ea79SLois Curfman McInnes   }
362c1dc657dSBarry Smith   nsends = 0;  for (i=0; i<size; i++) { nsends += nprocs[2*i+1];}
3638965ea79SLois Curfman McInnes 
3648965ea79SLois Curfman McInnes   /* inform other processors of number of messages and max length*/
365c1dc657dSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
3668965ea79SLois Curfman McInnes 
3678965ea79SLois Curfman McInnes   /* post receives:   */
36813f74950SBarry Smith   ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr);
369b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
3708965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
37113f74950SBarry Smith     ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr);
3728965ea79SLois Curfman McInnes   }
3738965ea79SLois Curfman McInnes 
3748965ea79SLois Curfman McInnes   /* do sends:
3758965ea79SLois Curfman McInnes       1) starts[i] gives the starting index in svalues for stuff going to
3768965ea79SLois Curfman McInnes          the ith processor
3778965ea79SLois Curfman McInnes   */
37813f74950SBarry Smith   ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr);
379b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
38013f74950SBarry Smith   ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
3818965ea79SLois Curfman McInnes   starts[0]  = 0;
382c1dc657dSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3838965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3848965ea79SLois Curfman McInnes     svalues[starts[owner[i]]++] = rows[i];
3858965ea79SLois Curfman McInnes   }
3868965ea79SLois Curfman McInnes 
3878965ea79SLois Curfman McInnes   starts[0] = 0;
388c1dc657dSBarry Smith   for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3898965ea79SLois Curfman McInnes   count = 0;
3908965ea79SLois Curfman McInnes   for (i=0; i<size; i++) {
391c1dc657dSBarry Smith     if (nprocs[2*i+1]) {
39213f74950SBarry Smith       ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr);
3938965ea79SLois Curfman McInnes     }
3948965ea79SLois Curfman McInnes   }
395606d414cSSatish Balay   ierr = PetscFree(starts);CHKERRQ(ierr);
3968965ea79SLois Curfman McInnes 
3978965ea79SLois Curfman McInnes   base = owners[rank];
3988965ea79SLois Curfman McInnes 
3998965ea79SLois Curfman McInnes   /*  wait on receives */
40013f74950SBarry Smith   ierr   = PetscMalloc(2*(nrecvs+1)*sizeof(PetscInt),&lens);CHKERRQ(ierr);
4018965ea79SLois Curfman McInnes   source = lens + nrecvs;
4028965ea79SLois Curfman McInnes   count  = nrecvs; slen = 0;
4038965ea79SLois Curfman McInnes   while (count) {
404ca161407SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
4058965ea79SLois Curfman McInnes     /* unpack receives into our local space */
40613f74950SBarry Smith     ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr);
4078965ea79SLois Curfman McInnes     source[imdex]  = recv_status.MPI_SOURCE;
4088965ea79SLois Curfman McInnes     lens[imdex]    = n;
4098965ea79SLois Curfman McInnes     slen += n;
4108965ea79SLois Curfman McInnes     count--;
4118965ea79SLois Curfman McInnes   }
412606d414cSSatish Balay   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
4138965ea79SLois Curfman McInnes 
4148965ea79SLois Curfman McInnes   /* move the data into the send scatter */
41513f74950SBarry Smith   ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr);
4168965ea79SLois Curfman McInnes   count = 0;
4178965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
4188965ea79SLois Curfman McInnes     values = rvalues + i*nmax;
4198965ea79SLois Curfman McInnes     for (j=0; j<lens[i]; j++) {
4208965ea79SLois Curfman McInnes       lrows[count++] = values[j] - base;
4218965ea79SLois Curfman McInnes     }
4228965ea79SLois Curfman McInnes   }
423606d414cSSatish Balay   ierr = PetscFree(rvalues);CHKERRQ(ierr);
424606d414cSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
425606d414cSSatish Balay   ierr = PetscFree(owner);CHKERRQ(ierr);
426606d414cSSatish Balay   ierr = PetscFree(nprocs);CHKERRQ(ierr);
4278965ea79SLois Curfman McInnes 
4288965ea79SLois Curfman McInnes   /* actually zap the local rows */
429f4df32b1SMatthew Knepley   ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr);
430606d414cSSatish Balay   ierr = PetscFree(lrows);CHKERRQ(ierr);
4318965ea79SLois Curfman McInnes 
4328965ea79SLois Curfman McInnes   /* wait on sends */
4338965ea79SLois Curfman McInnes   if (nsends) {
434b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
435ca161407SBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
436606d414cSSatish Balay     ierr = PetscFree(send_status);CHKERRQ(ierr);
4378965ea79SLois Curfman McInnes   }
438606d414cSSatish Balay   ierr = PetscFree(send_waits);CHKERRQ(ierr);
439606d414cSSatish Balay   ierr = PetscFree(svalues);CHKERRQ(ierr);
4408965ea79SLois Curfman McInnes 
4413a40ed3dSBarry Smith   PetscFunctionReturn(0);
4428965ea79SLois Curfman McInnes }
4438965ea79SLois Curfman McInnes 
4444a2ae208SSatish Balay #undef __FUNCT__
4454a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense"
446dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy)
4478965ea79SLois Curfman McInnes {
44839ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
449dfbe8321SBarry Smith   PetscErrorCode ierr;
450c456f294SBarry Smith 
4513a40ed3dSBarry Smith   PetscFunctionBegin;
452ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
453ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
45444cd7ae7SLois Curfman McInnes   ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr);
4553a40ed3dSBarry Smith   PetscFunctionReturn(0);
4568965ea79SLois Curfman McInnes }
4578965ea79SLois Curfman McInnes 
4584a2ae208SSatish Balay #undef __FUNCT__
4594a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense"
460dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz)
4618965ea79SLois Curfman McInnes {
46239ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
463dfbe8321SBarry Smith   PetscErrorCode ierr;
464c456f294SBarry Smith 
4653a40ed3dSBarry Smith   PetscFunctionBegin;
466ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
467ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
46844cd7ae7SLois Curfman McInnes   ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr);
4693a40ed3dSBarry Smith   PetscFunctionReturn(0);
4708965ea79SLois Curfman McInnes }
4718965ea79SLois Curfman McInnes 
4724a2ae208SSatish Balay #undef __FUNCT__
4734a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense"
474dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy)
475096963f5SLois Curfman McInnes {
476096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
477dfbe8321SBarry Smith   PetscErrorCode ierr;
47887828ca2SBarry Smith   PetscScalar    zero = 0.0;
479096963f5SLois Curfman McInnes 
4803a40ed3dSBarry Smith   PetscFunctionBegin;
4812dcb1b2aSMatthew Knepley   ierr = VecSet(yy,zero);CHKERRQ(ierr);
4827c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
483ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
484ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
4853a40ed3dSBarry Smith   PetscFunctionReturn(0);
486096963f5SLois Curfman McInnes }
487096963f5SLois Curfman McInnes 
4884a2ae208SSatish Balay #undef __FUNCT__
4894a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense"
490dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz)
491096963f5SLois Curfman McInnes {
492096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
493dfbe8321SBarry Smith   PetscErrorCode ierr;
494096963f5SLois Curfman McInnes 
4953a40ed3dSBarry Smith   PetscFunctionBegin;
4963501a2bdSLois Curfman McInnes   ierr = VecCopy(yy,zz);CHKERRQ(ierr);
4977c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
498ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
499ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
5003a40ed3dSBarry Smith   PetscFunctionReturn(0);
501096963f5SLois Curfman McInnes }
502096963f5SLois Curfman McInnes 
5034a2ae208SSatish Balay #undef __FUNCT__
5044a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense"
505dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v)
5068965ea79SLois Curfman McInnes {
50739ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
508096963f5SLois Curfman McInnes   Mat_SeqDense   *aloc = (Mat_SeqDense*)a->A->data;
509dfbe8321SBarry Smith   PetscErrorCode ierr;
510d0f46423SBarry Smith   PetscInt       len,i,n,m = A->rmap->n,radd;
51187828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
512ed3cc1f0SBarry Smith 
5133a40ed3dSBarry Smith   PetscFunctionBegin;
5142dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
5151ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
516096963f5SLois Curfman McInnes   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
517d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
518d0f46423SBarry Smith   len  = PetscMin(a->A->rmap->n,a->A->cmap->n);
519d0f46423SBarry Smith   radd = A->rmap->rstart*m;
52044cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
521096963f5SLois Curfman McInnes     x[i] = aloc->v[radd + i*m + i];
522096963f5SLois Curfman McInnes   }
5231ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
5243a40ed3dSBarry Smith   PetscFunctionReturn(0);
5258965ea79SLois Curfman McInnes }
5268965ea79SLois Curfman McInnes 
5274a2ae208SSatish Balay #undef __FUNCT__
5284a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense"
529dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat)
5308965ea79SLois Curfman McInnes {
5313501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
532dfbe8321SBarry Smith   PetscErrorCode ierr;
53301b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
53401b82886SBarry Smith   Mat_Plapack   *lu=(Mat_Plapack*)(mat->spptr);
53501b82886SBarry Smith #endif
536ed3cc1f0SBarry Smith 
5373a40ed3dSBarry Smith   PetscFunctionBegin;
53894d884c6SBarry Smith 
539aa482453SBarry Smith #if defined(PETSC_USE_LOG)
540d0f46423SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N);
5418965ea79SLois Curfman McInnes #endif
5428798bf22SSatish Balay   ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr);
5433501a2bdSLois Curfman McInnes   ierr = MatDestroy(mdn->A);CHKERRQ(ierr);
54405b42c5fSBarry Smith   if (mdn->lvec)   {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);}
54505b42c5fSBarry Smith   if (mdn->Mvctx)  {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);}
54601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
5472fbe02b9SBarry Smith   if (lu) {
54862b4c0b3SBarry Smith     ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr);
54962b4c0b3SBarry Smith     ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr);
55062b4c0b3SBarry Smith     ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr);
55109d27a7eSBarry Smith 
5522fbe02b9SBarry Smith     if (lu->is_pla) {
55301b82886SBarry Smith       ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr);
55401b82886SBarry Smith       ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr);
55501b82886SBarry Smith       ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr);
556622d7880SLois Curfman McInnes     }
5572fbe02b9SBarry Smith   }
55801b82886SBarry Smith #endif
55901b82886SBarry Smith 
560606d414cSSatish Balay   ierr = PetscFree(mdn);CHKERRQ(ierr);
561dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
562901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr);
563901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
5644ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5654ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5664ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5673a40ed3dSBarry Smith   PetscFunctionReturn(0);
5688965ea79SLois Curfman McInnes }
56939ddd567SLois Curfman McInnes 
5704a2ae208SSatish Balay #undef __FUNCT__
5714a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary"
5726849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer)
5738965ea79SLois Curfman McInnes {
57439ddd567SLois Curfman McInnes   Mat_MPIDense      *mdn = (Mat_MPIDense*)mat->data;
575dfbe8321SBarry Smith   PetscErrorCode    ierr;
576aa05aa95SBarry Smith   PetscViewerFormat format;
577aa05aa95SBarry Smith   int               fd;
578d0f46423SBarry Smith   PetscInt          header[4],mmax,N = mat->cmap->N,i,j,m,k;
579aa05aa95SBarry Smith   PetscMPIInt       rank,tag  = ((PetscObject)viewer)->tag,size;
580578230a0SSatish Balay   PetscScalar       *work,*v,*vv;
581aa05aa95SBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)mdn->A->data;
582aa05aa95SBarry Smith   MPI_Status        status;
5837056b6fcSBarry Smith 
5843a40ed3dSBarry Smith   PetscFunctionBegin;
58539ddd567SLois Curfman McInnes   if (mdn->size == 1) {
58639ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
587aa05aa95SBarry Smith   } else {
588aa05aa95SBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
5897adad957SLisandro Dalcin     ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr);
5907adad957SLisandro Dalcin     ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr);
591aa05aa95SBarry Smith 
592aa05aa95SBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
5937973ad04SBarry Smith     if (format == PETSC_VIEWER_NATIVE) {
594aa05aa95SBarry Smith 
595aa05aa95SBarry Smith       if (!rank) {
596aa05aa95SBarry Smith         /* store the matrix as a dense matrix */
597aa05aa95SBarry Smith         header[0] = MAT_FILE_COOKIE;
598d0f46423SBarry Smith         header[1] = mat->rmap->N;
599aa05aa95SBarry Smith         header[2] = N;
600aa05aa95SBarry Smith         header[3] = MATRIX_BINARY_FORMAT_DENSE;
601aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
602aa05aa95SBarry Smith 
603aa05aa95SBarry Smith         /* get largest work array needed for transposing array */
604d0f46423SBarry Smith         mmax = mat->rmap->n;
605aa05aa95SBarry Smith         for (i=1; i<size; i++) {
606d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
6078965ea79SLois Curfman McInnes         }
608aa05aa95SBarry Smith         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr);
609aa05aa95SBarry Smith 
610aa05aa95SBarry Smith         /* write out local array, by rows */
611d0f46423SBarry Smith         m    = mat->rmap->n;
612aa05aa95SBarry Smith         v    = a->v;
613aa05aa95SBarry Smith         for (j=0; j<N; j++) {
614aa05aa95SBarry Smith           for (i=0; i<m; i++) {
615578230a0SSatish Balay             work[j + i*N] = *v++;
616aa05aa95SBarry Smith           }
617aa05aa95SBarry Smith         }
618aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
619aa05aa95SBarry Smith         /* get largest work array to receive messages from other processes, excludes process zero */
620aa05aa95SBarry Smith         mmax = 0;
621aa05aa95SBarry Smith         for (i=1; i<size; i++) {
622d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
623aa05aa95SBarry Smith         }
624578230a0SSatish Balay         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr);
625aa05aa95SBarry Smith         for(k = 1; k < size; k++) {
626f8009846SMatthew Knepley           v    = vv;
627d0f46423SBarry Smith           m    = mat->rmap->range[k+1] - mat->rmap->range[k];
6287adad957SLisandro Dalcin           ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr);
629aa05aa95SBarry Smith 
630aa05aa95SBarry Smith           for(j = 0; j < N; j++) {
631aa05aa95SBarry Smith             for(i = 0; i < m; i++) {
632578230a0SSatish Balay               work[j + i*N] = *v++;
633aa05aa95SBarry Smith             }
634aa05aa95SBarry Smith           }
635aa05aa95SBarry Smith           ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
636aa05aa95SBarry Smith         }
637aa05aa95SBarry Smith         ierr = PetscFree(work);CHKERRQ(ierr);
638578230a0SSatish Balay         ierr = PetscFree(vv);CHKERRQ(ierr);
639aa05aa95SBarry Smith       } else {
640d0f46423SBarry Smith         ierr = MPI_Send(a->v,mat->rmap->n*mat->cmap->N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr);
641aa05aa95SBarry Smith       }
6424aa34b0aSBarry Smith     } else {
6437973ad04SBarry Smith       SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_NATIVE");
644aa05aa95SBarry Smith     }
645aa05aa95SBarry Smith   }
6463a40ed3dSBarry Smith   PetscFunctionReturn(0);
6478965ea79SLois Curfman McInnes }
6488965ea79SLois Curfman McInnes 
6494a2ae208SSatish Balay #undef __FUNCT__
6504a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket"
6516849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer)
6528965ea79SLois Curfman McInnes {
65339ddd567SLois Curfman McInnes   Mat_MPIDense          *mdn = (Mat_MPIDense*)mat->data;
654dfbe8321SBarry Smith   PetscErrorCode        ierr;
65532dcc486SBarry Smith   PetscMPIInt           size = mdn->size,rank = mdn->rank;
656a313700dSBarry Smith   const PetscViewerType vtype;
65732077d6dSBarry Smith   PetscTruth            iascii,isdraw;
658b0a32e0cSBarry Smith   PetscViewer           sviewer;
659f3ef73ceSBarry Smith   PetscViewerFormat     format;
66001b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
661*5d00a290SHong Zhang   Mat_Plapack           *lu;
66201b82886SBarry Smith #endif
6638965ea79SLois Curfman McInnes 
6643a40ed3dSBarry Smith   PetscFunctionBegin;
66532077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
666fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
66732077d6dSBarry Smith   if (iascii) {
668b0a32e0cSBarry Smith     ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr);
669b0a32e0cSBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
670456192e2SBarry Smith     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
6714e220ebcSLois Curfman McInnes       MatInfo info;
672888f2ed8SSatish Balay       ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr);
673d0f46423SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap->n,
67477431f27SBarry Smith                    (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr);
675b0a32e0cSBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
67601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
67701b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr);
678*5d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Processor mesh: nprows %d, npcols %d\n",Plapack_nprows, Plapack_npcols);CHKERRQ(ierr);
679*5d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Error checking: %d\n",Plapack_ierror);CHKERRQ(ierr);
680*5d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Algorithmic block size: %d\n",Plapack_nb_alg);CHKERRQ(ierr);
681*5d00a290SHong Zhang       if (mat->factor){
682*5d00a290SHong Zhang         lu=(Mat_Plapack*)(mat->spptr);
68301b82886SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr);
684*5d00a290SHong Zhang       }
685*5d00a290SHong Zhang #else
686*5d00a290SHong Zhang       ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr);
68701b82886SBarry Smith #endif
6883a40ed3dSBarry Smith       PetscFunctionReturn(0);
689fb9695e5SSatish Balay     } else if (format == PETSC_VIEWER_ASCII_INFO) {
6903a40ed3dSBarry Smith       PetscFunctionReturn(0);
6918965ea79SLois Curfman McInnes     }
692f1af5d2fSBarry Smith   } else if (isdraw) {
693b0a32e0cSBarry Smith     PetscDraw  draw;
694f1af5d2fSBarry Smith     PetscTruth isnull;
695f1af5d2fSBarry Smith 
696b0a32e0cSBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
697b0a32e0cSBarry Smith     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
698f1af5d2fSBarry Smith     if (isnull) PetscFunctionReturn(0);
699f1af5d2fSBarry Smith   }
70077ed5343SBarry Smith 
7018965ea79SLois Curfman McInnes   if (size == 1) {
70239ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
7033a40ed3dSBarry Smith   } else {
7048965ea79SLois Curfman McInnes     /* assemble the entire matrix onto first processor. */
7058965ea79SLois Curfman McInnes     Mat         A;
706d0f46423SBarry Smith     PetscInt    M = mat->rmap->N,N = mat->cmap->N,m,row,i,nz;
707ba8c8a56SBarry Smith     PetscInt    *cols;
708ba8c8a56SBarry Smith     PetscScalar *vals;
7098965ea79SLois Curfman McInnes 
7107adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr);
7118965ea79SLois Curfman McInnes     if (!rank) {
712f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr);
7133a40ed3dSBarry Smith     } else {
714f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr);
7158965ea79SLois Curfman McInnes     }
7167adad957SLisandro Dalcin     /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */
717878740d9SKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
718878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL);
71952e6d16bSBarry Smith     ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr);
7208965ea79SLois Curfman McInnes 
72139ddd567SLois Curfman McInnes     /* Copy the matrix ... This isn't the most efficient means,
72239ddd567SLois Curfman McInnes        but it's quick for now */
72351022da4SBarry Smith     A->insertmode = INSERT_VALUES;
724d0f46423SBarry Smith     row = mat->rmap->rstart; m = mdn->A->rmap->n;
7258965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
726ba8c8a56SBarry Smith       ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
727ba8c8a56SBarry Smith       ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
728ba8c8a56SBarry Smith       ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
72939ddd567SLois Curfman McInnes       row++;
7308965ea79SLois Curfman McInnes     }
7318965ea79SLois Curfman McInnes 
7326d4a8577SBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7336d4a8577SBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
734b0a32e0cSBarry Smith     ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr);
735b9b97703SBarry Smith     if (!rank) {
7366831982aSBarry Smith       ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr);
7378965ea79SLois Curfman McInnes     }
738b0a32e0cSBarry Smith     ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr);
739b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
7408965ea79SLois Curfman McInnes     ierr = MatDestroy(A);CHKERRQ(ierr);
7418965ea79SLois Curfman McInnes   }
7423a40ed3dSBarry Smith   PetscFunctionReturn(0);
7438965ea79SLois Curfman McInnes }
7448965ea79SLois Curfman McInnes 
7454a2ae208SSatish Balay #undef __FUNCT__
7464a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense"
747dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer)
7488965ea79SLois Curfman McInnes {
749dfbe8321SBarry Smith   PetscErrorCode ierr;
75032077d6dSBarry Smith   PetscTruth     iascii,isbinary,isdraw,issocket;
7518965ea79SLois Curfman McInnes 
752433994e6SBarry Smith   PetscFunctionBegin;
7530f5bd95cSBarry Smith 
75432077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
755fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
756b0a32e0cSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr);
757fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
7580f5bd95cSBarry Smith 
75932077d6dSBarry Smith   if (iascii || issocket || isdraw) {
760f1af5d2fSBarry Smith     ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr);
7610f5bd95cSBarry Smith   } else if (isbinary) {
7623a40ed3dSBarry Smith     ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr);
7635cd90555SBarry Smith   } else {
764958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name);
7658965ea79SLois Curfman McInnes   }
7663a40ed3dSBarry Smith   PetscFunctionReturn(0);
7678965ea79SLois Curfman McInnes }
7688965ea79SLois Curfman McInnes 
7694a2ae208SSatish Balay #undef __FUNCT__
7704a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense"
771dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info)
7728965ea79SLois Curfman McInnes {
7733501a2bdSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
7743501a2bdSLois Curfman McInnes   Mat            mdn = mat->A;
775dfbe8321SBarry Smith   PetscErrorCode ierr;
776329f5518SBarry Smith   PetscReal      isend[5],irecv[5];
7778965ea79SLois Curfman McInnes 
7783a40ed3dSBarry Smith   PetscFunctionBegin;
7794e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
7804e220ebcSLois Curfman McInnes   ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr);
7814e220ebcSLois Curfman McInnes   isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded;
7824e220ebcSLois Curfman McInnes   isend[3] = info->memory;  isend[4] = info->mallocs;
7838965ea79SLois Curfman McInnes   if (flag == MAT_LOCAL) {
7844e220ebcSLois Curfman McInnes     info->nz_used      = isend[0];
7854e220ebcSLois Curfman McInnes     info->nz_allocated = isend[1];
7864e220ebcSLois Curfman McInnes     info->nz_unneeded  = isend[2];
7874e220ebcSLois Curfman McInnes     info->memory       = isend[3];
7884e220ebcSLois Curfman McInnes     info->mallocs      = isend[4];
7898965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_MAX) {
7907adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
7914e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7924e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
7934e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
7944e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
7954e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
7968965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_SUM) {
7977adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
7984e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7994e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
8004e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
8014e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
8024e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
8038965ea79SLois Curfman McInnes   }
8044e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0; /* no parallel LU/ILU/Cholesky */
8054e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
8064e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
8073a40ed3dSBarry Smith   PetscFunctionReturn(0);
8088965ea79SLois Curfman McInnes }
8098965ea79SLois Curfman McInnes 
8104a2ae208SSatish Balay #undef __FUNCT__
8114a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense"
8124e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg)
8138965ea79SLois Curfman McInnes {
81439ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
815dfbe8321SBarry Smith   PetscErrorCode ierr;
8168965ea79SLois Curfman McInnes 
8173a40ed3dSBarry Smith   PetscFunctionBegin;
81812c028f9SKris Buschelman   switch (op) {
819512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
82012c028f9SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
82112c028f9SKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
8224e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82312c028f9SKris Buschelman     break;
82412c028f9SKris Buschelman   case MAT_ROW_ORIENTED:
8254e0d8c25SBarry Smith     a->roworiented = flg;
8264e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82712c028f9SKris Buschelman     break;
8284e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
82912c028f9SKris Buschelman   case MAT_USE_HASH_TABLE:
830290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
83112c028f9SKris Buschelman     break;
83212c028f9SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
8334e0d8c25SBarry Smith     a->donotstash = flg;
83412c028f9SKris Buschelman     break;
83577e54ba9SKris Buschelman   case MAT_SYMMETRIC:
83677e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
8379a4540c5SBarry Smith   case MAT_HERMITIAN:
8389a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
839600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
840290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
84177e54ba9SKris Buschelman     break;
84212c028f9SKris Buschelman   default:
843600fe468SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
8443a40ed3dSBarry Smith   }
8453a40ed3dSBarry Smith   PetscFunctionReturn(0);
8468965ea79SLois Curfman McInnes }
8478965ea79SLois Curfman McInnes 
8488965ea79SLois Curfman McInnes 
8494a2ae208SSatish Balay #undef __FUNCT__
8504a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense"
851dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr)
8525b2fa520SLois Curfman McInnes {
8535b2fa520SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8545b2fa520SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
85587828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
856dfbe8321SBarry Smith   PetscErrorCode ierr;
857d0f46423SBarry Smith   PetscInt       i,j,s2a,s3a,s2,s3,m=mdn->A->rmap->n,n=mdn->A->cmap->n;
8585b2fa520SLois Curfman McInnes 
8595b2fa520SLois Curfman McInnes   PetscFunctionBegin;
86072d926a5SLois Curfman McInnes   ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr);
8615b2fa520SLois Curfman McInnes   if (ll) {
86272d926a5SLois Curfman McInnes     ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr);
863175be7b4SMatthew Knepley     if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2);
8641ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
8655b2fa520SLois Curfman McInnes     for (i=0; i<m; i++) {
8665b2fa520SLois Curfman McInnes       x = l[i];
8675b2fa520SLois Curfman McInnes       v = mat->v + i;
8685b2fa520SLois Curfman McInnes       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
8695b2fa520SLois Curfman McInnes     }
8701ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
871efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8725b2fa520SLois Curfman McInnes   }
8735b2fa520SLois Curfman McInnes   if (rr) {
874175be7b4SMatthew Knepley     ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr);
875175be7b4SMatthew Knepley     if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3);
876ca9f406cSSatish Balay     ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
877ca9f406cSSatish Balay     ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
8781ebc52fbSHong Zhang     ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr);
8795b2fa520SLois Curfman McInnes     for (i=0; i<n; i++) {
8805b2fa520SLois Curfman McInnes       x = r[i];
8815b2fa520SLois Curfman McInnes       v = mat->v + i*m;
8825b2fa520SLois Curfman McInnes       for (j=0; j<m; j++) { (*v++) *= x;}
8835b2fa520SLois Curfman McInnes     }
8841ebc52fbSHong Zhang     ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr);
885efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8865b2fa520SLois Curfman McInnes   }
8875b2fa520SLois Curfman McInnes   PetscFunctionReturn(0);
8885b2fa520SLois Curfman McInnes }
8895b2fa520SLois Curfman McInnes 
8904a2ae208SSatish Balay #undef __FUNCT__
8914a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense"
892dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm)
893096963f5SLois Curfman McInnes {
8943501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8953501a2bdSLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
896dfbe8321SBarry Smith   PetscErrorCode ierr;
89713f74950SBarry Smith   PetscInt       i,j;
898329f5518SBarry Smith   PetscReal      sum = 0.0;
89987828ca2SBarry Smith   PetscScalar    *v = mat->v;
9003501a2bdSLois Curfman McInnes 
9013a40ed3dSBarry Smith   PetscFunctionBegin;
9023501a2bdSLois Curfman McInnes   if (mdn->size == 1) {
903064f8208SBarry Smith     ierr =  MatNorm(mdn->A,type,nrm);CHKERRQ(ierr);
9043501a2bdSLois Curfman McInnes   } else {
9053501a2bdSLois Curfman McInnes     if (type == NORM_FROBENIUS) {
906d0f46423SBarry Smith       for (i=0; i<mdn->A->cmap->n*mdn->A->rmap->n; i++) {
907aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
908329f5518SBarry Smith         sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
9093501a2bdSLois Curfman McInnes #else
9103501a2bdSLois Curfman McInnes         sum += (*v)*(*v); v++;
9113501a2bdSLois Curfman McInnes #endif
9123501a2bdSLois Curfman McInnes       }
9137adad957SLisandro Dalcin       ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
914064f8208SBarry Smith       *nrm = sqrt(*nrm);
915d0f46423SBarry Smith       ierr = PetscLogFlops(2*mdn->A->cmap->n*mdn->A->rmap->n);CHKERRQ(ierr);
9163a40ed3dSBarry Smith     } else if (type == NORM_1) {
917329f5518SBarry Smith       PetscReal *tmp,*tmp2;
918d0f46423SBarry Smith       ierr = PetscMalloc(2*A->cmap->N*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
919d0f46423SBarry Smith       tmp2 = tmp + A->cmap->N;
920d0f46423SBarry Smith       ierr = PetscMemzero(tmp,2*A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr);
921064f8208SBarry Smith       *nrm = 0.0;
9223501a2bdSLois Curfman McInnes       v = mat->v;
923d0f46423SBarry Smith       for (j=0; j<mdn->A->cmap->n; j++) {
924d0f46423SBarry Smith         for (i=0; i<mdn->A->rmap->n; i++) {
92567e560aaSBarry Smith           tmp[j] += PetscAbsScalar(*v);  v++;
9263501a2bdSLois Curfman McInnes         }
9273501a2bdSLois Curfman McInnes       }
928d0f46423SBarry Smith       ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
929d0f46423SBarry Smith       for (j=0; j<A->cmap->N; j++) {
930064f8208SBarry Smith         if (tmp2[j] > *nrm) *nrm = tmp2[j];
9313501a2bdSLois Curfman McInnes       }
932606d414cSSatish Balay       ierr = PetscFree(tmp);CHKERRQ(ierr);
933d0f46423SBarry Smith       ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
9343a40ed3dSBarry Smith     } else if (type == NORM_INFINITY) { /* max row norm */
935329f5518SBarry Smith       PetscReal ntemp;
9363501a2bdSLois Curfman McInnes       ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr);
9377adad957SLisandro Dalcin       ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
9383a40ed3dSBarry Smith     } else {
93929bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"No support for two norm");
9403501a2bdSLois Curfman McInnes     }
9413501a2bdSLois Curfman McInnes   }
9423a40ed3dSBarry Smith   PetscFunctionReturn(0);
9433501a2bdSLois Curfman McInnes }
9443501a2bdSLois Curfman McInnes 
9454a2ae208SSatish Balay #undef __FUNCT__
9464a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense"
947fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout)
9483501a2bdSLois Curfman McInnes {
9493501a2bdSLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
9503501a2bdSLois Curfman McInnes   Mat_SeqDense   *Aloc = (Mat_SeqDense*)a->A->data;
9513501a2bdSLois Curfman McInnes   Mat            B;
952d0f46423SBarry Smith   PetscInt       M = A->rmap->N,N = A->cmap->N,m,n,*rwork,rstart = A->rmap->rstart;
9536849ba73SBarry Smith   PetscErrorCode ierr;
95413f74950SBarry Smith   PetscInt       j,i;
95587828ca2SBarry Smith   PetscScalar    *v;
9563501a2bdSLois Curfman McInnes 
9573a40ed3dSBarry Smith   PetscFunctionBegin;
958e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place");
959fc4dec0aSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || A == *matout) {
9607adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
961d0f46423SBarry Smith     ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,N,M);CHKERRQ(ierr);
9627adad957SLisandro Dalcin     ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
963878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr);
964fc4dec0aSBarry Smith   } else {
965fc4dec0aSBarry Smith     B = *matout;
966fc4dec0aSBarry Smith   }
9673501a2bdSLois Curfman McInnes 
968d0f46423SBarry Smith   m = a->A->rmap->n; n = a->A->cmap->n; v = Aloc->v;
9691acff37aSSatish Balay   ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr);
9703501a2bdSLois Curfman McInnes   for (i=0; i<m; i++) rwork[i] = rstart + i;
9711acff37aSSatish Balay   for (j=0; j<n; j++) {
9723501a2bdSLois Curfman McInnes     ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr);
9733501a2bdSLois Curfman McInnes     v   += m;
9743501a2bdSLois Curfman McInnes   }
975606d414cSSatish Balay   ierr = PetscFree(rwork);CHKERRQ(ierr);
9766d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9776d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
978815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *matout != A) {
9793501a2bdSLois Curfman McInnes     *matout = B;
9803501a2bdSLois Curfman McInnes   } else {
981273d9f13SBarry Smith     ierr = MatHeaderCopy(A,B);CHKERRQ(ierr);
9823501a2bdSLois Curfman McInnes   }
9833a40ed3dSBarry Smith   PetscFunctionReturn(0);
984096963f5SLois Curfman McInnes }
985096963f5SLois Curfman McInnes 
986d9eff348SSatish Balay #include "petscblaslapack.h"
9874a2ae208SSatish Balay #undef __FUNCT__
9884a2ae208SSatish Balay #define __FUNCT__ "MatScale_MPIDense"
989f4df32b1SMatthew Knepley PetscErrorCode MatScale_MPIDense(Mat inA,PetscScalar alpha)
99044cd7ae7SLois Curfman McInnes {
99144cd7ae7SLois Curfman McInnes   Mat_MPIDense   *A = (Mat_MPIDense*)inA->data;
99244cd7ae7SLois Curfman McInnes   Mat_SeqDense   *a = (Mat_SeqDense*)A->A->data;
993f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
994efee365bSSatish Balay   PetscErrorCode ierr;
995d0f46423SBarry Smith   PetscBLASInt   one = 1,nz = PetscBLASIntCast(inA->rmap->n*inA->cmap->N);
99644cd7ae7SLois Curfman McInnes 
9973a40ed3dSBarry Smith   PetscFunctionBegin;
998f4df32b1SMatthew Knepley   BLASscal_(&nz,&oalpha,a->v,&one);
999efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
10003a40ed3dSBarry Smith   PetscFunctionReturn(0);
100144cd7ae7SLois Curfman McInnes }
100244cd7ae7SLois Curfman McInnes 
10036849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *);
10048965ea79SLois Curfman McInnes 
10054a2ae208SSatish Balay #undef __FUNCT__
10064a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_MPIDense"
1007dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_MPIDense(Mat A)
1008273d9f13SBarry Smith {
1009dfbe8321SBarry Smith   PetscErrorCode ierr;
1010273d9f13SBarry Smith 
1011273d9f13SBarry Smith   PetscFunctionBegin;
1012273d9f13SBarry Smith   ierr =  MatMPIDenseSetPreallocation(A,0);CHKERRQ(ierr);
1013273d9f13SBarry Smith   PetscFunctionReturn(0);
1014273d9f13SBarry Smith }
1015273d9f13SBarry Smith 
101601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1017eae6fb2eSBarry Smith 
1018eae6fb2eSBarry Smith #undef __FUNCT__
1019eae6fb2eSBarry Smith #define __FUNCT__ "MatMPIDenseCopyToPlapack"
1020eae6fb2eSBarry Smith PetscErrorCode MatMPIDenseCopyToPlapack(Mat A,Mat F)
1021eae6fb2eSBarry Smith {
1022eae6fb2eSBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
1023eae6fb2eSBarry Smith   PetscErrorCode ierr;
1024d0f46423SBarry Smith   PetscInt       M=A->cmap->N,m=A->rmap->n,rstart;
1025eae6fb2eSBarry Smith   PetscScalar    *array;
1026eae6fb2eSBarry Smith   PetscReal      one = 1.0;
1027eae6fb2eSBarry Smith 
1028eae6fb2eSBarry Smith   PetscFunctionBegin;
10292fbe02b9SBarry Smith   /* Copy A into F->lu->A */
1030eae6fb2eSBarry Smith   ierr = PLA_Obj_set_to_zero(lu->A);CHKERRQ(ierr);
1031eae6fb2eSBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
1032eae6fb2eSBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
103379b0a62dSBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
1034eae6fb2eSBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
1035eae6fb2eSBarry Smith   ierr = PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);CHKERRQ(ierr);
1036eae6fb2eSBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
1037eae6fb2eSBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
1038eae6fb2eSBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
1039eae6fb2eSBarry Smith   lu->rstart = rstart;
1040eae6fb2eSBarry Smith   PetscFunctionReturn(0);
1041eae6fb2eSBarry Smith }
1042eae6fb2eSBarry Smith 
104301b82886SBarry Smith #undef __FUNCT__
10442fbe02b9SBarry Smith #define __FUNCT__ "MatMPIDenseCopyFromPlapack"
10452fbe02b9SBarry Smith PetscErrorCode MatMPIDenseCopyFromPlapack(Mat F,Mat A)
10462fbe02b9SBarry Smith {
10472fbe02b9SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
10482fbe02b9SBarry Smith   PetscErrorCode ierr;
1049d0f46423SBarry Smith   PetscInt       M=A->cmap->N,m=A->rmap->n,rstart;
10502fbe02b9SBarry Smith   PetscScalar    *array;
10512fbe02b9SBarry Smith   PetscReal      one = 1.0;
10522fbe02b9SBarry Smith 
10532fbe02b9SBarry Smith   PetscFunctionBegin;
10542fbe02b9SBarry Smith   /* Copy F into A->lu->A */
105579b0a62dSBarry Smith   ierr = MatZeroEntries(A);CHKERRQ(ierr);
10562fbe02b9SBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
10572fbe02b9SBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
10582fbe02b9SBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
10592fbe02b9SBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
10602fbe02b9SBarry Smith   ierr = PLA_API_axpy_global_to_matrix(m,M, &one,lu->A,rstart,0,(void *)array,m);CHKERRQ(ierr);
10612fbe02b9SBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
10622fbe02b9SBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
10632fbe02b9SBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
10642fbe02b9SBarry Smith   lu->rstart = rstart;
10652fbe02b9SBarry Smith   PetscFunctionReturn(0);
10662fbe02b9SBarry Smith }
10672fbe02b9SBarry Smith 
10682fbe02b9SBarry Smith #undef __FUNCT__
10692fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultNumeric_MPIDense_MPIDense"
10702fbe02b9SBarry Smith PetscErrorCode MatMatMultNumeric_MPIDense_MPIDense(Mat A,Mat B,Mat C)
10712fbe02b9SBarry Smith {
10722fbe02b9SBarry Smith   PetscErrorCode ierr;
10732fbe02b9SBarry Smith   Mat_Plapack    *luA = (Mat_Plapack*)A->spptr;
10742fbe02b9SBarry Smith   Mat_Plapack    *luB = (Mat_Plapack*)B->spptr;
10752fbe02b9SBarry Smith   Mat_Plapack    *luC = (Mat_Plapack*)C->spptr;
10762fbe02b9SBarry Smith   PLA_Obj        alpha = NULL,beta = NULL;
10772fbe02b9SBarry Smith 
10782fbe02b9SBarry Smith   PetscFunctionBegin;
10792fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(A,A);CHKERRQ(ierr);
10802fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(B,B);CHKERRQ(ierr);
10812fbe02b9SBarry Smith 
10826e6f9017SBarry Smith   /*
1083cb2480eaSBarry Smith   ierr = PLA_Global_show("A = ",luA->A,"%g ","");CHKERRQ(ierr);
1084cb2480eaSBarry Smith   ierr = PLA_Global_show("B = ",luB->A,"%g ","");CHKERRQ(ierr);
10856e6f9017SBarry Smith   */
1086cb2480eaSBarry Smith 
10872fbe02b9SBarry Smith   /* do the multiply in PLA  */
108879b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luA->A,NULL,NULL,&alpha);CHKERRQ(ierr);
108979b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luC->A,NULL,&beta,NULL);CHKERRQ(ierr);
109079b0a62dSBarry Smith   CHKMEMQ;
10912fbe02b9SBarry Smith 
1092f0e3846dSSatish Balay   ierr = PLA_Gemm(PLA_NO_TRANSPOSE,PLA_NO_TRANSPOSE,alpha,luA->A,luB->A,beta,luC->A); /* CHKERRQ(ierr); */
109379b0a62dSBarry Smith   CHKMEMQ;
10942fbe02b9SBarry Smith   ierr = PLA_Obj_free(&alpha);CHKERRQ(ierr);
10952fbe02b9SBarry Smith   ierr = PLA_Obj_free(&beta);CHKERRQ(ierr);
10962fbe02b9SBarry Smith 
10976e6f9017SBarry Smith   /*
1098cb2480eaSBarry Smith   ierr = PLA_Global_show("C = ",luC->A,"%g ","");CHKERRQ(ierr);
10996e6f9017SBarry Smith   */
11002fbe02b9SBarry Smith   ierr = MatMPIDenseCopyFromPlapack(C,C);CHKERRQ(ierr);
11012fbe02b9SBarry Smith   PetscFunctionReturn(0);
11022fbe02b9SBarry Smith }
11032fbe02b9SBarry Smith 
11042fbe02b9SBarry Smith #undef __FUNCT__
11052fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultSymbolic_MPIDense_MPIDense"
11062fbe02b9SBarry Smith PetscErrorCode MatMatMultSymbolic_MPIDense_MPIDense(Mat A,Mat B,PetscReal fill,Mat *C)
11072fbe02b9SBarry Smith {
11082fbe02b9SBarry Smith   PetscErrorCode ierr;
1109d0f46423SBarry Smith   PetscInt       m=A->rmap->n,n=B->cmap->n;
11102fbe02b9SBarry Smith   Mat            Cmat;
11112fbe02b9SBarry Smith 
11122fbe02b9SBarry Smith   PetscFunctionBegin;
1113d0f46423SBarry 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);
11146e6f9017SBarry Smith   SETERRQ(PETSC_ERR_LIB,"Due to aparent bugs in PLAPACK,this is not currently supported");
11152fbe02b9SBarry Smith   ierr = MatCreate(((PetscObject)B)->comm,&Cmat);CHKERRQ(ierr);
1116d0f46423SBarry Smith   ierr = MatSetSizes(Cmat,m,n,A->rmap->N,B->cmap->N);CHKERRQ(ierr);
11172fbe02b9SBarry Smith   ierr = MatSetType(Cmat,MATMPIDENSE);CHKERRQ(ierr);
11182fbe02b9SBarry Smith   ierr = MatAssemblyBegin(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11192fbe02b9SBarry Smith   ierr = MatAssemblyEnd(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
112085bd4cefSHong Zhang 
11212fbe02b9SBarry Smith   *C = Cmat;
11222fbe02b9SBarry Smith   PetscFunctionReturn(0);
11232fbe02b9SBarry Smith }
11242fbe02b9SBarry Smith 
11252fbe02b9SBarry Smith #undef __FUNCT__
11262fbe02b9SBarry Smith #define __FUNCT__ "MatMatMult_MPIDense_MPIDense"
11272fbe02b9SBarry Smith PetscErrorCode MatMatMult_MPIDense_MPIDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
11282fbe02b9SBarry Smith {
11292fbe02b9SBarry Smith   PetscErrorCode ierr;
11302fbe02b9SBarry Smith 
11312fbe02b9SBarry Smith   PetscFunctionBegin;
11322fbe02b9SBarry Smith   if (scall == MAT_INITIAL_MATRIX){
11332fbe02b9SBarry Smith     ierr = MatMatMultSymbolic_MPIDense_MPIDense(A,B,fill,C);CHKERRQ(ierr);
11342fbe02b9SBarry Smith   }
11352fbe02b9SBarry Smith   ierr = MatMatMultNumeric_MPIDense_MPIDense(A,B,*C);CHKERRQ(ierr);
11362fbe02b9SBarry Smith   PetscFunctionReturn(0);
11372fbe02b9SBarry Smith }
11382fbe02b9SBarry Smith 
1139ab235cb6SHong Zhang #undef __FUNCT__
11406c995c7dSHong Zhang #define __FUNCT__ "MatSolve_MPIDense"
11416c995c7dSHong Zhang PetscErrorCode MatSolve_MPIDense(Mat A,Vec b,Vec x)
11426c995c7dSHong Zhang {
11436c995c7dSHong Zhang   MPI_Comm       comm = ((PetscObject)A)->comm;
11446c995c7dSHong Zhang   Mat_Plapack    *lu = (Mat_Plapack*)A->spptr;
11456c995c7dSHong Zhang   PetscErrorCode ierr;
1146d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,i,j,*idx_pla,*idx_petsc,loc_m,loc_stride;
11476c995c7dSHong Zhang   PetscScalar    *array;
11486c995c7dSHong Zhang   PetscReal      one = 1.0;
11496c995c7dSHong Zhang   PetscMPIInt    size,rank,r_rank,r_nproc,c_rank,c_nproc;;
11506c995c7dSHong Zhang   PLA_Obj        v_pla = NULL;
11516c995c7dSHong Zhang   PetscScalar    *loc_buf;
11526c995c7dSHong Zhang   Vec            loc_x;
11536c995c7dSHong Zhang 
11546c995c7dSHong Zhang   PetscFunctionBegin;
11556c995c7dSHong Zhang   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
11566c995c7dSHong Zhang   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
11576c995c7dSHong Zhang 
11586c995c7dSHong Zhang   /* Create PLAPACK vector objects, then copy b into PLAPACK b */
11596c995c7dSHong Zhang   PLA_Mvector_create(lu->datatype,M,1,lu->templ,PLA_ALIGN_FIRST,&v_pla);
11606c995c7dSHong Zhang   PLA_Obj_set_to_zero(v_pla);
11616c995c7dSHong Zhang 
11626c995c7dSHong Zhang   /* Copy b into rhs_pla */
11636c995c7dSHong Zhang   PLA_API_begin();
11646c995c7dSHong Zhang   PLA_Obj_API_open(v_pla);
11656c995c7dSHong Zhang   ierr = VecGetArray(b,&array);CHKERRQ(ierr);
11666c995c7dSHong Zhang   PLA_API_axpy_vector_to_global(m,&one,(void *)array,1,v_pla,lu->rstart);
11676c995c7dSHong Zhang   ierr = VecRestoreArray(b,&array);CHKERRQ(ierr);
11686c995c7dSHong Zhang   PLA_Obj_API_close(v_pla);
11696c995c7dSHong Zhang   PLA_API_end();
11706c995c7dSHong Zhang 
11716c995c7dSHong Zhang   if (A->factor == MAT_FACTOR_LU){
11726c995c7dSHong Zhang     /* Apply the permutations to the right hand sides */
11736c995c7dSHong Zhang     PLA_Apply_pivots_to_rows (v_pla,lu->pivots);
11746c995c7dSHong Zhang 
11756c995c7dSHong Zhang     /* Solve L y = b, overwriting b with y */
11766c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_UNIT_DIAG,lu->A,v_pla );
11776c995c7dSHong Zhang 
11786c995c7dSHong Zhang     /* Solve U x = y (=b), overwriting b with x */
11796c995c7dSHong Zhang     PLA_Trsv( PLA_UPPER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla );
11806c995c7dSHong Zhang   } else { /*MAT_FACTOR_CHOLESKY */
11816c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla);
11826c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,(lu->datatype == MPI_DOUBLE ? PLA_TRANSPOSE : PLA_CONJUGATE_TRANSPOSE),
11836c995c7dSHong Zhang                                     PLA_NONUNIT_DIAG,lu->A,v_pla);
11846c995c7dSHong Zhang   }
11856c995c7dSHong Zhang 
11866c995c7dSHong Zhang   /* Copy PLAPACK x into Petsc vector x  */
11876c995c7dSHong Zhang   PLA_Obj_local_length(v_pla, &loc_m);
11886c995c7dSHong Zhang   PLA_Obj_local_buffer(v_pla, (void**)&loc_buf);
11896c995c7dSHong Zhang   PLA_Obj_local_stride(v_pla, &loc_stride);
11906c995c7dSHong Zhang   /*
11916c995c7dSHong 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);
11926c995c7dSHong Zhang   */
11936c995c7dSHong Zhang   ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,loc_m*loc_stride,loc_buf,&loc_x);CHKERRQ(ierr);
11946c995c7dSHong Zhang   if (!lu->pla_solved){
11956c995c7dSHong Zhang 
1196*5d00a290SHong Zhang     PLA_Temp_comm_row_info(lu->templ,&Plapack_comm_2d,&r_rank,&r_nproc);
1197*5d00a290SHong Zhang     PLA_Temp_comm_col_info(lu->templ,&Plapack_comm_2d,&c_rank,&c_nproc);
11986c995c7dSHong Zhang 
11996c995c7dSHong Zhang     /* Create IS and cts for VecScatterring */
12006c995c7dSHong Zhang     PLA_Obj_local_length(v_pla, &loc_m);
12016c995c7dSHong Zhang     PLA_Obj_local_stride(v_pla, &loc_stride);
12026c995c7dSHong Zhang     ierr = PetscMalloc((2*loc_m+1)*sizeof(PetscInt),&idx_pla);CHKERRQ(ierr);
12036c995c7dSHong Zhang     idx_petsc = idx_pla + loc_m;
12046c995c7dSHong Zhang 
12056c995c7dSHong Zhang     rstart = (r_rank*c_nproc+c_rank)*lu->nb;
12066c995c7dSHong Zhang     for (i=0; i<loc_m; i+=lu->nb){
12076c995c7dSHong Zhang       j = 0;
12086c995c7dSHong Zhang       while (j < lu->nb && i+j < loc_m){
12096c995c7dSHong Zhang         idx_petsc[i+j] = rstart + j; j++;
12106c995c7dSHong Zhang       }
12116c995c7dSHong Zhang       rstart += size*lu->nb;
12126c995c7dSHong Zhang     }
12136c995c7dSHong Zhang 
12146c995c7dSHong Zhang     for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride;
12156c995c7dSHong Zhang 
12166c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr);
12176c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr);
12186c995c7dSHong Zhang     ierr = PetscFree(idx_pla);CHKERRQ(ierr);
12196c995c7dSHong Zhang     ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr);
12206c995c7dSHong Zhang   }
12216c995c7dSHong Zhang   ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12226c995c7dSHong Zhang   ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12236c995c7dSHong Zhang 
12246c995c7dSHong Zhang   /* Free data */
12256c995c7dSHong Zhang   ierr = VecDestroy(loc_x);CHKERRQ(ierr);
12266c995c7dSHong Zhang   PLA_Obj_free(&v_pla);
12276c995c7dSHong Zhang 
12286c995c7dSHong Zhang   lu->pla_solved = PETSC_TRUE;
12296c995c7dSHong Zhang   PetscFunctionReturn(0);
12306c995c7dSHong Zhang }
12316c995c7dSHong Zhang 
12326c995c7dSHong Zhang #undef __FUNCT__
12336c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense"
12340481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info)
12356c995c7dSHong Zhang {
1236719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
12376c995c7dSHong Zhang   PetscErrorCode ierr;
1238d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12396c995c7dSHong Zhang   PetscInt       info_pla=0;
12406c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12416c995c7dSHong Zhang 
12426c995c7dSHong Zhang   PetscFunctionBegin;
12436c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12446c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12456c995c7dSHong Zhang     PLA_Obj_free (&lu->pivots);
12466c995c7dSHong Zhang   }
12476c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12486c995c7dSHong Zhang   lu->A = NULL; lu->pivots = NULL;
12496c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12506c995c7dSHong Zhang   PLA_Obj_set_to_zero(lu->A);
12516c995c7dSHong Zhang   PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);
12526c995c7dSHong Zhang 
12536c995c7dSHong Zhang   /* Copy A into lu->A */
12546c995c7dSHong Zhang   PLA_API_begin();
12556c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
12566c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
12576c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
12586c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
12596c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
12606c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
12616c995c7dSHong Zhang   PLA_API_end();
12626c995c7dSHong Zhang 
12636c995c7dSHong Zhang   /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */
12646c995c7dSHong Zhang   info_pla = PLA_LU(lu->A,lu->pivots);
12656c995c7dSHong Zhang   if (info_pla != 0)
12666c995c7dSHong Zhang     SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla);
12676c995c7dSHong Zhang 
12686c995c7dSHong Zhang   lu->rstart         = rstart;
12696c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1270719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1271719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
12726c995c7dSHong Zhang   PetscFunctionReturn(0);
12736c995c7dSHong Zhang }
12746c995c7dSHong Zhang 
12756c995c7dSHong Zhang #undef __FUNCT__
12766c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense"
12770481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info)
12786c995c7dSHong Zhang {
1279719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)F->spptr;
12806c995c7dSHong Zhang   PetscErrorCode ierr;
1281d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12826c995c7dSHong Zhang   PetscInt       info_pla=0;
12836c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12846c995c7dSHong Zhang 
12856c995c7dSHong Zhang   PetscFunctionBegin;
12866c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12876c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12886c995c7dSHong Zhang   }
12896c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12906c995c7dSHong Zhang   lu->A      = NULL;
12916c995c7dSHong Zhang   lu->pivots = NULL;
12926c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12936c995c7dSHong Zhang 
12946c995c7dSHong Zhang   /* Copy A into lu->A */
12956c995c7dSHong Zhang   PLA_API_begin();
12966c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
12976c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
12986c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
12996c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
13006c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
13016c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
13026c995c7dSHong Zhang   PLA_API_end();
13036c995c7dSHong Zhang 
13046c995c7dSHong Zhang   /* Factor P A -> Chol */
13056c995c7dSHong Zhang   info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A);
13066c995c7dSHong Zhang   if (info_pla != 0)
13076c995c7dSHong Zhang     SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla);
13086c995c7dSHong Zhang 
13096c995c7dSHong Zhang   lu->rstart         = rstart;
13106c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1311719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1312719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
13136c995c7dSHong Zhang   PetscFunctionReturn(0);
13146c995c7dSHong Zhang }
13156c995c7dSHong Zhang 
1316b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */
13172fbe02b9SBarry Smith #undef __FUNCT__
13186c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense"
13190481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,const MatFactorInfo *info)
132001b82886SBarry Smith {
132101b82886SBarry Smith   PetscErrorCode ierr;
1322b24902e0SBarry Smith   PetscTruth     issymmetric,set;
132301b82886SBarry Smith 
132401b82886SBarry Smith   PetscFunctionBegin;
1325b24902e0SBarry Smith   ierr = MatIsSymmetricKnown(A,&set,&issymmetric); CHKERRQ(ierr);
1326b24902e0SBarry Smith   if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()");
1327719d5645SBarry Smith   F->ops->choleskyfactornumeric  = MatCholeskyFactorNumeric_MPIDense;
1328b24902e0SBarry Smith   PetscFunctionReturn(0);
132901b82886SBarry Smith }
133001b82886SBarry Smith 
1331b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */
1332b24902e0SBarry Smith #undef __FUNCT__
13336c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense"
13340481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
1335b24902e0SBarry Smith {
1336b24902e0SBarry Smith   PetscErrorCode ierr;
1337d0f46423SBarry Smith   PetscInt       M = A->rmap->N;
1338b24902e0SBarry Smith   Mat_Plapack    *lu;
133901b82886SBarry Smith 
1340b24902e0SBarry Smith   PetscFunctionBegin;
1341719d5645SBarry Smith   lu = (Mat_Plapack*)F->spptr;
1342b24902e0SBarry Smith   ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr);
1343719d5645SBarry Smith   F->ops->lufactornumeric  = MatLUFactorNumeric_MPIDense;
134401b82886SBarry Smith   PetscFunctionReturn(0);
134501b82886SBarry Smith }
134601b82886SBarry Smith 
134701b82886SBarry Smith #undef __FUNCT__
134885bd4cefSHong Zhang #define __FUNCT__ "MatGetFactor_mpidense_petsc"
134985bd4cefSHong Zhang PetscErrorCode MatGetFactor_mpidense_petsc(Mat A,MatFactorType ftype,Mat *F)
135001b82886SBarry Smith {
135101b82886SBarry Smith   PetscErrorCode ierr;
13526c995c7dSHong Zhang   Mat_Plapack    *lu;
13536c995c7dSHong Zhang   PetscMPIInt    size;
135485bd4cefSHong Zhang   PetscInt       M=A->rmap->N;
135501b82886SBarry Smith 
135601b82886SBarry Smith   PetscFunctionBegin;
135701b82886SBarry Smith   /* Create the factorization matrix */
1358eae6fb2eSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr);
1359d0f46423SBarry Smith   ierr = MatSetSizes(*F,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
1360eae6fb2eSBarry Smith   ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr);
136185bd4cefSHong Zhang   ierr = PetscNewLog(*F,Mat_Plapack,&lu);CHKERRQ(ierr);
136285bd4cefSHong Zhang   (*F)->spptr = (void*)lu;
136301b82886SBarry Smith 
1364b24902e0SBarry Smith   /* Set default Plapack parameters */
13656c995c7dSHong Zhang   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1366b24902e0SBarry Smith   lu->nb = M/size;
1367b24902e0SBarry Smith   if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */
136801b82886SBarry Smith 
1369b24902e0SBarry Smith   /* Set runtime options */
1370b24902e0SBarry Smith   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr);
1371b24902e0SBarry Smith     ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr);
1372b24902e0SBarry Smith   PetscOptionsEnd();
137301b82886SBarry Smith 
1374b24902e0SBarry Smith   /* Create object distribution template */
1375b24902e0SBarry Smith   lu->templ = NULL;
1376b24902e0SBarry Smith   ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr);
1377b24902e0SBarry Smith 
1378b24902e0SBarry Smith   /* Set the datatype */
1379b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX)
1380b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE_COMPLEX;
1381b24902e0SBarry Smith #else
1382b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE;
1383b24902e0SBarry Smith #endif
1384b24902e0SBarry Smith 
1385d0f46423SBarry Smith   ierr = PLA_Matrix_create(lu->datatype,M,A->cmap->N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr);
1386b24902e0SBarry Smith 
1387b24902e0SBarry Smith 
1388b24902e0SBarry Smith   lu->pla_solved     = PETSC_FALSE; /* MatSolve_Plapack() is called yet */
1389*5d00a290SHong Zhang   lu->mstruct        = DIFFERENT_NONZERO_PATTERN;
1390b24902e0SBarry Smith 
1391b24902e0SBarry Smith   if (ftype == MAT_FACTOR_LU) {
1392b24902e0SBarry Smith     (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense;
1393b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY) {
13946c995c7dSHong Zhang     (*F)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense;
1395b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices");
1396719d5645SBarry Smith   (*F)->factor = ftype;
139701b82886SBarry Smith   PetscFunctionReturn(0);
139801b82886SBarry Smith }
139901b82886SBarry Smith #endif
140001b82886SBarry Smith 
14018965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/
140209dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense,
140309dc0095SBarry Smith        MatGetRow_MPIDense,
140409dc0095SBarry Smith        MatRestoreRow_MPIDense,
140509dc0095SBarry Smith        MatMult_MPIDense,
140697304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense,
14077c922b88SBarry Smith        MatMultTranspose_MPIDense,
14087c922b88SBarry Smith        MatMultTransposeAdd_MPIDense,
14098965ea79SLois Curfman McInnes        0,
141009dc0095SBarry Smith        0,
141109dc0095SBarry Smith        0,
141297304618SKris Buschelman /*10*/ 0,
141309dc0095SBarry Smith        0,
141409dc0095SBarry Smith        0,
141509dc0095SBarry Smith        0,
141609dc0095SBarry Smith        MatTranspose_MPIDense,
141797304618SKris Buschelman /*15*/ MatGetInfo_MPIDense,
14186e4ee0c6SHong Zhang        MatEqual_MPIDense,
141909dc0095SBarry Smith        MatGetDiagonal_MPIDense,
14205b2fa520SLois Curfman McInnes        MatDiagonalScale_MPIDense,
142109dc0095SBarry Smith        MatNorm_MPIDense,
142297304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense,
142309dc0095SBarry Smith        MatAssemblyEnd_MPIDense,
142409dc0095SBarry Smith        0,
142509dc0095SBarry Smith        MatSetOption_MPIDense,
142609dc0095SBarry Smith        MatZeroEntries_MPIDense,
142797304618SKris Buschelman /*25*/ MatZeroRows_MPIDense,
1428919b68f7SBarry Smith        0,
142901b82886SBarry Smith        0,
143001b82886SBarry Smith        0,
143101b82886SBarry Smith        0,
143297304618SKris Buschelman /*30*/ MatSetUpPreallocation_MPIDense,
1433273d9f13SBarry Smith        0,
143409dc0095SBarry Smith        0,
143509dc0095SBarry Smith        MatGetArray_MPIDense,
143609dc0095SBarry Smith        MatRestoreArray_MPIDense,
143797304618SKris Buschelman /*35*/ MatDuplicate_MPIDense,
143809dc0095SBarry Smith        0,
143909dc0095SBarry Smith        0,
144009dc0095SBarry Smith        0,
144109dc0095SBarry Smith        0,
144297304618SKris Buschelman /*40*/ 0,
14432ce60cd0SSatish Balay        MatGetSubMatrices_MPIDense,
144409dc0095SBarry Smith        0,
144509dc0095SBarry Smith        MatGetValues_MPIDense,
144609dc0095SBarry Smith        0,
144797304618SKris Buschelman /*45*/ 0,
144809dc0095SBarry Smith        MatScale_MPIDense,
144909dc0095SBarry Smith        0,
145009dc0095SBarry Smith        0,
145109dc0095SBarry Smith        0,
1452521d7252SBarry Smith /*50*/ 0,
145309dc0095SBarry Smith        0,
145409dc0095SBarry Smith        0,
145509dc0095SBarry Smith        0,
145609dc0095SBarry Smith        0,
145797304618SKris Buschelman /*55*/ 0,
145809dc0095SBarry Smith        0,
145909dc0095SBarry Smith        0,
146009dc0095SBarry Smith        0,
146109dc0095SBarry Smith        0,
146297304618SKris Buschelman /*60*/ MatGetSubMatrix_MPIDense,
1463b9b97703SBarry Smith        MatDestroy_MPIDense,
1464b9b97703SBarry Smith        MatView_MPIDense,
1465357abbc8SBarry Smith        0,
146697304618SKris Buschelman        0,
146797304618SKris Buschelman /*65*/ 0,
146897304618SKris Buschelman        0,
146997304618SKris Buschelman        0,
147097304618SKris Buschelman        0,
147197304618SKris Buschelman        0,
147297304618SKris Buschelman /*70*/ 0,
147397304618SKris Buschelman        0,
147497304618SKris Buschelman        0,
147597304618SKris Buschelman        0,
147697304618SKris Buschelman        0,
147797304618SKris Buschelman /*75*/ 0,
147897304618SKris Buschelman        0,
147997304618SKris Buschelman        0,
148097304618SKris Buschelman        0,
148197304618SKris Buschelman        0,
148297304618SKris Buschelman /*80*/ 0,
148397304618SKris Buschelman        0,
148497304618SKris Buschelman        0,
148597304618SKris Buschelman        0,
1486865e5f61SKris Buschelman /*84*/ MatLoad_MPIDense,
1487865e5f61SKris Buschelman        0,
1488865e5f61SKris Buschelman        0,
1489865e5f61SKris Buschelman        0,
1490865e5f61SKris Buschelman        0,
1491865e5f61SKris Buschelman        0,
14922fbe02b9SBarry Smith /*90*/
14932fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
14942fbe02b9SBarry Smith        MatMatMult_MPIDense_MPIDense,
14954ae313f4SHong Zhang        MatMatMultSymbolic_MPIDense_MPIDense,
14962fbe02b9SBarry Smith        MatMatMultNumeric_MPIDense_MPIDense,
14972fbe02b9SBarry Smith #else
1498865e5f61SKris Buschelman        0,
1499865e5f61SKris Buschelman        0,
1500865e5f61SKris Buschelman        0,
15012fbe02b9SBarry Smith #endif
15022fbe02b9SBarry Smith        0,
1503865e5f61SKris Buschelman /*95*/ 0,
1504865e5f61SKris Buschelman        0,
1505865e5f61SKris Buschelman        0,
1506865e5f61SKris Buschelman        0};
15078965ea79SLois Curfman McInnes 
1508273d9f13SBarry Smith EXTERN_C_BEGIN
15094a2ae208SSatish Balay #undef __FUNCT__
1510a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense"
1511be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data)
1512a23d5eceSKris Buschelman {
1513a23d5eceSKris Buschelman   Mat_MPIDense   *a;
1514dfbe8321SBarry Smith   PetscErrorCode ierr;
1515a23d5eceSKris Buschelman 
1516a23d5eceSKris Buschelman   PetscFunctionBegin;
1517a23d5eceSKris Buschelman   mat->preallocated = PETSC_TRUE;
1518a23d5eceSKris Buschelman   /* Note:  For now, when data is specified above, this assumes the user correctly
1519a23d5eceSKris Buschelman    allocates the local dense storage space.  We should add error checking. */
1520a23d5eceSKris Buschelman 
1521a23d5eceSKris Buschelman   a    = (Mat_MPIDense*)mat->data;
1522f69a0ea3SMatthew Knepley   ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr);
1523d0f46423SBarry Smith   ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr);
15245c5985e7SKris Buschelman   ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr);
15255c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr);
152652e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
1527a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1528a23d5eceSKris Buschelman }
1529a23d5eceSKris Buschelman EXTERN_C_END
1530a23d5eceSKris Buschelman 
15310bad9183SKris Buschelman /*MC
1532fafad747SKris Buschelman    MATMPIDENSE - MATMPIDENSE = "mpidense" - A matrix type to be used for distributed dense matrices.
15330bad9183SKris Buschelman 
15340bad9183SKris Buschelman    Options Database Keys:
15350bad9183SKris Buschelman . -mat_type mpidense - sets the matrix type to "mpidense" during a call to MatSetFromOptions()
15360bad9183SKris Buschelman 
15370bad9183SKris Buschelman   Level: beginner
15380bad9183SKris Buschelman 
15397878bbefSBarry Smith   MATMPIDENSE matrices may use direct solvers (LU, Cholesky, and QR)
15407878bbefSBarry Smith   for parallel dense matrices via the external package PLAPACK, if PLAPACK is installed
15417878bbefSBarry Smith   (run config/configure.py with the option --download-plapack)
15427878bbefSBarry Smith 
15437878bbefSBarry Smith 
15447878bbefSBarry Smith   Options Database Keys:
15457878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition
15467878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition
15477878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector
15487878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size
15497878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag
15507878bbefSBarry Smith 
15517878bbefSBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE
15520bad9183SKris Buschelman M*/
15530bad9183SKris Buschelman 
1554a23d5eceSKris Buschelman EXTERN_C_BEGIN
1555a23d5eceSKris Buschelman #undef __FUNCT__
15564a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense"
1557be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat)
1558273d9f13SBarry Smith {
1559273d9f13SBarry Smith   Mat_MPIDense   *a;
1560dfbe8321SBarry Smith   PetscErrorCode ierr;
1561273d9f13SBarry Smith 
1562273d9f13SBarry Smith   PetscFunctionBegin;
156338f2d2fdSLisandro Dalcin   ierr              = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr);
1564b0a32e0cSBarry Smith   mat->data         = (void*)a;
1565273d9f13SBarry Smith   ierr              = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1566273d9f13SBarry Smith   mat->mapping      = 0;
1567273d9f13SBarry Smith 
1568273d9f13SBarry Smith   mat->insertmode = NOT_SET_VALUES;
15697adad957SLisandro Dalcin   ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr);
15707adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr);
1571273d9f13SBarry Smith 
1572d0f46423SBarry Smith   mat->rmap->bs = mat->cmap->bs = 1;
1573d0f46423SBarry Smith   ierr = PetscMapSetUp(mat->rmap);CHKERRQ(ierr);
1574d0f46423SBarry Smith   ierr = PetscMapSetUp(mat->cmap);CHKERRQ(ierr);
1575d0f46423SBarry Smith   a->nvec = mat->cmap->n;
1576273d9f13SBarry Smith 
1577273d9f13SBarry Smith   /* build cache for off array entries formed */
1578273d9f13SBarry Smith   a->donotstash = PETSC_FALSE;
15797adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr);
1580273d9f13SBarry Smith 
1581273d9f13SBarry Smith   /* stuff used for matrix vector multiply */
1582273d9f13SBarry Smith   a->lvec        = 0;
1583273d9f13SBarry Smith   a->Mvctx       = 0;
1584273d9f13SBarry Smith   a->roworiented = PETSC_TRUE;
1585273d9f13SBarry Smith 
1586273d9f13SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C",
1587273d9f13SBarry Smith                                      "MatGetDiagonalBlock_MPIDense",
1588273d9f13SBarry Smith                                      MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr);
1589a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C",
1590a23d5eceSKris Buschelman                                      "MatMPIDenseSetPreallocation_MPIDense",
1591a23d5eceSKris Buschelman                                      MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr);
15924ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C",
15934ae313f4SHong Zhang                                      "MatMatMult_MPIAIJ_MPIDense",
15944ae313f4SHong Zhang                                       MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr);
15954ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C",
15964ae313f4SHong Zhang                                      "MatMatMultSymbolic_MPIAIJ_MPIDense",
15974ae313f4SHong Zhang                                       MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr);
15984ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C",
15994ae313f4SHong Zhang                                      "MatMatMultNumeric_MPIAIJ_MPIDense",
16004ae313f4SHong Zhang                                       MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr);
16017878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
160285bd4cefSHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_petsc_C",
160385bd4cefSHong Zhang                                      "MatGetFactor_mpidense_petsc",
160485bd4cefSHong Zhang                                       MatGetFactor_mpidense_petsc);CHKERRQ(ierr);
1605db4efbfdSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1606db4efbfdSBarry Smith   ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr);
1607db4efbfdSBarry Smith #endif
1608db4efbfdSBarry Smith 
16097878bbefSBarry Smith #endif
161038aed534SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr);
161101b82886SBarry Smith 
1612273d9f13SBarry Smith   PetscFunctionReturn(0);
1613273d9f13SBarry Smith }
1614273d9f13SBarry Smith EXTERN_C_END
1615273d9f13SBarry Smith 
1616209238afSKris Buschelman /*MC
1617002d173eSKris Buschelman    MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices.
1618209238afSKris Buschelman 
1619209238afSKris Buschelman    This matrix type is identical to MATSEQDENSE when constructed with a single process communicator,
1620209238afSKris Buschelman    and MATMPIDENSE otherwise.
1621209238afSKris Buschelman 
1622209238afSKris Buschelman    Options Database Keys:
1623209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions()
1624209238afSKris Buschelman 
1625209238afSKris Buschelman   Level: beginner
1626209238afSKris Buschelman 
162701b82886SBarry Smith 
1628209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE
1629209238afSKris Buschelman M*/
1630209238afSKris Buschelman 
1631209238afSKris Buschelman EXTERN_C_BEGIN
1632209238afSKris Buschelman #undef __FUNCT__
1633209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense"
1634be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A)
1635dfbe8321SBarry Smith {
16366849ba73SBarry Smith   PetscErrorCode ierr;
163713f74950SBarry Smith   PetscMPIInt    size;
1638209238afSKris Buschelman 
1639209238afSKris Buschelman   PetscFunctionBegin;
16407adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1641209238afSKris Buschelman   if (size == 1) {
1642209238afSKris Buschelman     ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr);
1643209238afSKris Buschelman   } else {
1644209238afSKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
1645209238afSKris Buschelman   }
1646209238afSKris Buschelman   PetscFunctionReturn(0);
1647209238afSKris Buschelman }
1648209238afSKris Buschelman EXTERN_C_END
1649209238afSKris Buschelman 
16504a2ae208SSatish Balay #undef __FUNCT__
16514a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation"
1652273d9f13SBarry Smith /*@C
1653273d9f13SBarry Smith    MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries
1654273d9f13SBarry Smith 
1655273d9f13SBarry Smith    Not collective
1656273d9f13SBarry Smith 
1657273d9f13SBarry Smith    Input Parameters:
1658273d9f13SBarry Smith .  A - the matrix
1659273d9f13SBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1660273d9f13SBarry Smith    to control all matrix memory allocation.
1661273d9f13SBarry Smith 
1662273d9f13SBarry Smith    Notes:
1663273d9f13SBarry Smith    The dense format is fully compatible with standard Fortran 77
1664273d9f13SBarry Smith    storage by columns.
1665273d9f13SBarry Smith 
1666273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1667273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1668273d9f13SBarry Smith    set data=PETSC_NULL.
1669273d9f13SBarry Smith 
1670273d9f13SBarry Smith    Level: intermediate
1671273d9f13SBarry Smith 
1672273d9f13SBarry Smith .keywords: matrix,dense, parallel
1673273d9f13SBarry Smith 
1674273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
1675273d9f13SBarry Smith @*/
1676be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data)
1677273d9f13SBarry Smith {
1678dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar *);
1679273d9f13SBarry Smith 
1680273d9f13SBarry Smith   PetscFunctionBegin;
1681565567f0SKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1682a23d5eceSKris Buschelman   if (f) {
1683a23d5eceSKris Buschelman     ierr = (*f)(mat,data);CHKERRQ(ierr);
1684a23d5eceSKris Buschelman   }
1685273d9f13SBarry Smith   PetscFunctionReturn(0);
1686273d9f13SBarry Smith }
1687273d9f13SBarry Smith 
16884a2ae208SSatish Balay #undef __FUNCT__
16894a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense"
16908965ea79SLois Curfman McInnes /*@C
169139ddd567SLois Curfman McInnes    MatCreateMPIDense - Creates a sparse parallel matrix in dense format.
16928965ea79SLois Curfman McInnes 
1693db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1694db81eaa0SLois Curfman McInnes 
16958965ea79SLois Curfman McInnes    Input Parameters:
1696db81eaa0SLois Curfman McInnes +  comm - MPI communicator
16978965ea79SLois Curfman McInnes .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
1698db81eaa0SLois Curfman McInnes .  n - number of local columns (or PETSC_DECIDE to have calculated if N is given)
16998965ea79SLois Curfman McInnes .  M - number of global rows (or PETSC_DECIDE to have calculated if m is given)
1700db81eaa0SLois Curfman McInnes .  N - number of global columns (or PETSC_DECIDE to have calculated if n is given)
17017f5ff6fdSBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc
1702dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
17038965ea79SLois Curfman McInnes 
17048965ea79SLois Curfman McInnes    Output Parameter:
1705477f1c0bSLois Curfman McInnes .  A - the matrix
17068965ea79SLois Curfman McInnes 
1707b259b22eSLois Curfman McInnes    Notes:
170839ddd567SLois Curfman McInnes    The dense format is fully compatible with standard Fortran 77
170939ddd567SLois Curfman McInnes    storage by columns.
17108965ea79SLois Curfman McInnes 
171118f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
171218f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
17137f5ff6fdSBarry Smith    set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users).
171418f449edSLois Curfman McInnes 
17158965ea79SLois Curfman McInnes    The user MUST specify either the local or global matrix dimensions
17168965ea79SLois Curfman McInnes    (possibly both).
17178965ea79SLois Curfman McInnes 
1718027ccd11SLois Curfman McInnes    Level: intermediate
1719027ccd11SLois Curfman McInnes 
172039ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel
17218965ea79SLois Curfman McInnes 
172239ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
17238965ea79SLois Curfman McInnes @*/
1724be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A)
17258965ea79SLois Curfman McInnes {
17266849ba73SBarry Smith   PetscErrorCode ierr;
172713f74950SBarry Smith   PetscMPIInt    size;
17288965ea79SLois Curfman McInnes 
17293a40ed3dSBarry Smith   PetscFunctionBegin;
1730f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1731f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr);
1732273d9f13SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1733273d9f13SBarry Smith   if (size > 1) {
1734273d9f13SBarry Smith     ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr);
1735273d9f13SBarry Smith     ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1736273d9f13SBarry Smith   } else {
1737273d9f13SBarry Smith     ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1738273d9f13SBarry Smith     ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
17398c469469SLois Curfman McInnes   }
17403a40ed3dSBarry Smith   PetscFunctionReturn(0);
17418965ea79SLois Curfman McInnes }
17428965ea79SLois Curfman McInnes 
17434a2ae208SSatish Balay #undef __FUNCT__
17444a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense"
17456849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
17468965ea79SLois Curfman McInnes {
17478965ea79SLois Curfman McInnes   Mat            mat;
17483501a2bdSLois Curfman McInnes   Mat_MPIDense   *a,*oldmat = (Mat_MPIDense*)A->data;
1749dfbe8321SBarry Smith   PetscErrorCode ierr;
17508965ea79SLois Curfman McInnes 
17513a40ed3dSBarry Smith   PetscFunctionBegin;
17528965ea79SLois Curfman McInnes   *newmat       = 0;
17537adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr);
1754d0f46423SBarry Smith   ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
17557adad957SLisandro Dalcin   ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr);
1756834f8fabSBarry Smith   a                 = (Mat_MPIDense*)mat->data;
1757e04c1aa4SHong Zhang   ierr              = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
17583501a2bdSLois Curfman McInnes   mat->factor       = A->factor;
1759c456f294SBarry Smith   mat->assembled    = PETSC_TRUE;
1760273d9f13SBarry Smith   mat->preallocated = PETSC_TRUE;
17618965ea79SLois Curfman McInnes 
1762d0f46423SBarry Smith   mat->rmap->rstart     = A->rmap->rstart;
1763d0f46423SBarry Smith   mat->rmap->rend       = A->rmap->rend;
17648965ea79SLois Curfman McInnes   a->size              = oldmat->size;
17658965ea79SLois Curfman McInnes   a->rank              = oldmat->rank;
1766e0fa3b82SLois Curfman McInnes   mat->insertmode      = NOT_SET_VALUES;
1767b9b97703SBarry Smith   a->nvec              = oldmat->nvec;
17683782ba37SSatish Balay   a->donotstash        = oldmat->donotstash;
1769e04c1aa4SHong Zhang 
1770d0f46423SBarry Smith   ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
1771d0f46423SBarry Smith   ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
17727adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr);
17738965ea79SLois Curfman McInnes 
1774329f5518SBarry Smith   ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
17755609ef8eSBarry Smith   ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr);
177652e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
177701b82886SBarry Smith 
17788965ea79SLois Curfman McInnes   *newmat = mat;
17793a40ed3dSBarry Smith   PetscFunctionReturn(0);
17808965ea79SLois Curfman McInnes }
17818965ea79SLois Curfman McInnes 
1782e090d566SSatish Balay #include "petscsys.h"
17838965ea79SLois Curfman McInnes 
17844a2ae208SSatish Balay #undef __FUNCT__
17854a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile"
1786a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat)
178790ace30eSBarry Smith {
17886849ba73SBarry Smith   PetscErrorCode ierr;
178932dcc486SBarry Smith   PetscMPIInt    rank,size;
179013f74950SBarry Smith   PetscInt       *rowners,i,m,nz,j;
179187828ca2SBarry Smith   PetscScalar    *array,*vals,*vals_ptr;
179290ace30eSBarry Smith   MPI_Status     status;
179390ace30eSBarry Smith 
17943a40ed3dSBarry Smith   PetscFunctionBegin;
1795d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
1796d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
179790ace30eSBarry Smith 
179890ace30eSBarry Smith   /* determine ownership of all rows */
179990ace30eSBarry Smith   m          = M/size + ((M % size) > rank);
180013f74950SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr);
180113f74950SBarry Smith   ierr       = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr);
180290ace30eSBarry Smith   rowners[0] = 0;
180390ace30eSBarry Smith   for (i=2; i<=size; i++) {
180490ace30eSBarry Smith     rowners[i] += rowners[i-1];
180590ace30eSBarry Smith   }
180690ace30eSBarry Smith 
1807f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1808f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1809be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1810878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
181190ace30eSBarry Smith   ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr);
181290ace30eSBarry Smith 
181390ace30eSBarry Smith   if (!rank) {
181487828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
181590ace30eSBarry Smith 
181690ace30eSBarry Smith     /* read in my part of the matrix numerical values  */
18170752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr);
181890ace30eSBarry Smith 
181990ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
182090ace30eSBarry Smith     vals_ptr = vals;
182190ace30eSBarry Smith     for (i=0; i<m; i++) {
182290ace30eSBarry Smith       for (j=0; j<N; j++) {
182390ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
182490ace30eSBarry Smith       }
182590ace30eSBarry Smith     }
182690ace30eSBarry Smith 
182790ace30eSBarry Smith     /* read in other processors and ship out */
182890ace30eSBarry Smith     for (i=1; i<size; i++) {
182990ace30eSBarry Smith       nz   = (rowners[i+1] - rowners[i])*N;
18300752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
18317adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr);
183290ace30eSBarry Smith     }
18333a40ed3dSBarry Smith   } else {
183490ace30eSBarry Smith     /* receive numeric values */
183587828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
183690ace30eSBarry Smith 
183790ace30eSBarry Smith     /* receive message of values*/
18387adad957SLisandro Dalcin     ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr);
183990ace30eSBarry Smith 
184090ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
184190ace30eSBarry Smith     vals_ptr = vals;
184290ace30eSBarry Smith     for (i=0; i<m; i++) {
184390ace30eSBarry Smith       for (j=0; j<N; j++) {
184490ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
184590ace30eSBarry Smith       }
184690ace30eSBarry Smith     }
184790ace30eSBarry Smith   }
1848606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
1849606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
18506d4a8577SBarry Smith   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
18516d4a8577SBarry Smith   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
18523a40ed3dSBarry Smith   PetscFunctionReturn(0);
185390ace30eSBarry Smith }
185490ace30eSBarry Smith 
18554a2ae208SSatish Balay #undef __FUNCT__
18564a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense"
1857a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat)
18588965ea79SLois Curfman McInnes {
18598965ea79SLois Curfman McInnes   Mat            A;
186087828ca2SBarry Smith   PetscScalar    *vals,*svals;
186119bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
18628965ea79SLois Curfman McInnes   MPI_Status     status;
186313f74950SBarry Smith   PetscMPIInt    rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz;
186413f74950SBarry Smith   PetscInt       header[4],*rowlengths = 0,M,N,*cols;
186513f74950SBarry Smith   PetscInt       *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols;
186613f74950SBarry Smith   PetscInt       i,nz,j,rstart,rend;
186713f74950SBarry Smith   int            fd;
18686849ba73SBarry Smith   PetscErrorCode ierr;
18698965ea79SLois Curfman McInnes 
18703a40ed3dSBarry Smith   PetscFunctionBegin;
1871d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1872d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
18738965ea79SLois Curfman McInnes   if (!rank) {
1874b0a32e0cSBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
18750752156aSBarry Smith     ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr);
1876552e946dSBarry Smith     if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object");
18778965ea79SLois Curfman McInnes   }
18788965ea79SLois Curfman McInnes 
187913f74950SBarry Smith   ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr);
188090ace30eSBarry Smith   M = header[1]; N = header[2]; nz = header[3];
188190ace30eSBarry Smith 
188290ace30eSBarry Smith   /*
188390ace30eSBarry Smith        Handle case where matrix is stored on disk as a dense matrix
188490ace30eSBarry Smith   */
188590ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) {
1886be5d1d56SKris Buschelman     ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr);
18873a40ed3dSBarry Smith     PetscFunctionReturn(0);
188890ace30eSBarry Smith   }
188990ace30eSBarry Smith 
18908965ea79SLois Curfman McInnes   /* determine ownership of all rows */
1891e44c0bd4SBarry Smith   m          = PetscMPIIntCast(M/size + ((M % size) > rank));
1892e44c0bd4SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr);
1893ca161407SBarry Smith   ierr       = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr);
18948965ea79SLois Curfman McInnes   rowners[0] = 0;
18958965ea79SLois Curfman McInnes   for (i=2; i<=size; i++) {
18968965ea79SLois Curfman McInnes     rowners[i] += rowners[i-1];
18978965ea79SLois Curfman McInnes   }
18988965ea79SLois Curfman McInnes   rstart = rowners[rank];
18998965ea79SLois Curfman McInnes   rend   = rowners[rank+1];
19008965ea79SLois Curfman McInnes 
19018965ea79SLois Curfman McInnes   /* distribute row lengths to all processors */
190213f74950SBarry Smith   ierr    = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr);
19038965ea79SLois Curfman McInnes   offlens = ourlens + (rend-rstart);
19048965ea79SLois Curfman McInnes   if (!rank) {
190513f74950SBarry Smith     ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
19060752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
190713f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr);
19088965ea79SLois Curfman McInnes     for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i];
19091466f1e1SBarry Smith     ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
1910606d414cSSatish Balay     ierr = PetscFree(sndcounts);CHKERRQ(ierr);
1911ca161407SBarry Smith   } else {
19121466f1e1SBarry Smith     ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
19138965ea79SLois Curfman McInnes   }
19148965ea79SLois Curfman McInnes 
19158965ea79SLois Curfman McInnes   if (!rank) {
19168965ea79SLois Curfman McInnes     /* calculate the number of nonzeros on each processor */
191713f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr);
191813f74950SBarry Smith     ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr);
19198965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
19208965ea79SLois Curfman McInnes       for (j=rowners[i]; j< rowners[i+1]; j++) {
19218965ea79SLois Curfman McInnes         procsnz[i] += rowlengths[j];
19228965ea79SLois Curfman McInnes       }
19238965ea79SLois Curfman McInnes     }
1924606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
19258965ea79SLois Curfman McInnes 
19268965ea79SLois Curfman McInnes     /* determine max buffer needed and allocate it */
19278965ea79SLois Curfman McInnes     maxnz = 0;
19288965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
19290452661fSBarry Smith       maxnz = PetscMax(maxnz,procsnz[i]);
19308965ea79SLois Curfman McInnes     }
193113f74950SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr);
19328965ea79SLois Curfman McInnes 
19338965ea79SLois Curfman McInnes     /* read in my part of the matrix column indices  */
19348965ea79SLois Curfman McInnes     nz = procsnz[0];
193513f74950SBarry Smith     ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
19360752156aSBarry Smith     ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr);
19378965ea79SLois Curfman McInnes 
19388965ea79SLois Curfman McInnes     /* read in every one elses and ship off */
19398965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
19408965ea79SLois Curfman McInnes       nz   = procsnz[i];
19410752156aSBarry Smith       ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
194213f74950SBarry Smith       ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr);
19438965ea79SLois Curfman McInnes     }
1944606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
19453a40ed3dSBarry Smith   } else {
19468965ea79SLois Curfman McInnes     /* determine buffer space needed for message */
19478965ea79SLois Curfman McInnes     nz = 0;
19488965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
19498965ea79SLois Curfman McInnes       nz += ourlens[i];
19508965ea79SLois Curfman McInnes     }
195113f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
19528965ea79SLois Curfman McInnes 
19538965ea79SLois Curfman McInnes     /* receive message of column indices*/
195413f74950SBarry Smith     ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr);
195513f74950SBarry Smith     ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr);
195629bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
19578965ea79SLois Curfman McInnes   }
19588965ea79SLois Curfman McInnes 
19598965ea79SLois Curfman McInnes   /* loop over local rows, determining number of off diagonal entries */
196013f74950SBarry Smith   ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr);
19618965ea79SLois Curfman McInnes   jj = 0;
19628965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
19638965ea79SLois Curfman McInnes     for (j=0; j<ourlens[i]; j++) {
19648965ea79SLois Curfman McInnes       if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++;
19658965ea79SLois Curfman McInnes       jj++;
19668965ea79SLois Curfman McInnes     }
19678965ea79SLois Curfman McInnes   }
19688965ea79SLois Curfman McInnes 
19698965ea79SLois Curfman McInnes   /* create our matrix */
19708965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
19718965ea79SLois Curfman McInnes     ourlens[i] -= offlens[i];
19728965ea79SLois Curfman McInnes   }
1973f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1974f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1975be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1976878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
19778965ea79SLois Curfman McInnes   A = *newmat;
19788965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
19798965ea79SLois Curfman McInnes     ourlens[i] += offlens[i];
19808965ea79SLois Curfman McInnes   }
19818965ea79SLois Curfman McInnes 
19828965ea79SLois Curfman McInnes   if (!rank) {
198387828ca2SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
19848965ea79SLois Curfman McInnes 
19858965ea79SLois Curfman McInnes     /* read in my part of the matrix numerical values  */
19868965ea79SLois Curfman McInnes     nz = procsnz[0];
19870752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
19888965ea79SLois Curfman McInnes 
19898965ea79SLois Curfman McInnes     /* insert into matrix */
19908965ea79SLois Curfman McInnes     jj      = rstart;
19918965ea79SLois Curfman McInnes     smycols = mycols;
19928965ea79SLois Curfman McInnes     svals   = vals;
19938965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
19948965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
19958965ea79SLois Curfman McInnes       smycols += ourlens[i];
19968965ea79SLois Curfman McInnes       svals   += ourlens[i];
19978965ea79SLois Curfman McInnes       jj++;
19988965ea79SLois Curfman McInnes     }
19998965ea79SLois Curfman McInnes 
20008965ea79SLois Curfman McInnes     /* read in other processors and ship out */
20018965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
20028965ea79SLois Curfman McInnes       nz   = procsnz[i];
20030752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20047adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr);
20058965ea79SLois Curfman McInnes     }
2006606d414cSSatish Balay     ierr = PetscFree(procsnz);CHKERRQ(ierr);
20073a40ed3dSBarry Smith   } else {
20088965ea79SLois Curfman McInnes     /* receive numeric values */
200984d528b1SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20108965ea79SLois Curfman McInnes 
20118965ea79SLois Curfman McInnes     /* receive message of values*/
20127adad957SLisandro Dalcin     ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr);
2013ca161407SBarry Smith     ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr);
201429bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
20158965ea79SLois Curfman McInnes 
20168965ea79SLois Curfman McInnes     /* insert into matrix */
20178965ea79SLois Curfman McInnes     jj      = rstart;
20188965ea79SLois Curfman McInnes     smycols = mycols;
20198965ea79SLois Curfman McInnes     svals   = vals;
20208965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20218965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
20228965ea79SLois Curfman McInnes       smycols += ourlens[i];
20238965ea79SLois Curfman McInnes       svals   += ourlens[i];
20248965ea79SLois Curfman McInnes       jj++;
20258965ea79SLois Curfman McInnes     }
20268965ea79SLois Curfman McInnes   }
2027606d414cSSatish Balay   ierr = PetscFree(ourlens);CHKERRQ(ierr);
2028606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
2029606d414cSSatish Balay   ierr = PetscFree(mycols);CHKERRQ(ierr);
2030606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
20318965ea79SLois Curfman McInnes 
20326d4a8577SBarry Smith   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
20336d4a8577SBarry Smith   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
20343a40ed3dSBarry Smith   PetscFunctionReturn(0);
20358965ea79SLois Curfman McInnes }
203690ace30eSBarry Smith 
20376e4ee0c6SHong Zhang #undef __FUNCT__
20386e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense"
20396e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag)
20406e4ee0c6SHong Zhang {
20416e4ee0c6SHong Zhang   Mat_MPIDense   *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data;
20426e4ee0c6SHong Zhang   Mat            a,b;
20436e4ee0c6SHong Zhang   PetscTruth     flg;
20446e4ee0c6SHong Zhang   PetscErrorCode ierr;
204590ace30eSBarry Smith 
20466e4ee0c6SHong Zhang   PetscFunctionBegin;
20476e4ee0c6SHong Zhang   a = matA->A;
20486e4ee0c6SHong Zhang   b = matB->A;
20496e4ee0c6SHong Zhang   ierr = MatEqual(a,b,&flg);CHKERRQ(ierr);
20507adad957SLisandro Dalcin   ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr);
20516e4ee0c6SHong Zhang   PetscFunctionReturn(0);
20526e4ee0c6SHong Zhang }
205390ace30eSBarry Smith 
205409d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
205509d27a7eSBarry Smith 
205609d27a7eSBarry Smith #undef __FUNCT__
205709d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage"
205809d27a7eSBarry Smith /*@C
205909d27a7eSBarry Smith   PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK.
206009d27a7eSBarry Smith   Level: developer
206109d27a7eSBarry Smith 
206209d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK
206309d27a7eSBarry Smith .seealso: PetscFinalize()
206409d27a7eSBarry Smith @*/
206509d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void)
206609d27a7eSBarry Smith {
206709d27a7eSBarry Smith   PetscErrorCode ierr;
206809d27a7eSBarry Smith 
206909d27a7eSBarry Smith   PetscFunctionBegin;
207009d27a7eSBarry Smith   ierr = PLA_Finalize();CHKERRQ(ierr);
207109d27a7eSBarry Smith   PetscFunctionReturn(0);
207209d27a7eSBarry Smith }
207309d27a7eSBarry Smith 
207409d27a7eSBarry Smith #undef __FUNCT__
207509d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage"
207609d27a7eSBarry Smith /*@C
207709d27a7eSBarry Smith   PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is
2078db4efbfdSBarry Smith   called from MatCreate_MPIDense() the first time an MPI dense matrix is called.
207909d27a7eSBarry Smith 
208009d27a7eSBarry Smith   Input Parameter:
2081db4efbfdSBarry Smith .   comm - the communicator the matrix lives on
208209d27a7eSBarry Smith 
208309d27a7eSBarry Smith   Level: developer
208409d27a7eSBarry Smith 
2085db4efbfdSBarry Smith    Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the
2086db4efbfdSBarry Smith   same communicator (because there is some global state that is initialized and used for all matrices). In addition if
2087db4efbfdSBarry Smith   PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators
2088db4efbfdSBarry Smith   cannot overlap.
2089db4efbfdSBarry Smith 
209009d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK
209109d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize()
209209d27a7eSBarry Smith @*/
2093db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm)
209409d27a7eSBarry Smith {
2095eae6fb2eSBarry Smith   PetscMPIInt    size;
209609d27a7eSBarry Smith   PetscErrorCode ierr;
209709d27a7eSBarry Smith 
209809d27a7eSBarry Smith   PetscFunctionBegin;
209909d27a7eSBarry Smith   if (!PLA_Initialized(PETSC_NULL)) {
210009d27a7eSBarry Smith 
210109d27a7eSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
210204fea9ffSBarry Smith     Plapack_nprows = 1;
210304fea9ffSBarry Smith     Plapack_npcols = size;
210409d27a7eSBarry Smith 
2105aeccfd6fSBarry Smith     ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr);
2106*5d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr);
2107*5d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr);
210879b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG)
210979b0a62dSBarry Smith       Plapack_ierror = 3;
211079b0a62dSBarry Smith #else
211179b0a62dSBarry Smith       Plapack_ierror = 0;
211279b0a62dSBarry Smith #endif
2113*5d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr);
2114eae6fb2eSBarry Smith       if (Plapack_ierror){
2115eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr);
2116aeccfd6fSBarry Smith       } else {
2117eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr);
2118aeccfd6fSBarry Smith       }
2119aeccfd6fSBarry Smith 
2120eae6fb2eSBarry Smith       Plapack_nb_alg = 0;
2121*5d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr);
2122eae6fb2eSBarry Smith       if (Plapack_nb_alg) {
2123eae6fb2eSBarry Smith         ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr);
2124aeccfd6fSBarry Smith       }
2125aeccfd6fSBarry Smith     PetscOptionsEnd();
2126aeccfd6fSBarry Smith 
212704fea9ffSBarry Smith     ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr);
212804fea9ffSBarry Smith     ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr);
212909d27a7eSBarry Smith     ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr);
213009d27a7eSBarry Smith   }
213109d27a7eSBarry Smith   PetscFunctionReturn(0);
213209d27a7eSBarry Smith }
213390ace30eSBarry Smith 
213409d27a7eSBarry Smith #endif
2135