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 Level: developer 17db31f6deSJed Brown 18db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalFinalizePackage() 19db31f6deSJed Brown @*/ 20607a6623SBarry Smith PetscErrorCode PetscElementalInitializePackage(void) 21db31f6deSJed Brown { 22db31f6deSJed Brown PetscErrorCode ierr; 23db31f6deSJed Brown 24db31f6deSJed Brown PetscFunctionBegin; 25db31f6deSJed Brown if (elem::Initialized()) PetscFunctionReturn(0); 26db31f6deSJed 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 */ 27db31f6deSJed Brown int zero = 0; 28db31f6deSJed Brown char **nothing = 0; 29db31f6deSJed Brown elem::Initialize(zero,nothing); 30db31f6deSJed Brown } 31db31f6deSJed Brown ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr); 32db31f6deSJed Brown PetscFunctionReturn(0); 33db31f6deSJed Brown } 34db31f6deSJed Brown 35db31f6deSJed Brown #undef __FUNCT__ 36db31f6deSJed Brown #define __FUNCT__ "PetscElementalFinalizePackage" 37db31f6deSJed Brown /*@C 38db31f6deSJed Brown PetscElementalFinalizePackage - Finalize Elemental package 39db31f6deSJed Brown 40db31f6deSJed Brown Logically Collective 41db31f6deSJed Brown 42db31f6deSJed Brown Level: developer 43db31f6deSJed Brown 44db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalInitializePackage() 45db31f6deSJed Brown @*/ 46db31f6deSJed Brown PetscErrorCode PetscElementalFinalizePackage(void) 47db31f6deSJed Brown { 48db31f6deSJed Brown 49db31f6deSJed Brown PetscFunctionBegin; 50db31f6deSJed Brown elem::Finalize(); 51db31f6deSJed Brown PetscFunctionReturn(0); 52db31f6deSJed Brown } 53db31f6deSJed Brown 54db31f6deSJed Brown #undef __FUNCT__ 55db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental" 56db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer) 57db31f6deSJed Brown { 58db31f6deSJed Brown PetscErrorCode ierr; 59db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 60db31f6deSJed Brown PetscBool iascii; 61db31f6deSJed Brown 62db31f6deSJed Brown PetscFunctionBegin; 63db31f6deSJed Brown ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 64db31f6deSJed Brown if (iascii) { 65db31f6deSJed Brown PetscViewerFormat format; 66db31f6deSJed Brown ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 67db31f6deSJed Brown if (format == PETSC_VIEWER_ASCII_INFO) { 6879673f7bSHong Zhang /* call elemental viewing function */ 692d8adcc7SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr); 70ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr); 71ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr); 724fe7bbcaSHong Zhang if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 7379673f7bSHong Zhang /* call elemental viewing function */ 74ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"test matview_elemental 2\n");CHKERRQ(ierr); 754fe7bbcaSHong Zhang } 7679673f7bSHong Zhang 77db31f6deSJed Brown } else if (format == PETSC_VIEWER_DEFAULT) { 78db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 79dae58748SBarry Smith ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer);CHKERRQ(ierr); 807a583510SJack Poulson elem::Print( *a->emat, "Elemental matrix (cyclic ordering)" ); 81db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 82834d3fecSHong Zhang if (A->factortype == MAT_FACTOR_NONE){ 8361119200SXuan Zhou Mat Adense; 84ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 8561119200SXuan Zhou ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr); 8661119200SXuan Zhou ierr = MatView(Adense,viewer);CHKERRQ(ierr); 8761119200SXuan Zhou ierr = MatDestroy(&Adense);CHKERRQ(ierr); 88834d3fecSHong Zhang } 89ce94432eSBarry Smith } else SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Format"); 90d2daa67eSHong Zhang } else { 915cb544a0SHong Zhang /* convert to dense format and call MatView() */ 9261119200SXuan Zhou Mat Adense; 93ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 9461119200SXuan Zhou ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr); 9561119200SXuan Zhou ierr = MatView(Adense,viewer);CHKERRQ(ierr); 9661119200SXuan Zhou ierr = MatDestroy(&Adense);CHKERRQ(ierr); 97d2daa67eSHong Zhang } 98db31f6deSJed Brown PetscFunctionReturn(0); 99db31f6deSJed Brown } 100db31f6deSJed Brown 101db31f6deSJed Brown #undef __FUNCT__ 102180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental" 10315767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info) 104180a43e4SHong Zhang { 10515767789SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 10615767789SHong Zhang PetscMPIInt rank; 10715767789SHong Zhang 108180a43e4SHong Zhang PetscFunctionBegin; 109ce94432eSBarry Smith MPI_Comm_rank(PetscObjectComm((PetscObject)A),&rank); 11015767789SHong Zhang 11115767789SHong Zhang /* if (!rank) printf(" .........MatGetInfo_Elemental ...\n"); */ 1125cb544a0SHong Zhang info->block_size = 1.0; 11315767789SHong Zhang 11415767789SHong Zhang if (flag == MAT_LOCAL) { 11515767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 11615767789SHong Zhang info->nz_used = info->nz_allocated; 11715767789SHong Zhang } else if (flag == MAT_GLOBAL_MAX) { 118ce94432eSBarry Smith //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr); 11915767789SHong Zhang /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */ 12015767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet"); 12115767789SHong Zhang } else if (flag == MAT_GLOBAL_SUM) { 12215767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet"); 12315767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 12415767789SHong Zhang info->nz_used = info->nz_allocated; /* assume Elemental does accurate allocation */ 125ce94432eSBarry Smith //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,PetscObjectComm((PetscObject)A));CHKERRQ(ierr); 12615767789SHong Zhang //PetscPrintf(PETSC_COMM_SELF," ... [%d] locally allocated %g\n",rank,info->nz_allocated); 12715767789SHong Zhang } 12815767789SHong Zhang 12915767789SHong Zhang info->nz_unneeded = 0.0; 13015767789SHong Zhang info->assemblies = (double)A->num_ass; 13115767789SHong Zhang info->mallocs = 0; 13215767789SHong Zhang info->memory = ((PetscObject)A)->mem; 13315767789SHong Zhang info->fill_ratio_given = 0; /* determined by Elemental */ 13415767789SHong Zhang info->fill_ratio_needed = 0; 13515767789SHong Zhang info->factor_mallocs = 0; 136180a43e4SHong Zhang PetscFunctionReturn(0); 137180a43e4SHong Zhang } 138180a43e4SHong Zhang 139180a43e4SHong Zhang #undef __FUNCT__ 140db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental" 141e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode) 142db31f6deSJed Brown { 143db31f6deSJed Brown PetscErrorCode ierr; 144db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 145db31f6deSJed Brown PetscMPIInt rank; 146db31f6deSJed Brown PetscInt i,j,rrank,ridx,crank,cidx; 147db31f6deSJed Brown 148db31f6deSJed Brown PetscFunctionBegin; 149ce94432eSBarry Smith ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)A),&rank);CHKERRQ(ierr); 150db31f6deSJed Brown 151db31f6deSJed Brown const elem::Grid &grid = a->emat->Grid(); 152db31f6deSJed Brown for (i=0; i<nr; i++) { 153db31f6deSJed Brown PetscInt erow,ecol,elrow,elcol; 154db31f6deSJed Brown if (rows[i] < 0) continue; 155db31f6deSJed Brown P2RO(A,0,rows[i],&rrank,&ridx); 156db31f6deSJed Brown RO2E(A,0,rrank,ridx,&erow); 157ce94432eSBarry Smith if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation"); 158db31f6deSJed Brown for (j=0; j<nc; j++) { 159db31f6deSJed Brown if (cols[j] < 0) continue; 160db31f6deSJed Brown P2RO(A,1,cols[j],&crank,&cidx); 161db31f6deSJed Brown RO2E(A,1,crank,cidx,&ecol); 162ce94432eSBarry Smith if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation"); 163aae2c449SHong Zhang if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */ 164aae2c449SHong Zhang if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 165aae2c449SHong 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); */ 166e6dea9dbSXuan Zhou a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]); 167aae2c449SHong Zhang a->interface->Axpy(1.0,*(a->esubmat),erow,ecol); 168aae2c449SHong Zhang continue; 169ed36708cSHong Zhang } 170db31f6deSJed Brown elrow = erow / grid.MCSize(); 171db31f6deSJed Brown elcol = ecol / grid.MRSize(); 172db31f6deSJed Brown switch (imode) { 173e6dea9dbSXuan Zhou case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 174e6dea9dbSXuan Zhou case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 175ce94432eSBarry Smith default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 176db31f6deSJed Brown } 177db31f6deSJed Brown } 178db31f6deSJed Brown } 179db31f6deSJed Brown PetscFunctionReturn(0); 180db31f6deSJed Brown } 181db31f6deSJed Brown 182db31f6deSJed Brown #undef __FUNCT__ 183db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental" 184db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y) 185db31f6deSJed Brown { 186db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 187db31f6deSJed Brown PetscErrorCode ierr; 188e6dea9dbSXuan Zhou const PetscElemScalar *x; 189e6dea9dbSXuan Zhou PetscElemScalar *y; 190df311e6cSXuan Zhou PetscElemScalar one = 1,zero = 0; 191db31f6deSJed Brown 192db31f6deSJed Brown PetscFunctionBegin; 193e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 194e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 195db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 196efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,0,x,A->cmap->n,*a->grid); 197efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,0,y,A->rmap->n,*a->grid); 198db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye); 199db31f6deSJed Brown } 200e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 201e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 202db31f6deSJed Brown PetscFunctionReturn(0); 203db31f6deSJed Brown } 204db31f6deSJed Brown 205db31f6deSJed Brown #undef __FUNCT__ 2069426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental" 2079426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y) 2089426833fSXuan Zhou { 2099426833fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 2109426833fSXuan Zhou PetscErrorCode ierr; 211df311e6cSXuan Zhou const PetscElemScalar *x; 212df311e6cSXuan Zhou PetscElemScalar *y; 213e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 2149426833fSXuan Zhou 2159426833fSXuan Zhou PetscFunctionBegin; 216e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 217e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2189426833fSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 219efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,0,x,A->rmap->n,*a->grid); 220efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,0,y,A->cmap->n,*a->grid); 2219426833fSXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye); 2229426833fSXuan Zhou } 223e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 224e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2259426833fSXuan Zhou PetscFunctionReturn(0); 2269426833fSXuan Zhou } 2279426833fSXuan Zhou 2289426833fSXuan Zhou #undef __FUNCT__ 229db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental" 230db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 231db31f6deSJed Brown { 232db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 233db31f6deSJed Brown PetscErrorCode ierr; 234df311e6cSXuan Zhou const PetscElemScalar *x; 235df311e6cSXuan Zhou PetscElemScalar *z; 236e6dea9dbSXuan Zhou PetscElemScalar one = 1; 237db31f6deSJed Brown 238db31f6deSJed Brown PetscFunctionBegin; 239db31f6deSJed Brown if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 240e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 241e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 242db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 243efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,0,x,A->cmap->n,*a->grid); 244efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,0,z,A->rmap->n,*a->grid); 245db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze); 246db31f6deSJed Brown } 247e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 248e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 249db31f6deSJed Brown PetscFunctionReturn(0); 250db31f6deSJed Brown } 251db31f6deSJed Brown 252db31f6deSJed Brown #undef __FUNCT__ 253e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental" 254e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 255e883f9d5SXuan Zhou { 256e883f9d5SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 257e883f9d5SXuan Zhou PetscErrorCode ierr; 258df311e6cSXuan Zhou const PetscElemScalar *x; 259df311e6cSXuan Zhou PetscElemScalar *z; 260e6dea9dbSXuan Zhou PetscElemScalar one = 1; 261e883f9d5SXuan Zhou 262e883f9d5SXuan Zhou PetscFunctionBegin; 263e883f9d5SXuan Zhou if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 264e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 265e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 266e883f9d5SXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 267efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,0,x,A->rmap->n,*a->grid); 268efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,0,z,A->cmap->n,*a->grid); 269e883f9d5SXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze); 270e883f9d5SXuan Zhou } 271e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 272e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 273e883f9d5SXuan Zhou PetscFunctionReturn(0); 274e883f9d5SXuan Zhou } 275e883f9d5SXuan Zhou 276e883f9d5SXuan Zhou #undef __FUNCT__ 2779a9e8502SHong Zhang #define __FUNCT__ "MatMatMultNumeric_Elemental" 2789a9e8502SHong Zhang static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C) 279c1d1b975SXuan Zhou { 280c1d1b975SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 281c1d1b975SXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 2829a9e8502SHong Zhang Mat_Elemental *c = (Mat_Elemental*)C->data; 283e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 284c1d1b975SXuan Zhou 285c1d1b975SXuan Zhou PetscFunctionBegin; 286aae2c449SHong Zhang { /* Scoping so that constructor is called before pointer is returned */ 287c1d1b975SXuan Zhou elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat); 288aae2c449SHong Zhang } 2899a9e8502SHong Zhang C->assembled = PETSC_TRUE; 2909a9e8502SHong Zhang PetscFunctionReturn(0); 2919a9e8502SHong Zhang } 2929a9e8502SHong Zhang 2939a9e8502SHong Zhang #undef __FUNCT__ 2949a9e8502SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_Elemental" 2959a9e8502SHong Zhang static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 2969a9e8502SHong Zhang { 2979a9e8502SHong Zhang PetscErrorCode ierr; 2989a9e8502SHong Zhang Mat Ce; 299ce94432eSBarry Smith MPI_Comm comm; 3009a9e8502SHong Zhang 3019a9e8502SHong Zhang PetscFunctionBegin; 302ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 3039a9e8502SHong Zhang ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 3049a9e8502SHong Zhang ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 3059a9e8502SHong Zhang ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 3069a9e8502SHong Zhang ierr = MatSetUp(Ce);CHKERRQ(ierr); 3079a9e8502SHong Zhang *C = Ce; 3089a9e8502SHong Zhang PetscFunctionReturn(0); 3099a9e8502SHong Zhang } 3109a9e8502SHong Zhang 3119a9e8502SHong Zhang #undef __FUNCT__ 3129a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental" 3139a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 3149a9e8502SHong Zhang { 3159a9e8502SHong Zhang PetscErrorCode ierr; 3169a9e8502SHong Zhang 3179a9e8502SHong Zhang PetscFunctionBegin; 3189a9e8502SHong Zhang if (scall == MAT_INITIAL_MATRIX){ 3193ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3209a9e8502SHong Zhang ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 3213ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3229a9e8502SHong Zhang } 3233ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3249a9e8502SHong Zhang ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 3253ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 326c1d1b975SXuan Zhou PetscFunctionReturn(0); 327c1d1b975SXuan Zhou } 328c1d1b975SXuan Zhou 329c1d1b975SXuan Zhou #undef __FUNCT__ 330df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental" 331df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C) 332df311e6cSXuan Zhou { 333df311e6cSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 334df311e6cSXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 335df311e6cSXuan Zhou Mat_Elemental *c = (Mat_Elemental*)C->data; 336e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 337df311e6cSXuan Zhou 338df311e6cSXuan Zhou PetscFunctionBegin; 339df311e6cSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 340df311e6cSXuan Zhou elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat); 341df311e6cSXuan Zhou } 342df311e6cSXuan Zhou C->assembled = PETSC_TRUE; 343df311e6cSXuan Zhou PetscFunctionReturn(0); 344df311e6cSXuan Zhou } 345df311e6cSXuan Zhou 346df311e6cSXuan Zhou #undef __FUNCT__ 347df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental" 348df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 349df311e6cSXuan Zhou { 350df311e6cSXuan Zhou PetscErrorCode ierr; 351df311e6cSXuan Zhou Mat Ce; 352ce94432eSBarry Smith MPI_Comm comm; 353df311e6cSXuan Zhou 354df311e6cSXuan Zhou PetscFunctionBegin; 355ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 356df311e6cSXuan Zhou ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 357df311e6cSXuan Zhou ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 358df311e6cSXuan Zhou ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 359df311e6cSXuan Zhou ierr = MatSetUp(Ce);CHKERRQ(ierr); 360df311e6cSXuan Zhou *C = Ce; 361df311e6cSXuan Zhou PetscFunctionReturn(0); 362df311e6cSXuan Zhou } 363df311e6cSXuan Zhou 364df311e6cSXuan Zhou #undef __FUNCT__ 365df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental" 366df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 367df311e6cSXuan Zhou { 368df311e6cSXuan Zhou PetscErrorCode ierr; 369df311e6cSXuan Zhou 370df311e6cSXuan Zhou PetscFunctionBegin; 371df311e6cSXuan Zhou if (scall == MAT_INITIAL_MATRIX){ 372df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 373df311e6cSXuan Zhou ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 374df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 375df311e6cSXuan Zhou } 376df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 377df311e6cSXuan Zhou ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 378df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 379df311e6cSXuan Zhou PetscFunctionReturn(0); 380df311e6cSXuan Zhou } 381df311e6cSXuan Zhou 382df311e6cSXuan Zhou #undef __FUNCT__ 38361119200SXuan Zhou #define __FUNCT__ "MatGetDiagonal_Elemental" 384a9d89745SXuan Zhou static PetscErrorCode MatGetDiagonal_Elemental(Mat A,Vec D) 38561119200SXuan Zhou { 386a9d89745SXuan Zhou PetscInt i,nrows,ncols,nD,rrank,ridx,crank,cidx; 387a9d89745SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 38861119200SXuan Zhou PetscErrorCode ierr; 389a9d89745SXuan Zhou PetscElemScalar v; 390ce94432eSBarry Smith MPI_Comm comm; 39161119200SXuan Zhou 39261119200SXuan Zhou PetscFunctionBegin; 393ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 394a9d89745SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 395a9d89745SXuan Zhou nD = nrows>ncols ? ncols : nrows; 396a9d89745SXuan Zhou for (i=0; i<nD; i++) { 397a9d89745SXuan Zhou PetscInt erow,ecol; 398a9d89745SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 399a9d89745SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 400a9d89745SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 401a9d89745SXuan Zhou P2RO(A,1,i,&crank,&cidx); 402a9d89745SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 403a9d89745SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 404a9d89745SXuan Zhou v = a->emat->Get(erow,ecol); 4057c8b904fSSatish Balay ierr = VecSetValues(D,1,&i,(PetscScalar*)&v,INSERT_VALUES);CHKERRQ(ierr); 406ade3cc5eSXuan Zhou } 407a9d89745SXuan Zhou ierr = VecAssemblyBegin(D);CHKERRQ(ierr); 408a9d89745SXuan Zhou ierr = VecAssemblyEnd(D);CHKERRQ(ierr); 40961119200SXuan Zhou PetscFunctionReturn(0); 41061119200SXuan Zhou } 41161119200SXuan Zhou 41261119200SXuan Zhou #undef __FUNCT__ 413ade3cc5eSXuan Zhou #define __FUNCT__ "MatDiagonalScale_Elemental" 414ade3cc5eSXuan Zhou static PetscErrorCode MatDiagonalScale_Elemental(Mat X,Vec L,Vec R) 415ade3cc5eSXuan Zhou { 416ade3cc5eSXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 417ade3cc5eSXuan Zhou const PetscElemScalar *d; 418ade3cc5eSXuan Zhou PetscErrorCode ierr; 419ade3cc5eSXuan Zhou 420ade3cc5eSXuan Zhou PetscFunctionBegin; 4219065cd98SJed Brown if (R) { 422ade3cc5eSXuan Zhou ierr = VecGetArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr); 423efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> de(X->cmap->N,1,0,0,d,X->cmap->n,*x->grid); 424ade3cc5eSXuan Zhou elem::DiagonalScale(elem::RIGHT,elem::NORMAL,de,*x->emat); 425ade3cc5eSXuan Zhou ierr = VecRestoreArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr); 4269065cd98SJed Brown } 4279065cd98SJed Brown if (L) { 428ade3cc5eSXuan Zhou ierr = VecGetArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr); 429efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> de(X->rmap->N,1,0,0,d,X->rmap->n,*x->grid); 430ade3cc5eSXuan Zhou elem::DiagonalScale(elem::LEFT,elem::NORMAL,de,*x->emat); 431ade3cc5eSXuan Zhou ierr = VecRestoreArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr); 432ade3cc5eSXuan Zhou } 433ade3cc5eSXuan Zhou PetscFunctionReturn(0); 434ade3cc5eSXuan Zhou } 435ade3cc5eSXuan Zhou 436ade3cc5eSXuan Zhou #undef __FUNCT__ 4374ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental" 438e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a) 43965b78793SXuan Zhou { 44065b78793SXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 44165b78793SXuan Zhou 44265b78793SXuan Zhou PetscFunctionBegin; 443e6dea9dbSXuan Zhou elem::Scal((PetscElemScalar)a,*x->emat); 44465b78793SXuan Zhou PetscFunctionReturn(0); 44565b78793SXuan Zhou } 44665b78793SXuan Zhou 44765b78793SXuan Zhou #undef __FUNCT__ 4484ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental" 449e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str) 450e09a3074SHong Zhang { 451e09a3074SHong Zhang Mat_Elemental *x = (Mat_Elemental*)X->data; 452e09a3074SHong Zhang Mat_Elemental *y = (Mat_Elemental*)Y->data; 453*68446cf8SJed Brown PetscErrorCode ierr; 454e09a3074SHong Zhang 455e09a3074SHong Zhang PetscFunctionBegin; 456e6dea9dbSXuan Zhou elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat); 45721f6c9c4SJed Brown ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 458e09a3074SHong Zhang PetscFunctionReturn(0); 459e09a3074SHong Zhang } 460e09a3074SHong Zhang 461ae844d54SHong Zhang #undef __FUNCT__ 462d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental" 463d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str) 464d6223691SXuan Zhou { 465d6223691SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 466d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 467d6223691SXuan Zhou 468d6223691SXuan Zhou PetscFunctionBegin; 469d6223691SXuan Zhou elem::Copy(*a->emat,*b->emat); 470d6223691SXuan Zhou PetscFunctionReturn(0); 471d6223691SXuan Zhou } 472d6223691SXuan Zhou 473d6223691SXuan Zhou #undef __FUNCT__ 474df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental" 475df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B) 476df311e6cSXuan Zhou { 477df311e6cSXuan Zhou Mat Be; 478ce94432eSBarry Smith MPI_Comm comm; 479df311e6cSXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 480df311e6cSXuan Zhou PetscErrorCode ierr; 481df311e6cSXuan Zhou 482df311e6cSXuan Zhou PetscFunctionBegin; 483ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 484df311e6cSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 485df311e6cSXuan Zhou ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 486df311e6cSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 487df311e6cSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 488df311e6cSXuan Zhou *B = Be; 489df311e6cSXuan Zhou if (op == MAT_COPY_VALUES) { 490df311e6cSXuan Zhou Mat_Elemental *b=(Mat_Elemental*)Be->data; 491df311e6cSXuan Zhou elem::Copy(*a->emat,*b->emat); 492df311e6cSXuan Zhou } 493df311e6cSXuan Zhou Be->assembled = PETSC_TRUE; 494df311e6cSXuan Zhou PetscFunctionReturn(0); 495df311e6cSXuan Zhou } 496df311e6cSXuan Zhou 497df311e6cSXuan Zhou #undef __FUNCT__ 498d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental" 499d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 500d6223691SXuan Zhou { 5015262d616SXuan Zhou Mat Be; 5025262d616SXuan Zhou PetscErrorCode ierr; 503ce94432eSBarry Smith MPI_Comm comm; 5044fe7bbcaSHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 505d6223691SXuan Zhou 506d6223691SXuan Zhou PetscFunctionBegin; 507ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 5085cb544a0SHong Zhang /* Only out-of-place supported */ 5095262d616SXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 5105262d616SXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 5115262d616SXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 5125262d616SXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 5135262d616SXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 5145262d616SXuan Zhou *B = Be; 5155262d616SXuan Zhou } 5164fe7bbcaSHong Zhang b = (Mat_Elemental*)Be->data; 5175262d616SXuan Zhou elem::Transpose(*a->emat,*b->emat); 5185262d616SXuan Zhou Be->assembled = PETSC_TRUE; 519d6223691SXuan Zhou PetscFunctionReturn(0); 520d6223691SXuan Zhou } 521d6223691SXuan Zhou 522d6223691SXuan Zhou #undef __FUNCT__ 523dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental" 524dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A) 525dfcb0403SXuan Zhou { 526dfcb0403SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 527dfcb0403SXuan Zhou 528dfcb0403SXuan Zhou PetscFunctionBegin; 529dfcb0403SXuan Zhou elem::Conjugate(*a->emat); 530dfcb0403SXuan Zhou PetscFunctionReturn(0); 531dfcb0403SXuan Zhou } 532dfcb0403SXuan Zhou 533dfcb0403SXuan Zhou #undef __FUNCT__ 5344a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental" 5354a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 5364a29722dSXuan Zhou { 5374a29722dSXuan Zhou Mat Be; 5384a29722dSXuan Zhou PetscErrorCode ierr; 539ce94432eSBarry Smith MPI_Comm comm; 5404a29722dSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 5414a29722dSXuan Zhou 5424a29722dSXuan Zhou PetscFunctionBegin; 543ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 5445cb544a0SHong Zhang /* Only out-of-place supported */ 5454a29722dSXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 5464a29722dSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 5474a29722dSXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 5484a29722dSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 5494a29722dSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 5504a29722dSXuan Zhou *B = Be; 5514a29722dSXuan Zhou } 5524a29722dSXuan Zhou b = (Mat_Elemental*)Be->data; 5534a29722dSXuan Zhou elem::Adjoint(*a->emat,*b->emat); 5544a29722dSXuan Zhou Be->assembled = PETSC_TRUE; 5554a29722dSXuan Zhou PetscFunctionReturn(0); 5564a29722dSXuan Zhou } 5574a29722dSXuan Zhou 5584a29722dSXuan Zhou #undef __FUNCT__ 5591f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental" 5601f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X) 5611f881ff8SXuan Zhou { 5621f881ff8SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 5631f881ff8SXuan Zhou PetscErrorCode ierr; 564df311e6cSXuan Zhou PetscElemScalar *x; 5651f881ff8SXuan Zhou 5661f881ff8SXuan Zhou PetscFunctionBegin; 56745cf121fSXuan Zhou ierr = VecCopy(B,X);CHKERRQ(ierr); 568e6dea9dbSXuan Zhou ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 569efb79153SJack Poulson elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,0,x,A->rmap->n,*a->grid); 570df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe; 571fc54b460SXuan Zhou switch (A->factortype) { 572fc54b460SXuan Zhou case MAT_FACTOR_LU: 5731f881ff8SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 574324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,*a->pivot,xer); 57545cf121fSXuan Zhou elem::Copy(xer,xe); 576fc54b460SXuan Zhou } else { 577324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,xer); 57845cf121fSXuan Zhou elem::Copy(xer,xe); 5791f881ff8SXuan Zhou } 580fc54b460SXuan Zhou break; 581fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 582324645c7SJack Poulson elem::cholesky::SolveAfter(elem::UPPER,elem::NORMAL,*a->emat,xer); 583fc54b460SXuan Zhou elem::Copy(xer,xe); 584fc54b460SXuan Zhou break; 585fc54b460SXuan Zhou default: 5864fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 587fc54b460SXuan Zhou break; 5881f881ff8SXuan Zhou } 589e6dea9dbSXuan Zhou ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 5901f881ff8SXuan Zhou PetscFunctionReturn(0); 5911f881ff8SXuan Zhou } 5921f881ff8SXuan Zhou 5931f881ff8SXuan Zhou #undef __FUNCT__ 594df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental" 595df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X) 596df311e6cSXuan Zhou { 597df311e6cSXuan Zhou PetscErrorCode ierr; 598df311e6cSXuan Zhou 599df311e6cSXuan Zhou PetscFunctionBegin; 600df311e6cSXuan Zhou ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr); 6013d7f40dbSXuan Zhou ierr = VecAXPY(X,1,Y);CHKERRQ(ierr); 602df311e6cSXuan Zhou PetscFunctionReturn(0); 603df311e6cSXuan Zhou } 604df311e6cSXuan Zhou 605df311e6cSXuan Zhou #undef __FUNCT__ 606ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental" 607ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X) 608ae844d54SHong Zhang { 6091f0e42cfSHong Zhang Mat_Elemental *a=(Mat_Elemental*)A->data; 610d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 6111f0e42cfSHong Zhang Mat_Elemental *x=(Mat_Elemental*)X->data; 6121f0e42cfSHong Zhang 613ae844d54SHong Zhang PetscFunctionBegin; 614d6223691SXuan Zhou elem::Copy(*b->emat,*x->emat); 615fc54b460SXuan Zhou switch (A->factortype) { 616fc54b460SXuan Zhou case MAT_FACTOR_LU: 617d6223691SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 618324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,*a->pivot,*x->emat); 619fc54b460SXuan Zhou } else { 620324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,*x->emat); 621d6223691SXuan Zhou } 622fc54b460SXuan Zhou break; 623fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 624324645c7SJack Poulson elem::cholesky::SolveAfter(elem::UPPER,elem::NORMAL,*a->emat,*x->emat); 625fc54b460SXuan Zhou break; 626fc54b460SXuan Zhou default: 6274fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 628fc54b460SXuan Zhou break; 629fc54b460SXuan Zhou } 630ae844d54SHong Zhang PetscFunctionReturn(0); 631ae844d54SHong Zhang } 632ae844d54SHong Zhang 633ae844d54SHong Zhang #undef __FUNCT__ 634ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental" 635ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info) 636ae844d54SHong Zhang { 6377c920d81SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 6387c920d81SXuan Zhou 639ae844d54SHong Zhang PetscFunctionBegin; 640d6223691SXuan Zhou if (info->dtcol){ 6417c920d81SXuan Zhou elem::LU(*a->emat,*a->pivot); 6421293973dSHong Zhang } else { 643d6223691SXuan Zhou elem::LU(*a->emat); 644d6223691SXuan Zhou } 6451293973dSHong Zhang A->factortype = MAT_FACTOR_LU; 646834d3fecSHong Zhang A->assembled = PETSC_TRUE; 647ae844d54SHong Zhang PetscFunctionReturn(0); 648ae844d54SHong Zhang } 649ae844d54SHong Zhang 650d7c3f9d8SHong Zhang #undef __FUNCT__ 651d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental" 652d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 653d7c3f9d8SHong Zhang { 654d7c3f9d8SHong Zhang PetscErrorCode ierr; 655d7c3f9d8SHong Zhang 656d7c3f9d8SHong Zhang PetscFunctionBegin; 657d7c3f9d8SHong Zhang ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 658d7c3f9d8SHong Zhang ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr); 659d7c3f9d8SHong Zhang PetscFunctionReturn(0); 660d7c3f9d8SHong Zhang } 661d7c3f9d8SHong Zhang 662d7c3f9d8SHong Zhang #undef __FUNCT__ 663d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental" 664d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 665d7c3f9d8SHong Zhang { 666d7c3f9d8SHong Zhang PetscFunctionBegin; 667d7c3f9d8SHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 668d7c3f9d8SHong Zhang PetscFunctionReturn(0); 669d7c3f9d8SHong Zhang } 670d7c3f9d8SHong Zhang 67145cf121fSXuan Zhou #undef __FUNCT__ 67245cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental" 67345cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info) 67445cf121fSXuan Zhou { 67545cf121fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 676df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d; 67745cf121fSXuan Zhou 67845cf121fSXuan Zhou PetscFunctionBegin; 679c9fc186eSXuan Zhou elem::Cholesky(elem::UPPER,*a->emat); 680fc54b460SXuan Zhou A->factortype = MAT_FACTOR_CHOLESKY; 681834d3fecSHong Zhang A->assembled = PETSC_TRUE; 68245cf121fSXuan Zhou PetscFunctionReturn(0); 68345cf121fSXuan Zhou } 68445cf121fSXuan Zhou 68579673f7bSHong Zhang #undef __FUNCT__ 68679673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental" 68779673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 68879673f7bSHong Zhang { 689cb76c1d8SXuan Zhou PetscErrorCode ierr; 690cb76c1d8SXuan Zhou 691cb76c1d8SXuan Zhou PetscFunctionBegin; 692cb76c1d8SXuan Zhou ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 693cb76c1d8SXuan Zhou ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr); 694cb76c1d8SXuan Zhou PetscFunctionReturn(0); 69579673f7bSHong Zhang } 696cb76c1d8SXuan Zhou 69779673f7bSHong Zhang #undef __FUNCT__ 69879673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental" 69979673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info) 70079673f7bSHong Zhang { 70179673f7bSHong Zhang PetscFunctionBegin; 70279673f7bSHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 70379673f7bSHong Zhang PetscFunctionReturn(0); 70479673f7bSHong Zhang } 70579673f7bSHong Zhang 7061293973dSHong Zhang #undef __FUNCT__ 70715767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental" 70815767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type) 70915767789SHong Zhang { 71015767789SHong Zhang PetscFunctionBegin; 71115767789SHong Zhang *type = MATSOLVERELEMENTAL; 71215767789SHong Zhang PetscFunctionReturn(0); 71315767789SHong Zhang } 71415767789SHong Zhang 71515767789SHong Zhang #undef __FUNCT__ 71615767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental" 71715767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F) 7181293973dSHong Zhang { 7191293973dSHong Zhang Mat B; 7201293973dSHong Zhang PetscErrorCode ierr; 7211293973dSHong Zhang 7221293973dSHong Zhang PetscFunctionBegin; 7231293973dSHong Zhang /* Create the factorization matrix */ 724ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 7251293973dSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 7261293973dSHong Zhang ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr); 7271293973dSHong Zhang ierr = MatSetUp(B);CHKERRQ(ierr); 7281293973dSHong Zhang B->factortype = ftype; 729bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr); 7301293973dSHong Zhang *F = B; 7311293973dSHong Zhang PetscFunctionReturn(0); 7321293973dSHong Zhang } 7331293973dSHong Zhang 7341f881ff8SXuan Zhou #undef __FUNCT__ 7351f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental" 7361f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm) 7371f881ff8SXuan Zhou { 7381f881ff8SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 7391f881ff8SXuan Zhou 7401f881ff8SXuan Zhou PetscFunctionBegin; 7411f881ff8SXuan Zhou switch (type){ 7421f881ff8SXuan Zhou case NORM_1: 743324645c7SJack Poulson *nrm = elem::OneNorm(*a->emat); 7441f881ff8SXuan Zhou break; 7451f881ff8SXuan Zhou case NORM_FROBENIUS: 746324645c7SJack Poulson *nrm = elem::FrobeniusNorm(*a->emat); 7471f881ff8SXuan Zhou break; 7481f881ff8SXuan Zhou case NORM_INFINITY: 749324645c7SJack Poulson *nrm = elem::InfinityNorm(*a->emat); 7501f881ff8SXuan Zhou break; 7511f881ff8SXuan Zhou default: 7521f881ff8SXuan Zhou printf("Error: unsupported norm type!\n"); 7531f881ff8SXuan Zhou } 7541f881ff8SXuan Zhou PetscFunctionReturn(0); 7551f881ff8SXuan Zhou } 7561f881ff8SXuan Zhou 7575262d616SXuan Zhou #undef __FUNCT__ 7585262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental" 7595262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A) 7605262d616SXuan Zhou { 7615262d616SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 7625262d616SXuan Zhou 7635262d616SXuan Zhou PetscFunctionBegin; 7645262d616SXuan Zhou elem::Zero(*a->emat); 7655262d616SXuan Zhou PetscFunctionReturn(0); 7665262d616SXuan Zhou } 7675262d616SXuan Zhou 768e09a3074SHong Zhang #undef __FUNCT__ 769db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental" 770db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols) 771db31f6deSJed Brown { 772db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 773db31f6deSJed Brown PetscErrorCode ierr; 774db31f6deSJed Brown PetscInt i,m,shift,stride,*idx; 775db31f6deSJed Brown 776db31f6deSJed Brown PetscFunctionBegin; 777db31f6deSJed Brown if (rows) { 778db31f6deSJed Brown m = a->emat->LocalHeight(); 779db31f6deSJed Brown shift = a->emat->ColShift(); 780db31f6deSJed Brown stride = a->emat->ColStride(); 781785e854fSJed Brown ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr); 782db31f6deSJed Brown for (i=0; i<m; i++) { 783db31f6deSJed Brown PetscInt rank,offset; 784db31f6deSJed Brown E2RO(A,0,shift+i*stride,&rank,&offset); 785db31f6deSJed Brown RO2P(A,0,rank,offset,&idx[i]); 786db31f6deSJed Brown } 787db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr); 788db31f6deSJed Brown } 789db31f6deSJed Brown if (cols) { 790db31f6deSJed Brown m = a->emat->LocalWidth(); 791db31f6deSJed Brown shift = a->emat->RowShift(); 792db31f6deSJed Brown stride = a->emat->RowStride(); 793785e854fSJed Brown ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr); 794db31f6deSJed Brown for (i=0; i<m; i++) { 795db31f6deSJed Brown PetscInt rank,offset; 796db31f6deSJed Brown E2RO(A,1,shift+i*stride,&rank,&offset); 797db31f6deSJed Brown RO2P(A,1,rank,offset,&idx[i]); 798db31f6deSJed Brown } 799db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr); 800db31f6deSJed Brown } 801db31f6deSJed Brown PetscFunctionReturn(0); 802db31f6deSJed Brown } 803db31f6deSJed Brown 804db31f6deSJed Brown #undef __FUNCT__ 8052ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense" 80619fd82e9SBarry Smith static PetscErrorCode MatConvert_Elemental_Dense(Mat A,MatType newtype,MatReuse reuse,Mat *B) 807af295397SXuan Zhou { 8082ef0cf24SXuan Zhou Mat Bmpi; 809af295397SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 810ce94432eSBarry Smith MPI_Comm comm; 8112ef0cf24SXuan Zhou PetscErrorCode ierr; 8122ef0cf24SXuan Zhou PetscInt rrank,ridx,crank,cidx,nrows,ncols,i,j; 813df311e6cSXuan Zhou PetscElemScalar v; 814573b0fb4SBarry Smith PetscBool s1,s2,s3; 815af295397SXuan Zhou 816af295397SXuan Zhou PetscFunctionBegin; 817ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 818573b0fb4SBarry Smith ierr = PetscStrcmp(newtype,MATDENSE,&s1);CHKERRQ(ierr); 819573b0fb4SBarry Smith ierr = PetscStrcmp(newtype,MATSEQDENSE,&s2);CHKERRQ(ierr); 820573b0fb4SBarry Smith ierr = PetscStrcmp(newtype,MATMPIDENSE,&s3);CHKERRQ(ierr); 821573b0fb4SBarry Smith if (!s1 && !s2 && !s3) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE"); 822af295397SXuan Zhou ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr); 823af295397SXuan Zhou ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 8242ef0cf24SXuan Zhou ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr); 825af295397SXuan Zhou ierr = MatSetUp(Bmpi);CHKERRQ(ierr); 8262ef0cf24SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 8272ef0cf24SXuan Zhou for (i=0; i<nrows; i++) { 8282ef0cf24SXuan Zhou PetscInt erow,ecol; 8292ef0cf24SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 8302ef0cf24SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 8312ef0cf24SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 8322ef0cf24SXuan Zhou for (j=0; j<ncols; j++) { 8332ef0cf24SXuan Zhou P2RO(A,1,j,&crank,&cidx); 8342ef0cf24SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 8352ef0cf24SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 8362ef0cf24SXuan Zhou v = a->emat->Get(erow,ecol); 837e6dea9dbSXuan Zhou ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 8382ef0cf24SXuan Zhou } 8392ef0cf24SXuan Zhou } 840af295397SXuan Zhou ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 841af295397SXuan Zhou ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 842c4ad791aSXuan Zhou if (reuse == MAT_REUSE_MATRIX) { 843c4ad791aSXuan Zhou ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr); 844c4ad791aSXuan Zhou } else { 845c4ad791aSXuan Zhou *B = Bmpi; 846c4ad791aSXuan Zhou } 847af295397SXuan Zhou PetscFunctionReturn(0); 848af295397SXuan Zhou } 849af295397SXuan Zhou 850af295397SXuan Zhou #undef __FUNCT__ 851db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental" 852db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A) 853db31f6deSJed Brown { 854db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 855db31f6deSJed Brown PetscErrorCode ierr; 8565e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 8575e9f5b67SHong Zhang PetscBool flg; 8585e9f5b67SHong Zhang MPI_Comm icomm; 859db31f6deSJed Brown 860db31f6deSJed Brown PetscFunctionBegin; 861c1ee1e62SHong Zhang a->interface->Detach(); 862aae2c449SHong Zhang delete a->interface; 863aae2c449SHong Zhang delete a->esubmat; 864db31f6deSJed Brown delete a->emat; 8655e9f5b67SHong Zhang 866ce94432eSBarry Smith elem::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A)); 8670298fd71SBarry Smith ierr = PetscCommDuplicate(cxxcomm,&icomm,NULL);CHKERRQ(ierr); 8685e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 8695e9f5b67SHong Zhang if (--commgrid->grid_refct == 0) { 8705e9f5b67SHong Zhang delete commgrid->grid; 8715e9f5b67SHong Zhang ierr = PetscFree(commgrid);CHKERRQ(ierr); 8725e9f5b67SHong Zhang } 8735e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 874bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",NULL);CHKERRQ(ierr); 875bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetFactor_petsc_C",NULL);CHKERRQ(ierr); 876bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSolverPackage_C",NULL);CHKERRQ(ierr); 877db31f6deSJed Brown ierr = PetscFree(A->data);CHKERRQ(ierr); 878db31f6deSJed Brown PetscFunctionReturn(0); 879db31f6deSJed Brown } 880db31f6deSJed Brown 881db31f6deSJed Brown #undef __FUNCT__ 882db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental" 883db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A) 884db31f6deSJed Brown { 885db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 886db31f6deSJed Brown PetscErrorCode ierr; 887db31f6deSJed Brown PetscMPIInt rsize,csize; 888db31f6deSJed Brown 889db31f6deSJed Brown PetscFunctionBegin; 890db31f6deSJed Brown ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 891db31f6deSJed Brown ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 892db31f6deSJed Brown 893efb79153SJack Poulson a->emat->Resize(A->rmap->N,A->cmap->N);CHKERRQ(ierr); 894db31f6deSJed Brown elem::Zero(*a->emat); 895db31f6deSJed Brown 896db31f6deSJed Brown ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr); 897db31f6deSJed Brown ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr); 898ce94432eSBarry Smith if (csize != rsize) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes"); 899db31f6deSJed Brown a->commsize = rsize; 900db31f6deSJed Brown a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize; 901db31f6deSJed Brown a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize; 902db31f6deSJed Brown a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize); 903db31f6deSJed Brown a->m[1] = A->cmap->N / csize + (a->mr[1] != csize); 904db31f6deSJed Brown PetscFunctionReturn(0); 905db31f6deSJed Brown } 906db31f6deSJed Brown 907aae2c449SHong Zhang #undef __FUNCT__ 908aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental" 909aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type) 910aae2c449SHong Zhang { 911aae2c449SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 912aae2c449SHong Zhang 913aae2c449SHong Zhang PetscFunctionBegin; 914aae2c449SHong Zhang a->interface->Detach(); 915aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 916aae2c449SHong Zhang PetscFunctionReturn(0); 917aae2c449SHong Zhang } 918aae2c449SHong Zhang 919aae2c449SHong Zhang #undef __FUNCT__ 920aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental" 921aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type) 922aae2c449SHong Zhang { 923aae2c449SHong Zhang PetscFunctionBegin; 924aae2c449SHong Zhang /* Currently does nothing */ 925aae2c449SHong Zhang PetscFunctionReturn(0); 926aae2c449SHong Zhang } 927aae2c449SHong Zhang 92840d92e34SHong Zhang /* -------------------------------------------------------------------*/ 92940d92e34SHong Zhang static struct _MatOps MatOps_Values = { 93040d92e34SHong Zhang MatSetValues_Elemental, 93140d92e34SHong Zhang 0, 93240d92e34SHong Zhang 0, 93340d92e34SHong Zhang MatMult_Elemental, 93440d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental, 9359426833fSXuan Zhou MatMultTranspose_Elemental, 936e883f9d5SXuan Zhou MatMultTransposeAdd_Elemental, 93740d92e34SHong Zhang MatSolve_Elemental, 938df311e6cSXuan Zhou MatSolveAdd_Elemental, 93940d92e34SHong Zhang 0, //MatSolveTranspose_Elemental, 94040d92e34SHong Zhang /*10*/ 0, //MatSolveTransposeAdd_Elemental, 94140d92e34SHong Zhang MatLUFactor_Elemental, 94240d92e34SHong Zhang MatCholeskyFactor_Elemental, 94340d92e34SHong Zhang 0, 94440d92e34SHong Zhang MatTranspose_Elemental, 94540d92e34SHong Zhang /*15*/ MatGetInfo_Elemental, 94640d92e34SHong Zhang 0, 94761119200SXuan Zhou MatGetDiagonal_Elemental, 948ade3cc5eSXuan Zhou MatDiagonalScale_Elemental, 94940d92e34SHong Zhang MatNorm_Elemental, 95040d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental, 95140d92e34SHong Zhang MatAssemblyEnd_Elemental, 95240d92e34SHong Zhang 0, //MatSetOption_Elemental, 95340d92e34SHong Zhang MatZeroEntries_Elemental, 95440d92e34SHong Zhang /*24*/ 0, 95540d92e34SHong Zhang MatLUFactorSymbolic_Elemental, 95640d92e34SHong Zhang MatLUFactorNumeric_Elemental, 95740d92e34SHong Zhang MatCholeskyFactorSymbolic_Elemental, 95840d92e34SHong Zhang MatCholeskyFactorNumeric_Elemental, 95940d92e34SHong Zhang /*29*/ MatSetUp_Elemental, 96040d92e34SHong Zhang 0, 96140d92e34SHong Zhang 0, 96240d92e34SHong Zhang 0, 96340d92e34SHong Zhang 0, 964df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental, 96540d92e34SHong Zhang 0, 96640d92e34SHong Zhang 0, 96740d92e34SHong Zhang 0, 96840d92e34SHong Zhang 0, 96940d92e34SHong Zhang /*39*/ MatAXPY_Elemental, 97040d92e34SHong Zhang 0, 97140d92e34SHong Zhang 0, 97240d92e34SHong Zhang 0, 97340d92e34SHong Zhang MatCopy_Elemental, 97440d92e34SHong Zhang /*44*/ 0, 97540d92e34SHong Zhang MatScale_Elemental, 97640d92e34SHong Zhang 0, 97740d92e34SHong Zhang 0, 97840d92e34SHong Zhang 0, 97940d92e34SHong Zhang /*49*/ 0, 98040d92e34SHong Zhang 0, 98140d92e34SHong Zhang 0, 98240d92e34SHong Zhang 0, 98340d92e34SHong Zhang 0, 98440d92e34SHong Zhang /*54*/ 0, 98540d92e34SHong Zhang 0, 98640d92e34SHong Zhang 0, 98740d92e34SHong Zhang 0, 98840d92e34SHong Zhang 0, 98940d92e34SHong Zhang /*59*/ 0, 99040d92e34SHong Zhang MatDestroy_Elemental, 99140d92e34SHong Zhang MatView_Elemental, 99240d92e34SHong Zhang 0, 99340d92e34SHong Zhang 0, 99440d92e34SHong Zhang /*64*/ 0, 99540d92e34SHong Zhang 0, 99640d92e34SHong Zhang 0, 99740d92e34SHong Zhang 0, 99840d92e34SHong Zhang 0, 99940d92e34SHong Zhang /*69*/ 0, 100040d92e34SHong Zhang 0, 10012ef0cf24SXuan Zhou MatConvert_Elemental_Dense, 100240d92e34SHong Zhang 0, 100340d92e34SHong Zhang 0, 100440d92e34SHong Zhang /*74*/ 0, 100540d92e34SHong Zhang 0, 100640d92e34SHong Zhang 0, 100740d92e34SHong Zhang 0, 100840d92e34SHong Zhang 0, 100940d92e34SHong Zhang /*79*/ 0, 101040d92e34SHong Zhang 0, 101140d92e34SHong Zhang 0, 101240d92e34SHong Zhang 0, 101340d92e34SHong Zhang 0, 101440d92e34SHong Zhang /*84*/ 0, 101540d92e34SHong Zhang 0, 101640d92e34SHong Zhang 0, 101740d92e34SHong Zhang 0, 101840d92e34SHong Zhang 0, 101940d92e34SHong Zhang /*89*/ MatMatMult_Elemental, 102040d92e34SHong Zhang MatMatMultSymbolic_Elemental, 102140d92e34SHong Zhang MatMatMultNumeric_Elemental, 102240d92e34SHong Zhang 0, 102340d92e34SHong Zhang 0, 102440d92e34SHong Zhang /*94*/ 0, 1025df311e6cSXuan Zhou MatMatTransposeMult_Elemental, 1026df311e6cSXuan Zhou MatMatTransposeMultSymbolic_Elemental, 1027df311e6cSXuan Zhou MatMatTransposeMultNumeric_Elemental, 102840d92e34SHong Zhang 0, 102940d92e34SHong Zhang /*99*/ 0, 103040d92e34SHong Zhang 0, 103140d92e34SHong Zhang 0, 1032dfcb0403SXuan Zhou MatConjugate_Elemental, 103340d92e34SHong Zhang 0, 103440d92e34SHong Zhang /*104*/0, 103540d92e34SHong Zhang 0, 103640d92e34SHong Zhang 0, 103740d92e34SHong Zhang 0, 103840d92e34SHong Zhang 0, 103940d92e34SHong Zhang /*109*/MatMatSolve_Elemental, 104040d92e34SHong Zhang 0, 104140d92e34SHong Zhang 0, 104240d92e34SHong Zhang 0, 104340d92e34SHong Zhang 0, 104440d92e34SHong Zhang /*114*/0, 104540d92e34SHong Zhang 0, 104640d92e34SHong Zhang 0, 104740d92e34SHong Zhang 0, 104840d92e34SHong Zhang 0, 104940d92e34SHong Zhang /*119*/0, 10504a29722dSXuan Zhou MatHermitianTranspose_Elemental, 105140d92e34SHong Zhang 0, 105240d92e34SHong Zhang 0, 105340d92e34SHong Zhang 0, 105440d92e34SHong Zhang /*124*/0, 105540d92e34SHong Zhang 0, 105640d92e34SHong Zhang 0, 105740d92e34SHong Zhang 0, 105840d92e34SHong Zhang 0, 105940d92e34SHong Zhang /*129*/0, 106040d92e34SHong Zhang 0, 106140d92e34SHong Zhang 0, 106240d92e34SHong Zhang 0, 106340d92e34SHong Zhang 0, 106440d92e34SHong Zhang /*134*/0, 106540d92e34SHong Zhang 0, 106640d92e34SHong Zhang 0, 106740d92e34SHong Zhang 0, 106840d92e34SHong Zhang 0 106940d92e34SHong Zhang }; 107040d92e34SHong Zhang 1071ed36708cSHong Zhang /*MC 1072ed36708cSHong Zhang MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package 1073ed36708cSHong Zhang 1074ed36708cSHong Zhang Options Database Keys: 10755cc86fc1SJed Brown + -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions() 10765cc86fc1SJed Brown - -mat_elemental_grid_height - sets Grid Height for 2D cyclic ordering of internal matrix 1077ed36708cSHong Zhang 1078ed36708cSHong Zhang Level: beginner 1079ed36708cSHong Zhang 10805cb544a0SHong Zhang .seealso: MATDENSE 1081ed36708cSHong Zhang M*/ 10824a29722dSXuan Zhou 1083db31f6deSJed Brown #undef __FUNCT__ 1084db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental" 10858cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_Elemental(Mat A) 1086db31f6deSJed Brown { 1087db31f6deSJed Brown Mat_Elemental *a; 1088db31f6deSJed Brown PetscErrorCode ierr; 10895682a260SJack Poulson PetscBool flg,flg1; 10905e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 10915e9f5b67SHong Zhang MPI_Comm icomm; 10925682a260SJack Poulson PetscInt optv1; 1093db31f6deSJed Brown 1094db31f6deSJed Brown PetscFunctionBegin; 1095607a6623SBarry Smith ierr = PetscElementalInitializePackage();CHKERRQ(ierr); 109640d92e34SHong Zhang ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 109740d92e34SHong Zhang A->insertmode = NOT_SET_VALUES; 1098db31f6deSJed Brown 1099b00a9115SJed Brown ierr = PetscNewLog(A,&a);CHKERRQ(ierr); 1100db31f6deSJed Brown A->data = (void*)a; 1101db31f6deSJed Brown 1102db31f6deSJed Brown /* Set up the elemental matrix */ 1103ce94432eSBarry Smith elem::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A)); 11045e9f5b67SHong Zhang 11055e9f5b67SHong Zhang /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */ 11065e9f5b67SHong Zhang if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) { 1107180a43e4SHong Zhang ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); 11085e9f5b67SHong Zhang } 11090298fd71SBarry Smith ierr = PetscCommDuplicate(cxxcomm,&icomm,NULL);CHKERRQ(ierr); 11105e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 11115e9f5b67SHong Zhang if (!flg) { 1112b00a9115SJed Brown ierr = PetscNewLog(A,&commgrid);CHKERRQ(ierr); 11135cb544a0SHong Zhang 1114ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 11155cb544a0SHong Zhang /* displayed default grid sizes (CommSize,1) are set by us arbitrarily until elem::Grid() is called */ 11162ef0cf24SXuan Zhou ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr); 11175682a260SJack Poulson if (flg1) { 11185682a260SJack Poulson if (elem::mpi::CommSize(cxxcomm) % optv1 != 0) { 11195682a260SJack Poulson SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Grid Height %D must evenly divide CommSize %D",optv1,(PetscInt)elem::mpi::CommSize(cxxcomm)); 1120ed667823SXuan Zhou } 11215682a260SJack Poulson commgrid->grid = new elem::Grid(cxxcomm,optv1); /* use user-provided grid height */ 11222ef0cf24SXuan Zhou } else { 11232adf0be3SHong Zhang commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */ 1124ed667823SXuan Zhou } 11255e9f5b67SHong Zhang commgrid->grid_refct = 1; 11265e9f5b67SHong Zhang ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr); 11275cb544a0SHong Zhang PetscOptionsEnd(); 11285e9f5b67SHong Zhang } else { 11295e9f5b67SHong Zhang commgrid->grid_refct++; 11305e9f5b67SHong Zhang } 11315e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 11325e9f5b67SHong Zhang a->grid = commgrid->grid; 1133df311e6cSXuan Zhou a->emat = new elem::DistMatrix<PetscElemScalar>(*a->grid); 1134df311e6cSXuan Zhou a->esubmat = new elem::Matrix<PetscElemScalar>(1,1); 1135df311e6cSXuan Zhou a->interface = new elem::AxpyInterface<PetscElemScalar>; 11367c920d81SXuan Zhou a->pivot = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>; 1137db31f6deSJed Brown 1138db31f6deSJed Brown /* build cache for off array entries formed */ 1139aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 1140bafd5131SHong Zhang 1141bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",MatGetOwnershipIS_Elemental);CHKERRQ(ierr); 1142bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetFactor_elemental_C",MatGetFactor_elemental_elemental);CHKERRQ(ierr); 1143db31f6deSJed Brown 1144db31f6deSJed Brown ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr); 1145db31f6deSJed Brown PetscFunctionReturn(0); 1146db31f6deSJed Brown } 11474a29722dSXuan Zhou 1148