1be1d678aSKris Buschelman #define PETSCMAT_DLL 2be1d678aSKris Buschelman 3ed3cc1f0SBarry Smith /* 4ed3cc1f0SBarry Smith Basic functions for basic parallel dense matrices. 5ed3cc1f0SBarry Smith */ 6ed3cc1f0SBarry Smith 704fea9ffSBarry Smith 87c4f633dSBarry Smith #include "../src/mat/impls/dense/mpi/mpidense.h" /*I "petscmat.h" I*/ 9eae6fb2eSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 10eae6fb2eSBarry Smith static PetscMPIInt Plapack_nprows,Plapack_npcols,Plapack_ierror,Plapack_nb_alg; 11eae6fb2eSBarry Smith static MPI_Comm Plapack_comm_2d; 12eae6fb2eSBarry Smith #endif 138965ea79SLois Curfman McInnes 14ba8c8a56SBarry Smith #undef __FUNCT__ 15ab92ecdeSBarry Smith #define __FUNCT__ "MatDenseGetLocalMatrix" 16ab92ecdeSBarry Smith /*@ 17ab92ecdeSBarry Smith 18ab92ecdeSBarry Smith MatDenseGetLocalMatrix - For a MATMPIDENSE or MATSEQDENSE matrix returns the sequential 19ab92ecdeSBarry Smith matrix that represents the operator. For sequential matrices it returns itself. 20ab92ecdeSBarry Smith 21ab92ecdeSBarry Smith Input Parameter: 22ab92ecdeSBarry Smith . A - the Seq or MPI dense matrix 23ab92ecdeSBarry Smith 24ab92ecdeSBarry Smith Output Parameter: 25ab92ecdeSBarry Smith . B - the inner matrix 26ab92ecdeSBarry Smith 278e6c10adSSatish Balay Level: intermediate 288e6c10adSSatish Balay 29ab92ecdeSBarry Smith @*/ 30ab92ecdeSBarry Smith PetscErrorCode MatDenseGetLocalMatrix(Mat A,Mat *B) 31ab92ecdeSBarry Smith { 32ab92ecdeSBarry Smith Mat_MPIDense *mat = (Mat_MPIDense*)A->data; 33ab92ecdeSBarry Smith PetscErrorCode ierr; 34ab92ecdeSBarry Smith PetscTruth flg; 35ab92ecdeSBarry Smith 36ab92ecdeSBarry Smith PetscFunctionBegin; 37ab92ecdeSBarry Smith ierr = PetscTypeCompare((PetscObject)A,MATMPIDENSE,&flg);CHKERRQ(ierr); 38ab92ecdeSBarry Smith if (flg) { 39ab92ecdeSBarry Smith *B = mat->A; 40ab92ecdeSBarry Smith } else { 41ab92ecdeSBarry Smith *B = A; 42ab92ecdeSBarry Smith } 43ab92ecdeSBarry Smith PetscFunctionReturn(0); 44ab92ecdeSBarry Smith } 45ab92ecdeSBarry Smith 46ab92ecdeSBarry Smith #undef __FUNCT__ 47ba8c8a56SBarry Smith #define __FUNCT__ "MatGetRow_MPIDense" 48ba8c8a56SBarry Smith PetscErrorCode MatGetRow_MPIDense(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 49ba8c8a56SBarry Smith { 50ba8c8a56SBarry Smith Mat_MPIDense *mat = (Mat_MPIDense*)A->data; 51ba8c8a56SBarry Smith PetscErrorCode ierr; 52d0f46423SBarry Smith PetscInt lrow,rstart = A->rmap->rstart,rend = A->rmap->rend; 53ba8c8a56SBarry Smith 54ba8c8a56SBarry Smith PetscFunctionBegin; 55ba8c8a56SBarry Smith if (row < rstart || row >= rend) SETERRQ(PETSC_ERR_SUP,"only local rows") 56ba8c8a56SBarry Smith lrow = row - rstart; 57ba8c8a56SBarry Smith ierr = MatGetRow(mat->A,lrow,nz,(const PetscInt **)idx,(const PetscScalar **)v);CHKERRQ(ierr); 58ba8c8a56SBarry Smith PetscFunctionReturn(0); 59ba8c8a56SBarry Smith } 60ba8c8a56SBarry Smith 61ba8c8a56SBarry Smith #undef __FUNCT__ 62ba8c8a56SBarry Smith #define __FUNCT__ "MatRestoreRow_MPIDense" 63ba8c8a56SBarry Smith PetscErrorCode MatRestoreRow_MPIDense(Mat mat,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 64ba8c8a56SBarry Smith { 65ba8c8a56SBarry Smith PetscErrorCode ierr; 66ba8c8a56SBarry Smith 67ba8c8a56SBarry Smith PetscFunctionBegin; 68ba8c8a56SBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 69ba8c8a56SBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 70ba8c8a56SBarry Smith PetscFunctionReturn(0); 71ba8c8a56SBarry Smith } 72ba8c8a56SBarry Smith 730de54da6SSatish Balay EXTERN_C_BEGIN 744a2ae208SSatish Balay #undef __FUNCT__ 754a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonalBlock_MPIDense" 76be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatGetDiagonalBlock_MPIDense(Mat A,PetscTruth *iscopy,MatReuse reuse,Mat *B) 770de54da6SSatish Balay { 780de54da6SSatish Balay Mat_MPIDense *mdn = (Mat_MPIDense*)A->data; 796849ba73SBarry Smith PetscErrorCode ierr; 80d0f46423SBarry Smith PetscInt m = A->rmap->n,rstart = A->rmap->rstart; 8187828ca2SBarry Smith PetscScalar *array; 820de54da6SSatish Balay MPI_Comm comm; 830de54da6SSatish Balay 840de54da6SSatish Balay PetscFunctionBegin; 85d0f46423SBarry Smith if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_SUP,"Only square matrices supported."); 860de54da6SSatish Balay 870de54da6SSatish Balay /* The reuse aspect is not implemented efficiently */ 880de54da6SSatish Balay if (reuse) { ierr = MatDestroy(*B);CHKERRQ(ierr);} 890de54da6SSatish Balay 900de54da6SSatish Balay ierr = PetscObjectGetComm((PetscObject)(mdn->A),&comm);CHKERRQ(ierr); 910de54da6SSatish Balay ierr = MatGetArray(mdn->A,&array);CHKERRQ(ierr); 92f69a0ea3SMatthew Knepley ierr = MatCreate(comm,B);CHKERRQ(ierr); 93f69a0ea3SMatthew Knepley ierr = MatSetSizes(*B,m,m,m,m);CHKERRQ(ierr); 947adad957SLisandro Dalcin ierr = MatSetType(*B,((PetscObject)mdn->A)->type_name);CHKERRQ(ierr); 955c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(*B,array+m*rstart);CHKERRQ(ierr); 960de54da6SSatish Balay ierr = MatRestoreArray(mdn->A,&array);CHKERRQ(ierr); 970de54da6SSatish Balay ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 980de54da6SSatish Balay ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 990de54da6SSatish Balay 1000de54da6SSatish Balay *iscopy = PETSC_TRUE; 1010de54da6SSatish Balay PetscFunctionReturn(0); 1020de54da6SSatish Balay } 1030de54da6SSatish Balay EXTERN_C_END 1040de54da6SSatish Balay 1054a2ae208SSatish Balay #undef __FUNCT__ 1064a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_MPIDense" 10713f74950SBarry Smith PetscErrorCode MatSetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],const PetscScalar v[],InsertMode addv) 1088965ea79SLois Curfman McInnes { 10939b7565bSBarry Smith Mat_MPIDense *A = (Mat_MPIDense*)mat->data; 110dfbe8321SBarry Smith PetscErrorCode ierr; 111d0f46423SBarry Smith PetscInt i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row; 112273d9f13SBarry Smith PetscTruth roworiented = A->roworiented; 1138965ea79SLois Curfman McInnes 1143a40ed3dSBarry Smith PetscFunctionBegin; 1158965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 1165ef9f2a5SBarry Smith if (idxm[i] < 0) continue; 117d0f46423SBarry Smith if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large"); 1188965ea79SLois Curfman McInnes if (idxm[i] >= rstart && idxm[i] < rend) { 1198965ea79SLois Curfman McInnes row = idxm[i] - rstart; 12039b7565bSBarry Smith if (roworiented) { 12139b7565bSBarry Smith ierr = MatSetValues(A->A,1,&row,n,idxn,v+i*n,addv);CHKERRQ(ierr); 1223a40ed3dSBarry Smith } else { 1238965ea79SLois Curfman McInnes for (j=0; j<n; j++) { 1245ef9f2a5SBarry Smith if (idxn[j] < 0) continue; 125d0f46423SBarry Smith if (idxn[j] >= mat->cmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large"); 12639b7565bSBarry Smith ierr = MatSetValues(A->A,1,&row,1,&idxn[j],v+i+j*m,addv);CHKERRQ(ierr); 12739b7565bSBarry Smith } 1288965ea79SLois Curfman McInnes } 1293a40ed3dSBarry Smith } else { 1303782ba37SSatish Balay if (!A->donotstash) { 13139b7565bSBarry Smith if (roworiented) { 1328798bf22SSatish Balay ierr = MatStashValuesRow_Private(&mat->stash,idxm[i],n,idxn,v+i*n);CHKERRQ(ierr); 133d36fbae8SSatish Balay } else { 1348798bf22SSatish Balay ierr = MatStashValuesCol_Private(&mat->stash,idxm[i],n,idxn,v+i,m);CHKERRQ(ierr); 13539b7565bSBarry Smith } 136b49de8d1SLois Curfman McInnes } 137b49de8d1SLois Curfman McInnes } 1383782ba37SSatish Balay } 1393a40ed3dSBarry Smith PetscFunctionReturn(0); 140b49de8d1SLois Curfman McInnes } 141b49de8d1SLois Curfman McInnes 1424a2ae208SSatish Balay #undef __FUNCT__ 1434a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_MPIDense" 14413f74950SBarry Smith PetscErrorCode MatGetValues_MPIDense(Mat mat,PetscInt m,const PetscInt idxm[],PetscInt n,const PetscInt idxn[],PetscScalar v[]) 145b49de8d1SLois Curfman McInnes { 146b49de8d1SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 147dfbe8321SBarry Smith PetscErrorCode ierr; 148d0f46423SBarry Smith PetscInt i,j,rstart = mat->rmap->rstart,rend = mat->rmap->rend,row; 149b49de8d1SLois Curfman McInnes 1503a40ed3dSBarry Smith PetscFunctionBegin; 151b49de8d1SLois Curfman McInnes for (i=0; i<m; i++) { 15297e567efSBarry Smith if (idxm[i] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative row"); */ 153d0f46423SBarry Smith if (idxm[i] >= mat->rmap->N) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Row too large"); 154b49de8d1SLois Curfman McInnes if (idxm[i] >= rstart && idxm[i] < rend) { 155b49de8d1SLois Curfman McInnes row = idxm[i] - rstart; 156b49de8d1SLois Curfman McInnes for (j=0; j<n; j++) { 15797e567efSBarry Smith if (idxn[j] < 0) continue; /* SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Negative column"); */ 158d0f46423SBarry Smith if (idxn[j] >= mat->cmap->N) { 15929bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Column too large"); 160a8c6a408SBarry Smith } 161b49de8d1SLois Curfman McInnes ierr = MatGetValues(mdn->A,1,&row,1,&idxn[j],v+i*n+j);CHKERRQ(ierr); 162b49de8d1SLois Curfman McInnes } 163a8c6a408SBarry Smith } else { 16429bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"Only local values currently supported"); 1658965ea79SLois Curfman McInnes } 1668965ea79SLois Curfman McInnes } 1673a40ed3dSBarry Smith PetscFunctionReturn(0); 1688965ea79SLois Curfman McInnes } 1698965ea79SLois Curfman McInnes 1704a2ae208SSatish Balay #undef __FUNCT__ 1714a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_MPIDense" 172dfbe8321SBarry Smith PetscErrorCode MatGetArray_MPIDense(Mat A,PetscScalar *array[]) 173ff14e315SSatish Balay { 174ff14e315SSatish Balay Mat_MPIDense *a = (Mat_MPIDense*)A->data; 175dfbe8321SBarry Smith PetscErrorCode ierr; 176ff14e315SSatish Balay 1773a40ed3dSBarry Smith PetscFunctionBegin; 178ff14e315SSatish Balay ierr = MatGetArray(a->A,array);CHKERRQ(ierr); 1793a40ed3dSBarry Smith PetscFunctionReturn(0); 180ff14e315SSatish Balay } 181ff14e315SSatish Balay 1824a2ae208SSatish Balay #undef __FUNCT__ 1834a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_MPIDense" 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; 1895d0c19d7SBarry Smith PetscInt i,j,rstart,rend,nrows,ncols,nlrows,nlcols; 1905d0c19d7SBarry Smith const PetscInt *irow,*icol; 19187828ca2SBarry Smith PetscScalar *av,*bv,*v = lmat->v; 192ca3fa75bSLois Curfman McInnes Mat newmat; 193ca3fa75bSLois Curfman McInnes 194ca3fa75bSLois Curfman McInnes PetscFunctionBegin; 195ca3fa75bSLois Curfman McInnes ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 196ca3fa75bSLois Curfman McInnes ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 197b9b97703SBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 198b9b97703SBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 199ca3fa75bSLois Curfman McInnes 200ca3fa75bSLois Curfman McInnes /* No parallel redistribution currently supported! Should really check each index set 2017eba5e9cSLois Curfman McInnes to comfirm that it is OK. ... Currently supports only submatrix same partitioning as 2027eba5e9cSLois Curfman McInnes original matrix! */ 203ca3fa75bSLois Curfman McInnes 204ca3fa75bSLois Curfman McInnes ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr); 2057eba5e9cSLois Curfman McInnes ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 206ca3fa75bSLois Curfman McInnes 207ca3fa75bSLois Curfman McInnes /* Check submatrix call */ 208ca3fa75bSLois Curfman McInnes if (scall == MAT_REUSE_MATRIX) { 20929bbc08cSBarry Smith /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */ 2107eba5e9cSLois Curfman McInnes /* Really need to test rows and column sizes! */ 211ca3fa75bSLois Curfman McInnes newmat = *B; 212ca3fa75bSLois Curfman McInnes } else { 213ca3fa75bSLois Curfman McInnes /* Create and fill new matrix */ 2147adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr); 215f69a0ea3SMatthew Knepley ierr = MatSetSizes(newmat,nrows,cs,PETSC_DECIDE,ncols);CHKERRQ(ierr); 2167adad957SLisandro Dalcin ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 217878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr); 218ca3fa75bSLois Curfman McInnes } 219ca3fa75bSLois Curfman McInnes 220ca3fa75bSLois Curfman McInnes /* Now extract the data pointers and do the copy, column at a time */ 221ca3fa75bSLois Curfman McInnes newmatd = (Mat_MPIDense*)newmat->data; 222ca3fa75bSLois Curfman McInnes bv = ((Mat_SeqDense *)newmatd->A->data)->v; 223ca3fa75bSLois Curfman McInnes 224ca3fa75bSLois Curfman McInnes for (i=0; i<ncols; i++) { 225ca3fa75bSLois Curfman McInnes av = v + nlrows*icol[i]; 226ca3fa75bSLois Curfman McInnes for (j=0; j<nrows; j++) { 2277eba5e9cSLois Curfman McInnes *bv++ = av[irow[j] - rstart]; 228ca3fa75bSLois Curfman McInnes } 229ca3fa75bSLois Curfman McInnes } 230ca3fa75bSLois Curfman McInnes 231ca3fa75bSLois Curfman McInnes /* Assemble the matrices so that the correct flags are set */ 232ca3fa75bSLois Curfman McInnes ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 233ca3fa75bSLois Curfman McInnes ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 234ca3fa75bSLois Curfman McInnes 235ca3fa75bSLois Curfman McInnes /* Free work space */ 236ca3fa75bSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 237ca3fa75bSLois Curfman McInnes ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 238ca3fa75bSLois Curfman McInnes *B = newmat; 239ca3fa75bSLois Curfman McInnes PetscFunctionReturn(0); 240ca3fa75bSLois Curfman McInnes } 241ca3fa75bSLois Curfman McInnes 2424a2ae208SSatish Balay #undef __FUNCT__ 2434a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense" 244dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[]) 245ff14e315SSatish Balay { 2463a40ed3dSBarry Smith PetscFunctionBegin; 2473a40ed3dSBarry Smith PetscFunctionReturn(0); 248ff14e315SSatish Balay } 249ff14e315SSatish Balay 2504a2ae208SSatish Balay #undef __FUNCT__ 2514a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense" 252dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode) 2538965ea79SLois Curfman McInnes { 25439ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 2557adad957SLisandro Dalcin MPI_Comm comm = ((PetscObject)mat)->comm; 256dfbe8321SBarry Smith PetscErrorCode ierr; 25713f74950SBarry Smith PetscInt nstash,reallocs; 2588965ea79SLois Curfman McInnes InsertMode addv; 2598965ea79SLois Curfman McInnes 2603a40ed3dSBarry Smith PetscFunctionBegin; 2618965ea79SLois Curfman McInnes /* make sure all processors are either in INSERTMODE or ADDMODE */ 262ca161407SBarry Smith ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr); 2637056b6fcSBarry Smith if (addv == (ADD_VALUES|INSERT_VALUES)) { 26429bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs"); 2658965ea79SLois Curfman McInnes } 266e0fa3b82SLois Curfman McInnes mat->insertmode = addv; /* in case this processor had no cache */ 2678965ea79SLois Curfman McInnes 268d0f46423SBarry Smith ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr); 2698798bf22SSatish Balay ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr); 270ae15b995SBarry Smith ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr); 2713a40ed3dSBarry Smith PetscFunctionReturn(0); 2728965ea79SLois Curfman McInnes } 2738965ea79SLois Curfman McInnes 2744a2ae208SSatish Balay #undef __FUNCT__ 2754a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense" 276dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode) 2778965ea79SLois Curfman McInnes { 27839ddd567SLois Curfman McInnes Mat_MPIDense *mdn=(Mat_MPIDense*)mat->data; 2796849ba73SBarry Smith PetscErrorCode ierr; 28013f74950SBarry Smith PetscInt i,*row,*col,flg,j,rstart,ncols; 28113f74950SBarry Smith PetscMPIInt n; 28287828ca2SBarry Smith PetscScalar *val; 283e0fa3b82SLois Curfman McInnes InsertMode addv=mat->insertmode; 2848965ea79SLois Curfman McInnes 2853a40ed3dSBarry Smith PetscFunctionBegin; 2868965ea79SLois Curfman McInnes /* wait on receives */ 2877ef1d9bdSSatish Balay while (1) { 2888798bf22SSatish Balay ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr); 2897ef1d9bdSSatish Balay if (!flg) break; 2908965ea79SLois Curfman McInnes 2917ef1d9bdSSatish Balay for (i=0; i<n;) { 2927ef1d9bdSSatish Balay /* Now identify the consecutive vals belonging to the same row */ 2937ef1d9bdSSatish Balay for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; } 2947ef1d9bdSSatish Balay if (j < n) ncols = j-i; 2957ef1d9bdSSatish Balay else ncols = n-i; 2967ef1d9bdSSatish Balay /* Now assemble all these values with a single function call */ 2977ef1d9bdSSatish Balay ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr); 2987ef1d9bdSSatish Balay i = j; 2998965ea79SLois Curfman McInnes } 3007ef1d9bdSSatish Balay } 3018798bf22SSatish Balay ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr); 3028965ea79SLois Curfman McInnes 30339ddd567SLois Curfman McInnes ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr); 30439ddd567SLois Curfman McInnes ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr); 3058965ea79SLois Curfman McInnes 3066d4a8577SBarry Smith if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) { 30739ddd567SLois Curfman McInnes ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr); 3088965ea79SLois Curfman McInnes } 3093a40ed3dSBarry Smith PetscFunctionReturn(0); 3108965ea79SLois Curfman McInnes } 3118965ea79SLois Curfman McInnes 3124a2ae208SSatish Balay #undef __FUNCT__ 3134a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense" 314dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A) 3158965ea79SLois Curfman McInnes { 316dfbe8321SBarry Smith PetscErrorCode ierr; 31739ddd567SLois Curfman McInnes Mat_MPIDense *l = (Mat_MPIDense*)A->data; 3183a40ed3dSBarry Smith 3193a40ed3dSBarry Smith PetscFunctionBegin; 3203a40ed3dSBarry Smith ierr = MatZeroEntries(l->A);CHKERRQ(ierr); 3213a40ed3dSBarry Smith PetscFunctionReturn(0); 3228965ea79SLois Curfman McInnes } 3238965ea79SLois Curfman McInnes 3248965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the 3258965ea79SLois Curfman McInnes matrix is square and the column and row owerships are identical. 3268965ea79SLois Curfman McInnes This is a BUG. The only way to fix it seems to be to access 3273501a2bdSLois Curfman McInnes mdn->A and mdn->B directly and not through the MatZeroRows() 3288965ea79SLois Curfman McInnes routine. 3298965ea79SLois Curfman McInnes */ 3304a2ae208SSatish Balay #undef __FUNCT__ 3314a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense" 332f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag) 3338965ea79SLois Curfman McInnes { 33439ddd567SLois Curfman McInnes Mat_MPIDense *l = (Mat_MPIDense*)A->data; 3356849ba73SBarry Smith PetscErrorCode ierr; 336d0f46423SBarry Smith PetscInt i,*owners = A->rmap->range; 33713f74950SBarry Smith PetscInt *nprocs,j,idx,nsends; 33813f74950SBarry Smith PetscInt nmax,*svalues,*starts,*owner,nrecvs; 3397adad957SLisandro Dalcin PetscInt *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source; 34013f74950SBarry Smith PetscInt *lens,*lrows,*values; 34113f74950SBarry Smith PetscMPIInt n,imdex,rank = l->rank,size = l->size; 3427adad957SLisandro Dalcin MPI_Comm comm = ((PetscObject)A)->comm; 3438965ea79SLois Curfman McInnes MPI_Request *send_waits,*recv_waits; 3448965ea79SLois Curfman McInnes MPI_Status recv_status,*send_status; 34535d8aa7fSBarry Smith PetscTruth found; 3468965ea79SLois Curfman McInnes 3473a40ed3dSBarry Smith PetscFunctionBegin; 3488965ea79SLois Curfman McInnes /* first count number of contributors to each processor */ 34913f74950SBarry Smith ierr = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr); 35013f74950SBarry Smith ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr); 35113f74950SBarry Smith ierr = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/ 3528965ea79SLois Curfman McInnes for (i=0; i<N; i++) { 3538965ea79SLois Curfman McInnes idx = rows[i]; 35435d8aa7fSBarry Smith found = PETSC_FALSE; 3558965ea79SLois Curfman McInnes for (j=0; j<size; j++) { 3568965ea79SLois Curfman McInnes if (idx >= owners[j] && idx < owners[j+1]) { 357c1dc657dSBarry Smith nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break; 3588965ea79SLois Curfman McInnes } 3598965ea79SLois Curfman McInnes } 36029bbc08cSBarry Smith if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range"); 3618965ea79SLois Curfman McInnes } 362c1dc657dSBarry Smith nsends = 0; for (i=0; i<size; i++) { nsends += nprocs[2*i+1];} 3638965ea79SLois Curfman McInnes 3648965ea79SLois Curfman McInnes /* inform other processors of number of messages and max length*/ 365c1dc657dSBarry Smith ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr); 3668965ea79SLois Curfman McInnes 3678965ea79SLois Curfman McInnes /* post receives: */ 36813f74950SBarry Smith ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr); 369b0a32e0cSBarry Smith ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr); 3708965ea79SLois Curfman McInnes for (i=0; i<nrecvs; i++) { 37113f74950SBarry Smith ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr); 3728965ea79SLois Curfman McInnes } 3738965ea79SLois Curfman McInnes 3748965ea79SLois Curfman McInnes /* do sends: 3758965ea79SLois Curfman McInnes 1) starts[i] gives the starting index in svalues for stuff going to 3768965ea79SLois Curfman McInnes the ith processor 3778965ea79SLois Curfman McInnes */ 37813f74950SBarry Smith ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr); 379b0a32e0cSBarry Smith ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr); 38013f74950SBarry Smith ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr); 3818965ea79SLois Curfman McInnes starts[0] = 0; 382c1dc657dSBarry Smith for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];} 3838965ea79SLois Curfman McInnes for (i=0; i<N; i++) { 3848965ea79SLois Curfman McInnes svalues[starts[owner[i]]++] = rows[i]; 3858965ea79SLois Curfman McInnes } 3868965ea79SLois Curfman McInnes 3878965ea79SLois Curfman McInnes starts[0] = 0; 388c1dc657dSBarry Smith for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];} 3898965ea79SLois Curfman McInnes count = 0; 3908965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 391c1dc657dSBarry Smith if (nprocs[2*i+1]) { 39213f74950SBarry Smith ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr); 3938965ea79SLois Curfman McInnes } 3948965ea79SLois Curfman McInnes } 395606d414cSSatish Balay ierr = PetscFree(starts);CHKERRQ(ierr); 3968965ea79SLois Curfman McInnes 3978965ea79SLois Curfman McInnes base = owners[rank]; 3988965ea79SLois Curfman McInnes 3998965ea79SLois Curfman McInnes /* wait on receives */ 40013f74950SBarry Smith ierr = PetscMalloc(2*(nrecvs+1)*sizeof(PetscInt),&lens);CHKERRQ(ierr); 4018965ea79SLois Curfman McInnes source = lens + nrecvs; 4028965ea79SLois Curfman McInnes count = nrecvs; slen = 0; 4038965ea79SLois Curfman McInnes while (count) { 404ca161407SBarry Smith ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr); 4058965ea79SLois Curfman McInnes /* unpack receives into our local space */ 40613f74950SBarry Smith ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr); 4078965ea79SLois Curfman McInnes source[imdex] = recv_status.MPI_SOURCE; 4088965ea79SLois Curfman McInnes lens[imdex] = n; 4098965ea79SLois Curfman McInnes slen += n; 4108965ea79SLois Curfman McInnes count--; 4118965ea79SLois Curfman McInnes } 412606d414cSSatish Balay ierr = PetscFree(recv_waits);CHKERRQ(ierr); 4138965ea79SLois Curfman McInnes 4148965ea79SLois Curfman McInnes /* move the data into the send scatter */ 41513f74950SBarry Smith ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr); 4168965ea79SLois Curfman McInnes count = 0; 4178965ea79SLois Curfman McInnes for (i=0; i<nrecvs; i++) { 4188965ea79SLois Curfman McInnes values = rvalues + i*nmax; 4198965ea79SLois Curfman McInnes for (j=0; j<lens[i]; j++) { 4208965ea79SLois Curfman McInnes lrows[count++] = values[j] - base; 4218965ea79SLois Curfman McInnes } 4228965ea79SLois Curfman McInnes } 423606d414cSSatish Balay ierr = PetscFree(rvalues);CHKERRQ(ierr); 424606d414cSSatish Balay ierr = PetscFree(lens);CHKERRQ(ierr); 425606d414cSSatish Balay ierr = PetscFree(owner);CHKERRQ(ierr); 426606d414cSSatish Balay ierr = PetscFree(nprocs);CHKERRQ(ierr); 4278965ea79SLois Curfman McInnes 4288965ea79SLois Curfman McInnes /* actually zap the local rows */ 429f4df32b1SMatthew Knepley ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr); 430606d414cSSatish Balay ierr = PetscFree(lrows);CHKERRQ(ierr); 4318965ea79SLois Curfman McInnes 4328965ea79SLois Curfman McInnes /* wait on sends */ 4338965ea79SLois Curfman McInnes if (nsends) { 434b0a32e0cSBarry Smith ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr); 435ca161407SBarry Smith ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr); 436606d414cSSatish Balay ierr = PetscFree(send_status);CHKERRQ(ierr); 4378965ea79SLois Curfman McInnes } 438606d414cSSatish Balay ierr = PetscFree(send_waits);CHKERRQ(ierr); 439606d414cSSatish Balay ierr = PetscFree(svalues);CHKERRQ(ierr); 4408965ea79SLois Curfman McInnes 4413a40ed3dSBarry Smith PetscFunctionReturn(0); 4428965ea79SLois Curfman McInnes } 4438965ea79SLois Curfman McInnes 4444a2ae208SSatish Balay #undef __FUNCT__ 4454a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense" 446dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy) 4478965ea79SLois Curfman McInnes { 44839ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 449dfbe8321SBarry Smith PetscErrorCode ierr; 450c456f294SBarry Smith 4513a40ed3dSBarry Smith PetscFunctionBegin; 452ca9f406cSSatish Balay ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 453ca9f406cSSatish Balay ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 45444cd7ae7SLois Curfman McInnes ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr); 4553a40ed3dSBarry Smith PetscFunctionReturn(0); 4568965ea79SLois Curfman McInnes } 4578965ea79SLois Curfman McInnes 4584a2ae208SSatish Balay #undef __FUNCT__ 4594a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense" 460dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz) 4618965ea79SLois Curfman McInnes { 46239ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 463dfbe8321SBarry Smith PetscErrorCode ierr; 464c456f294SBarry Smith 4653a40ed3dSBarry Smith PetscFunctionBegin; 466ca9f406cSSatish Balay ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 467ca9f406cSSatish Balay ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 46844cd7ae7SLois Curfman McInnes ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr); 4693a40ed3dSBarry Smith PetscFunctionReturn(0); 4708965ea79SLois Curfman McInnes } 4718965ea79SLois Curfman McInnes 4724a2ae208SSatish Balay #undef __FUNCT__ 4734a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense" 474dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy) 475096963f5SLois Curfman McInnes { 476096963f5SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 477dfbe8321SBarry Smith PetscErrorCode ierr; 47887828ca2SBarry Smith PetscScalar zero = 0.0; 479096963f5SLois Curfman McInnes 4803a40ed3dSBarry Smith PetscFunctionBegin; 4812dcb1b2aSMatthew Knepley ierr = VecSet(yy,zero);CHKERRQ(ierr); 4827c922b88SBarry Smith ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr); 483ca9f406cSSatish Balay ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 484ca9f406cSSatish Balay ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 4853a40ed3dSBarry Smith PetscFunctionReturn(0); 486096963f5SLois Curfman McInnes } 487096963f5SLois Curfman McInnes 4884a2ae208SSatish Balay #undef __FUNCT__ 4894a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense" 490dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz) 491096963f5SLois Curfman McInnes { 492096963f5SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 493dfbe8321SBarry Smith PetscErrorCode ierr; 494096963f5SLois Curfman McInnes 4953a40ed3dSBarry Smith PetscFunctionBegin; 4963501a2bdSLois Curfman McInnes ierr = VecCopy(yy,zz);CHKERRQ(ierr); 4977c922b88SBarry Smith ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr); 498ca9f406cSSatish Balay ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 499ca9f406cSSatish Balay ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 5003a40ed3dSBarry Smith PetscFunctionReturn(0); 501096963f5SLois Curfman McInnes } 502096963f5SLois Curfman McInnes 5034a2ae208SSatish Balay #undef __FUNCT__ 5044a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense" 505dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v) 5068965ea79SLois Curfman McInnes { 50739ddd567SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 508096963f5SLois Curfman McInnes Mat_SeqDense *aloc = (Mat_SeqDense*)a->A->data; 509dfbe8321SBarry Smith PetscErrorCode ierr; 510d0f46423SBarry Smith PetscInt len,i,n,m = A->rmap->n,radd; 51187828ca2SBarry Smith PetscScalar *x,zero = 0.0; 512ed3cc1f0SBarry Smith 5133a40ed3dSBarry Smith PetscFunctionBegin; 5142dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 5151ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 516096963f5SLois Curfman McInnes ierr = VecGetSize(v,&n);CHKERRQ(ierr); 517d0f46423SBarry Smith if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec"); 518d0f46423SBarry Smith len = PetscMin(a->A->rmap->n,a->A->cmap->n); 519d0f46423SBarry Smith radd = A->rmap->rstart*m; 52044cd7ae7SLois Curfman McInnes for (i=0; i<len; i++) { 521096963f5SLois Curfman McInnes x[i] = aloc->v[radd + i*m + i]; 522096963f5SLois Curfman McInnes } 5231ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 5243a40ed3dSBarry Smith PetscFunctionReturn(0); 5258965ea79SLois Curfman McInnes } 5268965ea79SLois Curfman McInnes 5274a2ae208SSatish Balay #undef __FUNCT__ 5284a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense" 529dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat) 5308965ea79SLois Curfman McInnes { 5313501a2bdSLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 532dfbe8321SBarry Smith PetscErrorCode ierr; 53301b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 53401b82886SBarry Smith Mat_Plapack *lu=(Mat_Plapack*)(mat->spptr); 53501b82886SBarry Smith #endif 536ed3cc1f0SBarry Smith 5373a40ed3dSBarry Smith PetscFunctionBegin; 53894d884c6SBarry Smith 539aa482453SBarry Smith #if defined(PETSC_USE_LOG) 540d0f46423SBarry Smith PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N); 5418965ea79SLois Curfman McInnes #endif 5428798bf22SSatish Balay ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr); 5433501a2bdSLois Curfman McInnes ierr = MatDestroy(mdn->A);CHKERRQ(ierr); 54405b42c5fSBarry Smith if (mdn->lvec) {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);} 54505b42c5fSBarry Smith if (mdn->Mvctx) {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);} 54601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 5472fbe02b9SBarry Smith if (lu) { 54862b4c0b3SBarry Smith ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr); 54962b4c0b3SBarry Smith ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr); 55062b4c0b3SBarry Smith ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr); 55109d27a7eSBarry Smith 5522fbe02b9SBarry Smith if (lu->is_pla) { 55301b82886SBarry Smith ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr); 55401b82886SBarry Smith ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr); 55501b82886SBarry Smith ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr); 556622d7880SLois Curfman McInnes } 5572fbe02b9SBarry Smith } 55801b82886SBarry Smith #endif 55901b82886SBarry Smith 560606d414cSSatish Balay ierr = PetscFree(mdn);CHKERRQ(ierr); 561dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr); 562901853e0SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr); 563901853e0SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 5644ae313f4SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr); 5654ae313f4SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr); 5664ae313f4SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr); 5673a40ed3dSBarry Smith PetscFunctionReturn(0); 5688965ea79SLois Curfman McInnes } 56939ddd567SLois Curfman McInnes 5704a2ae208SSatish Balay #undef __FUNCT__ 5714a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary" 5726849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer) 5738965ea79SLois Curfman McInnes { 57439ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 575dfbe8321SBarry Smith PetscErrorCode ierr; 576aa05aa95SBarry Smith PetscViewerFormat format; 577aa05aa95SBarry Smith int fd; 578d0f46423SBarry Smith PetscInt header[4],mmax,N = mat->cmap->N,i,j,m,k; 579aa05aa95SBarry Smith PetscMPIInt rank,tag = ((PetscObject)viewer)->tag,size; 580578230a0SSatish Balay PetscScalar *work,*v,*vv; 581aa05aa95SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)mdn->A->data; 582aa05aa95SBarry Smith MPI_Status status; 5837056b6fcSBarry Smith 5843a40ed3dSBarry Smith PetscFunctionBegin; 58539ddd567SLois Curfman McInnes if (mdn->size == 1) { 58639ddd567SLois Curfman McInnes ierr = MatView(mdn->A,viewer);CHKERRQ(ierr); 587aa05aa95SBarry Smith } else { 588aa05aa95SBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 5897adad957SLisandro Dalcin ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr); 5907adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr); 591aa05aa95SBarry Smith 592aa05aa95SBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 5937973ad04SBarry Smith if (format == PETSC_VIEWER_NATIVE) { 594aa05aa95SBarry Smith 595aa05aa95SBarry Smith if (!rank) { 596aa05aa95SBarry Smith /* store the matrix as a dense matrix */ 597aa05aa95SBarry Smith header[0] = MAT_FILE_COOKIE; 598d0f46423SBarry Smith header[1] = mat->rmap->N; 599aa05aa95SBarry Smith header[2] = N; 600aa05aa95SBarry Smith header[3] = MATRIX_BINARY_FORMAT_DENSE; 601aa05aa95SBarry Smith ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 602aa05aa95SBarry Smith 603aa05aa95SBarry Smith /* get largest work array needed for transposing array */ 604d0f46423SBarry Smith mmax = mat->rmap->n; 605aa05aa95SBarry Smith for (i=1; i<size; i++) { 606d0f46423SBarry Smith mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]); 6078965ea79SLois Curfman McInnes } 608aa05aa95SBarry Smith ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr); 609aa05aa95SBarry Smith 610aa05aa95SBarry Smith /* write out local array, by rows */ 611d0f46423SBarry Smith m = mat->rmap->n; 612aa05aa95SBarry Smith v = a->v; 613aa05aa95SBarry Smith for (j=0; j<N; j++) { 614aa05aa95SBarry Smith for (i=0; i<m; i++) { 615578230a0SSatish Balay work[j + i*N] = *v++; 616aa05aa95SBarry Smith } 617aa05aa95SBarry Smith } 618aa05aa95SBarry Smith ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 619aa05aa95SBarry Smith /* get largest work array to receive messages from other processes, excludes process zero */ 620aa05aa95SBarry Smith mmax = 0; 621aa05aa95SBarry Smith for (i=1; i<size; i++) { 622d0f46423SBarry Smith mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]); 623aa05aa95SBarry Smith } 624578230a0SSatish Balay ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr); 625aa05aa95SBarry Smith for(k = 1; k < size; k++) { 626f8009846SMatthew Knepley v = vv; 627d0f46423SBarry Smith m = mat->rmap->range[k+1] - mat->rmap->range[k]; 6287adad957SLisandro Dalcin ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr); 629aa05aa95SBarry Smith 630aa05aa95SBarry Smith for(j = 0; j < N; j++) { 631aa05aa95SBarry Smith for(i = 0; i < m; i++) { 632578230a0SSatish Balay work[j + i*N] = *v++; 633aa05aa95SBarry Smith } 634aa05aa95SBarry Smith } 635aa05aa95SBarry Smith ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 636aa05aa95SBarry Smith } 637aa05aa95SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 638578230a0SSatish Balay ierr = PetscFree(vv);CHKERRQ(ierr); 639aa05aa95SBarry Smith } else { 640d0f46423SBarry Smith ierr = MPI_Send(a->v,mat->rmap->n*mat->cmap->N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr); 641aa05aa95SBarry Smith } 6424aa34b0aSBarry Smith } else { 6437973ad04SBarry Smith SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_NATIVE"); 644aa05aa95SBarry Smith } 645aa05aa95SBarry Smith } 6463a40ed3dSBarry Smith PetscFunctionReturn(0); 6478965ea79SLois Curfman McInnes } 6488965ea79SLois Curfman McInnes 6494a2ae208SSatish Balay #undef __FUNCT__ 6504a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket" 6516849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer) 6528965ea79SLois Curfman McInnes { 65339ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 654dfbe8321SBarry Smith PetscErrorCode ierr; 65532dcc486SBarry Smith PetscMPIInt size = mdn->size,rank = mdn->rank; 656a313700dSBarry Smith const PetscViewerType vtype; 65732077d6dSBarry Smith PetscTruth iascii,isdraw; 658b0a32e0cSBarry Smith PetscViewer sviewer; 659f3ef73ceSBarry Smith PetscViewerFormat format; 66001b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 661*5d00a290SHong Zhang Mat_Plapack *lu; 66201b82886SBarry Smith #endif 6638965ea79SLois Curfman McInnes 6643a40ed3dSBarry Smith PetscFunctionBegin; 66532077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 666fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 66732077d6dSBarry Smith if (iascii) { 668b0a32e0cSBarry Smith ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr); 669b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 670456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 6714e220ebcSLois Curfman McInnes MatInfo info; 672888f2ed8SSatish Balay ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr); 673d0f46423SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap->n, 67477431f27SBarry Smith (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr); 675b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 67601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 67701b82886SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr); 678*5d00a290SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Processor mesh: nprows %d, npcols %d\n",Plapack_nprows, Plapack_npcols);CHKERRQ(ierr); 679*5d00a290SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Error checking: %d\n",Plapack_ierror);CHKERRQ(ierr); 680*5d00a290SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Algorithmic block size: %d\n",Plapack_nb_alg);CHKERRQ(ierr); 681*5d00a290SHong Zhang if (mat->factor){ 682*5d00a290SHong Zhang lu=(Mat_Plapack*)(mat->spptr); 68301b82886SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr); 684*5d00a290SHong Zhang } 685*5d00a290SHong Zhang #else 686*5d00a290SHong Zhang ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr); 68701b82886SBarry Smith #endif 6883a40ed3dSBarry Smith PetscFunctionReturn(0); 689fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_INFO) { 6903a40ed3dSBarry Smith PetscFunctionReturn(0); 6918965ea79SLois Curfman McInnes } 692f1af5d2fSBarry Smith } else if (isdraw) { 693b0a32e0cSBarry Smith PetscDraw draw; 694f1af5d2fSBarry Smith PetscTruth isnull; 695f1af5d2fSBarry Smith 696b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 697b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 698f1af5d2fSBarry Smith if (isnull) PetscFunctionReturn(0); 699f1af5d2fSBarry Smith } 70077ed5343SBarry Smith 7018965ea79SLois Curfman McInnes if (size == 1) { 70239ddd567SLois Curfman McInnes ierr = MatView(mdn->A,viewer);CHKERRQ(ierr); 7033a40ed3dSBarry Smith } else { 7048965ea79SLois Curfman McInnes /* assemble the entire matrix onto first processor. */ 7058965ea79SLois Curfman McInnes Mat A; 706d0f46423SBarry Smith PetscInt M = mat->rmap->N,N = mat->cmap->N,m,row,i,nz; 707ba8c8a56SBarry Smith PetscInt *cols; 708ba8c8a56SBarry Smith PetscScalar *vals; 7098965ea79SLois Curfman McInnes 7107adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr); 7118965ea79SLois Curfman McInnes if (!rank) { 712f69a0ea3SMatthew Knepley ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr); 7133a40ed3dSBarry Smith } else { 714f69a0ea3SMatthew Knepley ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr); 7158965ea79SLois Curfman McInnes } 7167adad957SLisandro Dalcin /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */ 717878740d9SKris Buschelman ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr); 718878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL); 71952e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr); 7208965ea79SLois Curfman McInnes 72139ddd567SLois Curfman McInnes /* Copy the matrix ... This isn't the most efficient means, 72239ddd567SLois Curfman McInnes but it's quick for now */ 72351022da4SBarry Smith A->insertmode = INSERT_VALUES; 724d0f46423SBarry Smith row = mat->rmap->rstart; m = mdn->A->rmap->n; 7258965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 726ba8c8a56SBarry Smith ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr); 727ba8c8a56SBarry Smith ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 728ba8c8a56SBarry Smith ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr); 72939ddd567SLois Curfman McInnes row++; 7308965ea79SLois Curfman McInnes } 7318965ea79SLois Curfman McInnes 7326d4a8577SBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 7336d4a8577SBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 734b0a32e0cSBarry Smith ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr); 735b9b97703SBarry Smith if (!rank) { 7366831982aSBarry Smith ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr); 7378965ea79SLois Curfman McInnes } 738b0a32e0cSBarry Smith ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr); 739b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 7408965ea79SLois Curfman McInnes ierr = MatDestroy(A);CHKERRQ(ierr); 7418965ea79SLois Curfman McInnes } 7423a40ed3dSBarry Smith PetscFunctionReturn(0); 7438965ea79SLois Curfman McInnes } 7448965ea79SLois Curfman McInnes 7454a2ae208SSatish Balay #undef __FUNCT__ 7464a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense" 747dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer) 7488965ea79SLois Curfman McInnes { 749dfbe8321SBarry Smith PetscErrorCode ierr; 75032077d6dSBarry Smith PetscTruth iascii,isbinary,isdraw,issocket; 7518965ea79SLois Curfman McInnes 752433994e6SBarry Smith PetscFunctionBegin; 7530f5bd95cSBarry Smith 75432077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 755fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 756b0a32e0cSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 757fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 7580f5bd95cSBarry Smith 75932077d6dSBarry Smith if (iascii || issocket || isdraw) { 760f1af5d2fSBarry Smith ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr); 7610f5bd95cSBarry Smith } else if (isbinary) { 7623a40ed3dSBarry Smith ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr); 7635cd90555SBarry Smith } else { 764958c9bccSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name); 7658965ea79SLois Curfman McInnes } 7663a40ed3dSBarry Smith PetscFunctionReturn(0); 7678965ea79SLois Curfman McInnes } 7688965ea79SLois Curfman McInnes 7694a2ae208SSatish Balay #undef __FUNCT__ 7704a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense" 771dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info) 7728965ea79SLois Curfman McInnes { 7733501a2bdSLois Curfman McInnes Mat_MPIDense *mat = (Mat_MPIDense*)A->data; 7743501a2bdSLois Curfman McInnes Mat mdn = mat->A; 775dfbe8321SBarry Smith PetscErrorCode ierr; 776329f5518SBarry Smith PetscReal isend[5],irecv[5]; 7778965ea79SLois Curfman McInnes 7783a40ed3dSBarry Smith PetscFunctionBegin; 7794e220ebcSLois Curfman McInnes info->block_size = 1.0; 7804e220ebcSLois Curfman McInnes ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr); 7814e220ebcSLois Curfman McInnes isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded; 7824e220ebcSLois Curfman McInnes isend[3] = info->memory; isend[4] = info->mallocs; 7838965ea79SLois Curfman McInnes if (flag == MAT_LOCAL) { 7844e220ebcSLois Curfman McInnes info->nz_used = isend[0]; 7854e220ebcSLois Curfman McInnes info->nz_allocated = isend[1]; 7864e220ebcSLois Curfman McInnes info->nz_unneeded = isend[2]; 7874e220ebcSLois Curfman McInnes info->memory = isend[3]; 7884e220ebcSLois Curfman McInnes info->mallocs = isend[4]; 7898965ea79SLois Curfman McInnes } else if (flag == MAT_GLOBAL_MAX) { 7907adad957SLisandro Dalcin ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr); 7914e220ebcSLois Curfman McInnes info->nz_used = irecv[0]; 7924e220ebcSLois Curfman McInnes info->nz_allocated = irecv[1]; 7934e220ebcSLois Curfman McInnes info->nz_unneeded = irecv[2]; 7944e220ebcSLois Curfman McInnes info->memory = irecv[3]; 7954e220ebcSLois Curfman McInnes info->mallocs = irecv[4]; 7968965ea79SLois Curfman McInnes } else if (flag == MAT_GLOBAL_SUM) { 7977adad957SLisandro Dalcin ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 7984e220ebcSLois Curfman McInnes info->nz_used = irecv[0]; 7994e220ebcSLois Curfman McInnes info->nz_allocated = irecv[1]; 8004e220ebcSLois Curfman McInnes info->nz_unneeded = irecv[2]; 8014e220ebcSLois Curfman McInnes info->memory = irecv[3]; 8024e220ebcSLois Curfman McInnes info->mallocs = irecv[4]; 8038965ea79SLois Curfman McInnes } 8044e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; /* no parallel LU/ILU/Cholesky */ 8054e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 8064e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 8073a40ed3dSBarry Smith PetscFunctionReturn(0); 8088965ea79SLois Curfman McInnes } 8098965ea79SLois Curfman McInnes 8104a2ae208SSatish Balay #undef __FUNCT__ 8114a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense" 8124e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg) 8138965ea79SLois Curfman McInnes { 81439ddd567SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 815dfbe8321SBarry Smith PetscErrorCode ierr; 8168965ea79SLois Curfman McInnes 8173a40ed3dSBarry Smith PetscFunctionBegin; 81812c028f9SKris Buschelman switch (op) { 819512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 82012c028f9SKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 82112c028f9SKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 8224e0d8c25SBarry Smith ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 82312c028f9SKris Buschelman break; 82412c028f9SKris Buschelman case MAT_ROW_ORIENTED: 8254e0d8c25SBarry Smith a->roworiented = flg; 8264e0d8c25SBarry Smith ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 82712c028f9SKris Buschelman break; 8284e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 82912c028f9SKris Buschelman case MAT_USE_HASH_TABLE: 830290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 83112c028f9SKris Buschelman break; 83212c028f9SKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 8334e0d8c25SBarry Smith a->donotstash = flg; 83412c028f9SKris Buschelman break; 83577e54ba9SKris Buschelman case MAT_SYMMETRIC: 83677e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 8379a4540c5SBarry Smith case MAT_HERMITIAN: 8389a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 839600fe468SBarry Smith case MAT_IGNORE_LOWER_TRIANGULAR: 840290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 84177e54ba9SKris Buschelman break; 84212c028f9SKris Buschelman default: 843600fe468SBarry Smith SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]); 8443a40ed3dSBarry Smith } 8453a40ed3dSBarry Smith PetscFunctionReturn(0); 8468965ea79SLois Curfman McInnes } 8478965ea79SLois Curfman McInnes 8488965ea79SLois Curfman McInnes 8494a2ae208SSatish Balay #undef __FUNCT__ 8504a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense" 851dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr) 8525b2fa520SLois Curfman McInnes { 8535b2fa520SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)A->data; 8545b2fa520SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)mdn->A->data; 85587828ca2SBarry Smith PetscScalar *l,*r,x,*v; 856dfbe8321SBarry Smith PetscErrorCode ierr; 857d0f46423SBarry Smith PetscInt i,j,s2a,s3a,s2,s3,m=mdn->A->rmap->n,n=mdn->A->cmap->n; 8585b2fa520SLois Curfman McInnes 8595b2fa520SLois Curfman McInnes PetscFunctionBegin; 86072d926a5SLois Curfman McInnes ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr); 8615b2fa520SLois Curfman McInnes if (ll) { 86272d926a5SLois Curfman McInnes ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr); 863175be7b4SMatthew Knepley if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2); 8641ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 8655b2fa520SLois Curfman McInnes for (i=0; i<m; i++) { 8665b2fa520SLois Curfman McInnes x = l[i]; 8675b2fa520SLois Curfman McInnes v = mat->v + i; 8685b2fa520SLois Curfman McInnes for (j=0; j<n; j++) { (*v) *= x; v+= m;} 8695b2fa520SLois Curfman McInnes } 8701ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 871efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 8725b2fa520SLois Curfman McInnes } 8735b2fa520SLois Curfman McInnes if (rr) { 874175be7b4SMatthew Knepley ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr); 875175be7b4SMatthew Knepley if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3); 876ca9f406cSSatish Balay ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 877ca9f406cSSatish Balay ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 8781ebc52fbSHong Zhang ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr); 8795b2fa520SLois Curfman McInnes for (i=0; i<n; i++) { 8805b2fa520SLois Curfman McInnes x = r[i]; 8815b2fa520SLois Curfman McInnes v = mat->v + i*m; 8825b2fa520SLois Curfman McInnes for (j=0; j<m; j++) { (*v++) *= x;} 8835b2fa520SLois Curfman McInnes } 8841ebc52fbSHong Zhang ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr); 885efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 8865b2fa520SLois Curfman McInnes } 8875b2fa520SLois Curfman McInnes PetscFunctionReturn(0); 8885b2fa520SLois Curfman McInnes } 8895b2fa520SLois Curfman McInnes 8904a2ae208SSatish Balay #undef __FUNCT__ 8914a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense" 892dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm) 893096963f5SLois Curfman McInnes { 8943501a2bdSLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)A->data; 8953501a2bdSLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)mdn->A->data; 896dfbe8321SBarry Smith PetscErrorCode ierr; 89713f74950SBarry Smith PetscInt i,j; 898329f5518SBarry Smith PetscReal sum = 0.0; 89987828ca2SBarry Smith PetscScalar *v = mat->v; 9003501a2bdSLois Curfman McInnes 9013a40ed3dSBarry Smith PetscFunctionBegin; 9023501a2bdSLois Curfman McInnes if (mdn->size == 1) { 903064f8208SBarry Smith ierr = MatNorm(mdn->A,type,nrm);CHKERRQ(ierr); 9043501a2bdSLois Curfman McInnes } else { 9053501a2bdSLois Curfman McInnes if (type == NORM_FROBENIUS) { 906d0f46423SBarry Smith for (i=0; i<mdn->A->cmap->n*mdn->A->rmap->n; i++) { 907aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 908329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 9093501a2bdSLois Curfman McInnes #else 9103501a2bdSLois Curfman McInnes sum += (*v)*(*v); v++; 9113501a2bdSLois Curfman McInnes #endif 9123501a2bdSLois Curfman McInnes } 9137adad957SLisandro Dalcin ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 914064f8208SBarry Smith *nrm = sqrt(*nrm); 915d0f46423SBarry Smith ierr = PetscLogFlops(2*mdn->A->cmap->n*mdn->A->rmap->n);CHKERRQ(ierr); 9163a40ed3dSBarry Smith } else if (type == NORM_1) { 917329f5518SBarry Smith PetscReal *tmp,*tmp2; 918d0f46423SBarry Smith ierr = PetscMalloc(2*A->cmap->N*sizeof(PetscReal),&tmp);CHKERRQ(ierr); 919d0f46423SBarry Smith tmp2 = tmp + A->cmap->N; 920d0f46423SBarry Smith ierr = PetscMemzero(tmp,2*A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr); 921064f8208SBarry Smith *nrm = 0.0; 9223501a2bdSLois Curfman McInnes v = mat->v; 923d0f46423SBarry Smith for (j=0; j<mdn->A->cmap->n; j++) { 924d0f46423SBarry Smith for (i=0; i<mdn->A->rmap->n; i++) { 92567e560aaSBarry Smith tmp[j] += PetscAbsScalar(*v); v++; 9263501a2bdSLois Curfman McInnes } 9273501a2bdSLois Curfman McInnes } 928d0f46423SBarry Smith ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 929d0f46423SBarry Smith for (j=0; j<A->cmap->N; j++) { 930064f8208SBarry Smith if (tmp2[j] > *nrm) *nrm = tmp2[j]; 9313501a2bdSLois Curfman McInnes } 932606d414cSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 933d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr); 9343a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { /* max row norm */ 935329f5518SBarry Smith PetscReal ntemp; 9363501a2bdSLois Curfman McInnes ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr); 9377adad957SLisandro Dalcin ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr); 9383a40ed3dSBarry Smith } else { 93929bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"No support for two norm"); 9403501a2bdSLois Curfman McInnes } 9413501a2bdSLois Curfman McInnes } 9423a40ed3dSBarry Smith PetscFunctionReturn(0); 9433501a2bdSLois Curfman McInnes } 9443501a2bdSLois Curfman McInnes 9454a2ae208SSatish Balay #undef __FUNCT__ 9464a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense" 947fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout) 9483501a2bdSLois Curfman McInnes { 9493501a2bdSLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 9503501a2bdSLois Curfman McInnes Mat_SeqDense *Aloc = (Mat_SeqDense*)a->A->data; 9513501a2bdSLois Curfman McInnes Mat B; 952d0f46423SBarry Smith PetscInt M = A->rmap->N,N = A->cmap->N,m,n,*rwork,rstart = A->rmap->rstart; 9536849ba73SBarry Smith PetscErrorCode ierr; 95413f74950SBarry Smith PetscInt j,i; 95587828ca2SBarry Smith PetscScalar *v; 9563501a2bdSLois Curfman McInnes 9573a40ed3dSBarry Smith PetscFunctionBegin; 958e9695a30SBarry Smith if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place"); 959fc4dec0aSBarry Smith if (reuse == MAT_INITIAL_MATRIX || A == *matout) { 9607adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 961d0f46423SBarry Smith ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,N,M);CHKERRQ(ierr); 9627adad957SLisandro Dalcin ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 963878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr); 964fc4dec0aSBarry Smith } else { 965fc4dec0aSBarry Smith B = *matout; 966fc4dec0aSBarry Smith } 9673501a2bdSLois Curfman McInnes 968d0f46423SBarry Smith m = a->A->rmap->n; n = a->A->cmap->n; v = Aloc->v; 9691acff37aSSatish Balay ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr); 9703501a2bdSLois Curfman McInnes for (i=0; i<m; i++) rwork[i] = rstart + i; 9711acff37aSSatish Balay for (j=0; j<n; j++) { 9723501a2bdSLois Curfman McInnes ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr); 9733501a2bdSLois Curfman McInnes v += m; 9743501a2bdSLois Curfman McInnes } 975606d414cSSatish Balay ierr = PetscFree(rwork);CHKERRQ(ierr); 9766d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 9776d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 978815cbec1SBarry Smith if (reuse == MAT_INITIAL_MATRIX || *matout != A) { 9793501a2bdSLois Curfman McInnes *matout = B; 9803501a2bdSLois Curfman McInnes } else { 981273d9f13SBarry Smith ierr = MatHeaderCopy(A,B);CHKERRQ(ierr); 9823501a2bdSLois Curfman McInnes } 9833a40ed3dSBarry Smith PetscFunctionReturn(0); 984096963f5SLois Curfman McInnes } 985096963f5SLois Curfman McInnes 986d9eff348SSatish Balay #include "petscblaslapack.h" 9874a2ae208SSatish Balay #undef __FUNCT__ 9884a2ae208SSatish Balay #define __FUNCT__ "MatScale_MPIDense" 989f4df32b1SMatthew Knepley PetscErrorCode MatScale_MPIDense(Mat inA,PetscScalar alpha) 99044cd7ae7SLois Curfman McInnes { 99144cd7ae7SLois Curfman McInnes Mat_MPIDense *A = (Mat_MPIDense*)inA->data; 99244cd7ae7SLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*)A->A->data; 993f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 994efee365bSSatish Balay PetscErrorCode ierr; 995d0f46423SBarry Smith PetscBLASInt one = 1,nz = PetscBLASIntCast(inA->rmap->n*inA->cmap->N); 99644cd7ae7SLois Curfman McInnes 9973a40ed3dSBarry Smith PetscFunctionBegin; 998f4df32b1SMatthew Knepley BLASscal_(&nz,&oalpha,a->v,&one); 999efee365bSSatish Balay ierr = PetscLogFlops(nz);CHKERRQ(ierr); 10003a40ed3dSBarry Smith PetscFunctionReturn(0); 100144cd7ae7SLois Curfman McInnes } 100244cd7ae7SLois Curfman McInnes 10036849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *); 10048965ea79SLois Curfman McInnes 10054a2ae208SSatish Balay #undef __FUNCT__ 10064a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_MPIDense" 1007dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_MPIDense(Mat A) 1008273d9f13SBarry Smith { 1009dfbe8321SBarry Smith PetscErrorCode ierr; 1010273d9f13SBarry Smith 1011273d9f13SBarry Smith PetscFunctionBegin; 1012273d9f13SBarry Smith ierr = MatMPIDenseSetPreallocation(A,0);CHKERRQ(ierr); 1013273d9f13SBarry Smith PetscFunctionReturn(0); 1014273d9f13SBarry Smith } 1015273d9f13SBarry Smith 101601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 1017eae6fb2eSBarry Smith 1018eae6fb2eSBarry Smith #undef __FUNCT__ 1019eae6fb2eSBarry Smith #define __FUNCT__ "MatMPIDenseCopyToPlapack" 1020eae6fb2eSBarry Smith PetscErrorCode MatMPIDenseCopyToPlapack(Mat A,Mat F) 1021eae6fb2eSBarry Smith { 1022eae6fb2eSBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 1023eae6fb2eSBarry Smith PetscErrorCode ierr; 1024d0f46423SBarry Smith PetscInt M=A->cmap->N,m=A->rmap->n,rstart; 1025eae6fb2eSBarry Smith PetscScalar *array; 1026eae6fb2eSBarry Smith PetscReal one = 1.0; 1027eae6fb2eSBarry Smith 1028eae6fb2eSBarry Smith PetscFunctionBegin; 10292fbe02b9SBarry Smith /* Copy A into F->lu->A */ 1030eae6fb2eSBarry Smith ierr = PLA_Obj_set_to_zero(lu->A);CHKERRQ(ierr); 1031eae6fb2eSBarry Smith ierr = PLA_API_begin();CHKERRQ(ierr); 1032eae6fb2eSBarry Smith ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr); 103379b0a62dSBarry Smith ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr); 1034eae6fb2eSBarry Smith ierr = MatGetArray(A,&array);CHKERRQ(ierr); 1035eae6fb2eSBarry Smith ierr = PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);CHKERRQ(ierr); 1036eae6fb2eSBarry Smith ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 1037eae6fb2eSBarry Smith ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr); 1038eae6fb2eSBarry Smith ierr = PLA_API_end();CHKERRQ(ierr); 1039eae6fb2eSBarry Smith lu->rstart = rstart; 1040eae6fb2eSBarry Smith PetscFunctionReturn(0); 1041eae6fb2eSBarry Smith } 1042eae6fb2eSBarry Smith 104301b82886SBarry Smith #undef __FUNCT__ 10442fbe02b9SBarry Smith #define __FUNCT__ "MatMPIDenseCopyFromPlapack" 10452fbe02b9SBarry Smith PetscErrorCode MatMPIDenseCopyFromPlapack(Mat F,Mat A) 10462fbe02b9SBarry Smith { 10472fbe02b9SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 10482fbe02b9SBarry Smith PetscErrorCode ierr; 1049d0f46423SBarry Smith PetscInt M=A->cmap->N,m=A->rmap->n,rstart; 10502fbe02b9SBarry Smith PetscScalar *array; 10512fbe02b9SBarry Smith PetscReal one = 1.0; 10522fbe02b9SBarry Smith 10532fbe02b9SBarry Smith PetscFunctionBegin; 10542fbe02b9SBarry Smith /* Copy F into A->lu->A */ 105579b0a62dSBarry Smith ierr = MatZeroEntries(A);CHKERRQ(ierr); 10562fbe02b9SBarry Smith ierr = PLA_API_begin();CHKERRQ(ierr); 10572fbe02b9SBarry Smith ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr); 10582fbe02b9SBarry Smith ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr); 10592fbe02b9SBarry Smith ierr = MatGetArray(A,&array);CHKERRQ(ierr); 10602fbe02b9SBarry Smith ierr = PLA_API_axpy_global_to_matrix(m,M, &one,lu->A,rstart,0,(void *)array,m);CHKERRQ(ierr); 10612fbe02b9SBarry Smith ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 10622fbe02b9SBarry Smith ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr); 10632fbe02b9SBarry Smith ierr = PLA_API_end();CHKERRQ(ierr); 10642fbe02b9SBarry Smith lu->rstart = rstart; 10652fbe02b9SBarry Smith PetscFunctionReturn(0); 10662fbe02b9SBarry Smith } 10672fbe02b9SBarry Smith 10682fbe02b9SBarry Smith #undef __FUNCT__ 10692fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultNumeric_MPIDense_MPIDense" 10702fbe02b9SBarry Smith PetscErrorCode MatMatMultNumeric_MPIDense_MPIDense(Mat A,Mat B,Mat C) 10712fbe02b9SBarry Smith { 10722fbe02b9SBarry Smith PetscErrorCode ierr; 10732fbe02b9SBarry Smith Mat_Plapack *luA = (Mat_Plapack*)A->spptr; 10742fbe02b9SBarry Smith Mat_Plapack *luB = (Mat_Plapack*)B->spptr; 10752fbe02b9SBarry Smith Mat_Plapack *luC = (Mat_Plapack*)C->spptr; 10762fbe02b9SBarry Smith PLA_Obj alpha = NULL,beta = NULL; 10772fbe02b9SBarry Smith 10782fbe02b9SBarry Smith PetscFunctionBegin; 10792fbe02b9SBarry Smith ierr = MatMPIDenseCopyToPlapack(A,A);CHKERRQ(ierr); 10802fbe02b9SBarry Smith ierr = MatMPIDenseCopyToPlapack(B,B);CHKERRQ(ierr); 10812fbe02b9SBarry Smith 10826e6f9017SBarry Smith /* 1083cb2480eaSBarry Smith ierr = PLA_Global_show("A = ",luA->A,"%g ","");CHKERRQ(ierr); 1084cb2480eaSBarry Smith ierr = PLA_Global_show("B = ",luB->A,"%g ","");CHKERRQ(ierr); 10856e6f9017SBarry Smith */ 1086cb2480eaSBarry Smith 10872fbe02b9SBarry Smith /* do the multiply in PLA */ 108879b0a62dSBarry Smith ierr = PLA_Create_constants_conf_to(luA->A,NULL,NULL,&alpha);CHKERRQ(ierr); 108979b0a62dSBarry Smith ierr = PLA_Create_constants_conf_to(luC->A,NULL,&beta,NULL);CHKERRQ(ierr); 109079b0a62dSBarry Smith CHKMEMQ; 10912fbe02b9SBarry Smith 1092f0e3846dSSatish Balay ierr = PLA_Gemm(PLA_NO_TRANSPOSE,PLA_NO_TRANSPOSE,alpha,luA->A,luB->A,beta,luC->A); /* CHKERRQ(ierr); */ 109379b0a62dSBarry Smith CHKMEMQ; 10942fbe02b9SBarry Smith ierr = PLA_Obj_free(&alpha);CHKERRQ(ierr); 10952fbe02b9SBarry Smith ierr = PLA_Obj_free(&beta);CHKERRQ(ierr); 10962fbe02b9SBarry Smith 10976e6f9017SBarry Smith /* 1098cb2480eaSBarry Smith ierr = PLA_Global_show("C = ",luC->A,"%g ","");CHKERRQ(ierr); 10996e6f9017SBarry Smith */ 11002fbe02b9SBarry Smith ierr = MatMPIDenseCopyFromPlapack(C,C);CHKERRQ(ierr); 11012fbe02b9SBarry Smith PetscFunctionReturn(0); 11022fbe02b9SBarry Smith } 11032fbe02b9SBarry Smith 11042fbe02b9SBarry Smith #undef __FUNCT__ 11052fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultSymbolic_MPIDense_MPIDense" 11062fbe02b9SBarry Smith PetscErrorCode MatMatMultSymbolic_MPIDense_MPIDense(Mat A,Mat B,PetscReal fill,Mat *C) 11072fbe02b9SBarry Smith { 11082fbe02b9SBarry Smith PetscErrorCode ierr; 1109d0f46423SBarry Smith PetscInt m=A->rmap->n,n=B->cmap->n; 11102fbe02b9SBarry Smith Mat Cmat; 11112fbe02b9SBarry Smith 11122fbe02b9SBarry Smith PetscFunctionBegin; 1113d0f46423SBarry 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); 11146e6f9017SBarry Smith SETERRQ(PETSC_ERR_LIB,"Due to aparent bugs in PLAPACK,this is not currently supported"); 11152fbe02b9SBarry Smith ierr = MatCreate(((PetscObject)B)->comm,&Cmat);CHKERRQ(ierr); 1116d0f46423SBarry Smith ierr = MatSetSizes(Cmat,m,n,A->rmap->N,B->cmap->N);CHKERRQ(ierr); 11172fbe02b9SBarry Smith ierr = MatSetType(Cmat,MATMPIDENSE);CHKERRQ(ierr); 11182fbe02b9SBarry Smith ierr = MatAssemblyBegin(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11192fbe02b9SBarry Smith ierr = MatAssemblyEnd(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 112085bd4cefSHong Zhang 11212fbe02b9SBarry Smith *C = Cmat; 11222fbe02b9SBarry Smith PetscFunctionReturn(0); 11232fbe02b9SBarry Smith } 11242fbe02b9SBarry Smith 11252fbe02b9SBarry Smith #undef __FUNCT__ 11262fbe02b9SBarry Smith #define __FUNCT__ "MatMatMult_MPIDense_MPIDense" 11272fbe02b9SBarry Smith PetscErrorCode MatMatMult_MPIDense_MPIDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 11282fbe02b9SBarry Smith { 11292fbe02b9SBarry Smith PetscErrorCode ierr; 11302fbe02b9SBarry Smith 11312fbe02b9SBarry Smith PetscFunctionBegin; 11322fbe02b9SBarry Smith if (scall == MAT_INITIAL_MATRIX){ 11332fbe02b9SBarry Smith ierr = MatMatMultSymbolic_MPIDense_MPIDense(A,B,fill,C);CHKERRQ(ierr); 11342fbe02b9SBarry Smith } 11352fbe02b9SBarry Smith ierr = MatMatMultNumeric_MPIDense_MPIDense(A,B,*C);CHKERRQ(ierr); 11362fbe02b9SBarry Smith PetscFunctionReturn(0); 11372fbe02b9SBarry Smith } 11382fbe02b9SBarry Smith 1139ab235cb6SHong Zhang #undef __FUNCT__ 11406c995c7dSHong Zhang #define __FUNCT__ "MatSolve_MPIDense" 11416c995c7dSHong Zhang PetscErrorCode MatSolve_MPIDense(Mat A,Vec b,Vec x) 11426c995c7dSHong Zhang { 11436c995c7dSHong Zhang MPI_Comm comm = ((PetscObject)A)->comm; 11446c995c7dSHong Zhang Mat_Plapack *lu = (Mat_Plapack*)A->spptr; 11456c995c7dSHong Zhang PetscErrorCode ierr; 1146d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,i,j,*idx_pla,*idx_petsc,loc_m,loc_stride; 11476c995c7dSHong Zhang PetscScalar *array; 11486c995c7dSHong Zhang PetscReal one = 1.0; 11496c995c7dSHong Zhang PetscMPIInt size,rank,r_rank,r_nproc,c_rank,c_nproc;; 11506c995c7dSHong Zhang PLA_Obj v_pla = NULL; 11516c995c7dSHong Zhang PetscScalar *loc_buf; 11526c995c7dSHong Zhang Vec loc_x; 11536c995c7dSHong Zhang 11546c995c7dSHong Zhang PetscFunctionBegin; 11556c995c7dSHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 11566c995c7dSHong Zhang ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 11576c995c7dSHong Zhang 11586c995c7dSHong Zhang /* Create PLAPACK vector objects, then copy b into PLAPACK b */ 11596c995c7dSHong Zhang PLA_Mvector_create(lu->datatype,M,1,lu->templ,PLA_ALIGN_FIRST,&v_pla); 11606c995c7dSHong Zhang PLA_Obj_set_to_zero(v_pla); 11616c995c7dSHong Zhang 11626c995c7dSHong Zhang /* Copy b into rhs_pla */ 11636c995c7dSHong Zhang PLA_API_begin(); 11646c995c7dSHong Zhang PLA_Obj_API_open(v_pla); 11656c995c7dSHong Zhang ierr = VecGetArray(b,&array);CHKERRQ(ierr); 11666c995c7dSHong Zhang PLA_API_axpy_vector_to_global(m,&one,(void *)array,1,v_pla,lu->rstart); 11676c995c7dSHong Zhang ierr = VecRestoreArray(b,&array);CHKERRQ(ierr); 11686c995c7dSHong Zhang PLA_Obj_API_close(v_pla); 11696c995c7dSHong Zhang PLA_API_end(); 11706c995c7dSHong Zhang 11716c995c7dSHong Zhang if (A->factor == MAT_FACTOR_LU){ 11726c995c7dSHong Zhang /* Apply the permutations to the right hand sides */ 11736c995c7dSHong Zhang PLA_Apply_pivots_to_rows (v_pla,lu->pivots); 11746c995c7dSHong Zhang 11756c995c7dSHong Zhang /* Solve L y = b, overwriting b with y */ 11766c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_UNIT_DIAG,lu->A,v_pla ); 11776c995c7dSHong Zhang 11786c995c7dSHong Zhang /* Solve U x = y (=b), overwriting b with x */ 11796c995c7dSHong Zhang PLA_Trsv( PLA_UPPER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla ); 11806c995c7dSHong Zhang } else { /*MAT_FACTOR_CHOLESKY */ 11816c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla); 11826c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,(lu->datatype == MPI_DOUBLE ? PLA_TRANSPOSE : PLA_CONJUGATE_TRANSPOSE), 11836c995c7dSHong Zhang PLA_NONUNIT_DIAG,lu->A,v_pla); 11846c995c7dSHong Zhang } 11856c995c7dSHong Zhang 11866c995c7dSHong Zhang /* Copy PLAPACK x into Petsc vector x */ 11876c995c7dSHong Zhang PLA_Obj_local_length(v_pla, &loc_m); 11886c995c7dSHong Zhang PLA_Obj_local_buffer(v_pla, (void**)&loc_buf); 11896c995c7dSHong Zhang PLA_Obj_local_stride(v_pla, &loc_stride); 11906c995c7dSHong Zhang /* 11916c995c7dSHong 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); 11926c995c7dSHong Zhang */ 11936c995c7dSHong Zhang ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,loc_m*loc_stride,loc_buf,&loc_x);CHKERRQ(ierr); 11946c995c7dSHong Zhang if (!lu->pla_solved){ 11956c995c7dSHong Zhang 1196*5d00a290SHong Zhang PLA_Temp_comm_row_info(lu->templ,&Plapack_comm_2d,&r_rank,&r_nproc); 1197*5d00a290SHong Zhang PLA_Temp_comm_col_info(lu->templ,&Plapack_comm_2d,&c_rank,&c_nproc); 11986c995c7dSHong Zhang 11996c995c7dSHong Zhang /* Create IS and cts for VecScatterring */ 12006c995c7dSHong Zhang PLA_Obj_local_length(v_pla, &loc_m); 12016c995c7dSHong Zhang PLA_Obj_local_stride(v_pla, &loc_stride); 12026c995c7dSHong Zhang ierr = PetscMalloc((2*loc_m+1)*sizeof(PetscInt),&idx_pla);CHKERRQ(ierr); 12036c995c7dSHong Zhang idx_petsc = idx_pla + loc_m; 12046c995c7dSHong Zhang 12056c995c7dSHong Zhang rstart = (r_rank*c_nproc+c_rank)*lu->nb; 12066c995c7dSHong Zhang for (i=0; i<loc_m; i+=lu->nb){ 12076c995c7dSHong Zhang j = 0; 12086c995c7dSHong Zhang while (j < lu->nb && i+j < loc_m){ 12096c995c7dSHong Zhang idx_petsc[i+j] = rstart + j; j++; 12106c995c7dSHong Zhang } 12116c995c7dSHong Zhang rstart += size*lu->nb; 12126c995c7dSHong Zhang } 12136c995c7dSHong Zhang 12146c995c7dSHong Zhang for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride; 12156c995c7dSHong Zhang 12166c995c7dSHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr); 12176c995c7dSHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr); 12186c995c7dSHong Zhang ierr = PetscFree(idx_pla);CHKERRQ(ierr); 12196c995c7dSHong Zhang ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr); 12206c995c7dSHong Zhang } 12216c995c7dSHong Zhang ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12226c995c7dSHong Zhang ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12236c995c7dSHong Zhang 12246c995c7dSHong Zhang /* Free data */ 12256c995c7dSHong Zhang ierr = VecDestroy(loc_x);CHKERRQ(ierr); 12266c995c7dSHong Zhang PLA_Obj_free(&v_pla); 12276c995c7dSHong Zhang 12286c995c7dSHong Zhang lu->pla_solved = PETSC_TRUE; 12296c995c7dSHong Zhang PetscFunctionReturn(0); 12306c995c7dSHong Zhang } 12316c995c7dSHong Zhang 12326c995c7dSHong Zhang #undef __FUNCT__ 12336c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense" 12340481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info) 12356c995c7dSHong Zhang { 1236719d5645SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 12376c995c7dSHong Zhang PetscErrorCode ierr; 1238d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,rend; 12396c995c7dSHong Zhang PetscInt info_pla=0; 12406c995c7dSHong Zhang PetscScalar *array,one = 1.0; 12416c995c7dSHong Zhang 12426c995c7dSHong Zhang PetscFunctionBegin; 12436c995c7dSHong Zhang if (lu->mstruct == SAME_NONZERO_PATTERN){ 12446c995c7dSHong Zhang PLA_Obj_free(&lu->A); 12456c995c7dSHong Zhang PLA_Obj_free (&lu->pivots); 12466c995c7dSHong Zhang } 12476c995c7dSHong Zhang /* Create PLAPACK matrix object */ 12486c995c7dSHong Zhang lu->A = NULL; lu->pivots = NULL; 12496c995c7dSHong Zhang PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A); 12506c995c7dSHong Zhang PLA_Obj_set_to_zero(lu->A); 12516c995c7dSHong Zhang PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots); 12526c995c7dSHong Zhang 12536c995c7dSHong Zhang /* Copy A into lu->A */ 12546c995c7dSHong Zhang PLA_API_begin(); 12556c995c7dSHong Zhang PLA_Obj_API_open(lu->A); 12566c995c7dSHong Zhang ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 12576c995c7dSHong Zhang ierr = MatGetArray(A,&array);CHKERRQ(ierr); 12586c995c7dSHong Zhang PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0); 12596c995c7dSHong Zhang ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 12606c995c7dSHong Zhang PLA_Obj_API_close(lu->A); 12616c995c7dSHong Zhang PLA_API_end(); 12626c995c7dSHong Zhang 12636c995c7dSHong Zhang /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */ 12646c995c7dSHong Zhang info_pla = PLA_LU(lu->A,lu->pivots); 12656c995c7dSHong Zhang if (info_pla != 0) 12666c995c7dSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla); 12676c995c7dSHong Zhang 12686c995c7dSHong Zhang lu->rstart = rstart; 12696c995c7dSHong Zhang lu->mstruct = SAME_NONZERO_PATTERN; 1270719d5645SBarry Smith F->ops->solve = MatSolve_MPIDense; 1271719d5645SBarry Smith F->assembled = PETSC_TRUE; /* required by -ksp_view */ 12726c995c7dSHong Zhang PetscFunctionReturn(0); 12736c995c7dSHong Zhang } 12746c995c7dSHong Zhang 12756c995c7dSHong Zhang #undef __FUNCT__ 12766c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense" 12770481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info) 12786c995c7dSHong Zhang { 1279719d5645SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)F->spptr; 12806c995c7dSHong Zhang PetscErrorCode ierr; 1281d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,rend; 12826c995c7dSHong Zhang PetscInt info_pla=0; 12836c995c7dSHong Zhang PetscScalar *array,one = 1.0; 12846c995c7dSHong Zhang 12856c995c7dSHong Zhang PetscFunctionBegin; 12866c995c7dSHong Zhang if (lu->mstruct == SAME_NONZERO_PATTERN){ 12876c995c7dSHong Zhang PLA_Obj_free(&lu->A); 12886c995c7dSHong Zhang } 12896c995c7dSHong Zhang /* Create PLAPACK matrix object */ 12906c995c7dSHong Zhang lu->A = NULL; 12916c995c7dSHong Zhang lu->pivots = NULL; 12926c995c7dSHong Zhang PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A); 12936c995c7dSHong Zhang 12946c995c7dSHong Zhang /* Copy A into lu->A */ 12956c995c7dSHong Zhang PLA_API_begin(); 12966c995c7dSHong Zhang PLA_Obj_API_open(lu->A); 12976c995c7dSHong Zhang ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 12986c995c7dSHong Zhang ierr = MatGetArray(A,&array);CHKERRQ(ierr); 12996c995c7dSHong Zhang PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0); 13006c995c7dSHong Zhang ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 13016c995c7dSHong Zhang PLA_Obj_API_close(lu->A); 13026c995c7dSHong Zhang PLA_API_end(); 13036c995c7dSHong Zhang 13046c995c7dSHong Zhang /* Factor P A -> Chol */ 13056c995c7dSHong Zhang info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A); 13066c995c7dSHong Zhang if (info_pla != 0) 13076c995c7dSHong Zhang SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla); 13086c995c7dSHong Zhang 13096c995c7dSHong Zhang lu->rstart = rstart; 13106c995c7dSHong Zhang lu->mstruct = SAME_NONZERO_PATTERN; 1311719d5645SBarry Smith F->ops->solve = MatSolve_MPIDense; 1312719d5645SBarry Smith F->assembled = PETSC_TRUE; /* required by -ksp_view */ 13136c995c7dSHong Zhang PetscFunctionReturn(0); 13146c995c7dSHong Zhang } 13156c995c7dSHong Zhang 1316b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */ 13172fbe02b9SBarry Smith #undef __FUNCT__ 13186c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense" 13190481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,const MatFactorInfo *info) 132001b82886SBarry Smith { 132101b82886SBarry Smith PetscErrorCode ierr; 1322b24902e0SBarry Smith PetscTruth issymmetric,set; 132301b82886SBarry Smith 132401b82886SBarry Smith PetscFunctionBegin; 1325b24902e0SBarry Smith ierr = MatIsSymmetricKnown(A,&set,&issymmetric); CHKERRQ(ierr); 1326b24902e0SBarry Smith if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()"); 1327719d5645SBarry Smith F->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_MPIDense; 1328b24902e0SBarry Smith PetscFunctionReturn(0); 132901b82886SBarry Smith } 133001b82886SBarry Smith 1331b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */ 1332b24902e0SBarry Smith #undef __FUNCT__ 13336c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense" 13340481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 1335b24902e0SBarry Smith { 1336b24902e0SBarry Smith PetscErrorCode ierr; 1337d0f46423SBarry Smith PetscInt M = A->rmap->N; 1338b24902e0SBarry Smith Mat_Plapack *lu; 133901b82886SBarry Smith 1340b24902e0SBarry Smith PetscFunctionBegin; 1341719d5645SBarry Smith lu = (Mat_Plapack*)F->spptr; 1342b24902e0SBarry Smith ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr); 1343719d5645SBarry Smith F->ops->lufactornumeric = MatLUFactorNumeric_MPIDense; 134401b82886SBarry Smith PetscFunctionReturn(0); 134501b82886SBarry Smith } 134601b82886SBarry Smith 134701b82886SBarry Smith #undef __FUNCT__ 134885bd4cefSHong Zhang #define __FUNCT__ "MatGetFactor_mpidense_petsc" 134985bd4cefSHong Zhang PetscErrorCode MatGetFactor_mpidense_petsc(Mat A,MatFactorType ftype,Mat *F) 135001b82886SBarry Smith { 135101b82886SBarry Smith PetscErrorCode ierr; 13526c995c7dSHong Zhang Mat_Plapack *lu; 13536c995c7dSHong Zhang PetscMPIInt size; 135485bd4cefSHong Zhang PetscInt M=A->rmap->N; 135501b82886SBarry Smith 135601b82886SBarry Smith PetscFunctionBegin; 135701b82886SBarry Smith /* Create the factorization matrix */ 1358eae6fb2eSBarry Smith ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr); 1359d0f46423SBarry Smith ierr = MatSetSizes(*F,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1360eae6fb2eSBarry Smith ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr); 136185bd4cefSHong Zhang ierr = PetscNewLog(*F,Mat_Plapack,&lu);CHKERRQ(ierr); 136285bd4cefSHong Zhang (*F)->spptr = (void*)lu; 136301b82886SBarry Smith 1364b24902e0SBarry Smith /* Set default Plapack parameters */ 13656c995c7dSHong Zhang ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr); 1366b24902e0SBarry Smith lu->nb = M/size; 1367b24902e0SBarry Smith if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */ 136801b82886SBarry Smith 1369b24902e0SBarry Smith /* Set runtime options */ 1370b24902e0SBarry Smith ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr); 1371b24902e0SBarry Smith ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr); 1372b24902e0SBarry Smith PetscOptionsEnd(); 137301b82886SBarry Smith 1374b24902e0SBarry Smith /* Create object distribution template */ 1375b24902e0SBarry Smith lu->templ = NULL; 1376b24902e0SBarry Smith ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr); 1377b24902e0SBarry Smith 1378b24902e0SBarry Smith /* Set the datatype */ 1379b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX) 1380b24902e0SBarry Smith lu->datatype = MPI_DOUBLE_COMPLEX; 1381b24902e0SBarry Smith #else 1382b24902e0SBarry Smith lu->datatype = MPI_DOUBLE; 1383b24902e0SBarry Smith #endif 1384b24902e0SBarry Smith 1385d0f46423SBarry Smith ierr = PLA_Matrix_create(lu->datatype,M,A->cmap->N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr); 1386b24902e0SBarry Smith 1387b24902e0SBarry Smith 1388b24902e0SBarry Smith lu->pla_solved = PETSC_FALSE; /* MatSolve_Plapack() is called yet */ 1389*5d00a290SHong Zhang lu->mstruct = DIFFERENT_NONZERO_PATTERN; 1390b24902e0SBarry Smith 1391b24902e0SBarry Smith if (ftype == MAT_FACTOR_LU) { 1392b24902e0SBarry Smith (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense; 1393b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY) { 13946c995c7dSHong Zhang (*F)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense; 1395b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices"); 1396719d5645SBarry Smith (*F)->factor = ftype; 139701b82886SBarry Smith PetscFunctionReturn(0); 139801b82886SBarry Smith } 139901b82886SBarry Smith #endif 140001b82886SBarry Smith 14018965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/ 140209dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense, 140309dc0095SBarry Smith MatGetRow_MPIDense, 140409dc0095SBarry Smith MatRestoreRow_MPIDense, 140509dc0095SBarry Smith MatMult_MPIDense, 140697304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense, 14077c922b88SBarry Smith MatMultTranspose_MPIDense, 14087c922b88SBarry Smith MatMultTransposeAdd_MPIDense, 14098965ea79SLois Curfman McInnes 0, 141009dc0095SBarry Smith 0, 141109dc0095SBarry Smith 0, 141297304618SKris Buschelman /*10*/ 0, 141309dc0095SBarry Smith 0, 141409dc0095SBarry Smith 0, 141509dc0095SBarry Smith 0, 141609dc0095SBarry Smith MatTranspose_MPIDense, 141797304618SKris Buschelman /*15*/ MatGetInfo_MPIDense, 14186e4ee0c6SHong Zhang MatEqual_MPIDense, 141909dc0095SBarry Smith MatGetDiagonal_MPIDense, 14205b2fa520SLois Curfman McInnes MatDiagonalScale_MPIDense, 142109dc0095SBarry Smith MatNorm_MPIDense, 142297304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense, 142309dc0095SBarry Smith MatAssemblyEnd_MPIDense, 142409dc0095SBarry Smith 0, 142509dc0095SBarry Smith MatSetOption_MPIDense, 142609dc0095SBarry Smith MatZeroEntries_MPIDense, 142797304618SKris Buschelman /*25*/ MatZeroRows_MPIDense, 1428919b68f7SBarry Smith 0, 142901b82886SBarry Smith 0, 143001b82886SBarry Smith 0, 143101b82886SBarry Smith 0, 143297304618SKris Buschelman /*30*/ MatSetUpPreallocation_MPIDense, 1433273d9f13SBarry Smith 0, 143409dc0095SBarry Smith 0, 143509dc0095SBarry Smith MatGetArray_MPIDense, 143609dc0095SBarry Smith MatRestoreArray_MPIDense, 143797304618SKris Buschelman /*35*/ MatDuplicate_MPIDense, 143809dc0095SBarry Smith 0, 143909dc0095SBarry Smith 0, 144009dc0095SBarry Smith 0, 144109dc0095SBarry Smith 0, 144297304618SKris Buschelman /*40*/ 0, 14432ce60cd0SSatish Balay MatGetSubMatrices_MPIDense, 144409dc0095SBarry Smith 0, 144509dc0095SBarry Smith MatGetValues_MPIDense, 144609dc0095SBarry Smith 0, 144797304618SKris Buschelman /*45*/ 0, 144809dc0095SBarry Smith MatScale_MPIDense, 144909dc0095SBarry Smith 0, 145009dc0095SBarry Smith 0, 145109dc0095SBarry Smith 0, 1452521d7252SBarry Smith /*50*/ 0, 145309dc0095SBarry Smith 0, 145409dc0095SBarry Smith 0, 145509dc0095SBarry Smith 0, 145609dc0095SBarry Smith 0, 145797304618SKris Buschelman /*55*/ 0, 145809dc0095SBarry Smith 0, 145909dc0095SBarry Smith 0, 146009dc0095SBarry Smith 0, 146109dc0095SBarry Smith 0, 146297304618SKris Buschelman /*60*/ MatGetSubMatrix_MPIDense, 1463b9b97703SBarry Smith MatDestroy_MPIDense, 1464b9b97703SBarry Smith MatView_MPIDense, 1465357abbc8SBarry Smith 0, 146697304618SKris Buschelman 0, 146797304618SKris Buschelman /*65*/ 0, 146897304618SKris Buschelman 0, 146997304618SKris Buschelman 0, 147097304618SKris Buschelman 0, 147197304618SKris Buschelman 0, 147297304618SKris Buschelman /*70*/ 0, 147397304618SKris Buschelman 0, 147497304618SKris Buschelman 0, 147597304618SKris Buschelman 0, 147697304618SKris Buschelman 0, 147797304618SKris Buschelman /*75*/ 0, 147897304618SKris Buschelman 0, 147997304618SKris Buschelman 0, 148097304618SKris Buschelman 0, 148197304618SKris Buschelman 0, 148297304618SKris Buschelman /*80*/ 0, 148397304618SKris Buschelman 0, 148497304618SKris Buschelman 0, 148597304618SKris Buschelman 0, 1486865e5f61SKris Buschelman /*84*/ MatLoad_MPIDense, 1487865e5f61SKris Buschelman 0, 1488865e5f61SKris Buschelman 0, 1489865e5f61SKris Buschelman 0, 1490865e5f61SKris Buschelman 0, 1491865e5f61SKris Buschelman 0, 14922fbe02b9SBarry Smith /*90*/ 14932fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 14942fbe02b9SBarry Smith MatMatMult_MPIDense_MPIDense, 14954ae313f4SHong Zhang MatMatMultSymbolic_MPIDense_MPIDense, 14962fbe02b9SBarry Smith MatMatMultNumeric_MPIDense_MPIDense, 14972fbe02b9SBarry Smith #else 1498865e5f61SKris Buschelman 0, 1499865e5f61SKris Buschelman 0, 1500865e5f61SKris Buschelman 0, 15012fbe02b9SBarry Smith #endif 15022fbe02b9SBarry Smith 0, 1503865e5f61SKris Buschelman /*95*/ 0, 1504865e5f61SKris Buschelman 0, 1505865e5f61SKris Buschelman 0, 1506865e5f61SKris Buschelman 0}; 15078965ea79SLois Curfman McInnes 1508273d9f13SBarry Smith EXTERN_C_BEGIN 15094a2ae208SSatish Balay #undef __FUNCT__ 1510a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense" 1511be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data) 1512a23d5eceSKris Buschelman { 1513a23d5eceSKris Buschelman Mat_MPIDense *a; 1514dfbe8321SBarry Smith PetscErrorCode ierr; 1515a23d5eceSKris Buschelman 1516a23d5eceSKris Buschelman PetscFunctionBegin; 1517a23d5eceSKris Buschelman mat->preallocated = PETSC_TRUE; 1518a23d5eceSKris Buschelman /* Note: For now, when data is specified above, this assumes the user correctly 1519a23d5eceSKris Buschelman allocates the local dense storage space. We should add error checking. */ 1520a23d5eceSKris Buschelman 1521a23d5eceSKris Buschelman a = (Mat_MPIDense*)mat->data; 1522f69a0ea3SMatthew Knepley ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr); 1523d0f46423SBarry Smith ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr); 15245c5985e7SKris Buschelman ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr); 15255c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr); 152652e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr); 1527a23d5eceSKris Buschelman PetscFunctionReturn(0); 1528a23d5eceSKris Buschelman } 1529a23d5eceSKris Buschelman EXTERN_C_END 1530a23d5eceSKris Buschelman 15310bad9183SKris Buschelman /*MC 1532fafad747SKris Buschelman MATMPIDENSE - MATMPIDENSE = "mpidense" - A matrix type to be used for distributed dense matrices. 15330bad9183SKris Buschelman 15340bad9183SKris Buschelman Options Database Keys: 15350bad9183SKris Buschelman . -mat_type mpidense - sets the matrix type to "mpidense" during a call to MatSetFromOptions() 15360bad9183SKris Buschelman 15370bad9183SKris Buschelman Level: beginner 15380bad9183SKris Buschelman 15397878bbefSBarry Smith MATMPIDENSE matrices may use direct solvers (LU, Cholesky, and QR) 15407878bbefSBarry Smith for parallel dense matrices via the external package PLAPACK, if PLAPACK is installed 15417878bbefSBarry Smith (run config/configure.py with the option --download-plapack) 15427878bbefSBarry Smith 15437878bbefSBarry Smith 15447878bbefSBarry Smith Options Database Keys: 15457878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition 15467878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition 15477878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector 15487878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size 15497878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag 15507878bbefSBarry Smith 15517878bbefSBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE 15520bad9183SKris Buschelman M*/ 15530bad9183SKris Buschelman 1554a23d5eceSKris Buschelman EXTERN_C_BEGIN 1555a23d5eceSKris Buschelman #undef __FUNCT__ 15564a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense" 1557be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat) 1558273d9f13SBarry Smith { 1559273d9f13SBarry Smith Mat_MPIDense *a; 1560dfbe8321SBarry Smith PetscErrorCode ierr; 1561273d9f13SBarry Smith 1562273d9f13SBarry Smith PetscFunctionBegin; 156338f2d2fdSLisandro Dalcin ierr = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr); 1564b0a32e0cSBarry Smith mat->data = (void*)a; 1565273d9f13SBarry Smith ierr = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1566273d9f13SBarry Smith mat->mapping = 0; 1567273d9f13SBarry Smith 1568273d9f13SBarry Smith mat->insertmode = NOT_SET_VALUES; 15697adad957SLisandro Dalcin ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr); 15707adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr); 1571273d9f13SBarry Smith 1572d0f46423SBarry Smith mat->rmap->bs = mat->cmap->bs = 1; 1573d0f46423SBarry Smith ierr = PetscMapSetUp(mat->rmap);CHKERRQ(ierr); 1574d0f46423SBarry Smith ierr = PetscMapSetUp(mat->cmap);CHKERRQ(ierr); 1575d0f46423SBarry Smith a->nvec = mat->cmap->n; 1576273d9f13SBarry Smith 1577273d9f13SBarry Smith /* build cache for off array entries formed */ 1578273d9f13SBarry Smith a->donotstash = PETSC_FALSE; 15797adad957SLisandro Dalcin ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr); 1580273d9f13SBarry Smith 1581273d9f13SBarry Smith /* stuff used for matrix vector multiply */ 1582273d9f13SBarry Smith a->lvec = 0; 1583273d9f13SBarry Smith a->Mvctx = 0; 1584273d9f13SBarry Smith a->roworiented = PETSC_TRUE; 1585273d9f13SBarry Smith 1586273d9f13SBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C", 1587273d9f13SBarry Smith "MatGetDiagonalBlock_MPIDense", 1588273d9f13SBarry Smith MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr); 1589a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C", 1590a23d5eceSKris Buschelman "MatMPIDenseSetPreallocation_MPIDense", 1591a23d5eceSKris Buschelman MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr); 15924ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C", 15934ae313f4SHong Zhang "MatMatMult_MPIAIJ_MPIDense", 15944ae313f4SHong Zhang MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr); 15954ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C", 15964ae313f4SHong Zhang "MatMatMultSymbolic_MPIAIJ_MPIDense", 15974ae313f4SHong Zhang MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr); 15984ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C", 15994ae313f4SHong Zhang "MatMatMultNumeric_MPIAIJ_MPIDense", 16004ae313f4SHong Zhang MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr); 16017878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 160285bd4cefSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_petsc_C", 160385bd4cefSHong Zhang "MatGetFactor_mpidense_petsc", 160485bd4cefSHong Zhang MatGetFactor_mpidense_petsc);CHKERRQ(ierr); 1605db4efbfdSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 1606db4efbfdSBarry Smith ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr); 1607db4efbfdSBarry Smith #endif 1608db4efbfdSBarry Smith 16097878bbefSBarry Smith #endif 161038aed534SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr); 161101b82886SBarry Smith 1612273d9f13SBarry Smith PetscFunctionReturn(0); 1613273d9f13SBarry Smith } 1614273d9f13SBarry Smith EXTERN_C_END 1615273d9f13SBarry Smith 1616209238afSKris Buschelman /*MC 1617002d173eSKris Buschelman MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices. 1618209238afSKris Buschelman 1619209238afSKris Buschelman This matrix type is identical to MATSEQDENSE when constructed with a single process communicator, 1620209238afSKris Buschelman and MATMPIDENSE otherwise. 1621209238afSKris Buschelman 1622209238afSKris Buschelman Options Database Keys: 1623209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions() 1624209238afSKris Buschelman 1625209238afSKris Buschelman Level: beginner 1626209238afSKris Buschelman 162701b82886SBarry Smith 1628209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE 1629209238afSKris Buschelman M*/ 1630209238afSKris Buschelman 1631209238afSKris Buschelman EXTERN_C_BEGIN 1632209238afSKris Buschelman #undef __FUNCT__ 1633209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense" 1634be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A) 1635dfbe8321SBarry Smith { 16366849ba73SBarry Smith PetscErrorCode ierr; 163713f74950SBarry Smith PetscMPIInt size; 1638209238afSKris Buschelman 1639209238afSKris Buschelman PetscFunctionBegin; 16407adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr); 1641209238afSKris Buschelman if (size == 1) { 1642209238afSKris Buschelman ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr); 1643209238afSKris Buschelman } else { 1644209238afSKris Buschelman ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr); 1645209238afSKris Buschelman } 1646209238afSKris Buschelman PetscFunctionReturn(0); 1647209238afSKris Buschelman } 1648209238afSKris Buschelman EXTERN_C_END 1649209238afSKris Buschelman 16504a2ae208SSatish Balay #undef __FUNCT__ 16514a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation" 1652273d9f13SBarry Smith /*@C 1653273d9f13SBarry Smith MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries 1654273d9f13SBarry Smith 1655273d9f13SBarry Smith Not collective 1656273d9f13SBarry Smith 1657273d9f13SBarry Smith Input Parameters: 1658273d9f13SBarry Smith . A - the matrix 1659273d9f13SBarry Smith - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1660273d9f13SBarry Smith to control all matrix memory allocation. 1661273d9f13SBarry Smith 1662273d9f13SBarry Smith Notes: 1663273d9f13SBarry Smith The dense format is fully compatible with standard Fortran 77 1664273d9f13SBarry Smith storage by columns. 1665273d9f13SBarry Smith 1666273d9f13SBarry Smith The data input variable is intended primarily for Fortran programmers 1667273d9f13SBarry Smith who wish to allocate their own matrix memory space. Most users should 1668273d9f13SBarry Smith set data=PETSC_NULL. 1669273d9f13SBarry Smith 1670273d9f13SBarry Smith Level: intermediate 1671273d9f13SBarry Smith 1672273d9f13SBarry Smith .keywords: matrix,dense, parallel 1673273d9f13SBarry Smith 1674273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues() 1675273d9f13SBarry Smith @*/ 1676be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data) 1677273d9f13SBarry Smith { 1678dfbe8321SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscScalar *); 1679273d9f13SBarry Smith 1680273d9f13SBarry Smith PetscFunctionBegin; 1681565567f0SKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1682a23d5eceSKris Buschelman if (f) { 1683a23d5eceSKris Buschelman ierr = (*f)(mat,data);CHKERRQ(ierr); 1684a23d5eceSKris Buschelman } 1685273d9f13SBarry Smith PetscFunctionReturn(0); 1686273d9f13SBarry Smith } 1687273d9f13SBarry Smith 16884a2ae208SSatish Balay #undef __FUNCT__ 16894a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense" 16908965ea79SLois Curfman McInnes /*@C 169139ddd567SLois Curfman McInnes MatCreateMPIDense - Creates a sparse parallel matrix in dense format. 16928965ea79SLois Curfman McInnes 1693db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1694db81eaa0SLois Curfman McInnes 16958965ea79SLois Curfman McInnes Input Parameters: 1696db81eaa0SLois Curfman McInnes + comm - MPI communicator 16978965ea79SLois Curfman McInnes . m - number of local rows (or PETSC_DECIDE to have calculated if M is given) 1698db81eaa0SLois Curfman McInnes . n - number of local columns (or PETSC_DECIDE to have calculated if N is given) 16998965ea79SLois Curfman McInnes . M - number of global rows (or PETSC_DECIDE to have calculated if m is given) 1700db81eaa0SLois Curfman McInnes . N - number of global columns (or PETSC_DECIDE to have calculated if n is given) 17017f5ff6fdSBarry Smith - data - optional location of matrix data. Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc 1702dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 17038965ea79SLois Curfman McInnes 17048965ea79SLois Curfman McInnes Output Parameter: 1705477f1c0bSLois Curfman McInnes . A - the matrix 17068965ea79SLois Curfman McInnes 1707b259b22eSLois Curfman McInnes Notes: 170839ddd567SLois Curfman McInnes The dense format is fully compatible with standard Fortran 77 170939ddd567SLois Curfman McInnes storage by columns. 17108965ea79SLois Curfman McInnes 171118f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 171218f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 17137f5ff6fdSBarry Smith set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users). 171418f449edSLois Curfman McInnes 17158965ea79SLois Curfman McInnes The user MUST specify either the local or global matrix dimensions 17168965ea79SLois Curfman McInnes (possibly both). 17178965ea79SLois Curfman McInnes 1718027ccd11SLois Curfman McInnes Level: intermediate 1719027ccd11SLois Curfman McInnes 172039ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel 17218965ea79SLois Curfman McInnes 172239ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues() 17238965ea79SLois Curfman McInnes @*/ 1724be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A) 17258965ea79SLois Curfman McInnes { 17266849ba73SBarry Smith PetscErrorCode ierr; 172713f74950SBarry Smith PetscMPIInt size; 17288965ea79SLois Curfman McInnes 17293a40ed3dSBarry Smith PetscFunctionBegin; 1730f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 1731f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr); 1732273d9f13SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1733273d9f13SBarry Smith if (size > 1) { 1734273d9f13SBarry Smith ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr); 1735273d9f13SBarry Smith ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr); 1736273d9f13SBarry Smith } else { 1737273d9f13SBarry Smith ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr); 1738273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr); 17398c469469SLois Curfman McInnes } 17403a40ed3dSBarry Smith PetscFunctionReturn(0); 17418965ea79SLois Curfman McInnes } 17428965ea79SLois Curfman McInnes 17434a2ae208SSatish Balay #undef __FUNCT__ 17444a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense" 17456849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 17468965ea79SLois Curfman McInnes { 17478965ea79SLois Curfman McInnes Mat mat; 17483501a2bdSLois Curfman McInnes Mat_MPIDense *a,*oldmat = (Mat_MPIDense*)A->data; 1749dfbe8321SBarry Smith PetscErrorCode ierr; 17508965ea79SLois Curfman McInnes 17513a40ed3dSBarry Smith PetscFunctionBegin; 17528965ea79SLois Curfman McInnes *newmat = 0; 17537adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr); 1754d0f46423SBarry Smith ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 17557adad957SLisandro Dalcin ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr); 1756834f8fabSBarry Smith a = (Mat_MPIDense*)mat->data; 1757e04c1aa4SHong Zhang ierr = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 17583501a2bdSLois Curfman McInnes mat->factor = A->factor; 1759c456f294SBarry Smith mat->assembled = PETSC_TRUE; 1760273d9f13SBarry Smith mat->preallocated = PETSC_TRUE; 17618965ea79SLois Curfman McInnes 1762d0f46423SBarry Smith mat->rmap->rstart = A->rmap->rstart; 1763d0f46423SBarry Smith mat->rmap->rend = A->rmap->rend; 17648965ea79SLois Curfman McInnes a->size = oldmat->size; 17658965ea79SLois Curfman McInnes a->rank = oldmat->rank; 1766e0fa3b82SLois Curfman McInnes mat->insertmode = NOT_SET_VALUES; 1767b9b97703SBarry Smith a->nvec = oldmat->nvec; 17683782ba37SSatish Balay a->donotstash = oldmat->donotstash; 1769e04c1aa4SHong Zhang 1770d0f46423SBarry Smith ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr); 1771d0f46423SBarry Smith ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr); 17727adad957SLisandro Dalcin ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr); 17738965ea79SLois Curfman McInnes 1774329f5518SBarry Smith ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr); 17755609ef8eSBarry Smith ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr); 177652e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr); 177701b82886SBarry Smith 17788965ea79SLois Curfman McInnes *newmat = mat; 17793a40ed3dSBarry Smith PetscFunctionReturn(0); 17808965ea79SLois Curfman McInnes } 17818965ea79SLois Curfman McInnes 1782e090d566SSatish Balay #include "petscsys.h" 17838965ea79SLois Curfman McInnes 17844a2ae208SSatish Balay #undef __FUNCT__ 17854a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile" 1786a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat) 178790ace30eSBarry Smith { 17886849ba73SBarry Smith PetscErrorCode ierr; 178932dcc486SBarry Smith PetscMPIInt rank,size; 179013f74950SBarry Smith PetscInt *rowners,i,m,nz,j; 179187828ca2SBarry Smith PetscScalar *array,*vals,*vals_ptr; 179290ace30eSBarry Smith MPI_Status status; 179390ace30eSBarry Smith 17943a40ed3dSBarry Smith PetscFunctionBegin; 1795d132466eSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 1796d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 179790ace30eSBarry Smith 179890ace30eSBarry Smith /* determine ownership of all rows */ 179990ace30eSBarry Smith m = M/size + ((M % size) > rank); 180013f74950SBarry Smith ierr = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr); 180113f74950SBarry Smith ierr = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr); 180290ace30eSBarry Smith rowners[0] = 0; 180390ace30eSBarry Smith for (i=2; i<=size; i++) { 180490ace30eSBarry Smith rowners[i] += rowners[i-1]; 180590ace30eSBarry Smith } 180690ace30eSBarry Smith 1807f69a0ea3SMatthew Knepley ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 1808f69a0ea3SMatthew Knepley ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr); 1809be5d1d56SKris Buschelman ierr = MatSetType(*newmat,type);CHKERRQ(ierr); 1810878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr); 181190ace30eSBarry Smith ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr); 181290ace30eSBarry Smith 181390ace30eSBarry Smith if (!rank) { 181487828ca2SBarry Smith ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 181590ace30eSBarry Smith 181690ace30eSBarry Smith /* read in my part of the matrix numerical values */ 18170752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr); 181890ace30eSBarry Smith 181990ace30eSBarry Smith /* insert into matrix-by row (this is why cannot directly read into array */ 182090ace30eSBarry Smith vals_ptr = vals; 182190ace30eSBarry Smith for (i=0; i<m; i++) { 182290ace30eSBarry Smith for (j=0; j<N; j++) { 182390ace30eSBarry Smith array[i + j*m] = *vals_ptr++; 182490ace30eSBarry Smith } 182590ace30eSBarry Smith } 182690ace30eSBarry Smith 182790ace30eSBarry Smith /* read in other processors and ship out */ 182890ace30eSBarry Smith for (i=1; i<size; i++) { 182990ace30eSBarry Smith nz = (rowners[i+1] - rowners[i])*N; 18300752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 18317adad957SLisandro Dalcin ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr); 183290ace30eSBarry Smith } 18333a40ed3dSBarry Smith } else { 183490ace30eSBarry Smith /* receive numeric values */ 183587828ca2SBarry Smith ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 183690ace30eSBarry Smith 183790ace30eSBarry Smith /* receive message of values*/ 18387adad957SLisandro Dalcin ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr); 183990ace30eSBarry Smith 184090ace30eSBarry Smith /* insert into matrix-by row (this is why cannot directly read into array */ 184190ace30eSBarry Smith vals_ptr = vals; 184290ace30eSBarry Smith for (i=0; i<m; i++) { 184390ace30eSBarry Smith for (j=0; j<N; j++) { 184490ace30eSBarry Smith array[i + j*m] = *vals_ptr++; 184590ace30eSBarry Smith } 184690ace30eSBarry Smith } 184790ace30eSBarry Smith } 1848606d414cSSatish Balay ierr = PetscFree(rowners);CHKERRQ(ierr); 1849606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 18506d4a8577SBarry Smith ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 18516d4a8577SBarry Smith ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 18523a40ed3dSBarry Smith PetscFunctionReturn(0); 185390ace30eSBarry Smith } 185490ace30eSBarry Smith 18554a2ae208SSatish Balay #undef __FUNCT__ 18564a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense" 1857a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat) 18588965ea79SLois Curfman McInnes { 18598965ea79SLois Curfman McInnes Mat A; 186087828ca2SBarry Smith PetscScalar *vals,*svals; 186119bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 18628965ea79SLois Curfman McInnes MPI_Status status; 186313f74950SBarry Smith PetscMPIInt rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz; 186413f74950SBarry Smith PetscInt header[4],*rowlengths = 0,M,N,*cols; 186513f74950SBarry Smith PetscInt *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols; 186613f74950SBarry Smith PetscInt i,nz,j,rstart,rend; 186713f74950SBarry Smith int fd; 18686849ba73SBarry Smith PetscErrorCode ierr; 18698965ea79SLois Curfman McInnes 18703a40ed3dSBarry Smith PetscFunctionBegin; 1871d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1872d132466eSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 18738965ea79SLois Curfman McInnes if (!rank) { 1874b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 18750752156aSBarry Smith ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr); 1876552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object"); 18778965ea79SLois Curfman McInnes } 18788965ea79SLois Curfman McInnes 187913f74950SBarry Smith ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr); 188090ace30eSBarry Smith M = header[1]; N = header[2]; nz = header[3]; 188190ace30eSBarry Smith 188290ace30eSBarry Smith /* 188390ace30eSBarry Smith Handle case where matrix is stored on disk as a dense matrix 188490ace30eSBarry Smith */ 188590ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { 1886be5d1d56SKris Buschelman ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr); 18873a40ed3dSBarry Smith PetscFunctionReturn(0); 188890ace30eSBarry Smith } 188990ace30eSBarry Smith 18908965ea79SLois Curfman McInnes /* determine ownership of all rows */ 1891e44c0bd4SBarry Smith m = PetscMPIIntCast(M/size + ((M % size) > rank)); 1892e44c0bd4SBarry Smith ierr = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr); 1893ca161407SBarry Smith ierr = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr); 18948965ea79SLois Curfman McInnes rowners[0] = 0; 18958965ea79SLois Curfman McInnes for (i=2; i<=size; i++) { 18968965ea79SLois Curfman McInnes rowners[i] += rowners[i-1]; 18978965ea79SLois Curfman McInnes } 18988965ea79SLois Curfman McInnes rstart = rowners[rank]; 18998965ea79SLois Curfman McInnes rend = rowners[rank+1]; 19008965ea79SLois Curfman McInnes 19018965ea79SLois Curfman McInnes /* distribute row lengths to all processors */ 190213f74950SBarry Smith ierr = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr); 19038965ea79SLois Curfman McInnes offlens = ourlens + (rend-rstart); 19048965ea79SLois Curfman McInnes if (!rank) { 190513f74950SBarry Smith ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 19060752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 190713f74950SBarry Smith ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr); 19088965ea79SLois Curfman McInnes for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i]; 19091466f1e1SBarry Smith ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr); 1910606d414cSSatish Balay ierr = PetscFree(sndcounts);CHKERRQ(ierr); 1911ca161407SBarry Smith } else { 19121466f1e1SBarry Smith ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr); 19138965ea79SLois Curfman McInnes } 19148965ea79SLois Curfman McInnes 19158965ea79SLois Curfman McInnes if (!rank) { 19168965ea79SLois Curfman McInnes /* calculate the number of nonzeros on each processor */ 191713f74950SBarry Smith ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr); 191813f74950SBarry Smith ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr); 19198965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 19208965ea79SLois Curfman McInnes for (j=rowners[i]; j< rowners[i+1]; j++) { 19218965ea79SLois Curfman McInnes procsnz[i] += rowlengths[j]; 19228965ea79SLois Curfman McInnes } 19238965ea79SLois Curfman McInnes } 1924606d414cSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 19258965ea79SLois Curfman McInnes 19268965ea79SLois Curfman McInnes /* determine max buffer needed and allocate it */ 19278965ea79SLois Curfman McInnes maxnz = 0; 19288965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 19290452661fSBarry Smith maxnz = PetscMax(maxnz,procsnz[i]); 19308965ea79SLois Curfman McInnes } 193113f74950SBarry Smith ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr); 19328965ea79SLois Curfman McInnes 19338965ea79SLois Curfman McInnes /* read in my part of the matrix column indices */ 19348965ea79SLois Curfman McInnes nz = procsnz[0]; 193513f74950SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr); 19360752156aSBarry Smith ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr); 19378965ea79SLois Curfman McInnes 19388965ea79SLois Curfman McInnes /* read in every one elses and ship off */ 19398965ea79SLois Curfman McInnes for (i=1; i<size; i++) { 19408965ea79SLois Curfman McInnes nz = procsnz[i]; 19410752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr); 194213f74950SBarry Smith ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr); 19438965ea79SLois Curfman McInnes } 1944606d414cSSatish Balay ierr = PetscFree(cols);CHKERRQ(ierr); 19453a40ed3dSBarry Smith } else { 19468965ea79SLois Curfman McInnes /* determine buffer space needed for message */ 19478965ea79SLois Curfman McInnes nz = 0; 19488965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19498965ea79SLois Curfman McInnes nz += ourlens[i]; 19508965ea79SLois Curfman McInnes } 195113f74950SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr); 19528965ea79SLois Curfman McInnes 19538965ea79SLois Curfman McInnes /* receive message of column indices*/ 195413f74950SBarry Smith ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr); 195513f74950SBarry Smith ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr); 195629bbc08cSBarry Smith if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file"); 19578965ea79SLois Curfman McInnes } 19588965ea79SLois Curfman McInnes 19598965ea79SLois Curfman McInnes /* loop over local rows, determining number of off diagonal entries */ 196013f74950SBarry Smith ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr); 19618965ea79SLois Curfman McInnes jj = 0; 19628965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19638965ea79SLois Curfman McInnes for (j=0; j<ourlens[i]; j++) { 19648965ea79SLois Curfman McInnes if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++; 19658965ea79SLois Curfman McInnes jj++; 19668965ea79SLois Curfman McInnes } 19678965ea79SLois Curfman McInnes } 19688965ea79SLois Curfman McInnes 19698965ea79SLois Curfman McInnes /* create our matrix */ 19708965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19718965ea79SLois Curfman McInnes ourlens[i] -= offlens[i]; 19728965ea79SLois Curfman McInnes } 1973f69a0ea3SMatthew Knepley ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 1974f69a0ea3SMatthew Knepley ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr); 1975be5d1d56SKris Buschelman ierr = MatSetType(*newmat,type);CHKERRQ(ierr); 1976878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr); 19778965ea79SLois Curfman McInnes A = *newmat; 19788965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19798965ea79SLois Curfman McInnes ourlens[i] += offlens[i]; 19808965ea79SLois Curfman McInnes } 19818965ea79SLois Curfman McInnes 19828965ea79SLois Curfman McInnes if (!rank) { 198387828ca2SBarry Smith ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 19848965ea79SLois Curfman McInnes 19858965ea79SLois Curfman McInnes /* read in my part of the matrix numerical values */ 19868965ea79SLois Curfman McInnes nz = procsnz[0]; 19870752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 19888965ea79SLois Curfman McInnes 19898965ea79SLois Curfman McInnes /* insert into matrix */ 19908965ea79SLois Curfman McInnes jj = rstart; 19918965ea79SLois Curfman McInnes smycols = mycols; 19928965ea79SLois Curfman McInnes svals = vals; 19938965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19948965ea79SLois Curfman McInnes ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr); 19958965ea79SLois Curfman McInnes smycols += ourlens[i]; 19968965ea79SLois Curfman McInnes svals += ourlens[i]; 19978965ea79SLois Curfman McInnes jj++; 19988965ea79SLois Curfman McInnes } 19998965ea79SLois Curfman McInnes 20008965ea79SLois Curfman McInnes /* read in other processors and ship out */ 20018965ea79SLois Curfman McInnes for (i=1; i<size; i++) { 20028965ea79SLois Curfman McInnes nz = procsnz[i]; 20030752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 20047adad957SLisandro Dalcin ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr); 20058965ea79SLois Curfman McInnes } 2006606d414cSSatish Balay ierr = PetscFree(procsnz);CHKERRQ(ierr); 20073a40ed3dSBarry Smith } else { 20088965ea79SLois Curfman McInnes /* receive numeric values */ 200984d528b1SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 20108965ea79SLois Curfman McInnes 20118965ea79SLois Curfman McInnes /* receive message of values*/ 20127adad957SLisandro Dalcin ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr); 2013ca161407SBarry Smith ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr); 201429bbc08cSBarry Smith if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file"); 20158965ea79SLois Curfman McInnes 20168965ea79SLois Curfman McInnes /* insert into matrix */ 20178965ea79SLois Curfman McInnes jj = rstart; 20188965ea79SLois Curfman McInnes smycols = mycols; 20198965ea79SLois Curfman McInnes svals = vals; 20208965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20218965ea79SLois Curfman McInnes ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr); 20228965ea79SLois Curfman McInnes smycols += ourlens[i]; 20238965ea79SLois Curfman McInnes svals += ourlens[i]; 20248965ea79SLois Curfman McInnes jj++; 20258965ea79SLois Curfman McInnes } 20268965ea79SLois Curfman McInnes } 2027606d414cSSatish Balay ierr = PetscFree(ourlens);CHKERRQ(ierr); 2028606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 2029606d414cSSatish Balay ierr = PetscFree(mycols);CHKERRQ(ierr); 2030606d414cSSatish Balay ierr = PetscFree(rowners);CHKERRQ(ierr); 20318965ea79SLois Curfman McInnes 20326d4a8577SBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 20336d4a8577SBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 20343a40ed3dSBarry Smith PetscFunctionReturn(0); 20358965ea79SLois Curfman McInnes } 203690ace30eSBarry Smith 20376e4ee0c6SHong Zhang #undef __FUNCT__ 20386e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense" 20396e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag) 20406e4ee0c6SHong Zhang { 20416e4ee0c6SHong Zhang Mat_MPIDense *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data; 20426e4ee0c6SHong Zhang Mat a,b; 20436e4ee0c6SHong Zhang PetscTruth flg; 20446e4ee0c6SHong Zhang PetscErrorCode ierr; 204590ace30eSBarry Smith 20466e4ee0c6SHong Zhang PetscFunctionBegin; 20476e4ee0c6SHong Zhang a = matA->A; 20486e4ee0c6SHong Zhang b = matB->A; 20496e4ee0c6SHong Zhang ierr = MatEqual(a,b,&flg);CHKERRQ(ierr); 20507adad957SLisandro Dalcin ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr); 20516e4ee0c6SHong Zhang PetscFunctionReturn(0); 20526e4ee0c6SHong Zhang } 205390ace30eSBarry Smith 205409d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 205509d27a7eSBarry Smith 205609d27a7eSBarry Smith #undef __FUNCT__ 205709d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage" 205809d27a7eSBarry Smith /*@C 205909d27a7eSBarry Smith PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK. 206009d27a7eSBarry Smith Level: developer 206109d27a7eSBarry Smith 206209d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK 206309d27a7eSBarry Smith .seealso: PetscFinalize() 206409d27a7eSBarry Smith @*/ 206509d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void) 206609d27a7eSBarry Smith { 206709d27a7eSBarry Smith PetscErrorCode ierr; 206809d27a7eSBarry Smith 206909d27a7eSBarry Smith PetscFunctionBegin; 207009d27a7eSBarry Smith ierr = PLA_Finalize();CHKERRQ(ierr); 207109d27a7eSBarry Smith PetscFunctionReturn(0); 207209d27a7eSBarry Smith } 207309d27a7eSBarry Smith 207409d27a7eSBarry Smith #undef __FUNCT__ 207509d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage" 207609d27a7eSBarry Smith /*@C 207709d27a7eSBarry Smith PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is 2078db4efbfdSBarry Smith called from MatCreate_MPIDense() the first time an MPI dense matrix is called. 207909d27a7eSBarry Smith 208009d27a7eSBarry Smith Input Parameter: 2081db4efbfdSBarry Smith . comm - the communicator the matrix lives on 208209d27a7eSBarry Smith 208309d27a7eSBarry Smith Level: developer 208409d27a7eSBarry Smith 2085db4efbfdSBarry Smith Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the 2086db4efbfdSBarry Smith same communicator (because there is some global state that is initialized and used for all matrices). In addition if 2087db4efbfdSBarry Smith PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators 2088db4efbfdSBarry Smith cannot overlap. 2089db4efbfdSBarry Smith 209009d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK 209109d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize() 209209d27a7eSBarry Smith @*/ 2093db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm) 209409d27a7eSBarry Smith { 2095eae6fb2eSBarry Smith PetscMPIInt size; 209609d27a7eSBarry Smith PetscErrorCode ierr; 209709d27a7eSBarry Smith 209809d27a7eSBarry Smith PetscFunctionBegin; 209909d27a7eSBarry Smith if (!PLA_Initialized(PETSC_NULL)) { 210009d27a7eSBarry Smith 210109d27a7eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 210204fea9ffSBarry Smith Plapack_nprows = 1; 210304fea9ffSBarry Smith Plapack_npcols = size; 210409d27a7eSBarry Smith 2105aeccfd6fSBarry Smith ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr); 2106*5d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr); 2107*5d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr); 210879b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG) 210979b0a62dSBarry Smith Plapack_ierror = 3; 211079b0a62dSBarry Smith #else 211179b0a62dSBarry Smith Plapack_ierror = 0; 211279b0a62dSBarry Smith #endif 2113*5d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr); 2114eae6fb2eSBarry Smith if (Plapack_ierror){ 2115eae6fb2eSBarry Smith ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr); 2116aeccfd6fSBarry Smith } else { 2117eae6fb2eSBarry Smith ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr); 2118aeccfd6fSBarry Smith } 2119aeccfd6fSBarry Smith 2120eae6fb2eSBarry Smith Plapack_nb_alg = 0; 2121*5d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr); 2122eae6fb2eSBarry Smith if (Plapack_nb_alg) { 2123eae6fb2eSBarry Smith ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr); 2124aeccfd6fSBarry Smith } 2125aeccfd6fSBarry Smith PetscOptionsEnd(); 2126aeccfd6fSBarry Smith 212704fea9ffSBarry Smith ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr); 212804fea9ffSBarry Smith ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr); 212909d27a7eSBarry Smith ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr); 213009d27a7eSBarry Smith } 213109d27a7eSBarry Smith PetscFunctionReturn(0); 213209d27a7eSBarry Smith } 213390ace30eSBarry Smith 213409d27a7eSBarry Smith #endif 2135