1db31f6deSJed Brown #include <../src/mat/impls/elemental/matelemimpl.h> /*I "petscmat.h" I*/ 2db31f6deSJed Brown 35e9f5b67SHong Zhang /* 45e9f5b67SHong Zhang The variable Petsc_Elemental_keyval is used to indicate an MPI attribute that 55e9f5b67SHong Zhang is attached to a communicator, in this case the attribute is a Mat_Elemental_Grid 65e9f5b67SHong Zhang */ 75e9f5b67SHong Zhang static PetscMPIInt Petsc_Elemental_keyval = MPI_KEYVAL_INVALID; 85e9f5b67SHong Zhang 9db31f6deSJed Brown #undef __FUNCT__ 10db31f6deSJed Brown #define __FUNCT__ "PetscElementalInitializePackage" 11db31f6deSJed Brown /*@C 12db31f6deSJed Brown PetscElementalInitializePackage - Initialize Elemental package 13db31f6deSJed Brown 14db31f6deSJed Brown Logically Collective 15db31f6deSJed Brown 16db31f6deSJed Brown Input Arguments: 17db31f6deSJed Brown . path - the dynamic library path or PETSC_NULL 18db31f6deSJed Brown 19db31f6deSJed Brown Level: developer 20db31f6deSJed Brown 21db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalFinalizePackage() 22db31f6deSJed Brown @*/ 23db31f6deSJed Brown PetscErrorCode PetscElementalInitializePackage(const char *path) 24db31f6deSJed Brown { 25db31f6deSJed Brown PetscErrorCode ierr; 26db31f6deSJed Brown 27db31f6deSJed Brown PetscFunctionBegin; 28db31f6deSJed Brown if (elem::Initialized()) PetscFunctionReturn(0); 29db31f6deSJed Brown { /* We have already initialized MPI, so this song and dance is just to pass these variables (which won't be used by Elemental) through the interface that needs references */ 30db31f6deSJed Brown int zero = 0; 31db31f6deSJed Brown char **nothing = 0; 32db31f6deSJed Brown elem::Initialize(zero,nothing); 33db31f6deSJed Brown } 34db31f6deSJed Brown ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr); 35db31f6deSJed Brown PetscFunctionReturn(0); 36db31f6deSJed Brown } 37db31f6deSJed Brown 38db31f6deSJed Brown #undef __FUNCT__ 39db31f6deSJed Brown #define __FUNCT__ "PetscElementalFinalizePackage" 40db31f6deSJed Brown /*@C 41db31f6deSJed Brown PetscElementalFinalizePackage - Finalize Elemental package 42db31f6deSJed Brown 43db31f6deSJed Brown Logically Collective 44db31f6deSJed Brown 45db31f6deSJed Brown Level: developer 46db31f6deSJed Brown 47db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalInitializePackage() 48db31f6deSJed Brown @*/ 49db31f6deSJed Brown PetscErrorCode PetscElementalFinalizePackage(void) 50db31f6deSJed Brown { 51db31f6deSJed Brown 52db31f6deSJed Brown PetscFunctionBegin; 53db31f6deSJed Brown elem::Finalize(); 54db31f6deSJed Brown PetscFunctionReturn(0); 55db31f6deSJed Brown } 56db31f6deSJed Brown 57ed667823SXuan Zhou /* Sets Elemental options from the options database */ 58ed667823SXuan Zhou #undef __FUNCT__ 59ed667823SXuan Zhou #define __FUNCT__ "PetscSetElementalFromOptions" 60ed667823SXuan Zhou PetscErrorCode PetscSetElementalFromOptions(Mat A) 61ed667823SXuan Zhou { 62ed667823SXuan Zhou //Mat_Elemental *a = (Mat_Elemental*)A->data; 63ed667823SXuan Zhou PetscErrorCode ierr; 64ed667823SXuan Zhou //PetscInt optv1,optv2; 65ed667823SXuan Zhou //PetscBool flg; 66ed667823SXuan Zhou 67ed667823SXuan Zhou PetscFunctionBegin; 68ed667823SXuan Zhou ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 69ed667823SXuan Zhou //ierr = PetscOptionsInt("-mat_elemental_grid_height","Gird Height","None",(*a->emat).Grid().Height(),&optv1,&flg);CHKERRQ(ierr); 70ed667823SXuan Zhou //ierr = PetscOptionsInt("-mat_elemental_grid_width","Gird Width","None",(*a->emat).Grid().Width(),&optv2,&flg);CHKERRQ(ierr); 71ed667823SXuan Zhou //if (flg) (*a->emat).Grid().ResizeTo(optv1,optv2); 72ed667823SXuan Zhou // ierr = PetscOptionsInt("-mat_mumps_icntl_2","ICNTL(2): output stream for diagnostic printing, statistics, and warning","None",mumps->id.ICNTL(2),&icntl,&flg);CHKERRQ(ierr); 73ed667823SXuan Zhou // if (flg) mumps->id.ICNTL(2) = icntl; 74ed667823SXuan Zhou // ierr = PetscOptionsInt("-mat_mumps_icntl_3","ICNTL(3): output stream for global information, collected on the host","None",mumps->id.ICNTL(3),&icntl,&flg);CHKERRQ(ierr); 75ed667823SXuan Zhou // if (flg) mumps->id.ICNTL(3) = icntl; 76ed667823SXuan Zhou 77ed667823SXuan Zhou 78ed667823SXuan Zhou PetscOptionsEnd(); 79ed667823SXuan Zhou PetscFunctionReturn(0); 80ed667823SXuan Zhou } 81ed667823SXuan Zhou 82db31f6deSJed Brown #undef __FUNCT__ 83db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental" 84db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer) 85db31f6deSJed Brown { 86db31f6deSJed Brown PetscErrorCode ierr; 87db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 88db31f6deSJed Brown PetscBool iascii; 89db31f6deSJed Brown 90db31f6deSJed Brown PetscFunctionBegin; 91db31f6deSJed Brown ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 92db31f6deSJed Brown if (iascii) { 93db31f6deSJed Brown PetscViewerFormat format; 94db31f6deSJed Brown ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 95db31f6deSJed Brown if (format == PETSC_VIEWER_ASCII_INFO) { 9679673f7bSHong Zhang /* call elemental viewing function */ 972d8adcc7SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr); 98ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr); 99ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr); 1004fe7bbcaSHong Zhang if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 10179673f7bSHong Zhang /* call elemental viewing function */ 10280622e0cSXuan Zhou ierr = PetscPrintf(((PetscObject)viewer)->comm,"test matview_elemental 2\n");CHKERRQ(ierr); 1034fe7bbcaSHong Zhang } 10479673f7bSHong Zhang 105db31f6deSJed Brown } else if (format == PETSC_VIEWER_DEFAULT) { 106db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 107db31f6deSJed Brown ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr); 108db31f6deSJed Brown a->emat->Print("Elemental matrix (cyclic ordering)"); 109db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 110834d3fecSHong Zhang if (A->factortype == MAT_FACTOR_NONE){ 111d2daa67eSHong Zhang Mat Aaij; 1120022473aSHong Zhang ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 1130022473aSHong Zhang ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr); 114d2daa67eSHong Zhang ierr = MatView(Aaij,viewer);CHKERRQ(ierr); 1150022473aSHong Zhang ierr = MatDestroy(&Aaij);CHKERRQ(ierr); 116834d3fecSHong Zhang } 117db31f6deSJed Brown } else SETERRQ(((PetscObject)viewer)->comm,PETSC_ERR_SUP,"Format"); 118d2daa67eSHong Zhang } else { 119d2daa67eSHong Zhang /* convert to aij/mpidense format and call MatView() */ 120d2daa67eSHong Zhang Mat Aaij; 121d2daa67eSHong Zhang ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 122d2daa67eSHong Zhang ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr); 123d2daa67eSHong Zhang ierr = MatView(Aaij,viewer);CHKERRQ(ierr); 124d2daa67eSHong Zhang ierr = MatDestroy(&Aaij);CHKERRQ(ierr); 125d2daa67eSHong Zhang } 126db31f6deSJed Brown PetscFunctionReturn(0); 127db31f6deSJed Brown } 128db31f6deSJed Brown 129db31f6deSJed Brown #undef __FUNCT__ 130180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental" 13115767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info) 132180a43e4SHong Zhang { 13315767789SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 13415767789SHong Zhang PetscMPIInt rank; 13515767789SHong Zhang 136180a43e4SHong Zhang PetscFunctionBegin; 13715767789SHong Zhang MPI_Comm_rank(((PetscObject)A)->comm,&rank); 13815767789SHong Zhang 13915767789SHong Zhang /* if (!rank) printf(" .........MatGetInfo_Elemental ...\n"); */ 14015767789SHong Zhang info->block_size = 1.0; /* ? */ 14115767789SHong Zhang 14215767789SHong Zhang if (flag == MAT_LOCAL) { 14315767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 14415767789SHong Zhang info->nz_used = info->nz_allocated; 14515767789SHong Zhang } else if (flag == MAT_GLOBAL_MAX) { 14615767789SHong Zhang //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,((PetscObject)matin)->comm);CHKERRQ(ierr); 14715767789SHong Zhang /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */ 14815767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet"); 14915767789SHong Zhang } else if (flag == MAT_GLOBAL_SUM) { 15015767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet"); 15115767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 15215767789SHong Zhang info->nz_used = info->nz_allocated; /* assume Elemental does accurate allocation */ 15315767789SHong Zhang //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 15415767789SHong Zhang //PetscPrintf(PETSC_COMM_SELF," ... [%d] locally allocated %g\n",rank,info->nz_allocated); 15515767789SHong Zhang } 15615767789SHong Zhang 15715767789SHong Zhang info->nz_unneeded = 0.0; 15815767789SHong Zhang info->assemblies = (double)A->num_ass; 15915767789SHong Zhang info->mallocs = 0; 16015767789SHong Zhang info->memory = ((PetscObject)A)->mem; 16115767789SHong Zhang info->fill_ratio_given = 0; /* determined by Elemental */ 16215767789SHong Zhang info->fill_ratio_needed = 0; 16315767789SHong Zhang info->factor_mallocs = 0; 164180a43e4SHong Zhang PetscFunctionReturn(0); 165180a43e4SHong Zhang } 166180a43e4SHong Zhang 167180a43e4SHong Zhang #undef __FUNCT__ 168db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental" 169*e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode) 170db31f6deSJed Brown { 171db31f6deSJed Brown PetscErrorCode ierr; 172db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 173db31f6deSJed Brown PetscMPIInt rank; 174db31f6deSJed Brown PetscInt i,j,rrank,ridx,crank,cidx; 175db31f6deSJed Brown 176db31f6deSJed Brown PetscFunctionBegin; 177db31f6deSJed Brown ierr = MPI_Comm_rank(((PetscObject)A)->comm,&rank);CHKERRQ(ierr); 178db31f6deSJed Brown 179db31f6deSJed Brown const elem::Grid &grid = a->emat->Grid(); 180db31f6deSJed Brown for (i=0; i<nr; i++) { 181db31f6deSJed Brown PetscInt erow,ecol,elrow,elcol; 182db31f6deSJed Brown if (rows[i] < 0) continue; 183db31f6deSJed Brown P2RO(A,0,rows[i],&rrank,&ridx); 184db31f6deSJed Brown RO2E(A,0,rrank,ridx,&erow); 185db31f6deSJed Brown if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect row translation"); 186db31f6deSJed Brown for (j=0; j<nc; j++) { 187db31f6deSJed Brown if (cols[j] < 0) continue; 188db31f6deSJed Brown P2RO(A,1,cols[j],&crank,&cidx); 189db31f6deSJed Brown RO2E(A,1,crank,cidx,&ecol); 190db31f6deSJed Brown if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect col translation"); 191aae2c449SHong Zhang if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */ 192aae2c449SHong Zhang if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 193aae2c449SHong Zhang /* PetscPrintf(PETSC_COMM_SELF,"[%D] add off-proc entry (%D,%D, %g) (%D %D)\n",rank,rows[i],cols[j],*(vals+i*nc),erow,ecol); */ 194*e6dea9dbSXuan Zhou a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]); 195aae2c449SHong Zhang a->interface->Axpy(1.0,*(a->esubmat),erow,ecol); 196aae2c449SHong Zhang continue; 197ed36708cSHong Zhang } 198db31f6deSJed Brown elrow = erow / grid.MCSize(); 199db31f6deSJed Brown elcol = ecol / grid.MRSize(); 200db31f6deSJed Brown switch (imode) { 201*e6dea9dbSXuan Zhou case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 202*e6dea9dbSXuan Zhou case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 203db31f6deSJed Brown default: SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 204db31f6deSJed Brown } 205db31f6deSJed Brown } 206db31f6deSJed Brown } 207db31f6deSJed Brown PetscFunctionReturn(0); 208db31f6deSJed Brown } 209db31f6deSJed Brown 210db31f6deSJed Brown #undef __FUNCT__ 211db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental" 212db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y) 213db31f6deSJed Brown { 214db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 215db31f6deSJed Brown PetscErrorCode ierr; 216*e6dea9dbSXuan Zhou const PetscElemScalar *x; 217*e6dea9dbSXuan Zhou PetscElemScalar *y; 218df311e6cSXuan Zhou PetscElemScalar one = 1,zero = 0; 219db31f6deSJed Brown 220db31f6deSJed Brown PetscFunctionBegin; 221*e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 222*e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 223db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 224df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid); 225df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,y,A->rmap->n,*a->grid); 226db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye); 227db31f6deSJed Brown } 228*e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 229*e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 230db31f6deSJed Brown PetscFunctionReturn(0); 231db31f6deSJed Brown } 232db31f6deSJed Brown 233db31f6deSJed Brown #undef __FUNCT__ 2349426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental" 2359426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y) 2369426833fSXuan Zhou { 2379426833fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 2389426833fSXuan Zhou PetscErrorCode ierr; 239df311e6cSXuan Zhou const PetscElemScalar *x; 240df311e6cSXuan Zhou PetscElemScalar *y; 241*e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 2429426833fSXuan Zhou 2439426833fSXuan Zhou PetscFunctionBegin; 244*e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 245*e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2469426833fSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 247df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 248df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,y,A->cmap->n,*a->grid); 2499426833fSXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye); 2509426833fSXuan Zhou } 251*e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 252*e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2539426833fSXuan Zhou PetscFunctionReturn(0); 2549426833fSXuan Zhou } 2559426833fSXuan Zhou 2569426833fSXuan Zhou #undef __FUNCT__ 257db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental" 258db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 259db31f6deSJed Brown { 260db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 261db31f6deSJed Brown PetscErrorCode ierr; 262df311e6cSXuan Zhou const PetscElemScalar *x; 263df311e6cSXuan Zhou PetscElemScalar *z; 264*e6dea9dbSXuan Zhou PetscElemScalar one = 1; 265db31f6deSJed Brown 266db31f6deSJed Brown PetscFunctionBegin; 267db31f6deSJed Brown if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 268*e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 269*e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 270db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 271df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid); 272df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,z,A->rmap->n,*a->grid); 273db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze); 274db31f6deSJed Brown } 275*e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 276*e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 277db31f6deSJed Brown PetscFunctionReturn(0); 278db31f6deSJed Brown } 279db31f6deSJed Brown 280db31f6deSJed Brown #undef __FUNCT__ 281e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental" 282e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 283e883f9d5SXuan Zhou { 284e883f9d5SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 285e883f9d5SXuan Zhou PetscErrorCode ierr; 286df311e6cSXuan Zhou const PetscElemScalar *x; 287df311e6cSXuan Zhou PetscElemScalar *z; 288*e6dea9dbSXuan Zhou PetscElemScalar one = 1; 289e883f9d5SXuan Zhou 290e883f9d5SXuan Zhou PetscFunctionBegin; 291e883f9d5SXuan Zhou if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 292*e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 293*e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 294e883f9d5SXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 295df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 296df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,z,A->cmap->n,*a->grid); 297e883f9d5SXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze); 298e883f9d5SXuan Zhou } 299*e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 300*e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 301e883f9d5SXuan Zhou PetscFunctionReturn(0); 302e883f9d5SXuan Zhou } 303e883f9d5SXuan Zhou 304e883f9d5SXuan Zhou #undef __FUNCT__ 3059a9e8502SHong Zhang #define __FUNCT__ "MatMatMultNumeric_Elemental" 3069a9e8502SHong Zhang static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C) 307c1d1b975SXuan Zhou { 308c1d1b975SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 309c1d1b975SXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 3109a9e8502SHong Zhang Mat_Elemental *c = (Mat_Elemental*)C->data; 311*e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 312c1d1b975SXuan Zhou 313c1d1b975SXuan Zhou PetscFunctionBegin; 314aae2c449SHong Zhang { /* Scoping so that constructor is called before pointer is returned */ 315c1d1b975SXuan Zhou elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat); 316aae2c449SHong Zhang } 3179a9e8502SHong Zhang C->assembled = PETSC_TRUE; 3189a9e8502SHong Zhang PetscFunctionReturn(0); 3199a9e8502SHong Zhang } 3209a9e8502SHong Zhang 3219a9e8502SHong Zhang #undef __FUNCT__ 3229a9e8502SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_Elemental" 3239a9e8502SHong Zhang static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 3249a9e8502SHong Zhang { 3259a9e8502SHong Zhang PetscErrorCode ierr; 3269a9e8502SHong Zhang Mat Ce; 3279a9e8502SHong Zhang MPI_Comm comm=((PetscObject)A)->comm; 3289a9e8502SHong Zhang 3299a9e8502SHong Zhang PetscFunctionBegin; 3309a9e8502SHong Zhang ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 3319a9e8502SHong Zhang ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 3329a9e8502SHong Zhang ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 3339a9e8502SHong Zhang ierr = MatSetUp(Ce);CHKERRQ(ierr); 3349a9e8502SHong Zhang *C = Ce; 3359a9e8502SHong Zhang PetscFunctionReturn(0); 3369a9e8502SHong Zhang } 3379a9e8502SHong Zhang 3389a9e8502SHong Zhang #undef __FUNCT__ 3399a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental" 3409a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 3419a9e8502SHong Zhang { 3429a9e8502SHong Zhang PetscErrorCode ierr; 3439a9e8502SHong Zhang 3449a9e8502SHong Zhang PetscFunctionBegin; 3459a9e8502SHong Zhang if (scall == MAT_INITIAL_MATRIX){ 3469a9e8502SHong Zhang ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3479a9e8502SHong Zhang ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 3489a9e8502SHong Zhang ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3499a9e8502SHong Zhang } 3509a9e8502SHong Zhang ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3519a9e8502SHong Zhang ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 3529a9e8502SHong Zhang ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 353c1d1b975SXuan Zhou PetscFunctionReturn(0); 354c1d1b975SXuan Zhou } 355c1d1b975SXuan Zhou 356c1d1b975SXuan Zhou #undef __FUNCT__ 357df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental" 358df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C) 359df311e6cSXuan Zhou { 360df311e6cSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 361df311e6cSXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 362df311e6cSXuan Zhou Mat_Elemental *c = (Mat_Elemental*)C->data; 363*e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 364df311e6cSXuan Zhou 365df311e6cSXuan Zhou PetscFunctionBegin; 366df311e6cSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 367df311e6cSXuan Zhou elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat); 368df311e6cSXuan Zhou } 369df311e6cSXuan Zhou C->assembled = PETSC_TRUE; 370df311e6cSXuan Zhou PetscFunctionReturn(0); 371df311e6cSXuan Zhou } 372df311e6cSXuan Zhou 373df311e6cSXuan Zhou #undef __FUNCT__ 374df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental" 375df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 376df311e6cSXuan Zhou { 377df311e6cSXuan Zhou PetscErrorCode ierr; 378df311e6cSXuan Zhou Mat Ce; 379df311e6cSXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 380df311e6cSXuan Zhou 381df311e6cSXuan Zhou PetscFunctionBegin; 382df311e6cSXuan Zhou ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 383df311e6cSXuan Zhou ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 384df311e6cSXuan Zhou ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 385df311e6cSXuan Zhou ierr = MatSetUp(Ce);CHKERRQ(ierr); 386df311e6cSXuan Zhou *C = Ce; 387df311e6cSXuan Zhou PetscFunctionReturn(0); 388df311e6cSXuan Zhou } 389df311e6cSXuan Zhou 390df311e6cSXuan Zhou #undef __FUNCT__ 391df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental" 392df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 393df311e6cSXuan Zhou { 394df311e6cSXuan Zhou PetscErrorCode ierr; 395df311e6cSXuan Zhou 396df311e6cSXuan Zhou PetscFunctionBegin; 397df311e6cSXuan Zhou if (scall == MAT_INITIAL_MATRIX){ 398df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 399df311e6cSXuan Zhou ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 400df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 401df311e6cSXuan Zhou } 402df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 403df311e6cSXuan Zhou ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 404df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 405df311e6cSXuan Zhou PetscFunctionReturn(0); 406df311e6cSXuan Zhou } 407df311e6cSXuan Zhou 408df311e6cSXuan Zhou #undef __FUNCT__ 4094ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental" 410*e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a) 41165b78793SXuan Zhou { 41265b78793SXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 41365b78793SXuan Zhou 41465b78793SXuan Zhou PetscFunctionBegin; 415*e6dea9dbSXuan Zhou elem::Scal((PetscElemScalar)a,*x->emat); 41665b78793SXuan Zhou PetscFunctionReturn(0); 41765b78793SXuan Zhou } 41865b78793SXuan Zhou 41965b78793SXuan Zhou #undef __FUNCT__ 4204ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental" 421*e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str) 422e09a3074SHong Zhang { 423e09a3074SHong Zhang Mat_Elemental *x = (Mat_Elemental*)X->data; 424e09a3074SHong Zhang Mat_Elemental *y = (Mat_Elemental*)Y->data; 425e09a3074SHong Zhang 426e09a3074SHong Zhang PetscFunctionBegin; 427*e6dea9dbSXuan Zhou elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat); 428e09a3074SHong Zhang PetscFunctionReturn(0); 429e09a3074SHong Zhang } 430e09a3074SHong Zhang 431ae844d54SHong Zhang #undef __FUNCT__ 432d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental" 433d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str) 434d6223691SXuan Zhou { 435d6223691SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 436d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 437d6223691SXuan Zhou 438d6223691SXuan Zhou PetscFunctionBegin; 439d6223691SXuan Zhou elem::Copy(*a->emat,*b->emat); 440d6223691SXuan Zhou PetscFunctionReturn(0); 441d6223691SXuan Zhou } 442d6223691SXuan Zhou 443d6223691SXuan Zhou #undef __FUNCT__ 444df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental" 445df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B) 446df311e6cSXuan Zhou { 447df311e6cSXuan Zhou Mat Be; 448df311e6cSXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 449df311e6cSXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 450df311e6cSXuan Zhou PetscErrorCode ierr; 451df311e6cSXuan Zhou 452df311e6cSXuan Zhou PetscFunctionBegin; 453df311e6cSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 454df311e6cSXuan Zhou ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 455df311e6cSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 456df311e6cSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 457df311e6cSXuan Zhou *B = Be; 458df311e6cSXuan Zhou if (op == MAT_COPY_VALUES) { 459df311e6cSXuan Zhou Mat_Elemental *b=(Mat_Elemental*)Be->data; 460df311e6cSXuan Zhou elem::Copy(*a->emat,*b->emat); 461df311e6cSXuan Zhou } 462df311e6cSXuan Zhou Be->assembled = PETSC_TRUE; 463df311e6cSXuan Zhou PetscFunctionReturn(0); 464df311e6cSXuan Zhou } 465df311e6cSXuan Zhou 466df311e6cSXuan Zhou #undef __FUNCT__ 467d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental" 468d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 469d6223691SXuan Zhou { 4703512f328SXuan Zhou /* Only out-of-place supported */ 4715262d616SXuan Zhou Mat Be; 4725262d616SXuan Zhou PetscErrorCode ierr; 4735262d616SXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 4744fe7bbcaSHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 475d6223691SXuan Zhou 476d6223691SXuan Zhou PetscFunctionBegin; 4775262d616SXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 4785262d616SXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 4795262d616SXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 4805262d616SXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 4815262d616SXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 4825262d616SXuan Zhou *B = Be; 4835262d616SXuan Zhou } 4844fe7bbcaSHong Zhang b = (Mat_Elemental*)Be->data; 4855262d616SXuan Zhou elem::Transpose(*a->emat,*b->emat); 4865262d616SXuan Zhou Be->assembled = PETSC_TRUE; 487d6223691SXuan Zhou PetscFunctionReturn(0); 488d6223691SXuan Zhou } 489d6223691SXuan Zhou 490d6223691SXuan Zhou #undef __FUNCT__ 4911f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental" 4921f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X) 4931f881ff8SXuan Zhou { 4941f881ff8SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 4951f881ff8SXuan Zhou PetscErrorCode ierr; 496df311e6cSXuan Zhou PetscElemScalar *x; 4971f881ff8SXuan Zhou 4981f881ff8SXuan Zhou PetscFunctionBegin; 49945cf121fSXuan Zhou ierr = VecCopy(B,X);CHKERRQ(ierr); 500*e6dea9dbSXuan Zhou ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 501df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 502df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe; 503fc54b460SXuan Zhou switch (A->factortype) { 504fc54b460SXuan Zhou case MAT_FACTOR_LU: 5051f881ff8SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 5064b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer); 50745cf121fSXuan Zhou elem::Copy(xer,xe); 508fc54b460SXuan Zhou } else { 5094b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,xer); 51045cf121fSXuan Zhou elem::Copy(xer,xe); 5111f881ff8SXuan Zhou } 512fc54b460SXuan Zhou break; 513fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 514fc54b460SXuan Zhou elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer); 515fc54b460SXuan Zhou elem::Copy(xer,xe); 516fc54b460SXuan Zhou break; 517fc54b460SXuan Zhou default: 5184fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 519fc54b460SXuan Zhou break; 5201f881ff8SXuan Zhou } 521*e6dea9dbSXuan Zhou ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 5221f881ff8SXuan Zhou PetscFunctionReturn(0); 5231f881ff8SXuan Zhou } 5241f881ff8SXuan Zhou 5251f881ff8SXuan Zhou #undef __FUNCT__ 526df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental" 527df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X) 528df311e6cSXuan Zhou { 529df311e6cSXuan Zhou PetscErrorCode ierr; 530df311e6cSXuan Zhou 531df311e6cSXuan Zhou PetscFunctionBegin; 532df311e6cSXuan Zhou ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr); 5333d7f40dbSXuan Zhou ierr = VecAXPY(X,1,Y);CHKERRQ(ierr); 534df311e6cSXuan Zhou PetscFunctionReturn(0); 535df311e6cSXuan Zhou } 536df311e6cSXuan Zhou 537df311e6cSXuan Zhou #undef __FUNCT__ 538ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental" 539ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X) 540ae844d54SHong Zhang { 5411f0e42cfSHong Zhang Mat_Elemental *a=(Mat_Elemental*)A->data; 542d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 5431f0e42cfSHong Zhang Mat_Elemental *x=(Mat_Elemental*)X->data; 5441f0e42cfSHong Zhang 545ae844d54SHong Zhang PetscFunctionBegin; 546d6223691SXuan Zhou elem::Copy(*b->emat,*x->emat); 547fc54b460SXuan Zhou switch (A->factortype) { 548fc54b460SXuan Zhou case MAT_FACTOR_LU: 549d6223691SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 5504b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat); 551fc54b460SXuan Zhou } else { 5524b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat); 553d6223691SXuan Zhou } 554fc54b460SXuan Zhou break; 555fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 556fc54b460SXuan Zhou elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat); 557fc54b460SXuan Zhou break; 558fc54b460SXuan Zhou default: 5594fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 560fc54b460SXuan Zhou break; 561fc54b460SXuan Zhou } 562ae844d54SHong Zhang PetscFunctionReturn(0); 563ae844d54SHong Zhang } 564ae844d54SHong Zhang 565ae844d54SHong Zhang #undef __FUNCT__ 566ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental" 567ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info) 568ae844d54SHong Zhang { 5697c920d81SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 5707c920d81SXuan Zhou 571ae844d54SHong Zhang PetscFunctionBegin; 572d6223691SXuan Zhou if (info->dtcol){ 5737c920d81SXuan Zhou elem::LU(*a->emat,*a->pivot); 5741293973dSHong Zhang } else { 575d6223691SXuan Zhou elem::LU(*a->emat); 576d6223691SXuan Zhou } 5771293973dSHong Zhang A->factortype = MAT_FACTOR_LU; 578834d3fecSHong Zhang A->assembled = PETSC_TRUE; 579ae844d54SHong Zhang PetscFunctionReturn(0); 580ae844d54SHong Zhang } 581ae844d54SHong Zhang 582d7c3f9d8SHong Zhang #undef __FUNCT__ 583d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental" 584d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 585d7c3f9d8SHong Zhang { 586d7c3f9d8SHong Zhang PetscErrorCode ierr; 587d7c3f9d8SHong Zhang 588d7c3f9d8SHong Zhang PetscFunctionBegin; 589d7c3f9d8SHong Zhang ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 590d7c3f9d8SHong Zhang ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr); 591d7c3f9d8SHong Zhang PetscFunctionReturn(0); 592d7c3f9d8SHong Zhang } 593d7c3f9d8SHong Zhang 594d7c3f9d8SHong Zhang #undef __FUNCT__ 595d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental" 596d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 597d7c3f9d8SHong Zhang { 598d7c3f9d8SHong Zhang PetscFunctionBegin; 599d7c3f9d8SHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 600d7c3f9d8SHong Zhang PetscFunctionReturn(0); 601d7c3f9d8SHong Zhang } 602d7c3f9d8SHong Zhang 60345cf121fSXuan Zhou #undef __FUNCT__ 60445cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental" 60545cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info) 60645cf121fSXuan Zhou { 60745cf121fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 608df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d; 60945cf121fSXuan Zhou 61045cf121fSXuan Zhou PetscFunctionBegin; 611c9fc186eSXuan Zhou elem::Cholesky(elem::UPPER,*a->emat); 612fc54b460SXuan Zhou A->factortype = MAT_FACTOR_CHOLESKY; 613834d3fecSHong Zhang A->assembled = PETSC_TRUE; 61445cf121fSXuan Zhou PetscFunctionReturn(0); 61545cf121fSXuan Zhou } 61645cf121fSXuan Zhou 61779673f7bSHong Zhang #undef __FUNCT__ 61879673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental" 61979673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 62079673f7bSHong Zhang { 621cb76c1d8SXuan Zhou PetscErrorCode ierr; 622cb76c1d8SXuan Zhou 623cb76c1d8SXuan Zhou PetscFunctionBegin; 624cb76c1d8SXuan Zhou ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 625cb76c1d8SXuan Zhou ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr); 626cb76c1d8SXuan Zhou PetscFunctionReturn(0); 62779673f7bSHong Zhang } 628cb76c1d8SXuan Zhou 62979673f7bSHong Zhang #undef __FUNCT__ 63079673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental" 63179673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info) 63279673f7bSHong Zhang { 63379673f7bSHong Zhang PetscFunctionBegin; 63479673f7bSHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 63579673f7bSHong Zhang PetscFunctionReturn(0); 63679673f7bSHong Zhang } 63779673f7bSHong Zhang 63815767789SHong Zhang EXTERN_C_BEGIN 6391293973dSHong Zhang #undef __FUNCT__ 64015767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental" 64115767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type) 64215767789SHong Zhang { 64315767789SHong Zhang PetscFunctionBegin; 64415767789SHong Zhang *type = MATSOLVERELEMENTAL; 64515767789SHong Zhang PetscFunctionReturn(0); 64615767789SHong Zhang } 64715767789SHong Zhang EXTERN_C_END 64815767789SHong Zhang 64915767789SHong Zhang EXTERN_C_BEGIN 65015767789SHong Zhang #undef __FUNCT__ 65115767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental" 65215767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F) 6531293973dSHong Zhang { 6541293973dSHong Zhang Mat B; 6551293973dSHong Zhang PetscErrorCode ierr; 6561293973dSHong Zhang 6571293973dSHong Zhang PetscFunctionBegin; 6581293973dSHong Zhang /* Create the factorization matrix */ 6591293973dSHong Zhang ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 6601293973dSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 6611293973dSHong Zhang ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr); 6621293973dSHong Zhang ierr = MatSetUp(B);CHKERRQ(ierr); 6631293973dSHong Zhang B->factortype = ftype; 66415767789SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_elemental_elemental",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr); 6651293973dSHong Zhang *F = B; 6661293973dSHong Zhang PetscFunctionReturn(0); 6671293973dSHong Zhang } 66815767789SHong Zhang EXTERN_C_END 6691293973dSHong Zhang 6701f881ff8SXuan Zhou #undef __FUNCT__ 6711f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental" 6721f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm) 6731f881ff8SXuan Zhou { 6741f881ff8SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 6751f881ff8SXuan Zhou 6761f881ff8SXuan Zhou PetscFunctionBegin; 6771f881ff8SXuan Zhou switch (type){ 6781f881ff8SXuan Zhou case NORM_1: 6791f881ff8SXuan Zhou *nrm = elem::Norm(*a->emat,elem::ONE_NORM); 6801f881ff8SXuan Zhou break; 6811f881ff8SXuan Zhou case NORM_FROBENIUS: 6821f881ff8SXuan Zhou *nrm = elem::Norm(*a->emat,elem::FROBENIUS_NORM); 6831f881ff8SXuan Zhou break; 6841f881ff8SXuan Zhou case NORM_INFINITY: 6851f881ff8SXuan Zhou *nrm = elem::Norm(*a->emat,elem::INFINITY_NORM); 6861f881ff8SXuan Zhou break; 6871f881ff8SXuan Zhou default: 6881f881ff8SXuan Zhou printf("Error: unsupported norm type!\n"); 6891f881ff8SXuan Zhou } 6901f881ff8SXuan Zhou PetscFunctionReturn(0); 6911f881ff8SXuan Zhou } 6921f881ff8SXuan Zhou 6935262d616SXuan Zhou #undef __FUNCT__ 6945262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental" 6955262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A) 6965262d616SXuan Zhou { 6975262d616SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 6985262d616SXuan Zhou 6995262d616SXuan Zhou PetscFunctionBegin; 7005262d616SXuan Zhou elem::Zero(*a->emat); 7015262d616SXuan Zhou PetscFunctionReturn(0); 7025262d616SXuan Zhou } 7035262d616SXuan Zhou 7041293973dSHong Zhang EXTERN_C_BEGIN 705e09a3074SHong Zhang #undef __FUNCT__ 706db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental" 707db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols) 708db31f6deSJed Brown { 709db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 710db31f6deSJed Brown PetscErrorCode ierr; 711db31f6deSJed Brown PetscInt i,m,shift,stride,*idx; 712db31f6deSJed Brown 713db31f6deSJed Brown PetscFunctionBegin; 714db31f6deSJed Brown if (rows) { 715db31f6deSJed Brown m = a->emat->LocalHeight(); 716db31f6deSJed Brown shift = a->emat->ColShift(); 717db31f6deSJed Brown stride = a->emat->ColStride(); 718db31f6deSJed Brown ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr); 719db31f6deSJed Brown for (i=0; i<m; i++) { 720db31f6deSJed Brown PetscInt rank,offset; 721db31f6deSJed Brown E2RO(A,0,shift+i*stride,&rank,&offset); 722db31f6deSJed Brown RO2P(A,0,rank,offset,&idx[i]); 723db31f6deSJed Brown } 724db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr); 725db31f6deSJed Brown } 726db31f6deSJed Brown if (cols) { 727db31f6deSJed Brown m = a->emat->LocalWidth(); 728db31f6deSJed Brown shift = a->emat->RowShift(); 729db31f6deSJed Brown stride = a->emat->RowStride(); 730db31f6deSJed Brown ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr); 731db31f6deSJed Brown for (i=0; i<m; i++) { 732db31f6deSJed Brown PetscInt rank,offset; 733db31f6deSJed Brown E2RO(A,1,shift+i*stride,&rank,&offset); 734db31f6deSJed Brown RO2P(A,1,rank,offset,&idx[i]); 735db31f6deSJed Brown } 736db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr); 737db31f6deSJed Brown } 738db31f6deSJed Brown PetscFunctionReturn(0); 739db31f6deSJed Brown } 7401293973dSHong Zhang EXTERN_C_END 741db31f6deSJed Brown 742db31f6deSJed Brown #undef __FUNCT__ 7432ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense" 7442ef0cf24SXuan Zhou static PetscErrorCode MatConvert_Elemental_Dense(Mat A,const MatType newtype,MatReuse reuse,Mat *B) 745af295397SXuan Zhou { 7462ef0cf24SXuan Zhou Mat Bmpi; 747af295397SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 748af295397SXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 7492ef0cf24SXuan Zhou PetscErrorCode ierr; 7502ef0cf24SXuan Zhou PetscInt rrank,ridx,crank,cidx,nrows,ncols,i,j; 751df311e6cSXuan Zhou PetscElemScalar v; 752af295397SXuan Zhou 753af295397SXuan Zhou PetscFunctionBegin; 754c4ad791aSXuan Zhou if (strcmp(newtype,MATDENSE) && strcmp(newtype,MATSEQDENSE) && strcmp(newtype,MATMPIDENSE)) { 755c4ad791aSXuan Zhou SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE"); 756c4ad791aSXuan Zhou } 757af295397SXuan Zhou ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr); 758af295397SXuan Zhou ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 7592ef0cf24SXuan Zhou ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr); 760af295397SXuan Zhou ierr = MatSetUp(Bmpi);CHKERRQ(ierr); 7612ef0cf24SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 7622ef0cf24SXuan Zhou for (i=0; i<nrows; i++) { 7632ef0cf24SXuan Zhou PetscInt erow,ecol; 7642ef0cf24SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 7652ef0cf24SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 7662ef0cf24SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 7672ef0cf24SXuan Zhou for (j=0; j<ncols; j++) { 7682ef0cf24SXuan Zhou P2RO(A,1,j,&crank,&cidx); 7692ef0cf24SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 7702ef0cf24SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 7712ef0cf24SXuan Zhou v = a->emat->Get(erow,ecol); 772*e6dea9dbSXuan Zhou ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 7732ef0cf24SXuan Zhou } 7742ef0cf24SXuan Zhou } 775af295397SXuan Zhou ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 776af295397SXuan Zhou ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 777c4ad791aSXuan Zhou if (reuse == MAT_REUSE_MATRIX) { 778c4ad791aSXuan Zhou ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr); 779c4ad791aSXuan Zhou } else { 780c4ad791aSXuan Zhou *B = Bmpi; 781c4ad791aSXuan Zhou } 782af295397SXuan Zhou PetscFunctionReturn(0); 783af295397SXuan Zhou } 784af295397SXuan Zhou 785af295397SXuan Zhou #undef __FUNCT__ 786db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental" 787db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A) 788db31f6deSJed Brown { 789db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 790db31f6deSJed Brown PetscErrorCode ierr; 7915e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 7925e9f5b67SHong Zhang PetscBool flg; 7935e9f5b67SHong Zhang MPI_Comm icomm; 794db31f6deSJed Brown 795db31f6deSJed Brown PetscFunctionBegin; 796c1ee1e62SHong Zhang a->interface->Detach(); 797aae2c449SHong Zhang delete a->interface; 798aae2c449SHong Zhang delete a->esubmat; 799db31f6deSJed Brown delete a->emat; 8005e9f5b67SHong Zhang 801180a43e4SHong Zhang elem::mpi::Comm cxxcomm(((PetscObject)A)->comm); 8025e9f5b67SHong Zhang ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr); 8035e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 8045e9f5b67SHong Zhang if (--commgrid->grid_refct == 0) { 8055e9f5b67SHong Zhang delete commgrid->grid; 8065e9f5b67SHong Zhang ierr = PetscFree(commgrid);CHKERRQ(ierr); 8075e9f5b67SHong Zhang } 8085e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 809db31f6deSJed Brown ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","",PETSC_NULL);CHKERRQ(ierr); 8101293973dSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_petsc_C","",PETSC_NULL);CHKERRQ(ierr); 81115767789SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatFactorGetSolverPackage_C","",PETSC_NULL);CHKERRQ(ierr); 812db31f6deSJed Brown ierr = PetscFree(A->data);CHKERRQ(ierr); 813db31f6deSJed Brown PetscFunctionReturn(0); 814db31f6deSJed Brown } 815db31f6deSJed Brown 816db31f6deSJed Brown #undef __FUNCT__ 817db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental" 818db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A) 819db31f6deSJed Brown { 820db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 821db31f6deSJed Brown PetscErrorCode ierr; 822db31f6deSJed Brown PetscMPIInt rsize,csize; 823db31f6deSJed Brown 824db31f6deSJed Brown PetscFunctionBegin; 825db31f6deSJed Brown ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 826db31f6deSJed Brown ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 827db31f6deSJed Brown 828db31f6deSJed Brown a->emat->ResizeTo(A->rmap->N,A->cmap->N);CHKERRQ(ierr); 829db31f6deSJed Brown elem::Zero(*a->emat); 830db31f6deSJed Brown 831db31f6deSJed Brown ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr); 832db31f6deSJed Brown ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr); 833db31f6deSJed Brown if (csize != rsize) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes"); 834db31f6deSJed Brown a->commsize = rsize; 835db31f6deSJed Brown a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize; 836db31f6deSJed Brown a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize; 837db31f6deSJed Brown a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize); 838db31f6deSJed Brown a->m[1] = A->cmap->N / csize + (a->mr[1] != csize); 839db31f6deSJed Brown PetscFunctionReturn(0); 840db31f6deSJed Brown } 841db31f6deSJed Brown 842aae2c449SHong Zhang #undef __FUNCT__ 843aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental" 844aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type) 845aae2c449SHong Zhang { 846aae2c449SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 847aae2c449SHong Zhang 848aae2c449SHong Zhang PetscFunctionBegin; 849aae2c449SHong Zhang a->interface->Detach(); 850aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 851aae2c449SHong Zhang PetscFunctionReturn(0); 852aae2c449SHong Zhang } 853aae2c449SHong Zhang 854aae2c449SHong Zhang #undef __FUNCT__ 855aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental" 856aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type) 857aae2c449SHong Zhang { 858aae2c449SHong Zhang PetscFunctionBegin; 859aae2c449SHong Zhang /* Currently does nothing */ 860aae2c449SHong Zhang PetscFunctionReturn(0); 861aae2c449SHong Zhang } 862aae2c449SHong Zhang 86340d92e34SHong Zhang /* -------------------------------------------------------------------*/ 86440d92e34SHong Zhang static struct _MatOps MatOps_Values = { 86540d92e34SHong Zhang MatSetValues_Elemental, 86640d92e34SHong Zhang 0, 86740d92e34SHong Zhang 0, 86840d92e34SHong Zhang MatMult_Elemental, 86940d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental, 8709426833fSXuan Zhou MatMultTranspose_Elemental, 871e883f9d5SXuan Zhou MatMultTransposeAdd_Elemental, 87240d92e34SHong Zhang MatSolve_Elemental, 873df311e6cSXuan Zhou MatSolveAdd_Elemental, 87440d92e34SHong Zhang 0, //MatSolveTranspose_Elemental, 87540d92e34SHong Zhang /*10*/ 0, //MatSolveTransposeAdd_Elemental, 87640d92e34SHong Zhang MatLUFactor_Elemental, 87740d92e34SHong Zhang MatCholeskyFactor_Elemental, 87840d92e34SHong Zhang 0, 87940d92e34SHong Zhang MatTranspose_Elemental, 88040d92e34SHong Zhang /*15*/ MatGetInfo_Elemental, 88140d92e34SHong Zhang 0, 88240d92e34SHong Zhang 0, 88340d92e34SHong Zhang 0, 88440d92e34SHong Zhang MatNorm_Elemental, 88540d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental, 88640d92e34SHong Zhang MatAssemblyEnd_Elemental, 88740d92e34SHong Zhang 0, //MatSetOption_Elemental, 88840d92e34SHong Zhang MatZeroEntries_Elemental, 88940d92e34SHong Zhang /*24*/ 0, 89040d92e34SHong Zhang MatLUFactorSymbolic_Elemental, 89140d92e34SHong Zhang MatLUFactorNumeric_Elemental, 89240d92e34SHong Zhang MatCholeskyFactorSymbolic_Elemental, 89340d92e34SHong Zhang MatCholeskyFactorNumeric_Elemental, 89440d92e34SHong Zhang /*29*/ MatSetUp_Elemental, 89540d92e34SHong Zhang 0, 89640d92e34SHong Zhang 0, 89740d92e34SHong Zhang 0, 89840d92e34SHong Zhang 0, 899df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental, 90040d92e34SHong Zhang 0, 90140d92e34SHong Zhang 0, 90240d92e34SHong Zhang 0, 90340d92e34SHong Zhang 0, 90440d92e34SHong Zhang /*39*/ MatAXPY_Elemental, 90540d92e34SHong Zhang 0, 90640d92e34SHong Zhang 0, 90740d92e34SHong Zhang 0, 90840d92e34SHong Zhang MatCopy_Elemental, 90940d92e34SHong Zhang /*44*/ 0, 91040d92e34SHong Zhang MatScale_Elemental, 91140d92e34SHong Zhang 0, 91240d92e34SHong Zhang 0, 91340d92e34SHong Zhang 0, 91440d92e34SHong Zhang /*49*/ 0, 91540d92e34SHong Zhang 0, 91640d92e34SHong Zhang 0, 91740d92e34SHong Zhang 0, 91840d92e34SHong Zhang 0, 91940d92e34SHong Zhang /*54*/ 0, 92040d92e34SHong Zhang 0, 92140d92e34SHong Zhang 0, 92240d92e34SHong Zhang 0, 92340d92e34SHong Zhang 0, 92440d92e34SHong Zhang /*59*/ 0, 92540d92e34SHong Zhang MatDestroy_Elemental, 92640d92e34SHong Zhang MatView_Elemental, 92740d92e34SHong Zhang 0, 92840d92e34SHong Zhang 0, 92940d92e34SHong Zhang /*64*/ 0, 93040d92e34SHong Zhang 0, 93140d92e34SHong Zhang 0, 93240d92e34SHong Zhang 0, 93340d92e34SHong Zhang 0, 93440d92e34SHong Zhang /*69*/ 0, 93540d92e34SHong Zhang 0, 9362ef0cf24SXuan Zhou MatConvert_Elemental_Dense, 93740d92e34SHong Zhang 0, 93840d92e34SHong Zhang 0, 93940d92e34SHong Zhang /*74*/ 0, 94040d92e34SHong Zhang 0, 94140d92e34SHong Zhang 0, 94240d92e34SHong Zhang 0, 94340d92e34SHong Zhang 0, 94440d92e34SHong Zhang /*79*/ 0, 94540d92e34SHong Zhang 0, 94640d92e34SHong Zhang 0, 94740d92e34SHong Zhang 0, 94840d92e34SHong Zhang 0, 94940d92e34SHong Zhang /*84*/ 0, 95040d92e34SHong Zhang 0, 95140d92e34SHong Zhang 0, 95240d92e34SHong Zhang 0, 95340d92e34SHong Zhang 0, 95440d92e34SHong Zhang /*89*/ MatMatMult_Elemental, 95540d92e34SHong Zhang MatMatMultSymbolic_Elemental, 95640d92e34SHong Zhang MatMatMultNumeric_Elemental, 95740d92e34SHong Zhang 0, 95840d92e34SHong Zhang 0, 95940d92e34SHong Zhang /*94*/ 0, 960df311e6cSXuan Zhou MatMatTransposeMult_Elemental, 961df311e6cSXuan Zhou MatMatTransposeMultSymbolic_Elemental, 962df311e6cSXuan Zhou MatMatTransposeMultNumeric_Elemental, 96340d92e34SHong Zhang 0, 96440d92e34SHong Zhang /*99*/ 0, 96540d92e34SHong Zhang 0, 96640d92e34SHong Zhang 0, 96740d92e34SHong Zhang 0, 96840d92e34SHong Zhang 0, 96940d92e34SHong Zhang /*104*/0, 97040d92e34SHong Zhang 0, 97140d92e34SHong Zhang 0, 97240d92e34SHong Zhang 0, 97340d92e34SHong Zhang 0, 97440d92e34SHong Zhang /*109*/MatMatSolve_Elemental, 97540d92e34SHong Zhang 0, 97640d92e34SHong Zhang 0, 97740d92e34SHong Zhang 0, 97840d92e34SHong Zhang 0, 97940d92e34SHong Zhang /*114*/0, 98040d92e34SHong Zhang 0, 98140d92e34SHong Zhang 0, 98240d92e34SHong Zhang 0, 98340d92e34SHong Zhang 0, 98440d92e34SHong Zhang /*119*/0, 98540d92e34SHong Zhang 0, 98640d92e34SHong Zhang 0, 98740d92e34SHong Zhang 0, 98840d92e34SHong Zhang 0, 98940d92e34SHong Zhang /*124*/0, 99040d92e34SHong Zhang 0, 99140d92e34SHong Zhang 0, 99240d92e34SHong Zhang 0, 99340d92e34SHong Zhang 0, 99440d92e34SHong Zhang /*129*/0, 99540d92e34SHong Zhang 0, 99640d92e34SHong Zhang 0, 99740d92e34SHong Zhang 0, 99840d92e34SHong Zhang 0, 99940d92e34SHong Zhang /*134*/0, 100040d92e34SHong Zhang 0, 100140d92e34SHong Zhang 0, 100240d92e34SHong Zhang 0, 100340d92e34SHong Zhang 0 100440d92e34SHong Zhang }; 100540d92e34SHong Zhang 1006ed36708cSHong Zhang /*MC 1007ed36708cSHong Zhang MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package 1008ed36708cSHong Zhang 1009ed36708cSHong Zhang Options Database Keys: 1010ed36708cSHong Zhang . -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions() 1011ed36708cSHong Zhang 1012ed36708cSHong Zhang Level: beginner 1013ed36708cSHong Zhang 1014ed36708cSHong Zhang .seealso: MATDENSE,MatCreateElemental() 1015ed36708cSHong Zhang M*/ 1016ed36708cSHong Zhang EXTERN_C_BEGIN 1017db31f6deSJed Brown #undef __FUNCT__ 1018db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental" 1019db31f6deSJed Brown PETSC_EXTERN_C PetscErrorCode MatCreate_Elemental(Mat A) 1020db31f6deSJed Brown { 1021db31f6deSJed Brown Mat_Elemental *a; 1022db31f6deSJed Brown PetscErrorCode ierr; 1023ed667823SXuan Zhou PetscBool flg,flg1,flg2; 10245e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 10255e9f5b67SHong Zhang MPI_Comm icomm; 1026ed667823SXuan Zhou PetscInt optv1,optv2; 1027db31f6deSJed Brown 1028db31f6deSJed Brown PetscFunctionBegin; 1029db31f6deSJed Brown ierr = PetscElementalInitializePackage(PETSC_NULL);CHKERRQ(ierr); 103040d92e34SHong Zhang ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 103140d92e34SHong Zhang A->insertmode = NOT_SET_VALUES; 1032db31f6deSJed Brown 1033db31f6deSJed Brown ierr = PetscNewLog(A,Mat_Elemental,&a);CHKERRQ(ierr); 1034db31f6deSJed Brown A->data = (void*)a; 1035db31f6deSJed Brown 1036db31f6deSJed Brown /* Set up the elemental matrix */ 1037db31f6deSJed Brown elem::mpi::Comm cxxcomm(((PetscObject)A)->comm); 1038ed667823SXuan Zhou ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 10395e9f5b67SHong Zhang 10405e9f5b67SHong Zhang /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */ 10415e9f5b67SHong Zhang if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) { 1042180a43e4SHong Zhang ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); 10435e9f5b67SHong Zhang } 10445e9f5b67SHong Zhang ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr); 10455e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 10465e9f5b67SHong Zhang if (!flg) { 10475e9f5b67SHong Zhang ierr = PetscNewLog(A,Mat_Elemental_Grid,&commgrid);CHKERRQ(ierr); 10482ef0cf24SXuan Zhou ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr); 10492ef0cf24SXuan Zhou ierr = PetscOptionsInt("-mat_elemental_grid_width","Grid Width","None",1,&optv2,&flg2);CHKERRQ(ierr); 1050ed667823SXuan Zhou if (flg1 || flg2) { 1051ed667823SXuan Zhou if (optv1*optv2 != elem::mpi::CommSize(cxxcomm)) { 10522ef0cf24SXuan Zhou SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Grid Height times Grid Width must equal CommSize"); 1053ed667823SXuan Zhou } 1054ed667823SXuan Zhou commgrid->grid = new elem::Grid(cxxcomm,optv1,optv2); 10552ef0cf24SXuan Zhou } else { 10562ef0cf24SXuan Zhou commgrid->grid = new elem::Grid(cxxcomm); 1057ed667823SXuan Zhou } 10585e9f5b67SHong Zhang commgrid->grid_refct = 1; 10595e9f5b67SHong Zhang ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr); 10605e9f5b67SHong Zhang } else { 10615e9f5b67SHong Zhang commgrid->grid_refct++; 10625e9f5b67SHong Zhang } 10635e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 10645e9f5b67SHong Zhang a->grid = commgrid->grid; 1065df311e6cSXuan Zhou a->emat = new elem::DistMatrix<PetscElemScalar>(*a->grid); 1066df311e6cSXuan Zhou a->esubmat = new elem::Matrix<PetscElemScalar>(1,1); 1067df311e6cSXuan Zhou a->interface = new elem::AxpyInterface<PetscElemScalar>; 10687c920d81SXuan Zhou a->pivot = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>; 1069db31f6deSJed Brown 1070db31f6deSJed Brown /* build cache for off array entries formed */ 1071aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 1072bafd5131SHong Zhang 1073db31f6deSJed Brown ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","MatGetOwnershipIS_Elemental",MatGetOwnershipIS_Elemental);CHKERRQ(ierr); 107415767789SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_elemental_C","MatGetFactor_elemental_elemental",MatGetFactor_elemental_elemental);CHKERRQ(ierr); 1075db31f6deSJed Brown 1076db31f6deSJed Brown ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr); 1077ed667823SXuan Zhou PetscOptionsEnd(); 1078db31f6deSJed Brown PetscFunctionReturn(0); 1079db31f6deSJed Brown } 1080ed36708cSHong Zhang EXTERN_C_END 1081