xref: /petsc/src/mat/impls/dense/mpi/mpidense.c (revision 4aa3045d83914e00674d51ff6fb6b94ef17679f6)
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"
184*4aa3045dSJed Brown static PetscErrorCode MatGetSubMatrix_MPIDense(Mat A,IS isrow,IS iscol,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;
189*4aa3045dSJed Brown   PetscInt       i,j,rstart,rend,nrows,ncols,Ncols,nlrows,nlcols;
1905d0c19d7SBarry Smith   const PetscInt *irow,*icol;
19187828ca2SBarry Smith   PetscScalar    *av,*bv,*v = lmat->v;
192ca3fa75bSLois Curfman McInnes   Mat            newmat;
193*4aa3045dSJed Brown   IS             iscol_local;
194ca3fa75bSLois Curfman McInnes 
195ca3fa75bSLois Curfman McInnes   PetscFunctionBegin;
196*4aa3045dSJed Brown   ierr = ISAllGather(iscol,&iscol_local);CHKERRQ(ierr);
197ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
198*4aa3045dSJed Brown   ierr = ISGetIndices(iscol_local,&icol);CHKERRQ(ierr);
199b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
200b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
201*4aa3045dSJed Brown   ierr = ISGetSize(iscol,&Ncols);CHKERRQ(ierr); /* global number of columns, size of iscol_local */
202ca3fa75bSLois Curfman McInnes 
203ca3fa75bSLois Curfman McInnes   /* No parallel redistribution currently supported! Should really check each index set
2047eba5e9cSLois Curfman McInnes      to comfirm that it is OK.  ... Currently supports only submatrix same partitioning as
2057eba5e9cSLois Curfman McInnes      original matrix! */
206ca3fa75bSLois Curfman McInnes 
207ca3fa75bSLois Curfman McInnes   ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr);
2087eba5e9cSLois Curfman McInnes   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
209ca3fa75bSLois Curfman McInnes 
210ca3fa75bSLois Curfman McInnes   /* Check submatrix call */
211ca3fa75bSLois Curfman McInnes   if (scall == MAT_REUSE_MATRIX) {
21229bbc08cSBarry Smith     /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */
2137eba5e9cSLois Curfman McInnes     /* Really need to test rows and column sizes! */
214ca3fa75bSLois Curfman McInnes     newmat = *B;
215ca3fa75bSLois Curfman McInnes   } else {
216ca3fa75bSLois Curfman McInnes     /* Create and fill new matrix */
2177adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
218*4aa3045dSJed Brown     ierr = MatSetSizes(newmat,nrows,ncols,PETSC_DECIDE,Ncols);CHKERRQ(ierr);
2197adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
220878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
221ca3fa75bSLois Curfman McInnes   }
222ca3fa75bSLois Curfman McInnes 
223ca3fa75bSLois Curfman McInnes   /* Now extract the data pointers and do the copy, column at a time */
224ca3fa75bSLois Curfman McInnes   newmatd = (Mat_MPIDense*)newmat->data;
225ca3fa75bSLois Curfman McInnes   bv      = ((Mat_SeqDense *)newmatd->A->data)->v;
226ca3fa75bSLois Curfman McInnes 
227*4aa3045dSJed Brown   for (i=0; i<Ncols; i++) {
22825a33276SHong Zhang     av = v + ((Mat_SeqDense *)mat->A->data)->lda*icol[i];
229ca3fa75bSLois Curfman McInnes     for (j=0; j<nrows; j++) {
2307eba5e9cSLois Curfman McInnes       *bv++ = av[irow[j] - rstart];
231ca3fa75bSLois Curfman McInnes     }
232ca3fa75bSLois Curfman McInnes   }
233ca3fa75bSLois Curfman McInnes 
234ca3fa75bSLois Curfman McInnes   /* Assemble the matrices so that the correct flags are set */
235ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
236ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
237ca3fa75bSLois Curfman McInnes 
238ca3fa75bSLois Curfman McInnes   /* Free work space */
239ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
240ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
241ca3fa75bSLois Curfman McInnes   *B = newmat;
242ca3fa75bSLois Curfman McInnes   PetscFunctionReturn(0);
243ca3fa75bSLois Curfman McInnes }
244ca3fa75bSLois Curfman McInnes 
2454a2ae208SSatish Balay #undef __FUNCT__
2464a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense"
247dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[])
248ff14e315SSatish Balay {
2493a40ed3dSBarry Smith   PetscFunctionBegin;
2503a40ed3dSBarry Smith   PetscFunctionReturn(0);
251ff14e315SSatish Balay }
252ff14e315SSatish Balay 
2534a2ae208SSatish Balay #undef __FUNCT__
2544a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense"
255dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode)
2568965ea79SLois Curfman McInnes {
25739ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
2587adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)mat)->comm;
259dfbe8321SBarry Smith   PetscErrorCode ierr;
26013f74950SBarry Smith   PetscInt       nstash,reallocs;
2618965ea79SLois Curfman McInnes   InsertMode     addv;
2628965ea79SLois Curfman McInnes 
2633a40ed3dSBarry Smith   PetscFunctionBegin;
2648965ea79SLois Curfman McInnes   /* make sure all processors are either in INSERTMODE or ADDMODE */
265ca161407SBarry Smith   ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr);
2667056b6fcSBarry Smith   if (addv == (ADD_VALUES|INSERT_VALUES)) {
26729bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs");
2688965ea79SLois Curfman McInnes   }
269e0fa3b82SLois Curfman McInnes   mat->insertmode = addv; /* in case this processor had no cache */
2708965ea79SLois Curfman McInnes 
271d0f46423SBarry Smith   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr);
2728798bf22SSatish Balay   ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr);
273ae15b995SBarry Smith   ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr);
2743a40ed3dSBarry Smith   PetscFunctionReturn(0);
2758965ea79SLois Curfman McInnes }
2768965ea79SLois Curfman McInnes 
2774a2ae208SSatish Balay #undef __FUNCT__
2784a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense"
279dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode)
2808965ea79SLois Curfman McInnes {
28139ddd567SLois Curfman McInnes   Mat_MPIDense    *mdn=(Mat_MPIDense*)mat->data;
2826849ba73SBarry Smith   PetscErrorCode  ierr;
28313f74950SBarry Smith   PetscInt        i,*row,*col,flg,j,rstart,ncols;
28413f74950SBarry Smith   PetscMPIInt     n;
28587828ca2SBarry Smith   PetscScalar     *val;
286e0fa3b82SLois Curfman McInnes   InsertMode      addv=mat->insertmode;
2878965ea79SLois Curfman McInnes 
2883a40ed3dSBarry Smith   PetscFunctionBegin;
2898965ea79SLois Curfman McInnes   /*  wait on receives */
2907ef1d9bdSSatish Balay   while (1) {
2918798bf22SSatish Balay     ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr);
2927ef1d9bdSSatish Balay     if (!flg) break;
2938965ea79SLois Curfman McInnes 
2947ef1d9bdSSatish Balay     for (i=0; i<n;) {
2957ef1d9bdSSatish Balay       /* Now identify the consecutive vals belonging to the same row */
2967ef1d9bdSSatish Balay       for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; }
2977ef1d9bdSSatish Balay       if (j < n) ncols = j-i;
2987ef1d9bdSSatish Balay       else       ncols = n-i;
2997ef1d9bdSSatish Balay       /* Now assemble all these values with a single function call */
3007ef1d9bdSSatish Balay       ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr);
3017ef1d9bdSSatish Balay       i = j;
3028965ea79SLois Curfman McInnes     }
3037ef1d9bdSSatish Balay   }
3048798bf22SSatish Balay   ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr);
3058965ea79SLois Curfman McInnes 
30639ddd567SLois Curfman McInnes   ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr);
30739ddd567SLois Curfman McInnes   ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr);
3088965ea79SLois Curfman McInnes 
3096d4a8577SBarry Smith   if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) {
31039ddd567SLois Curfman McInnes     ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
3118965ea79SLois Curfman McInnes   }
3123a40ed3dSBarry Smith   PetscFunctionReturn(0);
3138965ea79SLois Curfman McInnes }
3148965ea79SLois Curfman McInnes 
3154a2ae208SSatish Balay #undef __FUNCT__
3164a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense"
317dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A)
3188965ea79SLois Curfman McInnes {
319dfbe8321SBarry Smith   PetscErrorCode ierr;
32039ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3213a40ed3dSBarry Smith 
3223a40ed3dSBarry Smith   PetscFunctionBegin;
3233a40ed3dSBarry Smith   ierr = MatZeroEntries(l->A);CHKERRQ(ierr);
3243a40ed3dSBarry Smith   PetscFunctionReturn(0);
3258965ea79SLois Curfman McInnes }
3268965ea79SLois Curfman McInnes 
3278965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the
3288965ea79SLois Curfman McInnes    matrix is square and the column and row owerships are identical.
3298965ea79SLois Curfman McInnes    This is a BUG. The only way to fix it seems to be to access
3303501a2bdSLois Curfman McInnes    mdn->A and mdn->B directly and not through the MatZeroRows()
3318965ea79SLois Curfman McInnes    routine.
3328965ea79SLois Curfman McInnes */
3334a2ae208SSatish Balay #undef __FUNCT__
3344a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense"
335f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
3368965ea79SLois Curfman McInnes {
33739ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3386849ba73SBarry Smith   PetscErrorCode ierr;
339d0f46423SBarry Smith   PetscInt       i,*owners = A->rmap->range;
34013f74950SBarry Smith   PetscInt       *nprocs,j,idx,nsends;
34113f74950SBarry Smith   PetscInt       nmax,*svalues,*starts,*owner,nrecvs;
3427adad957SLisandro Dalcin   PetscInt       *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source;
34313f74950SBarry Smith   PetscInt       *lens,*lrows,*values;
34413f74950SBarry Smith   PetscMPIInt    n,imdex,rank = l->rank,size = l->size;
3457adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)A)->comm;
3468965ea79SLois Curfman McInnes   MPI_Request    *send_waits,*recv_waits;
3478965ea79SLois Curfman McInnes   MPI_Status     recv_status,*send_status;
34835d8aa7fSBarry Smith   PetscTruth     found;
3498965ea79SLois Curfman McInnes 
3503a40ed3dSBarry Smith   PetscFunctionBegin;
3518965ea79SLois Curfman McInnes   /*  first count number of contributors to each processor */
35213f74950SBarry Smith   ierr  = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
35313f74950SBarry Smith   ierr  = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
35413f74950SBarry Smith   ierr  = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/
3558965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3568965ea79SLois Curfman McInnes     idx = rows[i];
35735d8aa7fSBarry Smith     found = PETSC_FALSE;
3588965ea79SLois Curfman McInnes     for (j=0; j<size; j++) {
3598965ea79SLois Curfman McInnes       if (idx >= owners[j] && idx < owners[j+1]) {
360c1dc657dSBarry Smith         nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break;
3618965ea79SLois Curfman McInnes       }
3628965ea79SLois Curfman McInnes     }
36329bbc08cSBarry Smith     if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range");
3648965ea79SLois Curfman McInnes   }
365c1dc657dSBarry Smith   nsends = 0;  for (i=0; i<size; i++) { nsends += nprocs[2*i+1];}
3668965ea79SLois Curfman McInnes 
3678965ea79SLois Curfman McInnes   /* inform other processors of number of messages and max length*/
368c1dc657dSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
3698965ea79SLois Curfman McInnes 
3708965ea79SLois Curfman McInnes   /* post receives:   */
37113f74950SBarry Smith   ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr);
372b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
3738965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
37413f74950SBarry Smith     ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr);
3758965ea79SLois Curfman McInnes   }
3768965ea79SLois Curfman McInnes 
3778965ea79SLois Curfman McInnes   /* do sends:
3788965ea79SLois Curfman McInnes       1) starts[i] gives the starting index in svalues for stuff going to
3798965ea79SLois Curfman McInnes          the ith processor
3808965ea79SLois Curfman McInnes   */
38113f74950SBarry Smith   ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr);
382b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
38313f74950SBarry Smith   ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
3848965ea79SLois Curfman McInnes   starts[0]  = 0;
385c1dc657dSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3868965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3878965ea79SLois Curfman McInnes     svalues[starts[owner[i]]++] = rows[i];
3888965ea79SLois Curfman McInnes   }
3898965ea79SLois Curfman McInnes 
3908965ea79SLois Curfman McInnes   starts[0] = 0;
391c1dc657dSBarry Smith   for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3928965ea79SLois Curfman McInnes   count = 0;
3938965ea79SLois Curfman McInnes   for (i=0; i<size; i++) {
394c1dc657dSBarry Smith     if (nprocs[2*i+1]) {
39513f74950SBarry Smith       ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr);
3968965ea79SLois Curfman McInnes     }
3978965ea79SLois Curfman McInnes   }
398606d414cSSatish Balay   ierr = PetscFree(starts);CHKERRQ(ierr);
3998965ea79SLois Curfman McInnes 
4008965ea79SLois Curfman McInnes   base = owners[rank];
4018965ea79SLois Curfman McInnes 
4028965ea79SLois Curfman McInnes   /*  wait on receives */
40313f74950SBarry Smith   ierr   = PetscMalloc(2*(nrecvs+1)*sizeof(PetscInt),&lens);CHKERRQ(ierr);
4048965ea79SLois Curfman McInnes   source = lens + nrecvs;
4058965ea79SLois Curfman McInnes   count  = nrecvs; slen = 0;
4068965ea79SLois Curfman McInnes   while (count) {
407ca161407SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
4088965ea79SLois Curfman McInnes     /* unpack receives into our local space */
40913f74950SBarry Smith     ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr);
4108965ea79SLois Curfman McInnes     source[imdex]  = recv_status.MPI_SOURCE;
4118965ea79SLois Curfman McInnes     lens[imdex]    = n;
4128965ea79SLois Curfman McInnes     slen += n;
4138965ea79SLois Curfman McInnes     count--;
4148965ea79SLois Curfman McInnes   }
415606d414cSSatish Balay   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
4168965ea79SLois Curfman McInnes 
4178965ea79SLois Curfman McInnes   /* move the data into the send scatter */
41813f74950SBarry Smith   ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr);
4198965ea79SLois Curfman McInnes   count = 0;
4208965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
4218965ea79SLois Curfman McInnes     values = rvalues + i*nmax;
4228965ea79SLois Curfman McInnes     for (j=0; j<lens[i]; j++) {
4238965ea79SLois Curfman McInnes       lrows[count++] = values[j] - base;
4248965ea79SLois Curfman McInnes     }
4258965ea79SLois Curfman McInnes   }
426606d414cSSatish Balay   ierr = PetscFree(rvalues);CHKERRQ(ierr);
427606d414cSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
428606d414cSSatish Balay   ierr = PetscFree(owner);CHKERRQ(ierr);
429606d414cSSatish Balay   ierr = PetscFree(nprocs);CHKERRQ(ierr);
4308965ea79SLois Curfman McInnes 
4318965ea79SLois Curfman McInnes   /* actually zap the local rows */
432f4df32b1SMatthew Knepley   ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr);
433606d414cSSatish Balay   ierr = PetscFree(lrows);CHKERRQ(ierr);
4348965ea79SLois Curfman McInnes 
4358965ea79SLois Curfman McInnes   /* wait on sends */
4368965ea79SLois Curfman McInnes   if (nsends) {
437b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
438ca161407SBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
439606d414cSSatish Balay     ierr = PetscFree(send_status);CHKERRQ(ierr);
4408965ea79SLois Curfman McInnes   }
441606d414cSSatish Balay   ierr = PetscFree(send_waits);CHKERRQ(ierr);
442606d414cSSatish Balay   ierr = PetscFree(svalues);CHKERRQ(ierr);
4438965ea79SLois Curfman McInnes 
4443a40ed3dSBarry Smith   PetscFunctionReturn(0);
4458965ea79SLois Curfman McInnes }
4468965ea79SLois Curfman McInnes 
4474a2ae208SSatish Balay #undef __FUNCT__
4484a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense"
449dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy)
4508965ea79SLois Curfman McInnes {
45139ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
452dfbe8321SBarry Smith   PetscErrorCode ierr;
453c456f294SBarry Smith 
4543a40ed3dSBarry Smith   PetscFunctionBegin;
455ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
456ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
45744cd7ae7SLois Curfman McInnes   ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr);
4583a40ed3dSBarry Smith   PetscFunctionReturn(0);
4598965ea79SLois Curfman McInnes }
4608965ea79SLois Curfman McInnes 
4614a2ae208SSatish Balay #undef __FUNCT__
4624a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense"
463dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz)
4648965ea79SLois Curfman McInnes {
46539ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
466dfbe8321SBarry Smith   PetscErrorCode ierr;
467c456f294SBarry Smith 
4683a40ed3dSBarry Smith   PetscFunctionBegin;
469ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
470ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
47144cd7ae7SLois Curfman McInnes   ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr);
4723a40ed3dSBarry Smith   PetscFunctionReturn(0);
4738965ea79SLois Curfman McInnes }
4748965ea79SLois Curfman McInnes 
4754a2ae208SSatish Balay #undef __FUNCT__
4764a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense"
477dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy)
478096963f5SLois Curfman McInnes {
479096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
480dfbe8321SBarry Smith   PetscErrorCode ierr;
48187828ca2SBarry Smith   PetscScalar    zero = 0.0;
482096963f5SLois Curfman McInnes 
4833a40ed3dSBarry Smith   PetscFunctionBegin;
4842dcb1b2aSMatthew Knepley   ierr = VecSet(yy,zero);CHKERRQ(ierr);
4857c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
486ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
487ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
4883a40ed3dSBarry Smith   PetscFunctionReturn(0);
489096963f5SLois Curfman McInnes }
490096963f5SLois Curfman McInnes 
4914a2ae208SSatish Balay #undef __FUNCT__
4924a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense"
493dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz)
494096963f5SLois Curfman McInnes {
495096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
496dfbe8321SBarry Smith   PetscErrorCode ierr;
497096963f5SLois Curfman McInnes 
4983a40ed3dSBarry Smith   PetscFunctionBegin;
4993501a2bdSLois Curfman McInnes   ierr = VecCopy(yy,zz);CHKERRQ(ierr);
5007c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
501ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
502ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
5033a40ed3dSBarry Smith   PetscFunctionReturn(0);
504096963f5SLois Curfman McInnes }
505096963f5SLois Curfman McInnes 
5064a2ae208SSatish Balay #undef __FUNCT__
5074a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense"
508dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v)
5098965ea79SLois Curfman McInnes {
51039ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
511096963f5SLois Curfman McInnes   Mat_SeqDense   *aloc = (Mat_SeqDense*)a->A->data;
512dfbe8321SBarry Smith   PetscErrorCode ierr;
513d0f46423SBarry Smith   PetscInt       len,i,n,m = A->rmap->n,radd;
51487828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
515ed3cc1f0SBarry Smith 
5163a40ed3dSBarry Smith   PetscFunctionBegin;
5172dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
5181ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
519096963f5SLois Curfman McInnes   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
520d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
521d0f46423SBarry Smith   len  = PetscMin(a->A->rmap->n,a->A->cmap->n);
522d0f46423SBarry Smith   radd = A->rmap->rstart*m;
52344cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
524096963f5SLois Curfman McInnes     x[i] = aloc->v[radd + i*m + i];
525096963f5SLois Curfman McInnes   }
5261ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
5273a40ed3dSBarry Smith   PetscFunctionReturn(0);
5288965ea79SLois Curfman McInnes }
5298965ea79SLois Curfman McInnes 
5304a2ae208SSatish Balay #undef __FUNCT__
5314a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense"
532dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat)
5338965ea79SLois Curfman McInnes {
5343501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
535dfbe8321SBarry Smith   PetscErrorCode ierr;
53601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
53701b82886SBarry Smith   Mat_Plapack   *lu=(Mat_Plapack*)(mat->spptr);
53801b82886SBarry Smith #endif
539ed3cc1f0SBarry Smith 
5403a40ed3dSBarry Smith   PetscFunctionBegin;
54194d884c6SBarry Smith 
542aa482453SBarry Smith #if defined(PETSC_USE_LOG)
543d0f46423SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N);
5448965ea79SLois Curfman McInnes #endif
5458798bf22SSatish Balay   ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr);
5463501a2bdSLois Curfman McInnes   ierr = MatDestroy(mdn->A);CHKERRQ(ierr);
54705b42c5fSBarry Smith   if (mdn->lvec)   {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);}
54805b42c5fSBarry Smith   if (mdn->Mvctx)  {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);}
54901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
5502fbe02b9SBarry Smith   if (lu) {
55162b4c0b3SBarry Smith     ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr);
55262b4c0b3SBarry Smith     ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr);
55362b4c0b3SBarry Smith     ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr);
55409d27a7eSBarry Smith 
5552fbe02b9SBarry Smith     if (lu->is_pla) {
55601b82886SBarry Smith       ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr);
55701b82886SBarry Smith       ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr);
55801b82886SBarry Smith       ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr);
559622d7880SLois Curfman McInnes     }
5602fbe02b9SBarry Smith   }
56101b82886SBarry Smith #endif
56201b82886SBarry Smith 
563606d414cSSatish Balay   ierr = PetscFree(mdn);CHKERRQ(ierr);
564dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
565901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr);
566901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
5674ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5684ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5694ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5703a40ed3dSBarry Smith   PetscFunctionReturn(0);
5718965ea79SLois Curfman McInnes }
57239ddd567SLois Curfman McInnes 
5734a2ae208SSatish Balay #undef __FUNCT__
5744a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary"
5756849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer)
5768965ea79SLois Curfman McInnes {
57739ddd567SLois Curfman McInnes   Mat_MPIDense      *mdn = (Mat_MPIDense*)mat->data;
578dfbe8321SBarry Smith   PetscErrorCode    ierr;
579aa05aa95SBarry Smith   PetscViewerFormat format;
580aa05aa95SBarry Smith   int               fd;
581d0f46423SBarry Smith   PetscInt          header[4],mmax,N = mat->cmap->N,i,j,m,k;
582aa05aa95SBarry Smith   PetscMPIInt       rank,tag  = ((PetscObject)viewer)->tag,size;
583578230a0SSatish Balay   PetscScalar       *work,*v,*vv;
584aa05aa95SBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)mdn->A->data;
585aa05aa95SBarry Smith   MPI_Status        status;
5867056b6fcSBarry Smith 
5873a40ed3dSBarry Smith   PetscFunctionBegin;
58839ddd567SLois Curfman McInnes   if (mdn->size == 1) {
58939ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
590aa05aa95SBarry Smith   } else {
591aa05aa95SBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
5927adad957SLisandro Dalcin     ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr);
5937adad957SLisandro Dalcin     ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr);
594aa05aa95SBarry Smith 
595aa05aa95SBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
5967973ad04SBarry Smith     if (format == PETSC_VIEWER_NATIVE) {
597aa05aa95SBarry Smith 
598aa05aa95SBarry Smith       if (!rank) {
599aa05aa95SBarry Smith         /* store the matrix as a dense matrix */
600aa05aa95SBarry Smith         header[0] = MAT_FILE_COOKIE;
601d0f46423SBarry Smith         header[1] = mat->rmap->N;
602aa05aa95SBarry Smith         header[2] = N;
603aa05aa95SBarry Smith         header[3] = MATRIX_BINARY_FORMAT_DENSE;
604aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
605aa05aa95SBarry Smith 
606aa05aa95SBarry Smith         /* get largest work array needed for transposing array */
607d0f46423SBarry Smith         mmax = mat->rmap->n;
608aa05aa95SBarry Smith         for (i=1; i<size; i++) {
609d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
6108965ea79SLois Curfman McInnes         }
611aa05aa95SBarry Smith         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr);
612aa05aa95SBarry Smith 
613aa05aa95SBarry Smith         /* write out local array, by rows */
614d0f46423SBarry Smith         m    = mat->rmap->n;
615aa05aa95SBarry Smith         v    = a->v;
616aa05aa95SBarry Smith         for (j=0; j<N; j++) {
617aa05aa95SBarry Smith           for (i=0; i<m; i++) {
618578230a0SSatish Balay             work[j + i*N] = *v++;
619aa05aa95SBarry Smith           }
620aa05aa95SBarry Smith         }
621aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
622aa05aa95SBarry Smith         /* get largest work array to receive messages from other processes, excludes process zero */
623aa05aa95SBarry Smith         mmax = 0;
624aa05aa95SBarry Smith         for (i=1; i<size; i++) {
625d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
626aa05aa95SBarry Smith         }
627578230a0SSatish Balay         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr);
628aa05aa95SBarry Smith         for(k = 1; k < size; k++) {
629f8009846SMatthew Knepley           v    = vv;
630d0f46423SBarry Smith           m    = mat->rmap->range[k+1] - mat->rmap->range[k];
6317adad957SLisandro Dalcin           ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr);
632aa05aa95SBarry Smith 
633aa05aa95SBarry Smith           for(j = 0; j < N; j++) {
634aa05aa95SBarry Smith             for(i = 0; i < m; i++) {
635578230a0SSatish Balay               work[j + i*N] = *v++;
636aa05aa95SBarry Smith             }
637aa05aa95SBarry Smith           }
638aa05aa95SBarry Smith           ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
639aa05aa95SBarry Smith         }
640aa05aa95SBarry Smith         ierr = PetscFree(work);CHKERRQ(ierr);
641578230a0SSatish Balay         ierr = PetscFree(vv);CHKERRQ(ierr);
642aa05aa95SBarry Smith       } else {
643d0f46423SBarry Smith         ierr = MPI_Send(a->v,mat->rmap->n*mat->cmap->N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr);
644aa05aa95SBarry Smith       }
6454aa34b0aSBarry Smith     } else {
6467973ad04SBarry Smith       SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_NATIVE");
647aa05aa95SBarry Smith     }
648aa05aa95SBarry Smith   }
6493a40ed3dSBarry Smith   PetscFunctionReturn(0);
6508965ea79SLois Curfman McInnes }
6518965ea79SLois Curfman McInnes 
6524a2ae208SSatish Balay #undef __FUNCT__
6534a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket"
6546849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer)
6558965ea79SLois Curfman McInnes {
65639ddd567SLois Curfman McInnes   Mat_MPIDense          *mdn = (Mat_MPIDense*)mat->data;
657dfbe8321SBarry Smith   PetscErrorCode        ierr;
65832dcc486SBarry Smith   PetscMPIInt           size = mdn->size,rank = mdn->rank;
659a313700dSBarry Smith   const PetscViewerType vtype;
66032077d6dSBarry Smith   PetscTruth            iascii,isdraw;
661b0a32e0cSBarry Smith   PetscViewer           sviewer;
662f3ef73ceSBarry Smith   PetscViewerFormat     format;
66301b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
6645d00a290SHong Zhang   Mat_Plapack           *lu;
66501b82886SBarry Smith #endif
6668965ea79SLois Curfman McInnes 
6673a40ed3dSBarry Smith   PetscFunctionBegin;
66832077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
669fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
67032077d6dSBarry Smith   if (iascii) {
671b0a32e0cSBarry Smith     ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr);
672b0a32e0cSBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
673456192e2SBarry Smith     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
6744e220ebcSLois Curfman McInnes       MatInfo info;
675888f2ed8SSatish Balay       ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr);
676d0f46423SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap->n,
67777431f27SBarry Smith                    (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr);
678b0a32e0cSBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
67901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
68001b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr);
6815d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Processor mesh: nprows %d, npcols %d\n",Plapack_nprows, Plapack_npcols);CHKERRQ(ierr);
6825d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Error checking: %d\n",Plapack_ierror);CHKERRQ(ierr);
6835d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Algorithmic block size: %d\n",Plapack_nb_alg);CHKERRQ(ierr);
6845d00a290SHong Zhang       if (mat->factor){
6855d00a290SHong Zhang         lu=(Mat_Plapack*)(mat->spptr);
68601b82886SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr);
6875d00a290SHong Zhang       }
6885d00a290SHong Zhang #else
6895d00a290SHong Zhang       ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr);
69001b82886SBarry Smith #endif
6913a40ed3dSBarry Smith       PetscFunctionReturn(0);
692fb9695e5SSatish Balay     } else if (format == PETSC_VIEWER_ASCII_INFO) {
6933a40ed3dSBarry Smith       PetscFunctionReturn(0);
6948965ea79SLois Curfman McInnes     }
695f1af5d2fSBarry Smith   } else if (isdraw) {
696b0a32e0cSBarry Smith     PetscDraw  draw;
697f1af5d2fSBarry Smith     PetscTruth isnull;
698f1af5d2fSBarry Smith 
699b0a32e0cSBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
700b0a32e0cSBarry Smith     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
701f1af5d2fSBarry Smith     if (isnull) PetscFunctionReturn(0);
702f1af5d2fSBarry Smith   }
70377ed5343SBarry Smith 
7048965ea79SLois Curfman McInnes   if (size == 1) {
70539ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
7063a40ed3dSBarry Smith   } else {
7078965ea79SLois Curfman McInnes     /* assemble the entire matrix onto first processor. */
7088965ea79SLois Curfman McInnes     Mat         A;
709d0f46423SBarry Smith     PetscInt    M = mat->rmap->N,N = mat->cmap->N,m,row,i,nz;
710ba8c8a56SBarry Smith     PetscInt    *cols;
711ba8c8a56SBarry Smith     PetscScalar *vals;
7128965ea79SLois Curfman McInnes 
7137adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr);
7148965ea79SLois Curfman McInnes     if (!rank) {
715f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr);
7163a40ed3dSBarry Smith     } else {
717f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr);
7188965ea79SLois Curfman McInnes     }
7197adad957SLisandro Dalcin     /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */
720878740d9SKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
721878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL);
72252e6d16bSBarry Smith     ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr);
7238965ea79SLois Curfman McInnes 
72439ddd567SLois Curfman McInnes     /* Copy the matrix ... This isn't the most efficient means,
72539ddd567SLois Curfman McInnes        but it's quick for now */
72651022da4SBarry Smith     A->insertmode = INSERT_VALUES;
727d0f46423SBarry Smith     row = mat->rmap->rstart; m = mdn->A->rmap->n;
7288965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
729ba8c8a56SBarry Smith       ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
730ba8c8a56SBarry Smith       ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
731ba8c8a56SBarry Smith       ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
73239ddd567SLois Curfman McInnes       row++;
7338965ea79SLois Curfman McInnes     }
7348965ea79SLois Curfman McInnes 
7356d4a8577SBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7366d4a8577SBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
737b0a32e0cSBarry Smith     ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr);
738b9b97703SBarry Smith     if (!rank) {
7396831982aSBarry Smith       ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr);
7408965ea79SLois Curfman McInnes     }
741b0a32e0cSBarry Smith     ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr);
742b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
7438965ea79SLois Curfman McInnes     ierr = MatDestroy(A);CHKERRQ(ierr);
7448965ea79SLois Curfman McInnes   }
7453a40ed3dSBarry Smith   PetscFunctionReturn(0);
7468965ea79SLois Curfman McInnes }
7478965ea79SLois Curfman McInnes 
7484a2ae208SSatish Balay #undef __FUNCT__
7494a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense"
750dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer)
7518965ea79SLois Curfman McInnes {
752dfbe8321SBarry Smith   PetscErrorCode ierr;
75332077d6dSBarry Smith   PetscTruth     iascii,isbinary,isdraw,issocket;
7548965ea79SLois Curfman McInnes 
755433994e6SBarry Smith   PetscFunctionBegin;
7560f5bd95cSBarry Smith 
75732077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
758fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
759b0a32e0cSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr);
760fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
7610f5bd95cSBarry Smith 
76232077d6dSBarry Smith   if (iascii || issocket || isdraw) {
763f1af5d2fSBarry Smith     ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr);
7640f5bd95cSBarry Smith   } else if (isbinary) {
7653a40ed3dSBarry Smith     ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr);
7665cd90555SBarry Smith   } else {
767958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name);
7688965ea79SLois Curfman McInnes   }
7693a40ed3dSBarry Smith   PetscFunctionReturn(0);
7708965ea79SLois Curfman McInnes }
7718965ea79SLois Curfman McInnes 
7724a2ae208SSatish Balay #undef __FUNCT__
7734a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense"
774dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info)
7758965ea79SLois Curfman McInnes {
7763501a2bdSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
7773501a2bdSLois Curfman McInnes   Mat            mdn = mat->A;
778dfbe8321SBarry Smith   PetscErrorCode ierr;
779329f5518SBarry Smith   PetscReal      isend[5],irecv[5];
7808965ea79SLois Curfman McInnes 
7813a40ed3dSBarry Smith   PetscFunctionBegin;
7824e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
7834e220ebcSLois Curfman McInnes   ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr);
7844e220ebcSLois Curfman McInnes   isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded;
7854e220ebcSLois Curfman McInnes   isend[3] = info->memory;  isend[4] = info->mallocs;
7868965ea79SLois Curfman McInnes   if (flag == MAT_LOCAL) {
7874e220ebcSLois Curfman McInnes     info->nz_used      = isend[0];
7884e220ebcSLois Curfman McInnes     info->nz_allocated = isend[1];
7894e220ebcSLois Curfman McInnes     info->nz_unneeded  = isend[2];
7904e220ebcSLois Curfman McInnes     info->memory       = isend[3];
7914e220ebcSLois Curfman McInnes     info->mallocs      = isend[4];
7928965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_MAX) {
7937adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
7944e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7954e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
7964e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
7974e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
7984e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
7998965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_SUM) {
8007adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
8014e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
8024e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
8034e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
8044e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
8054e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
8068965ea79SLois Curfman McInnes   }
8074e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0; /* no parallel LU/ILU/Cholesky */
8084e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
8094e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
8103a40ed3dSBarry Smith   PetscFunctionReturn(0);
8118965ea79SLois Curfman McInnes }
8128965ea79SLois Curfman McInnes 
8134a2ae208SSatish Balay #undef __FUNCT__
8144a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense"
8154e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg)
8168965ea79SLois Curfman McInnes {
81739ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
818dfbe8321SBarry Smith   PetscErrorCode ierr;
8198965ea79SLois Curfman McInnes 
8203a40ed3dSBarry Smith   PetscFunctionBegin;
82112c028f9SKris Buschelman   switch (op) {
822512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
82312c028f9SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
82412c028f9SKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
8254e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82612c028f9SKris Buschelman     break;
82712c028f9SKris Buschelman   case MAT_ROW_ORIENTED:
8284e0d8c25SBarry Smith     a->roworiented = flg;
8294e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
83012c028f9SKris Buschelman     break;
8314e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
83212c028f9SKris Buschelman   case MAT_USE_HASH_TABLE:
833290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
83412c028f9SKris Buschelman     break;
83512c028f9SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
8364e0d8c25SBarry Smith     a->donotstash = flg;
83712c028f9SKris Buschelman     break;
83877e54ba9SKris Buschelman   case MAT_SYMMETRIC:
83977e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
8409a4540c5SBarry Smith   case MAT_HERMITIAN:
8419a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
842600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
843290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
84477e54ba9SKris Buschelman     break;
84512c028f9SKris Buschelman   default:
846600fe468SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
8473a40ed3dSBarry Smith   }
8483a40ed3dSBarry Smith   PetscFunctionReturn(0);
8498965ea79SLois Curfman McInnes }
8508965ea79SLois Curfman McInnes 
8518965ea79SLois Curfman McInnes 
8524a2ae208SSatish Balay #undef __FUNCT__
8534a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense"
854dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr)
8555b2fa520SLois Curfman McInnes {
8565b2fa520SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8575b2fa520SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
85887828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
859dfbe8321SBarry Smith   PetscErrorCode ierr;
860d0f46423SBarry Smith   PetscInt       i,j,s2a,s3a,s2,s3,m=mdn->A->rmap->n,n=mdn->A->cmap->n;
8615b2fa520SLois Curfman McInnes 
8625b2fa520SLois Curfman McInnes   PetscFunctionBegin;
86372d926a5SLois Curfman McInnes   ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr);
8645b2fa520SLois Curfman McInnes   if (ll) {
86572d926a5SLois Curfman McInnes     ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr);
866175be7b4SMatthew Knepley     if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2);
8671ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
8685b2fa520SLois Curfman McInnes     for (i=0; i<m; i++) {
8695b2fa520SLois Curfman McInnes       x = l[i];
8705b2fa520SLois Curfman McInnes       v = mat->v + i;
8715b2fa520SLois Curfman McInnes       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
8725b2fa520SLois Curfman McInnes     }
8731ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
874efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8755b2fa520SLois Curfman McInnes   }
8765b2fa520SLois Curfman McInnes   if (rr) {
877175be7b4SMatthew Knepley     ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr);
878175be7b4SMatthew Knepley     if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3);
879ca9f406cSSatish Balay     ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
880ca9f406cSSatish Balay     ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
8811ebc52fbSHong Zhang     ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr);
8825b2fa520SLois Curfman McInnes     for (i=0; i<n; i++) {
8835b2fa520SLois Curfman McInnes       x = r[i];
8845b2fa520SLois Curfman McInnes       v = mat->v + i*m;
8855b2fa520SLois Curfman McInnes       for (j=0; j<m; j++) { (*v++) *= x;}
8865b2fa520SLois Curfman McInnes     }
8871ebc52fbSHong Zhang     ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr);
888efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8895b2fa520SLois Curfman McInnes   }
8905b2fa520SLois Curfman McInnes   PetscFunctionReturn(0);
8915b2fa520SLois Curfman McInnes }
8925b2fa520SLois Curfman McInnes 
8934a2ae208SSatish Balay #undef __FUNCT__
8944a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense"
895dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm)
896096963f5SLois Curfman McInnes {
8973501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8983501a2bdSLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
899dfbe8321SBarry Smith   PetscErrorCode ierr;
90013f74950SBarry Smith   PetscInt       i,j;
901329f5518SBarry Smith   PetscReal      sum = 0.0;
90287828ca2SBarry Smith   PetscScalar    *v = mat->v;
9033501a2bdSLois Curfman McInnes 
9043a40ed3dSBarry Smith   PetscFunctionBegin;
9053501a2bdSLois Curfman McInnes   if (mdn->size == 1) {
906064f8208SBarry Smith     ierr =  MatNorm(mdn->A,type,nrm);CHKERRQ(ierr);
9073501a2bdSLois Curfman McInnes   } else {
9083501a2bdSLois Curfman McInnes     if (type == NORM_FROBENIUS) {
909d0f46423SBarry Smith       for (i=0; i<mdn->A->cmap->n*mdn->A->rmap->n; i++) {
910aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
911329f5518SBarry Smith         sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
9123501a2bdSLois Curfman McInnes #else
9133501a2bdSLois Curfman McInnes         sum += (*v)*(*v); v++;
9143501a2bdSLois Curfman McInnes #endif
9153501a2bdSLois Curfman McInnes       }
9167adad957SLisandro Dalcin       ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
917064f8208SBarry Smith       *nrm = sqrt(*nrm);
918dc0b31edSSatish Balay       ierr = PetscLogFlops(2.0*mdn->A->cmap->n*mdn->A->rmap->n);CHKERRQ(ierr);
9193a40ed3dSBarry Smith     } else if (type == NORM_1) {
920329f5518SBarry Smith       PetscReal *tmp,*tmp2;
921d0f46423SBarry Smith       ierr = PetscMalloc(2*A->cmap->N*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
922d0f46423SBarry Smith       tmp2 = tmp + A->cmap->N;
923d0f46423SBarry Smith       ierr = PetscMemzero(tmp,2*A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr);
924064f8208SBarry Smith       *nrm = 0.0;
9253501a2bdSLois Curfman McInnes       v = mat->v;
926d0f46423SBarry Smith       for (j=0; j<mdn->A->cmap->n; j++) {
927d0f46423SBarry Smith         for (i=0; i<mdn->A->rmap->n; i++) {
92867e560aaSBarry Smith           tmp[j] += PetscAbsScalar(*v);  v++;
9293501a2bdSLois Curfman McInnes         }
9303501a2bdSLois Curfman McInnes       }
931d0f46423SBarry Smith       ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
932d0f46423SBarry Smith       for (j=0; j<A->cmap->N; j++) {
933064f8208SBarry Smith         if (tmp2[j] > *nrm) *nrm = tmp2[j];
9343501a2bdSLois Curfman McInnes       }
935606d414cSSatish Balay       ierr = PetscFree(tmp);CHKERRQ(ierr);
936d0f46423SBarry Smith       ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
9373a40ed3dSBarry Smith     } else if (type == NORM_INFINITY) { /* max row norm */
938329f5518SBarry Smith       PetscReal ntemp;
9393501a2bdSLois Curfman McInnes       ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr);
9407adad957SLisandro Dalcin       ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
9413a40ed3dSBarry Smith     } else {
94229bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"No support for two norm");
9433501a2bdSLois Curfman McInnes     }
9443501a2bdSLois Curfman McInnes   }
9453a40ed3dSBarry Smith   PetscFunctionReturn(0);
9463501a2bdSLois Curfman McInnes }
9473501a2bdSLois Curfman McInnes 
9484a2ae208SSatish Balay #undef __FUNCT__
9494a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense"
950fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout)
9513501a2bdSLois Curfman McInnes {
9523501a2bdSLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
9533501a2bdSLois Curfman McInnes   Mat_SeqDense   *Aloc = (Mat_SeqDense*)a->A->data;
9543501a2bdSLois Curfman McInnes   Mat            B;
955d0f46423SBarry Smith   PetscInt       M = A->rmap->N,N = A->cmap->N,m,n,*rwork,rstart = A->rmap->rstart;
9566849ba73SBarry Smith   PetscErrorCode ierr;
95713f74950SBarry Smith   PetscInt       j,i;
95887828ca2SBarry Smith   PetscScalar    *v;
9593501a2bdSLois Curfman McInnes 
9603a40ed3dSBarry Smith   PetscFunctionBegin;
961e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place");
962fc4dec0aSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || A == *matout) {
9637adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
964d0f46423SBarry Smith     ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,N,M);CHKERRQ(ierr);
9657adad957SLisandro Dalcin     ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
966878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr);
967fc4dec0aSBarry Smith   } else {
968fc4dec0aSBarry Smith     B = *matout;
969fc4dec0aSBarry Smith   }
9703501a2bdSLois Curfman McInnes 
971d0f46423SBarry Smith   m = a->A->rmap->n; n = a->A->cmap->n; v = Aloc->v;
9721acff37aSSatish Balay   ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr);
9733501a2bdSLois Curfman McInnes   for (i=0; i<m; i++) rwork[i] = rstart + i;
9741acff37aSSatish Balay   for (j=0; j<n; j++) {
9753501a2bdSLois Curfman McInnes     ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr);
9763501a2bdSLois Curfman McInnes     v   += m;
9773501a2bdSLois Curfman McInnes   }
978606d414cSSatish Balay   ierr = PetscFree(rwork);CHKERRQ(ierr);
9796d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9806d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
981815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *matout != A) {
9823501a2bdSLois Curfman McInnes     *matout = B;
9833501a2bdSLois Curfman McInnes   } else {
984273d9f13SBarry Smith     ierr = MatHeaderCopy(A,B);CHKERRQ(ierr);
9853501a2bdSLois Curfman McInnes   }
9863a40ed3dSBarry Smith   PetscFunctionReturn(0);
987096963f5SLois Curfman McInnes }
988096963f5SLois Curfman McInnes 
989d9eff348SSatish Balay #include "petscblaslapack.h"
9904a2ae208SSatish Balay #undef __FUNCT__
9914a2ae208SSatish Balay #define __FUNCT__ "MatScale_MPIDense"
992f4df32b1SMatthew Knepley PetscErrorCode MatScale_MPIDense(Mat inA,PetscScalar alpha)
99344cd7ae7SLois Curfman McInnes {
99444cd7ae7SLois Curfman McInnes   Mat_MPIDense   *A = (Mat_MPIDense*)inA->data;
99544cd7ae7SLois Curfman McInnes   Mat_SeqDense   *a = (Mat_SeqDense*)A->A->data;
996f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
997efee365bSSatish Balay   PetscErrorCode ierr;
998d0f46423SBarry Smith   PetscBLASInt   one = 1,nz = PetscBLASIntCast(inA->rmap->n*inA->cmap->N);
99944cd7ae7SLois Curfman McInnes 
10003a40ed3dSBarry Smith   PetscFunctionBegin;
1001f4df32b1SMatthew Knepley   BLASscal_(&nz,&oalpha,a->v,&one);
1002efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
10033a40ed3dSBarry Smith   PetscFunctionReturn(0);
100444cd7ae7SLois Curfman McInnes }
100544cd7ae7SLois Curfman McInnes 
10066849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *);
10078965ea79SLois Curfman McInnes 
10084a2ae208SSatish Balay #undef __FUNCT__
10094a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_MPIDense"
1010dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_MPIDense(Mat A)
1011273d9f13SBarry Smith {
1012dfbe8321SBarry Smith   PetscErrorCode ierr;
1013273d9f13SBarry Smith 
1014273d9f13SBarry Smith   PetscFunctionBegin;
1015273d9f13SBarry Smith   ierr =  MatMPIDenseSetPreallocation(A,0);CHKERRQ(ierr);
1016273d9f13SBarry Smith   PetscFunctionReturn(0);
1017273d9f13SBarry Smith }
1018273d9f13SBarry Smith 
101901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1020eae6fb2eSBarry Smith 
1021eae6fb2eSBarry Smith #undef __FUNCT__
1022eae6fb2eSBarry Smith #define __FUNCT__ "MatMPIDenseCopyToPlapack"
1023eae6fb2eSBarry Smith PetscErrorCode MatMPIDenseCopyToPlapack(Mat A,Mat F)
1024eae6fb2eSBarry Smith {
1025eae6fb2eSBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
1026eae6fb2eSBarry Smith   PetscErrorCode ierr;
1027d0f46423SBarry Smith   PetscInt       M=A->cmap->N,m=A->rmap->n,rstart;
1028eae6fb2eSBarry Smith   PetscScalar    *array;
1029eae6fb2eSBarry Smith   PetscReal      one = 1.0;
1030eae6fb2eSBarry Smith 
1031eae6fb2eSBarry Smith   PetscFunctionBegin;
10322fbe02b9SBarry Smith   /* Copy A into F->lu->A */
1033eae6fb2eSBarry Smith   ierr = PLA_Obj_set_to_zero(lu->A);CHKERRQ(ierr);
1034eae6fb2eSBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
1035eae6fb2eSBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
103679b0a62dSBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
1037eae6fb2eSBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
1038eae6fb2eSBarry Smith   ierr = PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);CHKERRQ(ierr);
1039eae6fb2eSBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
1040eae6fb2eSBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
1041eae6fb2eSBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
1042eae6fb2eSBarry Smith   lu->rstart = rstart;
1043eae6fb2eSBarry Smith   PetscFunctionReturn(0);
1044eae6fb2eSBarry Smith }
1045eae6fb2eSBarry Smith 
104601b82886SBarry Smith #undef __FUNCT__
10472fbe02b9SBarry Smith #define __FUNCT__ "MatMPIDenseCopyFromPlapack"
10482fbe02b9SBarry Smith PetscErrorCode MatMPIDenseCopyFromPlapack(Mat F,Mat A)
10492fbe02b9SBarry Smith {
10502fbe02b9SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
10512fbe02b9SBarry Smith   PetscErrorCode ierr;
1052d0f46423SBarry Smith   PetscInt       M=A->cmap->N,m=A->rmap->n,rstart;
10532fbe02b9SBarry Smith   PetscScalar    *array;
10542fbe02b9SBarry Smith   PetscReal      one = 1.0;
10552fbe02b9SBarry Smith 
10562fbe02b9SBarry Smith   PetscFunctionBegin;
10572fbe02b9SBarry Smith   /* Copy F into A->lu->A */
105879b0a62dSBarry Smith   ierr = MatZeroEntries(A);CHKERRQ(ierr);
10592fbe02b9SBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
10602fbe02b9SBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
10612fbe02b9SBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
10622fbe02b9SBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
10632fbe02b9SBarry Smith   ierr = PLA_API_axpy_global_to_matrix(m,M, &one,lu->A,rstart,0,(void *)array,m);CHKERRQ(ierr);
10642fbe02b9SBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
10652fbe02b9SBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
10662fbe02b9SBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
10672fbe02b9SBarry Smith   lu->rstart = rstart;
10682fbe02b9SBarry Smith   PetscFunctionReturn(0);
10692fbe02b9SBarry Smith }
10702fbe02b9SBarry Smith 
10712fbe02b9SBarry Smith #undef __FUNCT__
10722fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultNumeric_MPIDense_MPIDense"
10732fbe02b9SBarry Smith PetscErrorCode MatMatMultNumeric_MPIDense_MPIDense(Mat A,Mat B,Mat C)
10742fbe02b9SBarry Smith {
10752fbe02b9SBarry Smith   PetscErrorCode ierr;
10762fbe02b9SBarry Smith   Mat_Plapack    *luA = (Mat_Plapack*)A->spptr;
10772fbe02b9SBarry Smith   Mat_Plapack    *luB = (Mat_Plapack*)B->spptr;
10782fbe02b9SBarry Smith   Mat_Plapack    *luC = (Mat_Plapack*)C->spptr;
10792fbe02b9SBarry Smith   PLA_Obj        alpha = NULL,beta = NULL;
10802fbe02b9SBarry Smith 
10812fbe02b9SBarry Smith   PetscFunctionBegin;
10822fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(A,A);CHKERRQ(ierr);
10832fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(B,B);CHKERRQ(ierr);
10842fbe02b9SBarry Smith 
10856e6f9017SBarry Smith   /*
1086cb2480eaSBarry Smith   ierr = PLA_Global_show("A = ",luA->A,"%g ","");CHKERRQ(ierr);
1087cb2480eaSBarry Smith   ierr = PLA_Global_show("B = ",luB->A,"%g ","");CHKERRQ(ierr);
10886e6f9017SBarry Smith   */
1089cb2480eaSBarry Smith 
10902fbe02b9SBarry Smith   /* do the multiply in PLA  */
109179b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luA->A,NULL,NULL,&alpha);CHKERRQ(ierr);
109279b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luC->A,NULL,&beta,NULL);CHKERRQ(ierr);
109379b0a62dSBarry Smith   CHKMEMQ;
10942fbe02b9SBarry Smith 
1095f0e3846dSSatish Balay   ierr = PLA_Gemm(PLA_NO_TRANSPOSE,PLA_NO_TRANSPOSE,alpha,luA->A,luB->A,beta,luC->A); /* CHKERRQ(ierr); */
109679b0a62dSBarry Smith   CHKMEMQ;
10972fbe02b9SBarry Smith   ierr = PLA_Obj_free(&alpha);CHKERRQ(ierr);
10982fbe02b9SBarry Smith   ierr = PLA_Obj_free(&beta);CHKERRQ(ierr);
10992fbe02b9SBarry Smith 
11006e6f9017SBarry Smith   /*
1101cb2480eaSBarry Smith   ierr = PLA_Global_show("C = ",luC->A,"%g ","");CHKERRQ(ierr);
11026e6f9017SBarry Smith   */
11032fbe02b9SBarry Smith   ierr = MatMPIDenseCopyFromPlapack(C,C);CHKERRQ(ierr);
11042fbe02b9SBarry Smith   PetscFunctionReturn(0);
11052fbe02b9SBarry Smith }
11062fbe02b9SBarry Smith 
11072fbe02b9SBarry Smith #undef __FUNCT__
11082fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultSymbolic_MPIDense_MPIDense"
11092fbe02b9SBarry Smith PetscErrorCode MatMatMultSymbolic_MPIDense_MPIDense(Mat A,Mat B,PetscReal fill,Mat *C)
11102fbe02b9SBarry Smith {
11112fbe02b9SBarry Smith   PetscErrorCode ierr;
1112d0f46423SBarry Smith   PetscInt       m=A->rmap->n,n=B->cmap->n;
11132fbe02b9SBarry Smith   Mat            Cmat;
11142fbe02b9SBarry Smith 
11152fbe02b9SBarry Smith   PetscFunctionBegin;
1116d0f46423SBarry 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);
11176e6f9017SBarry Smith   SETERRQ(PETSC_ERR_LIB,"Due to aparent bugs in PLAPACK,this is not currently supported");
11182fbe02b9SBarry Smith   ierr = MatCreate(((PetscObject)B)->comm,&Cmat);CHKERRQ(ierr);
1119d0f46423SBarry Smith   ierr = MatSetSizes(Cmat,m,n,A->rmap->N,B->cmap->N);CHKERRQ(ierr);
11202fbe02b9SBarry Smith   ierr = MatSetType(Cmat,MATMPIDENSE);CHKERRQ(ierr);
11212fbe02b9SBarry Smith   ierr = MatAssemblyBegin(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11222fbe02b9SBarry Smith   ierr = MatAssemblyEnd(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
112385bd4cefSHong Zhang 
11242fbe02b9SBarry Smith   *C = Cmat;
11252fbe02b9SBarry Smith   PetscFunctionReturn(0);
11262fbe02b9SBarry Smith }
11272fbe02b9SBarry Smith 
11282fbe02b9SBarry Smith #undef __FUNCT__
11292fbe02b9SBarry Smith #define __FUNCT__ "MatMatMult_MPIDense_MPIDense"
11302fbe02b9SBarry Smith PetscErrorCode MatMatMult_MPIDense_MPIDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
11312fbe02b9SBarry Smith {
11322fbe02b9SBarry Smith   PetscErrorCode ierr;
11332fbe02b9SBarry Smith 
11342fbe02b9SBarry Smith   PetscFunctionBegin;
11352fbe02b9SBarry Smith   if (scall == MAT_INITIAL_MATRIX){
11362fbe02b9SBarry Smith     ierr = MatMatMultSymbolic_MPIDense_MPIDense(A,B,fill,C);CHKERRQ(ierr);
11372fbe02b9SBarry Smith   }
11382fbe02b9SBarry Smith   ierr = MatMatMultNumeric_MPIDense_MPIDense(A,B,*C);CHKERRQ(ierr);
11392fbe02b9SBarry Smith   PetscFunctionReturn(0);
11402fbe02b9SBarry Smith }
11412fbe02b9SBarry Smith 
1142ab235cb6SHong Zhang #undef __FUNCT__
11436c995c7dSHong Zhang #define __FUNCT__ "MatSolve_MPIDense"
11446c995c7dSHong Zhang PetscErrorCode MatSolve_MPIDense(Mat A,Vec b,Vec x)
11456c995c7dSHong Zhang {
11466c995c7dSHong Zhang   MPI_Comm       comm = ((PetscObject)A)->comm;
11476c995c7dSHong Zhang   Mat_Plapack    *lu = (Mat_Plapack*)A->spptr;
11486c995c7dSHong Zhang   PetscErrorCode ierr;
1149d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,i,j,*idx_pla,*idx_petsc,loc_m,loc_stride;
11506c995c7dSHong Zhang   PetscScalar    *array;
11516c995c7dSHong Zhang   PetscReal      one = 1.0;
11526c995c7dSHong Zhang   PetscMPIInt    size,rank,r_rank,r_nproc,c_rank,c_nproc;;
11536c995c7dSHong Zhang   PLA_Obj        v_pla = NULL;
11546c995c7dSHong Zhang   PetscScalar    *loc_buf;
11556c995c7dSHong Zhang   Vec            loc_x;
11566c995c7dSHong Zhang 
11576c995c7dSHong Zhang   PetscFunctionBegin;
11586c995c7dSHong Zhang   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
11596c995c7dSHong Zhang   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
11606c995c7dSHong Zhang 
11616c995c7dSHong Zhang   /* Create PLAPACK vector objects, then copy b into PLAPACK b */
11626c995c7dSHong Zhang   PLA_Mvector_create(lu->datatype,M,1,lu->templ,PLA_ALIGN_FIRST,&v_pla);
11636c995c7dSHong Zhang   PLA_Obj_set_to_zero(v_pla);
11646c995c7dSHong Zhang 
11656c995c7dSHong Zhang   /* Copy b into rhs_pla */
11666c995c7dSHong Zhang   PLA_API_begin();
11676c995c7dSHong Zhang   PLA_Obj_API_open(v_pla);
11686c995c7dSHong Zhang   ierr = VecGetArray(b,&array);CHKERRQ(ierr);
11696c995c7dSHong Zhang   PLA_API_axpy_vector_to_global(m,&one,(void *)array,1,v_pla,lu->rstart);
11706c995c7dSHong Zhang   ierr = VecRestoreArray(b,&array);CHKERRQ(ierr);
11716c995c7dSHong Zhang   PLA_Obj_API_close(v_pla);
11726c995c7dSHong Zhang   PLA_API_end();
11736c995c7dSHong Zhang 
11746c995c7dSHong Zhang   if (A->factor == MAT_FACTOR_LU){
11756c995c7dSHong Zhang     /* Apply the permutations to the right hand sides */
11766c995c7dSHong Zhang     PLA_Apply_pivots_to_rows (v_pla,lu->pivots);
11776c995c7dSHong Zhang 
11786c995c7dSHong Zhang     /* Solve L y = b, overwriting b with y */
11796c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_UNIT_DIAG,lu->A,v_pla );
11806c995c7dSHong Zhang 
11816c995c7dSHong Zhang     /* Solve U x = y (=b), overwriting b with x */
11826c995c7dSHong Zhang     PLA_Trsv( PLA_UPPER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla );
11836c995c7dSHong Zhang   } else { /* MAT_FACTOR_CHOLESKY */
11846c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla);
11856c995c7dSHong Zhang     PLA_Trsv( PLA_LOWER_TRIANGULAR,(lu->datatype == MPI_DOUBLE ? PLA_TRANSPOSE : PLA_CONJUGATE_TRANSPOSE),
11866c995c7dSHong Zhang                                     PLA_NONUNIT_DIAG,lu->A,v_pla);
11876c995c7dSHong Zhang   }
11886c995c7dSHong Zhang 
11896c995c7dSHong Zhang   /* Copy PLAPACK x into Petsc vector x  */
11906c995c7dSHong Zhang   PLA_Obj_local_length(v_pla, &loc_m);
11916c995c7dSHong Zhang   PLA_Obj_local_buffer(v_pla, (void**)&loc_buf);
11926c995c7dSHong Zhang   PLA_Obj_local_stride(v_pla, &loc_stride);
11936c995c7dSHong Zhang   /*
11946c995c7dSHong 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);
11956c995c7dSHong Zhang   */
11966c995c7dSHong Zhang   ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,loc_m*loc_stride,loc_buf,&loc_x);CHKERRQ(ierr);
11976c995c7dSHong Zhang   if (!lu->pla_solved){
11986c995c7dSHong Zhang 
11995d00a290SHong Zhang     PLA_Temp_comm_row_info(lu->templ,&Plapack_comm_2d,&r_rank,&r_nproc);
12005d00a290SHong Zhang     PLA_Temp_comm_col_info(lu->templ,&Plapack_comm_2d,&c_rank,&c_nproc);
12016c995c7dSHong Zhang 
12026c995c7dSHong Zhang     /* Create IS and cts for VecScatterring */
12036c995c7dSHong Zhang     PLA_Obj_local_length(v_pla, &loc_m);
12046c995c7dSHong Zhang     PLA_Obj_local_stride(v_pla, &loc_stride);
12056c995c7dSHong Zhang     ierr = PetscMalloc((2*loc_m+1)*sizeof(PetscInt),&idx_pla);CHKERRQ(ierr);
12066c995c7dSHong Zhang     idx_petsc = idx_pla + loc_m;
12076c995c7dSHong Zhang 
12086c995c7dSHong Zhang     rstart = (r_rank*c_nproc+c_rank)*lu->nb;
12096c995c7dSHong Zhang     for (i=0; i<loc_m; i+=lu->nb){
12106c995c7dSHong Zhang       j = 0;
12116c995c7dSHong Zhang       while (j < lu->nb && i+j < loc_m){
12126c995c7dSHong Zhang         idx_petsc[i+j] = rstart + j; j++;
12136c995c7dSHong Zhang       }
12146c995c7dSHong Zhang       rstart += size*lu->nb;
12156c995c7dSHong Zhang     }
12166c995c7dSHong Zhang 
12176c995c7dSHong Zhang     for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride;
12186c995c7dSHong Zhang 
12196c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr);
12206c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr);
12216c995c7dSHong Zhang     ierr = PetscFree(idx_pla);CHKERRQ(ierr);
12226c995c7dSHong Zhang     ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr);
12236c995c7dSHong Zhang   }
12246c995c7dSHong Zhang   ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12256c995c7dSHong Zhang   ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12266c995c7dSHong Zhang 
12276c995c7dSHong Zhang   /* Free data */
12286c995c7dSHong Zhang   ierr = VecDestroy(loc_x);CHKERRQ(ierr);
12296c995c7dSHong Zhang   PLA_Obj_free(&v_pla);
12306c995c7dSHong Zhang 
12316c995c7dSHong Zhang   lu->pla_solved = PETSC_TRUE;
12326c995c7dSHong Zhang   PetscFunctionReturn(0);
12336c995c7dSHong Zhang }
12346c995c7dSHong Zhang 
12356c995c7dSHong Zhang #undef __FUNCT__
12366c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense"
12370481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info)
12386c995c7dSHong Zhang {
1239719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
12406c995c7dSHong Zhang   PetscErrorCode ierr;
1241d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12426c995c7dSHong Zhang   PetscInt       info_pla=0;
12436c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12446c995c7dSHong Zhang 
12456c995c7dSHong Zhang   PetscFunctionBegin;
12466c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12476c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12486c995c7dSHong Zhang     PLA_Obj_free (&lu->pivots);
12496c995c7dSHong Zhang   }
12506c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12516c995c7dSHong Zhang   lu->A = NULL; lu->pivots = NULL;
12526c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12536c995c7dSHong Zhang   PLA_Obj_set_to_zero(lu->A);
12546c995c7dSHong Zhang   PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);
12556c995c7dSHong Zhang 
12566c995c7dSHong Zhang   /* Copy A into lu->A */
12576c995c7dSHong Zhang   PLA_API_begin();
12586c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
12596c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
12606c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
12616c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
12626c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
12636c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
12646c995c7dSHong Zhang   PLA_API_end();
12656c995c7dSHong Zhang 
12666c995c7dSHong Zhang   /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */
12676c995c7dSHong Zhang   info_pla = PLA_LU(lu->A,lu->pivots);
12686c995c7dSHong Zhang   if (info_pla != 0)
12696c995c7dSHong Zhang     SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla);
12706c995c7dSHong Zhang 
12716c995c7dSHong Zhang   lu->rstart         = rstart;
12726c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1273719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1274719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
12756c995c7dSHong Zhang   PetscFunctionReturn(0);
12766c995c7dSHong Zhang }
12776c995c7dSHong Zhang 
12786c995c7dSHong Zhang #undef __FUNCT__
12796c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense"
12800481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info)
12816c995c7dSHong Zhang {
1282719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)F->spptr;
12836c995c7dSHong Zhang   PetscErrorCode ierr;
1284d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12856c995c7dSHong Zhang   PetscInt       info_pla=0;
12866c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12876c995c7dSHong Zhang 
12886c995c7dSHong Zhang   PetscFunctionBegin;
12896c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12906c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12916c995c7dSHong Zhang   }
12926c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12936c995c7dSHong Zhang   lu->A      = NULL;
12946c995c7dSHong Zhang   lu->pivots = NULL;
12956c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12966c995c7dSHong Zhang 
12976c995c7dSHong Zhang   /* Copy A into lu->A */
12986c995c7dSHong Zhang   PLA_API_begin();
12996c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
13006c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
13016c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
13026c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
13036c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
13046c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
13056c995c7dSHong Zhang   PLA_API_end();
13066c995c7dSHong Zhang 
13076c995c7dSHong Zhang   /* Factor P A -> Chol */
13086c995c7dSHong Zhang   info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A);
13096c995c7dSHong Zhang   if (info_pla != 0)
13106c995c7dSHong Zhang     SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla);
13116c995c7dSHong Zhang 
13126c995c7dSHong Zhang   lu->rstart         = rstart;
13136c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1314719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1315719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
13166c995c7dSHong Zhang   PetscFunctionReturn(0);
13176c995c7dSHong Zhang }
13186c995c7dSHong Zhang 
1319b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */
13202fbe02b9SBarry Smith #undef __FUNCT__
13216c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense"
13220481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,const MatFactorInfo *info)
132301b82886SBarry Smith {
132401b82886SBarry Smith   PetscErrorCode ierr;
1325b24902e0SBarry Smith   PetscTruth     issymmetric,set;
132601b82886SBarry Smith 
132701b82886SBarry Smith   PetscFunctionBegin;
1328b24902e0SBarry Smith   ierr = MatIsSymmetricKnown(A,&set,&issymmetric);CHKERRQ(ierr);
1329b24902e0SBarry Smith   if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()");
1330719d5645SBarry Smith   F->ops->choleskyfactornumeric  = MatCholeskyFactorNumeric_MPIDense;
1331b24902e0SBarry Smith   PetscFunctionReturn(0);
133201b82886SBarry Smith }
133301b82886SBarry Smith 
1334b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */
1335b24902e0SBarry Smith #undef __FUNCT__
13366c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense"
13370481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
1338b24902e0SBarry Smith {
1339b24902e0SBarry Smith   PetscErrorCode ierr;
1340d0f46423SBarry Smith   PetscInt       M = A->rmap->N;
1341b24902e0SBarry Smith   Mat_Plapack    *lu;
134201b82886SBarry Smith 
1343b24902e0SBarry Smith   PetscFunctionBegin;
1344719d5645SBarry Smith   lu = (Mat_Plapack*)F->spptr;
1345b24902e0SBarry Smith   ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr);
1346719d5645SBarry Smith   F->ops->lufactornumeric  = MatLUFactorNumeric_MPIDense;
134701b82886SBarry Smith   PetscFunctionReturn(0);
134801b82886SBarry Smith }
134901b82886SBarry Smith 
1350b6806ab0SHong Zhang EXTERN_C_BEGIN
135101b82886SBarry Smith #undef __FUNCT__
1352b6806ab0SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_mpidense_plapack"
1353b6806ab0SHong Zhang PetscErrorCode MatFactorGetSolverPackage_mpidense_plapack(Mat A,const MatSolverPackage *type)
1354b6806ab0SHong Zhang {
1355b6806ab0SHong Zhang   PetscFunctionBegin;
1356b6806ab0SHong Zhang   *type = MAT_SOLVER_PLAPACK;
1357b6806ab0SHong Zhang   PetscFunctionReturn(0);
1358b6806ab0SHong Zhang }
1359b6806ab0SHong Zhang EXTERN_C_END
1360b6806ab0SHong Zhang 
1361b6806ab0SHong Zhang #undef __FUNCT__
1362b6806ab0SHong Zhang #define __FUNCT__ "MatGetFactor_mpidense_plapack"
1363b6806ab0SHong Zhang PetscErrorCode MatGetFactor_mpidense_plapack(Mat A,MatFactorType ftype,Mat *F)
136401b82886SBarry Smith {
136501b82886SBarry Smith   PetscErrorCode ierr;
13666c995c7dSHong Zhang   Mat_Plapack    *lu;
13676c995c7dSHong Zhang   PetscMPIInt    size;
136885bd4cefSHong Zhang   PetscInt       M=A->rmap->N;
136901b82886SBarry Smith 
137001b82886SBarry Smith   PetscFunctionBegin;
137101b82886SBarry Smith   /* Create the factorization matrix */
1372eae6fb2eSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr);
1373d0f46423SBarry Smith   ierr = MatSetSizes(*F,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
1374eae6fb2eSBarry Smith   ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr);
137585bd4cefSHong Zhang   ierr = PetscNewLog(*F,Mat_Plapack,&lu);CHKERRQ(ierr);
137685bd4cefSHong Zhang   (*F)->spptr = (void*)lu;
137701b82886SBarry Smith 
1378b24902e0SBarry Smith   /* Set default Plapack parameters */
13796c995c7dSHong Zhang   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1380b24902e0SBarry Smith   lu->nb = M/size;
1381b24902e0SBarry Smith   if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */
138201b82886SBarry Smith 
1383b24902e0SBarry Smith   /* Set runtime options */
1384b24902e0SBarry Smith   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr);
1385b24902e0SBarry Smith     ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr);
1386b24902e0SBarry Smith   PetscOptionsEnd();
138701b82886SBarry Smith 
1388b24902e0SBarry Smith   /* Create object distribution template */
1389b24902e0SBarry Smith   lu->templ = NULL;
1390b24902e0SBarry Smith   ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr);
1391b24902e0SBarry Smith 
1392b24902e0SBarry Smith   /* Set the datatype */
1393b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX)
1394b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE_COMPLEX;
1395b24902e0SBarry Smith #else
1396b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE;
1397b24902e0SBarry Smith #endif
1398b24902e0SBarry Smith 
1399d0f46423SBarry Smith   ierr = PLA_Matrix_create(lu->datatype,M,A->cmap->N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr);
1400b24902e0SBarry Smith 
1401b24902e0SBarry Smith 
1402b24902e0SBarry Smith   lu->pla_solved     = PETSC_FALSE; /* MatSolve_Plapack() is called yet */
14035d00a290SHong Zhang   lu->mstruct        = DIFFERENT_NONZERO_PATTERN;
1404b24902e0SBarry Smith 
1405b24902e0SBarry Smith   if (ftype == MAT_FACTOR_LU) {
1406b24902e0SBarry Smith     (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense;
1407b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY) {
14086c995c7dSHong Zhang     (*F)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense;
1409b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices");
1410719d5645SBarry Smith   (*F)->factor = ftype;
1411b6806ab0SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)(*F),"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_mpidense_plapack",MatFactorGetSolverPackage_mpidense_plapack);CHKERRQ(ierr);
141201b82886SBarry Smith   PetscFunctionReturn(0);
141301b82886SBarry Smith }
141401b82886SBarry Smith #endif
141501b82886SBarry Smith 
1416488007eeSBarry Smith #undef __FUNCT__
1417488007eeSBarry Smith #define __FUNCT__ "MatAXPY_MPIDense"
1418488007eeSBarry Smith PetscErrorCode MatAXPY_MPIDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str)
1419488007eeSBarry Smith {
1420488007eeSBarry Smith   PetscErrorCode ierr;
1421488007eeSBarry Smith   Mat_MPIDense   *A = (Mat_MPIDense*)Y->data, *B = (Mat_MPIDense*)X->data;
1422488007eeSBarry Smith 
1423488007eeSBarry Smith   PetscFunctionBegin;
1424488007eeSBarry Smith   ierr = MatAXPY(A->A,alpha,B->A,str);CHKERRQ(ierr);
1425488007eeSBarry Smith   PetscFunctionReturn(0);
1426488007eeSBarry Smith }
1427488007eeSBarry Smith 
14288965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/
142909dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense,
143009dc0095SBarry Smith        MatGetRow_MPIDense,
143109dc0095SBarry Smith        MatRestoreRow_MPIDense,
143209dc0095SBarry Smith        MatMult_MPIDense,
143397304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense,
14347c922b88SBarry Smith        MatMultTranspose_MPIDense,
14357c922b88SBarry Smith        MatMultTransposeAdd_MPIDense,
14368965ea79SLois Curfman McInnes        0,
143709dc0095SBarry Smith        0,
143809dc0095SBarry Smith        0,
143997304618SKris Buschelman /*10*/ 0,
144009dc0095SBarry Smith        0,
144109dc0095SBarry Smith        0,
144209dc0095SBarry Smith        0,
144309dc0095SBarry Smith        MatTranspose_MPIDense,
144497304618SKris Buschelman /*15*/ MatGetInfo_MPIDense,
14456e4ee0c6SHong Zhang        MatEqual_MPIDense,
144609dc0095SBarry Smith        MatGetDiagonal_MPIDense,
14475b2fa520SLois Curfman McInnes        MatDiagonalScale_MPIDense,
144809dc0095SBarry Smith        MatNorm_MPIDense,
144997304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense,
145009dc0095SBarry Smith        MatAssemblyEnd_MPIDense,
145109dc0095SBarry Smith        MatSetOption_MPIDense,
145209dc0095SBarry Smith        MatZeroEntries_MPIDense,
1453d519adbfSMatthew Knepley /*24*/ MatZeroRows_MPIDense,
1454919b68f7SBarry Smith        0,
145501b82886SBarry Smith        0,
145601b82886SBarry Smith        0,
145701b82886SBarry Smith        0,
1458d519adbfSMatthew Knepley /*29*/ MatSetUpPreallocation_MPIDense,
1459273d9f13SBarry Smith        0,
146009dc0095SBarry Smith        0,
146109dc0095SBarry Smith        MatGetArray_MPIDense,
146209dc0095SBarry Smith        MatRestoreArray_MPIDense,
1463d519adbfSMatthew Knepley /*34*/ MatDuplicate_MPIDense,
146409dc0095SBarry Smith        0,
146509dc0095SBarry Smith        0,
146609dc0095SBarry Smith        0,
146709dc0095SBarry Smith        0,
1468d519adbfSMatthew Knepley /*39*/ MatAXPY_MPIDense,
14692ce60cd0SSatish Balay        MatGetSubMatrices_MPIDense,
147009dc0095SBarry Smith        0,
147109dc0095SBarry Smith        MatGetValues_MPIDense,
147209dc0095SBarry Smith        0,
1473d519adbfSMatthew Knepley /*44*/ 0,
147409dc0095SBarry Smith        MatScale_MPIDense,
147509dc0095SBarry Smith        0,
147609dc0095SBarry Smith        0,
147709dc0095SBarry Smith        0,
1478d519adbfSMatthew Knepley /*49*/ 0,
147909dc0095SBarry Smith        0,
148009dc0095SBarry Smith        0,
148109dc0095SBarry Smith        0,
148209dc0095SBarry Smith        0,
1483d519adbfSMatthew Knepley /*54*/ 0,
148409dc0095SBarry Smith        0,
148509dc0095SBarry Smith        0,
148609dc0095SBarry Smith        0,
148709dc0095SBarry Smith        0,
1488d519adbfSMatthew Knepley /*59*/ MatGetSubMatrix_MPIDense,
1489b9b97703SBarry Smith        MatDestroy_MPIDense,
1490b9b97703SBarry Smith        MatView_MPIDense,
1491357abbc8SBarry Smith        0,
149297304618SKris Buschelman        0,
1493d519adbfSMatthew Knepley /*64*/ 0,
149497304618SKris Buschelman        0,
149597304618SKris Buschelman        0,
149697304618SKris Buschelman        0,
149797304618SKris Buschelman        0,
1498d519adbfSMatthew Knepley /*69*/ 0,
149997304618SKris Buschelman        0,
150097304618SKris Buschelman        0,
150197304618SKris Buschelman        0,
150297304618SKris Buschelman        0,
1503d519adbfSMatthew Knepley /*74*/ 0,
150497304618SKris Buschelman        0,
150597304618SKris Buschelman        0,
150697304618SKris Buschelman        0,
150797304618SKris Buschelman        0,
1508d519adbfSMatthew Knepley /*79*/ 0,
150997304618SKris Buschelman        0,
151097304618SKris Buschelman        0,
151197304618SKris Buschelman        0,
1512d519adbfSMatthew Knepley /*83*/ MatLoad_MPIDense,
1513865e5f61SKris Buschelman        0,
1514865e5f61SKris Buschelman        0,
1515865e5f61SKris Buschelman        0,
1516865e5f61SKris Buschelman        0,
1517865e5f61SKris Buschelman        0,
1518d519adbfSMatthew Knepley /*89*/
15192fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
15202fbe02b9SBarry Smith        MatMatMult_MPIDense_MPIDense,
15214ae313f4SHong Zhang        MatMatMultSymbolic_MPIDense_MPIDense,
15222fbe02b9SBarry Smith        MatMatMultNumeric_MPIDense_MPIDense,
15232fbe02b9SBarry Smith #else
1524865e5f61SKris Buschelman        0,
1525865e5f61SKris Buschelman        0,
1526865e5f61SKris Buschelman        0,
15272fbe02b9SBarry Smith #endif
15282fbe02b9SBarry Smith        0,
1529d519adbfSMatthew Knepley /*94*/ 0,
1530865e5f61SKris Buschelman        0,
1531865e5f61SKris Buschelman        0,
1532865e5f61SKris Buschelman        0};
15338965ea79SLois Curfman McInnes 
1534273d9f13SBarry Smith EXTERN_C_BEGIN
15354a2ae208SSatish Balay #undef __FUNCT__
1536a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense"
1537be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data)
1538a23d5eceSKris Buschelman {
1539a23d5eceSKris Buschelman   Mat_MPIDense   *a;
1540dfbe8321SBarry Smith   PetscErrorCode ierr;
1541a23d5eceSKris Buschelman 
1542a23d5eceSKris Buschelman   PetscFunctionBegin;
1543a23d5eceSKris Buschelman   mat->preallocated = PETSC_TRUE;
1544a23d5eceSKris Buschelman   /* Note:  For now, when data is specified above, this assumes the user correctly
1545a23d5eceSKris Buschelman    allocates the local dense storage space.  We should add error checking. */
1546a23d5eceSKris Buschelman 
1547a23d5eceSKris Buschelman   a    = (Mat_MPIDense*)mat->data;
1548f69a0ea3SMatthew Knepley   ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr);
1549d0f46423SBarry Smith   ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr);
15505c5985e7SKris Buschelman   ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr);
15515c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr);
155252e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
1553a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1554a23d5eceSKris Buschelman }
1555a23d5eceSKris Buschelman EXTERN_C_END
1556a23d5eceSKris Buschelman 
15570bad9183SKris Buschelman /*MC
155841c8de11SBarry Smith    MAT_SOLVER_PLAPACK = "mpidense" - Parallel LU and Cholesky factorization for MATMPIDENSE matrices
15590bad9183SKris Buschelman 
156041c8de11SBarry Smith   run config/configure.py with the option --download-plapack
15617878bbefSBarry Smith 
15627878bbefSBarry Smith 
15637878bbefSBarry Smith   Options Database Keys:
15647878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition
15657878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition
15667878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector
15677878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size
15687878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag
15697878bbefSBarry Smith 
157041c8de11SBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE, PCFactorSetSolverPackage(), MatSolverPackage
157141c8de11SBarry Smith 
15720bad9183SKris Buschelman M*/
15730bad9183SKris Buschelman 
1574a23d5eceSKris Buschelman EXTERN_C_BEGIN
1575a23d5eceSKris Buschelman #undef __FUNCT__
15764a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense"
1577be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat)
1578273d9f13SBarry Smith {
1579273d9f13SBarry Smith   Mat_MPIDense   *a;
1580dfbe8321SBarry Smith   PetscErrorCode ierr;
1581273d9f13SBarry Smith 
1582273d9f13SBarry Smith   PetscFunctionBegin;
158338f2d2fdSLisandro Dalcin   ierr              = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr);
1584b0a32e0cSBarry Smith   mat->data         = (void*)a;
1585273d9f13SBarry Smith   ierr              = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1586273d9f13SBarry Smith   mat->mapping      = 0;
1587273d9f13SBarry Smith 
1588273d9f13SBarry Smith   mat->insertmode = NOT_SET_VALUES;
15897adad957SLisandro Dalcin   ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr);
15907adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr);
1591273d9f13SBarry Smith 
15927408324eSLisandro Dalcin   ierr = PetscMapSetBlockSize(mat->rmap,1);CHKERRQ(ierr);
15937408324eSLisandro Dalcin   ierr = PetscMapSetBlockSize(mat->cmap,1);CHKERRQ(ierr);
1594d0f46423SBarry Smith   ierr = PetscMapSetUp(mat->rmap);CHKERRQ(ierr);
1595d0f46423SBarry Smith   ierr = PetscMapSetUp(mat->cmap);CHKERRQ(ierr);
1596d0f46423SBarry Smith   a->nvec = mat->cmap->n;
1597273d9f13SBarry Smith 
1598273d9f13SBarry Smith   /* build cache for off array entries formed */
1599273d9f13SBarry Smith   a->donotstash = PETSC_FALSE;
16007adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr);
1601273d9f13SBarry Smith 
1602273d9f13SBarry Smith   /* stuff used for matrix vector multiply */
1603273d9f13SBarry Smith   a->lvec        = 0;
1604273d9f13SBarry Smith   a->Mvctx       = 0;
1605273d9f13SBarry Smith   a->roworiented = PETSC_TRUE;
1606273d9f13SBarry Smith 
1607273d9f13SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C",
1608273d9f13SBarry Smith                                      "MatGetDiagonalBlock_MPIDense",
1609273d9f13SBarry Smith                                      MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr);
1610a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C",
1611a23d5eceSKris Buschelman                                      "MatMPIDenseSetPreallocation_MPIDense",
1612a23d5eceSKris Buschelman                                      MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr);
16134ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C",
16144ae313f4SHong Zhang                                      "MatMatMult_MPIAIJ_MPIDense",
16154ae313f4SHong Zhang                                       MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr);
16164ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C",
16174ae313f4SHong Zhang                                      "MatMatMultSymbolic_MPIAIJ_MPIDense",
16184ae313f4SHong Zhang                                       MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr);
16194ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C",
16204ae313f4SHong Zhang                                      "MatMatMultNumeric_MPIAIJ_MPIDense",
16214ae313f4SHong Zhang                                       MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr);
16227878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1623b6806ab0SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_plapack_C",
1624b6806ab0SHong Zhang                                      "MatGetFactor_mpidense_plapack",
1625b6806ab0SHong Zhang                                       MatGetFactor_mpidense_plapack);CHKERRQ(ierr);
1626db4efbfdSBarry Smith   ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr);
1627db4efbfdSBarry Smith #endif
162838aed534SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr);
162901b82886SBarry Smith 
1630273d9f13SBarry Smith   PetscFunctionReturn(0);
1631273d9f13SBarry Smith }
1632273d9f13SBarry Smith EXTERN_C_END
1633273d9f13SBarry Smith 
1634209238afSKris Buschelman /*MC
1635002d173eSKris Buschelman    MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices.
1636209238afSKris Buschelman 
1637209238afSKris Buschelman    This matrix type is identical to MATSEQDENSE when constructed with a single process communicator,
1638209238afSKris Buschelman    and MATMPIDENSE otherwise.
1639209238afSKris Buschelman 
1640209238afSKris Buschelman    Options Database Keys:
1641209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions()
1642209238afSKris Buschelman 
1643209238afSKris Buschelman   Level: beginner
1644209238afSKris Buschelman 
164501b82886SBarry Smith 
1646209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE
1647209238afSKris Buschelman M*/
1648209238afSKris Buschelman 
1649209238afSKris Buschelman EXTERN_C_BEGIN
1650209238afSKris Buschelman #undef __FUNCT__
1651209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense"
1652be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A)
1653dfbe8321SBarry Smith {
16546849ba73SBarry Smith   PetscErrorCode ierr;
165513f74950SBarry Smith   PetscMPIInt    size;
1656209238afSKris Buschelman 
1657209238afSKris Buschelman   PetscFunctionBegin;
16587adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1659209238afSKris Buschelman   if (size == 1) {
1660209238afSKris Buschelman     ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr);
1661209238afSKris Buschelman   } else {
1662209238afSKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
1663209238afSKris Buschelman   }
1664209238afSKris Buschelman   PetscFunctionReturn(0);
1665209238afSKris Buschelman }
1666209238afSKris Buschelman EXTERN_C_END
1667209238afSKris Buschelman 
16684a2ae208SSatish Balay #undef __FUNCT__
16694a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation"
1670273d9f13SBarry Smith /*@C
1671273d9f13SBarry Smith    MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries
1672273d9f13SBarry Smith 
1673273d9f13SBarry Smith    Not collective
1674273d9f13SBarry Smith 
1675273d9f13SBarry Smith    Input Parameters:
1676273d9f13SBarry Smith .  A - the matrix
1677273d9f13SBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1678273d9f13SBarry Smith    to control all matrix memory allocation.
1679273d9f13SBarry Smith 
1680273d9f13SBarry Smith    Notes:
1681273d9f13SBarry Smith    The dense format is fully compatible with standard Fortran 77
1682273d9f13SBarry Smith    storage by columns.
1683273d9f13SBarry Smith 
1684273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1685273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1686273d9f13SBarry Smith    set data=PETSC_NULL.
1687273d9f13SBarry Smith 
1688273d9f13SBarry Smith    Level: intermediate
1689273d9f13SBarry Smith 
1690273d9f13SBarry Smith .keywords: matrix,dense, parallel
1691273d9f13SBarry Smith 
1692273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
1693273d9f13SBarry Smith @*/
1694be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data)
1695273d9f13SBarry Smith {
1696dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar *);
1697273d9f13SBarry Smith 
1698273d9f13SBarry Smith   PetscFunctionBegin;
1699565567f0SKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1700a23d5eceSKris Buschelman   if (f) {
1701a23d5eceSKris Buschelman     ierr = (*f)(mat,data);CHKERRQ(ierr);
1702a23d5eceSKris Buschelman   }
1703273d9f13SBarry Smith   PetscFunctionReturn(0);
1704273d9f13SBarry Smith }
1705273d9f13SBarry Smith 
17064a2ae208SSatish Balay #undef __FUNCT__
17074a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense"
17088965ea79SLois Curfman McInnes /*@C
170939ddd567SLois Curfman McInnes    MatCreateMPIDense - Creates a sparse parallel matrix in dense format.
17108965ea79SLois Curfman McInnes 
1711db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1712db81eaa0SLois Curfman McInnes 
17138965ea79SLois Curfman McInnes    Input Parameters:
1714db81eaa0SLois Curfman McInnes +  comm - MPI communicator
17158965ea79SLois Curfman McInnes .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
1716db81eaa0SLois Curfman McInnes .  n - number of local columns (or PETSC_DECIDE to have calculated if N is given)
17178965ea79SLois Curfman McInnes .  M - number of global rows (or PETSC_DECIDE to have calculated if m is given)
1718db81eaa0SLois Curfman McInnes .  N - number of global columns (or PETSC_DECIDE to have calculated if n is given)
17197f5ff6fdSBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc
1720dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
17218965ea79SLois Curfman McInnes 
17228965ea79SLois Curfman McInnes    Output Parameter:
1723477f1c0bSLois Curfman McInnes .  A - the matrix
17248965ea79SLois Curfman McInnes 
1725b259b22eSLois Curfman McInnes    Notes:
172639ddd567SLois Curfman McInnes    The dense format is fully compatible with standard Fortran 77
172739ddd567SLois Curfman McInnes    storage by columns.
17288965ea79SLois Curfman McInnes 
172918f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
173018f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
17317f5ff6fdSBarry Smith    set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users).
173218f449edSLois Curfman McInnes 
17338965ea79SLois Curfman McInnes    The user MUST specify either the local or global matrix dimensions
17348965ea79SLois Curfman McInnes    (possibly both).
17358965ea79SLois Curfman McInnes 
1736027ccd11SLois Curfman McInnes    Level: intermediate
1737027ccd11SLois Curfman McInnes 
173839ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel
17398965ea79SLois Curfman McInnes 
174039ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
17418965ea79SLois Curfman McInnes @*/
1742be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A)
17438965ea79SLois Curfman McInnes {
17446849ba73SBarry Smith   PetscErrorCode ierr;
174513f74950SBarry Smith   PetscMPIInt    size;
17468965ea79SLois Curfman McInnes 
17473a40ed3dSBarry Smith   PetscFunctionBegin;
1748f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1749f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr);
1750273d9f13SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1751273d9f13SBarry Smith   if (size > 1) {
1752273d9f13SBarry Smith     ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr);
1753273d9f13SBarry Smith     ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1754273d9f13SBarry Smith   } else {
1755273d9f13SBarry Smith     ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1756273d9f13SBarry Smith     ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
17578c469469SLois Curfman McInnes   }
17583a40ed3dSBarry Smith   PetscFunctionReturn(0);
17598965ea79SLois Curfman McInnes }
17608965ea79SLois Curfman McInnes 
17614a2ae208SSatish Balay #undef __FUNCT__
17624a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense"
17636849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
17648965ea79SLois Curfman McInnes {
17658965ea79SLois Curfman McInnes   Mat            mat;
17663501a2bdSLois Curfman McInnes   Mat_MPIDense   *a,*oldmat = (Mat_MPIDense*)A->data;
1767dfbe8321SBarry Smith   PetscErrorCode ierr;
17688965ea79SLois Curfman McInnes 
17693a40ed3dSBarry Smith   PetscFunctionBegin;
17708965ea79SLois Curfman McInnes   *newmat       = 0;
17717adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr);
1772d0f46423SBarry Smith   ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
17737adad957SLisandro Dalcin   ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr);
1774834f8fabSBarry Smith   a                 = (Mat_MPIDense*)mat->data;
1775e04c1aa4SHong Zhang   ierr              = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
17763501a2bdSLois Curfman McInnes   mat->factor       = A->factor;
1777c456f294SBarry Smith   mat->assembled    = PETSC_TRUE;
1778273d9f13SBarry Smith   mat->preallocated = PETSC_TRUE;
17798965ea79SLois Curfman McInnes 
1780d0f46423SBarry Smith   mat->rmap->rstart     = A->rmap->rstart;
1781d0f46423SBarry Smith   mat->rmap->rend       = A->rmap->rend;
17828965ea79SLois Curfman McInnes   a->size              = oldmat->size;
17838965ea79SLois Curfman McInnes   a->rank              = oldmat->rank;
1784e0fa3b82SLois Curfman McInnes   mat->insertmode      = NOT_SET_VALUES;
1785b9b97703SBarry Smith   a->nvec              = oldmat->nvec;
17863782ba37SSatish Balay   a->donotstash        = oldmat->donotstash;
1787e04c1aa4SHong Zhang 
1788d0f46423SBarry Smith   ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
1789d0f46423SBarry Smith   ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
17907adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr);
17918965ea79SLois Curfman McInnes 
1792329f5518SBarry Smith   ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
17935609ef8eSBarry Smith   ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr);
179452e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
179501b82886SBarry Smith 
17968965ea79SLois Curfman McInnes   *newmat = mat;
17973a40ed3dSBarry Smith   PetscFunctionReturn(0);
17988965ea79SLois Curfman McInnes }
17998965ea79SLois Curfman McInnes 
1800e090d566SSatish Balay #include "petscsys.h"
18018965ea79SLois Curfman McInnes 
18024a2ae208SSatish Balay #undef __FUNCT__
18034a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile"
1804a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat)
180590ace30eSBarry Smith {
18066849ba73SBarry Smith   PetscErrorCode ierr;
180732dcc486SBarry Smith   PetscMPIInt    rank,size;
180813f74950SBarry Smith   PetscInt       *rowners,i,m,nz,j;
180987828ca2SBarry Smith   PetscScalar    *array,*vals,*vals_ptr;
181090ace30eSBarry Smith   MPI_Status     status;
181190ace30eSBarry Smith 
18123a40ed3dSBarry Smith   PetscFunctionBegin;
1813d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
1814d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
181590ace30eSBarry Smith 
181690ace30eSBarry Smith   /* determine ownership of all rows */
181790ace30eSBarry Smith   m          = M/size + ((M % size) > rank);
181813f74950SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr);
181913f74950SBarry Smith   ierr       = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr);
182090ace30eSBarry Smith   rowners[0] = 0;
182190ace30eSBarry Smith   for (i=2; i<=size; i++) {
182290ace30eSBarry Smith     rowners[i] += rowners[i-1];
182390ace30eSBarry Smith   }
182490ace30eSBarry Smith 
1825f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1826f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1827be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1828878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
182990ace30eSBarry Smith   ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr);
183090ace30eSBarry Smith 
183190ace30eSBarry Smith   if (!rank) {
183287828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
183390ace30eSBarry Smith 
183490ace30eSBarry Smith     /* read in my part of the matrix numerical values  */
18350752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr);
183690ace30eSBarry Smith 
183790ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
183890ace30eSBarry Smith     vals_ptr = vals;
183990ace30eSBarry Smith     for (i=0; i<m; i++) {
184090ace30eSBarry Smith       for (j=0; j<N; j++) {
184190ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
184290ace30eSBarry Smith       }
184390ace30eSBarry Smith     }
184490ace30eSBarry Smith 
184590ace30eSBarry Smith     /* read in other processors and ship out */
184690ace30eSBarry Smith     for (i=1; i<size; i++) {
184790ace30eSBarry Smith       nz   = (rowners[i+1] - rowners[i])*N;
18480752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
18497adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr);
185090ace30eSBarry Smith     }
18513a40ed3dSBarry Smith   } else {
185290ace30eSBarry Smith     /* receive numeric values */
185387828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
185490ace30eSBarry Smith 
185590ace30eSBarry Smith     /* receive message of values*/
18567adad957SLisandro Dalcin     ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr);
185790ace30eSBarry Smith 
185890ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
185990ace30eSBarry Smith     vals_ptr = vals;
186090ace30eSBarry Smith     for (i=0; i<m; i++) {
186190ace30eSBarry Smith       for (j=0; j<N; j++) {
186290ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
186390ace30eSBarry Smith       }
186490ace30eSBarry Smith     }
186590ace30eSBarry Smith   }
1866606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
1867606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
18686d4a8577SBarry Smith   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
18696d4a8577SBarry Smith   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
18703a40ed3dSBarry Smith   PetscFunctionReturn(0);
187190ace30eSBarry Smith }
187290ace30eSBarry Smith 
18734a2ae208SSatish Balay #undef __FUNCT__
18744a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense"
1875a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat)
18768965ea79SLois Curfman McInnes {
18778965ea79SLois Curfman McInnes   Mat            A;
187887828ca2SBarry Smith   PetscScalar    *vals,*svals;
187919bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
18808965ea79SLois Curfman McInnes   MPI_Status     status;
188113f74950SBarry Smith   PetscMPIInt    rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz;
188213f74950SBarry Smith   PetscInt       header[4],*rowlengths = 0,M,N,*cols;
188313f74950SBarry Smith   PetscInt       *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols;
188413f74950SBarry Smith   PetscInt       i,nz,j,rstart,rend;
188513f74950SBarry Smith   int            fd;
18866849ba73SBarry Smith   PetscErrorCode ierr;
18878965ea79SLois Curfman McInnes 
18883a40ed3dSBarry Smith   PetscFunctionBegin;
1889d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1890d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
18918965ea79SLois Curfman McInnes   if (!rank) {
1892b0a32e0cSBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
18930752156aSBarry Smith     ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr);
1894552e946dSBarry Smith     if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object");
18958965ea79SLois Curfman McInnes   }
18968965ea79SLois Curfman McInnes 
189713f74950SBarry Smith   ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr);
189890ace30eSBarry Smith   M = header[1]; N = header[2]; nz = header[3];
189990ace30eSBarry Smith 
190090ace30eSBarry Smith   /*
190190ace30eSBarry Smith        Handle case where matrix is stored on disk as a dense matrix
190290ace30eSBarry Smith   */
190390ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) {
1904be5d1d56SKris Buschelman     ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr);
19053a40ed3dSBarry Smith     PetscFunctionReturn(0);
190690ace30eSBarry Smith   }
190790ace30eSBarry Smith 
19088965ea79SLois Curfman McInnes   /* determine ownership of all rows */
1909e44c0bd4SBarry Smith   m          = PetscMPIIntCast(M/size + ((M % size) > rank));
1910e44c0bd4SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr);
1911ca161407SBarry Smith   ierr       = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr);
19128965ea79SLois Curfman McInnes   rowners[0] = 0;
19138965ea79SLois Curfman McInnes   for (i=2; i<=size; i++) {
19148965ea79SLois Curfman McInnes     rowners[i] += rowners[i-1];
19158965ea79SLois Curfman McInnes   }
19168965ea79SLois Curfman McInnes   rstart = rowners[rank];
19178965ea79SLois Curfman McInnes   rend   = rowners[rank+1];
19188965ea79SLois Curfman McInnes 
19198965ea79SLois Curfman McInnes   /* distribute row lengths to all processors */
192013f74950SBarry Smith   ierr    = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr);
19218965ea79SLois Curfman McInnes   offlens = ourlens + (rend-rstart);
19228965ea79SLois Curfman McInnes   if (!rank) {
192313f74950SBarry Smith     ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
19240752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
192513f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr);
19268965ea79SLois Curfman McInnes     for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i];
19271466f1e1SBarry Smith     ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
1928606d414cSSatish Balay     ierr = PetscFree(sndcounts);CHKERRQ(ierr);
1929ca161407SBarry Smith   } else {
19301466f1e1SBarry Smith     ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
19318965ea79SLois Curfman McInnes   }
19328965ea79SLois Curfman McInnes 
19338965ea79SLois Curfman McInnes   if (!rank) {
19348965ea79SLois Curfman McInnes     /* calculate the number of nonzeros on each processor */
193513f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr);
193613f74950SBarry Smith     ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr);
19378965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
19388965ea79SLois Curfman McInnes       for (j=rowners[i]; j< rowners[i+1]; j++) {
19398965ea79SLois Curfman McInnes         procsnz[i] += rowlengths[j];
19408965ea79SLois Curfman McInnes       }
19418965ea79SLois Curfman McInnes     }
1942606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
19438965ea79SLois Curfman McInnes 
19448965ea79SLois Curfman McInnes     /* determine max buffer needed and allocate it */
19458965ea79SLois Curfman McInnes     maxnz = 0;
19468965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
19470452661fSBarry Smith       maxnz = PetscMax(maxnz,procsnz[i]);
19488965ea79SLois Curfman McInnes     }
194913f74950SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr);
19508965ea79SLois Curfman McInnes 
19518965ea79SLois Curfman McInnes     /* read in my part of the matrix column indices  */
19528965ea79SLois Curfman McInnes     nz = procsnz[0];
195313f74950SBarry Smith     ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
19540752156aSBarry Smith     ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr);
19558965ea79SLois Curfman McInnes 
19568965ea79SLois Curfman McInnes     /* read in every one elses and ship off */
19578965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
19588965ea79SLois Curfman McInnes       nz   = procsnz[i];
19590752156aSBarry Smith       ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
196013f74950SBarry Smith       ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr);
19618965ea79SLois Curfman McInnes     }
1962606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
19633a40ed3dSBarry Smith   } else {
19648965ea79SLois Curfman McInnes     /* determine buffer space needed for message */
19658965ea79SLois Curfman McInnes     nz = 0;
19668965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
19678965ea79SLois Curfman McInnes       nz += ourlens[i];
19688965ea79SLois Curfman McInnes     }
196913f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
19708965ea79SLois Curfman McInnes 
19718965ea79SLois Curfman McInnes     /* receive message of column indices*/
197213f74950SBarry Smith     ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr);
197313f74950SBarry Smith     ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr);
197429bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
19758965ea79SLois Curfman McInnes   }
19768965ea79SLois Curfman McInnes 
19778965ea79SLois Curfman McInnes   /* loop over local rows, determining number of off diagonal entries */
197813f74950SBarry Smith   ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr);
19798965ea79SLois Curfman McInnes   jj = 0;
19808965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
19818965ea79SLois Curfman McInnes     for (j=0; j<ourlens[i]; j++) {
19828965ea79SLois Curfman McInnes       if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++;
19838965ea79SLois Curfman McInnes       jj++;
19848965ea79SLois Curfman McInnes     }
19858965ea79SLois Curfman McInnes   }
19868965ea79SLois Curfman McInnes 
19878965ea79SLois Curfman McInnes   /* create our matrix */
19888965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
19898965ea79SLois Curfman McInnes     ourlens[i] -= offlens[i];
19908965ea79SLois Curfman McInnes   }
1991f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1992f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1993be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1994878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
19958965ea79SLois Curfman McInnes   A = *newmat;
19968965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
19978965ea79SLois Curfman McInnes     ourlens[i] += offlens[i];
19988965ea79SLois Curfman McInnes   }
19998965ea79SLois Curfman McInnes 
20008965ea79SLois Curfman McInnes   if (!rank) {
200187828ca2SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20028965ea79SLois Curfman McInnes 
20038965ea79SLois Curfman McInnes     /* read in my part of the matrix numerical values  */
20048965ea79SLois Curfman McInnes     nz = procsnz[0];
20050752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20068965ea79SLois Curfman McInnes 
20078965ea79SLois Curfman McInnes     /* insert into matrix */
20088965ea79SLois Curfman McInnes     jj      = rstart;
20098965ea79SLois Curfman McInnes     smycols = mycols;
20108965ea79SLois Curfman McInnes     svals   = vals;
20118965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20128965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
20138965ea79SLois Curfman McInnes       smycols += ourlens[i];
20148965ea79SLois Curfman McInnes       svals   += ourlens[i];
20158965ea79SLois Curfman McInnes       jj++;
20168965ea79SLois Curfman McInnes     }
20178965ea79SLois Curfman McInnes 
20188965ea79SLois Curfman McInnes     /* read in other processors and ship out */
20198965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
20208965ea79SLois Curfman McInnes       nz   = procsnz[i];
20210752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20227adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr);
20238965ea79SLois Curfman McInnes     }
2024606d414cSSatish Balay     ierr = PetscFree(procsnz);CHKERRQ(ierr);
20253a40ed3dSBarry Smith   } else {
20268965ea79SLois Curfman McInnes     /* receive numeric values */
202784d528b1SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20288965ea79SLois Curfman McInnes 
20298965ea79SLois Curfman McInnes     /* receive message of values*/
20307adad957SLisandro Dalcin     ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr);
2031ca161407SBarry Smith     ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr);
203229bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
20338965ea79SLois Curfman McInnes 
20348965ea79SLois Curfman McInnes     /* insert into matrix */
20358965ea79SLois Curfman McInnes     jj      = rstart;
20368965ea79SLois Curfman McInnes     smycols = mycols;
20378965ea79SLois Curfman McInnes     svals   = vals;
20388965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20398965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
20408965ea79SLois Curfman McInnes       smycols += ourlens[i];
20418965ea79SLois Curfman McInnes       svals   += ourlens[i];
20428965ea79SLois Curfman McInnes       jj++;
20438965ea79SLois Curfman McInnes     }
20448965ea79SLois Curfman McInnes   }
2045606d414cSSatish Balay   ierr = PetscFree(ourlens);CHKERRQ(ierr);
2046606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
2047606d414cSSatish Balay   ierr = PetscFree(mycols);CHKERRQ(ierr);
2048606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
20498965ea79SLois Curfman McInnes 
20506d4a8577SBarry Smith   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
20516d4a8577SBarry Smith   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
20523a40ed3dSBarry Smith   PetscFunctionReturn(0);
20538965ea79SLois Curfman McInnes }
205490ace30eSBarry Smith 
20556e4ee0c6SHong Zhang #undef __FUNCT__
20566e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense"
20576e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag)
20586e4ee0c6SHong Zhang {
20596e4ee0c6SHong Zhang   Mat_MPIDense   *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data;
20606e4ee0c6SHong Zhang   Mat            a,b;
20616e4ee0c6SHong Zhang   PetscTruth     flg;
20626e4ee0c6SHong Zhang   PetscErrorCode ierr;
206390ace30eSBarry Smith 
20646e4ee0c6SHong Zhang   PetscFunctionBegin;
20656e4ee0c6SHong Zhang   a = matA->A;
20666e4ee0c6SHong Zhang   b = matB->A;
20676e4ee0c6SHong Zhang   ierr = MatEqual(a,b,&flg);CHKERRQ(ierr);
20687adad957SLisandro Dalcin   ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr);
20696e4ee0c6SHong Zhang   PetscFunctionReturn(0);
20706e4ee0c6SHong Zhang }
207190ace30eSBarry Smith 
207209d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
207309d27a7eSBarry Smith 
207409d27a7eSBarry Smith #undef __FUNCT__
207509d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage"
207609d27a7eSBarry Smith /*@C
207709d27a7eSBarry Smith   PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK.
207809d27a7eSBarry Smith   Level: developer
207909d27a7eSBarry Smith 
208009d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK
208109d27a7eSBarry Smith .seealso: PetscFinalize()
208209d27a7eSBarry Smith @*/
208309d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void)
208409d27a7eSBarry Smith {
208509d27a7eSBarry Smith   PetscErrorCode ierr;
208609d27a7eSBarry Smith 
208709d27a7eSBarry Smith   PetscFunctionBegin;
208809d27a7eSBarry Smith   ierr = PLA_Finalize();CHKERRQ(ierr);
208909d27a7eSBarry Smith   PetscFunctionReturn(0);
209009d27a7eSBarry Smith }
209109d27a7eSBarry Smith 
209209d27a7eSBarry Smith #undef __FUNCT__
209309d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage"
209409d27a7eSBarry Smith /*@C
209509d27a7eSBarry Smith   PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is
2096db4efbfdSBarry Smith   called from MatCreate_MPIDense() the first time an MPI dense matrix is called.
209709d27a7eSBarry Smith 
209809d27a7eSBarry Smith   Input Parameter:
2099db4efbfdSBarry Smith .   comm - the communicator the matrix lives on
210009d27a7eSBarry Smith 
210109d27a7eSBarry Smith   Level: developer
210209d27a7eSBarry Smith 
2103db4efbfdSBarry Smith    Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the
2104db4efbfdSBarry Smith   same communicator (because there is some global state that is initialized and used for all matrices). In addition if
2105db4efbfdSBarry Smith   PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators
2106db4efbfdSBarry Smith   cannot overlap.
2107db4efbfdSBarry Smith 
210809d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK
210909d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize()
211009d27a7eSBarry Smith @*/
2111db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm)
211209d27a7eSBarry Smith {
2113eae6fb2eSBarry Smith   PetscMPIInt    size;
211409d27a7eSBarry Smith   PetscErrorCode ierr;
211509d27a7eSBarry Smith 
211609d27a7eSBarry Smith   PetscFunctionBegin;
211709d27a7eSBarry Smith   if (!PLA_Initialized(PETSC_NULL)) {
211809d27a7eSBarry Smith 
211909d27a7eSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
212004fea9ffSBarry Smith     Plapack_nprows = 1;
212104fea9ffSBarry Smith     Plapack_npcols = size;
212209d27a7eSBarry Smith 
2123aeccfd6fSBarry Smith     ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr);
21245d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr);
21255d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr);
212679b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG)
212779b0a62dSBarry Smith       Plapack_ierror = 3;
212879b0a62dSBarry Smith #else
212979b0a62dSBarry Smith       Plapack_ierror = 0;
213079b0a62dSBarry Smith #endif
21315d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr);
2132eae6fb2eSBarry Smith       if (Plapack_ierror){
2133eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr);
2134aeccfd6fSBarry Smith       } else {
2135eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr);
2136aeccfd6fSBarry Smith       }
2137aeccfd6fSBarry Smith 
2138eae6fb2eSBarry Smith       Plapack_nb_alg = 0;
21395d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr);
2140eae6fb2eSBarry Smith       if (Plapack_nb_alg) {
2141eae6fb2eSBarry Smith         ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr);
2142aeccfd6fSBarry Smith       }
2143aeccfd6fSBarry Smith     PetscOptionsEnd();
2144aeccfd6fSBarry Smith 
214504fea9ffSBarry Smith     ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr);
214604fea9ffSBarry Smith     ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr);
214709d27a7eSBarry Smith     ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr);
214809d27a7eSBarry Smith   }
214909d27a7eSBarry Smith   PetscFunctionReturn(0);
215009d27a7eSBarry Smith }
215190ace30eSBarry Smith 
215209d27a7eSBarry Smith #endif
2153