xref: /petsc/src/mat/impls/dense/mpi/mpidense.c (revision ab235cb6cf7d025c14f3fc9704eaf5364952fc4b)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2be1d678aSKris Buschelman 
3ed3cc1f0SBarry Smith /*
4ed3cc1f0SBarry Smith    Basic functions for basic parallel dense matrices.
5ed3cc1f0SBarry Smith */
6ed3cc1f0SBarry Smith 
704fea9ffSBarry Smith 
889ac3759SBarry Smith #include "src/mat/impls/dense/mpi/mpidense.h"    /*I   "petscmat.h"  I*/
9eae6fb2eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
10eae6fb2eSBarry Smith static PetscMPIInt Plapack_nprows,Plapack_npcols,Plapack_ierror,Plapack_nb_alg;
11eae6fb2eSBarry Smith static MPI_Comm Plapack_comm_2d;
12eae6fb2eSBarry Smith #endif
138965ea79SLois Curfman McInnes 
14ba8c8a56SBarry Smith #undef __FUNCT__
15ab92ecdeSBarry Smith #define __FUNCT__ "MatDenseGetLocalMatrix"
16ab92ecdeSBarry Smith /*@
17ab92ecdeSBarry Smith 
18ab92ecdeSBarry Smith       MatDenseGetLocalMatrix - For a MATMPIDENSE or MATSEQDENSE matrix returns the sequential
19ab92ecdeSBarry Smith               matrix that represents the operator. For sequential matrices it returns itself.
20ab92ecdeSBarry Smith 
21ab92ecdeSBarry Smith     Input Parameter:
22ab92ecdeSBarry Smith .      A - the Seq or MPI dense matrix
23ab92ecdeSBarry Smith 
24ab92ecdeSBarry Smith     Output Parameter:
25ab92ecdeSBarry Smith .      B - the inner matrix
26ab92ecdeSBarry Smith 
278e6c10adSSatish Balay     Level: intermediate
288e6c10adSSatish Balay 
29ab92ecdeSBarry Smith @*/
30ab92ecdeSBarry Smith PetscErrorCode MatDenseGetLocalMatrix(Mat A,Mat *B)
31ab92ecdeSBarry Smith {
32ab92ecdeSBarry Smith   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
33ab92ecdeSBarry Smith   PetscErrorCode ierr;
34ab92ecdeSBarry Smith   PetscTruth     flg;
35ab92ecdeSBarry Smith 
36ab92ecdeSBarry Smith   PetscFunctionBegin;
37ab92ecdeSBarry Smith   ierr = PetscTypeCompare((PetscObject)A,MATMPIDENSE,&flg);CHKERRQ(ierr);
38ab92ecdeSBarry Smith   if (flg) {
39ab92ecdeSBarry Smith     *B = mat->A;
40ab92ecdeSBarry Smith   } else {
41ab92ecdeSBarry Smith     *B = A;
42ab92ecdeSBarry Smith   }
43ab92ecdeSBarry Smith   PetscFunctionReturn(0);
44ab92ecdeSBarry Smith }
45ab92ecdeSBarry Smith 
46ab92ecdeSBarry Smith #undef __FUNCT__
47ba8c8a56SBarry Smith #define __FUNCT__ "MatGetRow_MPIDense"
48ba8c8a56SBarry Smith PetscErrorCode MatGetRow_MPIDense(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
49ba8c8a56SBarry Smith {
50ba8c8a56SBarry Smith   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
51ba8c8a56SBarry Smith   PetscErrorCode ierr;
52899cda47SBarry Smith   PetscInt       lrow,rstart = A->rmap.rstart,rend = A->rmap.rend;
53ba8c8a56SBarry Smith 
54ba8c8a56SBarry Smith   PetscFunctionBegin;
55ba8c8a56SBarry Smith   if (row < rstart || row >= rend) SETERRQ(PETSC_ERR_SUP,"only local rows")
56ba8c8a56SBarry Smith   lrow = row - rstart;
57ba8c8a56SBarry Smith   ierr = MatGetRow(mat->A,lrow,nz,(const PetscInt **)idx,(const PetscScalar **)v);CHKERRQ(ierr);
58ba8c8a56SBarry Smith   PetscFunctionReturn(0);
59ba8c8a56SBarry Smith }
60ba8c8a56SBarry Smith 
61ba8c8a56SBarry Smith #undef __FUNCT__
62ba8c8a56SBarry Smith #define __FUNCT__ "MatRestoreRow_MPIDense"
63ba8c8a56SBarry Smith PetscErrorCode MatRestoreRow_MPIDense(Mat mat,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
64ba8c8a56SBarry Smith {
65ba8c8a56SBarry Smith   PetscErrorCode ierr;
66ba8c8a56SBarry Smith 
67ba8c8a56SBarry Smith   PetscFunctionBegin;
68ba8c8a56SBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
69ba8c8a56SBarry Smith   if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);}
70ba8c8a56SBarry Smith   PetscFunctionReturn(0);
71ba8c8a56SBarry Smith }
72ba8c8a56SBarry Smith 
730de54da6SSatish Balay EXTERN_C_BEGIN
744a2ae208SSatish Balay #undef __FUNCT__
754a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonalBlock_MPIDense"
76be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatGetDiagonalBlock_MPIDense(Mat A,PetscTruth *iscopy,MatReuse reuse,Mat *B)
770de54da6SSatish Balay {
780de54da6SSatish Balay   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
796849ba73SBarry Smith   PetscErrorCode ierr;
80899cda47SBarry Smith   PetscInt       m = A->rmap.n,rstart = A->rmap.rstart;
8187828ca2SBarry Smith   PetscScalar    *array;
820de54da6SSatish Balay   MPI_Comm       comm;
830de54da6SSatish Balay 
840de54da6SSatish Balay   PetscFunctionBegin;
85899cda47SBarry Smith   if (A->rmap.N != A->cmap.N) SETERRQ(PETSC_ERR_SUP,"Only square matrices supported.");
860de54da6SSatish Balay 
870de54da6SSatish Balay   /* The reuse aspect is not implemented efficiently */
880de54da6SSatish Balay   if (reuse) { ierr = MatDestroy(*B);CHKERRQ(ierr);}
890de54da6SSatish Balay 
900de54da6SSatish Balay   ierr = PetscObjectGetComm((PetscObject)(mdn->A),&comm);CHKERRQ(ierr);
910de54da6SSatish Balay   ierr = MatGetArray(mdn->A,&array);CHKERRQ(ierr);
92f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,B);CHKERRQ(ierr);
93f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*B,m,m,m,m);CHKERRQ(ierr);
947adad957SLisandro Dalcin   ierr = MatSetType(*B,((PetscObject)mdn->A)->type_name);CHKERRQ(ierr);
955c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(*B,array+m*rstart);CHKERRQ(ierr);
960de54da6SSatish Balay   ierr = MatRestoreArray(mdn->A,&array);CHKERRQ(ierr);
970de54da6SSatish Balay   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
980de54da6SSatish Balay   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
990de54da6SSatish Balay 
1000de54da6SSatish Balay   *iscopy = PETSC_TRUE;
1010de54da6SSatish Balay   PetscFunctionReturn(0);
1020de54da6SSatish Balay }
1030de54da6SSatish Balay EXTERN_C_END
1040de54da6SSatish Balay 
1054a2ae208SSatish Balay #undef __FUNCT__
1064a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_MPIDense"
10713f74950SBarry Smith PetscErrorCode MatSetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv)
1088965ea79SLois Curfman McInnes {
10939b7565bSBarry Smith   Mat_MPIDense   *A = (Mat_MPIDense*)mat->data;
110dfbe8321SBarry Smith   PetscErrorCode ierr;
111899cda47SBarry Smith   PetscInt       i,j,rstart = mat->rmap.rstart,rend = mat->rmap.rend,row;
112273d9f13SBarry Smith   PetscTruth     roworiented = A->roworiented;
1138965ea79SLois Curfman McInnes 
1143a40ed3dSBarry Smith   PetscFunctionBegin;
1158965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
1165ef9f2a5SBarry Smith     if (idxm[i] < 0) continue;
117899cda47SBarry Smith     if (idxm[i] >= mat->rmap.N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large");
1188965ea79SLois Curfman McInnes     if (idxm[i] >= rstart && idxm[i] < rend) {
1198965ea79SLois Curfman McInnes       row = idxm[i] - rstart;
12039b7565bSBarry Smith       if (roworiented) {
12139b7565bSBarry Smith         ierr = MatSetValues(A->A,1,&row,n,idxn,v+i*n,addv);CHKERRQ(ierr);
1223a40ed3dSBarry Smith       } else {
1238965ea79SLois Curfman McInnes         for (j=0; j<n; j++) {
1245ef9f2a5SBarry Smith           if (idxn[j] < 0) continue;
125899cda47SBarry Smith           if (idxn[j] >= mat->cmap.N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large");
12639b7565bSBarry Smith           ierr = MatSetValues(A->A,1,&row,1,&idxn[j],v+i+j*m,addv);CHKERRQ(ierr);
12739b7565bSBarry Smith         }
1288965ea79SLois Curfman McInnes       }
1293a40ed3dSBarry Smith     } else {
1303782ba37SSatish Balay       if (!A->donotstash) {
13139b7565bSBarry Smith         if (roworiented) {
1328798bf22SSatish Balay           ierr = MatStashValuesRow_Private(&mat->stash,idxm[i],n,idxn,v+i*n);CHKERRQ(ierr);
133d36fbae8SSatish Balay         } else {
1348798bf22SSatish Balay           ierr = MatStashValuesCol_Private(&mat->stash,idxm[i],n,idxn,v+i,m);CHKERRQ(ierr);
13539b7565bSBarry Smith         }
136b49de8d1SLois Curfman McInnes       }
137b49de8d1SLois Curfman McInnes     }
1383782ba37SSatish Balay   }
1393a40ed3dSBarry Smith   PetscFunctionReturn(0);
140b49de8d1SLois Curfman McInnes }
141b49de8d1SLois Curfman McInnes 
1424a2ae208SSatish Balay #undef __FUNCT__
1434a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_MPIDense"
14413f74950SBarry Smith PetscErrorCode MatGetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[])
145b49de8d1SLois Curfman McInnes {
146b49de8d1SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
147dfbe8321SBarry Smith   PetscErrorCode ierr;
148899cda47SBarry Smith   PetscInt       i,j,rstart = mat->rmap.rstart,rend = mat->rmap.rend,row;
149b49de8d1SLois Curfman McInnes 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
151b49de8d1SLois Curfman McInnes   for (i=0; i<m; i++) {
15297e567efSBarry Smith     if (idxm[i] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative row"); */
153899cda47SBarry Smith     if (idxm[i] >= mat->rmap.N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large");
154b49de8d1SLois Curfman McInnes     if (idxm[i] >= rstart && idxm[i] < rend) {
155b49de8d1SLois Curfman McInnes       row = idxm[i] - rstart;
156b49de8d1SLois Curfman McInnes       for (j=0; j<n; j++) {
15797e567efSBarry Smith         if (idxn[j] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative column"); */
158899cda47SBarry Smith         if (idxn[j] >= mat->cmap.N) {
15929bbc08cSBarry Smith           SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large");
160a8c6a408SBarry Smith         }
161b49de8d1SLois Curfman McInnes         ierr = MatGetValues(mdn->A,1,&row,1,&idxn[j],v+i*n+j);CHKERRQ(ierr);
162b49de8d1SLois Curfman McInnes       }
163a8c6a408SBarry Smith     } else {
16429bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"Only local values currently supported");
1658965ea79SLois Curfman McInnes     }
1668965ea79SLois Curfman McInnes   }
1673a40ed3dSBarry Smith   PetscFunctionReturn(0);
1688965ea79SLois Curfman McInnes }
1698965ea79SLois Curfman McInnes 
1704a2ae208SSatish Balay #undef __FUNCT__
1714a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_MPIDense"
172dfbe8321SBarry Smith PetscErrorCode MatGetArray_MPIDense(Mat A,PetscScalar *array[])
173ff14e315SSatish Balay {
174ff14e315SSatish Balay   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
175dfbe8321SBarry Smith   PetscErrorCode ierr;
176ff14e315SSatish Balay 
1773a40ed3dSBarry Smith   PetscFunctionBegin;
178ff14e315SSatish Balay   ierr = MatGetArray(a->A,array);CHKERRQ(ierr);
1793a40ed3dSBarry Smith   PetscFunctionReturn(0);
180ff14e315SSatish Balay }
181ff14e315SSatish Balay 
1824a2ae208SSatish Balay #undef __FUNCT__
1834a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_MPIDense"
18413f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_MPIDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B)
185ca3fa75bSLois Curfman McInnes {
186ca3fa75bSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data,*newmatd;
187ca3fa75bSLois Curfman McInnes   Mat_SeqDense   *lmat = (Mat_SeqDense*)mat->A->data;
1886849ba73SBarry Smith   PetscErrorCode ierr;
18913f74950SBarry Smith   PetscInt       i,j,*irow,*icol,rstart,rend,nrows,ncols,nlrows,nlcols;
19087828ca2SBarry Smith   PetscScalar    *av,*bv,*v = lmat->v;
191ca3fa75bSLois Curfman McInnes   Mat            newmat;
192ca3fa75bSLois Curfman McInnes 
193ca3fa75bSLois Curfman McInnes   PetscFunctionBegin;
194ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
195ca3fa75bSLois Curfman McInnes   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
196b9b97703SBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
197b9b97703SBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
198ca3fa75bSLois Curfman McInnes 
199ca3fa75bSLois Curfman McInnes   /* No parallel redistribution currently supported! Should really check each index set
2007eba5e9cSLois Curfman McInnes      to comfirm that it is OK.  ... Currently supports only submatrix same partitioning as
2017eba5e9cSLois Curfman McInnes      original matrix! */
202ca3fa75bSLois Curfman McInnes 
203ca3fa75bSLois Curfman McInnes   ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr);
2047eba5e9cSLois Curfman McInnes   ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr);
205ca3fa75bSLois Curfman McInnes 
206ca3fa75bSLois Curfman McInnes   /* Check submatrix call */
207ca3fa75bSLois Curfman McInnes   if (scall == MAT_REUSE_MATRIX) {
20829bbc08cSBarry Smith     /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */
2097eba5e9cSLois Curfman McInnes     /* Really need to test rows and column sizes! */
210ca3fa75bSLois Curfman McInnes     newmat = *B;
211ca3fa75bSLois Curfman McInnes   } else {
212ca3fa75bSLois Curfman McInnes     /* Create and fill new matrix */
2137adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
214f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,cs,PETSC_DECIDE,ncols);CHKERRQ(ierr);
2157adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
216878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
217ca3fa75bSLois Curfman McInnes   }
218ca3fa75bSLois Curfman McInnes 
219ca3fa75bSLois Curfman McInnes   /* Now extract the data pointers and do the copy, column at a time */
220ca3fa75bSLois Curfman McInnes   newmatd = (Mat_MPIDense*)newmat->data;
221ca3fa75bSLois Curfman McInnes   bv      = ((Mat_SeqDense *)newmatd->A->data)->v;
222ca3fa75bSLois Curfman McInnes 
223ca3fa75bSLois Curfman McInnes   for (i=0; i<ncols; i++) {
224ca3fa75bSLois Curfman McInnes     av = v + nlrows*icol[i];
225ca3fa75bSLois Curfman McInnes     for (j=0; j<nrows; j++) {
2267eba5e9cSLois Curfman McInnes       *bv++ = av[irow[j] - rstart];
227ca3fa75bSLois Curfman McInnes     }
228ca3fa75bSLois Curfman McInnes   }
229ca3fa75bSLois Curfman McInnes 
230ca3fa75bSLois Curfman McInnes   /* Assemble the matrices so that the correct flags are set */
231ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
232ca3fa75bSLois Curfman McInnes   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
233ca3fa75bSLois Curfman McInnes 
234ca3fa75bSLois Curfman McInnes   /* Free work space */
235ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
236ca3fa75bSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
237ca3fa75bSLois Curfman McInnes   *B = newmat;
238ca3fa75bSLois Curfman McInnes   PetscFunctionReturn(0);
239ca3fa75bSLois Curfman McInnes }
240ca3fa75bSLois Curfman McInnes 
2414a2ae208SSatish Balay #undef __FUNCT__
2424a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense"
243dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[])
244ff14e315SSatish Balay {
2453a40ed3dSBarry Smith   PetscFunctionBegin;
2463a40ed3dSBarry Smith   PetscFunctionReturn(0);
247ff14e315SSatish Balay }
248ff14e315SSatish Balay 
2494a2ae208SSatish Balay #undef __FUNCT__
2504a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense"
251dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode)
2528965ea79SLois Curfman McInnes {
25339ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
2547adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)mat)->comm;
255dfbe8321SBarry Smith   PetscErrorCode ierr;
25613f74950SBarry Smith   PetscInt       nstash,reallocs;
2578965ea79SLois Curfman McInnes   InsertMode     addv;
2588965ea79SLois Curfman McInnes 
2593a40ed3dSBarry Smith   PetscFunctionBegin;
2608965ea79SLois Curfman McInnes   /* make sure all processors are either in INSERTMODE or ADDMODE */
261ca161407SBarry Smith   ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr);
2627056b6fcSBarry Smith   if (addv == (ADD_VALUES|INSERT_VALUES)) {
26329bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs");
2648965ea79SLois Curfman McInnes   }
265e0fa3b82SLois Curfman McInnes   mat->insertmode = addv; /* in case this processor had no cache */
2668965ea79SLois Curfman McInnes 
2671e2582c4SBarry Smith   ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap.range);CHKERRQ(ierr);
2688798bf22SSatish Balay   ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr);
269ae15b995SBarry Smith   ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr);
2703a40ed3dSBarry Smith   PetscFunctionReturn(0);
2718965ea79SLois Curfman McInnes }
2728965ea79SLois Curfman McInnes 
2734a2ae208SSatish Balay #undef __FUNCT__
2744a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense"
275dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode)
2768965ea79SLois Curfman McInnes {
27739ddd567SLois Curfman McInnes   Mat_MPIDense    *mdn=(Mat_MPIDense*)mat->data;
2786849ba73SBarry Smith   PetscErrorCode  ierr;
27913f74950SBarry Smith   PetscInt        i,*row,*col,flg,j,rstart,ncols;
28013f74950SBarry Smith   PetscMPIInt     n;
28187828ca2SBarry Smith   PetscScalar     *val;
282e0fa3b82SLois Curfman McInnes   InsertMode      addv=mat->insertmode;
2838965ea79SLois Curfman McInnes 
2843a40ed3dSBarry Smith   PetscFunctionBegin;
2858965ea79SLois Curfman McInnes   /*  wait on receives */
2867ef1d9bdSSatish Balay   while (1) {
2878798bf22SSatish Balay     ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr);
2887ef1d9bdSSatish Balay     if (!flg) break;
2898965ea79SLois Curfman McInnes 
2907ef1d9bdSSatish Balay     for (i=0; i<n;) {
2917ef1d9bdSSatish Balay       /* Now identify the consecutive vals belonging to the same row */
2927ef1d9bdSSatish Balay       for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; }
2937ef1d9bdSSatish Balay       if (j < n) ncols = j-i;
2947ef1d9bdSSatish Balay       else       ncols = n-i;
2957ef1d9bdSSatish Balay       /* Now assemble all these values with a single function call */
2967ef1d9bdSSatish Balay       ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr);
2977ef1d9bdSSatish Balay       i = j;
2988965ea79SLois Curfman McInnes     }
2997ef1d9bdSSatish Balay   }
3008798bf22SSatish Balay   ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr);
3018965ea79SLois Curfman McInnes 
30239ddd567SLois Curfman McInnes   ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr);
30339ddd567SLois Curfman McInnes   ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr);
3048965ea79SLois Curfman McInnes 
3056d4a8577SBarry Smith   if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) {
30639ddd567SLois Curfman McInnes     ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
3078965ea79SLois Curfman McInnes   }
3083a40ed3dSBarry Smith   PetscFunctionReturn(0);
3098965ea79SLois Curfman McInnes }
3108965ea79SLois Curfman McInnes 
3114a2ae208SSatish Balay #undef __FUNCT__
3124a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense"
313dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A)
3148965ea79SLois Curfman McInnes {
315dfbe8321SBarry Smith   PetscErrorCode ierr;
31639ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3173a40ed3dSBarry Smith 
3183a40ed3dSBarry Smith   PetscFunctionBegin;
3193a40ed3dSBarry Smith   ierr = MatZeroEntries(l->A);CHKERRQ(ierr);
3203a40ed3dSBarry Smith   PetscFunctionReturn(0);
3218965ea79SLois Curfman McInnes }
3228965ea79SLois Curfman McInnes 
3238965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the
3248965ea79SLois Curfman McInnes    matrix is square and the column and row owerships are identical.
3258965ea79SLois Curfman McInnes    This is a BUG. The only way to fix it seems to be to access
3263501a2bdSLois Curfman McInnes    mdn->A and mdn->B directly and not through the MatZeroRows()
3278965ea79SLois Curfman McInnes    routine.
3288965ea79SLois Curfman McInnes */
3294a2ae208SSatish Balay #undef __FUNCT__
3304a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense"
331f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
3328965ea79SLois Curfman McInnes {
33339ddd567SLois Curfman McInnes   Mat_MPIDense   *l = (Mat_MPIDense*)A->data;
3346849ba73SBarry Smith   PetscErrorCode ierr;
335899cda47SBarry Smith   PetscInt       i,*owners = A->rmap.range;
33613f74950SBarry Smith   PetscInt       *nprocs,j,idx,nsends;
33713f74950SBarry Smith   PetscInt       nmax,*svalues,*starts,*owner,nrecvs;
3387adad957SLisandro Dalcin   PetscInt       *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source;
33913f74950SBarry Smith   PetscInt       *lens,*lrows,*values;
34013f74950SBarry Smith   PetscMPIInt    n,imdex,rank = l->rank,size = l->size;
3417adad957SLisandro Dalcin   MPI_Comm       comm = ((PetscObject)A)->comm;
3428965ea79SLois Curfman McInnes   MPI_Request    *send_waits,*recv_waits;
3438965ea79SLois Curfman McInnes   MPI_Status     recv_status,*send_status;
34435d8aa7fSBarry Smith   PetscTruth     found;
3458965ea79SLois Curfman McInnes 
3463a40ed3dSBarry Smith   PetscFunctionBegin;
3478965ea79SLois Curfman McInnes   /*  first count number of contributors to each processor */
34813f74950SBarry Smith   ierr  = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr);
34913f74950SBarry Smith   ierr  = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr);
35013f74950SBarry Smith   ierr  = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/
3518965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3528965ea79SLois Curfman McInnes     idx = rows[i];
35335d8aa7fSBarry Smith     found = PETSC_FALSE;
3548965ea79SLois Curfman McInnes     for (j=0; j<size; j++) {
3558965ea79SLois Curfman McInnes       if (idx >= owners[j] && idx < owners[j+1]) {
356c1dc657dSBarry Smith         nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break;
3578965ea79SLois Curfman McInnes       }
3588965ea79SLois Curfman McInnes     }
35929bbc08cSBarry Smith     if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range");
3608965ea79SLois Curfman McInnes   }
361c1dc657dSBarry Smith   nsends = 0;  for (i=0; i<size; i++) { nsends += nprocs[2*i+1];}
3628965ea79SLois Curfman McInnes 
3638965ea79SLois Curfman McInnes   /* inform other processors of number of messages and max length*/
364c1dc657dSBarry Smith   ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr);
3658965ea79SLois Curfman McInnes 
3668965ea79SLois Curfman McInnes   /* post receives:   */
36713f74950SBarry Smith   ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr);
368b0a32e0cSBarry Smith   ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr);
3698965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
37013f74950SBarry Smith     ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr);
3718965ea79SLois Curfman McInnes   }
3728965ea79SLois Curfman McInnes 
3738965ea79SLois Curfman McInnes   /* do sends:
3748965ea79SLois Curfman McInnes       1) starts[i] gives the starting index in svalues for stuff going to
3758965ea79SLois Curfman McInnes          the ith processor
3768965ea79SLois Curfman McInnes   */
37713f74950SBarry Smith   ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr);
378b0a32e0cSBarry Smith   ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr);
37913f74950SBarry Smith   ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr);
3808965ea79SLois Curfman McInnes   starts[0]  = 0;
381c1dc657dSBarry Smith   for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3828965ea79SLois Curfman McInnes   for (i=0; i<N; i++) {
3838965ea79SLois Curfman McInnes     svalues[starts[owner[i]]++] = rows[i];
3848965ea79SLois Curfman McInnes   }
3858965ea79SLois Curfman McInnes 
3868965ea79SLois Curfman McInnes   starts[0] = 0;
387c1dc657dSBarry Smith   for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];}
3888965ea79SLois Curfman McInnes   count = 0;
3898965ea79SLois Curfman McInnes   for (i=0; i<size; i++) {
390c1dc657dSBarry Smith     if (nprocs[2*i+1]) {
39113f74950SBarry Smith       ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr);
3928965ea79SLois Curfman McInnes     }
3938965ea79SLois Curfman McInnes   }
394606d414cSSatish Balay   ierr = PetscFree(starts);CHKERRQ(ierr);
3958965ea79SLois Curfman McInnes 
3968965ea79SLois Curfman McInnes   base = owners[rank];
3978965ea79SLois Curfman McInnes 
3988965ea79SLois Curfman McInnes   /*  wait on receives */
39913f74950SBarry Smith   ierr   = PetscMalloc(2*(nrecvs+1)*sizeof(PetscInt),&lens);CHKERRQ(ierr);
4008965ea79SLois Curfman McInnes   source = lens + nrecvs;
4018965ea79SLois Curfman McInnes   count  = nrecvs; slen = 0;
4028965ea79SLois Curfman McInnes   while (count) {
403ca161407SBarry Smith     ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr);
4048965ea79SLois Curfman McInnes     /* unpack receives into our local space */
40513f74950SBarry Smith     ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr);
4068965ea79SLois Curfman McInnes     source[imdex]  = recv_status.MPI_SOURCE;
4078965ea79SLois Curfman McInnes     lens[imdex]    = n;
4088965ea79SLois Curfman McInnes     slen += n;
4098965ea79SLois Curfman McInnes     count--;
4108965ea79SLois Curfman McInnes   }
411606d414cSSatish Balay   ierr = PetscFree(recv_waits);CHKERRQ(ierr);
4128965ea79SLois Curfman McInnes 
4138965ea79SLois Curfman McInnes   /* move the data into the send scatter */
41413f74950SBarry Smith   ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr);
4158965ea79SLois Curfman McInnes   count = 0;
4168965ea79SLois Curfman McInnes   for (i=0; i<nrecvs; i++) {
4178965ea79SLois Curfman McInnes     values = rvalues + i*nmax;
4188965ea79SLois Curfman McInnes     for (j=0; j<lens[i]; j++) {
4198965ea79SLois Curfman McInnes       lrows[count++] = values[j] - base;
4208965ea79SLois Curfman McInnes     }
4218965ea79SLois Curfman McInnes   }
422606d414cSSatish Balay   ierr = PetscFree(rvalues);CHKERRQ(ierr);
423606d414cSSatish Balay   ierr = PetscFree(lens);CHKERRQ(ierr);
424606d414cSSatish Balay   ierr = PetscFree(owner);CHKERRQ(ierr);
425606d414cSSatish Balay   ierr = PetscFree(nprocs);CHKERRQ(ierr);
4268965ea79SLois Curfman McInnes 
4278965ea79SLois Curfman McInnes   /* actually zap the local rows */
428f4df32b1SMatthew Knepley   ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr);
429606d414cSSatish Balay   ierr = PetscFree(lrows);CHKERRQ(ierr);
4308965ea79SLois Curfman McInnes 
4318965ea79SLois Curfman McInnes   /* wait on sends */
4328965ea79SLois Curfman McInnes   if (nsends) {
433b0a32e0cSBarry Smith     ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr);
434ca161407SBarry Smith     ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr);
435606d414cSSatish Balay     ierr = PetscFree(send_status);CHKERRQ(ierr);
4368965ea79SLois Curfman McInnes   }
437606d414cSSatish Balay   ierr = PetscFree(send_waits);CHKERRQ(ierr);
438606d414cSSatish Balay   ierr = PetscFree(svalues);CHKERRQ(ierr);
4398965ea79SLois Curfman McInnes 
4403a40ed3dSBarry Smith   PetscFunctionReturn(0);
4418965ea79SLois Curfman McInnes }
4428965ea79SLois Curfman McInnes 
4434a2ae208SSatish Balay #undef __FUNCT__
4444a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense"
445dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy)
4468965ea79SLois Curfman McInnes {
44739ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
448dfbe8321SBarry Smith   PetscErrorCode ierr;
449c456f294SBarry Smith 
4503a40ed3dSBarry Smith   PetscFunctionBegin;
451ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
452ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
45344cd7ae7SLois Curfman McInnes   ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr);
4543a40ed3dSBarry Smith   PetscFunctionReturn(0);
4558965ea79SLois Curfman McInnes }
4568965ea79SLois Curfman McInnes 
4574a2ae208SSatish Balay #undef __FUNCT__
4584a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense"
459dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz)
4608965ea79SLois Curfman McInnes {
46139ddd567SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
462dfbe8321SBarry Smith   PetscErrorCode ierr;
463c456f294SBarry Smith 
4643a40ed3dSBarry Smith   PetscFunctionBegin;
465ca9f406cSSatish Balay   ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
466ca9f406cSSatish Balay   ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
46744cd7ae7SLois Curfman McInnes   ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr);
4683a40ed3dSBarry Smith   PetscFunctionReturn(0);
4698965ea79SLois Curfman McInnes }
4708965ea79SLois Curfman McInnes 
4714a2ae208SSatish Balay #undef __FUNCT__
4724a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense"
473dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy)
474096963f5SLois Curfman McInnes {
475096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
476dfbe8321SBarry Smith   PetscErrorCode ierr;
47787828ca2SBarry Smith   PetscScalar    zero = 0.0;
478096963f5SLois Curfman McInnes 
4793a40ed3dSBarry Smith   PetscFunctionBegin;
4802dcb1b2aSMatthew Knepley   ierr = VecSet(yy,zero);CHKERRQ(ierr);
4817c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
482ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
483ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
4843a40ed3dSBarry Smith   PetscFunctionReturn(0);
485096963f5SLois Curfman McInnes }
486096963f5SLois Curfman McInnes 
4874a2ae208SSatish Balay #undef __FUNCT__
4884a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense"
489dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz)
490096963f5SLois Curfman McInnes {
491096963f5SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
492dfbe8321SBarry Smith   PetscErrorCode ierr;
493096963f5SLois Curfman McInnes 
4943a40ed3dSBarry Smith   PetscFunctionBegin;
4953501a2bdSLois Curfman McInnes   ierr = VecCopy(yy,zz);CHKERRQ(ierr);
4967c922b88SBarry Smith   ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr);
497ca9f406cSSatish Balay   ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
498ca9f406cSSatish Balay   ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr);
4993a40ed3dSBarry Smith   PetscFunctionReturn(0);
500096963f5SLois Curfman McInnes }
501096963f5SLois Curfman McInnes 
5024a2ae208SSatish Balay #undef __FUNCT__
5034a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense"
504dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v)
5058965ea79SLois Curfman McInnes {
50639ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
507096963f5SLois Curfman McInnes   Mat_SeqDense   *aloc = (Mat_SeqDense*)a->A->data;
508dfbe8321SBarry Smith   PetscErrorCode ierr;
509899cda47SBarry Smith   PetscInt       len,i,n,m = A->rmap.n,radd;
51087828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
511ed3cc1f0SBarry Smith 
5123a40ed3dSBarry Smith   PetscFunctionBegin;
5132dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
5141ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
515096963f5SLois Curfman McInnes   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
516899cda47SBarry Smith   if (n != A->rmap.N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
517899cda47SBarry Smith   len  = PetscMin(a->A->rmap.n,a->A->cmap.n);
518899cda47SBarry Smith   radd = A->rmap.rstart*m;
51944cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
520096963f5SLois Curfman McInnes     x[i] = aloc->v[radd + i*m + i];
521096963f5SLois Curfman McInnes   }
5221ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
5233a40ed3dSBarry Smith   PetscFunctionReturn(0);
5248965ea79SLois Curfman McInnes }
5258965ea79SLois Curfman McInnes 
5264a2ae208SSatish Balay #undef __FUNCT__
5274a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense"
528dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat)
5298965ea79SLois Curfman McInnes {
5303501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)mat->data;
531dfbe8321SBarry Smith   PetscErrorCode ierr;
53201b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
53301b82886SBarry Smith   Mat_Plapack   *lu=(Mat_Plapack*)(mat->spptr);
53401b82886SBarry Smith #endif
535ed3cc1f0SBarry Smith 
5363a40ed3dSBarry Smith   PetscFunctionBegin;
53794d884c6SBarry Smith 
538aa482453SBarry Smith #if defined(PETSC_USE_LOG)
539899cda47SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap.N,mat->cmap.N);
5408965ea79SLois Curfman McInnes #endif
5418798bf22SSatish Balay   ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr);
5423501a2bdSLois Curfman McInnes   ierr = MatDestroy(mdn->A);CHKERRQ(ierr);
54305b42c5fSBarry Smith   if (mdn->lvec)   {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);}
54405b42c5fSBarry Smith   if (mdn->Mvctx)  {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);}
54501b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
5462fbe02b9SBarry Smith   if (lu) {
54762b4c0b3SBarry Smith     ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr);
54862b4c0b3SBarry Smith     ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr);
54962b4c0b3SBarry Smith     ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr);
55009d27a7eSBarry Smith 
5512fbe02b9SBarry Smith     if (lu->is_pla) {
55201b82886SBarry Smith       ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr);
55301b82886SBarry Smith       ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr);
55401b82886SBarry Smith       ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr);
555622d7880SLois Curfman McInnes     }
5562fbe02b9SBarry Smith   }
55701b82886SBarry Smith #endif
55801b82886SBarry Smith 
559606d414cSSatish Balay   ierr = PetscFree(mdn);CHKERRQ(ierr);
560dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
561901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr);
562901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
5634ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5644ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5654ae313f4SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr);
5663a40ed3dSBarry Smith   PetscFunctionReturn(0);
5678965ea79SLois Curfman McInnes }
56839ddd567SLois Curfman McInnes 
5694a2ae208SSatish Balay #undef __FUNCT__
5704a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary"
5716849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer)
5728965ea79SLois Curfman McInnes {
57339ddd567SLois Curfman McInnes   Mat_MPIDense      *mdn = (Mat_MPIDense*)mat->data;
574dfbe8321SBarry Smith   PetscErrorCode    ierr;
575aa05aa95SBarry Smith   PetscViewerFormat format;
576aa05aa95SBarry Smith   int               fd;
577aa05aa95SBarry Smith   PetscInt          header[4],mmax,N = mat->cmap.N,i,j,m,k;
578aa05aa95SBarry Smith   PetscMPIInt       rank,tag  = ((PetscObject)viewer)->tag,size;
579578230a0SSatish Balay   PetscScalar       *work,*v,*vv;
580aa05aa95SBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)mdn->A->data;
581aa05aa95SBarry Smith   MPI_Status        status;
5827056b6fcSBarry Smith 
5833a40ed3dSBarry Smith   PetscFunctionBegin;
58439ddd567SLois Curfman McInnes   if (mdn->size == 1) {
58539ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
586aa05aa95SBarry Smith   } else {
587aa05aa95SBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
5887adad957SLisandro Dalcin     ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr);
5897adad957SLisandro Dalcin     ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr);
590aa05aa95SBarry Smith 
591aa05aa95SBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
592aa05aa95SBarry Smith     if (format == PETSC_VIEWER_BINARY_NATIVE) {
593aa05aa95SBarry Smith 
594aa05aa95SBarry Smith       if (!rank) {
595aa05aa95SBarry Smith         /* store the matrix as a dense matrix */
596aa05aa95SBarry Smith         header[0] = MAT_FILE_COOKIE;
597aa05aa95SBarry Smith         header[1] = mat->rmap.N;
598aa05aa95SBarry Smith         header[2] = N;
599aa05aa95SBarry Smith         header[3] = MATRIX_BINARY_FORMAT_DENSE;
600aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
601aa05aa95SBarry Smith 
602aa05aa95SBarry Smith         /* get largest work array needed for transposing array */
603aa05aa95SBarry Smith         mmax = mat->rmap.n;
604aa05aa95SBarry Smith         for (i=1; i<size; i++) {
605aa05aa95SBarry Smith           mmax = PetscMax(mmax,mat->rmap.range[i+1] - mat->rmap.range[i]);
6068965ea79SLois Curfman McInnes         }
607aa05aa95SBarry Smith         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr);
608aa05aa95SBarry Smith 
609aa05aa95SBarry Smith         /* write out local array, by rows */
610aa05aa95SBarry Smith         m    = mat->rmap.n;
611aa05aa95SBarry Smith         v    = a->v;
612aa05aa95SBarry Smith         for (j=0; j<N; j++) {
613aa05aa95SBarry Smith           for (i=0; i<m; i++) {
614578230a0SSatish Balay             work[j + i*N] = *v++;
615aa05aa95SBarry Smith           }
616aa05aa95SBarry Smith         }
617aa05aa95SBarry Smith         ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
618aa05aa95SBarry Smith         /* get largest work array to receive messages from other processes, excludes process zero */
619aa05aa95SBarry Smith         mmax = 0;
620aa05aa95SBarry Smith         for (i=1; i<size; i++) {
621aa05aa95SBarry Smith           mmax = PetscMax(mmax,mat->rmap.range[i+1] - mat->rmap.range[i]);
622aa05aa95SBarry Smith         }
623578230a0SSatish Balay         ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr);
624aa05aa95SBarry Smith         for(k = 1; k < size; k++) {
625f8009846SMatthew Knepley           v    = vv;
626aa05aa95SBarry Smith           m    = mat->rmap.range[k+1] - mat->rmap.range[k];
6277adad957SLisandro Dalcin           ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr);
628aa05aa95SBarry Smith 
629aa05aa95SBarry Smith           for(j = 0; j < N; j++) {
630aa05aa95SBarry Smith             for(i = 0; i < m; i++) {
631578230a0SSatish Balay               work[j + i*N] = *v++;
632aa05aa95SBarry Smith             }
633aa05aa95SBarry Smith           }
634aa05aa95SBarry Smith           ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
635aa05aa95SBarry Smith         }
636aa05aa95SBarry Smith         ierr = PetscFree(work);CHKERRQ(ierr);
637578230a0SSatish Balay         ierr = PetscFree(vv);CHKERRQ(ierr);
638aa05aa95SBarry Smith       } else {
6397adad957SLisandro Dalcin         ierr = MPI_Send(a->v,mat->rmap.n*mat->cmap.N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr);
640aa05aa95SBarry Smith       }
6414aa34b0aSBarry Smith     } else {
642a542b6e8SBarry Smith       SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_BINARY_NATIVE");
643aa05aa95SBarry Smith     }
644aa05aa95SBarry Smith   }
6453a40ed3dSBarry Smith   PetscFunctionReturn(0);
6468965ea79SLois Curfman McInnes }
6478965ea79SLois Curfman McInnes 
6484a2ae208SSatish Balay #undef __FUNCT__
6494a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket"
6506849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer)
6518965ea79SLois Curfman McInnes {
65239ddd567SLois Curfman McInnes   Mat_MPIDense          *mdn = (Mat_MPIDense*)mat->data;
653dfbe8321SBarry Smith   PetscErrorCode        ierr;
65432dcc486SBarry Smith   PetscMPIInt           size = mdn->size,rank = mdn->rank;
655a313700dSBarry Smith   const PetscViewerType vtype;
65632077d6dSBarry Smith   PetscTruth            iascii,isdraw;
657b0a32e0cSBarry Smith   PetscViewer           sviewer;
658f3ef73ceSBarry Smith   PetscViewerFormat     format;
65901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
66001b82886SBarry Smith   Mat_Plapack           *lu=(Mat_Plapack*)(mat->spptr);
66101b82886SBarry Smith #endif
6628965ea79SLois Curfman McInnes 
6633a40ed3dSBarry Smith   PetscFunctionBegin;
66432077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
665fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
66632077d6dSBarry Smith   if (iascii) {
667b0a32e0cSBarry Smith     ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr);
668b0a32e0cSBarry Smith     ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
669456192e2SBarry Smith     if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
6704e220ebcSLois Curfman McInnes       MatInfo info;
671888f2ed8SSatish Balay       ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr);
672899cda47SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"  [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap.n,
67377431f27SBarry Smith                    (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr);
674b0a32e0cSBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
6753501a2bdSLois Curfman McInnes       ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr);
67601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
67701b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr);
67804fea9ffSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Processor mesh: nprows %d, npcols %d\n",Plapack_nprows, Plapack_npcols);CHKERRQ(ierr);
67901b82886SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr);
680eae6fb2eSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Error checking: %d\n",Plapack_ierror);CHKERRQ(ierr);
681eae6fb2eSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Algorithmic block size: %d\n",Plapack_nb_alg);CHKERRQ(ierr);
68201b82886SBarry Smith #endif
6833a40ed3dSBarry Smith       PetscFunctionReturn(0);
684fb9695e5SSatish Balay     } else if (format == PETSC_VIEWER_ASCII_INFO) {
6853a40ed3dSBarry Smith       PetscFunctionReturn(0);
6868965ea79SLois Curfman McInnes     }
687f1af5d2fSBarry Smith   } else if (isdraw) {
688b0a32e0cSBarry Smith     PetscDraw  draw;
689f1af5d2fSBarry Smith     PetscTruth isnull;
690f1af5d2fSBarry Smith 
691b0a32e0cSBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
692b0a32e0cSBarry Smith     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
693f1af5d2fSBarry Smith     if (isnull) PetscFunctionReturn(0);
694f1af5d2fSBarry Smith   }
69577ed5343SBarry Smith 
6968965ea79SLois Curfman McInnes   if (size == 1) {
69739ddd567SLois Curfman McInnes     ierr = MatView(mdn->A,viewer);CHKERRQ(ierr);
6983a40ed3dSBarry Smith   } else {
6998965ea79SLois Curfman McInnes     /* assemble the entire matrix onto first processor. */
7008965ea79SLois Curfman McInnes     Mat         A;
701899cda47SBarry Smith     PetscInt    M = mat->rmap.N,N = mat->cmap.N,m,row,i,nz;
702ba8c8a56SBarry Smith     PetscInt    *cols;
703ba8c8a56SBarry Smith     PetscScalar *vals;
7048965ea79SLois Curfman McInnes 
7057adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr);
7068965ea79SLois Curfman McInnes     if (!rank) {
707f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr);
7083a40ed3dSBarry Smith     } else {
709f69a0ea3SMatthew Knepley       ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr);
7108965ea79SLois Curfman McInnes     }
7117adad957SLisandro Dalcin     /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */
712878740d9SKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
713878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL);
71452e6d16bSBarry Smith     ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr);
7158965ea79SLois Curfman McInnes 
71639ddd567SLois Curfman McInnes     /* Copy the matrix ... This isn't the most efficient means,
71739ddd567SLois Curfman McInnes        but it's quick for now */
71851022da4SBarry Smith     A->insertmode = INSERT_VALUES;
719899cda47SBarry Smith     row = mat->rmap.rstart; m = mdn->A->rmap.n;
7208965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
721ba8c8a56SBarry Smith       ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
722ba8c8a56SBarry Smith       ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr);
723ba8c8a56SBarry Smith       ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr);
72439ddd567SLois Curfman McInnes       row++;
7258965ea79SLois Curfman McInnes     }
7268965ea79SLois Curfman McInnes 
7276d4a8577SBarry Smith     ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7286d4a8577SBarry Smith     ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
729b0a32e0cSBarry Smith     ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr);
730b9b97703SBarry Smith     if (!rank) {
7316831982aSBarry Smith       ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr);
7328965ea79SLois Curfman McInnes     }
733b0a32e0cSBarry Smith     ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr);
734b0a32e0cSBarry Smith     ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
7358965ea79SLois Curfman McInnes     ierr = MatDestroy(A);CHKERRQ(ierr);
7368965ea79SLois Curfman McInnes   }
7373a40ed3dSBarry Smith   PetscFunctionReturn(0);
7388965ea79SLois Curfman McInnes }
7398965ea79SLois Curfman McInnes 
7404a2ae208SSatish Balay #undef __FUNCT__
7414a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense"
742dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer)
7438965ea79SLois Curfman McInnes {
744dfbe8321SBarry Smith   PetscErrorCode ierr;
74532077d6dSBarry Smith   PetscTruth     iascii,isbinary,isdraw,issocket;
7468965ea79SLois Curfman McInnes 
747433994e6SBarry Smith   PetscFunctionBegin;
7480f5bd95cSBarry Smith 
74932077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
750fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
751b0a32e0cSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr);
752fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
7530f5bd95cSBarry Smith 
75432077d6dSBarry Smith   if (iascii || issocket || isdraw) {
755f1af5d2fSBarry Smith     ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr);
7560f5bd95cSBarry Smith   } else if (isbinary) {
7573a40ed3dSBarry Smith     ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr);
7585cd90555SBarry Smith   } else {
759958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name);
7608965ea79SLois Curfman McInnes   }
7613a40ed3dSBarry Smith   PetscFunctionReturn(0);
7628965ea79SLois Curfman McInnes }
7638965ea79SLois Curfman McInnes 
7644a2ae208SSatish Balay #undef __FUNCT__
7654a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense"
766dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info)
7678965ea79SLois Curfman McInnes {
7683501a2bdSLois Curfman McInnes   Mat_MPIDense   *mat = (Mat_MPIDense*)A->data;
7693501a2bdSLois Curfman McInnes   Mat            mdn = mat->A;
770dfbe8321SBarry Smith   PetscErrorCode ierr;
771329f5518SBarry Smith   PetscReal      isend[5],irecv[5];
7728965ea79SLois Curfman McInnes 
7733a40ed3dSBarry Smith   PetscFunctionBegin;
774899cda47SBarry Smith   info->rows_global    = (double)A->rmap.N;
775899cda47SBarry Smith   info->columns_global = (double)A->cmap.N;
776899cda47SBarry Smith   info->rows_local     = (double)A->rmap.n;
777899cda47SBarry Smith   info->columns_local  = (double)A->cmap.N;
7784e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
7794e220ebcSLois Curfman McInnes   ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr);
7804e220ebcSLois Curfman McInnes   isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded;
7814e220ebcSLois Curfman McInnes   isend[3] = info->memory;  isend[4] = info->mallocs;
7828965ea79SLois Curfman McInnes   if (flag == MAT_LOCAL) {
7834e220ebcSLois Curfman McInnes     info->nz_used      = isend[0];
7844e220ebcSLois Curfman McInnes     info->nz_allocated = isend[1];
7854e220ebcSLois Curfman McInnes     info->nz_unneeded  = isend[2];
7864e220ebcSLois Curfman McInnes     info->memory       = isend[3];
7874e220ebcSLois Curfman McInnes     info->mallocs      = isend[4];
7888965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_MAX) {
7897adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
7904e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7914e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
7924e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
7934e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
7944e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
7958965ea79SLois Curfman McInnes   } else if (flag == MAT_GLOBAL_SUM) {
7967adad957SLisandro Dalcin     ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
7974e220ebcSLois Curfman McInnes     info->nz_used      = irecv[0];
7984e220ebcSLois Curfman McInnes     info->nz_allocated = irecv[1];
7994e220ebcSLois Curfman McInnes     info->nz_unneeded  = irecv[2];
8004e220ebcSLois Curfman McInnes     info->memory       = irecv[3];
8014e220ebcSLois Curfman McInnes     info->mallocs      = irecv[4];
8028965ea79SLois Curfman McInnes   }
8034e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0; /* no parallel LU/ILU/Cholesky */
8044e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
8054e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
8063a40ed3dSBarry Smith   PetscFunctionReturn(0);
8078965ea79SLois Curfman McInnes }
8088965ea79SLois Curfman McInnes 
8094a2ae208SSatish Balay #undef __FUNCT__
8104a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense"
8114e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg)
8128965ea79SLois Curfman McInnes {
81339ddd567SLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
814dfbe8321SBarry Smith   PetscErrorCode ierr;
8158965ea79SLois Curfman McInnes 
8163a40ed3dSBarry Smith   PetscFunctionBegin;
81712c028f9SKris Buschelman   switch (op) {
818512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
81912c028f9SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
82012c028f9SKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
8214e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82212c028f9SKris Buschelman     break;
82312c028f9SKris Buschelman   case MAT_ROW_ORIENTED:
8244e0d8c25SBarry Smith     a->roworiented = flg;
8254e0d8c25SBarry Smith     ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr);
82612c028f9SKris Buschelman     break;
8274e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
82812c028f9SKris Buschelman   case MAT_USE_HASH_TABLE:
829290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
83012c028f9SKris Buschelman     break;
83112c028f9SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
8324e0d8c25SBarry Smith     a->donotstash = flg;
83312c028f9SKris Buschelman     break;
83477e54ba9SKris Buschelman   case MAT_SYMMETRIC:
83577e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
8369a4540c5SBarry Smith   case MAT_HERMITIAN:
8379a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
838600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
839290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
84077e54ba9SKris Buschelman     break;
84112c028f9SKris Buschelman   default:
842600fe468SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
8433a40ed3dSBarry Smith   }
8443a40ed3dSBarry Smith   PetscFunctionReturn(0);
8458965ea79SLois Curfman McInnes }
8468965ea79SLois Curfman McInnes 
8478965ea79SLois Curfman McInnes 
8484a2ae208SSatish Balay #undef __FUNCT__
8494a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense"
850dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr)
8515b2fa520SLois Curfman McInnes {
8525b2fa520SLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8535b2fa520SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
85487828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
855dfbe8321SBarry Smith   PetscErrorCode ierr;
856899cda47SBarry Smith   PetscInt       i,j,s2a,s3a,s2,s3,m=mdn->A->rmap.n,n=mdn->A->cmap.n;
8575b2fa520SLois Curfman McInnes 
8585b2fa520SLois Curfman McInnes   PetscFunctionBegin;
85972d926a5SLois Curfman McInnes   ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr);
8605b2fa520SLois Curfman McInnes   if (ll) {
86172d926a5SLois Curfman McInnes     ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr);
862175be7b4SMatthew Knepley     if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2);
8631ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
8645b2fa520SLois Curfman McInnes     for (i=0; i<m; i++) {
8655b2fa520SLois Curfman McInnes       x = l[i];
8665b2fa520SLois Curfman McInnes       v = mat->v + i;
8675b2fa520SLois Curfman McInnes       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
8685b2fa520SLois Curfman McInnes     }
8691ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
870efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8715b2fa520SLois Curfman McInnes   }
8725b2fa520SLois Curfman McInnes   if (rr) {
873175be7b4SMatthew Knepley     ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr);
874175be7b4SMatthew Knepley     if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3);
875ca9f406cSSatish Balay     ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
876ca9f406cSSatish Balay     ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
8771ebc52fbSHong Zhang     ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr);
8785b2fa520SLois Curfman McInnes     for (i=0; i<n; i++) {
8795b2fa520SLois Curfman McInnes       x = r[i];
8805b2fa520SLois Curfman McInnes       v = mat->v + i*m;
8815b2fa520SLois Curfman McInnes       for (j=0; j<m; j++) { (*v++) *= x;}
8825b2fa520SLois Curfman McInnes     }
8831ebc52fbSHong Zhang     ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr);
884efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
8855b2fa520SLois Curfman McInnes   }
8865b2fa520SLois Curfman McInnes   PetscFunctionReturn(0);
8875b2fa520SLois Curfman McInnes }
8885b2fa520SLois Curfman McInnes 
8894a2ae208SSatish Balay #undef __FUNCT__
8904a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense"
891dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm)
892096963f5SLois Curfman McInnes {
8933501a2bdSLois Curfman McInnes   Mat_MPIDense   *mdn = (Mat_MPIDense*)A->data;
8943501a2bdSLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)mdn->A->data;
895dfbe8321SBarry Smith   PetscErrorCode ierr;
89613f74950SBarry Smith   PetscInt       i,j;
897329f5518SBarry Smith   PetscReal      sum = 0.0;
89887828ca2SBarry Smith   PetscScalar    *v = mat->v;
8993501a2bdSLois Curfman McInnes 
9003a40ed3dSBarry Smith   PetscFunctionBegin;
9013501a2bdSLois Curfman McInnes   if (mdn->size == 1) {
902064f8208SBarry Smith     ierr =  MatNorm(mdn->A,type,nrm);CHKERRQ(ierr);
9033501a2bdSLois Curfman McInnes   } else {
9043501a2bdSLois Curfman McInnes     if (type == NORM_FROBENIUS) {
905899cda47SBarry Smith       for (i=0; i<mdn->A->cmap.n*mdn->A->rmap.n; i++) {
906aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
907329f5518SBarry Smith         sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
9083501a2bdSLois Curfman McInnes #else
9093501a2bdSLois Curfman McInnes         sum += (*v)*(*v); v++;
9103501a2bdSLois Curfman McInnes #endif
9113501a2bdSLois Curfman McInnes       }
9127adad957SLisandro Dalcin       ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
913064f8208SBarry Smith       *nrm = sqrt(*nrm);
914899cda47SBarry Smith       ierr = PetscLogFlops(2*mdn->A->cmap.n*mdn->A->rmap.n);CHKERRQ(ierr);
9153a40ed3dSBarry Smith     } else if (type == NORM_1) {
916329f5518SBarry Smith       PetscReal *tmp,*tmp2;
917899cda47SBarry Smith       ierr = PetscMalloc(2*A->cmap.N*sizeof(PetscReal),&tmp);CHKERRQ(ierr);
918899cda47SBarry Smith       tmp2 = tmp + A->cmap.N;
919899cda47SBarry Smith       ierr = PetscMemzero(tmp,2*A->cmap.N*sizeof(PetscReal));CHKERRQ(ierr);
920064f8208SBarry Smith       *nrm = 0.0;
9213501a2bdSLois Curfman McInnes       v = mat->v;
922899cda47SBarry Smith       for (j=0; j<mdn->A->cmap.n; j++) {
923899cda47SBarry Smith         for (i=0; i<mdn->A->rmap.n; i++) {
92467e560aaSBarry Smith           tmp[j] += PetscAbsScalar(*v);  v++;
9253501a2bdSLois Curfman McInnes         }
9263501a2bdSLois Curfman McInnes       }
9277adad957SLisandro Dalcin       ierr = MPI_Allreduce(tmp,tmp2,A->cmap.N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr);
928899cda47SBarry Smith       for (j=0; j<A->cmap.N; j++) {
929064f8208SBarry Smith         if (tmp2[j] > *nrm) *nrm = tmp2[j];
9303501a2bdSLois Curfman McInnes       }
931606d414cSSatish Balay       ierr = PetscFree(tmp);CHKERRQ(ierr);
932899cda47SBarry Smith       ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr);
9333a40ed3dSBarry Smith     } else if (type == NORM_INFINITY) { /* max row norm */
934329f5518SBarry Smith       PetscReal ntemp;
9353501a2bdSLois Curfman McInnes       ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr);
9367adad957SLisandro Dalcin       ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr);
9373a40ed3dSBarry Smith     } else {
93829bbc08cSBarry Smith       SETERRQ(PETSC_ERR_SUP,"No support for two norm");
9393501a2bdSLois Curfman McInnes     }
9403501a2bdSLois Curfman McInnes   }
9413a40ed3dSBarry Smith   PetscFunctionReturn(0);
9423501a2bdSLois Curfman McInnes }
9433501a2bdSLois Curfman McInnes 
9444a2ae208SSatish Balay #undef __FUNCT__
9454a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense"
946fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout)
9473501a2bdSLois Curfman McInnes {
9483501a2bdSLois Curfman McInnes   Mat_MPIDense   *a = (Mat_MPIDense*)A->data;
9493501a2bdSLois Curfman McInnes   Mat_SeqDense   *Aloc = (Mat_SeqDense*)a->A->data;
9503501a2bdSLois Curfman McInnes   Mat            B;
951899cda47SBarry Smith   PetscInt       M = A->rmap.N,N = A->cmap.N,m,n,*rwork,rstart = A->rmap.rstart;
9526849ba73SBarry Smith   PetscErrorCode ierr;
95313f74950SBarry Smith   PetscInt       j,i;
95487828ca2SBarry Smith   PetscScalar    *v;
9553501a2bdSLois Curfman McInnes 
9563a40ed3dSBarry Smith   PetscFunctionBegin;
957e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place");
958fc4dec0aSBarry Smith   if (reuse == MAT_INITIAL_MATRIX || A == *matout) {
9597adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr);
960f69a0ea3SMatthew Knepley     ierr = MatSetSizes(B,PETSC_DECIDE,PETSC_DECIDE,N,M);CHKERRQ(ierr);
9617adad957SLisandro Dalcin     ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr);
962878740d9SKris Buschelman     ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr);
963fc4dec0aSBarry Smith   } else {
964fc4dec0aSBarry Smith     B = *matout;
965fc4dec0aSBarry Smith   }
9663501a2bdSLois Curfman McInnes 
967899cda47SBarry Smith   m = a->A->rmap.n; n = a->A->cmap.n; v = Aloc->v;
9681acff37aSSatish Balay   ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr);
9693501a2bdSLois Curfman McInnes   for (i=0; i<m; i++) rwork[i] = rstart + i;
9701acff37aSSatish Balay   for (j=0; j<n; j++) {
9713501a2bdSLois Curfman McInnes     ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr);
9723501a2bdSLois Curfman McInnes     v   += m;
9733501a2bdSLois Curfman McInnes   }
974606d414cSSatish Balay   ierr = PetscFree(rwork);CHKERRQ(ierr);
9756d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
9766d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
977815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *matout != A) {
9783501a2bdSLois Curfman McInnes     *matout = B;
9793501a2bdSLois Curfman McInnes   } else {
980273d9f13SBarry Smith     ierr = MatHeaderCopy(A,B);CHKERRQ(ierr);
9813501a2bdSLois Curfman McInnes   }
9823a40ed3dSBarry Smith   PetscFunctionReturn(0);
983096963f5SLois Curfman McInnes }
984096963f5SLois Curfman McInnes 
985d9eff348SSatish Balay #include "petscblaslapack.h"
9864a2ae208SSatish Balay #undef __FUNCT__
9874a2ae208SSatish Balay #define __FUNCT__ "MatScale_MPIDense"
988f4df32b1SMatthew Knepley PetscErrorCode MatScale_MPIDense(Mat inA,PetscScalar alpha)
98944cd7ae7SLois Curfman McInnes {
99044cd7ae7SLois Curfman McInnes   Mat_MPIDense   *A = (Mat_MPIDense*)inA->data;
99144cd7ae7SLois Curfman McInnes   Mat_SeqDense   *a = (Mat_SeqDense*)A->A->data;
992f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
993efee365bSSatish Balay   PetscErrorCode ierr;
9940805154bSBarry Smith   PetscBLASInt   one = 1,nz = PetscBLASIntCast(inA->rmap.n*inA->cmap.N);
99544cd7ae7SLois Curfman McInnes 
9963a40ed3dSBarry Smith   PetscFunctionBegin;
997f4df32b1SMatthew Knepley   BLASscal_(&nz,&oalpha,a->v,&one);
998efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
9993a40ed3dSBarry Smith   PetscFunctionReturn(0);
100044cd7ae7SLois Curfman McInnes }
100144cd7ae7SLois Curfman McInnes 
10026849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *);
10038965ea79SLois Curfman McInnes 
10044a2ae208SSatish Balay #undef __FUNCT__
10054a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_MPIDense"
1006dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_MPIDense(Mat A)
1007273d9f13SBarry Smith {
1008dfbe8321SBarry Smith   PetscErrorCode ierr;
1009273d9f13SBarry Smith 
1010273d9f13SBarry Smith   PetscFunctionBegin;
1011273d9f13SBarry Smith   ierr =  MatMPIDenseSetPreallocation(A,0);CHKERRQ(ierr);
1012273d9f13SBarry Smith   PetscFunctionReturn(0);
1013273d9f13SBarry Smith }
1014273d9f13SBarry Smith 
101501b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1016eae6fb2eSBarry Smith 
1017eae6fb2eSBarry Smith #undef __FUNCT__
1018eae6fb2eSBarry Smith #define __FUNCT__ "MatMPIDenseCopyToPlapack"
1019eae6fb2eSBarry Smith PetscErrorCode MatMPIDenseCopyToPlapack(Mat A,Mat F)
1020eae6fb2eSBarry Smith {
1021eae6fb2eSBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
1022eae6fb2eSBarry Smith   PetscErrorCode ierr;
102379b0a62dSBarry Smith   PetscInt       M=A->cmap.N,m=A->rmap.n,rstart;
1024eae6fb2eSBarry Smith   PetscScalar    *array;
1025eae6fb2eSBarry Smith   PetscReal      one = 1.0;
1026eae6fb2eSBarry Smith 
1027eae6fb2eSBarry Smith   PetscFunctionBegin;
10282fbe02b9SBarry Smith   /* Copy A into F->lu->A */
1029eae6fb2eSBarry Smith   ierr = PLA_Obj_set_to_zero(lu->A);CHKERRQ(ierr);
1030eae6fb2eSBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
1031eae6fb2eSBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
103279b0a62dSBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
1033eae6fb2eSBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
1034eae6fb2eSBarry Smith   ierr = PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);CHKERRQ(ierr);
1035eae6fb2eSBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
1036eae6fb2eSBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
1037eae6fb2eSBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
1038eae6fb2eSBarry Smith   lu->rstart = rstart;
1039eae6fb2eSBarry Smith   PetscFunctionReturn(0);
1040eae6fb2eSBarry Smith }
1041eae6fb2eSBarry Smith 
104201b82886SBarry Smith #undef __FUNCT__
10432fbe02b9SBarry Smith #define __FUNCT__ "MatMPIDenseCopyFromPlapack"
10442fbe02b9SBarry Smith PetscErrorCode MatMPIDenseCopyFromPlapack(Mat F,Mat A)
10452fbe02b9SBarry Smith {
10462fbe02b9SBarry Smith   Mat_Plapack    *lu = (Mat_Plapack*)(F)->spptr;
10472fbe02b9SBarry Smith   PetscErrorCode ierr;
104879b0a62dSBarry Smith   PetscInt       M=A->cmap.N,m=A->rmap.n,rstart;
10492fbe02b9SBarry Smith   PetscScalar    *array;
10502fbe02b9SBarry Smith   PetscReal      one = 1.0;
10512fbe02b9SBarry Smith 
10522fbe02b9SBarry Smith   PetscFunctionBegin;
10532fbe02b9SBarry Smith   /* Copy F into A->lu->A */
105479b0a62dSBarry Smith   ierr = MatZeroEntries(A);CHKERRQ(ierr);
10552fbe02b9SBarry Smith   ierr = PLA_API_begin();CHKERRQ(ierr);
10562fbe02b9SBarry Smith   ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr);
10572fbe02b9SBarry Smith   ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr);
10582fbe02b9SBarry Smith   ierr = MatGetArray(A,&array);CHKERRQ(ierr);
10592fbe02b9SBarry Smith   ierr = PLA_API_axpy_global_to_matrix(m,M, &one,lu->A,rstart,0,(void *)array,m);CHKERRQ(ierr);
10602fbe02b9SBarry Smith   ierr = MatRestoreArray(A,&array);CHKERRQ(ierr);
10612fbe02b9SBarry Smith   ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr);
10622fbe02b9SBarry Smith   ierr = PLA_API_end();CHKERRQ(ierr);
10632fbe02b9SBarry Smith   lu->rstart = rstart;
10642fbe02b9SBarry Smith   PetscFunctionReturn(0);
10652fbe02b9SBarry Smith }
10662fbe02b9SBarry Smith 
10672fbe02b9SBarry Smith #undef __FUNCT__
10682fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultNumeric_MPIDense_MPIDense"
10692fbe02b9SBarry Smith PetscErrorCode MatMatMultNumeric_MPIDense_MPIDense(Mat A,Mat B,Mat C)
10702fbe02b9SBarry Smith {
10712fbe02b9SBarry Smith   PetscErrorCode ierr;
10722fbe02b9SBarry Smith   Mat_Plapack    *luA = (Mat_Plapack*)A->spptr;
10732fbe02b9SBarry Smith   Mat_Plapack    *luB = (Mat_Plapack*)B->spptr;
10742fbe02b9SBarry Smith   Mat_Plapack    *luC = (Mat_Plapack*)C->spptr;
10752fbe02b9SBarry Smith   PLA_Obj        alpha = NULL,beta = NULL;
10762fbe02b9SBarry Smith 
10772fbe02b9SBarry Smith   PetscFunctionBegin;
10782fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(A,A);CHKERRQ(ierr);
10792fbe02b9SBarry Smith   ierr = MatMPIDenseCopyToPlapack(B,B);CHKERRQ(ierr);
10802fbe02b9SBarry Smith 
10816e6f9017SBarry Smith   /*
1082cb2480eaSBarry Smith   ierr = PLA_Global_show("A = ",luA->A,"%g ","");CHKERRQ(ierr);
1083cb2480eaSBarry Smith   ierr = PLA_Global_show("B = ",luB->A,"%g ","");CHKERRQ(ierr);
10846e6f9017SBarry Smith   */
1085cb2480eaSBarry Smith 
10862fbe02b9SBarry Smith   /* do the multiply in PLA  */
108779b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luA->A,NULL,NULL,&alpha);CHKERRQ(ierr);
108879b0a62dSBarry Smith   ierr = PLA_Create_constants_conf_to(luC->A,NULL,&beta,NULL);CHKERRQ(ierr);
108979b0a62dSBarry Smith   CHKMEMQ;
10902fbe02b9SBarry Smith 
1091f0e3846dSSatish Balay   ierr = PLA_Gemm(PLA_NO_TRANSPOSE,PLA_NO_TRANSPOSE,alpha,luA->A,luB->A,beta,luC->A); /* CHKERRQ(ierr); */
109279b0a62dSBarry Smith   CHKMEMQ;
10932fbe02b9SBarry Smith   ierr = PLA_Obj_free(&alpha);CHKERRQ(ierr);
10942fbe02b9SBarry Smith   ierr = PLA_Obj_free(&beta);CHKERRQ(ierr);
10952fbe02b9SBarry Smith 
10966e6f9017SBarry Smith   /*
1097cb2480eaSBarry Smith   ierr = PLA_Global_show("C = ",luC->A,"%g ","");CHKERRQ(ierr);
10986e6f9017SBarry Smith   */
10992fbe02b9SBarry Smith   ierr = MatMPIDenseCopyFromPlapack(C,C);CHKERRQ(ierr);
11002fbe02b9SBarry Smith   PetscFunctionReturn(0);
11012fbe02b9SBarry Smith }
11022fbe02b9SBarry Smith 
11032fbe02b9SBarry Smith #undef __FUNCT__
11042fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultSymbolic_MPIDense_MPIDense"
11052fbe02b9SBarry Smith PetscErrorCode MatMatMultSymbolic_MPIDense_MPIDense(Mat A,Mat B,PetscReal fill,Mat *C)
11062fbe02b9SBarry Smith {
11072fbe02b9SBarry Smith   PetscErrorCode ierr;
11082fbe02b9SBarry Smith   PetscInt       m=A->rmap.n,n=B->cmap.n;
11092fbe02b9SBarry Smith   Mat            Cmat;
11102fbe02b9SBarry Smith 
11112fbe02b9SBarry Smith   PetscFunctionBegin;
11122fbe02b9SBarry Smith   if (A->cmap.n != B->rmap.n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->cmap.n %d != B->rmap.n %d\n",A->cmap.n,B->rmap.n);
11136e6f9017SBarry Smith   SETERRQ(PETSC_ERR_LIB,"Due to aparent bugs in PLAPACK,this is not currently supported");
11142fbe02b9SBarry Smith   ierr = MatCreate(((PetscObject)B)->comm,&Cmat);CHKERRQ(ierr);
11152fbe02b9SBarry Smith   ierr = MatSetSizes(Cmat,m,n,A->rmap.N,B->cmap.N);CHKERRQ(ierr);
11162fbe02b9SBarry Smith   ierr = MatSetType(Cmat,MATMPIDENSE);CHKERRQ(ierr);
11172fbe02b9SBarry Smith   ierr = MatAssemblyBegin(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11182fbe02b9SBarry Smith   ierr = MatAssemblyEnd(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1119*ab235cb6SHong Zhang   //ierr = MatMPIDenseCreatePlapack(A);CHKERRQ(ierr);
1120*ab235cb6SHong Zhang   //ierr = MatMPIDenseCreatePlapack(B);CHKERRQ(ierr);
1121*ab235cb6SHong Zhang   //ierr = MatMPIDenseCreatePlapack(Cmat);CHKERRQ(ierr);
11222fbe02b9SBarry Smith   *C = Cmat;
11232fbe02b9SBarry Smith   PetscFunctionReturn(0);
11242fbe02b9SBarry Smith }
11252fbe02b9SBarry Smith 
11262fbe02b9SBarry Smith #undef __FUNCT__
11272fbe02b9SBarry Smith #define __FUNCT__ "MatMatMult_MPIDense_MPIDense"
11282fbe02b9SBarry Smith PetscErrorCode MatMatMult_MPIDense_MPIDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
11292fbe02b9SBarry Smith {
11302fbe02b9SBarry Smith   PetscErrorCode ierr;
11312fbe02b9SBarry Smith 
11322fbe02b9SBarry Smith   PetscFunctionBegin;
11332fbe02b9SBarry Smith   if (scall == MAT_INITIAL_MATRIX){
11342fbe02b9SBarry Smith     ierr = MatMatMultSymbolic_MPIDense_MPIDense(A,B,fill,C);CHKERRQ(ierr);
11352fbe02b9SBarry Smith   }
11362fbe02b9SBarry Smith   ierr = MatMatMultNumeric_MPIDense_MPIDense(A,B,*C);CHKERRQ(ierr);
11372fbe02b9SBarry Smith   PetscFunctionReturn(0);
11382fbe02b9SBarry Smith }
11392fbe02b9SBarry Smith 
1140*ab235cb6SHong Zhang #undef __FUNCT__
1141*ab235cb6SHong Zhang #define __FUNCT__ "MatFactorSymbolic_Plapack_Private"
1142*ab235cb6SHong Zhang PetscErrorCode MatFactorSymbolic_Plapack_Private(Mat A,MatFactorInfo *info,Mat *F)
1143*ab235cb6SHong Zhang {
1144*ab235cb6SHong Zhang   Mat            B = *F;
1145*ab235cb6SHong Zhang   Mat_Plapack    *lu;
1146*ab235cb6SHong Zhang   PetscErrorCode ierr;
1147*ab235cb6SHong Zhang   PetscInt       M=A->rmap.N,N=A->cmap.N;
1148*ab235cb6SHong Zhang   MPI_Comm       comm=((PetscObject)A)->comm,comm_2d;
1149*ab235cb6SHong Zhang   PetscMPIInt    size;
1150*ab235cb6SHong Zhang   PetscInt       ierror;
1151*ab235cb6SHong Zhang 
1152*ab235cb6SHong Zhang   PetscFunctionBegin;
1153*ab235cb6SHong Zhang   lu = (Mat_Plapack*)(B->spptr);
1154*ab235cb6SHong Zhang 
1155*ab235cb6SHong Zhang   /* Set default Plapack parameters */
1156*ab235cb6SHong Zhang   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1157*ab235cb6SHong Zhang   lu->nprows = 1; lu->npcols = size;
1158*ab235cb6SHong Zhang   ierror = 0;
1159*ab235cb6SHong Zhang   lu->nb     = M/size;
1160*ab235cb6SHong Zhang   if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */
1161*ab235cb6SHong Zhang 
1162*ab235cb6SHong Zhang   /* Set runtime options */
1163*ab235cb6SHong Zhang   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr);
1164*ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",lu->nprows,&lu->nprows,PETSC_NULL);CHKERRQ(ierr);
1165*ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",lu->npcols,&lu->npcols,PETSC_NULL);CHKERRQ(ierr);
1166*ab235cb6SHong Zhang 
1167*ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr);
1168*ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",ierror,&ierror,PETSC_NULL);CHKERRQ(ierr);
1169*ab235cb6SHong Zhang   if (ierror){
1170*ab235cb6SHong Zhang     PLA_Set_error_checking(ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );
1171*ab235cb6SHong Zhang   } else {
1172*ab235cb6SHong Zhang     PLA_Set_error_checking(ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );
1173*ab235cb6SHong Zhang   }
1174*ab235cb6SHong Zhang   lu->ierror = ierror;
1175*ab235cb6SHong Zhang 
1176*ab235cb6SHong Zhang   lu->nb_alg = 0;
1177*ab235cb6SHong Zhang   ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",lu->nb_alg,&lu->nb_alg,PETSC_NULL);CHKERRQ(ierr);
1178*ab235cb6SHong Zhang   if (lu->nb_alg){
1179*ab235cb6SHong Zhang     pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,lu->nb_alg);
1180*ab235cb6SHong Zhang   }
1181*ab235cb6SHong Zhang   PetscOptionsEnd();
1182*ab235cb6SHong Zhang 
1183*ab235cb6SHong Zhang 
1184*ab235cb6SHong Zhang   /* Create a 2D communicator */
1185*ab235cb6SHong Zhang   PLA_Comm_1D_to_2D(comm,lu->nprows,lu->npcols,&comm_2d);
1186*ab235cb6SHong Zhang   lu->comm_2d = comm_2d;
1187*ab235cb6SHong Zhang 
1188*ab235cb6SHong Zhang   /* Initialize PLAPACK */
1189*ab235cb6SHong Zhang   PLA_Init(comm_2d);
1190*ab235cb6SHong Zhang 
1191*ab235cb6SHong Zhang   /* Create object distribution template */
1192*ab235cb6SHong Zhang   lu->templ = NULL;
1193*ab235cb6SHong Zhang   PLA_Temp_create(lu->nb, 0, &lu->templ);
1194*ab235cb6SHong Zhang 
1195*ab235cb6SHong Zhang   /* Use suggested nb_alg if it is not provided by user */
1196*ab235cb6SHong Zhang   if (lu->nb_alg == 0){
1197*ab235cb6SHong Zhang     PLA_Environ_nb_alg(PLA_OP_PAN_PAN,lu->templ,&lu->nb_alg);
1198*ab235cb6SHong Zhang     pla_Environ_set_nb_alg(PLA_OP_ALL_ALG,lu->nb_alg);
1199*ab235cb6SHong Zhang   }
1200*ab235cb6SHong Zhang 
1201*ab235cb6SHong Zhang   /* Set the datatype */
1202*ab235cb6SHong Zhang #if defined(PETSC_USE_COMPLEX)
1203*ab235cb6SHong Zhang   lu->datatype = MPI_DOUBLE_COMPLEX;
1204*ab235cb6SHong Zhang #else
1205*ab235cb6SHong Zhang   lu->datatype = MPI_DOUBLE;
1206*ab235cb6SHong Zhang #endif
1207*ab235cb6SHong Zhang 
1208*ab235cb6SHong Zhang   lu->pla_solved     = PETSC_FALSE; /* MatSolve_Plapack() is called yet */
1209*ab235cb6SHong Zhang   lu->mstruct        = DIFFERENT_NONZERO_PATTERN;
1210*ab235cb6SHong Zhang   lu->CleanUpPlapack = PETSC_TRUE;
1211*ab235cb6SHong Zhang   *F                 = B;
1212*ab235cb6SHong Zhang   PetscFunctionReturn(0);
1213*ab235cb6SHong Zhang }
1214*ab235cb6SHong Zhang 
1215b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */
12162fbe02b9SBarry Smith #undef __FUNCT__
1217b24902e0SBarry Smith #define __FUNCT__ "MatCholeskyFactorSymbolic_Plapack"
1218b24902e0SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_Plapack(Mat A,IS perm,MatFactorInfo *info,Mat *F)
121901b82886SBarry Smith {
122001b82886SBarry Smith   PetscErrorCode ierr;
1221b24902e0SBarry Smith   PetscTruth     issymmetric,set;
122201b82886SBarry Smith 
122301b82886SBarry Smith   PetscFunctionBegin;
1224b24902e0SBarry Smith   ierr = MatIsSymmetricKnown(A,&set,&issymmetric); CHKERRQ(ierr);
1225b24902e0SBarry Smith   if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()");
1226b24902e0SBarry Smith   ierr = MatFactorSymbolic_Plapack_Private(A,info,F);CHKERRQ(ierr);
12275c9eb25fSBarry Smith   (*F)->factor = MAT_FACTOR_CHOLESKY;
1228b24902e0SBarry Smith   PetscFunctionReturn(0);
122901b82886SBarry Smith }
123001b82886SBarry Smith 
1231b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */
1232b24902e0SBarry Smith #undef __FUNCT__
1233b24902e0SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_Plapack"
1234b24902e0SBarry Smith PetscErrorCode MatLUFactorSymbolic_Plapack(Mat A,IS r,IS c,MatFactorInfo *info,Mat *F)
1235b24902e0SBarry Smith {
1236b24902e0SBarry Smith   PetscErrorCode ierr;
1237b24902e0SBarry Smith   PetscInt       M = A->rmap.N;
1238b24902e0SBarry Smith   Mat_Plapack    *lu;
123901b82886SBarry Smith 
1240b24902e0SBarry Smith   PetscFunctionBegin;
1241b24902e0SBarry Smith   ierr = MatFactorSymbolic_Plapack_Private(A,info,F);CHKERRQ(ierr);
1242b24902e0SBarry Smith   lu = (Mat_Plapack*)(*F)->spptr;
1243b24902e0SBarry Smith   ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr);
12445c9eb25fSBarry Smith   (*F)->factor = MAT_FACTOR_LU;
124501b82886SBarry Smith   PetscFunctionReturn(0);
124601b82886SBarry Smith }
124701b82886SBarry Smith 
124801b82886SBarry Smith #undef __FUNCT__
1249b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_mpidense_plapack"
1250b24902e0SBarry Smith PetscErrorCode MatGetFactor_mpidense_plapack(Mat A,MatFactorType ftype,Mat *F)
125101b82886SBarry Smith {
125201b82886SBarry Smith   PetscErrorCode ierr;
125301b82886SBarry Smith 
125401b82886SBarry Smith   PetscFunctionBegin;
125501b82886SBarry Smith   /* Create the factorization matrix */
1256eae6fb2eSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr);
1257eae6fb2eSBarry Smith   ierr = MatSetSizes(*F,A->rmap.n,A->cmap.n,A->rmap.N,A->cmap.N);CHKERRQ(ierr);
1258eae6fb2eSBarry Smith   ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr);
1259b24902e0SBarry Smith   ierr = PetscNewLog(A,Mat_Plapack,&lu);CHKERRQ(ierr);
1260b24902e0SBarry Smith   A->spptr = (void*)lu;
126101b82886SBarry Smith 
1262b24902e0SBarry Smith   lu = (Mat_Plapack*)(A->spptr);
126301b82886SBarry Smith 
1264b24902e0SBarry Smith   /* Set default Plapack parameters */
1265b24902e0SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1266b24902e0SBarry Smith   lu->nb     = M/size;
1267b24902e0SBarry Smith   if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */
126801b82886SBarry Smith 
1269b24902e0SBarry Smith   /* Set runtime options */
1270b24902e0SBarry Smith   ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr);
1271b24902e0SBarry Smith     ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr);
1272b24902e0SBarry Smith   PetscOptionsEnd();
127301b82886SBarry Smith 
1274b24902e0SBarry Smith   /* Create object distribution template */
1275b24902e0SBarry Smith   lu->templ = NULL;
1276b24902e0SBarry Smith   ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr);
1277b24902e0SBarry Smith 
1278b24902e0SBarry Smith   /* Set the datatype */
1279b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX)
1280b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE_COMPLEX;
1281b24902e0SBarry Smith #else
1282b24902e0SBarry Smith   lu->datatype = MPI_DOUBLE;
1283b24902e0SBarry Smith #endif
1284b24902e0SBarry Smith 
1285b24902e0SBarry Smith   ierr = PLA_Matrix_create(lu->datatype,M,A->cmap.N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr);
1286b24902e0SBarry Smith 
1287b24902e0SBarry Smith 
1288b24902e0SBarry Smith   lu->pla_solved     = PETSC_FALSE; /* MatSolve_Plapack() is called yet */
1289b24902e0SBarry Smith 
1290b24902e0SBarry Smith   if (ftype == MAT_FACTOR_LU) {
1291b24902e0SBarry Smith     (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense;
1292b24902e0SBarry Smith     (*F)->ops->lufactornumeric  = MatLUFactorNumeric_MPIDense;
1293b24902e0SBarry Smith     (*F)->ops->solve            = MatSolve_MPIDense;
1294b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY) {
1295b24902e0SBarry Smith     (*F)->ops->choleksyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense;
1296b24902e0SBarry Smith     (*F)->ops->choleskyfactornumeric  = MatCholeksyFactorNumeric_MPIDense;
1297b24902e0SBarry Smith     (*F)->ops->solve                  = MatSolve_MPIDense;
1298b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices");
1299b24902e0SBarry Smith 
130001b82886SBarry Smith   PetscFunctionReturn(0);
130101b82886SBarry Smith }
130201b82886SBarry Smith #endif
130301b82886SBarry Smith 
13048965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/
130509dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense,
130609dc0095SBarry Smith        MatGetRow_MPIDense,
130709dc0095SBarry Smith        MatRestoreRow_MPIDense,
130809dc0095SBarry Smith        MatMult_MPIDense,
130997304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense,
13107c922b88SBarry Smith        MatMultTranspose_MPIDense,
13117c922b88SBarry Smith        MatMultTransposeAdd_MPIDense,
131201b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
131301b82886SBarry Smith        MatSolve_MPIDense,
131401b82886SBarry Smith #else
13158965ea79SLois Curfman McInnes        0,
131601b82886SBarry Smith #endif
131709dc0095SBarry Smith        0,
131809dc0095SBarry Smith        0,
131997304618SKris Buschelman /*10*/ 0,
132009dc0095SBarry Smith        0,
132109dc0095SBarry Smith        0,
132209dc0095SBarry Smith        0,
132309dc0095SBarry Smith        MatTranspose_MPIDense,
132497304618SKris Buschelman /*15*/ MatGetInfo_MPIDense,
13256e4ee0c6SHong Zhang        MatEqual_MPIDense,
132609dc0095SBarry Smith        MatGetDiagonal_MPIDense,
13275b2fa520SLois Curfman McInnes        MatDiagonalScale_MPIDense,
132809dc0095SBarry Smith        MatNorm_MPIDense,
132997304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense,
133009dc0095SBarry Smith        MatAssemblyEnd_MPIDense,
133109dc0095SBarry Smith        0,
133209dc0095SBarry Smith        MatSetOption_MPIDense,
133309dc0095SBarry Smith        MatZeroEntries_MPIDense,
133497304618SKris Buschelman /*25*/ MatZeroRows_MPIDense,
133501b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
13360ad19415SHong Zhang        MatLUFactorSymbolic_MPIDense,
133701b82886SBarry Smith        MatLUFactorNumeric_MPIDense,
13380ad19415SHong Zhang        MatCholeskyFactorSymbolic_MPIDense,
133901b82886SBarry Smith        MatCholeskyFactorNumeric_MPIDense,
134001b82886SBarry Smith #else
1341919b68f7SBarry Smith        0,
134201b82886SBarry Smith        0,
134301b82886SBarry Smith        0,
134401b82886SBarry Smith        0,
134501b82886SBarry Smith #endif
134697304618SKris Buschelman /*30*/ MatSetUpPreallocation_MPIDense,
1347273d9f13SBarry Smith        0,
134809dc0095SBarry Smith        0,
134909dc0095SBarry Smith        MatGetArray_MPIDense,
135009dc0095SBarry Smith        MatRestoreArray_MPIDense,
135197304618SKris Buschelman /*35*/ MatDuplicate_MPIDense,
135209dc0095SBarry Smith        0,
135309dc0095SBarry Smith        0,
135409dc0095SBarry Smith        0,
135509dc0095SBarry Smith        0,
135697304618SKris Buschelman /*40*/ 0,
13572ce60cd0SSatish Balay        MatGetSubMatrices_MPIDense,
135809dc0095SBarry Smith        0,
135909dc0095SBarry Smith        MatGetValues_MPIDense,
136009dc0095SBarry Smith        0,
136197304618SKris Buschelman /*45*/ 0,
136209dc0095SBarry Smith        MatScale_MPIDense,
136309dc0095SBarry Smith        0,
136409dc0095SBarry Smith        0,
136509dc0095SBarry Smith        0,
1366521d7252SBarry Smith /*50*/ 0,
136709dc0095SBarry Smith        0,
136809dc0095SBarry Smith        0,
136909dc0095SBarry Smith        0,
137009dc0095SBarry Smith        0,
137197304618SKris Buschelman /*55*/ 0,
137209dc0095SBarry Smith        0,
137309dc0095SBarry Smith        0,
137409dc0095SBarry Smith        0,
137509dc0095SBarry Smith        0,
137697304618SKris Buschelman /*60*/ MatGetSubMatrix_MPIDense,
1377b9b97703SBarry Smith        MatDestroy_MPIDense,
1378b9b97703SBarry Smith        MatView_MPIDense,
1379357abbc8SBarry Smith        0,
138097304618SKris Buschelman        0,
138197304618SKris Buschelman /*65*/ 0,
138297304618SKris Buschelman        0,
138397304618SKris Buschelman        0,
138497304618SKris Buschelman        0,
138597304618SKris Buschelman        0,
138697304618SKris Buschelman /*70*/ 0,
138797304618SKris Buschelman        0,
138897304618SKris Buschelman        0,
138997304618SKris Buschelman        0,
139097304618SKris Buschelman        0,
139197304618SKris Buschelman /*75*/ 0,
139297304618SKris Buschelman        0,
139397304618SKris Buschelman        0,
139497304618SKris Buschelman        0,
139597304618SKris Buschelman        0,
139697304618SKris Buschelman /*80*/ 0,
139797304618SKris Buschelman        0,
139897304618SKris Buschelman        0,
139997304618SKris Buschelman        0,
1400865e5f61SKris Buschelman /*84*/ MatLoad_MPIDense,
1401865e5f61SKris Buschelman        0,
1402865e5f61SKris Buschelman        0,
1403865e5f61SKris Buschelman        0,
1404865e5f61SKris Buschelman        0,
1405865e5f61SKris Buschelman        0,
14062fbe02b9SBarry Smith /*90*/
14072fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
14082fbe02b9SBarry Smith        MatMatMult_MPIDense_MPIDense,
14094ae313f4SHong Zhang        MatMatMultSymbolic_MPIDense_MPIDense,
14102fbe02b9SBarry Smith        MatMatMultNumeric_MPIDense_MPIDense,
14112fbe02b9SBarry Smith #else
1412865e5f61SKris Buschelman        0,
1413865e5f61SKris Buschelman        0,
1414865e5f61SKris Buschelman        0,
14152fbe02b9SBarry Smith #endif
14162fbe02b9SBarry Smith        0,
1417865e5f61SKris Buschelman /*95*/ 0,
1418865e5f61SKris Buschelman        0,
1419865e5f61SKris Buschelman        0,
1420865e5f61SKris Buschelman        0};
14218965ea79SLois Curfman McInnes 
1422273d9f13SBarry Smith EXTERN_C_BEGIN
14234a2ae208SSatish Balay #undef __FUNCT__
1424a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense"
1425be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data)
1426a23d5eceSKris Buschelman {
1427a23d5eceSKris Buschelman   Mat_MPIDense   *a;
1428dfbe8321SBarry Smith   PetscErrorCode ierr;
1429a23d5eceSKris Buschelman 
1430a23d5eceSKris Buschelman   PetscFunctionBegin;
1431a23d5eceSKris Buschelman   mat->preallocated = PETSC_TRUE;
1432a23d5eceSKris Buschelman   /* Note:  For now, when data is specified above, this assumes the user correctly
1433a23d5eceSKris Buschelman    allocates the local dense storage space.  We should add error checking. */
1434a23d5eceSKris Buschelman 
1435a23d5eceSKris Buschelman   a    = (Mat_MPIDense*)mat->data;
1436f69a0ea3SMatthew Knepley   ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr);
1437899cda47SBarry Smith   ierr = MatSetSizes(a->A,mat->rmap.n,mat->cmap.N,mat->rmap.n,mat->cmap.N);CHKERRQ(ierr);
14385c5985e7SKris Buschelman   ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr);
14395c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr);
144052e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
1441a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1442a23d5eceSKris Buschelman }
1443a23d5eceSKris Buschelman EXTERN_C_END
1444a23d5eceSKris Buschelman 
14450bad9183SKris Buschelman /*MC
1446fafad747SKris Buschelman    MATMPIDENSE - MATMPIDENSE = "mpidense" - A matrix type to be used for distributed dense matrices.
14470bad9183SKris Buschelman 
14480bad9183SKris Buschelman    Options Database Keys:
14490bad9183SKris Buschelman . -mat_type mpidense - sets the matrix type to "mpidense" during a call to MatSetFromOptions()
14500bad9183SKris Buschelman 
14510bad9183SKris Buschelman   Level: beginner
14520bad9183SKris Buschelman 
14537878bbefSBarry Smith   MATMPIDENSE matrices may use direct solvers (LU, Cholesky, and QR)
14547878bbefSBarry Smith   for parallel dense matrices via the external package PLAPACK, if PLAPACK is installed
14557878bbefSBarry Smith   (run config/configure.py with the option --download-plapack)
14567878bbefSBarry Smith 
14577878bbefSBarry Smith 
14587878bbefSBarry Smith   Options Database Keys:
14597878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition
14607878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition
14617878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector
14627878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size
14637878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag
14647878bbefSBarry Smith 
14657878bbefSBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE
14660bad9183SKris Buschelman M*/
14670bad9183SKris Buschelman 
1468a23d5eceSKris Buschelman EXTERN_C_BEGIN
1469a23d5eceSKris Buschelman #undef __FUNCT__
14704a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense"
1471be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat)
1472273d9f13SBarry Smith {
1473273d9f13SBarry Smith   Mat_MPIDense   *a;
1474dfbe8321SBarry Smith   PetscErrorCode ierr;
1475273d9f13SBarry Smith 
1476273d9f13SBarry Smith   PetscFunctionBegin;
147738f2d2fdSLisandro Dalcin   ierr              = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr);
1478b0a32e0cSBarry Smith   mat->data         = (void*)a;
1479273d9f13SBarry Smith   ierr              = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1480273d9f13SBarry Smith   mat->mapping      = 0;
1481273d9f13SBarry Smith 
1482273d9f13SBarry Smith   mat->insertmode = NOT_SET_VALUES;
14837adad957SLisandro Dalcin   ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr);
14847adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr);
1485273d9f13SBarry Smith 
1486899cda47SBarry Smith   mat->rmap.bs = mat->cmap.bs = 1;
14876148ca0dSBarry Smith   ierr = PetscMapSetUp(&mat->rmap);CHKERRQ(ierr);
14886148ca0dSBarry Smith   ierr = PetscMapSetUp(&mat->cmap);CHKERRQ(ierr);
1489899cda47SBarry Smith   a->nvec = mat->cmap.n;
1490273d9f13SBarry Smith 
1491273d9f13SBarry Smith   /* build cache for off array entries formed */
1492273d9f13SBarry Smith   a->donotstash = PETSC_FALSE;
14937adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr);
1494273d9f13SBarry Smith 
1495273d9f13SBarry Smith   /* stuff used for matrix vector multiply */
1496273d9f13SBarry Smith   a->lvec        = 0;
1497273d9f13SBarry Smith   a->Mvctx       = 0;
1498273d9f13SBarry Smith   a->roworiented = PETSC_TRUE;
1499273d9f13SBarry Smith 
1500273d9f13SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C",
1501273d9f13SBarry Smith                                      "MatGetDiagonalBlock_MPIDense",
1502273d9f13SBarry Smith                                      MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr);
1503a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C",
1504a23d5eceSKris Buschelman                                      "MatMPIDenseSetPreallocation_MPIDense",
1505a23d5eceSKris Buschelman                                      MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr);
15064ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C",
15074ae313f4SHong Zhang                                      "MatMatMult_MPIAIJ_MPIDense",
15084ae313f4SHong Zhang                                       MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr);
15094ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C",
15104ae313f4SHong Zhang                                      "MatMatMultSymbolic_MPIAIJ_MPIDense",
15114ae313f4SHong Zhang                                       MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr);
15124ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C",
15134ae313f4SHong Zhang                                      "MatMatMultNumeric_MPIAIJ_MPIDense",
15144ae313f4SHong Zhang                                       MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr);
15157878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
1516b24902e0SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_plapack_C",
1517b24902e0SBarry Smith                                      "MatGetFactor_mpidense_plapack",
1518b24902e0SBarry Smith                                       MatGetFactor_mpidense_plapack);CHKERRQ(ierr);
15197878bbefSBarry Smith #endif
152038aed534SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr);
152101b82886SBarry Smith 
1522273d9f13SBarry Smith   PetscFunctionReturn(0);
1523273d9f13SBarry Smith }
1524273d9f13SBarry Smith EXTERN_C_END
1525273d9f13SBarry Smith 
1526209238afSKris Buschelman /*MC
1527002d173eSKris Buschelman    MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices.
1528209238afSKris Buschelman 
1529209238afSKris Buschelman    This matrix type is identical to MATSEQDENSE when constructed with a single process communicator,
1530209238afSKris Buschelman    and MATMPIDENSE otherwise.
1531209238afSKris Buschelman 
1532209238afSKris Buschelman    Options Database Keys:
1533209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions()
1534209238afSKris Buschelman 
1535209238afSKris Buschelman   Level: beginner
1536209238afSKris Buschelman 
153701b82886SBarry Smith 
1538209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE
1539209238afSKris Buschelman M*/
1540209238afSKris Buschelman 
1541209238afSKris Buschelman EXTERN_C_BEGIN
1542209238afSKris Buschelman #undef __FUNCT__
1543209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense"
1544be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A)
1545dfbe8321SBarry Smith {
15466849ba73SBarry Smith   PetscErrorCode ierr;
154713f74950SBarry Smith   PetscMPIInt    size;
1548209238afSKris Buschelman 
1549209238afSKris Buschelman   PetscFunctionBegin;
15507adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr);
1551209238afSKris Buschelman   if (size == 1) {
1552209238afSKris Buschelman     ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr);
1553209238afSKris Buschelman   } else {
1554209238afSKris Buschelman     ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr);
1555209238afSKris Buschelman   }
1556209238afSKris Buschelman   PetscFunctionReturn(0);
1557209238afSKris Buschelman }
1558209238afSKris Buschelman EXTERN_C_END
1559209238afSKris Buschelman 
15604a2ae208SSatish Balay #undef __FUNCT__
15614a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation"
1562273d9f13SBarry Smith /*@C
1563273d9f13SBarry Smith    MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries
1564273d9f13SBarry Smith 
1565273d9f13SBarry Smith    Not collective
1566273d9f13SBarry Smith 
1567273d9f13SBarry Smith    Input Parameters:
1568273d9f13SBarry Smith .  A - the matrix
1569273d9f13SBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1570273d9f13SBarry Smith    to control all matrix memory allocation.
1571273d9f13SBarry Smith 
1572273d9f13SBarry Smith    Notes:
1573273d9f13SBarry Smith    The dense format is fully compatible with standard Fortran 77
1574273d9f13SBarry Smith    storage by columns.
1575273d9f13SBarry Smith 
1576273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1577273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1578273d9f13SBarry Smith    set data=PETSC_NULL.
1579273d9f13SBarry Smith 
1580273d9f13SBarry Smith    Level: intermediate
1581273d9f13SBarry Smith 
1582273d9f13SBarry Smith .keywords: matrix,dense, parallel
1583273d9f13SBarry Smith 
1584273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
1585273d9f13SBarry Smith @*/
1586be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data)
1587273d9f13SBarry Smith {
1588dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar *);
1589273d9f13SBarry Smith 
1590273d9f13SBarry Smith   PetscFunctionBegin;
1591565567f0SKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1592a23d5eceSKris Buschelman   if (f) {
1593a23d5eceSKris Buschelman     ierr = (*f)(mat,data);CHKERRQ(ierr);
1594a23d5eceSKris Buschelman   }
1595273d9f13SBarry Smith   PetscFunctionReturn(0);
1596273d9f13SBarry Smith }
1597273d9f13SBarry Smith 
15984a2ae208SSatish Balay #undef __FUNCT__
15994a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense"
16008965ea79SLois Curfman McInnes /*@C
160139ddd567SLois Curfman McInnes    MatCreateMPIDense - Creates a sparse parallel matrix in dense format.
16028965ea79SLois Curfman McInnes 
1603db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1604db81eaa0SLois Curfman McInnes 
16058965ea79SLois Curfman McInnes    Input Parameters:
1606db81eaa0SLois Curfman McInnes +  comm - MPI communicator
16078965ea79SLois Curfman McInnes .  m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
1608db81eaa0SLois Curfman McInnes .  n - number of local columns (or PETSC_DECIDE to have calculated if N is given)
16098965ea79SLois Curfman McInnes .  M - number of global rows (or PETSC_DECIDE to have calculated if m is given)
1610db81eaa0SLois Curfman McInnes .  N - number of global columns (or PETSC_DECIDE to have calculated if n is given)
16117f5ff6fdSBarry Smith -  data - optional location of matrix data.  Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc
1612dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
16138965ea79SLois Curfman McInnes 
16148965ea79SLois Curfman McInnes    Output Parameter:
1615477f1c0bSLois Curfman McInnes .  A - the matrix
16168965ea79SLois Curfman McInnes 
1617b259b22eSLois Curfman McInnes    Notes:
161839ddd567SLois Curfman McInnes    The dense format is fully compatible with standard Fortran 77
161939ddd567SLois Curfman McInnes    storage by columns.
16208965ea79SLois Curfman McInnes 
162118f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
162218f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
16237f5ff6fdSBarry Smith    set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users).
162418f449edSLois Curfman McInnes 
16258965ea79SLois Curfman McInnes    The user MUST specify either the local or global matrix dimensions
16268965ea79SLois Curfman McInnes    (possibly both).
16278965ea79SLois Curfman McInnes 
1628027ccd11SLois Curfman McInnes    Level: intermediate
1629027ccd11SLois Curfman McInnes 
163039ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel
16318965ea79SLois Curfman McInnes 
163239ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues()
16338965ea79SLois Curfman McInnes @*/
1634be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A)
16358965ea79SLois Curfman McInnes {
16366849ba73SBarry Smith   PetscErrorCode ierr;
163713f74950SBarry Smith   PetscMPIInt    size;
16388965ea79SLois Curfman McInnes 
16393a40ed3dSBarry Smith   PetscFunctionBegin;
1640f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1641f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr);
1642273d9f13SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1643273d9f13SBarry Smith   if (size > 1) {
1644273d9f13SBarry Smith     ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr);
1645273d9f13SBarry Smith     ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1646273d9f13SBarry Smith   } else {
1647273d9f13SBarry Smith     ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1648273d9f13SBarry Smith     ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
16498c469469SLois Curfman McInnes   }
16503a40ed3dSBarry Smith   PetscFunctionReturn(0);
16518965ea79SLois Curfman McInnes }
16528965ea79SLois Curfman McInnes 
16534a2ae208SSatish Balay #undef __FUNCT__
16544a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense"
16556849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
16568965ea79SLois Curfman McInnes {
16578965ea79SLois Curfman McInnes   Mat            mat;
16583501a2bdSLois Curfman McInnes   Mat_MPIDense   *a,*oldmat = (Mat_MPIDense*)A->data;
1659dfbe8321SBarry Smith   PetscErrorCode ierr;
16608965ea79SLois Curfman McInnes 
16613a40ed3dSBarry Smith   PetscFunctionBegin;
16628965ea79SLois Curfman McInnes   *newmat       = 0;
16637adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr);
1664899cda47SBarry Smith   ierr = MatSetSizes(mat,A->rmap.n,A->cmap.n,A->rmap.N,A->cmap.N);CHKERRQ(ierr);
16657adad957SLisandro Dalcin   ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr);
1666834f8fabSBarry Smith   a                 = (Mat_MPIDense*)mat->data;
1667e04c1aa4SHong Zhang   ierr              = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
16683501a2bdSLois Curfman McInnes   mat->factor       = A->factor;
1669c456f294SBarry Smith   mat->assembled    = PETSC_TRUE;
1670273d9f13SBarry Smith   mat->preallocated = PETSC_TRUE;
16718965ea79SLois Curfman McInnes 
1672899cda47SBarry Smith   mat->rmap.rstart     = A->rmap.rstart;
1673899cda47SBarry Smith   mat->rmap.rend       = A->rmap.rend;
16748965ea79SLois Curfman McInnes   a->size              = oldmat->size;
16758965ea79SLois Curfman McInnes   a->rank              = oldmat->rank;
1676e0fa3b82SLois Curfman McInnes   mat->insertmode      = NOT_SET_VALUES;
1677b9b97703SBarry Smith   a->nvec              = oldmat->nvec;
16783782ba37SSatish Balay   a->donotstash        = oldmat->donotstash;
1679e04c1aa4SHong Zhang 
1680899cda47SBarry Smith   ierr = PetscMemcpy(mat->rmap.range,A->rmap.range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
1681899cda47SBarry Smith   ierr = PetscMemcpy(mat->cmap.range,A->cmap.range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr);
16827adad957SLisandro Dalcin   ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr);
16838965ea79SLois Curfman McInnes 
1684329f5518SBarry Smith   ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr);
16855609ef8eSBarry Smith   ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr);
168652e6d16bSBarry Smith   ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr);
168701b82886SBarry Smith 
168801b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK)
168901b82886SBarry Smith   ierr = PetscMemcpy(mat->spptr,A->spptr,sizeof(Mat_Plapack));CHKERRQ(ierr);
169001b82886SBarry Smith #endif
16918965ea79SLois Curfman McInnes   *newmat = mat;
16923a40ed3dSBarry Smith   PetscFunctionReturn(0);
16938965ea79SLois Curfman McInnes }
16948965ea79SLois Curfman McInnes 
1695e090d566SSatish Balay #include "petscsys.h"
16968965ea79SLois Curfman McInnes 
16974a2ae208SSatish Balay #undef __FUNCT__
16984a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile"
1699a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat)
170090ace30eSBarry Smith {
17016849ba73SBarry Smith   PetscErrorCode ierr;
170232dcc486SBarry Smith   PetscMPIInt    rank,size;
170313f74950SBarry Smith   PetscInt       *rowners,i,m,nz,j;
170487828ca2SBarry Smith   PetscScalar    *array,*vals,*vals_ptr;
170590ace30eSBarry Smith   MPI_Status     status;
170690ace30eSBarry Smith 
17073a40ed3dSBarry Smith   PetscFunctionBegin;
1708d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
1709d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
171090ace30eSBarry Smith 
171190ace30eSBarry Smith   /* determine ownership of all rows */
171290ace30eSBarry Smith   m          = M/size + ((M % size) > rank);
171313f74950SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr);
171413f74950SBarry Smith   ierr       = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr);
171590ace30eSBarry Smith   rowners[0] = 0;
171690ace30eSBarry Smith   for (i=2; i<=size; i++) {
171790ace30eSBarry Smith     rowners[i] += rowners[i-1];
171890ace30eSBarry Smith   }
171990ace30eSBarry Smith 
1720f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1721f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1722be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1723878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
172490ace30eSBarry Smith   ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr);
172590ace30eSBarry Smith 
172690ace30eSBarry Smith   if (!rank) {
172787828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
172890ace30eSBarry Smith 
172990ace30eSBarry Smith     /* read in my part of the matrix numerical values  */
17300752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr);
173190ace30eSBarry Smith 
173290ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
173390ace30eSBarry Smith     vals_ptr = vals;
173490ace30eSBarry Smith     for (i=0; i<m; i++) {
173590ace30eSBarry Smith       for (j=0; j<N; j++) {
173690ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
173790ace30eSBarry Smith       }
173890ace30eSBarry Smith     }
173990ace30eSBarry Smith 
174090ace30eSBarry Smith     /* read in other processors and ship out */
174190ace30eSBarry Smith     for (i=1; i<size; i++) {
174290ace30eSBarry Smith       nz   = (rowners[i+1] - rowners[i])*N;
17430752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
17447adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr);
174590ace30eSBarry Smith     }
17463a40ed3dSBarry Smith   } else {
174790ace30eSBarry Smith     /* receive numeric values */
174887828ca2SBarry Smith     ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
174990ace30eSBarry Smith 
175090ace30eSBarry Smith     /* receive message of values*/
17517adad957SLisandro Dalcin     ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr);
175290ace30eSBarry Smith 
175390ace30eSBarry Smith     /* insert into matrix-by row (this is why cannot directly read into array */
175490ace30eSBarry Smith     vals_ptr = vals;
175590ace30eSBarry Smith     for (i=0; i<m; i++) {
175690ace30eSBarry Smith       for (j=0; j<N; j++) {
175790ace30eSBarry Smith         array[i + j*m] = *vals_ptr++;
175890ace30eSBarry Smith       }
175990ace30eSBarry Smith     }
176090ace30eSBarry Smith   }
1761606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
1762606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
17636d4a8577SBarry Smith   ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
17646d4a8577SBarry Smith   ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
17653a40ed3dSBarry Smith   PetscFunctionReturn(0);
176690ace30eSBarry Smith }
176790ace30eSBarry Smith 
17684a2ae208SSatish Balay #undef __FUNCT__
17694a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense"
1770a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat)
17718965ea79SLois Curfman McInnes {
17728965ea79SLois Curfman McInnes   Mat            A;
177387828ca2SBarry Smith   PetscScalar    *vals,*svals;
177419bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
17758965ea79SLois Curfman McInnes   MPI_Status     status;
177613f74950SBarry Smith   PetscMPIInt    rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz;
177713f74950SBarry Smith   PetscInt       header[4],*rowlengths = 0,M,N,*cols;
177813f74950SBarry Smith   PetscInt       *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols;
177913f74950SBarry Smith   PetscInt       i,nz,j,rstart,rend;
178013f74950SBarry Smith   int            fd;
17816849ba73SBarry Smith   PetscErrorCode ierr;
17828965ea79SLois Curfman McInnes 
17833a40ed3dSBarry Smith   PetscFunctionBegin;
1784d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
1785d132466eSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
17868965ea79SLois Curfman McInnes   if (!rank) {
1787b0a32e0cSBarry Smith     ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
17880752156aSBarry Smith     ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr);
1789552e946dSBarry Smith     if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object");
17908965ea79SLois Curfman McInnes   }
17918965ea79SLois Curfman McInnes 
179213f74950SBarry Smith   ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr);
179390ace30eSBarry Smith   M = header[1]; N = header[2]; nz = header[3];
179490ace30eSBarry Smith 
179590ace30eSBarry Smith   /*
179690ace30eSBarry Smith        Handle case where matrix is stored on disk as a dense matrix
179790ace30eSBarry Smith   */
179890ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) {
1799be5d1d56SKris Buschelman     ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr);
18003a40ed3dSBarry Smith     PetscFunctionReturn(0);
180190ace30eSBarry Smith   }
180290ace30eSBarry Smith 
18038965ea79SLois Curfman McInnes   /* determine ownership of all rows */
1804e44c0bd4SBarry Smith   m          = PetscMPIIntCast(M/size + ((M % size) > rank));
1805e44c0bd4SBarry Smith   ierr       = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr);
1806ca161407SBarry Smith   ierr       = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr);
18078965ea79SLois Curfman McInnes   rowners[0] = 0;
18088965ea79SLois Curfman McInnes   for (i=2; i<=size; i++) {
18098965ea79SLois Curfman McInnes     rowners[i] += rowners[i-1];
18108965ea79SLois Curfman McInnes   }
18118965ea79SLois Curfman McInnes   rstart = rowners[rank];
18128965ea79SLois Curfman McInnes   rend   = rowners[rank+1];
18138965ea79SLois Curfman McInnes 
18148965ea79SLois Curfman McInnes   /* distribute row lengths to all processors */
181513f74950SBarry Smith   ierr    = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr);
18168965ea79SLois Curfman McInnes   offlens = ourlens + (rend-rstart);
18178965ea79SLois Curfman McInnes   if (!rank) {
181813f74950SBarry Smith     ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
18190752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
182013f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr);
18218965ea79SLois Curfman McInnes     for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i];
18221466f1e1SBarry Smith     ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
1823606d414cSSatish Balay     ierr = PetscFree(sndcounts);CHKERRQ(ierr);
1824ca161407SBarry Smith   } else {
18251466f1e1SBarry Smith     ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr);
18268965ea79SLois Curfman McInnes   }
18278965ea79SLois Curfman McInnes 
18288965ea79SLois Curfman McInnes   if (!rank) {
18298965ea79SLois Curfman McInnes     /* calculate the number of nonzeros on each processor */
183013f74950SBarry Smith     ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr);
183113f74950SBarry Smith     ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr);
18328965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
18338965ea79SLois Curfman McInnes       for (j=rowners[i]; j< rowners[i+1]; j++) {
18348965ea79SLois Curfman McInnes         procsnz[i] += rowlengths[j];
18358965ea79SLois Curfman McInnes       }
18368965ea79SLois Curfman McInnes     }
1837606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
18388965ea79SLois Curfman McInnes 
18398965ea79SLois Curfman McInnes     /* determine max buffer needed and allocate it */
18408965ea79SLois Curfman McInnes     maxnz = 0;
18418965ea79SLois Curfman McInnes     for (i=0; i<size; i++) {
18420452661fSBarry Smith       maxnz = PetscMax(maxnz,procsnz[i]);
18438965ea79SLois Curfman McInnes     }
184413f74950SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr);
18458965ea79SLois Curfman McInnes 
18468965ea79SLois Curfman McInnes     /* read in my part of the matrix column indices  */
18478965ea79SLois Curfman McInnes     nz = procsnz[0];
184813f74950SBarry Smith     ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
18490752156aSBarry Smith     ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr);
18508965ea79SLois Curfman McInnes 
18518965ea79SLois Curfman McInnes     /* read in every one elses and ship off */
18528965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
18538965ea79SLois Curfman McInnes       nz   = procsnz[i];
18540752156aSBarry Smith       ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
185513f74950SBarry Smith       ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr);
18568965ea79SLois Curfman McInnes     }
1857606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
18583a40ed3dSBarry Smith   } else {
18598965ea79SLois Curfman McInnes     /* determine buffer space needed for message */
18608965ea79SLois Curfman McInnes     nz = 0;
18618965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
18628965ea79SLois Curfman McInnes       nz += ourlens[i];
18638965ea79SLois Curfman McInnes     }
186413f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr);
18658965ea79SLois Curfman McInnes 
18668965ea79SLois Curfman McInnes     /* receive message of column indices*/
186713f74950SBarry Smith     ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr);
186813f74950SBarry Smith     ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr);
186929bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
18708965ea79SLois Curfman McInnes   }
18718965ea79SLois Curfman McInnes 
18728965ea79SLois Curfman McInnes   /* loop over local rows, determining number of off diagonal entries */
187313f74950SBarry Smith   ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr);
18748965ea79SLois Curfman McInnes   jj = 0;
18758965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
18768965ea79SLois Curfman McInnes     for (j=0; j<ourlens[i]; j++) {
18778965ea79SLois Curfman McInnes       if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++;
18788965ea79SLois Curfman McInnes       jj++;
18798965ea79SLois Curfman McInnes     }
18808965ea79SLois Curfman McInnes   }
18818965ea79SLois Curfman McInnes 
18828965ea79SLois Curfman McInnes   /* create our matrix */
18838965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
18848965ea79SLois Curfman McInnes     ourlens[i] -= offlens[i];
18858965ea79SLois Curfman McInnes   }
1886f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,newmat);CHKERRQ(ierr);
1887f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr);
1888be5d1d56SKris Buschelman   ierr = MatSetType(*newmat,type);CHKERRQ(ierr);
1889878740d9SKris Buschelman   ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
18908965ea79SLois Curfman McInnes   A = *newmat;
18918965ea79SLois Curfman McInnes   for (i=0; i<m; i++) {
18928965ea79SLois Curfman McInnes     ourlens[i] += offlens[i];
18938965ea79SLois Curfman McInnes   }
18948965ea79SLois Curfman McInnes 
18958965ea79SLois Curfman McInnes   if (!rank) {
189687828ca2SBarry Smith     ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
18978965ea79SLois Curfman McInnes 
18988965ea79SLois Curfman McInnes     /* read in my part of the matrix numerical values  */
18998965ea79SLois Curfman McInnes     nz = procsnz[0];
19000752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
19018965ea79SLois Curfman McInnes 
19028965ea79SLois Curfman McInnes     /* insert into matrix */
19038965ea79SLois Curfman McInnes     jj      = rstart;
19048965ea79SLois Curfman McInnes     smycols = mycols;
19058965ea79SLois Curfman McInnes     svals   = vals;
19068965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
19078965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
19088965ea79SLois Curfman McInnes       smycols += ourlens[i];
19098965ea79SLois Curfman McInnes       svals   += ourlens[i];
19108965ea79SLois Curfman McInnes       jj++;
19118965ea79SLois Curfman McInnes     }
19128965ea79SLois Curfman McInnes 
19138965ea79SLois Curfman McInnes     /* read in other processors and ship out */
19148965ea79SLois Curfman McInnes     for (i=1; i<size; i++) {
19158965ea79SLois Curfman McInnes       nz   = procsnz[i];
19160752156aSBarry Smith       ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
19177adad957SLisandro Dalcin       ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr);
19188965ea79SLois Curfman McInnes     }
1919606d414cSSatish Balay     ierr = PetscFree(procsnz);CHKERRQ(ierr);
19203a40ed3dSBarry Smith   } else {
19218965ea79SLois Curfman McInnes     /* receive numeric values */
192284d528b1SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
19238965ea79SLois Curfman McInnes 
19248965ea79SLois Curfman McInnes     /* receive message of values*/
19257adad957SLisandro Dalcin     ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr);
1926ca161407SBarry Smith     ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr);
192729bbc08cSBarry Smith     if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file");
19288965ea79SLois Curfman McInnes 
19298965ea79SLois Curfman McInnes     /* insert into matrix */
19308965ea79SLois Curfman McInnes     jj      = rstart;
19318965ea79SLois Curfman McInnes     smycols = mycols;
19328965ea79SLois Curfman McInnes     svals   = vals;
19338965ea79SLois Curfman McInnes     for (i=0; i<m; i++) {
19348965ea79SLois Curfman McInnes       ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr);
19358965ea79SLois Curfman McInnes       smycols += ourlens[i];
19368965ea79SLois Curfman McInnes       svals   += ourlens[i];
19378965ea79SLois Curfman McInnes       jj++;
19388965ea79SLois Curfman McInnes     }
19398965ea79SLois Curfman McInnes   }
1940606d414cSSatish Balay   ierr = PetscFree(ourlens);CHKERRQ(ierr);
1941606d414cSSatish Balay   ierr = PetscFree(vals);CHKERRQ(ierr);
1942606d414cSSatish Balay   ierr = PetscFree(mycols);CHKERRQ(ierr);
1943606d414cSSatish Balay   ierr = PetscFree(rowners);CHKERRQ(ierr);
19448965ea79SLois Curfman McInnes 
19456d4a8577SBarry Smith   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19466d4a8577SBarry Smith   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19473a40ed3dSBarry Smith   PetscFunctionReturn(0);
19488965ea79SLois Curfman McInnes }
194990ace30eSBarry Smith 
19506e4ee0c6SHong Zhang #undef __FUNCT__
19516e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense"
19526e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag)
19536e4ee0c6SHong Zhang {
19546e4ee0c6SHong Zhang   Mat_MPIDense   *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data;
19556e4ee0c6SHong Zhang   Mat            a,b;
19566e4ee0c6SHong Zhang   PetscTruth     flg;
19576e4ee0c6SHong Zhang   PetscErrorCode ierr;
195890ace30eSBarry Smith 
19596e4ee0c6SHong Zhang   PetscFunctionBegin;
19606e4ee0c6SHong Zhang   a = matA->A;
19616e4ee0c6SHong Zhang   b = matB->A;
19626e4ee0c6SHong Zhang   ierr = MatEqual(a,b,&flg);CHKERRQ(ierr);
19637adad957SLisandro Dalcin   ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr);
19646e4ee0c6SHong Zhang   PetscFunctionReturn(0);
19656e4ee0c6SHong Zhang }
196690ace30eSBarry Smith 
196709d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK)
196809d27a7eSBarry Smith 
196909d27a7eSBarry Smith #undef __FUNCT__
197009d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage"
197109d27a7eSBarry Smith /*@C
197209d27a7eSBarry Smith   PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK.
197309d27a7eSBarry Smith   Level: developer
197409d27a7eSBarry Smith 
197509d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK
197609d27a7eSBarry Smith .seealso: PetscFinalize()
197709d27a7eSBarry Smith @*/
197809d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void)
197909d27a7eSBarry Smith {
198009d27a7eSBarry Smith   PetscErrorCode ierr;
198109d27a7eSBarry Smith 
198209d27a7eSBarry Smith   PetscFunctionBegin;
198309d27a7eSBarry Smith   ierr = PLA_Finalize();CHKERRQ(ierr);
198409d27a7eSBarry Smith   PetscFunctionReturn(0);
198509d27a7eSBarry Smith }
198609d27a7eSBarry Smith 
198709d27a7eSBarry Smith #undef __FUNCT__
198809d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage"
198909d27a7eSBarry Smith /*@C
199009d27a7eSBarry Smith   PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is
199109d27a7eSBarry Smith   called from PetscDLLibraryRegister() when using dynamic libraries, and on the call to PetscInitialize()
199209d27a7eSBarry Smith   when using static libraries.
199309d27a7eSBarry Smith 
199409d27a7eSBarry Smith   Input Parameter:
199509d27a7eSBarry Smith   path - The dynamic library path, or PETSC_NULL
199609d27a7eSBarry Smith 
199709d27a7eSBarry Smith   Level: developer
199809d27a7eSBarry Smith 
199909d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK
200009d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize()
200109d27a7eSBarry Smith @*/
200209d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(const char path[])
200309d27a7eSBarry Smith {
200404fea9ffSBarry Smith   MPI_Comm       comm = PETSC_COMM_WORLD;
2005eae6fb2eSBarry Smith   PetscMPIInt    size;
200609d27a7eSBarry Smith   PetscErrorCode ierr;
200709d27a7eSBarry Smith 
200809d27a7eSBarry Smith   PetscFunctionBegin;
200909d27a7eSBarry Smith   if (!PLA_Initialized(PETSC_NULL)) {
201009d27a7eSBarry Smith 
201109d27a7eSBarry Smith     ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
201204fea9ffSBarry Smith     Plapack_nprows = 1;
201304fea9ffSBarry Smith     Plapack_npcols = size;
201409d27a7eSBarry Smith 
2015aeccfd6fSBarry Smith     ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr);
2016aeccfd6fSBarry Smith       ierr = PetscOptionsInt("-plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr);
2017aeccfd6fSBarry Smith       ierr = PetscOptionsInt("-plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr);
201879b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG)
201979b0a62dSBarry Smith       Plapack_ierror = 3;
202079b0a62dSBarry Smith #else
202179b0a62dSBarry Smith       Plapack_ierror = 0;
202279b0a62dSBarry Smith #endif
2023eae6fb2eSBarry Smith       ierr = PetscOptionsInt("-plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr);
2024eae6fb2eSBarry Smith       if (Plapack_ierror){
2025eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr);
2026aeccfd6fSBarry Smith       } else {
2027eae6fb2eSBarry Smith 	ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr);
2028aeccfd6fSBarry Smith       }
2029aeccfd6fSBarry Smith 
2030eae6fb2eSBarry Smith       Plapack_nb_alg = 0;
203179b0a62dSBarry Smith       ierr = PetscOptionsInt("-plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr);
2032eae6fb2eSBarry Smith       if (Plapack_nb_alg) {
2033eae6fb2eSBarry Smith         ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr);
2034aeccfd6fSBarry Smith       }
2035aeccfd6fSBarry Smith     PetscOptionsEnd();
2036aeccfd6fSBarry Smith 
203704fea9ffSBarry Smith     ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr);
203804fea9ffSBarry Smith     ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr);
203909d27a7eSBarry Smith     ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr);
204009d27a7eSBarry Smith   }
204109d27a7eSBarry Smith   PetscFunctionReturn(0);
204209d27a7eSBarry Smith }
204390ace30eSBarry Smith 
204490ace30eSBarry Smith 
204509d27a7eSBarry Smith #endif
2046