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) { 132b400d20cSBarry Smith ierr = MatStashValuesRow_Private(&mat->stash,idxm[i],n,idxn,v+i*n,PETSC_FALSE);CHKERRQ(ierr); 133d36fbae8SSatish Balay } else { 134b400d20cSBarry Smith ierr = MatStashValuesCol_Private(&mat->stash,idxm[i],n,idxn,v+i,m,PETSC_FALSE);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" 1844aa3045dSJed Brown static PetscErrorCode MatGetSubMatrix_MPIDense(Mat A,IS isrow,IS iscol,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; 1894aa3045dSJed Brown PetscInt i,j,rstart,rend,nrows,ncols,Ncols,nlrows,nlcols; 1905d0c19d7SBarry Smith const PetscInt *irow,*icol; 19187828ca2SBarry Smith PetscScalar *av,*bv,*v = lmat->v; 192ca3fa75bSLois Curfman McInnes Mat newmat; 1934aa3045dSJed Brown IS iscol_local; 194ca3fa75bSLois Curfman McInnes 195ca3fa75bSLois Curfman McInnes PetscFunctionBegin; 1964aa3045dSJed Brown ierr = ISAllGather(iscol,&iscol_local);CHKERRQ(ierr); 197ca3fa75bSLois Curfman McInnes ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 1984aa3045dSJed Brown ierr = ISGetIndices(iscol_local,&icol);CHKERRQ(ierr); 199b9b97703SBarry Smith ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr); 200b9b97703SBarry Smith ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr); 2014aa3045dSJed Brown ierr = ISGetSize(iscol,&Ncols);CHKERRQ(ierr); /* global number of columns, size of iscol_local */ 202ca3fa75bSLois Curfman McInnes 203ca3fa75bSLois Curfman McInnes /* No parallel redistribution currently supported! Should really check each index set 2047eba5e9cSLois Curfman McInnes to comfirm that it is OK. ... Currently supports only submatrix same partitioning as 2057eba5e9cSLois Curfman McInnes original matrix! */ 206ca3fa75bSLois Curfman McInnes 207ca3fa75bSLois Curfman McInnes ierr = MatGetLocalSize(A,&nlrows,&nlcols);CHKERRQ(ierr); 2087eba5e9cSLois Curfman McInnes ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 209ca3fa75bSLois Curfman McInnes 210ca3fa75bSLois Curfman McInnes /* Check submatrix call */ 211ca3fa75bSLois Curfman McInnes if (scall == MAT_REUSE_MATRIX) { 21229bbc08cSBarry Smith /* SETERRQ(PETSC_ERR_ARG_SIZ,"Reused submatrix wrong size"); */ 2137eba5e9cSLois Curfman McInnes /* Really need to test rows and column sizes! */ 214ca3fa75bSLois Curfman McInnes newmat = *B; 215ca3fa75bSLois Curfman McInnes } else { 216ca3fa75bSLois Curfman McInnes /* Create and fill new matrix */ 2177adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr); 2184aa3045dSJed Brown ierr = MatSetSizes(newmat,nrows,ncols,PETSC_DECIDE,Ncols);CHKERRQ(ierr); 2197adad957SLisandro Dalcin ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr); 220878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr); 221ca3fa75bSLois Curfman McInnes } 222ca3fa75bSLois Curfman McInnes 223ca3fa75bSLois Curfman McInnes /* Now extract the data pointers and do the copy, column at a time */ 224ca3fa75bSLois Curfman McInnes newmatd = (Mat_MPIDense*)newmat->data; 225ca3fa75bSLois Curfman McInnes bv = ((Mat_SeqDense *)newmatd->A->data)->v; 226ca3fa75bSLois Curfman McInnes 2274aa3045dSJed Brown for (i=0; i<Ncols; i++) { 22825a33276SHong Zhang av = v + ((Mat_SeqDense *)mat->A->data)->lda*icol[i]; 229ca3fa75bSLois Curfman McInnes for (j=0; j<nrows; j++) { 2307eba5e9cSLois Curfman McInnes *bv++ = av[irow[j] - rstart]; 231ca3fa75bSLois Curfman McInnes } 232ca3fa75bSLois Curfman McInnes } 233ca3fa75bSLois Curfman McInnes 234ca3fa75bSLois Curfman McInnes /* Assemble the matrices so that the correct flags are set */ 235ca3fa75bSLois Curfman McInnes ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 236ca3fa75bSLois Curfman McInnes ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 237ca3fa75bSLois Curfman McInnes 238ca3fa75bSLois Curfman McInnes /* Free work space */ 239ca3fa75bSLois Curfman McInnes ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 240ca3fa75bSLois Curfman McInnes ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 241ca3fa75bSLois Curfman McInnes *B = newmat; 242ca3fa75bSLois Curfman McInnes PetscFunctionReturn(0); 243ca3fa75bSLois Curfman McInnes } 244ca3fa75bSLois Curfman McInnes 2454a2ae208SSatish Balay #undef __FUNCT__ 2464a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_MPIDense" 247dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_MPIDense(Mat A,PetscScalar *array[]) 248ff14e315SSatish Balay { 2493a40ed3dSBarry Smith PetscFunctionBegin; 2503a40ed3dSBarry Smith PetscFunctionReturn(0); 251ff14e315SSatish Balay } 252ff14e315SSatish Balay 2534a2ae208SSatish Balay #undef __FUNCT__ 2544a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyBegin_MPIDense" 255dfbe8321SBarry Smith PetscErrorCode MatAssemblyBegin_MPIDense(Mat mat,MatAssemblyType mode) 2568965ea79SLois Curfman McInnes { 25739ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 2587adad957SLisandro Dalcin MPI_Comm comm = ((PetscObject)mat)->comm; 259dfbe8321SBarry Smith PetscErrorCode ierr; 26013f74950SBarry Smith PetscInt nstash,reallocs; 2618965ea79SLois Curfman McInnes InsertMode addv; 2628965ea79SLois Curfman McInnes 2633a40ed3dSBarry Smith PetscFunctionBegin; 2648965ea79SLois Curfman McInnes /* make sure all processors are either in INSERTMODE or ADDMODE */ 265ca161407SBarry Smith ierr = MPI_Allreduce(&mat->insertmode,&addv,1,MPI_INT,MPI_BOR,comm);CHKERRQ(ierr); 2667056b6fcSBarry Smith if (addv == (ADD_VALUES|INSERT_VALUES)) { 26729bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Cannot mix adds/inserts on different procs"); 2688965ea79SLois Curfman McInnes } 269e0fa3b82SLois Curfman McInnes mat->insertmode = addv; /* in case this processor had no cache */ 2708965ea79SLois Curfman McInnes 271d0f46423SBarry Smith ierr = MatStashScatterBegin_Private(mat,&mat->stash,mat->rmap->range);CHKERRQ(ierr); 2728798bf22SSatish Balay ierr = MatStashGetInfo_Private(&mat->stash,&nstash,&reallocs);CHKERRQ(ierr); 273ae15b995SBarry Smith ierr = PetscInfo2(mdn->A,"Stash has %D entries, uses %D mallocs.\n",nstash,reallocs);CHKERRQ(ierr); 2743a40ed3dSBarry Smith PetscFunctionReturn(0); 2758965ea79SLois Curfman McInnes } 2768965ea79SLois Curfman McInnes 2774a2ae208SSatish Balay #undef __FUNCT__ 2784a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_MPIDense" 279dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_MPIDense(Mat mat,MatAssemblyType mode) 2808965ea79SLois Curfman McInnes { 28139ddd567SLois Curfman McInnes Mat_MPIDense *mdn=(Mat_MPIDense*)mat->data; 2826849ba73SBarry Smith PetscErrorCode ierr; 28313f74950SBarry Smith PetscInt i,*row,*col,flg,j,rstart,ncols; 28413f74950SBarry Smith PetscMPIInt n; 28587828ca2SBarry Smith PetscScalar *val; 286e0fa3b82SLois Curfman McInnes InsertMode addv=mat->insertmode; 2878965ea79SLois Curfman McInnes 2883a40ed3dSBarry Smith PetscFunctionBegin; 2898965ea79SLois Curfman McInnes /* wait on receives */ 2907ef1d9bdSSatish Balay while (1) { 2918798bf22SSatish Balay ierr = MatStashScatterGetMesg_Private(&mat->stash,&n,&row,&col,&val,&flg);CHKERRQ(ierr); 2927ef1d9bdSSatish Balay if (!flg) break; 2938965ea79SLois Curfman McInnes 2947ef1d9bdSSatish Balay for (i=0; i<n;) { 2957ef1d9bdSSatish Balay /* Now identify the consecutive vals belonging to the same row */ 2967ef1d9bdSSatish Balay for (j=i,rstart=row[j]; j<n; j++) { if (row[j] != rstart) break; } 2977ef1d9bdSSatish Balay if (j < n) ncols = j-i; 2987ef1d9bdSSatish Balay else ncols = n-i; 2997ef1d9bdSSatish Balay /* Now assemble all these values with a single function call */ 3007ef1d9bdSSatish Balay ierr = MatSetValues_MPIDense(mat,1,row+i,ncols,col+i,val+i,addv);CHKERRQ(ierr); 3017ef1d9bdSSatish Balay i = j; 3028965ea79SLois Curfman McInnes } 3037ef1d9bdSSatish Balay } 3048798bf22SSatish Balay ierr = MatStashScatterEnd_Private(&mat->stash);CHKERRQ(ierr); 3058965ea79SLois Curfman McInnes 30639ddd567SLois Curfman McInnes ierr = MatAssemblyBegin(mdn->A,mode);CHKERRQ(ierr); 30739ddd567SLois Curfman McInnes ierr = MatAssemblyEnd(mdn->A,mode);CHKERRQ(ierr); 3088965ea79SLois Curfman McInnes 3096d4a8577SBarry Smith if (!mat->was_assembled && mode == MAT_FINAL_ASSEMBLY) { 31039ddd567SLois Curfman McInnes ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr); 3118965ea79SLois Curfman McInnes } 3123a40ed3dSBarry Smith PetscFunctionReturn(0); 3138965ea79SLois Curfman McInnes } 3148965ea79SLois Curfman McInnes 3154a2ae208SSatish Balay #undef __FUNCT__ 3164a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_MPIDense" 317dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_MPIDense(Mat A) 3188965ea79SLois Curfman McInnes { 319dfbe8321SBarry Smith PetscErrorCode ierr; 32039ddd567SLois Curfman McInnes Mat_MPIDense *l = (Mat_MPIDense*)A->data; 3213a40ed3dSBarry Smith 3223a40ed3dSBarry Smith PetscFunctionBegin; 3233a40ed3dSBarry Smith ierr = MatZeroEntries(l->A);CHKERRQ(ierr); 3243a40ed3dSBarry Smith PetscFunctionReturn(0); 3258965ea79SLois Curfman McInnes } 3268965ea79SLois Curfman McInnes 3278965ea79SLois Curfman McInnes /* the code does not do the diagonal entries correctly unless the 3288965ea79SLois Curfman McInnes matrix is square and the column and row owerships are identical. 3298965ea79SLois Curfman McInnes This is a BUG. The only way to fix it seems to be to access 3303501a2bdSLois Curfman McInnes mdn->A and mdn->B directly and not through the MatZeroRows() 3318965ea79SLois Curfman McInnes routine. 3328965ea79SLois Curfman McInnes */ 3334a2ae208SSatish Balay #undef __FUNCT__ 3344a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_MPIDense" 335f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_MPIDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag) 3368965ea79SLois Curfman McInnes { 33739ddd567SLois Curfman McInnes Mat_MPIDense *l = (Mat_MPIDense*)A->data; 3386849ba73SBarry Smith PetscErrorCode ierr; 339d0f46423SBarry Smith PetscInt i,*owners = A->rmap->range; 34013f74950SBarry Smith PetscInt *nprocs,j,idx,nsends; 34113f74950SBarry Smith PetscInt nmax,*svalues,*starts,*owner,nrecvs; 3427adad957SLisandro Dalcin PetscInt *rvalues,tag = ((PetscObject)A)->tag,count,base,slen,*source; 34313f74950SBarry Smith PetscInt *lens,*lrows,*values; 34413f74950SBarry Smith PetscMPIInt n,imdex,rank = l->rank,size = l->size; 3457adad957SLisandro Dalcin MPI_Comm comm = ((PetscObject)A)->comm; 3468965ea79SLois Curfman McInnes MPI_Request *send_waits,*recv_waits; 3478965ea79SLois Curfman McInnes MPI_Status recv_status,*send_status; 34835d8aa7fSBarry Smith PetscTruth found; 3498965ea79SLois Curfman McInnes 3503a40ed3dSBarry Smith PetscFunctionBegin; 3518965ea79SLois Curfman McInnes /* first count number of contributors to each processor */ 35213f74950SBarry Smith ierr = PetscMalloc(2*size*sizeof(PetscInt),&nprocs);CHKERRQ(ierr); 35313f74950SBarry Smith ierr = PetscMemzero(nprocs,2*size*sizeof(PetscInt));CHKERRQ(ierr); 35413f74950SBarry Smith ierr = PetscMalloc((N+1)*sizeof(PetscInt),&owner);CHKERRQ(ierr); /* see note*/ 3558965ea79SLois Curfman McInnes for (i=0; i<N; i++) { 3568965ea79SLois Curfman McInnes idx = rows[i]; 35735d8aa7fSBarry Smith found = PETSC_FALSE; 3588965ea79SLois Curfman McInnes for (j=0; j<size; j++) { 3598965ea79SLois Curfman McInnes if (idx >= owners[j] && idx < owners[j+1]) { 360c1dc657dSBarry Smith nprocs[2*j]++; nprocs[2*j+1] = 1; owner[i] = j; found = PETSC_TRUE; break; 3618965ea79SLois Curfman McInnes } 3628965ea79SLois Curfman McInnes } 36329bbc08cSBarry Smith if (!found) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Index out of range"); 3648965ea79SLois Curfman McInnes } 365c1dc657dSBarry Smith nsends = 0; for (i=0; i<size; i++) { nsends += nprocs[2*i+1];} 3668965ea79SLois Curfman McInnes 3678965ea79SLois Curfman McInnes /* inform other processors of number of messages and max length*/ 368c1dc657dSBarry Smith ierr = PetscMaxSum(comm,nprocs,&nmax,&nrecvs);CHKERRQ(ierr); 3698965ea79SLois Curfman McInnes 3708965ea79SLois Curfman McInnes /* post receives: */ 37113f74950SBarry Smith ierr = PetscMalloc((nrecvs+1)*(nmax+1)*sizeof(PetscInt),&rvalues);CHKERRQ(ierr); 372b0a32e0cSBarry Smith ierr = PetscMalloc((nrecvs+1)*sizeof(MPI_Request),&recv_waits);CHKERRQ(ierr); 3738965ea79SLois Curfman McInnes for (i=0; i<nrecvs; i++) { 37413f74950SBarry Smith ierr = MPI_Irecv(rvalues+nmax*i,nmax,MPIU_INT,MPI_ANY_SOURCE,tag,comm,recv_waits+i);CHKERRQ(ierr); 3758965ea79SLois Curfman McInnes } 3768965ea79SLois Curfman McInnes 3778965ea79SLois Curfman McInnes /* do sends: 3788965ea79SLois Curfman McInnes 1) starts[i] gives the starting index in svalues for stuff going to 3798965ea79SLois Curfman McInnes the ith processor 3808965ea79SLois Curfman McInnes */ 38113f74950SBarry Smith ierr = PetscMalloc((N+1)*sizeof(PetscInt),&svalues);CHKERRQ(ierr); 382b0a32e0cSBarry Smith ierr = PetscMalloc((nsends+1)*sizeof(MPI_Request),&send_waits);CHKERRQ(ierr); 38313f74950SBarry Smith ierr = PetscMalloc((size+1)*sizeof(PetscInt),&starts);CHKERRQ(ierr); 3848965ea79SLois Curfman McInnes starts[0] = 0; 385c1dc657dSBarry Smith for (i=1; i<size; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];} 3868965ea79SLois Curfman McInnes for (i=0; i<N; i++) { 3878965ea79SLois Curfman McInnes svalues[starts[owner[i]]++] = rows[i]; 3888965ea79SLois Curfman McInnes } 3898965ea79SLois Curfman McInnes 3908965ea79SLois Curfman McInnes starts[0] = 0; 391c1dc657dSBarry Smith for (i=1; i<size+1; i++) { starts[i] = starts[i-1] + nprocs[2*i-2];} 3928965ea79SLois Curfman McInnes count = 0; 3938965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 394c1dc657dSBarry Smith if (nprocs[2*i+1]) { 39513f74950SBarry Smith ierr = MPI_Isend(svalues+starts[i],nprocs[2*i],MPIU_INT,i,tag,comm,send_waits+count++);CHKERRQ(ierr); 3968965ea79SLois Curfman McInnes } 3978965ea79SLois Curfman McInnes } 398606d414cSSatish Balay ierr = PetscFree(starts);CHKERRQ(ierr); 3998965ea79SLois Curfman McInnes 4008965ea79SLois Curfman McInnes base = owners[rank]; 4018965ea79SLois Curfman McInnes 4028965ea79SLois Curfman McInnes /* wait on receives */ 40313f74950SBarry Smith ierr = PetscMalloc(2*(nrecvs+1)*sizeof(PetscInt),&lens);CHKERRQ(ierr); 4048965ea79SLois Curfman McInnes source = lens + nrecvs; 4058965ea79SLois Curfman McInnes count = nrecvs; slen = 0; 4068965ea79SLois Curfman McInnes while (count) { 407ca161407SBarry Smith ierr = MPI_Waitany(nrecvs,recv_waits,&imdex,&recv_status);CHKERRQ(ierr); 4088965ea79SLois Curfman McInnes /* unpack receives into our local space */ 40913f74950SBarry Smith ierr = MPI_Get_count(&recv_status,MPIU_INT,&n);CHKERRQ(ierr); 4108965ea79SLois Curfman McInnes source[imdex] = recv_status.MPI_SOURCE; 4118965ea79SLois Curfman McInnes lens[imdex] = n; 4128965ea79SLois Curfman McInnes slen += n; 4138965ea79SLois Curfman McInnes count--; 4148965ea79SLois Curfman McInnes } 415606d414cSSatish Balay ierr = PetscFree(recv_waits);CHKERRQ(ierr); 4168965ea79SLois Curfman McInnes 4178965ea79SLois Curfman McInnes /* move the data into the send scatter */ 41813f74950SBarry Smith ierr = PetscMalloc((slen+1)*sizeof(PetscInt),&lrows);CHKERRQ(ierr); 4198965ea79SLois Curfman McInnes count = 0; 4208965ea79SLois Curfman McInnes for (i=0; i<nrecvs; i++) { 4218965ea79SLois Curfman McInnes values = rvalues + i*nmax; 4228965ea79SLois Curfman McInnes for (j=0; j<lens[i]; j++) { 4238965ea79SLois Curfman McInnes lrows[count++] = values[j] - base; 4248965ea79SLois Curfman McInnes } 4258965ea79SLois Curfman McInnes } 426606d414cSSatish Balay ierr = PetscFree(rvalues);CHKERRQ(ierr); 427606d414cSSatish Balay ierr = PetscFree(lens);CHKERRQ(ierr); 428606d414cSSatish Balay ierr = PetscFree(owner);CHKERRQ(ierr); 429606d414cSSatish Balay ierr = PetscFree(nprocs);CHKERRQ(ierr); 4308965ea79SLois Curfman McInnes 4318965ea79SLois Curfman McInnes /* actually zap the local rows */ 432f4df32b1SMatthew Knepley ierr = MatZeroRows(l->A,slen,lrows,diag);CHKERRQ(ierr); 433606d414cSSatish Balay ierr = PetscFree(lrows);CHKERRQ(ierr); 4348965ea79SLois Curfman McInnes 4358965ea79SLois Curfman McInnes /* wait on sends */ 4368965ea79SLois Curfman McInnes if (nsends) { 437b0a32e0cSBarry Smith ierr = PetscMalloc(nsends*sizeof(MPI_Status),&send_status);CHKERRQ(ierr); 438ca161407SBarry Smith ierr = MPI_Waitall(nsends,send_waits,send_status);CHKERRQ(ierr); 439606d414cSSatish Balay ierr = PetscFree(send_status);CHKERRQ(ierr); 4408965ea79SLois Curfman McInnes } 441606d414cSSatish Balay ierr = PetscFree(send_waits);CHKERRQ(ierr); 442606d414cSSatish Balay ierr = PetscFree(svalues);CHKERRQ(ierr); 4438965ea79SLois Curfman McInnes 4443a40ed3dSBarry Smith PetscFunctionReturn(0); 4458965ea79SLois Curfman McInnes } 4468965ea79SLois Curfman McInnes 4474a2ae208SSatish Balay #undef __FUNCT__ 4484a2ae208SSatish Balay #define __FUNCT__ "MatMult_MPIDense" 449dfbe8321SBarry Smith PetscErrorCode MatMult_MPIDense(Mat mat,Vec xx,Vec yy) 4508965ea79SLois Curfman McInnes { 45139ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 452dfbe8321SBarry Smith PetscErrorCode ierr; 453c456f294SBarry Smith 4543a40ed3dSBarry Smith PetscFunctionBegin; 455ca9f406cSSatish Balay ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 456ca9f406cSSatish Balay ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 45744cd7ae7SLois Curfman McInnes ierr = MatMult_SeqDense(mdn->A,mdn->lvec,yy);CHKERRQ(ierr); 4583a40ed3dSBarry Smith PetscFunctionReturn(0); 4598965ea79SLois Curfman McInnes } 4608965ea79SLois Curfman McInnes 4614a2ae208SSatish Balay #undef __FUNCT__ 4624a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_MPIDense" 463dfbe8321SBarry Smith PetscErrorCode MatMultAdd_MPIDense(Mat mat,Vec xx,Vec yy,Vec zz) 4648965ea79SLois Curfman McInnes { 46539ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 466dfbe8321SBarry Smith PetscErrorCode ierr; 467c456f294SBarry Smith 4683a40ed3dSBarry Smith PetscFunctionBegin; 469ca9f406cSSatish Balay ierr = VecScatterBegin(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 470ca9f406cSSatish Balay ierr = VecScatterEnd(mdn->Mvctx,xx,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 47144cd7ae7SLois Curfman McInnes ierr = MatMultAdd_SeqDense(mdn->A,mdn->lvec,yy,zz);CHKERRQ(ierr); 4723a40ed3dSBarry Smith PetscFunctionReturn(0); 4738965ea79SLois Curfman McInnes } 4748965ea79SLois Curfman McInnes 4754a2ae208SSatish Balay #undef __FUNCT__ 4764a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_MPIDense" 477dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_MPIDense(Mat A,Vec xx,Vec yy) 478096963f5SLois Curfman McInnes { 479096963f5SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 480dfbe8321SBarry Smith PetscErrorCode ierr; 48187828ca2SBarry Smith PetscScalar zero = 0.0; 482096963f5SLois Curfman McInnes 4833a40ed3dSBarry Smith PetscFunctionBegin; 4842dcb1b2aSMatthew Knepley ierr = VecSet(yy,zero);CHKERRQ(ierr); 4857c922b88SBarry Smith ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr); 486ca9f406cSSatish Balay ierr = VecScatterBegin(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 487ca9f406cSSatish Balay ierr = VecScatterEnd(a->Mvctx,a->lvec,yy,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 4883a40ed3dSBarry Smith PetscFunctionReturn(0); 489096963f5SLois Curfman McInnes } 490096963f5SLois Curfman McInnes 4914a2ae208SSatish Balay #undef __FUNCT__ 4924a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_MPIDense" 493dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_MPIDense(Mat A,Vec xx,Vec yy,Vec zz) 494096963f5SLois Curfman McInnes { 495096963f5SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 496dfbe8321SBarry Smith PetscErrorCode ierr; 497096963f5SLois Curfman McInnes 4983a40ed3dSBarry Smith PetscFunctionBegin; 4993501a2bdSLois Curfman McInnes ierr = VecCopy(yy,zz);CHKERRQ(ierr); 5007c922b88SBarry Smith ierr = MatMultTranspose_SeqDense(a->A,xx,a->lvec);CHKERRQ(ierr); 501ca9f406cSSatish Balay ierr = VecScatterBegin(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 502ca9f406cSSatish Balay ierr = VecScatterEnd(a->Mvctx,a->lvec,zz,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 5033a40ed3dSBarry Smith PetscFunctionReturn(0); 504096963f5SLois Curfman McInnes } 505096963f5SLois Curfman McInnes 5064a2ae208SSatish Balay #undef __FUNCT__ 5074a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_MPIDense" 508dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_MPIDense(Mat A,Vec v) 5098965ea79SLois Curfman McInnes { 51039ddd567SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 511096963f5SLois Curfman McInnes Mat_SeqDense *aloc = (Mat_SeqDense*)a->A->data; 512dfbe8321SBarry Smith PetscErrorCode ierr; 513d0f46423SBarry Smith PetscInt len,i,n,m = A->rmap->n,radd; 51487828ca2SBarry Smith PetscScalar *x,zero = 0.0; 515ed3cc1f0SBarry Smith 5163a40ed3dSBarry Smith PetscFunctionBegin; 5172dcb1b2aSMatthew Knepley ierr = VecSet(v,zero);CHKERRQ(ierr); 5181ebc52fbSHong Zhang ierr = VecGetArray(v,&x);CHKERRQ(ierr); 519096963f5SLois Curfman McInnes ierr = VecGetSize(v,&n);CHKERRQ(ierr); 520d0f46423SBarry Smith if (n != A->rmap->N) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec"); 521d0f46423SBarry Smith len = PetscMin(a->A->rmap->n,a->A->cmap->n); 522d0f46423SBarry Smith radd = A->rmap->rstart*m; 52344cd7ae7SLois Curfman McInnes for (i=0; i<len; i++) { 524096963f5SLois Curfman McInnes x[i] = aloc->v[radd + i*m + i]; 525096963f5SLois Curfman McInnes } 5261ebc52fbSHong Zhang ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 5273a40ed3dSBarry Smith PetscFunctionReturn(0); 5288965ea79SLois Curfman McInnes } 5298965ea79SLois Curfman McInnes 5304a2ae208SSatish Balay #undef __FUNCT__ 5314a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_MPIDense" 532dfbe8321SBarry Smith PetscErrorCode MatDestroy_MPIDense(Mat mat) 5338965ea79SLois Curfman McInnes { 5343501a2bdSLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 535dfbe8321SBarry Smith PetscErrorCode ierr; 53601b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 53701b82886SBarry Smith Mat_Plapack *lu=(Mat_Plapack*)(mat->spptr); 53801b82886SBarry Smith #endif 539ed3cc1f0SBarry Smith 5403a40ed3dSBarry Smith PetscFunctionBegin; 54194d884c6SBarry Smith 542aa482453SBarry Smith #if defined(PETSC_USE_LOG) 543d0f46423SBarry Smith PetscLogObjectState((PetscObject)mat,"Rows=%D, Cols=%D",mat->rmap->N,mat->cmap->N); 5448965ea79SLois Curfman McInnes #endif 5458798bf22SSatish Balay ierr = MatStashDestroy_Private(&mat->stash);CHKERRQ(ierr); 5463501a2bdSLois Curfman McInnes ierr = MatDestroy(mdn->A);CHKERRQ(ierr); 54705b42c5fSBarry Smith if (mdn->lvec) {ierr = VecDestroy(mdn->lvec);CHKERRQ(ierr);} 54805b42c5fSBarry Smith if (mdn->Mvctx) {ierr = VecScatterDestroy(mdn->Mvctx);CHKERRQ(ierr);} 54901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 5502fbe02b9SBarry Smith if (lu) { 55162b4c0b3SBarry Smith ierr = PLA_Obj_free(&lu->A);CHKERRQ(ierr); 55262b4c0b3SBarry Smith ierr = PLA_Obj_free (&lu->pivots);CHKERRQ(ierr); 55362b4c0b3SBarry Smith ierr = PLA_Temp_free(&lu->templ);CHKERRQ(ierr); 55409d27a7eSBarry Smith 5552fbe02b9SBarry Smith if (lu->is_pla) { 55601b82886SBarry Smith ierr = ISDestroy(lu->is_pla);CHKERRQ(ierr); 55701b82886SBarry Smith ierr = ISDestroy(lu->is_petsc);CHKERRQ(ierr); 55801b82886SBarry Smith ierr = VecScatterDestroy(lu->ctx);CHKERRQ(ierr); 559622d7880SLois Curfman McInnes } 5602fbe02b9SBarry Smith } 56101b82886SBarry Smith #endif 56201b82886SBarry Smith 563606d414cSSatish Balay ierr = PetscFree(mdn);CHKERRQ(ierr); 564dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr); 565901853e0SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C","",PETSC_NULL);CHKERRQ(ierr); 566901853e0SKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 5674ae313f4SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr); 5684ae313f4SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr); 5694ae313f4SHong Zhang ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C","",PETSC_NULL);CHKERRQ(ierr); 5703a40ed3dSBarry Smith PetscFunctionReturn(0); 5718965ea79SLois Curfman McInnes } 57239ddd567SLois Curfman McInnes 5734a2ae208SSatish Balay #undef __FUNCT__ 5744a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_Binary" 5756849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_Binary(Mat mat,PetscViewer viewer) 5768965ea79SLois Curfman McInnes { 57739ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 578dfbe8321SBarry Smith PetscErrorCode ierr; 579aa05aa95SBarry Smith PetscViewerFormat format; 580aa05aa95SBarry Smith int fd; 581d0f46423SBarry Smith PetscInt header[4],mmax,N = mat->cmap->N,i,j,m,k; 582aa05aa95SBarry Smith PetscMPIInt rank,tag = ((PetscObject)viewer)->tag,size; 583578230a0SSatish Balay PetscScalar *work,*v,*vv; 584aa05aa95SBarry Smith Mat_SeqDense *a = (Mat_SeqDense*)mdn->A->data; 585aa05aa95SBarry Smith MPI_Status status; 5867056b6fcSBarry Smith 5873a40ed3dSBarry Smith PetscFunctionBegin; 58839ddd567SLois Curfman McInnes if (mdn->size == 1) { 58939ddd567SLois Curfman McInnes ierr = MatView(mdn->A,viewer);CHKERRQ(ierr); 590aa05aa95SBarry Smith } else { 591aa05aa95SBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 5927adad957SLisandro Dalcin ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&rank);CHKERRQ(ierr); 5937adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)mat)->comm,&size);CHKERRQ(ierr); 594aa05aa95SBarry Smith 595aa05aa95SBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 5967973ad04SBarry Smith if (format == PETSC_VIEWER_NATIVE) { 597aa05aa95SBarry Smith 598aa05aa95SBarry Smith if (!rank) { 599aa05aa95SBarry Smith /* store the matrix as a dense matrix */ 600aa05aa95SBarry Smith header[0] = MAT_FILE_COOKIE; 601d0f46423SBarry Smith header[1] = mat->rmap->N; 602aa05aa95SBarry Smith header[2] = N; 603aa05aa95SBarry Smith header[3] = MATRIX_BINARY_FORMAT_DENSE; 604aa05aa95SBarry Smith ierr = PetscBinaryWrite(fd,header,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr); 605aa05aa95SBarry Smith 606aa05aa95SBarry Smith /* get largest work array needed for transposing array */ 607d0f46423SBarry Smith mmax = mat->rmap->n; 608aa05aa95SBarry Smith for (i=1; i<size; i++) { 609d0f46423SBarry Smith mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]); 6108965ea79SLois Curfman McInnes } 611aa05aa95SBarry Smith ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&work);CHKERRQ(ierr); 612aa05aa95SBarry Smith 613aa05aa95SBarry Smith /* write out local array, by rows */ 614d0f46423SBarry Smith m = mat->rmap->n; 615aa05aa95SBarry Smith v = a->v; 616aa05aa95SBarry Smith for (j=0; j<N; j++) { 617aa05aa95SBarry Smith for (i=0; i<m; i++) { 618578230a0SSatish Balay work[j + i*N] = *v++; 619aa05aa95SBarry Smith } 620aa05aa95SBarry Smith } 621aa05aa95SBarry Smith ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 622aa05aa95SBarry Smith /* get largest work array to receive messages from other processes, excludes process zero */ 623aa05aa95SBarry Smith mmax = 0; 624aa05aa95SBarry Smith for (i=1; i<size; i++) { 625d0f46423SBarry Smith mmax = PetscMax(mmax,mat->rmap->range[i+1] - mat->rmap->range[i]); 626aa05aa95SBarry Smith } 627578230a0SSatish Balay ierr = PetscMalloc(mmax*N*sizeof(PetscScalar),&vv);CHKERRQ(ierr); 628aa05aa95SBarry Smith for(k = 1; k < size; k++) { 629f8009846SMatthew Knepley v = vv; 630d0f46423SBarry Smith m = mat->rmap->range[k+1] - mat->rmap->range[k]; 6317adad957SLisandro Dalcin ierr = MPI_Recv(v,m*N,MPIU_SCALAR,k,tag,((PetscObject)mat)->comm,&status);CHKERRQ(ierr); 632aa05aa95SBarry Smith 633aa05aa95SBarry Smith for(j = 0; j < N; j++) { 634aa05aa95SBarry Smith for(i = 0; i < m; i++) { 635578230a0SSatish Balay work[j + i*N] = *v++; 636aa05aa95SBarry Smith } 637aa05aa95SBarry Smith } 638aa05aa95SBarry Smith ierr = PetscBinaryWrite(fd,work,m*N,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr); 639aa05aa95SBarry Smith } 640aa05aa95SBarry Smith ierr = PetscFree(work);CHKERRQ(ierr); 641578230a0SSatish Balay ierr = PetscFree(vv);CHKERRQ(ierr); 642aa05aa95SBarry Smith } else { 643d0f46423SBarry Smith ierr = MPI_Send(a->v,mat->rmap->n*mat->cmap->N,MPIU_SCALAR,0,tag,((PetscObject)mat)->comm);CHKERRQ(ierr); 644aa05aa95SBarry Smith } 6454aa34b0aSBarry Smith } else { 6467973ad04SBarry Smith SETERRQ(PETSC_ERR_SUP,"To store a parallel dense matrix you must first call PetscViewerSetFormat(viewer,PETSC_VIEWER_NATIVE"); 647aa05aa95SBarry Smith } 648aa05aa95SBarry Smith } 6493a40ed3dSBarry Smith PetscFunctionReturn(0); 6508965ea79SLois Curfman McInnes } 6518965ea79SLois Curfman McInnes 6524a2ae208SSatish Balay #undef __FUNCT__ 6534a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense_ASCIIorDraworSocket" 6546849ba73SBarry Smith static PetscErrorCode MatView_MPIDense_ASCIIorDraworSocket(Mat mat,PetscViewer viewer) 6558965ea79SLois Curfman McInnes { 65639ddd567SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)mat->data; 657dfbe8321SBarry Smith PetscErrorCode ierr; 65832dcc486SBarry Smith PetscMPIInt size = mdn->size,rank = mdn->rank; 659a313700dSBarry Smith const PetscViewerType vtype; 66032077d6dSBarry Smith PetscTruth iascii,isdraw; 661b0a32e0cSBarry Smith PetscViewer sviewer; 662f3ef73ceSBarry Smith PetscViewerFormat format; 66301b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 6645d00a290SHong Zhang Mat_Plapack *lu; 66501b82886SBarry Smith #endif 6668965ea79SLois Curfman McInnes 6673a40ed3dSBarry Smith PetscFunctionBegin; 66832077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 669fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 67032077d6dSBarry Smith if (iascii) { 671b0a32e0cSBarry Smith ierr = PetscViewerGetType(viewer,&vtype);CHKERRQ(ierr); 672b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 673456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 6744e220ebcSLois Curfman McInnes MatInfo info; 675888f2ed8SSatish Balay ierr = MatGetInfo(mat,MAT_LOCAL,&info);CHKERRQ(ierr); 676d0f46423SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer," [%d] local rows %D nz %D nz alloced %D mem %D \n",rank,mat->rmap->n, 67777431f27SBarry Smith (PetscInt)info.nz_used,(PetscInt)info.nz_allocated,(PetscInt)info.memory);CHKERRQ(ierr); 678b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 67901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 68001b82886SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"PLAPACK run parameters:\n");CHKERRQ(ierr); 6815d00a290SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Processor mesh: nprows %d, npcols %d\n",Plapack_nprows, Plapack_npcols);CHKERRQ(ierr); 6825d00a290SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Error checking: %d\n",Plapack_ierror);CHKERRQ(ierr); 6835d00a290SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," Algorithmic block size: %d\n",Plapack_nb_alg);CHKERRQ(ierr); 6845d00a290SHong Zhang if (mat->factor){ 6855d00a290SHong Zhang lu=(Mat_Plapack*)(mat->spptr); 68601b82886SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Distr. block size nb: %d \n",lu->nb);CHKERRQ(ierr); 6875d00a290SHong Zhang } 6885d00a290SHong Zhang #else 6895d00a290SHong Zhang ierr = VecScatterView(mdn->Mvctx,viewer);CHKERRQ(ierr); 69001b82886SBarry Smith #endif 6913a40ed3dSBarry Smith PetscFunctionReturn(0); 692fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_INFO) { 6933a40ed3dSBarry Smith PetscFunctionReturn(0); 6948965ea79SLois Curfman McInnes } 695f1af5d2fSBarry Smith } else if (isdraw) { 696b0a32e0cSBarry Smith PetscDraw draw; 697f1af5d2fSBarry Smith PetscTruth isnull; 698f1af5d2fSBarry Smith 699b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 700b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); 701f1af5d2fSBarry Smith if (isnull) PetscFunctionReturn(0); 702f1af5d2fSBarry Smith } 70377ed5343SBarry Smith 7048965ea79SLois Curfman McInnes if (size == 1) { 70539ddd567SLois Curfman McInnes ierr = MatView(mdn->A,viewer);CHKERRQ(ierr); 7063a40ed3dSBarry Smith } else { 7078965ea79SLois Curfman McInnes /* assemble the entire matrix onto first processor. */ 7088965ea79SLois Curfman McInnes Mat A; 709d0f46423SBarry Smith PetscInt M = mat->rmap->N,N = mat->cmap->N,m,row,i,nz; 710ba8c8a56SBarry Smith PetscInt *cols; 711ba8c8a56SBarry Smith PetscScalar *vals; 7128965ea79SLois Curfman McInnes 7137adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)mat)->comm,&A);CHKERRQ(ierr); 7148965ea79SLois Curfman McInnes if (!rank) { 715f69a0ea3SMatthew Knepley ierr = MatSetSizes(A,M,N,M,N);CHKERRQ(ierr); 7163a40ed3dSBarry Smith } else { 717f69a0ea3SMatthew Knepley ierr = MatSetSizes(A,0,0,M,N);CHKERRQ(ierr); 7188965ea79SLois Curfman McInnes } 7197adad957SLisandro Dalcin /* Since this is a temporary matrix, MATMPIDENSE instead of ((PetscObject)A)->type_name here is probably acceptable. */ 720878740d9SKris Buschelman ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr); 721878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(A,PETSC_NULL); 72252e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,A);CHKERRQ(ierr); 7238965ea79SLois Curfman McInnes 72439ddd567SLois Curfman McInnes /* Copy the matrix ... This isn't the most efficient means, 72539ddd567SLois Curfman McInnes but it's quick for now */ 72651022da4SBarry Smith A->insertmode = INSERT_VALUES; 727d0f46423SBarry Smith row = mat->rmap->rstart; m = mdn->A->rmap->n; 7288965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 729ba8c8a56SBarry Smith ierr = MatGetRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr); 730ba8c8a56SBarry Smith ierr = MatSetValues_MPIDense(A,1,&row,nz,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 731ba8c8a56SBarry Smith ierr = MatRestoreRow_MPIDense(mat,row,&nz,&cols,&vals);CHKERRQ(ierr); 73239ddd567SLois Curfman McInnes row++; 7338965ea79SLois Curfman McInnes } 7348965ea79SLois Curfman McInnes 7356d4a8577SBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 7366d4a8577SBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 737b0a32e0cSBarry Smith ierr = PetscViewerGetSingleton(viewer,&sviewer);CHKERRQ(ierr); 738b9b97703SBarry Smith if (!rank) { 7396831982aSBarry Smith ierr = MatView(((Mat_MPIDense*)(A->data))->A,sviewer);CHKERRQ(ierr); 7408965ea79SLois Curfman McInnes } 741b0a32e0cSBarry Smith ierr = PetscViewerRestoreSingleton(viewer,&sviewer);CHKERRQ(ierr); 742b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 7438965ea79SLois Curfman McInnes ierr = MatDestroy(A);CHKERRQ(ierr); 7448965ea79SLois Curfman McInnes } 7453a40ed3dSBarry Smith PetscFunctionReturn(0); 7468965ea79SLois Curfman McInnes } 7478965ea79SLois Curfman McInnes 7484a2ae208SSatish Balay #undef __FUNCT__ 7494a2ae208SSatish Balay #define __FUNCT__ "MatView_MPIDense" 750dfbe8321SBarry Smith PetscErrorCode MatView_MPIDense(Mat mat,PetscViewer viewer) 7518965ea79SLois Curfman McInnes { 752dfbe8321SBarry Smith PetscErrorCode ierr; 75332077d6dSBarry Smith PetscTruth iascii,isbinary,isdraw,issocket; 7548965ea79SLois Curfman McInnes 755433994e6SBarry Smith PetscFunctionBegin; 7560f5bd95cSBarry Smith 75732077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 758fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr); 759b0a32e0cSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_SOCKET,&issocket);CHKERRQ(ierr); 760fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 7610f5bd95cSBarry Smith 76232077d6dSBarry Smith if (iascii || issocket || isdraw) { 763f1af5d2fSBarry Smith ierr = MatView_MPIDense_ASCIIorDraworSocket(mat,viewer);CHKERRQ(ierr); 7640f5bd95cSBarry Smith } else if (isbinary) { 7653a40ed3dSBarry Smith ierr = MatView_MPIDense_Binary(mat,viewer);CHKERRQ(ierr); 7665cd90555SBarry Smith } else { 767958c9bccSBarry Smith SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by MPI dense matrix",((PetscObject)viewer)->type_name); 7688965ea79SLois Curfman McInnes } 7693a40ed3dSBarry Smith PetscFunctionReturn(0); 7708965ea79SLois Curfman McInnes } 7718965ea79SLois Curfman McInnes 7724a2ae208SSatish Balay #undef __FUNCT__ 7734a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_MPIDense" 774dfbe8321SBarry Smith PetscErrorCode MatGetInfo_MPIDense(Mat A,MatInfoType flag,MatInfo *info) 7758965ea79SLois Curfman McInnes { 7763501a2bdSLois Curfman McInnes Mat_MPIDense *mat = (Mat_MPIDense*)A->data; 7773501a2bdSLois Curfman McInnes Mat mdn = mat->A; 778dfbe8321SBarry Smith PetscErrorCode ierr; 779329f5518SBarry Smith PetscReal isend[5],irecv[5]; 7808965ea79SLois Curfman McInnes 7813a40ed3dSBarry Smith PetscFunctionBegin; 7824e220ebcSLois Curfman McInnes info->block_size = 1.0; 7834e220ebcSLois Curfman McInnes ierr = MatGetInfo(mdn,MAT_LOCAL,info);CHKERRQ(ierr); 7844e220ebcSLois Curfman McInnes isend[0] = info->nz_used; isend[1] = info->nz_allocated; isend[2] = info->nz_unneeded; 7854e220ebcSLois Curfman McInnes isend[3] = info->memory; isend[4] = info->mallocs; 7868965ea79SLois Curfman McInnes if (flag == MAT_LOCAL) { 7874e220ebcSLois Curfman McInnes info->nz_used = isend[0]; 7884e220ebcSLois Curfman McInnes info->nz_allocated = isend[1]; 7894e220ebcSLois Curfman McInnes info->nz_unneeded = isend[2]; 7904e220ebcSLois Curfman McInnes info->memory = isend[3]; 7914e220ebcSLois Curfman McInnes info->mallocs = isend[4]; 7928965ea79SLois Curfman McInnes } else if (flag == MAT_GLOBAL_MAX) { 7937adad957SLisandro Dalcin ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr); 7944e220ebcSLois Curfman McInnes info->nz_used = irecv[0]; 7954e220ebcSLois Curfman McInnes info->nz_allocated = irecv[1]; 7964e220ebcSLois Curfman McInnes info->nz_unneeded = irecv[2]; 7974e220ebcSLois Curfman McInnes info->memory = irecv[3]; 7984e220ebcSLois Curfman McInnes info->mallocs = irecv[4]; 7998965ea79SLois Curfman McInnes } else if (flag == MAT_GLOBAL_SUM) { 8007adad957SLisandro Dalcin ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 8014e220ebcSLois Curfman McInnes info->nz_used = irecv[0]; 8024e220ebcSLois Curfman McInnes info->nz_allocated = irecv[1]; 8034e220ebcSLois Curfman McInnes info->nz_unneeded = irecv[2]; 8044e220ebcSLois Curfman McInnes info->memory = irecv[3]; 8054e220ebcSLois Curfman McInnes info->mallocs = irecv[4]; 8068965ea79SLois Curfman McInnes } 8074e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; /* no parallel LU/ILU/Cholesky */ 8084e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 8094e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 8103a40ed3dSBarry Smith PetscFunctionReturn(0); 8118965ea79SLois Curfman McInnes } 8128965ea79SLois Curfman McInnes 8134a2ae208SSatish Balay #undef __FUNCT__ 8144a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_MPIDense" 8154e0d8c25SBarry Smith PetscErrorCode MatSetOption_MPIDense(Mat A,MatOption op,PetscTruth flg) 8168965ea79SLois Curfman McInnes { 81739ddd567SLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 818dfbe8321SBarry Smith PetscErrorCode ierr; 8198965ea79SLois Curfman McInnes 8203a40ed3dSBarry Smith PetscFunctionBegin; 82112c028f9SKris Buschelman switch (op) { 822512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 82312c028f9SKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 82412c028f9SKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 8254e0d8c25SBarry Smith ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 82612c028f9SKris Buschelman break; 82712c028f9SKris Buschelman case MAT_ROW_ORIENTED: 8284e0d8c25SBarry Smith a->roworiented = flg; 8294e0d8c25SBarry Smith ierr = MatSetOption(a->A,op,flg);CHKERRQ(ierr); 83012c028f9SKris Buschelman break; 8314e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 83212c028f9SKris Buschelman case MAT_USE_HASH_TABLE: 833290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 83412c028f9SKris Buschelman break; 83512c028f9SKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 8364e0d8c25SBarry Smith a->donotstash = flg; 83712c028f9SKris Buschelman break; 83877e54ba9SKris Buschelman case MAT_SYMMETRIC: 83977e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 8409a4540c5SBarry Smith case MAT_HERMITIAN: 8419a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 842600fe468SBarry Smith case MAT_IGNORE_LOWER_TRIANGULAR: 843290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 84477e54ba9SKris Buschelman break; 84512c028f9SKris Buschelman default: 846600fe468SBarry Smith SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]); 8473a40ed3dSBarry Smith } 8483a40ed3dSBarry Smith PetscFunctionReturn(0); 8498965ea79SLois Curfman McInnes } 8508965ea79SLois Curfman McInnes 8518965ea79SLois Curfman McInnes 8524a2ae208SSatish Balay #undef __FUNCT__ 8534a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_MPIDense" 854dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_MPIDense(Mat A,Vec ll,Vec rr) 8555b2fa520SLois Curfman McInnes { 8565b2fa520SLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)A->data; 8575b2fa520SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)mdn->A->data; 85887828ca2SBarry Smith PetscScalar *l,*r,x,*v; 859dfbe8321SBarry Smith PetscErrorCode ierr; 860d0f46423SBarry Smith PetscInt i,j,s2a,s3a,s2,s3,m=mdn->A->rmap->n,n=mdn->A->cmap->n; 8615b2fa520SLois Curfman McInnes 8625b2fa520SLois Curfman McInnes PetscFunctionBegin; 86372d926a5SLois Curfman McInnes ierr = MatGetLocalSize(A,&s2,&s3);CHKERRQ(ierr); 8645b2fa520SLois Curfman McInnes if (ll) { 86572d926a5SLois Curfman McInnes ierr = VecGetLocalSize(ll,&s2a);CHKERRQ(ierr); 866175be7b4SMatthew Knepley if (s2a != s2) SETERRQ2(PETSC_ERR_ARG_SIZ,"Left scaling vector non-conforming local size, %d != %d.", s2a, s2); 8671ebc52fbSHong Zhang ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 8685b2fa520SLois Curfman McInnes for (i=0; i<m; i++) { 8695b2fa520SLois Curfman McInnes x = l[i]; 8705b2fa520SLois Curfman McInnes v = mat->v + i; 8715b2fa520SLois Curfman McInnes for (j=0; j<n; j++) { (*v) *= x; v+= m;} 8725b2fa520SLois Curfman McInnes } 8731ebc52fbSHong Zhang ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 874efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 8755b2fa520SLois Curfman McInnes } 8765b2fa520SLois Curfman McInnes if (rr) { 877175be7b4SMatthew Knepley ierr = VecGetLocalSize(rr,&s3a);CHKERRQ(ierr); 878175be7b4SMatthew Knepley if (s3a != s3) SETERRQ2(PETSC_ERR_ARG_SIZ,"Right scaling vec non-conforming local size, %d != %d.", s3a, s3); 879ca9f406cSSatish Balay ierr = VecScatterBegin(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 880ca9f406cSSatish Balay ierr = VecScatterEnd(mdn->Mvctx,rr,mdn->lvec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 8811ebc52fbSHong Zhang ierr = VecGetArray(mdn->lvec,&r);CHKERRQ(ierr); 8825b2fa520SLois Curfman McInnes for (i=0; i<n; i++) { 8835b2fa520SLois Curfman McInnes x = r[i]; 8845b2fa520SLois Curfman McInnes v = mat->v + i*m; 8855b2fa520SLois Curfman McInnes for (j=0; j<m; j++) { (*v++) *= x;} 8865b2fa520SLois Curfman McInnes } 8871ebc52fbSHong Zhang ierr = VecRestoreArray(mdn->lvec,&r);CHKERRQ(ierr); 888efee365bSSatish Balay ierr = PetscLogFlops(n*m);CHKERRQ(ierr); 8895b2fa520SLois Curfman McInnes } 8905b2fa520SLois Curfman McInnes PetscFunctionReturn(0); 8915b2fa520SLois Curfman McInnes } 8925b2fa520SLois Curfman McInnes 8934a2ae208SSatish Balay #undef __FUNCT__ 8944a2ae208SSatish Balay #define __FUNCT__ "MatNorm_MPIDense" 895dfbe8321SBarry Smith PetscErrorCode MatNorm_MPIDense(Mat A,NormType type,PetscReal *nrm) 896096963f5SLois Curfman McInnes { 8973501a2bdSLois Curfman McInnes Mat_MPIDense *mdn = (Mat_MPIDense*)A->data; 8983501a2bdSLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense*)mdn->A->data; 899dfbe8321SBarry Smith PetscErrorCode ierr; 90013f74950SBarry Smith PetscInt i,j; 901329f5518SBarry Smith PetscReal sum = 0.0; 90287828ca2SBarry Smith PetscScalar *v = mat->v; 9033501a2bdSLois Curfman McInnes 9043a40ed3dSBarry Smith PetscFunctionBegin; 9053501a2bdSLois Curfman McInnes if (mdn->size == 1) { 906064f8208SBarry Smith ierr = MatNorm(mdn->A,type,nrm);CHKERRQ(ierr); 9073501a2bdSLois Curfman McInnes } else { 9083501a2bdSLois Curfman McInnes if (type == NORM_FROBENIUS) { 909d0f46423SBarry Smith for (i=0; i<mdn->A->cmap->n*mdn->A->rmap->n; i++) { 910aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 911329f5518SBarry Smith sum += PetscRealPart(PetscConj(*v)*(*v)); v++; 9123501a2bdSLois Curfman McInnes #else 9133501a2bdSLois Curfman McInnes sum += (*v)*(*v); v++; 9143501a2bdSLois Curfman McInnes #endif 9153501a2bdSLois Curfman McInnes } 9167adad957SLisandro Dalcin ierr = MPI_Allreduce(&sum,nrm,1,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 917064f8208SBarry Smith *nrm = sqrt(*nrm); 918dc0b31edSSatish Balay ierr = PetscLogFlops(2.0*mdn->A->cmap->n*mdn->A->rmap->n);CHKERRQ(ierr); 9193a40ed3dSBarry Smith } else if (type == NORM_1) { 920329f5518SBarry Smith PetscReal *tmp,*tmp2; 921d0f46423SBarry Smith ierr = PetscMalloc(2*A->cmap->N*sizeof(PetscReal),&tmp);CHKERRQ(ierr); 922d0f46423SBarry Smith tmp2 = tmp + A->cmap->N; 923d0f46423SBarry Smith ierr = PetscMemzero(tmp,2*A->cmap->N*sizeof(PetscReal));CHKERRQ(ierr); 924064f8208SBarry Smith *nrm = 0.0; 9253501a2bdSLois Curfman McInnes v = mat->v; 926d0f46423SBarry Smith for (j=0; j<mdn->A->cmap->n; j++) { 927d0f46423SBarry Smith for (i=0; i<mdn->A->rmap->n; i++) { 92867e560aaSBarry Smith tmp[j] += PetscAbsScalar(*v); v++; 9293501a2bdSLois Curfman McInnes } 9303501a2bdSLois Curfman McInnes } 931d0f46423SBarry Smith ierr = MPI_Allreduce(tmp,tmp2,A->cmap->N,MPIU_REAL,MPI_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 932d0f46423SBarry Smith for (j=0; j<A->cmap->N; j++) { 933064f8208SBarry Smith if (tmp2[j] > *nrm) *nrm = tmp2[j]; 9343501a2bdSLois Curfman McInnes } 935606d414cSSatish Balay ierr = PetscFree(tmp);CHKERRQ(ierr); 936d0f46423SBarry Smith ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr); 9373a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { /* max row norm */ 938329f5518SBarry Smith PetscReal ntemp; 9393501a2bdSLois Curfman McInnes ierr = MatNorm(mdn->A,type,&ntemp);CHKERRQ(ierr); 9407adad957SLisandro Dalcin ierr = MPI_Allreduce(&ntemp,nrm,1,MPIU_REAL,MPI_MAX,((PetscObject)A)->comm);CHKERRQ(ierr); 9413a40ed3dSBarry Smith } else { 94229bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"No support for two norm"); 9433501a2bdSLois Curfman McInnes } 9443501a2bdSLois Curfman McInnes } 9453a40ed3dSBarry Smith PetscFunctionReturn(0); 9463501a2bdSLois Curfman McInnes } 9473501a2bdSLois Curfman McInnes 9484a2ae208SSatish Balay #undef __FUNCT__ 9494a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_MPIDense" 950fc4dec0aSBarry Smith PetscErrorCode MatTranspose_MPIDense(Mat A,MatReuse reuse,Mat *matout) 9513501a2bdSLois Curfman McInnes { 9523501a2bdSLois Curfman McInnes Mat_MPIDense *a = (Mat_MPIDense*)A->data; 9533501a2bdSLois Curfman McInnes Mat_SeqDense *Aloc = (Mat_SeqDense*)a->A->data; 9543501a2bdSLois Curfman McInnes Mat B; 955d0f46423SBarry Smith PetscInt M = A->rmap->N,N = A->cmap->N,m,n,*rwork,rstart = A->rmap->rstart; 9566849ba73SBarry Smith PetscErrorCode ierr; 95713f74950SBarry Smith PetscInt j,i; 95887828ca2SBarry Smith PetscScalar *v; 9593501a2bdSLois Curfman McInnes 9603a40ed3dSBarry Smith PetscFunctionBegin; 961e9695a30SBarry Smith if (reuse == MAT_REUSE_MATRIX && A == *matout && M != N) SETERRQ(PETSC_ERR_SUP,"Supports square matrix only in-place"); 962fc4dec0aSBarry Smith if (reuse == MAT_INITIAL_MATRIX || A == *matout) { 9637adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 964d0f46423SBarry Smith ierr = MatSetSizes(B,A->cmap->n,A->rmap->n,N,M);CHKERRQ(ierr); 9657adad957SLisandro Dalcin ierr = MatSetType(B,((PetscObject)A)->type_name);CHKERRQ(ierr); 966878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(B,PETSC_NULL);CHKERRQ(ierr); 967fc4dec0aSBarry Smith } else { 968fc4dec0aSBarry Smith B = *matout; 969fc4dec0aSBarry Smith } 9703501a2bdSLois Curfman McInnes 971d0f46423SBarry Smith m = a->A->rmap->n; n = a->A->cmap->n; v = Aloc->v; 9721acff37aSSatish Balay ierr = PetscMalloc(m*sizeof(PetscInt),&rwork);CHKERRQ(ierr); 9733501a2bdSLois Curfman McInnes for (i=0; i<m; i++) rwork[i] = rstart + i; 9741acff37aSSatish Balay for (j=0; j<n; j++) { 9753501a2bdSLois Curfman McInnes ierr = MatSetValues(B,1,&j,m,rwork,v,INSERT_VALUES);CHKERRQ(ierr); 9763501a2bdSLois Curfman McInnes v += m; 9773501a2bdSLois Curfman McInnes } 978606d414cSSatish Balay ierr = PetscFree(rwork);CHKERRQ(ierr); 9796d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 9806d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 981815cbec1SBarry Smith if (reuse == MAT_INITIAL_MATRIX || *matout != A) { 9823501a2bdSLois Curfman McInnes *matout = B; 9833501a2bdSLois Curfman McInnes } else { 984273d9f13SBarry Smith ierr = MatHeaderCopy(A,B);CHKERRQ(ierr); 9853501a2bdSLois Curfman McInnes } 9863a40ed3dSBarry Smith PetscFunctionReturn(0); 987096963f5SLois Curfman McInnes } 988096963f5SLois Curfman McInnes 989d9eff348SSatish Balay #include "petscblaslapack.h" 9904a2ae208SSatish Balay #undef __FUNCT__ 9914a2ae208SSatish Balay #define __FUNCT__ "MatScale_MPIDense" 992f4df32b1SMatthew Knepley PetscErrorCode MatScale_MPIDense(Mat inA,PetscScalar alpha) 99344cd7ae7SLois Curfman McInnes { 99444cd7ae7SLois Curfman McInnes Mat_MPIDense *A = (Mat_MPIDense*)inA->data; 99544cd7ae7SLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*)A->A->data; 996f4df32b1SMatthew Knepley PetscScalar oalpha = alpha; 997efee365bSSatish Balay PetscErrorCode ierr; 998d0f46423SBarry Smith PetscBLASInt one = 1,nz = PetscBLASIntCast(inA->rmap->n*inA->cmap->N); 99944cd7ae7SLois Curfman McInnes 10003a40ed3dSBarry Smith PetscFunctionBegin; 1001f4df32b1SMatthew Knepley BLASscal_(&nz,&oalpha,a->v,&one); 1002efee365bSSatish Balay ierr = PetscLogFlops(nz);CHKERRQ(ierr); 10033a40ed3dSBarry Smith PetscFunctionReturn(0); 100444cd7ae7SLois Curfman McInnes } 100544cd7ae7SLois Curfman McInnes 10066849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat,MatDuplicateOption,Mat *); 10078965ea79SLois Curfman McInnes 10084a2ae208SSatish Balay #undef __FUNCT__ 10094a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_MPIDense" 1010dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_MPIDense(Mat A) 1011273d9f13SBarry Smith { 1012dfbe8321SBarry Smith PetscErrorCode ierr; 1013273d9f13SBarry Smith 1014273d9f13SBarry Smith PetscFunctionBegin; 1015273d9f13SBarry Smith ierr = MatMPIDenseSetPreallocation(A,0);CHKERRQ(ierr); 1016273d9f13SBarry Smith PetscFunctionReturn(0); 1017273d9f13SBarry Smith } 1018273d9f13SBarry Smith 101901b82886SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 1020eae6fb2eSBarry Smith 1021eae6fb2eSBarry Smith #undef __FUNCT__ 1022eae6fb2eSBarry Smith #define __FUNCT__ "MatMPIDenseCopyToPlapack" 1023eae6fb2eSBarry Smith PetscErrorCode MatMPIDenseCopyToPlapack(Mat A,Mat F) 1024eae6fb2eSBarry Smith { 1025eae6fb2eSBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 1026eae6fb2eSBarry Smith PetscErrorCode ierr; 1027d0f46423SBarry Smith PetscInt M=A->cmap->N,m=A->rmap->n,rstart; 1028eae6fb2eSBarry Smith PetscScalar *array; 1029eae6fb2eSBarry Smith PetscReal one = 1.0; 1030eae6fb2eSBarry Smith 1031eae6fb2eSBarry Smith PetscFunctionBegin; 10322fbe02b9SBarry Smith /* Copy A into F->lu->A */ 1033eae6fb2eSBarry Smith ierr = PLA_Obj_set_to_zero(lu->A);CHKERRQ(ierr); 1034eae6fb2eSBarry Smith ierr = PLA_API_begin();CHKERRQ(ierr); 1035eae6fb2eSBarry Smith ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr); 103679b0a62dSBarry Smith ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr); 1037eae6fb2eSBarry Smith ierr = MatGetArray(A,&array);CHKERRQ(ierr); 1038eae6fb2eSBarry Smith ierr = PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0);CHKERRQ(ierr); 1039eae6fb2eSBarry Smith ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 1040eae6fb2eSBarry Smith ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr); 1041eae6fb2eSBarry Smith ierr = PLA_API_end();CHKERRQ(ierr); 1042eae6fb2eSBarry Smith lu->rstart = rstart; 1043eae6fb2eSBarry Smith PetscFunctionReturn(0); 1044eae6fb2eSBarry Smith } 1045eae6fb2eSBarry Smith 104601b82886SBarry Smith #undef __FUNCT__ 10472fbe02b9SBarry Smith #define __FUNCT__ "MatMPIDenseCopyFromPlapack" 10482fbe02b9SBarry Smith PetscErrorCode MatMPIDenseCopyFromPlapack(Mat F,Mat A) 10492fbe02b9SBarry Smith { 10502fbe02b9SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 10512fbe02b9SBarry Smith PetscErrorCode ierr; 1052d0f46423SBarry Smith PetscInt M=A->cmap->N,m=A->rmap->n,rstart; 10532fbe02b9SBarry Smith PetscScalar *array; 10542fbe02b9SBarry Smith PetscReal one = 1.0; 10552fbe02b9SBarry Smith 10562fbe02b9SBarry Smith PetscFunctionBegin; 10572fbe02b9SBarry Smith /* Copy F into A->lu->A */ 105879b0a62dSBarry Smith ierr = MatZeroEntries(A);CHKERRQ(ierr); 10592fbe02b9SBarry Smith ierr = PLA_API_begin();CHKERRQ(ierr); 10602fbe02b9SBarry Smith ierr = PLA_Obj_API_open(lu->A);CHKERRQ(ierr); 10612fbe02b9SBarry Smith ierr = MatGetOwnershipRange(A,&rstart,PETSC_NULL);CHKERRQ(ierr); 10622fbe02b9SBarry Smith ierr = MatGetArray(A,&array);CHKERRQ(ierr); 10632fbe02b9SBarry Smith ierr = PLA_API_axpy_global_to_matrix(m,M, &one,lu->A,rstart,0,(void *)array,m);CHKERRQ(ierr); 10642fbe02b9SBarry Smith ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 10652fbe02b9SBarry Smith ierr = PLA_Obj_API_close(lu->A);CHKERRQ(ierr); 10662fbe02b9SBarry Smith ierr = PLA_API_end();CHKERRQ(ierr); 10672fbe02b9SBarry Smith lu->rstart = rstart; 10682fbe02b9SBarry Smith PetscFunctionReturn(0); 10692fbe02b9SBarry Smith } 10702fbe02b9SBarry Smith 10712fbe02b9SBarry Smith #undef __FUNCT__ 10722fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultNumeric_MPIDense_MPIDense" 10732fbe02b9SBarry Smith PetscErrorCode MatMatMultNumeric_MPIDense_MPIDense(Mat A,Mat B,Mat C) 10742fbe02b9SBarry Smith { 10752fbe02b9SBarry Smith PetscErrorCode ierr; 10762fbe02b9SBarry Smith Mat_Plapack *luA = (Mat_Plapack*)A->spptr; 10772fbe02b9SBarry Smith Mat_Plapack *luB = (Mat_Plapack*)B->spptr; 10782fbe02b9SBarry Smith Mat_Plapack *luC = (Mat_Plapack*)C->spptr; 10792fbe02b9SBarry Smith PLA_Obj alpha = NULL,beta = NULL; 10802fbe02b9SBarry Smith 10812fbe02b9SBarry Smith PetscFunctionBegin; 10822fbe02b9SBarry Smith ierr = MatMPIDenseCopyToPlapack(A,A);CHKERRQ(ierr); 10832fbe02b9SBarry Smith ierr = MatMPIDenseCopyToPlapack(B,B);CHKERRQ(ierr); 10842fbe02b9SBarry Smith 10856e6f9017SBarry Smith /* 1086cb2480eaSBarry Smith ierr = PLA_Global_show("A = ",luA->A,"%g ","");CHKERRQ(ierr); 1087cb2480eaSBarry Smith ierr = PLA_Global_show("B = ",luB->A,"%g ","");CHKERRQ(ierr); 10886e6f9017SBarry Smith */ 1089cb2480eaSBarry Smith 10902fbe02b9SBarry Smith /* do the multiply in PLA */ 109179b0a62dSBarry Smith ierr = PLA_Create_constants_conf_to(luA->A,NULL,NULL,&alpha);CHKERRQ(ierr); 109279b0a62dSBarry Smith ierr = PLA_Create_constants_conf_to(luC->A,NULL,&beta,NULL);CHKERRQ(ierr); 109379b0a62dSBarry Smith CHKMEMQ; 10942fbe02b9SBarry Smith 1095f0e3846dSSatish Balay ierr = PLA_Gemm(PLA_NO_TRANSPOSE,PLA_NO_TRANSPOSE,alpha,luA->A,luB->A,beta,luC->A); /* CHKERRQ(ierr); */ 109679b0a62dSBarry Smith CHKMEMQ; 10972fbe02b9SBarry Smith ierr = PLA_Obj_free(&alpha);CHKERRQ(ierr); 10982fbe02b9SBarry Smith ierr = PLA_Obj_free(&beta);CHKERRQ(ierr); 10992fbe02b9SBarry Smith 11006e6f9017SBarry Smith /* 1101cb2480eaSBarry Smith ierr = PLA_Global_show("C = ",luC->A,"%g ","");CHKERRQ(ierr); 11026e6f9017SBarry Smith */ 11032fbe02b9SBarry Smith ierr = MatMPIDenseCopyFromPlapack(C,C);CHKERRQ(ierr); 11042fbe02b9SBarry Smith PetscFunctionReturn(0); 11052fbe02b9SBarry Smith } 11062fbe02b9SBarry Smith 11072fbe02b9SBarry Smith #undef __FUNCT__ 11082fbe02b9SBarry Smith #define __FUNCT__ "MatMatMultSymbolic_MPIDense_MPIDense" 11092fbe02b9SBarry Smith PetscErrorCode MatMatMultSymbolic_MPIDense_MPIDense(Mat A,Mat B,PetscReal fill,Mat *C) 11102fbe02b9SBarry Smith { 11112fbe02b9SBarry Smith PetscErrorCode ierr; 1112d0f46423SBarry Smith PetscInt m=A->rmap->n,n=B->cmap->n; 11132fbe02b9SBarry Smith Mat Cmat; 11142fbe02b9SBarry Smith 11152fbe02b9SBarry Smith PetscFunctionBegin; 1116d0f46423SBarry Smith if (A->cmap->n != B->rmap->n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->cmap->n %d != B->rmap->n %d\n",A->cmap->n,B->rmap->n); 11176e6f9017SBarry Smith SETERRQ(PETSC_ERR_LIB,"Due to aparent bugs in PLAPACK,this is not currently supported"); 11182fbe02b9SBarry Smith ierr = MatCreate(((PetscObject)B)->comm,&Cmat);CHKERRQ(ierr); 1119d0f46423SBarry Smith ierr = MatSetSizes(Cmat,m,n,A->rmap->N,B->cmap->N);CHKERRQ(ierr); 11202fbe02b9SBarry Smith ierr = MatSetType(Cmat,MATMPIDENSE);CHKERRQ(ierr); 11212fbe02b9SBarry Smith ierr = MatAssemblyBegin(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11222fbe02b9SBarry Smith ierr = MatAssemblyEnd(Cmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 112385bd4cefSHong Zhang 11242fbe02b9SBarry Smith *C = Cmat; 11252fbe02b9SBarry Smith PetscFunctionReturn(0); 11262fbe02b9SBarry Smith } 11272fbe02b9SBarry Smith 11282fbe02b9SBarry Smith #undef __FUNCT__ 11292fbe02b9SBarry Smith #define __FUNCT__ "MatMatMult_MPIDense_MPIDense" 11302fbe02b9SBarry Smith PetscErrorCode MatMatMult_MPIDense_MPIDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 11312fbe02b9SBarry Smith { 11322fbe02b9SBarry Smith PetscErrorCode ierr; 11332fbe02b9SBarry Smith 11342fbe02b9SBarry Smith PetscFunctionBegin; 11352fbe02b9SBarry Smith if (scall == MAT_INITIAL_MATRIX){ 11362fbe02b9SBarry Smith ierr = MatMatMultSymbolic_MPIDense_MPIDense(A,B,fill,C);CHKERRQ(ierr); 11372fbe02b9SBarry Smith } 11382fbe02b9SBarry Smith ierr = MatMatMultNumeric_MPIDense_MPIDense(A,B,*C);CHKERRQ(ierr); 11392fbe02b9SBarry Smith PetscFunctionReturn(0); 11402fbe02b9SBarry Smith } 11412fbe02b9SBarry Smith 1142ab235cb6SHong Zhang #undef __FUNCT__ 11436c995c7dSHong Zhang #define __FUNCT__ "MatSolve_MPIDense" 11446c995c7dSHong Zhang PetscErrorCode MatSolve_MPIDense(Mat A,Vec b,Vec x) 11456c995c7dSHong Zhang { 11466c995c7dSHong Zhang MPI_Comm comm = ((PetscObject)A)->comm; 11476c995c7dSHong Zhang Mat_Plapack *lu = (Mat_Plapack*)A->spptr; 11486c995c7dSHong Zhang PetscErrorCode ierr; 1149d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,i,j,*idx_pla,*idx_petsc,loc_m,loc_stride; 11506c995c7dSHong Zhang PetscScalar *array; 11516c995c7dSHong Zhang PetscReal one = 1.0; 11526c995c7dSHong Zhang PetscMPIInt size,rank,r_rank,r_nproc,c_rank,c_nproc;; 11536c995c7dSHong Zhang PLA_Obj v_pla = NULL; 11546c995c7dSHong Zhang PetscScalar *loc_buf; 11556c995c7dSHong Zhang Vec loc_x; 11566c995c7dSHong Zhang 11576c995c7dSHong Zhang PetscFunctionBegin; 11586c995c7dSHong Zhang ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 11596c995c7dSHong Zhang ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 11606c995c7dSHong Zhang 11616c995c7dSHong Zhang /* Create PLAPACK vector objects, then copy b into PLAPACK b */ 11626c995c7dSHong Zhang PLA_Mvector_create(lu->datatype,M,1,lu->templ,PLA_ALIGN_FIRST,&v_pla); 11636c995c7dSHong Zhang PLA_Obj_set_to_zero(v_pla); 11646c995c7dSHong Zhang 11656c995c7dSHong Zhang /* Copy b into rhs_pla */ 11666c995c7dSHong Zhang PLA_API_begin(); 11676c995c7dSHong Zhang PLA_Obj_API_open(v_pla); 11686c995c7dSHong Zhang ierr = VecGetArray(b,&array);CHKERRQ(ierr); 11696c995c7dSHong Zhang PLA_API_axpy_vector_to_global(m,&one,(void *)array,1,v_pla,lu->rstart); 11706c995c7dSHong Zhang ierr = VecRestoreArray(b,&array);CHKERRQ(ierr); 11716c995c7dSHong Zhang PLA_Obj_API_close(v_pla); 11726c995c7dSHong Zhang PLA_API_end(); 11736c995c7dSHong Zhang 11746c995c7dSHong Zhang if (A->factor == MAT_FACTOR_LU){ 11756c995c7dSHong Zhang /* Apply the permutations to the right hand sides */ 11766c995c7dSHong Zhang PLA_Apply_pivots_to_rows (v_pla,lu->pivots); 11776c995c7dSHong Zhang 11786c995c7dSHong Zhang /* Solve L y = b, overwriting b with y */ 11796c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_UNIT_DIAG,lu->A,v_pla ); 11806c995c7dSHong Zhang 11816c995c7dSHong Zhang /* Solve U x = y (=b), overwriting b with x */ 11826c995c7dSHong Zhang PLA_Trsv( PLA_UPPER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla ); 11836c995c7dSHong Zhang } else { /* MAT_FACTOR_CHOLESKY */ 11846c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,PLA_NO_TRANSPOSE,PLA_NONUNIT_DIAG,lu->A,v_pla); 11856c995c7dSHong Zhang PLA_Trsv( PLA_LOWER_TRIANGULAR,(lu->datatype == MPI_DOUBLE ? PLA_TRANSPOSE : PLA_CONJUGATE_TRANSPOSE), 11866c995c7dSHong Zhang PLA_NONUNIT_DIAG,lu->A,v_pla); 11876c995c7dSHong Zhang } 11886c995c7dSHong Zhang 11896c995c7dSHong Zhang /* Copy PLAPACK x into Petsc vector x */ 11906c995c7dSHong Zhang PLA_Obj_local_length(v_pla, &loc_m); 11916c995c7dSHong Zhang PLA_Obj_local_buffer(v_pla, (void**)&loc_buf); 11926c995c7dSHong Zhang PLA_Obj_local_stride(v_pla, &loc_stride); 11936c995c7dSHong Zhang /* 11946c995c7dSHong Zhang PetscPrintf(PETSC_COMM_SELF," [%d] b - local_m %d local_stride %d, loc_buf: %g %g, nb: %d\n",rank,loc_m,loc_stride,loc_buf[0],loc_buf[(loc_m-1)*loc_stride],lu->nb); 11956c995c7dSHong Zhang */ 11966c995c7dSHong Zhang ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,loc_m*loc_stride,loc_buf,&loc_x);CHKERRQ(ierr); 11976c995c7dSHong Zhang if (!lu->pla_solved){ 11986c995c7dSHong Zhang 11995d00a290SHong Zhang PLA_Temp_comm_row_info(lu->templ,&Plapack_comm_2d,&r_rank,&r_nproc); 12005d00a290SHong Zhang PLA_Temp_comm_col_info(lu->templ,&Plapack_comm_2d,&c_rank,&c_nproc); 12016c995c7dSHong Zhang 12026c995c7dSHong Zhang /* Create IS and cts for VecScatterring */ 12036c995c7dSHong Zhang PLA_Obj_local_length(v_pla, &loc_m); 12046c995c7dSHong Zhang PLA_Obj_local_stride(v_pla, &loc_stride); 12056c995c7dSHong Zhang ierr = PetscMalloc((2*loc_m+1)*sizeof(PetscInt),&idx_pla);CHKERRQ(ierr); 12066c995c7dSHong Zhang idx_petsc = idx_pla + loc_m; 12076c995c7dSHong Zhang 12086c995c7dSHong Zhang rstart = (r_rank*c_nproc+c_rank)*lu->nb; 12096c995c7dSHong Zhang for (i=0; i<loc_m; i+=lu->nb){ 12106c995c7dSHong Zhang j = 0; 12116c995c7dSHong Zhang while (j < lu->nb && i+j < loc_m){ 12126c995c7dSHong Zhang idx_petsc[i+j] = rstart + j; j++; 12136c995c7dSHong Zhang } 12146c995c7dSHong Zhang rstart += size*lu->nb; 12156c995c7dSHong Zhang } 12166c995c7dSHong Zhang 12176c995c7dSHong Zhang for (i=0; i<loc_m; i++) idx_pla[i] = i*loc_stride; 12186c995c7dSHong Zhang 12196c995c7dSHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_pla,&lu->is_pla);CHKERRQ(ierr); 12206c995c7dSHong Zhang ierr = ISCreateGeneral(PETSC_COMM_SELF,loc_m,idx_petsc,&lu->is_petsc);CHKERRQ(ierr); 12216c995c7dSHong Zhang ierr = PetscFree(idx_pla);CHKERRQ(ierr); 12226c995c7dSHong Zhang ierr = VecScatterCreate(loc_x,lu->is_pla,x,lu->is_petsc,&lu->ctx);CHKERRQ(ierr); 12236c995c7dSHong Zhang } 12246c995c7dSHong Zhang ierr = VecScatterBegin(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12256c995c7dSHong Zhang ierr = VecScatterEnd(lu->ctx,loc_x,x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12266c995c7dSHong Zhang 12276c995c7dSHong Zhang /* Free data */ 12286c995c7dSHong Zhang ierr = VecDestroy(loc_x);CHKERRQ(ierr); 12296c995c7dSHong Zhang PLA_Obj_free(&v_pla); 12306c995c7dSHong Zhang 12316c995c7dSHong Zhang lu->pla_solved = PETSC_TRUE; 12326c995c7dSHong Zhang PetscFunctionReturn(0); 12336c995c7dSHong Zhang } 12346c995c7dSHong Zhang 12356c995c7dSHong Zhang #undef __FUNCT__ 12366c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_MPIDense" 12370481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info) 12386c995c7dSHong Zhang { 1239719d5645SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)(F)->spptr; 12406c995c7dSHong Zhang PetscErrorCode ierr; 1241d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,rend; 12426c995c7dSHong Zhang PetscInt info_pla=0; 12436c995c7dSHong Zhang PetscScalar *array,one = 1.0; 12446c995c7dSHong Zhang 12456c995c7dSHong Zhang PetscFunctionBegin; 12466c995c7dSHong Zhang if (lu->mstruct == SAME_NONZERO_PATTERN){ 12476c995c7dSHong Zhang PLA_Obj_free(&lu->A); 12486c995c7dSHong Zhang PLA_Obj_free (&lu->pivots); 12496c995c7dSHong Zhang } 12506c995c7dSHong Zhang /* Create PLAPACK matrix object */ 12516c995c7dSHong Zhang lu->A = NULL; lu->pivots = NULL; 12526c995c7dSHong Zhang PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A); 12536c995c7dSHong Zhang PLA_Obj_set_to_zero(lu->A); 12546c995c7dSHong Zhang PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots); 12556c995c7dSHong Zhang 12566c995c7dSHong Zhang /* Copy A into lu->A */ 12576c995c7dSHong Zhang PLA_API_begin(); 12586c995c7dSHong Zhang PLA_Obj_API_open(lu->A); 12596c995c7dSHong Zhang ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 12606c995c7dSHong Zhang ierr = MatGetArray(A,&array);CHKERRQ(ierr); 12616c995c7dSHong Zhang PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0); 12626c995c7dSHong Zhang ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 12636c995c7dSHong Zhang PLA_Obj_API_close(lu->A); 12646c995c7dSHong Zhang PLA_API_end(); 12656c995c7dSHong Zhang 12666c995c7dSHong Zhang /* Factor P A -> L U overwriting lower triangular portion of A with L, upper, U */ 12676c995c7dSHong Zhang info_pla = PLA_LU(lu->A,lu->pivots); 12686c995c7dSHong Zhang if (info_pla != 0) 12696c995c7dSHong Zhang SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot encountered at row %d from PLA_LU()",info_pla); 12706c995c7dSHong Zhang 12716c995c7dSHong Zhang lu->rstart = rstart; 12726c995c7dSHong Zhang lu->mstruct = SAME_NONZERO_PATTERN; 1273719d5645SBarry Smith F->ops->solve = MatSolve_MPIDense; 1274719d5645SBarry Smith F->assembled = PETSC_TRUE; /* required by -ksp_view */ 12756c995c7dSHong Zhang PetscFunctionReturn(0); 12766c995c7dSHong Zhang } 12776c995c7dSHong Zhang 12786c995c7dSHong Zhang #undef __FUNCT__ 12796c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_MPIDense" 12800481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_MPIDense(Mat F,Mat A,const MatFactorInfo *info) 12816c995c7dSHong Zhang { 1282719d5645SBarry Smith Mat_Plapack *lu = (Mat_Plapack*)F->spptr; 12836c995c7dSHong Zhang PetscErrorCode ierr; 1284d0f46423SBarry Smith PetscInt M=A->rmap->N,m=A->rmap->n,rstart,rend; 12856c995c7dSHong Zhang PetscInt info_pla=0; 12866c995c7dSHong Zhang PetscScalar *array,one = 1.0; 12876c995c7dSHong Zhang 12886c995c7dSHong Zhang PetscFunctionBegin; 12896c995c7dSHong Zhang if (lu->mstruct == SAME_NONZERO_PATTERN){ 12906c995c7dSHong Zhang PLA_Obj_free(&lu->A); 12916c995c7dSHong Zhang } 12926c995c7dSHong Zhang /* Create PLAPACK matrix object */ 12936c995c7dSHong Zhang lu->A = NULL; 12946c995c7dSHong Zhang lu->pivots = NULL; 12956c995c7dSHong Zhang PLA_Matrix_create(lu->datatype,M,M,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A); 12966c995c7dSHong Zhang 12976c995c7dSHong Zhang /* Copy A into lu->A */ 12986c995c7dSHong Zhang PLA_API_begin(); 12996c995c7dSHong Zhang PLA_Obj_API_open(lu->A); 13006c995c7dSHong Zhang ierr = MatGetOwnershipRange(A,&rstart,&rend);CHKERRQ(ierr); 13016c995c7dSHong Zhang ierr = MatGetArray(A,&array);CHKERRQ(ierr); 13026c995c7dSHong Zhang PLA_API_axpy_matrix_to_global(m,M, &one,(void *)array,m,lu->A,rstart,0); 13036c995c7dSHong Zhang ierr = MatRestoreArray(A,&array);CHKERRQ(ierr); 13046c995c7dSHong Zhang PLA_Obj_API_close(lu->A); 13056c995c7dSHong Zhang PLA_API_end(); 13066c995c7dSHong Zhang 13076c995c7dSHong Zhang /* Factor P A -> Chol */ 13086c995c7dSHong Zhang info_pla = PLA_Chol(PLA_LOWER_TRIANGULAR,lu->A); 13096c995c7dSHong Zhang if (info_pla != 0) 13106c995c7dSHong Zhang SETERRQ1( PETSC_ERR_MAT_CH_ZRPVT,"Nonpositive definite matrix detected at row %d from PLA_Chol()",info_pla); 13116c995c7dSHong Zhang 13126c995c7dSHong Zhang lu->rstart = rstart; 13136c995c7dSHong Zhang lu->mstruct = SAME_NONZERO_PATTERN; 1314719d5645SBarry Smith F->ops->solve = MatSolve_MPIDense; 1315719d5645SBarry Smith F->assembled = PETSC_TRUE; /* required by -ksp_view */ 13166c995c7dSHong Zhang PetscFunctionReturn(0); 13176c995c7dSHong Zhang } 13186c995c7dSHong Zhang 1319b24902e0SBarry Smith /* Note the Petsc perm permutation is ignored */ 13202fbe02b9SBarry Smith #undef __FUNCT__ 13216c995c7dSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_MPIDense" 13220481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_MPIDense(Mat F,Mat A,IS perm,const MatFactorInfo *info) 132301b82886SBarry Smith { 132401b82886SBarry Smith PetscErrorCode ierr; 1325b24902e0SBarry Smith PetscTruth issymmetric,set; 132601b82886SBarry Smith 132701b82886SBarry Smith PetscFunctionBegin; 1328b24902e0SBarry Smith ierr = MatIsSymmetricKnown(A,&set,&issymmetric);CHKERRQ(ierr); 1329b24902e0SBarry Smith if (!set || !issymmetric) SETERRQ(PETSC_ERR_USER,"Matrix must be set as MAT_SYMMETRIC for CholeskyFactor()"); 1330719d5645SBarry Smith F->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_MPIDense; 1331b24902e0SBarry Smith PetscFunctionReturn(0); 133201b82886SBarry Smith } 133301b82886SBarry Smith 1334b24902e0SBarry Smith /* Note the Petsc r and c permutations are ignored */ 1335b24902e0SBarry Smith #undef __FUNCT__ 13366c995c7dSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_MPIDense" 13370481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_MPIDense(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 1338b24902e0SBarry Smith { 1339b24902e0SBarry Smith PetscErrorCode ierr; 1340d0f46423SBarry Smith PetscInt M = A->rmap->N; 1341b24902e0SBarry Smith Mat_Plapack *lu; 134201b82886SBarry Smith 1343b24902e0SBarry Smith PetscFunctionBegin; 1344719d5645SBarry Smith lu = (Mat_Plapack*)F->spptr; 1345b24902e0SBarry Smith ierr = PLA_Mvector_create(MPI_INT,M,1,lu->templ,PLA_ALIGN_FIRST,&lu->pivots);CHKERRQ(ierr); 1346719d5645SBarry Smith F->ops->lufactornumeric = MatLUFactorNumeric_MPIDense; 134701b82886SBarry Smith PetscFunctionReturn(0); 134801b82886SBarry Smith } 134901b82886SBarry Smith 1350b6806ab0SHong Zhang EXTERN_C_BEGIN 135101b82886SBarry Smith #undef __FUNCT__ 1352b6806ab0SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_mpidense_plapack" 1353b6806ab0SHong Zhang PetscErrorCode MatFactorGetSolverPackage_mpidense_plapack(Mat A,const MatSolverPackage *type) 1354b6806ab0SHong Zhang { 1355b6806ab0SHong Zhang PetscFunctionBegin; 1356b6806ab0SHong Zhang *type = MAT_SOLVER_PLAPACK; 1357b6806ab0SHong Zhang PetscFunctionReturn(0); 1358b6806ab0SHong Zhang } 1359b6806ab0SHong Zhang EXTERN_C_END 1360b6806ab0SHong Zhang 1361*bb5747d9SMatthew Knepley EXTERN_C_BEGIN 1362b6806ab0SHong Zhang #undef __FUNCT__ 1363b6806ab0SHong Zhang #define __FUNCT__ "MatGetFactor_mpidense_plapack" 1364b6806ab0SHong Zhang PetscErrorCode MatGetFactor_mpidense_plapack(Mat A,MatFactorType ftype,Mat *F) 136501b82886SBarry Smith { 136601b82886SBarry Smith PetscErrorCode ierr; 13676c995c7dSHong Zhang Mat_Plapack *lu; 13686c995c7dSHong Zhang PetscMPIInt size; 136985bd4cefSHong Zhang PetscInt M=A->rmap->N; 137001b82886SBarry Smith 137101b82886SBarry Smith PetscFunctionBegin; 137201b82886SBarry Smith /* Create the factorization matrix */ 1373eae6fb2eSBarry Smith ierr = MatCreate(((PetscObject)A)->comm,F);CHKERRQ(ierr); 1374d0f46423SBarry Smith ierr = MatSetSizes(*F,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1375eae6fb2eSBarry Smith ierr = MatSetType(*F,((PetscObject)A)->type_name);CHKERRQ(ierr); 137685bd4cefSHong Zhang ierr = PetscNewLog(*F,Mat_Plapack,&lu);CHKERRQ(ierr); 137785bd4cefSHong Zhang (*F)->spptr = (void*)lu; 137801b82886SBarry Smith 1379b24902e0SBarry Smith /* Set default Plapack parameters */ 13806c995c7dSHong Zhang ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr); 1381b24902e0SBarry Smith lu->nb = M/size; 1382b24902e0SBarry Smith if (M - lu->nb*size) lu->nb++; /* without cyclic distribution */ 138301b82886SBarry Smith 1384b24902e0SBarry Smith /* Set runtime options */ 1385b24902e0SBarry Smith ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"PLAPACK Options","Mat");CHKERRQ(ierr); 1386b24902e0SBarry Smith ierr = PetscOptionsInt("-mat_plapack_nb","block size of template vector","None",lu->nb,&lu->nb,PETSC_NULL);CHKERRQ(ierr); 1387b24902e0SBarry Smith PetscOptionsEnd(); 138801b82886SBarry Smith 1389b24902e0SBarry Smith /* Create object distribution template */ 1390b24902e0SBarry Smith lu->templ = NULL; 1391b24902e0SBarry Smith ierr = PLA_Temp_create(lu->nb, 0, &lu->templ);CHKERRQ(ierr); 1392b24902e0SBarry Smith 1393b24902e0SBarry Smith /* Set the datatype */ 1394b24902e0SBarry Smith #if defined(PETSC_USE_COMPLEX) 1395b24902e0SBarry Smith lu->datatype = MPI_DOUBLE_COMPLEX; 1396b24902e0SBarry Smith #else 1397b24902e0SBarry Smith lu->datatype = MPI_DOUBLE; 1398b24902e0SBarry Smith #endif 1399b24902e0SBarry Smith 1400d0f46423SBarry Smith ierr = PLA_Matrix_create(lu->datatype,M,A->cmap->N,lu->templ,PLA_ALIGN_FIRST,PLA_ALIGN_FIRST,&lu->A);CHKERRQ(ierr); 1401b24902e0SBarry Smith 1402b24902e0SBarry Smith 1403b24902e0SBarry Smith lu->pla_solved = PETSC_FALSE; /* MatSolve_Plapack() is called yet */ 14045d00a290SHong Zhang lu->mstruct = DIFFERENT_NONZERO_PATTERN; 1405b24902e0SBarry Smith 1406b24902e0SBarry Smith if (ftype == MAT_FACTOR_LU) { 1407b24902e0SBarry Smith (*F)->ops->lufactorsymbolic = MatLUFactorSymbolic_MPIDense; 1408b24902e0SBarry Smith } else if (ftype == MAT_FACTOR_CHOLESKY) { 14096c995c7dSHong Zhang (*F)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_MPIDense; 1410b24902e0SBarry Smith } else SETERRQ(PETSC_ERR_SUP,"No incomplete factorizations for dense matrices"); 1411719d5645SBarry Smith (*F)->factor = ftype; 1412b6806ab0SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)(*F),"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_mpidense_plapack",MatFactorGetSolverPackage_mpidense_plapack);CHKERRQ(ierr); 141301b82886SBarry Smith PetscFunctionReturn(0); 141401b82886SBarry Smith } 1415*bb5747d9SMatthew Knepley EXTERN_C_END 141601b82886SBarry Smith #endif 141701b82886SBarry Smith 1418488007eeSBarry Smith #undef __FUNCT__ 1419488007eeSBarry Smith #define __FUNCT__ "MatAXPY_MPIDense" 1420488007eeSBarry Smith PetscErrorCode MatAXPY_MPIDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str) 1421488007eeSBarry Smith { 1422488007eeSBarry Smith PetscErrorCode ierr; 1423488007eeSBarry Smith Mat_MPIDense *A = (Mat_MPIDense*)Y->data, *B = (Mat_MPIDense*)X->data; 1424488007eeSBarry Smith 1425488007eeSBarry Smith PetscFunctionBegin; 1426488007eeSBarry Smith ierr = MatAXPY(A->A,alpha,B->A,str);CHKERRQ(ierr); 1427488007eeSBarry Smith PetscFunctionReturn(0); 1428488007eeSBarry Smith } 1429488007eeSBarry Smith 14308965ea79SLois Curfman McInnes /* -------------------------------------------------------------------*/ 143109dc0095SBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_MPIDense, 143209dc0095SBarry Smith MatGetRow_MPIDense, 143309dc0095SBarry Smith MatRestoreRow_MPIDense, 143409dc0095SBarry Smith MatMult_MPIDense, 143597304618SKris Buschelman /* 4*/ MatMultAdd_MPIDense, 14367c922b88SBarry Smith MatMultTranspose_MPIDense, 14377c922b88SBarry Smith MatMultTransposeAdd_MPIDense, 14388965ea79SLois Curfman McInnes 0, 143909dc0095SBarry Smith 0, 144009dc0095SBarry Smith 0, 144197304618SKris Buschelman /*10*/ 0, 144209dc0095SBarry Smith 0, 144309dc0095SBarry Smith 0, 144409dc0095SBarry Smith 0, 144509dc0095SBarry Smith MatTranspose_MPIDense, 144697304618SKris Buschelman /*15*/ MatGetInfo_MPIDense, 14476e4ee0c6SHong Zhang MatEqual_MPIDense, 144809dc0095SBarry Smith MatGetDiagonal_MPIDense, 14495b2fa520SLois Curfman McInnes MatDiagonalScale_MPIDense, 145009dc0095SBarry Smith MatNorm_MPIDense, 145197304618SKris Buschelman /*20*/ MatAssemblyBegin_MPIDense, 145209dc0095SBarry Smith MatAssemblyEnd_MPIDense, 145309dc0095SBarry Smith MatSetOption_MPIDense, 145409dc0095SBarry Smith MatZeroEntries_MPIDense, 1455d519adbfSMatthew Knepley /*24*/ MatZeroRows_MPIDense, 1456919b68f7SBarry Smith 0, 145701b82886SBarry Smith 0, 145801b82886SBarry Smith 0, 145901b82886SBarry Smith 0, 1460d519adbfSMatthew Knepley /*29*/ MatSetUpPreallocation_MPIDense, 1461273d9f13SBarry Smith 0, 146209dc0095SBarry Smith 0, 146309dc0095SBarry Smith MatGetArray_MPIDense, 146409dc0095SBarry Smith MatRestoreArray_MPIDense, 1465d519adbfSMatthew Knepley /*34*/ MatDuplicate_MPIDense, 146609dc0095SBarry Smith 0, 146709dc0095SBarry Smith 0, 146809dc0095SBarry Smith 0, 146909dc0095SBarry Smith 0, 1470d519adbfSMatthew Knepley /*39*/ MatAXPY_MPIDense, 14712ce60cd0SSatish Balay MatGetSubMatrices_MPIDense, 147209dc0095SBarry Smith 0, 147309dc0095SBarry Smith MatGetValues_MPIDense, 147409dc0095SBarry Smith 0, 1475d519adbfSMatthew Knepley /*44*/ 0, 147609dc0095SBarry Smith MatScale_MPIDense, 147709dc0095SBarry Smith 0, 147809dc0095SBarry Smith 0, 147909dc0095SBarry Smith 0, 1480d519adbfSMatthew Knepley /*49*/ 0, 148109dc0095SBarry Smith 0, 148209dc0095SBarry Smith 0, 148309dc0095SBarry Smith 0, 148409dc0095SBarry Smith 0, 1485d519adbfSMatthew Knepley /*54*/ 0, 148609dc0095SBarry Smith 0, 148709dc0095SBarry Smith 0, 148809dc0095SBarry Smith 0, 148909dc0095SBarry Smith 0, 1490d519adbfSMatthew Knepley /*59*/ MatGetSubMatrix_MPIDense, 1491b9b97703SBarry Smith MatDestroy_MPIDense, 1492b9b97703SBarry Smith MatView_MPIDense, 1493357abbc8SBarry Smith 0, 149497304618SKris Buschelman 0, 1495d519adbfSMatthew Knepley /*64*/ 0, 149697304618SKris Buschelman 0, 149797304618SKris Buschelman 0, 149897304618SKris Buschelman 0, 149997304618SKris Buschelman 0, 1500d519adbfSMatthew Knepley /*69*/ 0, 150197304618SKris Buschelman 0, 150297304618SKris Buschelman 0, 150397304618SKris Buschelman 0, 150497304618SKris Buschelman 0, 1505d519adbfSMatthew Knepley /*74*/ 0, 150697304618SKris Buschelman 0, 150797304618SKris Buschelman 0, 150897304618SKris Buschelman 0, 150997304618SKris Buschelman 0, 1510d519adbfSMatthew Knepley /*79*/ 0, 151197304618SKris Buschelman 0, 151297304618SKris Buschelman 0, 151397304618SKris Buschelman 0, 1514d519adbfSMatthew Knepley /*83*/ MatLoad_MPIDense, 1515865e5f61SKris Buschelman 0, 1516865e5f61SKris Buschelman 0, 1517865e5f61SKris Buschelman 0, 1518865e5f61SKris Buschelman 0, 1519865e5f61SKris Buschelman 0, 1520d519adbfSMatthew Knepley /*89*/ 15212fbe02b9SBarry Smith #if defined(PETSC_HAVE_PLAPACK) 15222fbe02b9SBarry Smith MatMatMult_MPIDense_MPIDense, 15234ae313f4SHong Zhang MatMatMultSymbolic_MPIDense_MPIDense, 15242fbe02b9SBarry Smith MatMatMultNumeric_MPIDense_MPIDense, 15252fbe02b9SBarry Smith #else 1526865e5f61SKris Buschelman 0, 1527865e5f61SKris Buschelman 0, 1528865e5f61SKris Buschelman 0, 15292fbe02b9SBarry Smith #endif 15302fbe02b9SBarry Smith 0, 1531d519adbfSMatthew Knepley /*94*/ 0, 1532865e5f61SKris Buschelman 0, 1533865e5f61SKris Buschelman 0, 1534865e5f61SKris Buschelman 0}; 15358965ea79SLois Curfman McInnes 1536273d9f13SBarry Smith EXTERN_C_BEGIN 15374a2ae208SSatish Balay #undef __FUNCT__ 1538a23d5eceSKris Buschelman #define __FUNCT__ "MatMPIDenseSetPreallocation_MPIDense" 1539be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation_MPIDense(Mat mat,PetscScalar *data) 1540a23d5eceSKris Buschelman { 1541a23d5eceSKris Buschelman Mat_MPIDense *a; 1542dfbe8321SBarry Smith PetscErrorCode ierr; 1543a23d5eceSKris Buschelman 1544a23d5eceSKris Buschelman PetscFunctionBegin; 1545a23d5eceSKris Buschelman mat->preallocated = PETSC_TRUE; 1546a23d5eceSKris Buschelman /* Note: For now, when data is specified above, this assumes the user correctly 1547a23d5eceSKris Buschelman allocates the local dense storage space. We should add error checking. */ 1548a23d5eceSKris Buschelman 1549a23d5eceSKris Buschelman a = (Mat_MPIDense*)mat->data; 1550f69a0ea3SMatthew Knepley ierr = MatCreate(PETSC_COMM_SELF,&a->A);CHKERRQ(ierr); 1551d0f46423SBarry Smith ierr = MatSetSizes(a->A,mat->rmap->n,mat->cmap->N,mat->rmap->n,mat->cmap->N);CHKERRQ(ierr); 15525c5985e7SKris Buschelman ierr = MatSetType(a->A,MATSEQDENSE);CHKERRQ(ierr); 15535c5985e7SKris Buschelman ierr = MatSeqDenseSetPreallocation(a->A,data);CHKERRQ(ierr); 155452e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr); 1555a23d5eceSKris Buschelman PetscFunctionReturn(0); 1556a23d5eceSKris Buschelman } 1557a23d5eceSKris Buschelman EXTERN_C_END 1558a23d5eceSKris Buschelman 15590bad9183SKris Buschelman /*MC 156041c8de11SBarry Smith MAT_SOLVER_PLAPACK = "mpidense" - Parallel LU and Cholesky factorization for MATMPIDENSE matrices 15610bad9183SKris Buschelman 156241c8de11SBarry Smith run config/configure.py with the option --download-plapack 15637878bbefSBarry Smith 15647878bbefSBarry Smith 15657878bbefSBarry Smith Options Database Keys: 15667878bbefSBarry Smith . -mat_plapack_nprows <n> - number of rows in processor partition 15677878bbefSBarry Smith . -mat_plapack_npcols <n> - number of columns in processor partition 15687878bbefSBarry Smith . -mat_plapack_nb <n> - block size of template vector 15697878bbefSBarry Smith . -mat_plapack_nb_alg <n> - algorithmic block size 15707878bbefSBarry Smith - -mat_plapack_ckerror <n> - error checking flag 15717878bbefSBarry Smith 157241c8de11SBarry Smith .seealso: MatCreateMPIDense(), MATDENSE, MATSEQDENSE, PCFactorSetSolverPackage(), MatSolverPackage 157341c8de11SBarry Smith 15740bad9183SKris Buschelman M*/ 15750bad9183SKris Buschelman 1576a23d5eceSKris Buschelman EXTERN_C_BEGIN 1577a23d5eceSKris Buschelman #undef __FUNCT__ 15784a2ae208SSatish Balay #define __FUNCT__ "MatCreate_MPIDense" 1579be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_MPIDense(Mat mat) 1580273d9f13SBarry Smith { 1581273d9f13SBarry Smith Mat_MPIDense *a; 1582dfbe8321SBarry Smith PetscErrorCode ierr; 1583273d9f13SBarry Smith 1584273d9f13SBarry Smith PetscFunctionBegin; 158538f2d2fdSLisandro Dalcin ierr = PetscNewLog(mat,Mat_MPIDense,&a);CHKERRQ(ierr); 1586b0a32e0cSBarry Smith mat->data = (void*)a; 1587273d9f13SBarry Smith ierr = PetscMemcpy(mat->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1588273d9f13SBarry Smith mat->mapping = 0; 1589273d9f13SBarry Smith 1590273d9f13SBarry Smith mat->insertmode = NOT_SET_VALUES; 15917adad957SLisandro Dalcin ierr = MPI_Comm_rank(((PetscObject)mat)->comm,&a->rank);CHKERRQ(ierr); 15927adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)mat)->comm,&a->size);CHKERRQ(ierr); 1593273d9f13SBarry Smith 15947408324eSLisandro Dalcin ierr = PetscMapSetBlockSize(mat->rmap,1);CHKERRQ(ierr); 15957408324eSLisandro Dalcin ierr = PetscMapSetBlockSize(mat->cmap,1);CHKERRQ(ierr); 1596d0f46423SBarry Smith ierr = PetscMapSetUp(mat->rmap);CHKERRQ(ierr); 1597d0f46423SBarry Smith ierr = PetscMapSetUp(mat->cmap);CHKERRQ(ierr); 1598d0f46423SBarry Smith a->nvec = mat->cmap->n; 1599273d9f13SBarry Smith 1600273d9f13SBarry Smith /* build cache for off array entries formed */ 1601273d9f13SBarry Smith a->donotstash = PETSC_FALSE; 16027adad957SLisandro Dalcin ierr = MatStashCreate_Private(((PetscObject)mat)->comm,1,&mat->stash);CHKERRQ(ierr); 1603273d9f13SBarry Smith 1604273d9f13SBarry Smith /* stuff used for matrix vector multiply */ 1605273d9f13SBarry Smith a->lvec = 0; 1606273d9f13SBarry Smith a->Mvctx = 0; 1607273d9f13SBarry Smith a->roworiented = PETSC_TRUE; 1608273d9f13SBarry Smith 1609273d9f13SBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetDiagonalBlock_C", 1610273d9f13SBarry Smith "MatGetDiagonalBlock_MPIDense", 1611273d9f13SBarry Smith MatGetDiagonalBlock_MPIDense);CHKERRQ(ierr); 1612a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMPIDenseSetPreallocation_C", 1613a23d5eceSKris Buschelman "MatMPIDenseSetPreallocation_MPIDense", 1614a23d5eceSKris Buschelman MatMPIDenseSetPreallocation_MPIDense);CHKERRQ(ierr); 16154ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_mpiaij_mpidense_C", 16164ae313f4SHong Zhang "MatMatMult_MPIAIJ_MPIDense", 16174ae313f4SHong Zhang MatMatMult_MPIAIJ_MPIDense);CHKERRQ(ierr); 16184ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_mpiaij_mpidense_C", 16194ae313f4SHong Zhang "MatMatMultSymbolic_MPIAIJ_MPIDense", 16204ae313f4SHong Zhang MatMatMultSymbolic_MPIAIJ_MPIDense);CHKERRQ(ierr); 16214ae313f4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_mpiaij_mpidense_C", 16224ae313f4SHong Zhang "MatMatMultNumeric_MPIAIJ_MPIDense", 16234ae313f4SHong Zhang MatMatMultNumeric_MPIAIJ_MPIDense);CHKERRQ(ierr); 16247878bbefSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 1625b6806ab0SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatGetFactor_mpidense_plapack_C", 1626b6806ab0SHong Zhang "MatGetFactor_mpidense_plapack", 1627b6806ab0SHong Zhang MatGetFactor_mpidense_plapack);CHKERRQ(ierr); 1628db4efbfdSBarry Smith ierr = PetscPLAPACKInitializePackage(((PetscObject)mat)->comm);CHKERRQ(ierr); 1629db4efbfdSBarry Smith #endif 163038aed534SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)mat,MATMPIDENSE);CHKERRQ(ierr); 163101b82886SBarry Smith 1632273d9f13SBarry Smith PetscFunctionReturn(0); 1633273d9f13SBarry Smith } 1634273d9f13SBarry Smith EXTERN_C_END 1635273d9f13SBarry Smith 1636209238afSKris Buschelman /*MC 1637002d173eSKris Buschelman MATDENSE - MATDENSE = "dense" - A matrix type to be used for dense matrices. 1638209238afSKris Buschelman 1639209238afSKris Buschelman This matrix type is identical to MATSEQDENSE when constructed with a single process communicator, 1640209238afSKris Buschelman and MATMPIDENSE otherwise. 1641209238afSKris Buschelman 1642209238afSKris Buschelman Options Database Keys: 1643209238afSKris Buschelman . -mat_type dense - sets the matrix type to "dense" during a call to MatSetFromOptions() 1644209238afSKris Buschelman 1645209238afSKris Buschelman Level: beginner 1646209238afSKris Buschelman 164701b82886SBarry Smith 1648209238afSKris Buschelman .seealso: MatCreateMPIDense,MATSEQDENSE,MATMPIDENSE 1649209238afSKris Buschelman M*/ 1650209238afSKris Buschelman 1651209238afSKris Buschelman EXTERN_C_BEGIN 1652209238afSKris Buschelman #undef __FUNCT__ 1653209238afSKris Buschelman #define __FUNCT__ "MatCreate_Dense" 1654be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_Dense(Mat A) 1655dfbe8321SBarry Smith { 16566849ba73SBarry Smith PetscErrorCode ierr; 165713f74950SBarry Smith PetscMPIInt size; 1658209238afSKris Buschelman 1659209238afSKris Buschelman PetscFunctionBegin; 16607adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)A)->comm,&size);CHKERRQ(ierr); 1661209238afSKris Buschelman if (size == 1) { 1662209238afSKris Buschelman ierr = MatSetType(A,MATSEQDENSE);CHKERRQ(ierr); 1663209238afSKris Buschelman } else { 1664209238afSKris Buschelman ierr = MatSetType(A,MATMPIDENSE);CHKERRQ(ierr); 1665209238afSKris Buschelman } 1666209238afSKris Buschelman PetscFunctionReturn(0); 1667209238afSKris Buschelman } 1668209238afSKris Buschelman EXTERN_C_END 1669209238afSKris Buschelman 16704a2ae208SSatish Balay #undef __FUNCT__ 16714a2ae208SSatish Balay #define __FUNCT__ "MatMPIDenseSetPreallocation" 1672273d9f13SBarry Smith /*@C 1673273d9f13SBarry Smith MatMPIDenseSetPreallocation - Sets the array used to store the matrix entries 1674273d9f13SBarry Smith 1675273d9f13SBarry Smith Not collective 1676273d9f13SBarry Smith 1677273d9f13SBarry Smith Input Parameters: 1678273d9f13SBarry Smith . A - the matrix 1679273d9f13SBarry Smith - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1680273d9f13SBarry Smith to control all matrix memory allocation. 1681273d9f13SBarry Smith 1682273d9f13SBarry Smith Notes: 1683273d9f13SBarry Smith The dense format is fully compatible with standard Fortran 77 1684273d9f13SBarry Smith storage by columns. 1685273d9f13SBarry Smith 1686273d9f13SBarry Smith The data input variable is intended primarily for Fortran programmers 1687273d9f13SBarry Smith who wish to allocate their own matrix memory space. Most users should 1688273d9f13SBarry Smith set data=PETSC_NULL. 1689273d9f13SBarry Smith 1690273d9f13SBarry Smith Level: intermediate 1691273d9f13SBarry Smith 1692273d9f13SBarry Smith .keywords: matrix,dense, parallel 1693273d9f13SBarry Smith 1694273d9f13SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues() 1695273d9f13SBarry Smith @*/ 1696be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatMPIDenseSetPreallocation(Mat mat,PetscScalar *data) 1697273d9f13SBarry Smith { 1698dfbe8321SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscScalar *); 1699273d9f13SBarry Smith 1700273d9f13SBarry Smith PetscFunctionBegin; 1701565567f0SKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)mat,"MatMPIDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1702a23d5eceSKris Buschelman if (f) { 1703a23d5eceSKris Buschelman ierr = (*f)(mat,data);CHKERRQ(ierr); 1704a23d5eceSKris Buschelman } 1705273d9f13SBarry Smith PetscFunctionReturn(0); 1706273d9f13SBarry Smith } 1707273d9f13SBarry Smith 17084a2ae208SSatish Balay #undef __FUNCT__ 17094a2ae208SSatish Balay #define __FUNCT__ "MatCreateMPIDense" 17108965ea79SLois Curfman McInnes /*@C 171139ddd567SLois Curfman McInnes MatCreateMPIDense - Creates a sparse parallel matrix in dense format. 17128965ea79SLois Curfman McInnes 1713db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1714db81eaa0SLois Curfman McInnes 17158965ea79SLois Curfman McInnes Input Parameters: 1716db81eaa0SLois Curfman McInnes + comm - MPI communicator 17178965ea79SLois Curfman McInnes . m - number of local rows (or PETSC_DECIDE to have calculated if M is given) 1718db81eaa0SLois Curfman McInnes . n - number of local columns (or PETSC_DECIDE to have calculated if N is given) 17198965ea79SLois Curfman McInnes . M - number of global rows (or PETSC_DECIDE to have calculated if m is given) 1720db81eaa0SLois Curfman McInnes . N - number of global columns (or PETSC_DECIDE to have calculated if n is given) 17217f5ff6fdSBarry Smith - data - optional location of matrix data. Set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users) for PETSc 1722dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 17238965ea79SLois Curfman McInnes 17248965ea79SLois Curfman McInnes Output Parameter: 1725477f1c0bSLois Curfman McInnes . A - the matrix 17268965ea79SLois Curfman McInnes 1727b259b22eSLois Curfman McInnes Notes: 172839ddd567SLois Curfman McInnes The dense format is fully compatible with standard Fortran 77 172939ddd567SLois Curfman McInnes storage by columns. 17308965ea79SLois Curfman McInnes 173118f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 173218f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 17337f5ff6fdSBarry Smith set data=PETSC_NULL (PETSC_NULL_SCALAR for Fortran users). 173418f449edSLois Curfman McInnes 17358965ea79SLois Curfman McInnes The user MUST specify either the local or global matrix dimensions 17368965ea79SLois Curfman McInnes (possibly both). 17378965ea79SLois Curfman McInnes 1738027ccd11SLois Curfman McInnes Level: intermediate 1739027ccd11SLois Curfman McInnes 174039ddd567SLois Curfman McInnes .keywords: matrix,dense, parallel 17418965ea79SLois Curfman McInnes 174239ddd567SLois Curfman McInnes .seealso: MatCreate(), MatCreateSeqDense(), MatSetValues() 17438965ea79SLois Curfman McInnes @*/ 1744be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateMPIDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,PetscScalar *data,Mat *A) 17458965ea79SLois Curfman McInnes { 17466849ba73SBarry Smith PetscErrorCode ierr; 174713f74950SBarry Smith PetscMPIInt size; 17488965ea79SLois Curfman McInnes 17493a40ed3dSBarry Smith PetscFunctionBegin; 1750f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 1751f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,M,N);CHKERRQ(ierr); 1752273d9f13SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1753273d9f13SBarry Smith if (size > 1) { 1754273d9f13SBarry Smith ierr = MatSetType(*A,MATMPIDENSE);CHKERRQ(ierr); 1755273d9f13SBarry Smith ierr = MatMPIDenseSetPreallocation(*A,data);CHKERRQ(ierr); 1756273d9f13SBarry Smith } else { 1757273d9f13SBarry Smith ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr); 1758273d9f13SBarry Smith ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr); 17598c469469SLois Curfman McInnes } 17603a40ed3dSBarry Smith PetscFunctionReturn(0); 17618965ea79SLois Curfman McInnes } 17628965ea79SLois Curfman McInnes 17634a2ae208SSatish Balay #undef __FUNCT__ 17644a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_MPIDense" 17656849ba73SBarry Smith static PetscErrorCode MatDuplicate_MPIDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 17668965ea79SLois Curfman McInnes { 17678965ea79SLois Curfman McInnes Mat mat; 17683501a2bdSLois Curfman McInnes Mat_MPIDense *a,*oldmat = (Mat_MPIDense*)A->data; 1769dfbe8321SBarry Smith PetscErrorCode ierr; 17708965ea79SLois Curfman McInnes 17713a40ed3dSBarry Smith PetscFunctionBegin; 17728965ea79SLois Curfman McInnes *newmat = 0; 17737adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&mat);CHKERRQ(ierr); 1774d0f46423SBarry Smith ierr = MatSetSizes(mat,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 17757adad957SLisandro Dalcin ierr = MatSetType(mat,((PetscObject)A)->type_name);CHKERRQ(ierr); 1776834f8fabSBarry Smith a = (Mat_MPIDense*)mat->data; 1777e04c1aa4SHong Zhang ierr = PetscMemcpy(mat->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 17783501a2bdSLois Curfman McInnes mat->factor = A->factor; 1779c456f294SBarry Smith mat->assembled = PETSC_TRUE; 1780273d9f13SBarry Smith mat->preallocated = PETSC_TRUE; 17818965ea79SLois Curfman McInnes 1782d0f46423SBarry Smith mat->rmap->rstart = A->rmap->rstart; 1783d0f46423SBarry Smith mat->rmap->rend = A->rmap->rend; 17848965ea79SLois Curfman McInnes a->size = oldmat->size; 17858965ea79SLois Curfman McInnes a->rank = oldmat->rank; 1786e0fa3b82SLois Curfman McInnes mat->insertmode = NOT_SET_VALUES; 1787b9b97703SBarry Smith a->nvec = oldmat->nvec; 17883782ba37SSatish Balay a->donotstash = oldmat->donotstash; 1789e04c1aa4SHong Zhang 1790d0f46423SBarry Smith ierr = PetscMemcpy(mat->rmap->range,A->rmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr); 1791d0f46423SBarry Smith ierr = PetscMemcpy(mat->cmap->range,A->cmap->range,(a->size+1)*sizeof(PetscInt));CHKERRQ(ierr); 17927adad957SLisandro Dalcin ierr = MatStashCreate_Private(((PetscObject)A)->comm,1,&mat->stash);CHKERRQ(ierr); 17938965ea79SLois Curfman McInnes 1794329f5518SBarry Smith ierr = MatSetUpMultiply_MPIDense(mat);CHKERRQ(ierr); 17955609ef8eSBarry Smith ierr = MatDuplicate(oldmat->A,cpvalues,&a->A);CHKERRQ(ierr); 179652e6d16bSBarry Smith ierr = PetscLogObjectParent(mat,a->A);CHKERRQ(ierr); 179701b82886SBarry Smith 17988965ea79SLois Curfman McInnes *newmat = mat; 17993a40ed3dSBarry Smith PetscFunctionReturn(0); 18008965ea79SLois Curfman McInnes } 18018965ea79SLois Curfman McInnes 1802e090d566SSatish Balay #include "petscsys.h" 18038965ea79SLois Curfman McInnes 18044a2ae208SSatish Balay #undef __FUNCT__ 18054a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense_DenseInFile" 1806a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense_DenseInFile(MPI_Comm comm,PetscInt fd,PetscInt M,PetscInt N, const MatType type,Mat *newmat) 180790ace30eSBarry Smith { 18086849ba73SBarry Smith PetscErrorCode ierr; 180932dcc486SBarry Smith PetscMPIInt rank,size; 181013f74950SBarry Smith PetscInt *rowners,i,m,nz,j; 181187828ca2SBarry Smith PetscScalar *array,*vals,*vals_ptr; 181290ace30eSBarry Smith MPI_Status status; 181390ace30eSBarry Smith 18143a40ed3dSBarry Smith PetscFunctionBegin; 1815d132466eSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 1816d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 181790ace30eSBarry Smith 181890ace30eSBarry Smith /* determine ownership of all rows */ 181990ace30eSBarry Smith m = M/size + ((M % size) > rank); 182013f74950SBarry Smith ierr = PetscMalloc((size+2)*sizeof(PetscInt),&rowners);CHKERRQ(ierr); 182113f74950SBarry Smith ierr = MPI_Allgather(&m,1,MPIU_INT,rowners+1,1,MPIU_INT,comm);CHKERRQ(ierr); 182290ace30eSBarry Smith rowners[0] = 0; 182390ace30eSBarry Smith for (i=2; i<=size; i++) { 182490ace30eSBarry Smith rowners[i] += rowners[i-1]; 182590ace30eSBarry Smith } 182690ace30eSBarry Smith 1827f69a0ea3SMatthew Knepley ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 1828f69a0ea3SMatthew Knepley ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr); 1829be5d1d56SKris Buschelman ierr = MatSetType(*newmat,type);CHKERRQ(ierr); 1830878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr); 183190ace30eSBarry Smith ierr = MatGetArray(*newmat,&array);CHKERRQ(ierr); 183290ace30eSBarry Smith 183390ace30eSBarry Smith if (!rank) { 183487828ca2SBarry Smith ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 183590ace30eSBarry Smith 183690ace30eSBarry Smith /* read in my part of the matrix numerical values */ 18370752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,m*N,PETSC_SCALAR);CHKERRQ(ierr); 183890ace30eSBarry Smith 183990ace30eSBarry Smith /* insert into matrix-by row (this is why cannot directly read into array */ 184090ace30eSBarry Smith vals_ptr = vals; 184190ace30eSBarry Smith for (i=0; i<m; i++) { 184290ace30eSBarry Smith for (j=0; j<N; j++) { 184390ace30eSBarry Smith array[i + j*m] = *vals_ptr++; 184490ace30eSBarry Smith } 184590ace30eSBarry Smith } 184690ace30eSBarry Smith 184790ace30eSBarry Smith /* read in other processors and ship out */ 184890ace30eSBarry Smith for (i=1; i<size; i++) { 184990ace30eSBarry Smith nz = (rowners[i+1] - rowners[i])*N; 18500752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 18517adad957SLisandro Dalcin ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)(*newmat))->tag,comm);CHKERRQ(ierr); 185290ace30eSBarry Smith } 18533a40ed3dSBarry Smith } else { 185490ace30eSBarry Smith /* receive numeric values */ 185587828ca2SBarry Smith ierr = PetscMalloc(m*N*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 185690ace30eSBarry Smith 185790ace30eSBarry Smith /* receive message of values*/ 18587adad957SLisandro Dalcin ierr = MPI_Recv(vals,m*N,MPIU_SCALAR,0,((PetscObject)(*newmat))->tag,comm,&status);CHKERRQ(ierr); 185990ace30eSBarry Smith 186090ace30eSBarry Smith /* insert into matrix-by row (this is why cannot directly read into array */ 186190ace30eSBarry Smith vals_ptr = vals; 186290ace30eSBarry Smith for (i=0; i<m; i++) { 186390ace30eSBarry Smith for (j=0; j<N; j++) { 186490ace30eSBarry Smith array[i + j*m] = *vals_ptr++; 186590ace30eSBarry Smith } 186690ace30eSBarry Smith } 186790ace30eSBarry Smith } 1868606d414cSSatish Balay ierr = PetscFree(rowners);CHKERRQ(ierr); 1869606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 18706d4a8577SBarry Smith ierr = MatAssemblyBegin(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 18716d4a8577SBarry Smith ierr = MatAssemblyEnd(*newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 18723a40ed3dSBarry Smith PetscFunctionReturn(0); 187390ace30eSBarry Smith } 187490ace30eSBarry Smith 18754a2ae208SSatish Balay #undef __FUNCT__ 18764a2ae208SSatish Balay #define __FUNCT__ "MatLoad_MPIDense" 1877a313700dSBarry Smith PetscErrorCode MatLoad_MPIDense(PetscViewer viewer,const MatType type,Mat *newmat) 18788965ea79SLois Curfman McInnes { 18798965ea79SLois Curfman McInnes Mat A; 188087828ca2SBarry Smith PetscScalar *vals,*svals; 188119bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 18828965ea79SLois Curfman McInnes MPI_Status status; 188313f74950SBarry Smith PetscMPIInt rank,size,tag = ((PetscObject)viewer)->tag,*rowners,*sndcounts,m,maxnz; 188413f74950SBarry Smith PetscInt header[4],*rowlengths = 0,M,N,*cols; 188513f74950SBarry Smith PetscInt *ourlens,*procsnz = 0,*offlens,jj,*mycols,*smycols; 188613f74950SBarry Smith PetscInt i,nz,j,rstart,rend; 188713f74950SBarry Smith int fd; 18886849ba73SBarry Smith PetscErrorCode ierr; 18898965ea79SLois Curfman McInnes 18903a40ed3dSBarry Smith PetscFunctionBegin; 1891d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1892d132466eSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 18938965ea79SLois Curfman McInnes if (!rank) { 1894b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 18950752156aSBarry Smith ierr = PetscBinaryRead(fd,(char *)header,4,PETSC_INT);CHKERRQ(ierr); 1896552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not matrix object"); 18978965ea79SLois Curfman McInnes } 18988965ea79SLois Curfman McInnes 189913f74950SBarry Smith ierr = MPI_Bcast(header+1,3,MPIU_INT,0,comm);CHKERRQ(ierr); 190090ace30eSBarry Smith M = header[1]; N = header[2]; nz = header[3]; 190190ace30eSBarry Smith 190290ace30eSBarry Smith /* 190390ace30eSBarry Smith Handle case where matrix is stored on disk as a dense matrix 190490ace30eSBarry Smith */ 190590ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { 1906be5d1d56SKris Buschelman ierr = MatLoad_MPIDense_DenseInFile(comm,fd,M,N,type,newmat);CHKERRQ(ierr); 19073a40ed3dSBarry Smith PetscFunctionReturn(0); 190890ace30eSBarry Smith } 190990ace30eSBarry Smith 19108965ea79SLois Curfman McInnes /* determine ownership of all rows */ 1911e44c0bd4SBarry Smith m = PetscMPIIntCast(M/size + ((M % size) > rank)); 1912e44c0bd4SBarry Smith ierr = PetscMalloc((size+2)*sizeof(PetscMPIInt),&rowners);CHKERRQ(ierr); 1913ca161407SBarry Smith ierr = MPI_Allgather(&m,1,MPI_INT,rowners+1,1,MPI_INT,comm);CHKERRQ(ierr); 19148965ea79SLois Curfman McInnes rowners[0] = 0; 19158965ea79SLois Curfman McInnes for (i=2; i<=size; i++) { 19168965ea79SLois Curfman McInnes rowners[i] += rowners[i-1]; 19178965ea79SLois Curfman McInnes } 19188965ea79SLois Curfman McInnes rstart = rowners[rank]; 19198965ea79SLois Curfman McInnes rend = rowners[rank+1]; 19208965ea79SLois Curfman McInnes 19218965ea79SLois Curfman McInnes /* distribute row lengths to all processors */ 192213f74950SBarry Smith ierr = PetscMalloc(2*(rend-rstart+1)*sizeof(PetscInt),&ourlens);CHKERRQ(ierr); 19238965ea79SLois Curfman McInnes offlens = ourlens + (rend-rstart); 19248965ea79SLois Curfman McInnes if (!rank) { 192513f74950SBarry Smith ierr = PetscMalloc(M*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 19260752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 192713f74950SBarry Smith ierr = PetscMalloc(size*sizeof(PetscMPIInt),&sndcounts);CHKERRQ(ierr); 19288965ea79SLois Curfman McInnes for (i=0; i<size; i++) sndcounts[i] = rowners[i+1] - rowners[i]; 19291466f1e1SBarry Smith ierr = MPI_Scatterv(rowlengths,sndcounts,rowners,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr); 1930606d414cSSatish Balay ierr = PetscFree(sndcounts);CHKERRQ(ierr); 1931ca161407SBarry Smith } else { 19321466f1e1SBarry Smith ierr = MPI_Scatterv(0,0,0,MPIU_INT,ourlens,rend-rstart,MPIU_INT,0,comm);CHKERRQ(ierr); 19338965ea79SLois Curfman McInnes } 19348965ea79SLois Curfman McInnes 19358965ea79SLois Curfman McInnes if (!rank) { 19368965ea79SLois Curfman McInnes /* calculate the number of nonzeros on each processor */ 193713f74950SBarry Smith ierr = PetscMalloc(size*sizeof(PetscInt),&procsnz);CHKERRQ(ierr); 193813f74950SBarry Smith ierr = PetscMemzero(procsnz,size*sizeof(PetscInt));CHKERRQ(ierr); 19398965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 19408965ea79SLois Curfman McInnes for (j=rowners[i]; j< rowners[i+1]; j++) { 19418965ea79SLois Curfman McInnes procsnz[i] += rowlengths[j]; 19428965ea79SLois Curfman McInnes } 19438965ea79SLois Curfman McInnes } 1944606d414cSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 19458965ea79SLois Curfman McInnes 19468965ea79SLois Curfman McInnes /* determine max buffer needed and allocate it */ 19478965ea79SLois Curfman McInnes maxnz = 0; 19488965ea79SLois Curfman McInnes for (i=0; i<size; i++) { 19490452661fSBarry Smith maxnz = PetscMax(maxnz,procsnz[i]); 19508965ea79SLois Curfman McInnes } 195113f74950SBarry Smith ierr = PetscMalloc(maxnz*sizeof(PetscInt),&cols);CHKERRQ(ierr); 19528965ea79SLois Curfman McInnes 19538965ea79SLois Curfman McInnes /* read in my part of the matrix column indices */ 19548965ea79SLois Curfman McInnes nz = procsnz[0]; 195513f74950SBarry Smith ierr = PetscMalloc(nz*sizeof(PetscInt),&mycols);CHKERRQ(ierr); 19560752156aSBarry Smith ierr = PetscBinaryRead(fd,mycols,nz,PETSC_INT);CHKERRQ(ierr); 19578965ea79SLois Curfman McInnes 19588965ea79SLois Curfman McInnes /* read in every one elses and ship off */ 19598965ea79SLois Curfman McInnes for (i=1; i<size; i++) { 19608965ea79SLois Curfman McInnes nz = procsnz[i]; 19610752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr); 196213f74950SBarry Smith ierr = MPI_Send(cols,nz,MPIU_INT,i,tag,comm);CHKERRQ(ierr); 19638965ea79SLois Curfman McInnes } 1964606d414cSSatish Balay ierr = PetscFree(cols);CHKERRQ(ierr); 19653a40ed3dSBarry Smith } else { 19668965ea79SLois Curfman McInnes /* determine buffer space needed for message */ 19678965ea79SLois Curfman McInnes nz = 0; 19688965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19698965ea79SLois Curfman McInnes nz += ourlens[i]; 19708965ea79SLois Curfman McInnes } 197113f74950SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&mycols);CHKERRQ(ierr); 19728965ea79SLois Curfman McInnes 19738965ea79SLois Curfman McInnes /* receive message of column indices*/ 197413f74950SBarry Smith ierr = MPI_Recv(mycols,nz,MPIU_INT,0,tag,comm,&status);CHKERRQ(ierr); 197513f74950SBarry Smith ierr = MPI_Get_count(&status,MPIU_INT,&maxnz);CHKERRQ(ierr); 197629bbc08cSBarry Smith if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file"); 19778965ea79SLois Curfman McInnes } 19788965ea79SLois Curfman McInnes 19798965ea79SLois Curfman McInnes /* loop over local rows, determining number of off diagonal entries */ 198013f74950SBarry Smith ierr = PetscMemzero(offlens,m*sizeof(PetscInt));CHKERRQ(ierr); 19818965ea79SLois Curfman McInnes jj = 0; 19828965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19838965ea79SLois Curfman McInnes for (j=0; j<ourlens[i]; j++) { 19848965ea79SLois Curfman McInnes if (mycols[jj] < rstart || mycols[jj] >= rend) offlens[i]++; 19858965ea79SLois Curfman McInnes jj++; 19868965ea79SLois Curfman McInnes } 19878965ea79SLois Curfman McInnes } 19888965ea79SLois Curfman McInnes 19898965ea79SLois Curfman McInnes /* create our matrix */ 19908965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19918965ea79SLois Curfman McInnes ourlens[i] -= offlens[i]; 19928965ea79SLois Curfman McInnes } 1993f69a0ea3SMatthew Knepley ierr = MatCreate(comm,newmat);CHKERRQ(ierr); 1994f69a0ea3SMatthew Knepley ierr = MatSetSizes(*newmat,m,PETSC_DECIDE,M,N);CHKERRQ(ierr); 1995be5d1d56SKris Buschelman ierr = MatSetType(*newmat,type);CHKERRQ(ierr); 1996878740d9SKris Buschelman ierr = MatMPIDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr); 19978965ea79SLois Curfman McInnes A = *newmat; 19988965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 19998965ea79SLois Curfman McInnes ourlens[i] += offlens[i]; 20008965ea79SLois Curfman McInnes } 20018965ea79SLois Curfman McInnes 20028965ea79SLois Curfman McInnes if (!rank) { 200387828ca2SBarry Smith ierr = PetscMalloc(maxnz*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 20048965ea79SLois Curfman McInnes 20058965ea79SLois Curfman McInnes /* read in my part of the matrix numerical values */ 20068965ea79SLois Curfman McInnes nz = procsnz[0]; 20070752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 20088965ea79SLois Curfman McInnes 20098965ea79SLois Curfman McInnes /* insert into matrix */ 20108965ea79SLois Curfman McInnes jj = rstart; 20118965ea79SLois Curfman McInnes smycols = mycols; 20128965ea79SLois Curfman McInnes svals = vals; 20138965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20148965ea79SLois Curfman McInnes ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr); 20158965ea79SLois Curfman McInnes smycols += ourlens[i]; 20168965ea79SLois Curfman McInnes svals += ourlens[i]; 20178965ea79SLois Curfman McInnes jj++; 20188965ea79SLois Curfman McInnes } 20198965ea79SLois Curfman McInnes 20208965ea79SLois Curfman McInnes /* read in other processors and ship out */ 20218965ea79SLois Curfman McInnes for (i=1; i<size; i++) { 20228965ea79SLois Curfman McInnes nz = procsnz[i]; 20230752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 20247adad957SLisandro Dalcin ierr = MPI_Send(vals,nz,MPIU_SCALAR,i,((PetscObject)A)->tag,comm);CHKERRQ(ierr); 20258965ea79SLois Curfman McInnes } 2026606d414cSSatish Balay ierr = PetscFree(procsnz);CHKERRQ(ierr); 20273a40ed3dSBarry Smith } else { 20288965ea79SLois Curfman McInnes /* receive numeric values */ 202984d528b1SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr); 20308965ea79SLois Curfman McInnes 20318965ea79SLois Curfman McInnes /* receive message of values*/ 20327adad957SLisandro Dalcin ierr = MPI_Recv(vals,nz,MPIU_SCALAR,0,((PetscObject)A)->tag,comm,&status);CHKERRQ(ierr); 2033ca161407SBarry Smith ierr = MPI_Get_count(&status,MPIU_SCALAR,&maxnz);CHKERRQ(ierr); 203429bbc08cSBarry Smith if (maxnz != nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"something is wrong with file"); 20358965ea79SLois Curfman McInnes 20368965ea79SLois Curfman McInnes /* insert into matrix */ 20378965ea79SLois Curfman McInnes jj = rstart; 20388965ea79SLois Curfman McInnes smycols = mycols; 20398965ea79SLois Curfman McInnes svals = vals; 20408965ea79SLois Curfman McInnes for (i=0; i<m; i++) { 20418965ea79SLois Curfman McInnes ierr = MatSetValues(A,1,&jj,ourlens[i],smycols,svals,INSERT_VALUES);CHKERRQ(ierr); 20428965ea79SLois Curfman McInnes smycols += ourlens[i]; 20438965ea79SLois Curfman McInnes svals += ourlens[i]; 20448965ea79SLois Curfman McInnes jj++; 20458965ea79SLois Curfman McInnes } 20468965ea79SLois Curfman McInnes } 2047606d414cSSatish Balay ierr = PetscFree(ourlens);CHKERRQ(ierr); 2048606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 2049606d414cSSatish Balay ierr = PetscFree(mycols);CHKERRQ(ierr); 2050606d414cSSatish Balay ierr = PetscFree(rowners);CHKERRQ(ierr); 20518965ea79SLois Curfman McInnes 20526d4a8577SBarry Smith ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 20536d4a8577SBarry Smith ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 20543a40ed3dSBarry Smith PetscFunctionReturn(0); 20558965ea79SLois Curfman McInnes } 205690ace30eSBarry Smith 20576e4ee0c6SHong Zhang #undef __FUNCT__ 20586e4ee0c6SHong Zhang #define __FUNCT__ "MatEqual_MPIDense" 20596e4ee0c6SHong Zhang PetscErrorCode MatEqual_MPIDense(Mat A,Mat B,PetscTruth *flag) 20606e4ee0c6SHong Zhang { 20616e4ee0c6SHong Zhang Mat_MPIDense *matB = (Mat_MPIDense*)B->data,*matA = (Mat_MPIDense*)A->data; 20626e4ee0c6SHong Zhang Mat a,b; 20636e4ee0c6SHong Zhang PetscTruth flg; 20646e4ee0c6SHong Zhang PetscErrorCode ierr; 206590ace30eSBarry Smith 20666e4ee0c6SHong Zhang PetscFunctionBegin; 20676e4ee0c6SHong Zhang a = matA->A; 20686e4ee0c6SHong Zhang b = matB->A; 20696e4ee0c6SHong Zhang ierr = MatEqual(a,b,&flg);CHKERRQ(ierr); 20707adad957SLisandro Dalcin ierr = MPI_Allreduce(&flg,flag,1,MPI_INT,MPI_LAND,((PetscObject)A)->comm);CHKERRQ(ierr); 20716e4ee0c6SHong Zhang PetscFunctionReturn(0); 20726e4ee0c6SHong Zhang } 207390ace30eSBarry Smith 207409d27a7eSBarry Smith #if defined(PETSC_HAVE_PLAPACK) 207509d27a7eSBarry Smith 207609d27a7eSBarry Smith #undef __FUNCT__ 207709d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKFinalizePackage" 207809d27a7eSBarry Smith /*@C 207909d27a7eSBarry Smith PetscPLAPACKFinalizePackage - This function destroys everything in the Petsc interface to PLAPACK. 208009d27a7eSBarry Smith Level: developer 208109d27a7eSBarry Smith 208209d27a7eSBarry Smith .keywords: Petsc, destroy, package, PLAPACK 208309d27a7eSBarry Smith .seealso: PetscFinalize() 208409d27a7eSBarry Smith @*/ 208509d27a7eSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKFinalizePackage(void) 208609d27a7eSBarry Smith { 208709d27a7eSBarry Smith PetscErrorCode ierr; 208809d27a7eSBarry Smith 208909d27a7eSBarry Smith PetscFunctionBegin; 209009d27a7eSBarry Smith ierr = PLA_Finalize();CHKERRQ(ierr); 209109d27a7eSBarry Smith PetscFunctionReturn(0); 209209d27a7eSBarry Smith } 209309d27a7eSBarry Smith 209409d27a7eSBarry Smith #undef __FUNCT__ 209509d27a7eSBarry Smith #define __FUNCT__ "PetscPLAPACKInitializePackage" 209609d27a7eSBarry Smith /*@C 209709d27a7eSBarry Smith PetscPLAPACKInitializePackage - This function initializes everything in the Petsc interface to PLAPACK. It is 2098db4efbfdSBarry Smith called from MatCreate_MPIDense() the first time an MPI dense matrix is called. 209909d27a7eSBarry Smith 210009d27a7eSBarry Smith Input Parameter: 2101db4efbfdSBarry Smith . comm - the communicator the matrix lives on 210209d27a7eSBarry Smith 210309d27a7eSBarry Smith Level: developer 210409d27a7eSBarry Smith 2105db4efbfdSBarry Smith Notes: PLAPACK does not have a good fit with MPI communicators; all (parallel) PLAPACK objects have to live in the 2106db4efbfdSBarry Smith same communicator (because there is some global state that is initialized and used for all matrices). In addition if 2107db4efbfdSBarry Smith PLAPACK is initialized (that is the initial matrices created) are on subcommunicators of MPI_COMM_WORLD, these subcommunicators 2108db4efbfdSBarry Smith cannot overlap. 2109db4efbfdSBarry Smith 211009d27a7eSBarry Smith .keywords: Petsc, initialize, package, PLAPACK 211109d27a7eSBarry Smith .seealso: PetscInitializePackage(), PetscInitialize() 211209d27a7eSBarry Smith @*/ 2113db4efbfdSBarry Smith PetscErrorCode PETSC_DLLEXPORT PetscPLAPACKInitializePackage(MPI_Comm comm) 211409d27a7eSBarry Smith { 2115eae6fb2eSBarry Smith PetscMPIInt size; 211609d27a7eSBarry Smith PetscErrorCode ierr; 211709d27a7eSBarry Smith 211809d27a7eSBarry Smith PetscFunctionBegin; 211909d27a7eSBarry Smith if (!PLA_Initialized(PETSC_NULL)) { 212009d27a7eSBarry Smith 212109d27a7eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 212204fea9ffSBarry Smith Plapack_nprows = 1; 212304fea9ffSBarry Smith Plapack_npcols = size; 212409d27a7eSBarry Smith 2125aeccfd6fSBarry Smith ierr = PetscOptionsBegin(comm,PETSC_NULL,"PLAPACK Options","Mat");CHKERRQ(ierr); 21265d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nprows","row dimension of 2D processor mesh","None",Plapack_nprows,&Plapack_nprows,PETSC_NULL);CHKERRQ(ierr); 21275d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_npcols","column dimension of 2D processor mesh","None",Plapack_npcols,&Plapack_npcols,PETSC_NULL);CHKERRQ(ierr); 212879b0a62dSBarry Smith #if defined(PETSC_USE_DEBUG) 212979b0a62dSBarry Smith Plapack_ierror = 3; 213079b0a62dSBarry Smith #else 213179b0a62dSBarry Smith Plapack_ierror = 0; 213279b0a62dSBarry Smith #endif 21335d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_ckerror","error checking flag","None",Plapack_ierror,&Plapack_ierror,PETSC_NULL);CHKERRQ(ierr); 2134eae6fb2eSBarry Smith if (Plapack_ierror){ 2135eae6fb2eSBarry Smith ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_TRUE,PETSC_TRUE,PETSC_FALSE );CHKERRQ(ierr); 2136aeccfd6fSBarry Smith } else { 2137eae6fb2eSBarry Smith ierr = PLA_Set_error_checking(Plapack_ierror,PETSC_FALSE,PETSC_FALSE,PETSC_FALSE );CHKERRQ(ierr); 2138aeccfd6fSBarry Smith } 2139aeccfd6fSBarry Smith 2140eae6fb2eSBarry Smith Plapack_nb_alg = 0; 21415d00a290SHong Zhang ierr = PetscOptionsInt("-mat_plapack_nb_alg","algorithmic block size","None",Plapack_nb_alg,&Plapack_nb_alg,PETSC_NULL);CHKERRQ(ierr); 2142eae6fb2eSBarry Smith if (Plapack_nb_alg) { 2143eae6fb2eSBarry Smith ierr = pla_Environ_set_nb_alg (PLA_OP_ALL_ALG,Plapack_nb_alg);CHKERRQ(ierr); 2144aeccfd6fSBarry Smith } 2145aeccfd6fSBarry Smith PetscOptionsEnd(); 2146aeccfd6fSBarry Smith 214704fea9ffSBarry Smith ierr = PLA_Comm_1D_to_2D(comm,Plapack_nprows,Plapack_npcols,&Plapack_comm_2d);CHKERRQ(ierr); 214804fea9ffSBarry Smith ierr = PLA_Init(Plapack_comm_2d);CHKERRQ(ierr); 214909d27a7eSBarry Smith ierr = PetscRegisterFinalize(PetscPLAPACKFinalizePackage);CHKERRQ(ierr); 215009d27a7eSBarry Smith } 215109d27a7eSBarry Smith PetscFunctionReturn(0); 215209d27a7eSBarry Smith } 215390ace30eSBarry Smith 215409d27a7eSBarry Smith #endif 2155