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; 52d0f46423SBarry Smith PetscInt lrow,rstart = A->rmap->rstart,rend = A->rmap->rend; 53ba8c8a56SBarry Smith 54ba8c8a56SBarry Smith PetscFunctionBegin; 55ba8c8a56SBarry Smith if (row < rstart || row >= rend) SETERRQ(PETSC_ERR_SUP,"only local rows") 56ba8c8a56SBarry Smith lrow = row - rstart; 57ba8c8a56SBarry Smith ierr = MatGetRow(mat->A,lrow,nz,(const PetscInt **)idx,(const PetscScalar **)v);CHKERRQ(ierr); 58ba8c8a56SBarry Smith PetscFunctionReturn(0); 59ba8c8a56SBarry Smith } 60ba8c8a56SBarry Smith 61ba8c8a56SBarry Smith #undef __FUNCT__ 62ba8c8a56SBarry Smith #define __FUNCT__ "MatRestoreRow_MPIDense" 63ba8c8a56SBarry Smith PetscErrorCode MatRestoreRow_MPIDense(Mat mat,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 64ba8c8a56SBarry Smith { 65ba8c8a56SBarry Smith PetscErrorCode ierr; 66ba8c8a56SBarry Smith 67ba8c8a56SBarry Smith PetscFunctionBegin; 68ba8c8a56SBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 69ba8c8a56SBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 70ba8c8a56SBarry Smith PetscFunctionReturn(0); 71ba8c8a56SBarry Smith } 72ba8c8a56SBarry Smith 730de54da6SSatish Balay EXTERN_C_BEGIN 744a2ae208SSatish Balay #undef __FUNCT__ 754a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonalBlock_MPIDense" 76be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatGetDiagonalBlock_MPIDense(Mat A,PetscTruth *iscopy,MatReuse reuse,Mat *B) 770de54da6SSatish Balay { 780de54da6SSatish Balay Mat_MPIDense *mdn = (Mat_MPIDense*)A->data; 796849ba73SBarry Smith PetscErrorCode ierr; 80d0f46423SBarry Smith PetscInt m = A->rmap->n,rstart = A->rmap->rstart; 8187828ca2SBarry Smith PetscScalar *array; 820de54da6SSatish Balay MPI_Comm comm; 830de54da6SSatish Balay 840de54da6SSatish Balay PetscFunctionBegin; 85d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_SUP,"Only square matrices supported."); 860de54da6SSatish Balay 870de54da6SSatish Balay /* The reuse aspect is not implemented efficiently */ 880de54da6SSatish Balay if (reuse) { ierr = MatDestroy(*B);CHKERRQ(ierr);} 890de54da6SSatish Balay 900de54da6SSatish Balay ierr = PetscObjectGetComm((PetscObject)(mdn->A),&comm);CHKERRQ(ierr); 910de54da6SSatish Balay ierr = MatGetArray(mdn->A,&array);CHKERRQ(ierr); 92f69a0ea3SMatthew Knepley ierr = MatCreate(comm,B);CHKERRQ(ierr); 93f69a0ea3SMatthew Knepley ierr = MatSetSizes(*B,m,m,m,m);CHKERRQ(ierr); 947adad957SLisandro Dalcin ierr = MatSetType(*B,((PetscObject)mdn->A)->type_name);CHKERRQ(ierr); 955c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(*B,array+m*rstart);CHKERRQ(ierr); 960de54da6SSatish Balay ierr = MatRestoreArray(mdn->A,&array);CHKERRQ(ierr); 970de54da6SSatish Balay ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 980de54da6SSatish Balay ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 990de54da6SSatish Balay 1000de54da6SSatish Balay *iscopy = PETSC_TRUE; 1010de54da6SSatish Balay PetscFunctionReturn(0); 1020de54da6SSatish Balay } 1030de54da6SSatish Balay EXTERN_C_END 1040de54da6SSatish Balay 1054a2ae208SSatish Balay #undef __FUNCT__ 1064a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_MPIDense" 10713f74950SBarry Smith PetscErrorCode MatSetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv) 1088965ea79SLois Curfman McInnes { 10939b7565bSBarry Smith Mat_MPIDense *A = (Mat_MPIDense*)mat->data; 110dfbe8321SBarry Smith PetscErrorCode ierr; 111d0f46423SBarry Smith PetscInt i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row; 112273d9f13SBarry Smith PetscTruth roworiented = A->roworiented; 1138965ea79SLois Curfman McInnes 1143a40ed3dSBarry Smith PetscFunctionBegin; 1158965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 1165ef9f2a5SBarry Smith if (idxm[i] < 0) continue; 117d0f46423SBarry Smith if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large"); 1188965ea79SLois Curfman McInnes if (idxm[i] >= rstart && idxm[i] < rend) { 1198965ea79SLois Curfman McInnes row = idxm[i] - rstart; 12039b7565bSBarry Smith if (roworiented) { 12139b7565bSBarry Smith ierr = MatSetValues(A->A,1,&row,n,idxn,v+i*n,addv);CHKERRQ(ierr); 1223a40ed3dSBarry Smith } else { 1238965ea79SLois Curfman McInnes for (j=0; j<n; j++) { 1245ef9f2a5SBarry Smith if (idxn[j] < 0) continue; 125d0f46423SBarry Smith if (idxn[j] >= mat->cmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large"); 12639b7565bSBarry Smith ierr = MatSetValues(A->A,1,&row,1,&idxn[j],v+i+j*m,addv);CHKERRQ(ierr); 12739b7565bSBarry Smith } 1288965ea79SLois Curfman McInnes } 1293a40ed3dSBarry Smith } else { 1303782ba37SSatish Balay if (!A->donotstash) { 13139b7565bSBarry Smith if (roworiented) { 1328798bf22SSatish Balay ierr = MatStashValuesRow_Private(&mat->stash,idxm[i],n,idxn,v+i*n);CHKERRQ(ierr); 133d36fbae8SSatish Balay } else { 1348798bf22SSatish Balay ierr = MatStashValuesCol_Private(&mat->stash,idxm[i],n,idxn,v+i,m);CHKERRQ(ierr); 13539b7565bSBarry Smith } 136b49de8d1SLois Curfman McInnes } 137b49de8d1SLois Curfman McInnes } 1383782ba37SSatish Balay } 1393a40ed3dSBarry Smith PetscFunctionReturn(0); 140b49de8d1SLois Curfman McInnes } 141b49de8d1SLois Curfman McInnes 1424a2ae208SSatish Balay #undef __FUNCT__ 1434a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_MPIDense" 14413f74950SBarry Smith PetscErrorCode MatGetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[]) 145b49de8d1SLois Curfman McInnes { 146b49de8d1SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 147dfbe8321SBarry Smith PetscErrorCode ierr; 148d0f46423SBarry Smith PetscInt i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row; 149b49de8d1SLois Curfman McInnes 1503a40ed3dSBarry Smith PetscFunctionBegin; 151b49de8d1SLois Curfman McInnes for (i=0; i<m; i++) { 15297e567efSBarry Smith if (idxm[i] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative row"); */ 153d0f46423SBarry Smith if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large"); 154b49de8d1SLois Curfman McInnes if (idxm[i] >= rstart && idxm[i] < rend) { 155b49de8d1SLois Curfman McInnes row = idxm[i] - rstart; 156b49de8d1SLois Curfman McInnes for (j=0; j<n; j++) { 15797e567efSBarry Smith if (idxn[j] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative column"); */ 158d0f46423SBarry Smith if (idxn[j] >= mat->cmap->N) { 15929bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large"); 160a8c6a408SBarry Smith } 161b49de8d1SLois Curfman McInnes ierr = MatGetValues(mdn->A,1,&row,1,&idxn[j],v+i*n+j);CHKERRQ(ierr); 162b49de8d1SLois Curfman McInnes } 163a8c6a408SBarry Smith } else { 16429bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"Only local values currently supported"); 1658965ea79SLois Curfman McInnes } 1668965ea79SLois Curfman McInnes } 1673a40ed3dSBarry Smith PetscFunctionReturn(0); 1688965ea79SLois Curfman McInnes } 1698965ea79SLois Curfman McInnes 1704a2ae208SSatish Balay #undef __FUNCT__ 1714a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_MPIDense" 172dfbe8321SBarry Smith PetscErrorCode MatGetArray_MPIDense(Mat A,PetscScalar *array[]) 173ff14e315SSatish Balay { 174ff14e315SSatish Balay Mat_MPIDense *a = (Mat_MPIDense*)A->data; 175dfbe8321SBarry Smith PetscErrorCode ierr; 176ff14e315SSatish Balay 1773a40ed3dSBarry Smith PetscFunctionBegin; 178ff14e315SSatish Balay ierr = MatGetArray(a->A,array);CHKERRQ(ierr); 1793a40ed3dSBarry Smith PetscFunctionReturn(0); 180ff14e315SSatish Balay } 181ff14e315SSatish Balay 1824a2ae208SSatish Balay #undef __FUNCT__ 1834a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_MPIDense" 18413f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_MPIDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B) 185ca3fa75bSLois Curfman McInnes { 186ca3fa75bSLois Curfman McInnes Mat_MPIDense *mat = (Mat_MPIDense*)A->data,*newmatd; 187ca3fa75bSLois Curfman McInnes Mat_SeqDense *lmat = (Mat_SeqDense*)mat->A->data; 1886849ba73SBarry Smith PetscErrorCode ierr; 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 267d0f46423SBarry 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; 335d0f46423SBarry 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; 509d0f46423SBarry 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); 516d0f46423SBarry Smith if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec"); 517d0f46423SBarry Smith len = PetscMin(a->A->rmap->n,a->A->cmap->n); 518d0f46423SBarry 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) 539d0f46423SBarry 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; 577d0f46423SBarry 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; 597d0f46423SBarry 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 */ 603d0f46423SBarry Smith mmax = mat->rmap->n; 604aa05aa95SBarry Smith for (i=1; i<size; i++) { 605d0f46423SBarry 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 */ 610d0f46423SBarry 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++) { 621d0f46423SBarry 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; 626d0f46423SBarry 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 { 639d0f46423SBarry Smith 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); 672d0f46423SBarry 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); 6786c995c7dSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Processor mesh: nprows %d, npcols %d\n",lu->Plapack_nprows, lu->Plapack_npcols);CHKERRQ(ierr); 67901b82886SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr); 6806c995c7dSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Error checking: %d\n",lu->Plapack_ierror);CHKERRQ(ierr); 6816c995c7dSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Algorithmic block size: %d\n",lu->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; 701d0f46423SBarry 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; 719d0f46423SBarry 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; 774d0f46423SBarry Smith info->rows_global = (double)A->rmap->N; 775d0f46423SBarry Smith info->columns_global = (double)A->cmap->N; 776d0f46423SBarry Smith info->rows_local = (double)A->rmap->n; 777d0f46423SBarry 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; 856d0f46423SBarry 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) { 905d0f46423SBarry 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); 914d0f46423SBarry 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; 917d0f46423SBarry Smith ierr = PetscMalloc(2*A->cmap->N*sizeof(PetscReal),&tmp);CHKERRQ(ierr); 918d0f46423SBarry Smith tmp2 = tmp + A->cmap->N; 919d0f46423SBarry Smith ierr = PetscMemzero(tmp,2*A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr); 920064f8208SBarry Smith *nrm = 0.0; 9213501a2bdSLois Curfman McInnes v = mat->v; 922d0f46423SBarry Smith for (j=0; j<mdn->A->cmap->n; j++) { 923d0f46423SBarry Smith for (i=0; i<mdn->A->rmap->n; i++) { 92467e560aaSBarry Smith tmp[j] += PetscAbsScalar(*v); v++; 9253501a2bdSLois Curfman McInnes } 9263501a2bdSLois Curfman McInnes } 927d0f46423SBarry Smith ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 928d0f46423SBarry 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); 932d0f46423SBarry 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; 951d0f46423SBarry 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); 960d0f46423SBarry Smith ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,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 967d0f46423SBarry 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; 994d0f46423SBarry 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; 1023d0f46423SBarry 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; 1048d0f46423SBarry 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; 1108d0f46423SBarry Smith PetscInt m=A->rmap->n,n=B->cmap->n; 11092fbe02b9SBarry Smith Mat Cmat; 11102fbe02b9SBarry Smith 11112fbe02b9SBarry Smith PetscFunctionBegin; 1112d0f46423SBarry 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); 1115d0f46423SBarry 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); 1119ab235cb6SHong Zhang //ierr = MatMPIDenseCreatePlapack(A);CHKERRQ(ierr); 1120ab235cb6SHong Zhang //ierr = MatMPIDenseCreatePlapack(B);CHKERRQ(ierr); 1121ab235cb6SHong 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 1140ab235cb6SHong Zhang #undef __FUNCT__ 11416c995c7dSHong Zhang #define __FUNCT__ "MatSolve_MPIDense" 11426c995c7dSHong Zhang PetscErrorCode MatSolve_MPIDense(Mat A,Vec b,Vec x) 11436c995c7dSHong Zhang { 11446c995c7dSHong Zhang MPI_Comm comm = ((PetscObject)A)->comm; 11456c995c7dSHong Zhang Mat_Plapack *lu = (Mat_Plapack*)A->spptr; 11466c995c7dSHong Zhang PetscErrorCode ierr; 1147d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,i,j,*idx_pla,*idx_petsc,loc_m,loc_stride; 11486c995c7dSHong Zhang PetscScalar *array; 11496c995c7dSHong Zhang PetscReal one = 1.0; 11506c995c7dSHong Zhang PetscMPIInt size,rank,r_rank,r_nproc,c_rank,c_nproc;; 11516c995c7dSHong Zhang PLA_Obj v_pla = NULL; 11526c995c7dSHong Zhang PetscScalar *loc_buf; 11536c995c7dSHong Zhang Vec loc_x; 11546c995c7dSHong Zhang 11556c995c7dSHong Zhang PetscFunctionBegin; 11566c995c7dSHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 11576c995c7dSHong Zhang ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 11586c995c7dSHong Zhang 11596c995c7dSHong Zhang /* Create PLAPACK vector objects, then copy b into PLAPACK b */ 11606c995c7dSHong Zhang PLA_Mvector_create(lu->datatype,M,1,lu->templ,PLA_ALIGN_FIRST,&v_pla); 11616c995c7dSHong Zhang PLA_Obj_set_to_zero(v_pla); 11626c995c7dSHong Zhang 11636c995c7dSHong Zhang /* Copy b into rhs_pla */ 11646c995c7dSHong Zhang PLA_API_begin(); 11656c995c7dSHong Zhang PLA_Obj_API_open(v_pla); 11666c995c7dSHong Zhang ierr = VecGetArray(b,&array);CHKERRQ(ierr); 11676c995c7dSHong Zhang PLA_API_axpy_vector_to_global(m,&one,(void *)array,1,v_pla,lu->rstart); 11686c995c7dSHong Zhang ierr = VecRestoreArray(b,&array);CHKERRQ(ierr); 11696c995c7dSHong Zhang PLA_Obj_API_close(v_pla); 11706c995c7dSHong Zhang PLA_API_end(); 11716c995c7dSHong Zhang 11726c995c7dSHong Zhang if (A->factor == MAT_FACTOR_LU){ 11736c995c7dSHong Zhang /* Apply the permutations to the right hand sides */ 11746c995c7dSHong Zhang PLA_Apply_pivots_to_rows (v_pla,lu->pivots); 11756c995c7dSHong Zhang 11766c995c7dSHong Zhang /* Solve L y = b, overwriting b with y */ 11776c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_UNIT_DIAG,lu->A,v_pla ); 11786c995c7dSHong Zhang 11796c995c7dSHong Zhang /* Solve U x = y (=b), overwriting b with x */ 11806c995c7dSHong Zhang PLA_Trsv( PLA_UPPER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla ); 11816c995c7dSHong Zhang } else { /*MAT_FACTOR_CHOLESKY */ 11826c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla); 11836c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,(lu->datatype == MPI_DOUBLE ? PLA_TRANSPOSE : PLA_CONJUGATE_TRANSPOSE), 11846c995c7dSHong Zhang PLA_NONUNIT_DIAG,lu->A,v_pla); 11856c995c7dSHong Zhang } 11866c995c7dSHong Zhang 11876c995c7dSHong Zhang /* Copy PLAPACK x into Petsc vector x */ 11886c995c7dSHong Zhang PLA_Obj_local_length(v_pla, &loc_m); 11896c995c7dSHong Zhang PLA_Obj_local_buffer(v_pla, (void**)&loc_buf); 11906c995c7dSHong Zhang PLA_Obj_local_stride(v_pla, &loc_stride); 11916c995c7dSHong Zhang /* 11926c995c7dSHong Zhang PetscPrintf(PETSC_COMM_SELF," [%d] b - local_m %d local_stride %d, loc_buf: %g %g, nb: %d\n",rank,loc_m,loc_stride,loc_buf[0],loc_buf[(loc_m-1)*loc_stride],lu->nb); 11936c995c7dSHong Zhang */ 11946c995c7dSHong Zhang ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,loc_m*loc_stride,loc_buf,&loc_x);CHKERRQ(ierr); 11956c995c7dSHong Zhang if (!lu->pla_solved){ 11966c995c7dSHong Zhang 11976c995c7dSHong Zhang PLA_Temp_comm_row_info(lu->templ,&lu->comm_2d,&r_rank,&r_nproc); 11986c995c7dSHong Zhang PLA_Temp_comm_col_info(lu->templ,&lu->comm_2d,&c_rank,&c_nproc); 11996c995c7dSHong Zhang /* printf(" [%d] rank: %d %d, nproc: %d %d\n",rank,r_rank,c_rank,r_nproc,c_nproc); */ 12006c995c7dSHong Zhang 12016c995c7dSHong Zhang /* Create IS and cts for VecScatterring */ 12026c995c7dSHong Zhang PLA_Obj_local_length(v_pla, &loc_m); 12036c995c7dSHong Zhang PLA_Obj_local_stride(v_pla, &loc_stride); 12046c995c7dSHong Zhang ierr = PetscMalloc((2*loc_m+1)*sizeof(PetscInt),&idx_pla);CHKERRQ(ierr); 12056c995c7dSHong Zhang idx_petsc = idx_pla + loc_m; 12066c995c7dSHong Zhang 12076c995c7dSHong Zhang rstart = (r_rank*c_nproc+c_rank)*lu->nb; 12086c995c7dSHong Zhang for (i=0; i<loc_m; i+=lu->nb){ 12096c995c7dSHong Zhang j = 0; 12106c995c7dSHong Zhang while (j < lu->nb && i+j < loc_m){ 12116c995c7dSHong Zhang idx_petsc[i+j] = rstart + j; j++; 12126c995c7dSHong Zhang } 12136c995c7dSHong Zhang rstart += size*lu->nb; 12146c995c7dSHong Zhang } 12156c995c7dSHong Zhang 12166c995c7dSHong Zhang for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride; 12176c995c7dSHong Zhang 12186c995c7dSHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr); 12196c995c7dSHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr); 12206c995c7dSHong Zhang ierr = PetscFree(idx_pla);CHKERRQ(ierr); 12216c995c7dSHong Zhang ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr); 12226c995c7dSHong Zhang } 12236c995c7dSHong Zhang ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12246c995c7dSHong Zhang ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12256c995c7dSHong Zhang 12266c995c7dSHong Zhang /* Free data */ 12276c995c7dSHong Zhang ierr = VecDestroy(loc_x);CHKERRQ(ierr); 12286c995c7dSHong Zhang PLA_Obj_free(&v_pla); 12296c995c7dSHong Zhang 12306c995c7dSHong Zhang lu->pla_solved = PETSC_TRUE; 12316c995c7dSHong Zhang PetscFunctionReturn(0); 12326c995c7dSHong Zhang } 12336c995c7dSHong Zhang 12346c995c7dSHong Zhang #undef __FUNCT__ 12356c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense" 1236*719d5645SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,MatFactorInfo *info) 12376c995c7dSHong Zhang { 1238*719d5645SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 12396c995c7dSHong Zhang PetscErrorCode ierr; 1240d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,rend; 12416c995c7dSHong Zhang PetscInt info_pla=0; 12426c995c7dSHong Zhang PetscScalar *array,one = 1.0; 12436c995c7dSHong Zhang 12446c995c7dSHong Zhang PetscFunctionBegin; 12456c995c7dSHong Zhang if (lu->mstruct == SAME_NONZERO_PATTERN){ 12466c995c7dSHong Zhang PLA_Obj_free(&lu->A); 12476c995c7dSHong Zhang PLA_Obj_free (&lu->pivots); 12486c995c7dSHong Zhang } 12496c995c7dSHong Zhang /* Create PLAPACK matrix object */ 12506c995c7dSHong Zhang lu->A = NULL; lu->pivots = NULL; 12516c995c7dSHong Zhang PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A); 12526c995c7dSHong Zhang PLA_Obj_set_to_zero(lu->A); 12536c995c7dSHong Zhang PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots); 12546c995c7dSHong Zhang 12556c995c7dSHong Zhang /* Copy A into lu->A */ 12566c995c7dSHong Zhang PLA_API_begin(); 12576c995c7dSHong Zhang PLA_Obj_API_open(lu->A); 12586c995c7dSHong Zhang ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 12596c995c7dSHong Zhang ierr = MatGetArray(A,&array);CHKERRQ(ierr); 12606c995c7dSHong Zhang PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0); 12616c995c7dSHong Zhang ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 12626c995c7dSHong Zhang PLA_Obj_API_close(lu->A); 12636c995c7dSHong Zhang PLA_API_end(); 12646c995c7dSHong Zhang 12656c995c7dSHong Zhang /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */ 12666c995c7dSHong Zhang info_pla = PLA_LU(lu->A,lu->pivots); 12676c995c7dSHong Zhang if (info_pla != 0) 12686c995c7dSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla); 12696c995c7dSHong Zhang 12706c995c7dSHong Zhang lu->CleanUpPlapack = PETSC_TRUE; 12716c995c7dSHong Zhang lu->rstart = rstart; 12726c995c7dSHong Zhang lu->mstruct = SAME_NONZERO_PATTERN; 1273*719d5645SBarry Smith F->ops->solve = MatSolve_MPIDense; 1274*719d5645SBarry Smith F->assembled = PETSC_TRUE; /* required by -ksp_view */ 12756c995c7dSHong Zhang PetscFunctionReturn(0); 12766c995c7dSHong Zhang } 12776c995c7dSHong Zhang 12786c995c7dSHong Zhang #undef __FUNCT__ 12796c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense" 1280*719d5645SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,MatFactorInfo *info) 12816c995c7dSHong Zhang { 1282*719d5645SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)F->spptr; 12836c995c7dSHong Zhang PetscErrorCode ierr; 1284d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,rend; 12856c995c7dSHong Zhang PetscInt info_pla=0; 12866c995c7dSHong Zhang PetscScalar *array,one = 1.0; 12876c995c7dSHong Zhang 12886c995c7dSHong Zhang PetscFunctionBegin; 12896c995c7dSHong Zhang if (lu->mstruct == SAME_NONZERO_PATTERN){ 12906c995c7dSHong Zhang PLA_Obj_free(&lu->A); 12916c995c7dSHong Zhang } 12926c995c7dSHong Zhang /* Create PLAPACK matrix object */ 12936c995c7dSHong Zhang lu->A = NULL; 12946c995c7dSHong Zhang lu->pivots = NULL; 12956c995c7dSHong Zhang PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A); 12966c995c7dSHong Zhang 12976c995c7dSHong Zhang /* Copy A into lu->A */ 12986c995c7dSHong Zhang PLA_API_begin(); 12996c995c7dSHong Zhang PLA_Obj_API_open(lu->A); 13006c995c7dSHong Zhang ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 13016c995c7dSHong Zhang ierr = MatGetArray(A,&array);CHKERRQ(ierr); 13026c995c7dSHong Zhang PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0); 13036c995c7dSHong Zhang ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 13046c995c7dSHong Zhang PLA_Obj_API_close(lu->A); 13056c995c7dSHong Zhang PLA_API_end(); 13066c995c7dSHong Zhang 13076c995c7dSHong Zhang /* Factor P A -> Chol */ 13086c995c7dSHong Zhang info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A); 13096c995c7dSHong Zhang if (info_pla != 0) 13106c995c7dSHong Zhang SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla); 13116c995c7dSHong Zhang 13126c995c7dSHong Zhang lu->CleanUpPlapack = PETSC_TRUE; 13136c995c7dSHong Zhang lu->rstart = rstart; 13146c995c7dSHong Zhang lu->mstruct = SAME_NONZERO_PATTERN; 1315*719d5645SBarry Smith F->ops->solve = MatSolve_MPIDense; 1316*719d5645SBarry Smith F->assembled = PETSC_TRUE; /* required by -ksp_view */ 13176c995c7dSHong Zhang PetscFunctionReturn(0); 13186c995c7dSHong Zhang } 13196c995c7dSHong Zhang 13206c995c7dSHong Zhang #undef __FUNCT__ 1321ab235cb6SHong Zhang #define __FUNCT__ "MatFactorSymbolic_Plapack_Private" 1322ab235cb6SHong Zhang PetscErrorCode MatFactorSymbolic_Plapack_Private(Mat A,MatFactorInfo *info,Mat *F) 1323ab235cb6SHong Zhang { 1324ab235cb6SHong Zhang Mat B = *F; 1325ab235cb6SHong Zhang Mat_Plapack *lu; 1326ab235cb6SHong Zhang PetscErrorCode ierr; 1327d0f46423SBarry Smith PetscInt M=A->rmap->N; 1328ab235cb6SHong Zhang MPI_Comm comm=((PetscObject)A)->comm,comm_2d; 1329ab235cb6SHong Zhang PetscMPIInt size; 1330ab235cb6SHong Zhang PetscInt ierror; 1331ab235cb6SHong Zhang 1332ab235cb6SHong Zhang PetscFunctionBegin; 1333ab235cb6SHong Zhang lu = (Mat_Plapack*)(B->spptr); 1334ab235cb6SHong Zhang 1335ab235cb6SHong Zhang /* Set default Plapack parameters */ 1336ab235cb6SHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1337ab235cb6SHong Zhang lu->nprows = 1; lu->npcols = size; 1338ab235cb6SHong Zhang ierror = 0; 1339ab235cb6SHong Zhang lu->nb = M/size; 1340ab235cb6SHong Zhang if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */ 1341ab235cb6SHong Zhang 1342ab235cb6SHong Zhang /* Set runtime options */ 1343ab235cb6SHong Zhang ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr); 1344ab235cb6SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",lu->nprows,&lu->nprows,PETSC_NULL);CHKERRQ(ierr); 1345ab235cb6SHong Zhang ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",lu->npcols,&lu->npcols,PETSC_NULL);CHKERRQ(ierr); 1346ab235cb6SHong Zhang 1347ab235cb6SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr); 1348ab235cb6SHong Zhang ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",ierror,&ierror,PETSC_NULL);CHKERRQ(ierr); 1349ab235cb6SHong Zhang if (ierror){ 1350ab235cb6SHong Zhang PLA_Set_error_checking(ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE ); 1351ab235cb6SHong Zhang } else { 1352ab235cb6SHong Zhang PLA_Set_error_checking(ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE ); 1353ab235cb6SHong Zhang } 1354ab235cb6SHong Zhang lu->ierror = ierror; 1355ab235cb6SHong Zhang 1356ab235cb6SHong Zhang lu->nb_alg = 0; 1357ab235cb6SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",lu->nb_alg,&lu->nb_alg,PETSC_NULL);CHKERRQ(ierr); 1358ab235cb6SHong Zhang if (lu->nb_alg){ 1359ab235cb6SHong Zhang pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,lu->nb_alg); 1360ab235cb6SHong Zhang } 1361ab235cb6SHong Zhang PetscOptionsEnd(); 1362ab235cb6SHong Zhang 1363ab235cb6SHong Zhang 1364ab235cb6SHong Zhang /* Create a 2D communicator */ 1365ab235cb6SHong Zhang PLA_Comm_1D_to_2D(comm,lu->nprows,lu->npcols,&comm_2d); 1366ab235cb6SHong Zhang lu->comm_2d = comm_2d; 1367ab235cb6SHong Zhang 1368ab235cb6SHong Zhang /* Initialize PLAPACK */ 1369ab235cb6SHong Zhang PLA_Init(comm_2d); 1370ab235cb6SHong Zhang 1371ab235cb6SHong Zhang /* Create object distribution template */ 1372ab235cb6SHong Zhang lu->templ = NULL; 1373ab235cb6SHong Zhang PLA_Temp_create(lu->nb, 0, &lu->templ); 1374ab235cb6SHong Zhang 1375ab235cb6SHong Zhang /* Use suggested nb_alg if it is not provided by user */ 1376ab235cb6SHong Zhang if (lu->nb_alg == 0){ 1377ab235cb6SHong Zhang PLA_Environ_nb_alg(PLA_OP_PAN_PAN,lu->templ,&lu->nb_alg); 1378ab235cb6SHong Zhang pla_Environ_set_nb_alg(PLA_OP_ALL_ALG,lu->nb_alg); 1379ab235cb6SHong Zhang } 1380ab235cb6SHong Zhang 1381ab235cb6SHong Zhang /* Set the datatype */ 1382ab235cb6SHong Zhang #if defined(PETSC_USE_COMPLEX) 1383ab235cb6SHong Zhang lu->datatype = MPI_DOUBLE_COMPLEX; 1384ab235cb6SHong Zhang #else 1385ab235cb6SHong Zhang lu->datatype = MPI_DOUBLE; 1386ab235cb6SHong Zhang #endif 1387ab235cb6SHong Zhang 1388ab235cb6SHong Zhang lu->pla_solved = PETSC_FALSE; /* MatSolve_Plapack() is called yet */ 1389ab235cb6SHong Zhang lu->mstruct = DIFFERENT_NONZERO_PATTERN; 1390ab235cb6SHong Zhang lu->CleanUpPlapack = PETSC_TRUE; 1391ab235cb6SHong Zhang *F = B; 1392ab235cb6SHong Zhang PetscFunctionReturn(0); 1393ab235cb6SHong Zhang } 1394ab235cb6SHong Zhang 1395b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */ 13962fbe02b9SBarry Smith #undef __FUNCT__ 13976c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense" 1398*719d5645SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,MatFactorInfo *info) 139901b82886SBarry Smith { 140001b82886SBarry Smith PetscErrorCode ierr; 1401b24902e0SBarry Smith PetscTruth issymmetric,set; 140201b82886SBarry Smith 140301b82886SBarry Smith PetscFunctionBegin; 1404b24902e0SBarry Smith ierr = MatIsSymmetricKnown(A,&set,&issymmetric); CHKERRQ(ierr); 1405b24902e0SBarry Smith if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()"); 1406*719d5645SBarry Smith ierr = MatFactorSymbolic_Plapack_Private(F,A,info);CHKERRQ(ierr); 1407*719d5645SBarry Smith F->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_MPIDense; 1408b24902e0SBarry Smith PetscFunctionReturn(0); 140901b82886SBarry Smith } 141001b82886SBarry Smith 1411b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */ 1412b24902e0SBarry Smith #undef __FUNCT__ 14136c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense" 1414*719d5645SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,MatFactorInfo *info) 1415b24902e0SBarry Smith { 1416b24902e0SBarry Smith PetscErrorCode ierr; 1417d0f46423SBarry Smith PetscInt M = A->rmap->N; 1418b24902e0SBarry Smith Mat_Plapack *lu; 141901b82886SBarry Smith 1420b24902e0SBarry Smith PetscFunctionBegin; 1421b24902e0SBarry Smith ierr = MatFactorSymbolic_Plapack_Private(A,info,F);CHKERRQ(ierr); 1422*719d5645SBarry Smith lu = (Mat_Plapack*)F->spptr; 1423b24902e0SBarry Smith ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr); 1424*719d5645SBarry Smith F->ops->lufactornumeric = MatLUFactorNumeric_MPIDense; 142501b82886SBarry Smith PetscFunctionReturn(0); 142601b82886SBarry Smith } 142701b82886SBarry Smith 142801b82886SBarry Smith #undef __FUNCT__ 1429b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_mpidense_plapack" 1430b24902e0SBarry Smith PetscErrorCode MatGetFactor_mpidense_plapack(Mat A,MatFactorType ftype,Mat *F) 143101b82886SBarry Smith { 143201b82886SBarry Smith PetscErrorCode ierr; 14336c995c7dSHong Zhang Mat_Plapack *lu; 14346c995c7dSHong Zhang PetscMPIInt size; 14356c995c7dSHong Zhang PetscInt M; 143601b82886SBarry Smith 143701b82886SBarry Smith PetscFunctionBegin; 143801b82886SBarry Smith /* Create the factorization matrix */ 1439eae6fb2eSBarry Smith ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr); 1440d0f46423SBarry Smith ierr = MatSetSizes(*F,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1441eae6fb2eSBarry Smith ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr); 1442b24902e0SBarry Smith ierr = PetscNewLog(A,Mat_Plapack,&lu);CHKERRQ(ierr); 1443b24902e0SBarry Smith A->spptr = (void*)lu; 144401b82886SBarry Smith 1445b24902e0SBarry Smith lu = (Mat_Plapack*)(A->spptr); 144601b82886SBarry Smith 1447b24902e0SBarry Smith /* Set default Plapack parameters */ 14486c995c7dSHong Zhang ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr); 1449b24902e0SBarry Smith lu->nb = M/size; 1450b24902e0SBarry Smith if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */ 145101b82886SBarry Smith 1452b24902e0SBarry Smith /* Set runtime options */ 1453b24902e0SBarry Smith ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr); 1454b24902e0SBarry Smith ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr); 1455b24902e0SBarry Smith PetscOptionsEnd(); 145601b82886SBarry Smith 1457b24902e0SBarry Smith /* Create object distribution template */ 1458b24902e0SBarry Smith lu->templ = NULL; 1459b24902e0SBarry Smith ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr); 1460b24902e0SBarry Smith 1461b24902e0SBarry Smith /* Set the datatype */ 1462b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX) 1463b24902e0SBarry Smith lu->datatype = MPI_DOUBLE_COMPLEX; 1464b24902e0SBarry Smith #else 1465b24902e0SBarry Smith lu->datatype = MPI_DOUBLE; 1466b24902e0SBarry Smith #endif 1467b24902e0SBarry Smith 1468d0f46423SBarry Smith ierr = PLA_Matrix_create(lu->datatype,M,A->cmap->N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr); 1469b24902e0SBarry Smith 1470b24902e0SBarry Smith 1471b24902e0SBarry Smith lu->pla_solved = PETSC_FALSE; /* MatSolve_Plapack() is called yet */ 1472b24902e0SBarry Smith 1473b24902e0SBarry Smith if (ftype == MAT_FACTOR_LU) { 1474b24902e0SBarry Smith (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense; 1475b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY) { 14766c995c7dSHong Zhang (*F)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense; 1477b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices"); 1478*719d5645SBarry Smith (*F)->factor = ftype; 147901b82886SBarry Smith PetscFunctionReturn(0); 148001b82886SBarry Smith } 148101b82886SBarry Smith #endif 148201b82886SBarry Smith 14838965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/ 148409dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense, 148509dc0095SBarry Smith MatGetRow_MPIDense, 148609dc0095SBarry Smith MatRestoreRow_MPIDense, 148709dc0095SBarry Smith MatMult_MPIDense, 148897304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense, 14897c922b88SBarry Smith MatMultTranspose_MPIDense, 14907c922b88SBarry Smith MatMultTransposeAdd_MPIDense, 14918965ea79SLois Curfman McInnes 0, 149209dc0095SBarry Smith 0, 149309dc0095SBarry Smith 0, 149497304618SKris Buschelman /*10*/ 0, 149509dc0095SBarry Smith 0, 149609dc0095SBarry Smith 0, 149709dc0095SBarry Smith 0, 149809dc0095SBarry Smith MatTranspose_MPIDense, 149997304618SKris Buschelman /*15*/ MatGetInfo_MPIDense, 15006e4ee0c6SHong Zhang MatEqual_MPIDense, 150109dc0095SBarry Smith MatGetDiagonal_MPIDense, 15025b2fa520SLois Curfman McInnes MatDiagonalScale_MPIDense, 150309dc0095SBarry Smith MatNorm_MPIDense, 150497304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense, 150509dc0095SBarry Smith MatAssemblyEnd_MPIDense, 150609dc0095SBarry Smith 0, 150709dc0095SBarry Smith MatSetOption_MPIDense, 150809dc0095SBarry Smith MatZeroEntries_MPIDense, 150997304618SKris Buschelman /*25*/ MatZeroRows_MPIDense, 1510919b68f7SBarry Smith 0, 151101b82886SBarry Smith 0, 151201b82886SBarry Smith 0, 151301b82886SBarry Smith 0, 151497304618SKris Buschelman /*30*/ MatSetUpPreallocation_MPIDense, 1515273d9f13SBarry Smith 0, 151609dc0095SBarry Smith 0, 151709dc0095SBarry Smith MatGetArray_MPIDense, 151809dc0095SBarry Smith MatRestoreArray_MPIDense, 151997304618SKris Buschelman /*35*/ MatDuplicate_MPIDense, 152009dc0095SBarry Smith 0, 152109dc0095SBarry Smith 0, 152209dc0095SBarry Smith 0, 152309dc0095SBarry Smith 0, 152497304618SKris Buschelman /*40*/ 0, 15252ce60cd0SSatish Balay MatGetSubMatrices_MPIDense, 152609dc0095SBarry Smith 0, 152709dc0095SBarry Smith MatGetValues_MPIDense, 152809dc0095SBarry Smith 0, 152997304618SKris Buschelman /*45*/ 0, 153009dc0095SBarry Smith MatScale_MPIDense, 153109dc0095SBarry Smith 0, 153209dc0095SBarry Smith 0, 153309dc0095SBarry Smith 0, 1534521d7252SBarry Smith /*50*/ 0, 153509dc0095SBarry Smith 0, 153609dc0095SBarry Smith 0, 153709dc0095SBarry Smith 0, 153809dc0095SBarry Smith 0, 153997304618SKris Buschelman /*55*/ 0, 154009dc0095SBarry Smith 0, 154109dc0095SBarry Smith 0, 154209dc0095SBarry Smith 0, 154309dc0095SBarry Smith 0, 154497304618SKris Buschelman /*60*/ MatGetSubMatrix_MPIDense, 1545b9b97703SBarry Smith MatDestroy_MPIDense, 1546b9b97703SBarry Smith MatView_MPIDense, 1547357abbc8SBarry Smith 0, 154897304618SKris Buschelman 0, 154997304618SKris Buschelman /*65*/ 0, 155097304618SKris Buschelman 0, 155197304618SKris Buschelman 0, 155297304618SKris Buschelman 0, 155397304618SKris Buschelman 0, 155497304618SKris Buschelman /*70*/ 0, 155597304618SKris Buschelman 0, 155697304618SKris Buschelman 0, 155797304618SKris Buschelman 0, 155897304618SKris Buschelman 0, 155997304618SKris Buschelman /*75*/ 0, 156097304618SKris Buschelman 0, 156197304618SKris Buschelman 0, 156297304618SKris Buschelman 0, 156397304618SKris Buschelman 0, 156497304618SKris Buschelman /*80*/ 0, 156597304618SKris Buschelman 0, 156697304618SKris Buschelman 0, 156797304618SKris Buschelman 0, 1568865e5f61SKris Buschelman /*84*/ MatLoad_MPIDense, 1569865e5f61SKris Buschelman 0, 1570865e5f61SKris Buschelman 0, 1571865e5f61SKris Buschelman 0, 1572865e5f61SKris Buschelman 0, 1573865e5f61SKris Buschelman 0, 15742fbe02b9SBarry Smith /*90*/ 15752fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 15762fbe02b9SBarry Smith MatMatMult_MPIDense_MPIDense, 15774ae313f4SHong Zhang MatMatMultSymbolic_MPIDense_MPIDense, 15782fbe02b9SBarry Smith MatMatMultNumeric_MPIDense_MPIDense, 15792fbe02b9SBarry Smith #else 1580865e5f61SKris Buschelman 0, 1581865e5f61SKris Buschelman 0, 1582865e5f61SKris Buschelman 0, 15832fbe02b9SBarry Smith #endif 15842fbe02b9SBarry Smith 0, 1585865e5f61SKris Buschelman /*95*/ 0, 1586865e5f61SKris Buschelman 0, 1587865e5f61SKris Buschelman 0, 1588865e5f61SKris Buschelman 0}; 15898965ea79SLois Curfman McInnes 1590273d9f13SBarry Smith EXTERN_C_BEGIN 15914a2ae208SSatish Balay #undef __FUNCT__ 1592a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense" 1593be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data) 1594a23d5eceSKris Buschelman { 1595a23d5eceSKris Buschelman Mat_MPIDense *a; 1596dfbe8321SBarry Smith PetscErrorCode ierr; 1597a23d5eceSKris Buschelman 1598a23d5eceSKris Buschelman PetscFunctionBegin; 1599a23d5eceSKris Buschelman mat->preallocated = PETSC_TRUE; 1600a23d5eceSKris Buschelman /* Note: For now, when data is specified above, this assumes the user correctly 1601a23d5eceSKris Buschelman allocates the local dense storage space. We should add error checking. */ 1602a23d5eceSKris Buschelman 1603a23d5eceSKris Buschelman a = (Mat_MPIDense*)mat->data; 1604f69a0ea3SMatthew Knepley ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr); 1605d0f46423SBarry Smith ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr); 16065c5985e7SKris Buschelman ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr); 16075c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr); 160852e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr); 1609a23d5eceSKris Buschelman PetscFunctionReturn(0); 1610a23d5eceSKris Buschelman } 1611a23d5eceSKris Buschelman EXTERN_C_END 1612a23d5eceSKris Buschelman 16130bad9183SKris Buschelman /*MC 1614fafad747SKris Buschelman MATMPIDENSE - MATMPIDENSE = "mpidense" - A matrix type to be used for distributed dense matrices. 16150bad9183SKris Buschelman 16160bad9183SKris Buschelman Options Database Keys: 16170bad9183SKris Buschelman . -mat_type mpidense - sets the matrix type to "mpidense" during a call to MatSetFromOptions() 16180bad9183SKris Buschelman 16190bad9183SKris Buschelman Level: beginner 16200bad9183SKris Buschelman 16217878bbefSBarry Smith MATMPIDENSE matrices may use direct solvers (LU, Cholesky, and QR) 16227878bbefSBarry Smith for parallel dense matrices via the external package PLAPACK, if PLAPACK is installed 16237878bbefSBarry Smith (run config/configure.py with the option --download-plapack) 16247878bbefSBarry Smith 16257878bbefSBarry Smith 16267878bbefSBarry Smith Options Database Keys: 16277878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition 16287878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition 16297878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector 16307878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size 16317878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag 16327878bbefSBarry Smith 16337878bbefSBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE 16340bad9183SKris Buschelman M*/ 16350bad9183SKris Buschelman 1636a23d5eceSKris Buschelman EXTERN_C_BEGIN 1637a23d5eceSKris Buschelman #undef __FUNCT__ 16384a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense" 1639be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat) 1640273d9f13SBarry Smith { 1641273d9f13SBarry Smith Mat_MPIDense *a; 1642dfbe8321SBarry Smith PetscErrorCode ierr; 1643273d9f13SBarry Smith 1644273d9f13SBarry Smith PetscFunctionBegin; 164538f2d2fdSLisandro Dalcin ierr = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr); 1646b0a32e0cSBarry Smith mat->data = (void*)a; 1647273d9f13SBarry Smith ierr = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1648273d9f13SBarry Smith mat->mapping = 0; 1649273d9f13SBarry Smith 1650273d9f13SBarry Smith mat->insertmode = NOT_SET_VALUES; 16517adad957SLisandro Dalcin ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr); 16527adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr); 1653273d9f13SBarry Smith 1654d0f46423SBarry Smith mat->rmap->bs = mat->cmap->bs = 1; 1655d0f46423SBarry Smith ierr = PetscMapSetUp(mat->rmap);CHKERRQ(ierr); 1656d0f46423SBarry Smith ierr = PetscMapSetUp(mat->cmap);CHKERRQ(ierr); 1657d0f46423SBarry Smith a->nvec = mat->cmap->n; 1658273d9f13SBarry Smith 1659273d9f13SBarry Smith /* build cache for off array entries formed */ 1660273d9f13SBarry Smith a->donotstash = PETSC_FALSE; 16617adad957SLisandro Dalcin ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr); 1662273d9f13SBarry Smith 1663273d9f13SBarry Smith /* stuff used for matrix vector multiply */ 1664273d9f13SBarry Smith a->lvec = 0; 1665273d9f13SBarry Smith a->Mvctx = 0; 1666273d9f13SBarry Smith a->roworiented = PETSC_TRUE; 1667273d9f13SBarry Smith 1668273d9f13SBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C", 1669273d9f13SBarry Smith "MatGetDiagonalBlock_MPIDense", 1670273d9f13SBarry Smith MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr); 1671a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C", 1672a23d5eceSKris Buschelman "MatMPIDenseSetPreallocation_MPIDense", 1673a23d5eceSKris Buschelman MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr); 16744ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C", 16754ae313f4SHong Zhang "MatMatMult_MPIAIJ_MPIDense", 16764ae313f4SHong Zhang MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr); 16774ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C", 16784ae313f4SHong Zhang "MatMatMultSymbolic_MPIAIJ_MPIDense", 16794ae313f4SHong Zhang MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr); 16804ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C", 16814ae313f4SHong Zhang "MatMatMultNumeric_MPIAIJ_MPIDense", 16824ae313f4SHong Zhang MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr); 16837878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 1684b24902e0SBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_plapack_C", 1685b24902e0SBarry Smith "MatGetFactor_mpidense_plapack", 1686b24902e0SBarry Smith MatGetFactor_mpidense_plapack);CHKERRQ(ierr); 1687db4efbfdSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 1688db4efbfdSBarry Smith ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr); 1689db4efbfdSBarry Smith #endif 1690db4efbfdSBarry Smith 16917878bbefSBarry Smith #endif 169238aed534SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr); 169301b82886SBarry Smith 1694273d9f13SBarry Smith PetscFunctionReturn(0); 1695273d9f13SBarry Smith } 1696273d9f13SBarry Smith EXTERN_C_END 1697273d9f13SBarry Smith 1698209238afSKris Buschelman /*MC 1699002d173eSKris Buschelman MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices. 1700209238afSKris Buschelman 1701209238afSKris Buschelman This matrix type is identical to MATSEQDENSE when constructed with a single process communicator, 1702209238afSKris Buschelman and MATMPIDENSE otherwise. 1703209238afSKris Buschelman 1704209238afSKris Buschelman Options Database Keys: 1705209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions() 1706209238afSKris Buschelman 1707209238afSKris Buschelman Level: beginner 1708209238afSKris Buschelman 170901b82886SBarry Smith 1710209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE 1711209238afSKris Buschelman M*/ 1712209238afSKris Buschelman 1713209238afSKris Buschelman EXTERN_C_BEGIN 1714209238afSKris Buschelman #undef __FUNCT__ 1715209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense" 1716be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A) 1717dfbe8321SBarry Smith { 17186849ba73SBarry Smith PetscErrorCode ierr; 171913f74950SBarry Smith PetscMPIInt size; 1720209238afSKris Buschelman 1721209238afSKris Buschelman PetscFunctionBegin; 17227adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr); 1723209238afSKris Buschelman if (size == 1) { 1724209238afSKris Buschelman ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr); 1725209238afSKris Buschelman } else { 1726209238afSKris Buschelman ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr); 1727209238afSKris Buschelman } 1728209238afSKris Buschelman PetscFunctionReturn(0); 1729209238afSKris Buschelman } 1730209238afSKris Buschelman EXTERN_C_END 1731209238afSKris Buschelman 17324a2ae208SSatish Balay #undef __FUNCT__ 17334a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation" 1734273d9f13SBarry Smith /*@C 1735273d9f13SBarry Smith MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries 1736273d9f13SBarry Smith 1737273d9f13SBarry Smith Not collective 1738273d9f13SBarry Smith 1739273d9f13SBarry Smith Input Parameters: 1740273d9f13SBarry Smith . A - the matrix 1741273d9f13SBarry Smith - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1742273d9f13SBarry Smith to control all matrix memory allocation. 1743273d9f13SBarry Smith 1744273d9f13SBarry Smith Notes: 1745273d9f13SBarry Smith The dense format is fully compatible with standard Fortran 77 1746273d9f13SBarry Smith storage by columns. 1747273d9f13SBarry Smith 1748273d9f13SBarry Smith The data input variable is intended primarily for Fortran programmers 1749273d9f13SBarry Smith who wish to allocate their own matrix memory space. Most users should 1750273d9f13SBarry Smith set data=PETSC_NULL. 1751273d9f13SBarry Smith 1752273d9f13SBarry Smith Level: intermediate 1753273d9f13SBarry Smith 1754273d9f13SBarry Smith .keywords: matrix,dense, parallel 1755273d9f13SBarry Smith 1756273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues() 1757273d9f13SBarry Smith @*/ 1758be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data) 1759273d9f13SBarry Smith { 1760dfbe8321SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscScalar *); 1761273d9f13SBarry Smith 1762273d9f13SBarry Smith PetscFunctionBegin; 1763565567f0SKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1764a23d5eceSKris Buschelman if (f) { 1765a23d5eceSKris Buschelman ierr = (*f)(mat,data);CHKERRQ(ierr); 1766a23d5eceSKris Buschelman } 1767273d9f13SBarry Smith PetscFunctionReturn(0); 1768273d9f13SBarry Smith } 1769273d9f13SBarry Smith 17704a2ae208SSatish Balay #undef __FUNCT__ 17714a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense" 17728965ea79SLois Curfman McInnes /*@C 177339ddd567SLois Curfman McInnes MatCreateMPIDense - Creates a sparse parallel matrix in dense format. 17748965ea79SLois Curfman McInnes 1775db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1776db81eaa0SLois Curfman McInnes 17778965ea79SLois Curfman McInnes Input Parameters: 1778db81eaa0SLois Curfman McInnes + comm - MPI communicator 17798965ea79SLois Curfman McInnes . m - number of local rows (or PETSC_DECIDE to have calculated if M is given) 1780db81eaa0SLois Curfman McInnes . n - number of local columns (or PETSC_DECIDE to have calculated if N is given) 17818965ea79SLois Curfman McInnes . M - number of global rows (or PETSC_DECIDE to have calculated if m is given) 1782db81eaa0SLois Curfman McInnes . N - number of global columns (or PETSC_DECIDE to have calculated if n is given) 17837f5ff6fdSBarry Smith - data - optional location of matrix data. Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc 1784dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 17858965ea79SLois Curfman McInnes 17868965ea79SLois Curfman McInnes Output Parameter: 1787477f1c0bSLois Curfman McInnes . A - the matrix 17888965ea79SLois Curfman McInnes 1789b259b22eSLois Curfman McInnes Notes: 179039ddd567SLois Curfman McInnes The dense format is fully compatible with standard Fortran 77 179139ddd567SLois Curfman McInnes storage by columns. 17928965ea79SLois Curfman McInnes 179318f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 179418f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 17957f5ff6fdSBarry Smith set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users). 179618f449edSLois Curfman McInnes 17978965ea79SLois Curfman McInnes The user MUST specify either the local or global matrix dimensions 17988965ea79SLois Curfman McInnes (possibly both). 17998965ea79SLois Curfman McInnes 1800027ccd11SLois Curfman McInnes Level: intermediate 1801027ccd11SLois Curfman McInnes 180239ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel 18038965ea79SLois Curfman McInnes 180439ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues() 18058965ea79SLois Curfman McInnes @*/ 1806be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A) 18078965ea79SLois Curfman McInnes { 18086849ba73SBarry Smith PetscErrorCode ierr; 180913f74950SBarry Smith PetscMPIInt size; 18108965ea79SLois Curfman McInnes 18113a40ed3dSBarry Smith PetscFunctionBegin; 1812f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 1813f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr); 1814273d9f13SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1815273d9f13SBarry Smith if (size > 1) { 1816273d9f13SBarry Smith ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr); 1817273d9f13SBarry Smith ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr); 1818273d9f13SBarry Smith } else { 1819273d9f13SBarry Smith ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr); 1820273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr); 18218c469469SLois Curfman McInnes } 18223a40ed3dSBarry Smith PetscFunctionReturn(0); 18238965ea79SLois Curfman McInnes } 18248965ea79SLois Curfman McInnes 18254a2ae208SSatish Balay #undef __FUNCT__ 18264a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense" 18276849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 18288965ea79SLois Curfman McInnes { 18298965ea79SLois Curfman McInnes Mat mat; 18303501a2bdSLois Curfman McInnes Mat_MPIDense *a,*oldmat = (Mat_MPIDense*)A->data; 1831dfbe8321SBarry Smith PetscErrorCode ierr; 18328965ea79SLois Curfman McInnes 18333a40ed3dSBarry Smith PetscFunctionBegin; 18348965ea79SLois Curfman McInnes *newmat = 0; 18357adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr); 1836d0f46423SBarry Smith ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 18377adad957SLisandro Dalcin ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr); 1838834f8fabSBarry Smith a = (Mat_MPIDense*)mat->data; 1839e04c1aa4SHong Zhang ierr = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 18403501a2bdSLois Curfman McInnes mat->factor = A->factor; 1841c456f294SBarry Smith mat->assembled = PETSC_TRUE; 1842273d9f13SBarry Smith mat->preallocated = PETSC_TRUE; 18438965ea79SLois Curfman McInnes 1844d0f46423SBarry Smith mat->rmap->rstart = A->rmap->rstart; 1845d0f46423SBarry Smith mat->rmap->rend = A->rmap->rend; 18468965ea79SLois Curfman McInnes a->size = oldmat->size; 18478965ea79SLois Curfman McInnes a->rank = oldmat->rank; 1848e0fa3b82SLois Curfman McInnes mat->insertmode = NOT_SET_VALUES; 1849b9b97703SBarry Smith a->nvec = oldmat->nvec; 18503782ba37SSatish Balay a->donotstash = oldmat->donotstash; 1851e04c1aa4SHong Zhang 1852d0f46423SBarry Smith ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr); 1853d0f46423SBarry Smith ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr); 18547adad957SLisandro Dalcin ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr); 18558965ea79SLois Curfman McInnes 1856329f5518SBarry Smith ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr); 18575609ef8eSBarry Smith ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr); 185852e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr); 185901b82886SBarry Smith 186001b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 186101b82886SBarry Smith ierr = PetscMemcpy(mat->spptr,A->spptr,sizeof(Mat_Plapack));CHKERRQ(ierr); 186201b82886SBarry Smith #endif 18638965ea79SLois Curfman McInnes *newmat = mat; 18643a40ed3dSBarry Smith PetscFunctionReturn(0); 18658965ea79SLois Curfman McInnes } 18668965ea79SLois Curfman McInnes 1867e090d566SSatish Balay #include "petscsys.h" 18688965ea79SLois Curfman McInnes 18694a2ae208SSatish Balay #undef __FUNCT__ 18704a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile" 1871a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat) 187290ace30eSBarry Smith { 18736849ba73SBarry Smith PetscErrorCode ierr; 187432dcc486SBarry Smith PetscMPIInt rank,size; 187513f74950SBarry Smith PetscInt *rowners,i,m,nz,j; 187687828ca2SBarry Smith PetscScalar *array,*vals,*vals_ptr; 187790ace30eSBarry Smith MPI_Status status; 187890ace30eSBarry Smith 18793a40ed3dSBarry Smith PetscFunctionBegin; 1880d132466eSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 1881d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 188290ace30eSBarry Smith 188390ace30eSBarry Smith /* determine ownership of all rows */ 188490ace30eSBarry Smith m = M/size + ((M % size) > rank); 188513f74950SBarry Smith ierr = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr); 188613f74950SBarry Smith ierr = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr); 188790ace30eSBarry Smith rowners[0] = 0; 188890ace30eSBarry Smith for (i=2; i<=size; i++) { 188990ace30eSBarry Smith rowners[i] += rowners[i-1]; 189090ace30eSBarry Smith } 189190ace30eSBarry Smith 1892f69a0ea3SMatthew Knepley ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 1893f69a0ea3SMatthew Knepley ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr); 1894be5d1d56SKris Buschelman ierr = MatSetType(*newmat,type);CHKERRQ(ierr); 1895878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr); 189690ace30eSBarry Smith ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr); 189790ace30eSBarry Smith 189890ace30eSBarry Smith if (!rank) { 189987828ca2SBarry Smith ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 190090ace30eSBarry Smith 190190ace30eSBarry Smith /* read in my part of the matrix numerical values */ 19020752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr); 190390ace30eSBarry Smith 190490ace30eSBarry Smith /* insert into matrix-by row (this is why cannot directly read into array */ 190590ace30eSBarry Smith vals_ptr = vals; 190690ace30eSBarry Smith for (i=0; i<m; i++) { 190790ace30eSBarry Smith for (j=0; j<N; j++) { 190890ace30eSBarry Smith array[i + j*m] = *vals_ptr++; 190990ace30eSBarry Smith } 191090ace30eSBarry Smith } 191190ace30eSBarry Smith 191290ace30eSBarry Smith /* read in other processors and ship out */ 191390ace30eSBarry Smith for (i=1; i<size; i++) { 191490ace30eSBarry Smith nz = (rowners[i+1] - rowners[i])*N; 19150752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 19167adad957SLisandro Dalcin ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr); 191790ace30eSBarry Smith } 19183a40ed3dSBarry Smith } else { 191990ace30eSBarry Smith /* receive numeric values */ 192087828ca2SBarry Smith ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 192190ace30eSBarry Smith 192290ace30eSBarry Smith /* receive message of values*/ 19237adad957SLisandro Dalcin ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr); 192490ace30eSBarry Smith 192590ace30eSBarry Smith /* insert into matrix-by row (this is why cannot directly read into array */ 192690ace30eSBarry Smith vals_ptr = vals; 192790ace30eSBarry Smith for (i=0; i<m; i++) { 192890ace30eSBarry Smith for (j=0; j<N; j++) { 192990ace30eSBarry Smith array[i + j*m] = *vals_ptr++; 193090ace30eSBarry Smith } 193190ace30eSBarry Smith } 193290ace30eSBarry Smith } 1933606d414cSSatish Balay ierr = PetscFree(rowners);CHKERRQ(ierr); 1934606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 19356d4a8577SBarry Smith ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 19366d4a8577SBarry Smith ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 19373a40ed3dSBarry Smith PetscFunctionReturn(0); 193890ace30eSBarry Smith } 193990ace30eSBarry Smith 19404a2ae208SSatish Balay #undef __FUNCT__ 19414a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense" 1942a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat) 19438965ea79SLois Curfman McInnes { 19448965ea79SLois Curfman McInnes Mat A; 194587828ca2SBarry Smith PetscScalar *vals,*svals; 194619bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 19478965ea79SLois Curfman McInnes MPI_Status status; 194813f74950SBarry Smith PetscMPIInt rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz; 194913f74950SBarry Smith PetscInt header[4],*rowlengths = 0,M,N,*cols; 195013f74950SBarry Smith PetscInt *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols; 195113f74950SBarry Smith PetscInt i,nz,j,rstart,rend; 195213f74950SBarry Smith int fd; 19536849ba73SBarry Smith PetscErrorCode ierr; 19548965ea79SLois Curfman McInnes 19553a40ed3dSBarry Smith PetscFunctionBegin; 1956d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1957d132466eSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 19588965ea79SLois Curfman McInnes if (!rank) { 1959b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 19600752156aSBarry Smith ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr); 1961552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object"); 19628965ea79SLois Curfman McInnes } 19638965ea79SLois Curfman McInnes 196413f74950SBarry Smith ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr); 196590ace30eSBarry Smith M = header[1]; N = header[2]; nz = header[3]; 196690ace30eSBarry Smith 196790ace30eSBarry Smith /* 196890ace30eSBarry Smith Handle case where matrix is stored on disk as a dense matrix 196990ace30eSBarry Smith */ 197090ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { 1971be5d1d56SKris Buschelman ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr); 19723a40ed3dSBarry Smith PetscFunctionReturn(0); 197390ace30eSBarry Smith } 197490ace30eSBarry Smith 19758965ea79SLois Curfman McInnes /* determine ownership of all rows */ 1976e44c0bd4SBarry Smith m = PetscMPIIntCast(M/size + ((M % size) > rank)); 1977e44c0bd4SBarry Smith ierr = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr); 1978ca161407SBarry Smith ierr = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr); 19798965ea79SLois Curfman McInnes rowners[0] = 0; 19808965ea79SLois Curfman McInnes for (i=2; i<=size; i++) { 19818965ea79SLois Curfman McInnes rowners[i] += rowners[i-1]; 19828965ea79SLois Curfman McInnes } 19838965ea79SLois Curfman McInnes rstart = rowners[rank]; 19848965ea79SLois Curfman McInnes rend = rowners[rank+1]; 19858965ea79SLois Curfman McInnes 19868965ea79SLois Curfman McInnes /* distribute row lengths to all processors */ 198713f74950SBarry Smith ierr = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr); 19888965ea79SLois Curfman McInnes offlens = ourlens + (rend-rstart); 19898965ea79SLois Curfman McInnes if (!rank) { 199013f74950SBarry Smith ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 19910752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 199213f74950SBarry Smith ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr); 19938965ea79SLois Curfman McInnes for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i]; 19941466f1e1SBarry Smith ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr); 1995606d414cSSatish Balay ierr = PetscFree(sndcounts);CHKERRQ(ierr); 1996ca161407SBarry Smith } else { 19971466f1e1SBarry Smith ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr); 19988965ea79SLois Curfman McInnes } 19998965ea79SLois Curfman McInnes 20008965ea79SLois Curfman McInnes if (!rank) { 20018965ea79SLois Curfman McInnes /* calculate the number of nonzeros on each processor */ 200213f74950SBarry Smith ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr); 200313f74950SBarry Smith ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr); 20048965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 20058965ea79SLois Curfman McInnes for (j=rowners[i]; j< rowners[i+1]; j++) { 20068965ea79SLois Curfman McInnes procsnz[i] += rowlengths[j]; 20078965ea79SLois Curfman McInnes } 20088965ea79SLois Curfman McInnes } 2009606d414cSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 20108965ea79SLois Curfman McInnes 20118965ea79SLois Curfman McInnes /* determine max buffer needed and allocate it */ 20128965ea79SLois Curfman McInnes maxnz = 0; 20138965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 20140452661fSBarry Smith maxnz = PetscMax(maxnz,procsnz[i]); 20158965ea79SLois Curfman McInnes } 201613f74950SBarry Smith ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr); 20178965ea79SLois Curfman McInnes 20188965ea79SLois Curfman McInnes /* read in my part of the matrix column indices */ 20198965ea79SLois Curfman McInnes nz = procsnz[0]; 202013f74950SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr); 20210752156aSBarry Smith ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr); 20228965ea79SLois Curfman McInnes 20238965ea79SLois Curfman McInnes /* read in every one elses and ship off */ 20248965ea79SLois Curfman McInnes for (i=1; i<size; i++) { 20258965ea79SLois Curfman McInnes nz = procsnz[i]; 20260752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr); 202713f74950SBarry Smith ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr); 20288965ea79SLois Curfman McInnes } 2029606d414cSSatish Balay ierr = PetscFree(cols);CHKERRQ(ierr); 20303a40ed3dSBarry Smith } else { 20318965ea79SLois Curfman McInnes /* determine buffer space needed for message */ 20328965ea79SLois Curfman McInnes nz = 0; 20338965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20348965ea79SLois Curfman McInnes nz += ourlens[i]; 20358965ea79SLois Curfman McInnes } 203613f74950SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr); 20378965ea79SLois Curfman McInnes 20388965ea79SLois Curfman McInnes /* receive message of column indices*/ 203913f74950SBarry Smith ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr); 204013f74950SBarry Smith ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr); 204129bbc08cSBarry Smith if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file"); 20428965ea79SLois Curfman McInnes } 20438965ea79SLois Curfman McInnes 20448965ea79SLois Curfman McInnes /* loop over local rows, determining number of off diagonal entries */ 204513f74950SBarry Smith ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr); 20468965ea79SLois Curfman McInnes jj = 0; 20478965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20488965ea79SLois Curfman McInnes for (j=0; j<ourlens[i]; j++) { 20498965ea79SLois Curfman McInnes if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++; 20508965ea79SLois Curfman McInnes jj++; 20518965ea79SLois Curfman McInnes } 20528965ea79SLois Curfman McInnes } 20538965ea79SLois Curfman McInnes 20548965ea79SLois Curfman McInnes /* create our matrix */ 20558965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20568965ea79SLois Curfman McInnes ourlens[i] -= offlens[i]; 20578965ea79SLois Curfman McInnes } 2058f69a0ea3SMatthew Knepley ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 2059f69a0ea3SMatthew Knepley ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr); 2060be5d1d56SKris Buschelman ierr = MatSetType(*newmat,type);CHKERRQ(ierr); 2061878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr); 20628965ea79SLois Curfman McInnes A = *newmat; 20638965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20648965ea79SLois Curfman McInnes ourlens[i] += offlens[i]; 20658965ea79SLois Curfman McInnes } 20668965ea79SLois Curfman McInnes 20678965ea79SLois Curfman McInnes if (!rank) { 206887828ca2SBarry Smith ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 20698965ea79SLois Curfman McInnes 20708965ea79SLois Curfman McInnes /* read in my part of the matrix numerical values */ 20718965ea79SLois Curfman McInnes nz = procsnz[0]; 20720752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 20738965ea79SLois Curfman McInnes 20748965ea79SLois Curfman McInnes /* insert into matrix */ 20758965ea79SLois Curfman McInnes jj = rstart; 20768965ea79SLois Curfman McInnes smycols = mycols; 20778965ea79SLois Curfman McInnes svals = vals; 20788965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20798965ea79SLois Curfman McInnes ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr); 20808965ea79SLois Curfman McInnes smycols += ourlens[i]; 20818965ea79SLois Curfman McInnes svals += ourlens[i]; 20828965ea79SLois Curfman McInnes jj++; 20838965ea79SLois Curfman McInnes } 20848965ea79SLois Curfman McInnes 20858965ea79SLois Curfman McInnes /* read in other processors and ship out */ 20868965ea79SLois Curfman McInnes for (i=1; i<size; i++) { 20878965ea79SLois Curfman McInnes nz = procsnz[i]; 20880752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 20897adad957SLisandro Dalcin ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr); 20908965ea79SLois Curfman McInnes } 2091606d414cSSatish Balay ierr = PetscFree(procsnz);CHKERRQ(ierr); 20923a40ed3dSBarry Smith } else { 20938965ea79SLois Curfman McInnes /* receive numeric values */ 209484d528b1SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 20958965ea79SLois Curfman McInnes 20968965ea79SLois Curfman McInnes /* receive message of values*/ 20977adad957SLisandro Dalcin ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr); 2098ca161407SBarry Smith ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr); 209929bbc08cSBarry Smith if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file"); 21008965ea79SLois Curfman McInnes 21018965ea79SLois Curfman McInnes /* insert into matrix */ 21028965ea79SLois Curfman McInnes jj = rstart; 21038965ea79SLois Curfman McInnes smycols = mycols; 21048965ea79SLois Curfman McInnes svals = vals; 21058965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 21068965ea79SLois Curfman McInnes ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr); 21078965ea79SLois Curfman McInnes smycols += ourlens[i]; 21088965ea79SLois Curfman McInnes svals += ourlens[i]; 21098965ea79SLois Curfman McInnes jj++; 21108965ea79SLois Curfman McInnes } 21118965ea79SLois Curfman McInnes } 2112606d414cSSatish Balay ierr = PetscFree(ourlens);CHKERRQ(ierr); 2113606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 2114606d414cSSatish Balay ierr = PetscFree(mycols);CHKERRQ(ierr); 2115606d414cSSatish Balay ierr = PetscFree(rowners);CHKERRQ(ierr); 21168965ea79SLois Curfman McInnes 21176d4a8577SBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 21186d4a8577SBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 21193a40ed3dSBarry Smith PetscFunctionReturn(0); 21208965ea79SLois Curfman McInnes } 212190ace30eSBarry Smith 21226e4ee0c6SHong Zhang #undef __FUNCT__ 21236e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense" 21246e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag) 21256e4ee0c6SHong Zhang { 21266e4ee0c6SHong Zhang Mat_MPIDense *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data; 21276e4ee0c6SHong Zhang Mat a,b; 21286e4ee0c6SHong Zhang PetscTruth flg; 21296e4ee0c6SHong Zhang PetscErrorCode ierr; 213090ace30eSBarry Smith 21316e4ee0c6SHong Zhang PetscFunctionBegin; 21326e4ee0c6SHong Zhang a = matA->A; 21336e4ee0c6SHong Zhang b = matB->A; 21346e4ee0c6SHong Zhang ierr = MatEqual(a,b,&flg);CHKERRQ(ierr); 21357adad957SLisandro Dalcin ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr); 21366e4ee0c6SHong Zhang PetscFunctionReturn(0); 21376e4ee0c6SHong Zhang } 213890ace30eSBarry Smith 213909d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 214009d27a7eSBarry Smith 214109d27a7eSBarry Smith #undef __FUNCT__ 214209d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage" 214309d27a7eSBarry Smith /*@C 214409d27a7eSBarry Smith PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK. 214509d27a7eSBarry Smith Level: developer 214609d27a7eSBarry Smith 214709d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK 214809d27a7eSBarry Smith .seealso: PetscFinalize() 214909d27a7eSBarry Smith @*/ 215009d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void) 215109d27a7eSBarry Smith { 215209d27a7eSBarry Smith PetscErrorCode ierr; 215309d27a7eSBarry Smith 215409d27a7eSBarry Smith PetscFunctionBegin; 215509d27a7eSBarry Smith ierr = PLA_Finalize();CHKERRQ(ierr); 215609d27a7eSBarry Smith PetscFunctionReturn(0); 215709d27a7eSBarry Smith } 215809d27a7eSBarry Smith 215909d27a7eSBarry Smith #undef __FUNCT__ 216009d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage" 216109d27a7eSBarry Smith /*@C 216209d27a7eSBarry Smith PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is 2163db4efbfdSBarry Smith called from MatCreate_MPIDense() the first time an MPI dense matrix is called. 216409d27a7eSBarry Smith 216509d27a7eSBarry Smith Input Parameter: 2166db4efbfdSBarry Smith . comm - the communicator the matrix lives on 216709d27a7eSBarry Smith 216809d27a7eSBarry Smith Level: developer 216909d27a7eSBarry Smith 2170db4efbfdSBarry Smith Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the 2171db4efbfdSBarry Smith same communicator (because there is some global state that is initialized and used for all matrices). In addition if 2172db4efbfdSBarry Smith PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators 2173db4efbfdSBarry Smith cannot overlap. 2174db4efbfdSBarry Smith 217509d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK 217609d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize() 217709d27a7eSBarry Smith @*/ 2178db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm) 217909d27a7eSBarry Smith { 2180eae6fb2eSBarry Smith PetscMPIInt size; 218109d27a7eSBarry Smith PetscErrorCode ierr; 218209d27a7eSBarry Smith 218309d27a7eSBarry Smith PetscFunctionBegin; 218409d27a7eSBarry Smith if (!PLA_Initialized(PETSC_NULL)) { 218509d27a7eSBarry Smith 218609d27a7eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 218704fea9ffSBarry Smith Plapack_nprows = 1; 218804fea9ffSBarry Smith Plapack_npcols = size; 218909d27a7eSBarry Smith 2190aeccfd6fSBarry Smith ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr); 2191aeccfd6fSBarry Smith ierr = PetscOptionsInt("-plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr); 2192aeccfd6fSBarry Smith ierr = PetscOptionsInt("-plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr); 219379b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG) 219479b0a62dSBarry Smith Plapack_ierror = 3; 219579b0a62dSBarry Smith #else 219679b0a62dSBarry Smith Plapack_ierror = 0; 219779b0a62dSBarry Smith #endif 2198eae6fb2eSBarry Smith ierr = PetscOptionsInt("-plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr); 2199eae6fb2eSBarry Smith if (Plapack_ierror){ 2200eae6fb2eSBarry Smith ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr); 2201aeccfd6fSBarry Smith } else { 2202eae6fb2eSBarry Smith ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr); 2203aeccfd6fSBarry Smith } 2204aeccfd6fSBarry Smith 2205eae6fb2eSBarry Smith Plapack_nb_alg = 0; 220679b0a62dSBarry Smith ierr = PetscOptionsInt("-plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr); 2207eae6fb2eSBarry Smith if (Plapack_nb_alg) { 2208eae6fb2eSBarry Smith ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr); 2209aeccfd6fSBarry Smith } 2210aeccfd6fSBarry Smith PetscOptionsEnd(); 2211aeccfd6fSBarry Smith 221204fea9ffSBarry Smith ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr); 221304fea9ffSBarry Smith ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr); 221409d27a7eSBarry Smith ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr); 221509d27a7eSBarry Smith } 221609d27a7eSBarry Smith PetscFunctionReturn(0); 221709d27a7eSBarry Smith } 221890ace30eSBarry Smith 221990ace30eSBarry Smith 222009d27a7eSBarry Smith #endif 2221