xref: /petsc/src/mat/impls/dense/mpi/mpidense.c (revision ba337c4413f444c9b07b767e9700a1bd83f660a1)
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;
12d84338a6SBarry Smith EXTERN_C_BEGIN
13d84338a6SBarry Smith #include "PLA.h"
14d84338a6SBarry Smith EXTERN_C_END
15d84338a6SBarry Smith 
16d84338a6SBarry Smith typedef struct {
17d84338a6SBarry Smith   PLA_Obj        A,pivots;
18d84338a6SBarry Smith   PLA_Template   templ;
19d84338a6SBarry Smith   MPI_Datatype   datatype;
20d84338a6SBarry Smith   PetscInt       nb,rstart;
21d84338a6SBarry Smith   VecScatter     ctx;
22d84338a6SBarry Smith   IS             is_pla,is_petsc;
23d84338a6SBarry Smith   PetscTruth     pla_solved;
24d84338a6SBarry Smith   MatStructure   mstruct;
25d84338a6SBarry Smith } Mat_Plapack;
26eae6fb2eSBarry Smith #endif
278965ea79SLois Curfman McInnes 
28ba8c8a56SBarry Smith #undef __FUNCT__
29ab92ecdeSBarry Smith #define __FUNCT__ "MatDenseGetLocalMatrix"
30ab92ecdeSBarry Smith /*@
31ab92ecdeSBarry Smith 
32ab92ecdeSBarry Smith       MatDenseGetLocalMatrix - For a MATMPIDENSE or MATSEQDENSE matrix returns the sequential
33ab92ecdeSBarry Smith               matrix that represents the operator. For sequential matrices it returns itself.
34ab92ecdeSBarry Smith 
35ab92ecdeSBarry Smith     Input Parameter:
36ab92ecdeSBarry Smith .      A - the Seq or MPI dense matrix
37ab92ecdeSBarry Smith 
38ab92ecdeSBarry Smith     Output Parameter:
39ab92ecdeSBarry Smith .      B - the inner matrix
40ab92ecdeSBarry Smith 
418e6c10adSSatish Balay     Level: intermediate
428e6c10adSSatish Balay 
43ab92ecdeSBarry Smith @*/
44ab92ecdeSBarry Smith PetscErrorCode MatDenseGetLocalMatrix(Mat A,Mat *B)
45ab92ecdeSBarry Smith {
46ab92ecdeSBarry Smith   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
47ab92ecdeSBarry Smith   PetscErrorCode ierr;
48ab92ecdeSBarry Smith   PetscTruth     flg;
49ab92ecdeSBarry Smith 
50ab92ecdeSBarry Smith   PetscFunctionBegin;
51ab92ecdeSBarry Smith   ierr = PetscTypeCompare((PetscObject)A,MATMPIDENSE,&flg);CHKERRQ(ierr);
52ab92ecdeSBarry Smith   if (flg) {
53ab92ecdeSBarry Smith     *B = mat->A;
54ab92ecdeSBarry Smith   } else {
55ab92ecdeSBarry Smith     *B = A;
56ab92ecdeSBarry Smith   }
57ab92ecdeSBarry Smith   PetscFunctionReturn(0);
58ab92ecdeSBarry Smith }
59ab92ecdeSBarry Smith 
60ab92ecdeSBarry Smith #undef __FUNCT__
61ba8c8a56SBarry Smith #define __FUNCT__ "MatGetRow_MPIDense"
62ba8c8a56SBarry Smith PetscErrorCode MatGetRow_MPIDense(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
63ba8c8a56SBarry Smith {
64ba8c8a56SBarry Smith   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
65ba8c8a56SBarry Smith   PetscErrorCode ierr;
66d0f46423SBarry Smith   PetscInt       lrow,rstart = A->rmap->rstart,rend = A->rmap->rend;
67ba8c8a56SBarry Smith 
68ba8c8a56SBarry Smith   PetscFunctionBegin;
69ba8c8a56SBarry Smith   if (row < rstart || row >= rend) SETERRQ(PETSC_ERR_SUP,"only local rows")
70ba8c8a56SBarry Smith   lrow = row - rstart;
71ba8c8a56SBarry Smith   ierr = MatGetRow(mat->A,lrow,nz,(const PetscInt **)idx,(const PetscScalar **)v);CHKERRQ(ierr);
72ba8c8a56SBarry Smith   PetscFunctionReturn(0);
73ba8c8a56SBarry Smith }
74ba8c8a56SBarry Smith 
75ba8c8a56SBarry Smith #undef __FUNCT__
76ba8c8a56SBarry Smith #define __FUNCT__ "MatRestoreRow_MPIDense"
77ba8c8a56SBarry Smith PetscErrorCode MatRestoreRow_MPIDense(Mat mat,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
78ba8c8a56SBarry Smith {
79ba8c8a56SBarry Smith   PetscErrorCode ierr;
80ba8c8a56SBarry Smith 
81ba8c8a56SBarry Smith   PetscFunctionBegin;
82ba8c8a56SBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
83ba8c8a56SBarry Smith   if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);}
84ba8c8a56SBarry Smith   PetscFunctionReturn(0);
85ba8c8a56SBarry Smith }
86ba8c8a56SBarry Smith 
870de54da6SSatish Balay EXTERN_C_BEGIN
884a2ae208SSatish Balay #undef __FUNCT__
894a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonalBlock_MPIDense"
90be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatGetDiagonalBlock_MPIDense(Mat A,PetscTruth *iscopy,MatReuse reuse,Mat *B)
910de54da6SSatish Balay {
920de54da6SSatish Balay   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
936849ba73SBarry Smith   PetscErrorCode ierr;
94d0f46423SBarry Smith   PetscInt       m = A->rmap->n,rstart = A->rmap->rstart;
9587828ca2SBarry Smith   PetscScalar    *array;
960de54da6SSatish Balay   MPI_Comm       comm;
970de54da6SSatish Balay 
980de54da6SSatish Balay   PetscFunctionBegin;
99d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_SUP,"Only square matrices supported.");
1000de54da6SSatish Balay 
1010de54da6SSatish Balay   /* The reuse aspect is not implemented efficiently */
1020de54da6SSatish Balay   if (reuse) { ierr = MatDestroy(*B);CHKERRQ(ierr);}
1030de54da6SSatish Balay 
1040de54da6SSatish Balay   ierr = PetscObjectGetComm((PetscObject)(mdn->A),&comm);CHKERRQ(ierr);
1050de54da6SSatish Balay   ierr = MatGetArray(mdn->A,&array);CHKERRQ(ierr);
106f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,B);CHKERRQ(ierr);
107f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*B,m,m,m,m);CHKERRQ(ierr);
1087adad957SLisandro Dalcin   ierr = MatSetType(*B,((PetscObject)mdn->A)->type_name);CHKERRQ(ierr);
1095c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(*B,array+m*rstart);CHKERRQ(ierr);
1100de54da6SSatish Balay   ierr = MatRestoreArray(mdn->A,&array);CHKERRQ(ierr);
1110de54da6SSatish Balay   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1120de54da6SSatish Balay   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1130de54da6SSatish Balay 
1140de54da6SSatish Balay   *iscopy = PETSC_TRUE;
1150de54da6SSatish Balay   PetscFunctionReturn(0);
1160de54da6SSatish Balay }
1170de54da6SSatish Balay EXTERN_C_END
1180de54da6SSatish Balay 
1194a2ae208SSatish Balay #undef __FUNCT__
1204a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_MPIDense"
12113f74950SBarry Smith PetscErrorCode MatSetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1228965ea79SLois Curfman McInnes {
12339b7565bSBarry Smith   Mat_MPIDense   *A = (Mat_MPIDense*)mat->data;
124dfbe8321SBarry Smith   PetscErrorCode ierr;
125d0f46423SBarry Smith   PetscInt       i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row;
126273d9f13SBarry Smith   PetscTruth     roworiented = A->roworiented;
1278965ea79SLois Curfman McInnes 
1283a40ed3dSBarry Smith   PetscFunctionBegin;
12971fd2e92SBarry Smith   if (v) PetscValidScalarPointer(v,6);
1308965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
1315ef9f2a5SBarry Smith     if (idxm[i] < 0) continue;
132d0f46423SBarry Smith     if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large");
1338965ea79SLois Curfman McInnes     if (idxm[i] >= rstart && idxm[i] < rend) {
1348965ea79SLois Curfman McInnes       row = idxm[i] - rstart;
13539b7565bSBarry Smith       if (roworiented) {
13639b7565bSBarry Smith         ierr = MatSetValues(A->A,1,&row,n,idxn,v+i*n,addv);CHKERRQ(ierr);
1373a40ed3dSBarry Smith       } else {
1388965ea79SLois Curfman McInnes         for (j=0; j<n; j++) {
1395ef9f2a5SBarry Smith           if (idxn[j] < 0) continue;
140d0f46423SBarry Smith           if (idxn[j] >= mat->cmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large");
14139b7565bSBarry Smith           ierr = MatSetValues(A->A,1,&row,1,&idxn[j],v+i+j*m,addv);CHKERRQ(ierr);
14239b7565bSBarry Smith         }
1438965ea79SLois Curfman McInnes       }
1443a40ed3dSBarry Smith     } else {
1453782ba37SSatish Balay       if (!A->donotstash) {
14639b7565bSBarry Smith         if (roworiented) {
147b400d20cSBarry Smith           ierr = MatStashValuesRow_Private(&mat->stash,idxm[i],n,idxn,v+i*n,PETSC_FALSE);CHKERRQ(ierr);
148d36fbae8SSatish Balay         } else {
149b400d20cSBarry Smith           ierr = MatStashValuesCol_Private(&mat->stash,idxm[i],n,idxn,v+i,m,PETSC_FALSE);CHKERRQ(ierr);
15039b7565bSBarry Smith         }
151b49de8d1SLois Curfman McInnes       }
152b49de8d1SLois Curfman McInnes     }
1533782ba37SSatish Balay   }
1543a40ed3dSBarry Smith   PetscFunctionReturn(0);
155b49de8d1SLois Curfman McInnes }
156b49de8d1SLois Curfman McInnes 
1574a2ae208SSatish Balay #undef __FUNCT__
1584a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_MPIDense"
15913f74950SBarry Smith PetscErrorCode MatGetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
160b49de8d1SLois Curfman McInnes {
161b49de8d1SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
162dfbe8321SBarry Smith   PetscErrorCode ierr;
163d0f46423SBarry Smith   PetscInt       i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row;
164b49de8d1SLois Curfman McInnes 
1653a40ed3dSBarry Smith   PetscFunctionBegin;
166b49de8d1SLois Curfman McInnes   for (i=0; i<m; i++) {
16797e567efSBarry Smith     if (idxm[i] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative row"); */
168d0f46423SBarry Smith     if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large");
169b49de8d1SLois Curfman McInnes     if (idxm[i] >= rstart && idxm[i] < rend) {
170b49de8d1SLois Curfman McInnes       row = idxm[i] - rstart;
171b49de8d1SLois Curfman McInnes       for (j=0; j<n; j++) {
17297e567efSBarry Smith         if (idxn[j] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative column"); */
173d0f46423SBarry Smith         if (idxn[j] >= mat->cmap->N) {
17429bbc08cSBarry Smith           SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large");
175a8c6a408SBarry Smith         }
176b49de8d1SLois Curfman McInnes         ierr = MatGetValues(mdn->A,1,&row,1,&idxn[j],v+i*n+j);CHKERRQ(ierr);
177b49de8d1SLois Curfman McInnes       }
178a8c6a408SBarry Smith     } else {
17929bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"Only local values currently supported");
1808965ea79SLois Curfman McInnes     }
1818965ea79SLois Curfman McInnes   }
1823a40ed3dSBarry Smith   PetscFunctionReturn(0);
1838965ea79SLois Curfman McInnes }
1848965ea79SLois Curfman McInnes 
1854a2ae208SSatish Balay #undef __FUNCT__
1864a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_MPIDense"
187dfbe8321SBarry Smith PetscErrorCode MatGetArray_MPIDense(Mat A,PetscScalar *array[])
188ff14e315SSatish Balay {
189ff14e315SSatish Balay   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
190dfbe8321SBarry Smith   PetscErrorCode ierr;
191ff14e315SSatish Balay 
1923a40ed3dSBarry Smith   PetscFunctionBegin;
193ff14e315SSatish Balay   ierr = MatGetArray(a->A,array);CHKERRQ(ierr);
1943a40ed3dSBarry Smith   PetscFunctionReturn(0);
195ff14e315SSatish Balay }
196ff14e315SSatish Balay 
1974a2ae208SSatish Balay #undef __FUNCT__
1984a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_MPIDense"
1994aa3045dSJed Brown static PetscErrorCode MatGetSubMatrix_MPIDense(Mat A,IS isrow,IS iscol,MatReuse scall,Mat *B)
200ca3fa75bSLois Curfman McInnes {
201ca3fa75bSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data,*newmatd;
202ca3fa75bSLois Curfman McInnes   Mat_SeqDense   *lmat = (Mat_SeqDense*)mat->A->data;
2036849ba73SBarry Smith   PetscErrorCode ierr;
2044aa3045dSJed Brown   PetscInt       i,j,rstart,rend,nrows,ncols,Ncols,nlrows,nlcols;
2055d0c19d7SBarry Smith   const PetscInt *irow,*icol;
20687828ca2SBarry Smith   PetscScalar    *av,*bv,*v = lmat->v;
207ca3fa75bSLois Curfman McInnes   Mat            newmat;
2084aa3045dSJed Brown   IS             iscol_local;
209ca3fa75bSLois Curfman McInnes 
210ca3fa75bSLois Curfman McInnes   PetscFunctionBegin;
2114aa3045dSJed Brown   ierr = ISAllGather(iscol,&iscol_local);CHKERRQ(ierr);
212ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
2134aa3045dSJed Brown   ierr = ISGetIndices(iscol_local,&icol);CHKERRQ(ierr);
214b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
215b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
2164aa3045dSJed Brown   ierr = ISGetSize(iscol,&Ncols);CHKERRQ(ierr); /* global number of columns, size of iscol_local */
217ca3fa75bSLois Curfman McInnes 
218ca3fa75bSLois Curfman McInnes   /* No parallel redistribution currently supported! Should really check each index set
2197eba5e9cSLois Curfman McInnes      to comfirm that it is OK.  ... Currently supports only submatrix same partitioning as
2207eba5e9cSLois Curfman McInnes      original matrix! */
221ca3fa75bSLois Curfman McInnes 
222ca3fa75bSLois Curfman McInnes   ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr);
2237eba5e9cSLois Curfman McInnes   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
224ca3fa75bSLois Curfman McInnes 
225ca3fa75bSLois Curfman McInnes   /* Check submatrix call */
226ca3fa75bSLois Curfman McInnes   if (scall == MAT_REUSE_MATRIX) {
22729bbc08cSBarry Smith     /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */
2287eba5e9cSLois Curfman McInnes     /* Really need to test rows and column sizes! */
229ca3fa75bSLois Curfman McInnes     newmat = *B;
230ca3fa75bSLois Curfman McInnes   } else {
231ca3fa75bSLois Curfman McInnes     /* Create and fill new matrix */
2327adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
2334aa3045dSJed Brown     ierr = MatSetSizes(newmat,nrows,ncols,PETSC_DECIDE,Ncols);CHKERRQ(ierr);
2347adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
235878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
236ca3fa75bSLois Curfman McInnes   }
237ca3fa75bSLois Curfman McInnes 
238ca3fa75bSLois Curfman McInnes   /* Now extract the data pointers and do the copy, column at a time */
239ca3fa75bSLois Curfman McInnes   newmatd = (Mat_MPIDense*)newmat->data;
240ca3fa75bSLois Curfman McInnes   bv      = ((Mat_SeqDense *)newmatd->A->data)->v;
241ca3fa75bSLois Curfman McInnes 
2424aa3045dSJed Brown   for (i=0; i<Ncols; i++) {
24325a33276SHong Zhang     av = v + ((Mat_SeqDense *)mat->A->data)->lda*icol[i];
244ca3fa75bSLois Curfman McInnes     for (j=0; j<nrows; j++) {
2457eba5e9cSLois Curfman McInnes       *bv++ = av[irow[j] - rstart];
246ca3fa75bSLois Curfman McInnes     }
247ca3fa75bSLois Curfman McInnes   }
248ca3fa75bSLois Curfman McInnes 
249ca3fa75bSLois Curfman McInnes   /* Assemble the matrices so that the correct flags are set */
250ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
251ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
252ca3fa75bSLois Curfman McInnes 
253ca3fa75bSLois Curfman McInnes   /* Free work space */
254ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
255ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
256ca3fa75bSLois Curfman McInnes   *B = newmat;
257ca3fa75bSLois Curfman McInnes   PetscFunctionReturn(0);
258ca3fa75bSLois Curfman McInnes }
259ca3fa75bSLois Curfman McInnes 
2604a2ae208SSatish Balay #undef __FUNCT__
2614a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense"
262dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[])
263ff14e315SSatish Balay {
2643a40ed3dSBarry Smith   PetscFunctionBegin;
2653a40ed3dSBarry Smith   PetscFunctionReturn(0);
266ff14e315SSatish Balay }
267ff14e315SSatish Balay 
2684a2ae208SSatish Balay #undef __FUNCT__
2694a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense"
270dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode)
2718965ea79SLois Curfman McInnes {
27239ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
2737adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)mat)->comm;
274dfbe8321SBarry Smith   PetscErrorCode ierr;
27513f74950SBarry Smith   PetscInt       nstash,reallocs;
2768965ea79SLois Curfman McInnes   InsertMode     addv;
2778965ea79SLois Curfman McInnes 
2783a40ed3dSBarry Smith   PetscFunctionBegin;
2798965ea79SLois Curfman McInnes   /* make sure all processors are either in INSERTMODE or ADDMODE */
280ca161407SBarry Smith   ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr);
2817056b6fcSBarry Smith   if (addv == (ADD_VALUES|INSERT_VALUES)) {
28229bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs");
2838965ea79SLois Curfman McInnes   }
284e0fa3b82SLois Curfman McInnes   mat->insertmode = addv; /* in case this processor had no cache */
2858965ea79SLois Curfman McInnes 
286d0f46423SBarry Smith   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr);
2878798bf22SSatish Balay   ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr);
288ae15b995SBarry Smith   ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr);
2893a40ed3dSBarry Smith   PetscFunctionReturn(0);
2908965ea79SLois Curfman McInnes }
2918965ea79SLois Curfman McInnes 
2924a2ae208SSatish Balay #undef __FUNCT__
2934a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense"
294dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode)
2958965ea79SLois Curfman McInnes {
29639ddd567SLois Curfman McInnes   Mat_MPIDense    *mdn=(Mat_MPIDense*)mat->data;
2976849ba73SBarry Smith   PetscErrorCode  ierr;
29813f74950SBarry Smith   PetscInt        i,*row,*col,flg,j,rstart,ncols;
29913f74950SBarry Smith   PetscMPIInt     n;
30087828ca2SBarry Smith   PetscScalar     *val;
301e0fa3b82SLois Curfman McInnes   InsertMode      addv=mat->insertmode;
3028965ea79SLois Curfman McInnes 
3033a40ed3dSBarry Smith   PetscFunctionBegin;
3048965ea79SLois Curfman McInnes   /*  wait on receives */
3057ef1d9bdSSatish Balay   while (1) {
3068798bf22SSatish Balay     ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr);
3077ef1d9bdSSatish Balay     if (!flg) break;
3088965ea79SLois Curfman McInnes 
3097ef1d9bdSSatish Balay     for (i=0; i<n;) {
3107ef1d9bdSSatish Balay       /* Now identify the consecutive vals belonging to the same row */
3117ef1d9bdSSatish Balay       for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; }
3127ef1d9bdSSatish Balay       if (j < n) ncols = j-i;
3137ef1d9bdSSatish Balay       else       ncols = n-i;
3147ef1d9bdSSatish Balay       /* Now assemble all these values with a single function call */
3157ef1d9bdSSatish Balay       ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr);
3167ef1d9bdSSatish Balay       i = j;
3178965ea79SLois Curfman McInnes     }
3187ef1d9bdSSatish Balay   }
3198798bf22SSatish Balay   ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr);
3208965ea79SLois Curfman McInnes 
32139ddd567SLois Curfman McInnes   ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr);
32239ddd567SLois Curfman McInnes   ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr);
3238965ea79SLois Curfman McInnes 
3246d4a8577SBarry Smith   if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) {
32539ddd567SLois Curfman McInnes     ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
3268965ea79SLois Curfman McInnes   }
3273a40ed3dSBarry Smith   PetscFunctionReturn(0);
3288965ea79SLois Curfman McInnes }
3298965ea79SLois Curfman McInnes 
3304a2ae208SSatish Balay #undef __FUNCT__
3314a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense"
332dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A)
3338965ea79SLois Curfman McInnes {
334dfbe8321SBarry Smith   PetscErrorCode ierr;
33539ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3363a40ed3dSBarry Smith 
3373a40ed3dSBarry Smith   PetscFunctionBegin;
3383a40ed3dSBarry Smith   ierr = MatZeroEntries(l->A);CHKERRQ(ierr);
3393a40ed3dSBarry Smith   PetscFunctionReturn(0);
3408965ea79SLois Curfman McInnes }
3418965ea79SLois Curfman McInnes 
3428965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the
3438965ea79SLois Curfman McInnes    matrix is square and the column and row owerships are identical.
3448965ea79SLois Curfman McInnes    This is a BUG. The only way to fix it seems to be to access
3453501a2bdSLois Curfman McInnes    mdn->A and mdn->B directly and not through the MatZeroRows()
3468965ea79SLois Curfman McInnes    routine.
3478965ea79SLois Curfman McInnes */
3484a2ae208SSatish Balay #undef __FUNCT__
3494a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense"
350f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
3518965ea79SLois Curfman McInnes {
35239ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3536849ba73SBarry Smith   PetscErrorCode ierr;
354d0f46423SBarry Smith   PetscInt       i,*owners = A->rmap->range;
35513f74950SBarry Smith   PetscInt       *nprocs,j,idx,nsends;
35613f74950SBarry Smith   PetscInt       nmax,*svalues,*starts,*owner,nrecvs;
3577adad957SLisandro Dalcin   PetscInt       *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source;
35813f74950SBarry Smith   PetscInt       *lens,*lrows,*values;
35913f74950SBarry Smith   PetscMPIInt    n,imdex,rank = l->rank,size = l->size;
3607adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)A)->comm;
3618965ea79SLois Curfman McInnes   MPI_Request    *send_waits,*recv_waits;
3628965ea79SLois Curfman McInnes   MPI_Status     recv_status,*send_status;
36335d8aa7fSBarry Smith   PetscTruth     found;
3648965ea79SLois Curfman McInnes 
3653a40ed3dSBarry Smith   PetscFunctionBegin;
3668965ea79SLois Curfman McInnes   /*  first count number of contributors to each processor */
36713f74950SBarry Smith   ierr  = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
36813f74950SBarry Smith   ierr  = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
36913f74950SBarry Smith   ierr  = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/
3708965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3718965ea79SLois Curfman McInnes     idx = rows[i];
37235d8aa7fSBarry Smith     found = PETSC_FALSE;
3738965ea79SLois Curfman McInnes     for (j=0; j<size; j++) {
3748965ea79SLois Curfman McInnes       if (idx >= owners[j] && idx < owners[j+1]) {
375c1dc657dSBarry Smith         nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break;
3768965ea79SLois Curfman McInnes       }
3778965ea79SLois Curfman McInnes     }
37829bbc08cSBarry Smith     if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range");
3798965ea79SLois Curfman McInnes   }
380c1dc657dSBarry Smith   nsends = 0;  for (i=0; i<size; i++) { nsends += nprocs[2*i+1];}
3818965ea79SLois Curfman McInnes 
3828965ea79SLois Curfman McInnes   /* inform other processors of number of messages and max length*/
383c1dc657dSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
3848965ea79SLois Curfman McInnes 
3858965ea79SLois Curfman McInnes   /* post receives:   */
38613f74950SBarry Smith   ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr);
387b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
3888965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
38913f74950SBarry Smith     ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr);
3908965ea79SLois Curfman McInnes   }
3918965ea79SLois Curfman McInnes 
3928965ea79SLois Curfman McInnes   /* do sends:
3938965ea79SLois Curfman McInnes       1) starts[i] gives the starting index in svalues for stuff going to
3948965ea79SLois Curfman McInnes          the ith processor
3958965ea79SLois Curfman McInnes   */
39613f74950SBarry Smith   ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr);
397b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
39813f74950SBarry Smith   ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
3998965ea79SLois Curfman McInnes   starts[0]  = 0;
400c1dc657dSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
4018965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
4028965ea79SLois Curfman McInnes     svalues[starts[owner[i]]++] = rows[i];
4038965ea79SLois Curfman McInnes   }
4048965ea79SLois Curfman McInnes 
4058965ea79SLois Curfman McInnes   starts[0] = 0;
406c1dc657dSBarry Smith   for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
4078965ea79SLois Curfman McInnes   count = 0;
4088965ea79SLois Curfman McInnes   for (i=0; i<size; i++) {
409c1dc657dSBarry Smith     if (nprocs[2*i+1]) {
41013f74950SBarry Smith       ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr);
4118965ea79SLois Curfman McInnes     }
4128965ea79SLois Curfman McInnes   }
413606d414cSSatish Balay   ierr = PetscFree(starts);CHKERRQ(ierr);
4148965ea79SLois Curfman McInnes 
4158965ea79SLois Curfman McInnes   base = owners[rank];
4168965ea79SLois Curfman McInnes 
4178965ea79SLois Curfman McInnes   /*  wait on receives */
41874ed9c26SBarry Smith   ierr   = PetscMalloc2(nrecvs,PetscInt,&lens,nrecvs,PetscInt,&source);CHKERRQ(ierr);
41974ed9c26SBarry Smith   count  = nrecvs;
42074ed9c26SBarry Smith   slen   = 0;
4218965ea79SLois Curfman McInnes   while (count) {
422ca161407SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
4238965ea79SLois Curfman McInnes     /* unpack receives into our local space */
42413f74950SBarry Smith     ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr);
4258965ea79SLois Curfman McInnes     source[imdex]  = recv_status.MPI_SOURCE;
4268965ea79SLois Curfman McInnes     lens[imdex]    = n;
4278965ea79SLois Curfman McInnes     slen += n;
4288965ea79SLois Curfman McInnes     count--;
4298965ea79SLois Curfman McInnes   }
430606d414cSSatish Balay   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
4318965ea79SLois Curfman McInnes 
4328965ea79SLois Curfman McInnes   /* move the data into the send scatter */
43313f74950SBarry Smith   ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr);
4348965ea79SLois Curfman McInnes   count = 0;
4358965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
4368965ea79SLois Curfman McInnes     values = rvalues + i*nmax;
4378965ea79SLois Curfman McInnes     for (j=0; j<lens[i]; j++) {
4388965ea79SLois Curfman McInnes       lrows[count++] = values[j] - base;
4398965ea79SLois Curfman McInnes     }
4408965ea79SLois Curfman McInnes   }
441606d414cSSatish Balay   ierr = PetscFree(rvalues);CHKERRQ(ierr);
44274ed9c26SBarry Smith   ierr = PetscFree2(lens,source);CHKERRQ(ierr);
443606d414cSSatish Balay   ierr = PetscFree(owner);CHKERRQ(ierr);
444606d414cSSatish Balay   ierr = PetscFree(nprocs);CHKERRQ(ierr);
4458965ea79SLois Curfman McInnes 
4468965ea79SLois Curfman McInnes   /* actually zap the local rows */
447f4df32b1SMatthew Knepley   ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr);
448606d414cSSatish Balay   ierr = PetscFree(lrows);CHKERRQ(ierr);
4498965ea79SLois Curfman McInnes 
4508965ea79SLois Curfman McInnes   /* wait on sends */
4518965ea79SLois Curfman McInnes   if (nsends) {
452b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
453ca161407SBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
454606d414cSSatish Balay     ierr = PetscFree(send_status);CHKERRQ(ierr);
4558965ea79SLois Curfman McInnes   }
456606d414cSSatish Balay   ierr = PetscFree(send_waits);CHKERRQ(ierr);
457606d414cSSatish Balay   ierr = PetscFree(svalues);CHKERRQ(ierr);
4588965ea79SLois Curfman McInnes 
4593a40ed3dSBarry Smith   PetscFunctionReturn(0);
4608965ea79SLois Curfman McInnes }
4618965ea79SLois Curfman McInnes 
4624a2ae208SSatish Balay #undef __FUNCT__
4634a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense"
464dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy)
4658965ea79SLois Curfman McInnes {
46639ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
467dfbe8321SBarry Smith   PetscErrorCode ierr;
468c456f294SBarry Smith 
4693a40ed3dSBarry Smith   PetscFunctionBegin;
470ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
471ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
47244cd7ae7SLois Curfman McInnes   ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr);
4733a40ed3dSBarry Smith   PetscFunctionReturn(0);
4748965ea79SLois Curfman McInnes }
4758965ea79SLois Curfman McInnes 
4764a2ae208SSatish Balay #undef __FUNCT__
4774a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense"
478dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz)
4798965ea79SLois Curfman McInnes {
48039ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
481dfbe8321SBarry Smith   PetscErrorCode ierr;
482c456f294SBarry Smith 
4833a40ed3dSBarry Smith   PetscFunctionBegin;
484ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
485ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
48644cd7ae7SLois Curfman McInnes   ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr);
4873a40ed3dSBarry Smith   PetscFunctionReturn(0);
4888965ea79SLois Curfman McInnes }
4898965ea79SLois Curfman McInnes 
4904a2ae208SSatish Balay #undef __FUNCT__
4914a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense"
492dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy)
493096963f5SLois Curfman McInnes {
494096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
495dfbe8321SBarry Smith   PetscErrorCode ierr;
49687828ca2SBarry Smith   PetscScalar    zero = 0.0;
497096963f5SLois Curfman McInnes 
4983a40ed3dSBarry Smith   PetscFunctionBegin;
4992dcb1b2aSMatthew Knepley   ierr = VecSet(yy,zero);CHKERRQ(ierr);
5007c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
501ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
502ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
5033a40ed3dSBarry Smith   PetscFunctionReturn(0);
504096963f5SLois Curfman McInnes }
505096963f5SLois Curfman McInnes 
5064a2ae208SSatish Balay #undef __FUNCT__
5074a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense"
508dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz)
509096963f5SLois Curfman McInnes {
510096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
511dfbe8321SBarry Smith   PetscErrorCode ierr;
512096963f5SLois Curfman McInnes 
5133a40ed3dSBarry Smith   PetscFunctionBegin;
5143501a2bdSLois Curfman McInnes   ierr = VecCopy(yy,zz);CHKERRQ(ierr);
5157c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
516ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
517ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
5183a40ed3dSBarry Smith   PetscFunctionReturn(0);
519096963f5SLois Curfman McInnes }
520096963f5SLois Curfman McInnes 
5214a2ae208SSatish Balay #undef __FUNCT__
5224a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense"
523dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v)
5248965ea79SLois Curfman McInnes {
52539ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
526096963f5SLois Curfman McInnes   Mat_SeqDense   *aloc = (Mat_SeqDense*)a->A->data;
527dfbe8321SBarry Smith   PetscErrorCode ierr;
528d0f46423SBarry Smith   PetscInt       len,i,n,m = A->rmap->n,radd;
52987828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
530ed3cc1f0SBarry Smith 
5313a40ed3dSBarry Smith   PetscFunctionBegin;
5322dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
5331ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
534096963f5SLois Curfman McInnes   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
535d0f46423SBarry Smith   if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
536d0f46423SBarry Smith   len  = PetscMin(a->A->rmap->n,a->A->cmap->n);
537d0f46423SBarry Smith   radd = A->rmap->rstart*m;
53844cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
539096963f5SLois Curfman McInnes     x[i] = aloc->v[radd + i*m + i];
540096963f5SLois Curfman McInnes   }
5411ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
5423a40ed3dSBarry Smith   PetscFunctionReturn(0);
5438965ea79SLois Curfman McInnes }
5448965ea79SLois Curfman McInnes 
5454a2ae208SSatish Balay #undef __FUNCT__
5464a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense"
547dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat)
5488965ea79SLois Curfman McInnes {
5493501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
550dfbe8321SBarry Smith   PetscErrorCode ierr;
55101b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
55201b82886SBarry Smith   Mat_Plapack   *lu=(Mat_Plapack*)(mat->spptr);
55301b82886SBarry Smith #endif
554ed3cc1f0SBarry Smith 
5553a40ed3dSBarry Smith   PetscFunctionBegin;
55694d884c6SBarry Smith 
557aa482453SBarry Smith #if defined(PETSC_USE_LOG)
558d0f46423SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N);
5598965ea79SLois Curfman McInnes #endif
5608798bf22SSatish Balay   ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr);
5613501a2bdSLois Curfman McInnes   ierr = MatDestroy(mdn->A);CHKERRQ(ierr);
56205b42c5fSBarry Smith   if (mdn->lvec)   {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);}
56305b42c5fSBarry Smith   if (mdn->Mvctx)  {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);}
56401b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
5652fbe02b9SBarry Smith   if (lu) {
56662b4c0b3SBarry Smith     ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr);
56762b4c0b3SBarry Smith     ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr);
56862b4c0b3SBarry Smith     ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr);
56909d27a7eSBarry Smith 
5702fbe02b9SBarry Smith     if (lu->is_pla) {
57101b82886SBarry Smith       ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr);
57201b82886SBarry Smith       ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr);
57301b82886SBarry Smith       ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr);
574622d7880SLois Curfman McInnes     }
5752fbe02b9SBarry Smith   }
57601b82886SBarry Smith #endif
57701b82886SBarry Smith 
578606d414cSSatish Balay   ierr = PetscFree(mdn);CHKERRQ(ierr);
579dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
580901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr);
581901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
5824ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5834ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5844ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5853a40ed3dSBarry Smith   PetscFunctionReturn(0);
5868965ea79SLois Curfman McInnes }
58739ddd567SLois Curfman McInnes 
5884a2ae208SSatish Balay #undef __FUNCT__
5894a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary"
5906849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer)
5918965ea79SLois Curfman McInnes {
59239ddd567SLois Curfman McInnes   Mat_MPIDense      *mdn = (Mat_MPIDense*)mat->data;
593dfbe8321SBarry Smith   PetscErrorCode    ierr;
594aa05aa95SBarry Smith   PetscViewerFormat format;
595aa05aa95SBarry Smith   int               fd;
596d0f46423SBarry Smith   PetscInt          header[4],mmax,N = mat->cmap->N,i,j,m,k;
597aa05aa95SBarry Smith   PetscMPIInt       rank,tag  = ((PetscObject)viewer)->tag,size;
598578230a0SSatish Balay   PetscScalar       *work,*v,*vv;
599aa05aa95SBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)mdn->A->data;
600aa05aa95SBarry Smith   MPI_Status        status;
6017056b6fcSBarry Smith 
6023a40ed3dSBarry Smith   PetscFunctionBegin;
60339ddd567SLois Curfman McInnes   if (mdn->size == 1) {
60439ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
605aa05aa95SBarry Smith   } else {
606aa05aa95SBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
6077adad957SLisandro Dalcin     ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr);
6087adad957SLisandro Dalcin     ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr);
609aa05aa95SBarry Smith 
610aa05aa95SBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
6117973ad04SBarry Smith     if (format == PETSC_VIEWER_NATIVE) {
612aa05aa95SBarry Smith 
613aa05aa95SBarry Smith       if (!rank) {
614aa05aa95SBarry Smith         /* store the matrix as a dense matrix */
615aa05aa95SBarry Smith         header[0] = MAT_FILE_COOKIE;
616d0f46423SBarry Smith         header[1] = mat->rmap->N;
617aa05aa95SBarry Smith         header[2] = N;
618aa05aa95SBarry Smith         header[3] = MATRIX_BINARY_FORMAT_DENSE;
619aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
620aa05aa95SBarry Smith 
621aa05aa95SBarry Smith         /* get largest work array needed for transposing array */
622d0f46423SBarry Smith         mmax = mat->rmap->n;
623aa05aa95SBarry Smith         for (i=1; i<size; i++) {
624d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
6258965ea79SLois Curfman McInnes         }
626aa05aa95SBarry Smith         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr);
627aa05aa95SBarry Smith 
628aa05aa95SBarry Smith         /* write out local array, by rows */
629d0f46423SBarry Smith         m    = mat->rmap->n;
630aa05aa95SBarry Smith         v    = a->v;
631aa05aa95SBarry Smith         for (j=0; j<N; j++) {
632aa05aa95SBarry Smith           for (i=0; i<m; i++) {
633578230a0SSatish Balay             work[j + i*N] = *v++;
634aa05aa95SBarry Smith           }
635aa05aa95SBarry Smith         }
636aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
637aa05aa95SBarry Smith         /* get largest work array to receive messages from other processes, excludes process zero */
638aa05aa95SBarry Smith         mmax = 0;
639aa05aa95SBarry Smith         for (i=1; i<size; i++) {
640d0f46423SBarry Smith           mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]);
641aa05aa95SBarry Smith         }
642578230a0SSatish Balay         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr);
643aa05aa95SBarry Smith         for(k = 1; k < size; k++) {
644f8009846SMatthew Knepley           v    = vv;
645d0f46423SBarry Smith           m    = mat->rmap->range[k+1] - mat->rmap->range[k];
6467adad957SLisandro Dalcin           ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr);
647aa05aa95SBarry Smith 
648aa05aa95SBarry Smith           for(j = 0; j < N; j++) {
649aa05aa95SBarry Smith             for(i = 0; i < m; i++) {
650578230a0SSatish Balay               work[j + i*N] = *v++;
651aa05aa95SBarry Smith             }
652aa05aa95SBarry Smith           }
653aa05aa95SBarry Smith           ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
654aa05aa95SBarry Smith         }
655aa05aa95SBarry Smith         ierr = PetscFree(work);CHKERRQ(ierr);
656578230a0SSatish Balay         ierr = PetscFree(vv);CHKERRQ(ierr);
657aa05aa95SBarry Smith       } else {
658d0f46423SBarry Smith         ierr = MPI_Send(a->v,mat->rmap->n*mat->cmap->N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr);
659aa05aa95SBarry Smith       }
6604aa34b0aSBarry Smith     } else {
6617973ad04SBarry Smith       SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_NATIVE");
662aa05aa95SBarry Smith     }
663aa05aa95SBarry Smith   }
6643a40ed3dSBarry Smith   PetscFunctionReturn(0);
6658965ea79SLois Curfman McInnes }
6668965ea79SLois Curfman McInnes 
6674a2ae208SSatish Balay #undef __FUNCT__
6684a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket"
6696849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer)
6708965ea79SLois Curfman McInnes {
67139ddd567SLois Curfman McInnes   Mat_MPIDense          *mdn = (Mat_MPIDense*)mat->data;
672dfbe8321SBarry Smith   PetscErrorCode        ierr;
67332dcc486SBarry Smith   PetscMPIInt           size = mdn->size,rank = mdn->rank;
674a313700dSBarry Smith   const PetscViewerType vtype;
67532077d6dSBarry Smith   PetscTruth            iascii,isdraw;
676b0a32e0cSBarry Smith   PetscViewer           sviewer;
677f3ef73ceSBarry Smith   PetscViewerFormat     format;
67801b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
6795d00a290SHong Zhang   Mat_Plapack           *lu;
68001b82886SBarry Smith #endif
6818965ea79SLois Curfman McInnes 
6823a40ed3dSBarry Smith   PetscFunctionBegin;
68332077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
684fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
68532077d6dSBarry Smith   if (iascii) {
686b0a32e0cSBarry Smith     ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr);
687b0a32e0cSBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
688456192e2SBarry Smith     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
6894e220ebcSLois Curfman McInnes       MatInfo info;
690888f2ed8SSatish Balay       ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr);
691d0f46423SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap->n,
69277431f27SBarry Smith                    (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr);
693b0a32e0cSBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
69401b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
69501b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr);
6965d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Processor mesh: nprows %d, npcols %d\n",Plapack_nprows, Plapack_npcols);CHKERRQ(ierr);
6975d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Error checking: %d\n",Plapack_ierror);CHKERRQ(ierr);
6985d00a290SHong Zhang       ierr = PetscViewerASCIIPrintf(viewer,"  Algorithmic block size: %d\n",Plapack_nb_alg);CHKERRQ(ierr);
6995d00a290SHong Zhang       if (mat->factor){
7005d00a290SHong Zhang         lu=(Mat_Plapack*)(mat->spptr);
70101b82886SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"  Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr);
7025d00a290SHong Zhang       }
7035d00a290SHong Zhang #else
7045d00a290SHong Zhang       ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr);
70501b82886SBarry Smith #endif
7063a40ed3dSBarry Smith       PetscFunctionReturn(0);
707fb9695e5SSatish Balay     } else if (format == PETSC_VIEWER_ASCII_INFO) {
7083a40ed3dSBarry Smith       PetscFunctionReturn(0);
7098965ea79SLois Curfman McInnes     }
710f1af5d2fSBarry Smith   } else if (isdraw) {
711b0a32e0cSBarry Smith     PetscDraw  draw;
712f1af5d2fSBarry Smith     PetscTruth isnull;
713f1af5d2fSBarry Smith 
714b0a32e0cSBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
715b0a32e0cSBarry Smith     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
716f1af5d2fSBarry Smith     if (isnull) PetscFunctionReturn(0);
717f1af5d2fSBarry Smith   }
71877ed5343SBarry Smith 
7198965ea79SLois Curfman McInnes   if (size == 1) {
72039ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
7213a40ed3dSBarry Smith   } else {
7228965ea79SLois Curfman McInnes     /* assemble the entire matrix onto first processor. */
7238965ea79SLois Curfman McInnes     Mat         A;
724d0f46423SBarry Smith     PetscInt    M = mat->rmap->N,N = mat->cmap->N,m,row,i,nz;
725ba8c8a56SBarry Smith     PetscInt    *cols;
726ba8c8a56SBarry Smith     PetscScalar *vals;
7278965ea79SLois Curfman McInnes 
7287adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr);
7298965ea79SLois Curfman McInnes     if (!rank) {
730f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr);
7313a40ed3dSBarry Smith     } else {
732f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr);
7338965ea79SLois Curfman McInnes     }
7347adad957SLisandro Dalcin     /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */
735878740d9SKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
736878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL);
73752e6d16bSBarry Smith     ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr);
7388965ea79SLois Curfman McInnes 
73939ddd567SLois Curfman McInnes     /* Copy the matrix ... This isn't the most efficient means,
74039ddd567SLois Curfman McInnes        but it's quick for now */
74151022da4SBarry Smith     A->insertmode = INSERT_VALUES;
742d0f46423SBarry Smith     row = mat->rmap->rstart; m = mdn->A->rmap->n;
7438965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
744ba8c8a56SBarry Smith       ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
745ba8c8a56SBarry Smith       ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
746ba8c8a56SBarry Smith       ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
74739ddd567SLois Curfman McInnes       row++;
7488965ea79SLois Curfman McInnes     }
7498965ea79SLois Curfman McInnes 
7506d4a8577SBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7516d4a8577SBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
752b0a32e0cSBarry Smith     ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr);
753b9b97703SBarry Smith     if (!rank) {
7546831982aSBarry Smith       ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr);
7558965ea79SLois Curfman McInnes     }
756b0a32e0cSBarry Smith     ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr);
757b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
7588965ea79SLois Curfman McInnes     ierr = MatDestroy(A);CHKERRQ(ierr);
7598965ea79SLois Curfman McInnes   }
7603a40ed3dSBarry Smith   PetscFunctionReturn(0);
7618965ea79SLois Curfman McInnes }
7628965ea79SLois Curfman McInnes 
7634a2ae208SSatish Balay #undef __FUNCT__
7644a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense"
765dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer)
7668965ea79SLois Curfman McInnes {
767dfbe8321SBarry Smith   PetscErrorCode ierr;
76832077d6dSBarry Smith   PetscTruth     iascii,isbinary,isdraw,issocket;
7698965ea79SLois Curfman McInnes 
770433994e6SBarry Smith   PetscFunctionBegin;
7710f5bd95cSBarry Smith 
77232077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
773fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
774b0a32e0cSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr);
775fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
7760f5bd95cSBarry Smith 
77732077d6dSBarry Smith   if (iascii || issocket || isdraw) {
778f1af5d2fSBarry Smith     ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr);
7790f5bd95cSBarry Smith   } else if (isbinary) {
7803a40ed3dSBarry Smith     ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr);
7815cd90555SBarry Smith   } else {
782958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name);
7838965ea79SLois Curfman McInnes   }
7843a40ed3dSBarry Smith   PetscFunctionReturn(0);
7858965ea79SLois Curfman McInnes }
7868965ea79SLois Curfman McInnes 
7874a2ae208SSatish Balay #undef __FUNCT__
7884a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense"
789dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info)
7908965ea79SLois Curfman McInnes {
7913501a2bdSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
7923501a2bdSLois Curfman McInnes   Mat            mdn = mat->A;
793dfbe8321SBarry Smith   PetscErrorCode ierr;
794329f5518SBarry Smith   PetscReal      isend[5],irecv[5];
7958965ea79SLois Curfman McInnes 
7963a40ed3dSBarry Smith   PetscFunctionBegin;
7974e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
7984e220ebcSLois Curfman McInnes   ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr);
7994e220ebcSLois Curfman McInnes   isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded;
8004e220ebcSLois Curfman McInnes   isend[3] = info->memory;  isend[4] = info->mallocs;
8018965ea79SLois Curfman McInnes   if (flag == MAT_LOCAL) {
8024e220ebcSLois Curfman McInnes     info->nz_used      = isend[0];
8034e220ebcSLois Curfman McInnes     info->nz_allocated = isend[1];
8044e220ebcSLois Curfman McInnes     info->nz_unneeded  = isend[2];
8054e220ebcSLois Curfman McInnes     info->memory       = isend[3];
8064e220ebcSLois Curfman McInnes     info->mallocs      = isend[4];
8078965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_MAX) {
8087adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
8094e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
8104e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
8114e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
8124e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
8134e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
8148965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_SUM) {
8157adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
8164e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
8174e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
8184e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
8194e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
8204e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
8218965ea79SLois Curfman McInnes   }
8224e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0; /* no parallel LU/ILU/Cholesky */
8234e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
8244e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
8253a40ed3dSBarry Smith   PetscFunctionReturn(0);
8268965ea79SLois Curfman McInnes }
8278965ea79SLois Curfman McInnes 
8284a2ae208SSatish Balay #undef __FUNCT__
8294a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense"
8304e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg)
8318965ea79SLois Curfman McInnes {
83239ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
833dfbe8321SBarry Smith   PetscErrorCode ierr;
8348965ea79SLois Curfman McInnes 
8353a40ed3dSBarry Smith   PetscFunctionBegin;
83612c028f9SKris Buschelman   switch (op) {
837512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
83812c028f9SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
83912c028f9SKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
8404e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
84112c028f9SKris Buschelman     break;
84212c028f9SKris Buschelman   case MAT_ROW_ORIENTED:
8434e0d8c25SBarry Smith     a->roworiented = flg;
8444e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
84512c028f9SKris Buschelman     break;
8464e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
84712c028f9SKris Buschelman   case MAT_USE_HASH_TABLE:
848290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
84912c028f9SKris Buschelman     break;
85012c028f9SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
8514e0d8c25SBarry Smith     a->donotstash = flg;
85212c028f9SKris Buschelman     break;
85377e54ba9SKris Buschelman   case MAT_SYMMETRIC:
85477e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
8559a4540c5SBarry Smith   case MAT_HERMITIAN:
8569a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
857600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
858290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
85977e54ba9SKris Buschelman     break;
86012c028f9SKris Buschelman   default:
861600fe468SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
8623a40ed3dSBarry Smith   }
8633a40ed3dSBarry Smith   PetscFunctionReturn(0);
8648965ea79SLois Curfman McInnes }
8658965ea79SLois Curfman McInnes 
8668965ea79SLois Curfman McInnes 
8674a2ae208SSatish Balay #undef __FUNCT__
8684a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense"
869dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr)
8705b2fa520SLois Curfman McInnes {
8715b2fa520SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8725b2fa520SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
87387828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
874dfbe8321SBarry Smith   PetscErrorCode ierr;
875d0f46423SBarry Smith   PetscInt       i,j,s2a,s3a,s2,s3,m=mdn->A->rmap->n,n=mdn->A->cmap->n;
8765b2fa520SLois Curfman McInnes 
8775b2fa520SLois Curfman McInnes   PetscFunctionBegin;
87872d926a5SLois Curfman McInnes   ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr);
8795b2fa520SLois Curfman McInnes   if (ll) {
88072d926a5SLois Curfman McInnes     ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr);
881175be7b4SMatthew Knepley     if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2);
8821ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
8835b2fa520SLois Curfman McInnes     for (i=0; i<m; i++) {
8845b2fa520SLois Curfman McInnes       x = l[i];
8855b2fa520SLois Curfman McInnes       v = mat->v + i;
8865b2fa520SLois Curfman McInnes       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
8875b2fa520SLois Curfman McInnes     }
8881ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
889efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8905b2fa520SLois Curfman McInnes   }
8915b2fa520SLois Curfman McInnes   if (rr) {
892175be7b4SMatthew Knepley     ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr);
893175be7b4SMatthew Knepley     if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3);
894ca9f406cSSatish Balay     ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
895ca9f406cSSatish Balay     ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
8961ebc52fbSHong Zhang     ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr);
8975b2fa520SLois Curfman McInnes     for (i=0; i<n; i++) {
8985b2fa520SLois Curfman McInnes       x = r[i];
8995b2fa520SLois Curfman McInnes       v = mat->v + i*m;
9005b2fa520SLois Curfman McInnes       for (j=0; j<m; j++) { (*v++) *= x;}
9015b2fa520SLois Curfman McInnes     }
9021ebc52fbSHong Zhang     ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr);
903efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
9045b2fa520SLois Curfman McInnes   }
9055b2fa520SLois Curfman McInnes   PetscFunctionReturn(0);
9065b2fa520SLois Curfman McInnes }
9075b2fa520SLois Curfman McInnes 
9084a2ae208SSatish Balay #undef __FUNCT__
9094a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense"
910dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm)
911096963f5SLois Curfman McInnes {
9123501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
9133501a2bdSLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
914dfbe8321SBarry Smith   PetscErrorCode ierr;
91513f74950SBarry Smith   PetscInt       i,j;
916329f5518SBarry Smith   PetscReal      sum = 0.0;
91787828ca2SBarry Smith   PetscScalar    *v = mat->v;
9183501a2bdSLois Curfman McInnes 
9193a40ed3dSBarry Smith   PetscFunctionBegin;
9203501a2bdSLois Curfman McInnes   if (mdn->size == 1) {
921064f8208SBarry Smith     ierr =  MatNorm(mdn->A,type,nrm);CHKERRQ(ierr);
9223501a2bdSLois Curfman McInnes   } else {
9233501a2bdSLois Curfman McInnes     if (type == NORM_FROBENIUS) {
924d0f46423SBarry Smith       for (i=0; i<mdn->A->cmap->n*mdn->A->rmap->n; i++) {
925aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
926329f5518SBarry Smith         sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
9273501a2bdSLois Curfman McInnes #else
9283501a2bdSLois Curfman McInnes         sum += (*v)*(*v); v++;
9293501a2bdSLois Curfman McInnes #endif
9303501a2bdSLois Curfman McInnes       }
9317adad957SLisandro Dalcin       ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
932064f8208SBarry Smith       *nrm = sqrt(*nrm);
933dc0b31edSSatish Balay       ierr = PetscLogFlops(2.0*mdn->A->cmap->n*mdn->A->rmap->n);CHKERRQ(ierr);
9343a40ed3dSBarry Smith     } else if (type == NORM_1) {
935329f5518SBarry Smith       PetscReal *tmp,*tmp2;
93674ed9c26SBarry Smith       ierr = PetscMalloc2(A->cmap->N,PetscReal,&tmp,A->cmap->N,PetscReal,&tmp2);CHKERRQ(ierr);
93774ed9c26SBarry Smith       ierr = PetscMemzero(tmp,A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr);
93874ed9c26SBarry Smith       ierr = PetscMemzero(tmp2,A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr);
939064f8208SBarry Smith       *nrm = 0.0;
9403501a2bdSLois Curfman McInnes       v = mat->v;
941d0f46423SBarry Smith       for (j=0; j<mdn->A->cmap->n; j++) {
942d0f46423SBarry Smith         for (i=0; i<mdn->A->rmap->n; i++) {
94367e560aaSBarry Smith           tmp[j] += PetscAbsScalar(*v);  v++;
9443501a2bdSLois Curfman McInnes         }
9453501a2bdSLois Curfman McInnes       }
946d0f46423SBarry Smith       ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
947d0f46423SBarry Smith       for (j=0; j<A->cmap->N; j++) {
948064f8208SBarry Smith         if (tmp2[j] > *nrm) *nrm = tmp2[j];
9493501a2bdSLois Curfman McInnes       }
95074ed9c26SBarry Smith       ierr = PetscFree2(tmp,tmp);CHKERRQ(ierr);
951d0f46423SBarry Smith       ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
9523a40ed3dSBarry Smith     } else if (type == NORM_INFINITY) { /* max row norm */
953329f5518SBarry Smith       PetscReal ntemp;
9543501a2bdSLois Curfman McInnes       ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr);
9557adad957SLisandro Dalcin       ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
9563a40ed3dSBarry Smith     } else {
95729bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"No support for two norm");
9583501a2bdSLois Curfman McInnes     }
9593501a2bdSLois Curfman McInnes   }
9603a40ed3dSBarry Smith   PetscFunctionReturn(0);
9613501a2bdSLois Curfman McInnes }
9623501a2bdSLois Curfman McInnes 
9634a2ae208SSatish Balay #undef __FUNCT__
9644a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense"
965fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout)
9663501a2bdSLois Curfman McInnes {
9673501a2bdSLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
9683501a2bdSLois Curfman McInnes   Mat_SeqDense   *Aloc = (Mat_SeqDense*)a->A->data;
9693501a2bdSLois Curfman McInnes   Mat            B;
970d0f46423SBarry Smith   PetscInt       M = A->rmap->N,N = A->cmap->N,m,n,*rwork,rstart = A->rmap->rstart;
9716849ba73SBarry Smith   PetscErrorCode ierr;
97213f74950SBarry Smith   PetscInt       j,i;
97387828ca2SBarry Smith   PetscScalar    *v;
9743501a2bdSLois Curfman McInnes 
9753a40ed3dSBarry Smith   PetscFunctionBegin;
976e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place");
977fc4dec0aSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || A == *matout) {
9787adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
979d0f46423SBarry Smith     ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,N,M);CHKERRQ(ierr);
9807adad957SLisandro Dalcin     ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
981878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr);
982fc4dec0aSBarry Smith   } else {
983fc4dec0aSBarry Smith     B = *matout;
984fc4dec0aSBarry Smith   }
9853501a2bdSLois Curfman McInnes 
986d0f46423SBarry Smith   m = a->A->rmap->n; n = a->A->cmap->n; v = Aloc->v;
9871acff37aSSatish Balay   ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr);
9883501a2bdSLois Curfman McInnes   for (i=0; i<m; i++) rwork[i] = rstart + i;
9891acff37aSSatish Balay   for (j=0; j<n; j++) {
9903501a2bdSLois Curfman McInnes     ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr);
9913501a2bdSLois Curfman McInnes     v   += m;
9923501a2bdSLois Curfman McInnes   }
993606d414cSSatish Balay   ierr = PetscFree(rwork);CHKERRQ(ierr);
9946d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9956d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
996815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *matout != A) {
9973501a2bdSLois Curfman McInnes     *matout = B;
9983501a2bdSLois Curfman McInnes   } else {
999273d9f13SBarry Smith     ierr = MatHeaderCopy(A,B);CHKERRQ(ierr);
10003501a2bdSLois Curfman McInnes   }
10013a40ed3dSBarry Smith   PetscFunctionReturn(0);
1002096963f5SLois Curfman McInnes }
1003096963f5SLois Curfman McInnes 
100444cd7ae7SLois Curfman McInnes 
10056849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *);
1006d84338a6SBarry Smith extern PetscErrorCode MatScale_MPIDense(Mat,PetscScalar);
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);
1117d2a212eaSBarry Smith   SETERRQ(PETSC_ERR_LIB,"Due to apparent 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);
120574ed9c26SBarry Smith     ierr = PetscMalloc2(loc_m,PetscInt,&idx_pla,loc_m,PetscInt,&idx_petsc);CHKERRQ(ierr);
12066c995c7dSHong Zhang 
12076c995c7dSHong Zhang     rstart = (r_rank*c_nproc+c_rank)*lu->nb;
12086c995c7dSHong Zhang     for (i=0; i<loc_m; i+=lu->nb){
12096c995c7dSHong Zhang       j = 0;
12106c995c7dSHong Zhang       while (j < lu->nb && i+j < loc_m){
12116c995c7dSHong Zhang         idx_petsc[i+j] = rstart + j; j++;
12126c995c7dSHong Zhang       }
12136c995c7dSHong Zhang       rstart += size*lu->nb;
12146c995c7dSHong Zhang     }
12156c995c7dSHong Zhang 
12166c995c7dSHong Zhang     for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride;
12176c995c7dSHong Zhang 
12186c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr);
12196c995c7dSHong Zhang     ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr);
122074ed9c26SBarry Smith     ierr = PetscFree2(idx_pla,idx_petsc);CHKERRQ(ierr);
12216c995c7dSHong Zhang     ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr);
12226c995c7dSHong Zhang   }
12236c995c7dSHong Zhang   ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12246c995c7dSHong Zhang   ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
12256c995c7dSHong Zhang 
12266c995c7dSHong Zhang   /* Free data */
12276c995c7dSHong Zhang   ierr = VecDestroy(loc_x);CHKERRQ(ierr);
12286c995c7dSHong Zhang   PLA_Obj_free(&v_pla);
12296c995c7dSHong Zhang 
12306c995c7dSHong Zhang   lu->pla_solved = PETSC_TRUE;
12316c995c7dSHong Zhang   PetscFunctionReturn(0);
12326c995c7dSHong Zhang }
12336c995c7dSHong Zhang 
12346c995c7dSHong Zhang #undef __FUNCT__
12356c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense"
12360481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info)
12376c995c7dSHong Zhang {
1238719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
12396c995c7dSHong Zhang   PetscErrorCode ierr;
1240d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12416c995c7dSHong Zhang   PetscInt       info_pla=0;
12426c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12436c995c7dSHong Zhang 
12446c995c7dSHong Zhang   PetscFunctionBegin;
12456c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12466c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12476c995c7dSHong Zhang     PLA_Obj_free (&lu->pivots);
12486c995c7dSHong Zhang   }
12496c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12506c995c7dSHong Zhang   lu->A = NULL; lu->pivots = NULL;
12516c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12526c995c7dSHong Zhang   PLA_Obj_set_to_zero(lu->A);
12536c995c7dSHong Zhang   PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);
12546c995c7dSHong Zhang 
12556c995c7dSHong Zhang   /* Copy A into lu->A */
12566c995c7dSHong Zhang   PLA_API_begin();
12576c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
12586c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
12596c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
12606c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
12616c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
12626c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
12636c995c7dSHong Zhang   PLA_API_end();
12646c995c7dSHong Zhang 
12656c995c7dSHong Zhang   /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */
12666c995c7dSHong Zhang   info_pla = PLA_LU(lu->A,lu->pivots);
12676c995c7dSHong Zhang   if (info_pla != 0)
12686c995c7dSHong Zhang     SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla);
12696c995c7dSHong Zhang 
12706c995c7dSHong Zhang   lu->rstart         = rstart;
12716c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1272719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1273719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
12746c995c7dSHong Zhang   PetscFunctionReturn(0);
12756c995c7dSHong Zhang }
12766c995c7dSHong Zhang 
12776c995c7dSHong Zhang #undef __FUNCT__
12786c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense"
12790481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info)
12806c995c7dSHong Zhang {
1281719d5645SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)F->spptr;
12826c995c7dSHong Zhang   PetscErrorCode ierr;
1283d0f46423SBarry Smith   PetscInt       M=A->rmap->N,m=A->rmap->n,rstart,rend;
12846c995c7dSHong Zhang   PetscInt       info_pla=0;
12856c995c7dSHong Zhang   PetscScalar    *array,one = 1.0;
12866c995c7dSHong Zhang 
12876c995c7dSHong Zhang   PetscFunctionBegin;
12886c995c7dSHong Zhang   if (lu->mstruct == SAME_NONZERO_PATTERN){
12896c995c7dSHong Zhang     PLA_Obj_free(&lu->A);
12906c995c7dSHong Zhang   }
12916c995c7dSHong Zhang   /* Create PLAPACK matrix object */
12926c995c7dSHong Zhang   lu->A      = NULL;
12936c995c7dSHong Zhang   lu->pivots = NULL;
12946c995c7dSHong Zhang   PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);
12956c995c7dSHong Zhang 
12966c995c7dSHong Zhang   /* Copy A into lu->A */
12976c995c7dSHong Zhang   PLA_API_begin();
12986c995c7dSHong Zhang   PLA_Obj_API_open(lu->A);
12996c995c7dSHong Zhang   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
13006c995c7dSHong Zhang   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
13016c995c7dSHong Zhang   PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);
13026c995c7dSHong Zhang   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
13036c995c7dSHong Zhang   PLA_Obj_API_close(lu->A);
13046c995c7dSHong Zhang   PLA_API_end();
13056c995c7dSHong Zhang 
13066c995c7dSHong Zhang   /* Factor P A -> Chol */
13076c995c7dSHong Zhang   info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A);
13086c995c7dSHong Zhang   if (info_pla != 0)
13096c995c7dSHong Zhang     SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla);
13106c995c7dSHong Zhang 
13116c995c7dSHong Zhang   lu->rstart         = rstart;
13126c995c7dSHong Zhang   lu->mstruct        = SAME_NONZERO_PATTERN;
1313719d5645SBarry Smith   F->ops->solve      = MatSolve_MPIDense;
1314719d5645SBarry Smith   F->assembled       = PETSC_TRUE;  /* required by -ksp_view */
13156c995c7dSHong Zhang   PetscFunctionReturn(0);
13166c995c7dSHong Zhang }
13176c995c7dSHong Zhang 
1318b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */
13192fbe02b9SBarry Smith #undef __FUNCT__
13206c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense"
13210481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,const MatFactorInfo *info)
132201b82886SBarry Smith {
132301b82886SBarry Smith   PetscErrorCode ierr;
1324b24902e0SBarry Smith   PetscTruth     issymmetric,set;
132501b82886SBarry Smith 
132601b82886SBarry Smith   PetscFunctionBegin;
1327b24902e0SBarry Smith   ierr = MatIsSymmetricKnown(A,&set,&issymmetric);CHKERRQ(ierr);
1328b24902e0SBarry Smith   if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()");
1329719d5645SBarry Smith   F->ops->choleskyfactornumeric  = MatCholeskyFactorNumeric_MPIDense;
1330b24902e0SBarry Smith   PetscFunctionReturn(0);
133101b82886SBarry Smith }
133201b82886SBarry Smith 
1333b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */
1334b24902e0SBarry Smith #undef __FUNCT__
13356c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense"
13360481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info)
1337b24902e0SBarry Smith {
1338b24902e0SBarry Smith   PetscErrorCode ierr;
1339d0f46423SBarry Smith   PetscInt       M = A->rmap->N;
1340b24902e0SBarry Smith   Mat_Plapack    *lu;
134101b82886SBarry Smith 
1342b24902e0SBarry Smith   PetscFunctionBegin;
1343719d5645SBarry Smith   lu = (Mat_Plapack*)F->spptr;
1344b24902e0SBarry Smith   ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr);
1345719d5645SBarry Smith   F->ops->lufactornumeric  = MatLUFactorNumeric_MPIDense;
134601b82886SBarry Smith   PetscFunctionReturn(0);
134701b82886SBarry Smith }
134801b82886SBarry Smith 
1349b6806ab0SHong Zhang EXTERN_C_BEGIN
135001b82886SBarry Smith #undef __FUNCT__
1351b6806ab0SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_mpidense_plapack"
1352b6806ab0SHong Zhang PetscErrorCode MatFactorGetSolverPackage_mpidense_plapack(Mat A,const MatSolverPackage *type)
1353b6806ab0SHong Zhang {
1354b6806ab0SHong Zhang   PetscFunctionBegin;
1355b6806ab0SHong Zhang   *type = MAT_SOLVER_PLAPACK;
1356b6806ab0SHong Zhang   PetscFunctionReturn(0);
1357b6806ab0SHong Zhang }
1358b6806ab0SHong Zhang EXTERN_C_END
1359b6806ab0SHong Zhang 
1360bb5747d9SMatthew Knepley EXTERN_C_BEGIN
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 }
1414bb5747d9SMatthew Knepley EXTERN_C_END
141501b82886SBarry Smith #endif
141601b82886SBarry Smith 
1417488007eeSBarry Smith #undef __FUNCT__
141830716080SHong Zhang #define __FUNCT__ "MatGetFactor_mpidense_petsc"
141930716080SHong Zhang PetscErrorCode MatGetFactor_mpidense_petsc(Mat A,MatFactorType ftype,Mat *F)
142030716080SHong Zhang {
142130716080SHong Zhang #if defined(PETSC_HAVE_PLAPACK)
142230716080SHong Zhang   PetscErrorCode ierr;
142330716080SHong Zhang #endif
142430716080SHong Zhang 
142530716080SHong Zhang   PetscFunctionBegin;
142630716080SHong Zhang #if defined(PETSC_HAVE_PLAPACK)
142730716080SHong Zhang   ierr = MatGetFactor_mpidense_plapack(A,ftype,F);CHKERRQ(ierr);
142830716080SHong Zhang #else
142930716080SHong Zhang   SETERRQ1(PETSC_ERR_SUP,"Matrix format %s uses PLAPACK direct solver. Install PLAPACK",((PetscObject)A)->type_name);
143030716080SHong Zhang #endif
143130716080SHong Zhang   PetscFunctionReturn(0);
143230716080SHong Zhang }
143330716080SHong Zhang 
143430716080SHong Zhang #undef __FUNCT__
1435488007eeSBarry Smith #define __FUNCT__ "MatAXPY_MPIDense"
1436488007eeSBarry Smith PetscErrorCode MatAXPY_MPIDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str)
1437488007eeSBarry Smith {
1438488007eeSBarry Smith   PetscErrorCode ierr;
1439488007eeSBarry Smith   Mat_MPIDense   *A = (Mat_MPIDense*)Y->data, *B = (Mat_MPIDense*)X->data;
1440488007eeSBarry Smith 
1441488007eeSBarry Smith   PetscFunctionBegin;
1442488007eeSBarry Smith   ierr = MatAXPY(A->A,alpha,B->A,str);CHKERRQ(ierr);
1443488007eeSBarry Smith   PetscFunctionReturn(0);
1444488007eeSBarry Smith }
1445488007eeSBarry Smith 
1446*ba337c44SJed Brown #undef __FUNCT__
1447*ba337c44SJed Brown #define __FUNCT__ "MatConjugate_MPIDense"
1448*ba337c44SJed Brown PetscErrorCode PETSCMAT_DLLEXPORT MatConjugate_MPIDense(Mat mat)
1449*ba337c44SJed Brown {
1450*ba337c44SJed Brown   Mat_MPIDense   *a = (Mat_MPIDense *)mat->data;
1451*ba337c44SJed Brown   PetscErrorCode ierr;
1452*ba337c44SJed Brown 
1453*ba337c44SJed Brown   PetscFunctionBegin;
1454*ba337c44SJed Brown   ierr = MatConjugate(a->A);CHKERRQ(ierr);
1455*ba337c44SJed Brown   PetscFunctionReturn(0);
1456*ba337c44SJed Brown }
1457*ba337c44SJed Brown 
1458*ba337c44SJed Brown #undef __FUNCT__
1459*ba337c44SJed Brown #define __FUNCT__ "MatRealPart_MPIDense"
1460*ba337c44SJed Brown PetscErrorCode MatRealPart_MPIDense(Mat A)
1461*ba337c44SJed Brown {
1462*ba337c44SJed Brown   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
1463*ba337c44SJed Brown   PetscErrorCode ierr;
1464*ba337c44SJed Brown 
1465*ba337c44SJed Brown   PetscFunctionBegin;
1466*ba337c44SJed Brown   ierr = MatRealPart(a->A);CHKERRQ(ierr);
1467*ba337c44SJed Brown   PetscFunctionReturn(0);
1468*ba337c44SJed Brown }
1469*ba337c44SJed Brown 
1470*ba337c44SJed Brown #undef __FUNCT__
1471*ba337c44SJed Brown #define __FUNCT__ "MatImaginaryPart_MPIDense"
1472*ba337c44SJed Brown PetscErrorCode MatImaginaryPart_MPIDense(Mat A)
1473*ba337c44SJed Brown {
1474*ba337c44SJed Brown   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
1475*ba337c44SJed Brown   PetscErrorCode ierr;
1476*ba337c44SJed Brown 
1477*ba337c44SJed Brown   PetscFunctionBegin;
1478*ba337c44SJed Brown   ierr = MatImaginaryPart(a->A);CHKERRQ(ierr);
1479*ba337c44SJed Brown   PetscFunctionReturn(0);
1480*ba337c44SJed Brown }
1481*ba337c44SJed Brown 
14828965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/
148309dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense,
148409dc0095SBarry Smith        MatGetRow_MPIDense,
148509dc0095SBarry Smith        MatRestoreRow_MPIDense,
148609dc0095SBarry Smith        MatMult_MPIDense,
148797304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense,
14887c922b88SBarry Smith        MatMultTranspose_MPIDense,
14897c922b88SBarry Smith        MatMultTransposeAdd_MPIDense,
14908965ea79SLois Curfman McInnes        0,
149109dc0095SBarry Smith        0,
149209dc0095SBarry Smith        0,
149397304618SKris Buschelman /*10*/ 0,
149409dc0095SBarry Smith        0,
149509dc0095SBarry Smith        0,
149609dc0095SBarry Smith        0,
149709dc0095SBarry Smith        MatTranspose_MPIDense,
149897304618SKris Buschelman /*15*/ MatGetInfo_MPIDense,
14996e4ee0c6SHong Zhang        MatEqual_MPIDense,
150009dc0095SBarry Smith        MatGetDiagonal_MPIDense,
15015b2fa520SLois Curfman McInnes        MatDiagonalScale_MPIDense,
150209dc0095SBarry Smith        MatNorm_MPIDense,
150397304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense,
150409dc0095SBarry Smith        MatAssemblyEnd_MPIDense,
150509dc0095SBarry Smith        MatSetOption_MPIDense,
150609dc0095SBarry Smith        MatZeroEntries_MPIDense,
1507d519adbfSMatthew Knepley /*24*/ MatZeroRows_MPIDense,
1508919b68f7SBarry Smith        0,
150901b82886SBarry Smith        0,
151001b82886SBarry Smith        0,
151101b82886SBarry Smith        0,
1512d519adbfSMatthew Knepley /*29*/ MatSetUpPreallocation_MPIDense,
1513273d9f13SBarry Smith        0,
151409dc0095SBarry Smith        0,
151509dc0095SBarry Smith        MatGetArray_MPIDense,
151609dc0095SBarry Smith        MatRestoreArray_MPIDense,
1517d519adbfSMatthew Knepley /*34*/ MatDuplicate_MPIDense,
151809dc0095SBarry Smith        0,
151909dc0095SBarry Smith        0,
152009dc0095SBarry Smith        0,
152109dc0095SBarry Smith        0,
1522d519adbfSMatthew Knepley /*39*/ MatAXPY_MPIDense,
15232ce60cd0SSatish Balay        MatGetSubMatrices_MPIDense,
152409dc0095SBarry Smith        0,
152509dc0095SBarry Smith        MatGetValues_MPIDense,
152609dc0095SBarry Smith        0,
1527d519adbfSMatthew Knepley /*44*/ 0,
152809dc0095SBarry Smith        MatScale_MPIDense,
152909dc0095SBarry Smith        0,
153009dc0095SBarry Smith        0,
153109dc0095SBarry Smith        0,
1532d519adbfSMatthew Knepley /*49*/ 0,
153309dc0095SBarry Smith        0,
153409dc0095SBarry Smith        0,
153509dc0095SBarry Smith        0,
153609dc0095SBarry Smith        0,
1537d519adbfSMatthew Knepley /*54*/ 0,
153809dc0095SBarry Smith        0,
153909dc0095SBarry Smith        0,
154009dc0095SBarry Smith        0,
154109dc0095SBarry Smith        0,
1542d519adbfSMatthew Knepley /*59*/ MatGetSubMatrix_MPIDense,
1543b9b97703SBarry Smith        MatDestroy_MPIDense,
1544b9b97703SBarry Smith        MatView_MPIDense,
1545357abbc8SBarry Smith        0,
154697304618SKris Buschelman        0,
1547d519adbfSMatthew Knepley /*64*/ 0,
154897304618SKris Buschelman        0,
154997304618SKris Buschelman        0,
155097304618SKris Buschelman        0,
155197304618SKris Buschelman        0,
1552d519adbfSMatthew Knepley /*69*/ 0,
155397304618SKris Buschelman        0,
155497304618SKris Buschelman        0,
155597304618SKris Buschelman        0,
155697304618SKris Buschelman        0,
1557d519adbfSMatthew Knepley /*74*/ 0,
155897304618SKris Buschelman        0,
155997304618SKris Buschelman        0,
156097304618SKris Buschelman        0,
156197304618SKris Buschelman        0,
1562d519adbfSMatthew Knepley /*79*/ 0,
156397304618SKris Buschelman        0,
156497304618SKris Buschelman        0,
156597304618SKris Buschelman        0,
1566d519adbfSMatthew Knepley /*83*/ MatLoad_MPIDense,
1567865e5f61SKris Buschelman        0,
1568865e5f61SKris Buschelman        0,
1569865e5f61SKris Buschelman        0,
1570865e5f61SKris Buschelman        0,
1571865e5f61SKris Buschelman        0,
1572d519adbfSMatthew Knepley /*89*/
15732fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
15742fbe02b9SBarry Smith        MatMatMult_MPIDense_MPIDense,
15754ae313f4SHong Zhang        MatMatMultSymbolic_MPIDense_MPIDense,
15762fbe02b9SBarry Smith        MatMatMultNumeric_MPIDense_MPIDense,
15772fbe02b9SBarry Smith #else
1578865e5f61SKris Buschelman        0,
1579865e5f61SKris Buschelman        0,
1580865e5f61SKris Buschelman        0,
15812fbe02b9SBarry Smith #endif
15822fbe02b9SBarry Smith        0,
1583*ba337c44SJed Brown        0,
1584d519adbfSMatthew Knepley /*94*/ 0,
1585865e5f61SKris Buschelman        0,
1586865e5f61SKris Buschelman        0,
1587*ba337c44SJed Brown        0,
1588*ba337c44SJed Brown        0,
1589*ba337c44SJed Brown /*99*/ 0,
1590*ba337c44SJed Brown        0,
1591*ba337c44SJed Brown        0,
1592*ba337c44SJed Brown        MatConjugate_MPIDense,
1593*ba337c44SJed Brown        0,
1594*ba337c44SJed Brown /*104*/0,
1595*ba337c44SJed Brown        MatRealPart_MPIDense,
1596*ba337c44SJed Brown        MatImaginaryPart_MPIDense,
1597*ba337c44SJed Brown        0
1598*ba337c44SJed Brown };
15998965ea79SLois Curfman McInnes 
1600273d9f13SBarry Smith EXTERN_C_BEGIN
16014a2ae208SSatish Balay #undef __FUNCT__
1602a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense"
1603be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data)
1604a23d5eceSKris Buschelman {
1605a23d5eceSKris Buschelman   Mat_MPIDense   *a;
1606dfbe8321SBarry Smith   PetscErrorCode ierr;
1607a23d5eceSKris Buschelman 
1608a23d5eceSKris Buschelman   PetscFunctionBegin;
1609a23d5eceSKris Buschelman   mat->preallocated = PETSC_TRUE;
1610a23d5eceSKris Buschelman   /* Note:  For now, when data is specified above, this assumes the user correctly
1611a23d5eceSKris Buschelman    allocates the local dense storage space.  We should add error checking. */
1612a23d5eceSKris Buschelman 
1613a23d5eceSKris Buschelman   a    = (Mat_MPIDense*)mat->data;
1614f69a0ea3SMatthew Knepley   ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr);
1615d0f46423SBarry Smith   ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr);
16165c5985e7SKris Buschelman   ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr);
16175c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr);
161852e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
1619a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1620a23d5eceSKris Buschelman }
1621a23d5eceSKris Buschelman EXTERN_C_END
1622a23d5eceSKris Buschelman 
16230bad9183SKris Buschelman /*MC
162441c8de11SBarry Smith    MAT_SOLVER_PLAPACK = "mpidense" - Parallel LU and Cholesky factorization for MATMPIDENSE matrices
16250bad9183SKris Buschelman 
162641c8de11SBarry Smith   run config/configure.py with the option --download-plapack
16277878bbefSBarry Smith 
16287878bbefSBarry Smith 
16297878bbefSBarry Smith   Options Database Keys:
16307878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition
16317878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition
16327878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector
16337878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size
16347878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag
16357878bbefSBarry Smith 
163641c8de11SBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE, PCFactorSetSolverPackage(), MatSolverPackage
163741c8de11SBarry Smith 
16380bad9183SKris Buschelman M*/
16390bad9183SKris Buschelman 
1640a23d5eceSKris Buschelman EXTERN_C_BEGIN
1641a23d5eceSKris Buschelman #undef __FUNCT__
16424a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense"
1643be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat)
1644273d9f13SBarry Smith {
1645273d9f13SBarry Smith   Mat_MPIDense   *a;
1646dfbe8321SBarry Smith   PetscErrorCode ierr;
1647273d9f13SBarry Smith 
1648273d9f13SBarry Smith   PetscFunctionBegin;
164938f2d2fdSLisandro Dalcin   ierr              = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr);
1650b0a32e0cSBarry Smith   mat->data         = (void*)a;
1651273d9f13SBarry Smith   ierr              = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1652273d9f13SBarry Smith   mat->mapping      = 0;
1653273d9f13SBarry Smith 
1654273d9f13SBarry Smith   mat->insertmode = NOT_SET_VALUES;
16557adad957SLisandro Dalcin   ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr);
16567adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr);
1657273d9f13SBarry Smith 
165826283091SBarry Smith   ierr = PetscLayoutSetBlockSize(mat->rmap,1);CHKERRQ(ierr);
165926283091SBarry Smith   ierr = PetscLayoutSetBlockSize(mat->cmap,1);CHKERRQ(ierr);
166026283091SBarry Smith   ierr = PetscLayoutSetUp(mat->rmap);CHKERRQ(ierr);
166126283091SBarry Smith   ierr = PetscLayoutSetUp(mat->cmap);CHKERRQ(ierr);
1662d0f46423SBarry Smith   a->nvec = mat->cmap->n;
1663273d9f13SBarry Smith 
1664273d9f13SBarry Smith   /* build cache for off array entries formed */
1665273d9f13SBarry Smith   a->donotstash = PETSC_FALSE;
16667adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr);
1667273d9f13SBarry Smith 
1668273d9f13SBarry Smith   /* stuff used for matrix vector multiply */
1669273d9f13SBarry Smith   a->lvec        = 0;
1670273d9f13SBarry Smith   a->Mvctx       = 0;
1671273d9f13SBarry Smith   a->roworiented = PETSC_TRUE;
1672273d9f13SBarry Smith 
1673273d9f13SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C",
1674273d9f13SBarry Smith                                      "MatGetDiagonalBlock_MPIDense",
1675273d9f13SBarry Smith                                      MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr);
1676a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C",
1677a23d5eceSKris Buschelman                                      "MatMPIDenseSetPreallocation_MPIDense",
1678a23d5eceSKris Buschelman                                      MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr);
16794ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C",
16804ae313f4SHong Zhang                                      "MatMatMult_MPIAIJ_MPIDense",
16814ae313f4SHong Zhang                                       MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr);
16824ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C",
16834ae313f4SHong Zhang                                      "MatMatMultSymbolic_MPIAIJ_MPIDense",
16844ae313f4SHong Zhang                                       MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr);
16854ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C",
16864ae313f4SHong Zhang                                      "MatMatMultNumeric_MPIAIJ_MPIDense",
16874ae313f4SHong Zhang                                       MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr);
1688ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_petsc_C",
168930716080SHong Zhang                                      "MatGetFactor_mpidense_petsc",
169030716080SHong Zhang                                       MatGetFactor_mpidense_petsc);CHKERRQ(ierr);
16917878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1692ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_plapack_C",
1693b6806ab0SHong Zhang                                      "MatGetFactor_mpidense_plapack",
1694b6806ab0SHong Zhang                                       MatGetFactor_mpidense_plapack);CHKERRQ(ierr);
1695db4efbfdSBarry Smith   ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr);
1696db4efbfdSBarry Smith #endif
169738aed534SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr);
169801b82886SBarry Smith 
1699273d9f13SBarry Smith   PetscFunctionReturn(0);
1700273d9f13SBarry Smith }
1701273d9f13SBarry Smith EXTERN_C_END
1702273d9f13SBarry Smith 
1703209238afSKris Buschelman /*MC
1704002d173eSKris Buschelman    MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices.
1705209238afSKris Buschelman 
1706209238afSKris Buschelman    This matrix type is identical to MATSEQDENSE when constructed with a single process communicator,
1707209238afSKris Buschelman    and MATMPIDENSE otherwise.
1708209238afSKris Buschelman 
1709209238afSKris Buschelman    Options Database Keys:
1710209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions()
1711209238afSKris Buschelman 
1712209238afSKris Buschelman   Level: beginner
1713209238afSKris Buschelman 
171401b82886SBarry Smith 
1715209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE
1716209238afSKris Buschelman M*/
1717209238afSKris Buschelman 
1718209238afSKris Buschelman EXTERN_C_BEGIN
1719209238afSKris Buschelman #undef __FUNCT__
1720209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense"
1721be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A)
1722dfbe8321SBarry Smith {
17236849ba73SBarry Smith   PetscErrorCode ierr;
172413f74950SBarry Smith   PetscMPIInt    size;
1725209238afSKris Buschelman 
1726209238afSKris Buschelman   PetscFunctionBegin;
17277adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1728209238afSKris Buschelman   if (size == 1) {
1729209238afSKris Buschelman     ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr);
1730209238afSKris Buschelman   } else {
1731209238afSKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
1732209238afSKris Buschelman   }
1733209238afSKris Buschelman   PetscFunctionReturn(0);
1734209238afSKris Buschelman }
1735209238afSKris Buschelman EXTERN_C_END
1736209238afSKris Buschelman 
17374a2ae208SSatish Balay #undef __FUNCT__
17384a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation"
1739273d9f13SBarry Smith /*@C
1740273d9f13SBarry Smith    MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries
1741273d9f13SBarry Smith 
1742273d9f13SBarry Smith    Not collective
1743273d9f13SBarry Smith 
1744273d9f13SBarry Smith    Input Parameters:
1745273d9f13SBarry Smith .  A - the matrix
1746273d9f13SBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1747273d9f13SBarry Smith    to control all matrix memory allocation.
1748273d9f13SBarry Smith 
1749273d9f13SBarry Smith    Notes:
1750273d9f13SBarry Smith    The dense format is fully compatible with standard Fortran 77
1751273d9f13SBarry Smith    storage by columns.
1752273d9f13SBarry Smith 
1753273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1754273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1755273d9f13SBarry Smith    set data=PETSC_NULL.
1756273d9f13SBarry Smith 
1757273d9f13SBarry Smith    Level: intermediate
1758273d9f13SBarry Smith 
1759273d9f13SBarry Smith .keywords: matrix,dense, parallel
1760273d9f13SBarry Smith 
1761273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
1762273d9f13SBarry Smith @*/
1763be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data)
1764273d9f13SBarry Smith {
1765dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar *);
1766273d9f13SBarry Smith 
1767273d9f13SBarry Smith   PetscFunctionBegin;
1768565567f0SKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1769a23d5eceSKris Buschelman   if (f) {
1770a23d5eceSKris Buschelman     ierr = (*f)(mat,data);CHKERRQ(ierr);
1771a23d5eceSKris Buschelman   }
1772273d9f13SBarry Smith   PetscFunctionReturn(0);
1773273d9f13SBarry Smith }
1774273d9f13SBarry Smith 
17754a2ae208SSatish Balay #undef __FUNCT__
17764a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense"
17778965ea79SLois Curfman McInnes /*@C
177839ddd567SLois Curfman McInnes    MatCreateMPIDense - Creates a sparse parallel matrix in dense format.
17798965ea79SLois Curfman McInnes 
1780db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1781db81eaa0SLois Curfman McInnes 
17828965ea79SLois Curfman McInnes    Input Parameters:
1783db81eaa0SLois Curfman McInnes +  comm - MPI communicator
17848965ea79SLois Curfman McInnes .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
1785db81eaa0SLois Curfman McInnes .  n - number of local columns (or PETSC_DECIDE to have calculated if N is given)
17868965ea79SLois Curfman McInnes .  M - number of global rows (or PETSC_DECIDE to have calculated if m is given)
1787db81eaa0SLois Curfman McInnes .  N - number of global columns (or PETSC_DECIDE to have calculated if n is given)
17887f5ff6fdSBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc
1789dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
17908965ea79SLois Curfman McInnes 
17918965ea79SLois Curfman McInnes    Output Parameter:
1792477f1c0bSLois Curfman McInnes .  A - the matrix
17938965ea79SLois Curfman McInnes 
1794b259b22eSLois Curfman McInnes    Notes:
179539ddd567SLois Curfman McInnes    The dense format is fully compatible with standard Fortran 77
179639ddd567SLois Curfman McInnes    storage by columns.
17978965ea79SLois Curfman McInnes 
179818f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
179918f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
18007f5ff6fdSBarry Smith    set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users).
180118f449edSLois Curfman McInnes 
18028965ea79SLois Curfman McInnes    The user MUST specify either the local or global matrix dimensions
18038965ea79SLois Curfman McInnes    (possibly both).
18048965ea79SLois Curfman McInnes 
1805027ccd11SLois Curfman McInnes    Level: intermediate
1806027ccd11SLois Curfman McInnes 
180739ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel
18088965ea79SLois Curfman McInnes 
180939ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
18108965ea79SLois Curfman McInnes @*/
1811be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A)
18128965ea79SLois Curfman McInnes {
18136849ba73SBarry Smith   PetscErrorCode ierr;
181413f74950SBarry Smith   PetscMPIInt    size;
18158965ea79SLois Curfman McInnes 
18163a40ed3dSBarry Smith   PetscFunctionBegin;
1817f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1818f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr);
1819273d9f13SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1820273d9f13SBarry Smith   if (size > 1) {
1821273d9f13SBarry Smith     ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr);
1822273d9f13SBarry Smith     ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1823273d9f13SBarry Smith   } else {
1824273d9f13SBarry Smith     ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1825273d9f13SBarry Smith     ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
18268c469469SLois Curfman McInnes   }
18273a40ed3dSBarry Smith   PetscFunctionReturn(0);
18288965ea79SLois Curfman McInnes }
18298965ea79SLois Curfman McInnes 
18304a2ae208SSatish Balay #undef __FUNCT__
18314a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense"
18326849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
18338965ea79SLois Curfman McInnes {
18348965ea79SLois Curfman McInnes   Mat            mat;
18353501a2bdSLois Curfman McInnes   Mat_MPIDense   *a,*oldmat = (Mat_MPIDense*)A->data;
1836dfbe8321SBarry Smith   PetscErrorCode ierr;
18378965ea79SLois Curfman McInnes 
18383a40ed3dSBarry Smith   PetscFunctionBegin;
18398965ea79SLois Curfman McInnes   *newmat       = 0;
18407adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr);
1841d0f46423SBarry Smith   ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
18427adad957SLisandro Dalcin   ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr);
1843834f8fabSBarry Smith   a                 = (Mat_MPIDense*)mat->data;
1844e04c1aa4SHong Zhang   ierr              = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
18453501a2bdSLois Curfman McInnes   mat->factor       = A->factor;
1846c456f294SBarry Smith   mat->assembled    = PETSC_TRUE;
1847273d9f13SBarry Smith   mat->preallocated = PETSC_TRUE;
18488965ea79SLois Curfman McInnes 
1849d0f46423SBarry Smith   mat->rmap->rstart     = A->rmap->rstart;
1850d0f46423SBarry Smith   mat->rmap->rend       = A->rmap->rend;
18518965ea79SLois Curfman McInnes   a->size              = oldmat->size;
18528965ea79SLois Curfman McInnes   a->rank              = oldmat->rank;
1853e0fa3b82SLois Curfman McInnes   mat->insertmode      = NOT_SET_VALUES;
1854b9b97703SBarry Smith   a->nvec              = oldmat->nvec;
18553782ba37SSatish Balay   a->donotstash        = oldmat->donotstash;
1856e04c1aa4SHong Zhang 
1857d0f46423SBarry Smith   ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
1858d0f46423SBarry Smith   ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
18598965ea79SLois Curfman McInnes 
1860329f5518SBarry Smith   ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
18615609ef8eSBarry Smith   ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr);
186252e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
186301b82886SBarry Smith 
18648965ea79SLois Curfman McInnes   *newmat = mat;
18653a40ed3dSBarry Smith   PetscFunctionReturn(0);
18668965ea79SLois Curfman McInnes }
18678965ea79SLois Curfman McInnes 
1868e090d566SSatish Balay #include "petscsys.h"
18698965ea79SLois Curfman McInnes 
18704a2ae208SSatish Balay #undef __FUNCT__
18714a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile"
1872a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat)
187390ace30eSBarry Smith {
18746849ba73SBarry Smith   PetscErrorCode ierr;
187532dcc486SBarry Smith   PetscMPIInt    rank,size;
187613f74950SBarry Smith   PetscInt       *rowners,i,m,nz,j;
187787828ca2SBarry Smith   PetscScalar    *array,*vals,*vals_ptr;
187890ace30eSBarry Smith   MPI_Status     status;
187990ace30eSBarry Smith 
18803a40ed3dSBarry Smith   PetscFunctionBegin;
1881d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
1882d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
188390ace30eSBarry Smith 
188490ace30eSBarry Smith   /* determine ownership of all rows */
188590ace30eSBarry Smith   m          = M/size + ((M % size) > rank);
188613f74950SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr);
188713f74950SBarry Smith   ierr       = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr);
188890ace30eSBarry Smith   rowners[0] = 0;
188990ace30eSBarry Smith   for (i=2; i<=size; i++) {
189090ace30eSBarry Smith     rowners[i] += rowners[i-1];
189190ace30eSBarry Smith   }
189290ace30eSBarry Smith 
1893f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1894f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1895be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1896878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
189790ace30eSBarry Smith   ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr);
189890ace30eSBarry Smith 
189990ace30eSBarry Smith   if (!rank) {
190087828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
190190ace30eSBarry Smith 
190290ace30eSBarry Smith     /* read in my part of the matrix numerical values  */
19030752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr);
190490ace30eSBarry Smith 
190590ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
190690ace30eSBarry Smith     vals_ptr = vals;
190790ace30eSBarry Smith     for (i=0; i<m; i++) {
190890ace30eSBarry Smith       for (j=0; j<N; j++) {
190990ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
191090ace30eSBarry Smith       }
191190ace30eSBarry Smith     }
191290ace30eSBarry Smith 
191390ace30eSBarry Smith     /* read in other processors and ship out */
191490ace30eSBarry Smith     for (i=1; i<size; i++) {
191590ace30eSBarry Smith       nz   = (rowners[i+1] - rowners[i])*N;
19160752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
19177adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr);
191890ace30eSBarry Smith     }
19193a40ed3dSBarry Smith   } else {
192090ace30eSBarry Smith     /* receive numeric values */
192187828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
192290ace30eSBarry Smith 
192390ace30eSBarry Smith     /* receive message of values*/
19247adad957SLisandro Dalcin     ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr);
192590ace30eSBarry Smith 
192690ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
192790ace30eSBarry Smith     vals_ptr = vals;
192890ace30eSBarry Smith     for (i=0; i<m; i++) {
192990ace30eSBarry Smith       for (j=0; j<N; j++) {
193090ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
193190ace30eSBarry Smith       }
193290ace30eSBarry Smith     }
193390ace30eSBarry Smith   }
1934606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
1935606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
19366d4a8577SBarry Smith   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19376d4a8577SBarry Smith   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19383a40ed3dSBarry Smith   PetscFunctionReturn(0);
193990ace30eSBarry Smith }
194090ace30eSBarry Smith 
19414a2ae208SSatish Balay #undef __FUNCT__
19424a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense"
1943a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat)
19448965ea79SLois Curfman McInnes {
19458965ea79SLois Curfman McInnes   Mat            A;
194687828ca2SBarry Smith   PetscScalar    *vals,*svals;
194719bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
19488965ea79SLois Curfman McInnes   MPI_Status     status;
194913f74950SBarry Smith   PetscMPIInt    rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz;
195013f74950SBarry Smith   PetscInt       header[4],*rowlengths = 0,M,N,*cols;
195113f74950SBarry Smith   PetscInt       *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols;
195213f74950SBarry Smith   PetscInt       i,nz,j,rstart,rend;
195313f74950SBarry Smith   int            fd;
19546849ba73SBarry Smith   PetscErrorCode ierr;
19558965ea79SLois Curfman McInnes 
19563a40ed3dSBarry Smith   PetscFunctionBegin;
1957d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1958d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
19598965ea79SLois Curfman McInnes   if (!rank) {
1960b0a32e0cSBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
19610752156aSBarry Smith     ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr);
1962552e946dSBarry Smith     if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object");
19638965ea79SLois Curfman McInnes   }
19648965ea79SLois Curfman McInnes 
196513f74950SBarry Smith   ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr);
196690ace30eSBarry Smith   M = header[1]; N = header[2]; nz = header[3];
196790ace30eSBarry Smith 
196890ace30eSBarry Smith   /*
196990ace30eSBarry Smith        Handle case where matrix is stored on disk as a dense matrix
197090ace30eSBarry Smith   */
197190ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) {
1972be5d1d56SKris Buschelman     ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr);
19733a40ed3dSBarry Smith     PetscFunctionReturn(0);
197490ace30eSBarry Smith   }
197590ace30eSBarry Smith 
19768965ea79SLois Curfman McInnes   /* determine ownership of all rows */
1977e44c0bd4SBarry Smith   m          = PetscMPIIntCast(M/size + ((M % size) > rank));
1978e44c0bd4SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr);
1979ca161407SBarry Smith   ierr       = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr);
19808965ea79SLois Curfman McInnes   rowners[0] = 0;
19818965ea79SLois Curfman McInnes   for (i=2; i<=size; i++) {
19828965ea79SLois Curfman McInnes     rowners[i] += rowners[i-1];
19838965ea79SLois Curfman McInnes   }
19848965ea79SLois Curfman McInnes   rstart = rowners[rank];
19858965ea79SLois Curfman McInnes   rend   = rowners[rank+1];
19868965ea79SLois Curfman McInnes 
19878965ea79SLois Curfman McInnes   /* distribute row lengths to all processors */
198874ed9c26SBarry Smith   ierr    = PetscMalloc2(rend-rstart,PetscInt,&ourlens,rend-rstart,PetscInt,&offlens);CHKERRQ(ierr);
19898965ea79SLois Curfman McInnes   if (!rank) {
199013f74950SBarry Smith     ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
19910752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
199213f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr);
19938965ea79SLois Curfman McInnes     for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i];
19941466f1e1SBarry Smith     ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
1995606d414cSSatish Balay     ierr = PetscFree(sndcounts);CHKERRQ(ierr);
1996ca161407SBarry Smith   } else {
19971466f1e1SBarry Smith     ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
19988965ea79SLois Curfman McInnes   }
19998965ea79SLois Curfman McInnes 
20008965ea79SLois Curfman McInnes   if (!rank) {
20018965ea79SLois Curfman McInnes     /* calculate the number of nonzeros on each processor */
200213f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr);
200313f74950SBarry Smith     ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr);
20048965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
20058965ea79SLois Curfman McInnes       for (j=rowners[i]; j< rowners[i+1]; j++) {
20068965ea79SLois Curfman McInnes         procsnz[i] += rowlengths[j];
20078965ea79SLois Curfman McInnes       }
20088965ea79SLois Curfman McInnes     }
2009606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
20108965ea79SLois Curfman McInnes 
20118965ea79SLois Curfman McInnes     /* determine max buffer needed and allocate it */
20128965ea79SLois Curfman McInnes     maxnz = 0;
20138965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
20140452661fSBarry Smith       maxnz = PetscMax(maxnz,procsnz[i]);
20158965ea79SLois Curfman McInnes     }
201613f74950SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr);
20178965ea79SLois Curfman McInnes 
20188965ea79SLois Curfman McInnes     /* read in my part of the matrix column indices  */
20198965ea79SLois Curfman McInnes     nz = procsnz[0];
202013f74950SBarry Smith     ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
20210752156aSBarry Smith     ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr);
20228965ea79SLois Curfman McInnes 
20238965ea79SLois Curfman McInnes     /* read in every one elses and ship off */
20248965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
20258965ea79SLois Curfman McInnes       nz   = procsnz[i];
20260752156aSBarry Smith       ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
202713f74950SBarry Smith       ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr);
20288965ea79SLois Curfman McInnes     }
2029606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
20303a40ed3dSBarry Smith   } else {
20318965ea79SLois Curfman McInnes     /* determine buffer space needed for message */
20328965ea79SLois Curfman McInnes     nz = 0;
20338965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20348965ea79SLois Curfman McInnes       nz += ourlens[i];
20358965ea79SLois Curfman McInnes     }
203613f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
20378965ea79SLois Curfman McInnes 
20388965ea79SLois Curfman McInnes     /* receive message of column indices*/
203913f74950SBarry Smith     ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr);
204013f74950SBarry Smith     ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr);
204129bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
20428965ea79SLois Curfman McInnes   }
20438965ea79SLois Curfman McInnes 
20448965ea79SLois Curfman McInnes   /* loop over local rows, determining number of off diagonal entries */
204513f74950SBarry Smith   ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr);
20468965ea79SLois Curfman McInnes   jj = 0;
20478965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
20488965ea79SLois Curfman McInnes     for (j=0; j<ourlens[i]; j++) {
20498965ea79SLois Curfman McInnes       if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++;
20508965ea79SLois Curfman McInnes       jj++;
20518965ea79SLois Curfman McInnes     }
20528965ea79SLois Curfman McInnes   }
20538965ea79SLois Curfman McInnes 
20548965ea79SLois Curfman McInnes   /* create our matrix */
20558965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
20568965ea79SLois Curfman McInnes     ourlens[i] -= offlens[i];
20578965ea79SLois Curfman McInnes   }
2058f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
2059f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
2060be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
2061878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
20628965ea79SLois Curfman McInnes   A = *newmat;
20638965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
20648965ea79SLois Curfman McInnes     ourlens[i] += offlens[i];
20658965ea79SLois Curfman McInnes   }
20668965ea79SLois Curfman McInnes 
20678965ea79SLois Curfman McInnes   if (!rank) {
206887828ca2SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20698965ea79SLois Curfman McInnes 
20708965ea79SLois Curfman McInnes     /* read in my part of the matrix numerical values  */
20718965ea79SLois Curfman McInnes     nz = procsnz[0];
20720752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20738965ea79SLois Curfman McInnes 
20748965ea79SLois Curfman McInnes     /* insert into matrix */
20758965ea79SLois Curfman McInnes     jj      = rstart;
20768965ea79SLois Curfman McInnes     smycols = mycols;
20778965ea79SLois Curfman McInnes     svals   = vals;
20788965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
20798965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
20808965ea79SLois Curfman McInnes       smycols += ourlens[i];
20818965ea79SLois Curfman McInnes       svals   += ourlens[i];
20828965ea79SLois Curfman McInnes       jj++;
20838965ea79SLois Curfman McInnes     }
20848965ea79SLois Curfman McInnes 
20858965ea79SLois Curfman McInnes     /* read in other processors and ship out */
20868965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
20878965ea79SLois Curfman McInnes       nz   = procsnz[i];
20880752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
20897adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr);
20908965ea79SLois Curfman McInnes     }
2091606d414cSSatish Balay     ierr = PetscFree(procsnz);CHKERRQ(ierr);
20923a40ed3dSBarry Smith   } else {
20938965ea79SLois Curfman McInnes     /* receive numeric values */
209484d528b1SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
20958965ea79SLois Curfman McInnes 
20968965ea79SLois Curfman McInnes     /* receive message of values*/
20977adad957SLisandro Dalcin     ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr);
2098ca161407SBarry Smith     ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr);
209929bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
21008965ea79SLois Curfman McInnes 
21018965ea79SLois Curfman McInnes     /* insert into matrix */
21028965ea79SLois Curfman McInnes     jj      = rstart;
21038965ea79SLois Curfman McInnes     smycols = mycols;
21048965ea79SLois Curfman McInnes     svals   = vals;
21058965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
21068965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
21078965ea79SLois Curfman McInnes       smycols += ourlens[i];
21088965ea79SLois Curfman McInnes       svals   += ourlens[i];
21098965ea79SLois Curfman McInnes       jj++;
21108965ea79SLois Curfman McInnes     }
21118965ea79SLois Curfman McInnes   }
211274ed9c26SBarry Smith   ierr = PetscFree2(ourlens,offlens);CHKERRQ(ierr);
2113606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
2114606d414cSSatish Balay   ierr = PetscFree(mycols);CHKERRQ(ierr);
2115606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
21168965ea79SLois Curfman McInnes 
21176d4a8577SBarry Smith   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
21186d4a8577SBarry Smith   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
21193a40ed3dSBarry Smith   PetscFunctionReturn(0);
21208965ea79SLois Curfman McInnes }
212190ace30eSBarry Smith 
21226e4ee0c6SHong Zhang #undef __FUNCT__
21236e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense"
21246e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag)
21256e4ee0c6SHong Zhang {
21266e4ee0c6SHong Zhang   Mat_MPIDense   *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data;
21276e4ee0c6SHong Zhang   Mat            a,b;
21286e4ee0c6SHong Zhang   PetscTruth     flg;
21296e4ee0c6SHong Zhang   PetscErrorCode ierr;
213090ace30eSBarry Smith 
21316e4ee0c6SHong Zhang   PetscFunctionBegin;
21326e4ee0c6SHong Zhang   a = matA->A;
21336e4ee0c6SHong Zhang   b = matB->A;
21346e4ee0c6SHong Zhang   ierr = MatEqual(a,b,&flg);CHKERRQ(ierr);
21357adad957SLisandro Dalcin   ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr);
21366e4ee0c6SHong Zhang   PetscFunctionReturn(0);
21376e4ee0c6SHong Zhang }
213890ace30eSBarry Smith 
213909d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
214009d27a7eSBarry Smith 
214109d27a7eSBarry Smith #undef __FUNCT__
214209d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage"
214309d27a7eSBarry Smith /*@C
214409d27a7eSBarry Smith   PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK.
214509d27a7eSBarry Smith   Level: developer
214609d27a7eSBarry Smith 
214709d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK
214809d27a7eSBarry Smith .seealso: PetscFinalize()
214909d27a7eSBarry Smith @*/
215009d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void)
215109d27a7eSBarry Smith {
215209d27a7eSBarry Smith   PetscErrorCode ierr;
215309d27a7eSBarry Smith 
215409d27a7eSBarry Smith   PetscFunctionBegin;
215509d27a7eSBarry Smith   ierr = PLA_Finalize();CHKERRQ(ierr);
215609d27a7eSBarry Smith   PetscFunctionReturn(0);
215709d27a7eSBarry Smith }
215809d27a7eSBarry Smith 
215909d27a7eSBarry Smith #undef __FUNCT__
216009d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage"
216109d27a7eSBarry Smith /*@C
216209d27a7eSBarry Smith   PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is
2163db4efbfdSBarry Smith   called from MatCreate_MPIDense() the first time an MPI dense matrix is called.
216409d27a7eSBarry Smith 
216509d27a7eSBarry Smith   Input Parameter:
2166db4efbfdSBarry Smith .   comm - the communicator the matrix lives on
216709d27a7eSBarry Smith 
216809d27a7eSBarry Smith   Level: developer
216909d27a7eSBarry Smith 
2170db4efbfdSBarry Smith    Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the
2171db4efbfdSBarry Smith   same communicator (because there is some global state that is initialized and used for all matrices). In addition if
2172db4efbfdSBarry Smith   PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators
2173db4efbfdSBarry Smith   cannot overlap.
2174db4efbfdSBarry Smith 
217509d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK
217609d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize()
217709d27a7eSBarry Smith @*/
2178db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm)
217909d27a7eSBarry Smith {
2180eae6fb2eSBarry Smith   PetscMPIInt    size;
218109d27a7eSBarry Smith   PetscErrorCode ierr;
218209d27a7eSBarry Smith 
218309d27a7eSBarry Smith   PetscFunctionBegin;
218409d27a7eSBarry Smith   if (!PLA_Initialized(PETSC_NULL)) {
218509d27a7eSBarry Smith 
218609d27a7eSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
218704fea9ffSBarry Smith     Plapack_nprows = 1;
218804fea9ffSBarry Smith     Plapack_npcols = size;
218909d27a7eSBarry Smith 
2190aeccfd6fSBarry Smith     ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr);
21915d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr);
21925d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr);
219379b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG)
219479b0a62dSBarry Smith       Plapack_ierror = 3;
219579b0a62dSBarry Smith #else
219679b0a62dSBarry Smith       Plapack_ierror = 0;
219779b0a62dSBarry Smith #endif
21985d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr);
2199eae6fb2eSBarry Smith       if (Plapack_ierror){
2200eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr);
2201aeccfd6fSBarry Smith       } else {
2202eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr);
2203aeccfd6fSBarry Smith       }
2204aeccfd6fSBarry Smith 
2205eae6fb2eSBarry Smith       Plapack_nb_alg = 0;
22065d00a290SHong Zhang       ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr);
2207eae6fb2eSBarry Smith       if (Plapack_nb_alg) {
2208eae6fb2eSBarry Smith         ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr);
2209aeccfd6fSBarry Smith       }
2210aeccfd6fSBarry Smith     PetscOptionsEnd();
2211aeccfd6fSBarry Smith 
221204fea9ffSBarry Smith     ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr);
221304fea9ffSBarry Smith     ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr);
221409d27a7eSBarry Smith     ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr);
221509d27a7eSBarry Smith   }
221609d27a7eSBarry Smith   PetscFunctionReturn(0);
221709d27a7eSBarry Smith }
221890ace30eSBarry Smith 
221909d27a7eSBarry Smith #endif
2220