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