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 57db31f6deSJed Brown #undef __FUNCT__ 58db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental" 59db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer) 60db31f6deSJed Brown { 61db31f6deSJed Brown PetscErrorCode ierr; 62db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 63db31f6deSJed Brown PetscBool iascii; 64db31f6deSJed Brown 65db31f6deSJed Brown PetscFunctionBegin; 66db31f6deSJed Brown ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 67db31f6deSJed Brown if (iascii) { 68db31f6deSJed Brown PetscViewerFormat format; 69db31f6deSJed Brown ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 70db31f6deSJed Brown if (format == PETSC_VIEWER_ASCII_INFO) { 7179673f7bSHong Zhang /* call elemental viewing function */ 722d8adcc7SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr); 73ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr); 74ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr); 754fe7bbcaSHong Zhang if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 7679673f7bSHong Zhang /* call elemental viewing function */ 7780622e0cSXuan Zhou ierr = PetscPrintf(((PetscObject)viewer)->comm,"test matview_elemental 2\n");CHKERRQ(ierr); 784fe7bbcaSHong Zhang } 7979673f7bSHong Zhang 80db31f6deSJed Brown } else if (format == PETSC_VIEWER_DEFAULT) { 81db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 82db31f6deSJed Brown ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr); 83db31f6deSJed Brown a->emat->Print("Elemental matrix (cyclic ordering)"); 84db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 85834d3fecSHong Zhang if (A->factortype == MAT_FACTOR_NONE){ 86d2daa67eSHong Zhang Mat Aaij; 870022473aSHong Zhang ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 880022473aSHong Zhang ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr); 89d2daa67eSHong Zhang ierr = MatView(Aaij,viewer);CHKERRQ(ierr); 900022473aSHong Zhang ierr = MatDestroy(&Aaij);CHKERRQ(ierr); 91834d3fecSHong Zhang } 92db31f6deSJed Brown } else SETERRQ(((PetscObject)viewer)->comm,PETSC_ERR_SUP,"Format"); 93d2daa67eSHong Zhang } else { 94d2daa67eSHong Zhang /* convert to aij/mpidense format and call MatView() */ 95d2daa67eSHong Zhang Mat Aaij; 96d2daa67eSHong Zhang ierr = PetscPrintf(((PetscObject)viewer)->comm,"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 97d2daa67eSHong Zhang ierr = MatComputeExplicitOperator(A,&Aaij);CHKERRQ(ierr); 98d2daa67eSHong Zhang ierr = MatView(Aaij,viewer);CHKERRQ(ierr); 99d2daa67eSHong Zhang ierr = MatDestroy(&Aaij);CHKERRQ(ierr); 100d2daa67eSHong Zhang } 101db31f6deSJed Brown PetscFunctionReturn(0); 102db31f6deSJed Brown } 103db31f6deSJed Brown 104db31f6deSJed Brown #undef __FUNCT__ 105180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental" 10615767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info) 107180a43e4SHong Zhang { 10815767789SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 10915767789SHong Zhang PetscMPIInt rank; 11015767789SHong Zhang 111180a43e4SHong Zhang PetscFunctionBegin; 11215767789SHong Zhang MPI_Comm_rank(((PetscObject)A)->comm,&rank); 11315767789SHong Zhang 11415767789SHong Zhang /* if (!rank) printf(" .........MatGetInfo_Elemental ...\n"); */ 11515767789SHong Zhang info->block_size = 1.0; /* ? */ 11615767789SHong Zhang 11715767789SHong Zhang if (flag == MAT_LOCAL) { 11815767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 11915767789SHong Zhang info->nz_used = info->nz_allocated; 12015767789SHong Zhang } else if (flag == MAT_GLOBAL_MAX) { 12115767789SHong Zhang //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,((PetscObject)matin)->comm);CHKERRQ(ierr); 12215767789SHong Zhang /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */ 12315767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet"); 12415767789SHong Zhang } else if (flag == MAT_GLOBAL_SUM) { 12515767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet"); 12615767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 12715767789SHong Zhang info->nz_used = info->nz_allocated; /* assume Elemental does accurate allocation */ 12815767789SHong Zhang //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,((PetscObject)A)->comm);CHKERRQ(ierr); 12915767789SHong Zhang //PetscPrintf(PETSC_COMM_SELF," ... [%d] locally allocated %g\n",rank,info->nz_allocated); 13015767789SHong Zhang } 13115767789SHong Zhang 13215767789SHong Zhang info->nz_unneeded = 0.0; 13315767789SHong Zhang info->assemblies = (double)A->num_ass; 13415767789SHong Zhang info->mallocs = 0; 13515767789SHong Zhang info->memory = ((PetscObject)A)->mem; 13615767789SHong Zhang info->fill_ratio_given = 0; /* determined by Elemental */ 13715767789SHong Zhang info->fill_ratio_needed = 0; 13815767789SHong Zhang info->factor_mallocs = 0; 139180a43e4SHong Zhang PetscFunctionReturn(0); 140180a43e4SHong Zhang } 141180a43e4SHong Zhang 142180a43e4SHong Zhang #undef __FUNCT__ 143db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental" 144e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode) 145db31f6deSJed Brown { 146db31f6deSJed Brown PetscErrorCode ierr; 147db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 148db31f6deSJed Brown PetscMPIInt rank; 149db31f6deSJed Brown PetscInt i,j,rrank,ridx,crank,cidx; 150db31f6deSJed Brown 151db31f6deSJed Brown PetscFunctionBegin; 152db31f6deSJed Brown ierr = MPI_Comm_rank(((PetscObject)A)->comm,&rank);CHKERRQ(ierr); 153db31f6deSJed Brown 154db31f6deSJed Brown const elem::Grid &grid = a->emat->Grid(); 155db31f6deSJed Brown for (i=0; i<nr; i++) { 156db31f6deSJed Brown PetscInt erow,ecol,elrow,elcol; 157db31f6deSJed Brown if (rows[i] < 0) continue; 158db31f6deSJed Brown P2RO(A,0,rows[i],&rrank,&ridx); 159db31f6deSJed Brown RO2E(A,0,rrank,ridx,&erow); 160db31f6deSJed Brown if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect row translation"); 161db31f6deSJed Brown for (j=0; j<nc; j++) { 162db31f6deSJed Brown if (cols[j] < 0) continue; 163db31f6deSJed Brown P2RO(A,1,cols[j],&crank,&cidx); 164db31f6deSJed Brown RO2E(A,1,crank,cidx,&ecol); 165db31f6deSJed Brown if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_PLIB,"Incorrect col translation"); 166aae2c449SHong Zhang if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */ 167aae2c449SHong Zhang if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 168aae2c449SHong 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); */ 169e6dea9dbSXuan Zhou a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]); 170aae2c449SHong Zhang a->interface->Axpy(1.0,*(a->esubmat),erow,ecol); 171aae2c449SHong Zhang continue; 172ed36708cSHong Zhang } 173db31f6deSJed Brown elrow = erow / grid.MCSize(); 174db31f6deSJed Brown elcol = ecol / grid.MRSize(); 175db31f6deSJed Brown switch (imode) { 176e6dea9dbSXuan Zhou case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 177e6dea9dbSXuan Zhou case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 178db31f6deSJed Brown default: SETERRQ1(((PetscObject)A)->comm,PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 179db31f6deSJed Brown } 180db31f6deSJed Brown } 181db31f6deSJed Brown } 182db31f6deSJed Brown PetscFunctionReturn(0); 183db31f6deSJed Brown } 184db31f6deSJed Brown 185db31f6deSJed Brown #undef __FUNCT__ 186db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental" 187db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y) 188db31f6deSJed Brown { 189db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 190db31f6deSJed Brown PetscErrorCode ierr; 191e6dea9dbSXuan Zhou const PetscElemScalar *x; 192e6dea9dbSXuan Zhou PetscElemScalar *y; 193df311e6cSXuan Zhou PetscElemScalar one = 1,zero = 0; 194db31f6deSJed Brown 195db31f6deSJed Brown PetscFunctionBegin; 196e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 197e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 198db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 199df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid); 200df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->rmap->N,1,0,y,A->rmap->n,*a->grid); 201db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,zero,ye); 202db31f6deSJed Brown } 203e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 204e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 205db31f6deSJed Brown PetscFunctionReturn(0); 206db31f6deSJed Brown } 207db31f6deSJed Brown 208db31f6deSJed Brown #undef __FUNCT__ 2099426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental" 2109426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y) 2119426833fSXuan Zhou { 2129426833fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 2139426833fSXuan Zhou PetscErrorCode ierr; 214df311e6cSXuan Zhou const PetscElemScalar *x; 215df311e6cSXuan Zhou PetscElemScalar *y; 216e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 2179426833fSXuan Zhou 2189426833fSXuan Zhou PetscFunctionBegin; 219e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 220e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2219426833fSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 222df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 223df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ye(A->cmap->N,1,0,y,A->cmap->n,*a->grid); 2249426833fSXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye); 2259426833fSXuan Zhou } 226e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 227e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2289426833fSXuan Zhou PetscFunctionReturn(0); 2299426833fSXuan Zhou } 2309426833fSXuan Zhou 2319426833fSXuan Zhou #undef __FUNCT__ 232db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental" 233db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 234db31f6deSJed Brown { 235db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 236db31f6deSJed Brown PetscErrorCode ierr; 237df311e6cSXuan Zhou const PetscElemScalar *x; 238df311e6cSXuan Zhou PetscElemScalar *z; 239e6dea9dbSXuan Zhou PetscElemScalar one = 1; 240db31f6deSJed Brown 241db31f6deSJed Brown PetscFunctionBegin; 242db31f6deSJed Brown if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 243e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 244e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 245db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 246df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->cmap->N,1,0,x,A->cmap->n,*a->grid); 247df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->rmap->N,1,0,z,A->rmap->n,*a->grid); 248db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze); 249db31f6deSJed Brown } 250e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 251e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 252db31f6deSJed Brown PetscFunctionReturn(0); 253db31f6deSJed Brown } 254db31f6deSJed Brown 255db31f6deSJed Brown #undef __FUNCT__ 256e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental" 257e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 258e883f9d5SXuan Zhou { 259e883f9d5SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 260e883f9d5SXuan Zhou PetscErrorCode ierr; 261df311e6cSXuan Zhou const PetscElemScalar *x; 262df311e6cSXuan Zhou PetscElemScalar *z; 263e6dea9dbSXuan Zhou PetscElemScalar one = 1; 264e883f9d5SXuan Zhou 265e883f9d5SXuan Zhou PetscFunctionBegin; 266e883f9d5SXuan Zhou if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 267e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 268e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 269e883f9d5SXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 270df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 271df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> ze(A->cmap->N,1,0,z,A->cmap->n,*a->grid); 272e883f9d5SXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,one,ze); 273e883f9d5SXuan Zhou } 274e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 275e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 276e883f9d5SXuan Zhou PetscFunctionReturn(0); 277e883f9d5SXuan Zhou } 278e883f9d5SXuan Zhou 279e883f9d5SXuan Zhou #undef __FUNCT__ 2809a9e8502SHong Zhang #define __FUNCT__ "MatMatMultNumeric_Elemental" 2819a9e8502SHong Zhang static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C) 282c1d1b975SXuan Zhou { 283c1d1b975SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 284c1d1b975SXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 2859a9e8502SHong Zhang Mat_Elemental *c = (Mat_Elemental*)C->data; 286e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 287c1d1b975SXuan Zhou 288c1d1b975SXuan Zhou PetscFunctionBegin; 289aae2c449SHong Zhang { /* Scoping so that constructor is called before pointer is returned */ 290c1d1b975SXuan Zhou elem::Gemm(elem::NORMAL,elem::NORMAL,one,*a->emat,*b->emat,zero,*c->emat); 291aae2c449SHong Zhang } 2929a9e8502SHong Zhang C->assembled = PETSC_TRUE; 2939a9e8502SHong Zhang PetscFunctionReturn(0); 2949a9e8502SHong Zhang } 2959a9e8502SHong Zhang 2969a9e8502SHong Zhang #undef __FUNCT__ 2979a9e8502SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_Elemental" 2989a9e8502SHong Zhang static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 2999a9e8502SHong Zhang { 3009a9e8502SHong Zhang PetscErrorCode ierr; 3019a9e8502SHong Zhang Mat Ce; 3029a9e8502SHong Zhang MPI_Comm comm=((PetscObject)A)->comm; 3039a9e8502SHong Zhang 3049a9e8502SHong Zhang PetscFunctionBegin; 3059a9e8502SHong Zhang ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 3069a9e8502SHong Zhang ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 3079a9e8502SHong Zhang ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 3089a9e8502SHong Zhang ierr = MatSetUp(Ce);CHKERRQ(ierr); 3099a9e8502SHong Zhang *C = Ce; 3109a9e8502SHong Zhang PetscFunctionReturn(0); 3119a9e8502SHong Zhang } 3129a9e8502SHong Zhang 3139a9e8502SHong Zhang #undef __FUNCT__ 3149a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental" 3159a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 3169a9e8502SHong Zhang { 3179a9e8502SHong Zhang PetscErrorCode ierr; 3189a9e8502SHong Zhang 3199a9e8502SHong Zhang PetscFunctionBegin; 3209a9e8502SHong Zhang if (scall == MAT_INITIAL_MATRIX){ 3219a9e8502SHong Zhang ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3229a9e8502SHong Zhang ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 3239a9e8502SHong Zhang ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3249a9e8502SHong Zhang } 3259a9e8502SHong Zhang ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3269a9e8502SHong Zhang ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 3279a9e8502SHong Zhang ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 328c1d1b975SXuan Zhou PetscFunctionReturn(0); 329c1d1b975SXuan Zhou } 330c1d1b975SXuan Zhou 331c1d1b975SXuan Zhou #undef __FUNCT__ 332df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental" 333df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C) 334df311e6cSXuan Zhou { 335df311e6cSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 336df311e6cSXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 337df311e6cSXuan Zhou Mat_Elemental *c = (Mat_Elemental*)C->data; 338e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 339df311e6cSXuan Zhou 340df311e6cSXuan Zhou PetscFunctionBegin; 341df311e6cSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 342df311e6cSXuan Zhou elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat); 343df311e6cSXuan Zhou } 344df311e6cSXuan Zhou C->assembled = PETSC_TRUE; 345df311e6cSXuan Zhou PetscFunctionReturn(0); 346df311e6cSXuan Zhou } 347df311e6cSXuan Zhou 348df311e6cSXuan Zhou #undef __FUNCT__ 349df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental" 350df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 351df311e6cSXuan Zhou { 352df311e6cSXuan Zhou PetscErrorCode ierr; 353df311e6cSXuan Zhou Mat Ce; 354df311e6cSXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 355df311e6cSXuan Zhou 356df311e6cSXuan Zhou PetscFunctionBegin; 357df311e6cSXuan Zhou ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 358df311e6cSXuan Zhou ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 359df311e6cSXuan Zhou ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 360df311e6cSXuan Zhou ierr = MatSetUp(Ce);CHKERRQ(ierr); 361df311e6cSXuan Zhou *C = Ce; 362df311e6cSXuan Zhou PetscFunctionReturn(0); 363df311e6cSXuan Zhou } 364df311e6cSXuan Zhou 365df311e6cSXuan Zhou #undef __FUNCT__ 366df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental" 367df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 368df311e6cSXuan Zhou { 369df311e6cSXuan Zhou PetscErrorCode ierr; 370df311e6cSXuan Zhou 371df311e6cSXuan Zhou PetscFunctionBegin; 372df311e6cSXuan Zhou if (scall == MAT_INITIAL_MATRIX){ 373df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 374df311e6cSXuan Zhou ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 375df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 376df311e6cSXuan Zhou } 377df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 378df311e6cSXuan Zhou ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 379df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 380df311e6cSXuan Zhou PetscFunctionReturn(0); 381df311e6cSXuan Zhou } 382df311e6cSXuan Zhou 383df311e6cSXuan Zhou #undef __FUNCT__ 3844ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental" 385e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a) 38665b78793SXuan Zhou { 38765b78793SXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 38865b78793SXuan Zhou 38965b78793SXuan Zhou PetscFunctionBegin; 390e6dea9dbSXuan Zhou elem::Scal((PetscElemScalar)a,*x->emat); 39165b78793SXuan Zhou PetscFunctionReturn(0); 39265b78793SXuan Zhou } 39365b78793SXuan Zhou 39465b78793SXuan Zhou #undef __FUNCT__ 3954ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental" 396e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str) 397e09a3074SHong Zhang { 398e09a3074SHong Zhang Mat_Elemental *x = (Mat_Elemental*)X->data; 399e09a3074SHong Zhang Mat_Elemental *y = (Mat_Elemental*)Y->data; 400e09a3074SHong Zhang 401e09a3074SHong Zhang PetscFunctionBegin; 402e6dea9dbSXuan Zhou elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat); 403e09a3074SHong Zhang PetscFunctionReturn(0); 404e09a3074SHong Zhang } 405e09a3074SHong Zhang 406ae844d54SHong Zhang #undef __FUNCT__ 407d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental" 408d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str) 409d6223691SXuan Zhou { 410d6223691SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 411d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 412d6223691SXuan Zhou 413d6223691SXuan Zhou PetscFunctionBegin; 414d6223691SXuan Zhou elem::Copy(*a->emat,*b->emat); 415d6223691SXuan Zhou PetscFunctionReturn(0); 416d6223691SXuan Zhou } 417d6223691SXuan Zhou 418d6223691SXuan Zhou #undef __FUNCT__ 419df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental" 420df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B) 421df311e6cSXuan Zhou { 422df311e6cSXuan Zhou Mat Be; 423df311e6cSXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 424df311e6cSXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 425df311e6cSXuan Zhou PetscErrorCode ierr; 426df311e6cSXuan Zhou 427df311e6cSXuan Zhou PetscFunctionBegin; 428df311e6cSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 429df311e6cSXuan Zhou ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 430df311e6cSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 431df311e6cSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 432df311e6cSXuan Zhou *B = Be; 433df311e6cSXuan Zhou if (op == MAT_COPY_VALUES) { 434df311e6cSXuan Zhou Mat_Elemental *b=(Mat_Elemental*)Be->data; 435df311e6cSXuan Zhou elem::Copy(*a->emat,*b->emat); 436df311e6cSXuan Zhou } 437df311e6cSXuan Zhou Be->assembled = PETSC_TRUE; 438df311e6cSXuan Zhou PetscFunctionReturn(0); 439df311e6cSXuan Zhou } 440df311e6cSXuan Zhou 441df311e6cSXuan Zhou #undef __FUNCT__ 442d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental" 443d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 444d6223691SXuan Zhou { 4453512f328SXuan Zhou /* Only out-of-place supported */ 4465262d616SXuan Zhou Mat Be; 4475262d616SXuan Zhou PetscErrorCode ierr; 4485262d616SXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 4494fe7bbcaSHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 450d6223691SXuan Zhou 451d6223691SXuan Zhou PetscFunctionBegin; 4525262d616SXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 4535262d616SXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 4545262d616SXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 4555262d616SXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 4565262d616SXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 4575262d616SXuan Zhou *B = Be; 4585262d616SXuan Zhou } 4594fe7bbcaSHong Zhang b = (Mat_Elemental*)Be->data; 4605262d616SXuan Zhou elem::Transpose(*a->emat,*b->emat); 4615262d616SXuan Zhou Be->assembled = PETSC_TRUE; 462d6223691SXuan Zhou PetscFunctionReturn(0); 463d6223691SXuan Zhou } 464d6223691SXuan Zhou 465d6223691SXuan Zhou #undef __FUNCT__ 466dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental" 467dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A) 468dfcb0403SXuan Zhou { 469dfcb0403SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 470dfcb0403SXuan Zhou 471dfcb0403SXuan Zhou PetscFunctionBegin; 472dfcb0403SXuan Zhou elem::Conjugate(*a->emat); 473dfcb0403SXuan Zhou PetscFunctionReturn(0); 474dfcb0403SXuan Zhou } 475dfcb0403SXuan Zhou 476dfcb0403SXuan Zhou #undef __FUNCT__ 4774a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental" 4784a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 4794a29722dSXuan Zhou { 4804a29722dSXuan Zhou /* Only out-of-place supported */ 4814a29722dSXuan Zhou Mat Be; 4824a29722dSXuan Zhou PetscErrorCode ierr; 4834a29722dSXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 4844a29722dSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 4854a29722dSXuan Zhou 4864a29722dSXuan Zhou PetscFunctionBegin; 4874a29722dSXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 4884a29722dSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 4894a29722dSXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 4904a29722dSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 4914a29722dSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 4924a29722dSXuan Zhou *B = Be; 4934a29722dSXuan Zhou } 4944a29722dSXuan Zhou b = (Mat_Elemental*)Be->data; 4954a29722dSXuan Zhou elem::Adjoint(*a->emat,*b->emat); 4964a29722dSXuan Zhou Be->assembled = PETSC_TRUE; 4974a29722dSXuan Zhou PetscFunctionReturn(0); 4984a29722dSXuan Zhou } 4994a29722dSXuan Zhou 5004a29722dSXuan Zhou #undef __FUNCT__ 5011f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental" 5021f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X) 5031f881ff8SXuan Zhou { 5041f881ff8SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 5051f881ff8SXuan Zhou PetscErrorCode ierr; 506df311e6cSXuan Zhou PetscElemScalar *x; 5071f881ff8SXuan Zhou 5081f881ff8SXuan Zhou PetscFunctionBegin; 50945cf121fSXuan Zhou ierr = VecCopy(B,X);CHKERRQ(ierr); 510e6dea9dbSXuan Zhou ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 511df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe(A->rmap->N,1,0,x,A->rmap->n,*a->grid); 512df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer = xe; 513fc54b460SXuan Zhou switch (A->factortype) { 514fc54b460SXuan Zhou case MAT_FACTOR_LU: 5151f881ff8SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 5164b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,xer); 51745cf121fSXuan Zhou elem::Copy(xer,xe); 518fc54b460SXuan Zhou } else { 5194b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,xer); 52045cf121fSXuan Zhou elem::Copy(xer,xe); 5211f881ff8SXuan Zhou } 522fc54b460SXuan Zhou break; 523fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 524fc54b460SXuan Zhou elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,xer); 525fc54b460SXuan Zhou elem::Copy(xer,xe); 526fc54b460SXuan Zhou break; 527fc54b460SXuan Zhou default: 5284fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 529fc54b460SXuan Zhou break; 5301f881ff8SXuan Zhou } 531e6dea9dbSXuan Zhou ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 5321f881ff8SXuan Zhou PetscFunctionReturn(0); 5331f881ff8SXuan Zhou } 5341f881ff8SXuan Zhou 5351f881ff8SXuan Zhou #undef __FUNCT__ 536df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental" 537df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X) 538df311e6cSXuan Zhou { 539df311e6cSXuan Zhou PetscErrorCode ierr; 540df311e6cSXuan Zhou 541df311e6cSXuan Zhou PetscFunctionBegin; 542df311e6cSXuan Zhou ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr); 5433d7f40dbSXuan Zhou ierr = VecAXPY(X,1,Y);CHKERRQ(ierr); 544df311e6cSXuan Zhou PetscFunctionReturn(0); 545df311e6cSXuan Zhou } 546df311e6cSXuan Zhou 547df311e6cSXuan Zhou #undef __FUNCT__ 548ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental" 549ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X) 550ae844d54SHong Zhang { 5511f0e42cfSHong Zhang Mat_Elemental *a=(Mat_Elemental*)A->data; 552d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 5531f0e42cfSHong Zhang Mat_Elemental *x=(Mat_Elemental*)X->data; 5541f0e42cfSHong Zhang 555ae844d54SHong Zhang PetscFunctionBegin; 556d6223691SXuan Zhou elem::Copy(*b->emat,*x->emat); 557fc54b460SXuan Zhou switch (A->factortype) { 558fc54b460SXuan Zhou case MAT_FACTOR_LU: 559d6223691SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 5604b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,*a->pivot,*x->emat); 561fc54b460SXuan Zhou } else { 5624b28e8e8SXuan Zhou elem::SolveAfterLU(elem::NORMAL,*a->emat,*x->emat); 563d6223691SXuan Zhou } 564fc54b460SXuan Zhou break; 565fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 566fc54b460SXuan Zhou elem::SolveAfterCholesky(elem::UPPER,elem::NORMAL,*a->emat,*x->emat); 567fc54b460SXuan Zhou break; 568fc54b460SXuan Zhou default: 5694fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 570fc54b460SXuan Zhou break; 571fc54b460SXuan Zhou } 572ae844d54SHong Zhang PetscFunctionReturn(0); 573ae844d54SHong Zhang } 574ae844d54SHong Zhang 575ae844d54SHong Zhang #undef __FUNCT__ 576ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental" 577ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info) 578ae844d54SHong Zhang { 5797c920d81SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 5807c920d81SXuan Zhou 581ae844d54SHong Zhang PetscFunctionBegin; 582d6223691SXuan Zhou if (info->dtcol){ 5837c920d81SXuan Zhou elem::LU(*a->emat,*a->pivot); 5841293973dSHong Zhang } else { 585d6223691SXuan Zhou elem::LU(*a->emat); 586d6223691SXuan Zhou } 5871293973dSHong Zhang A->factortype = MAT_FACTOR_LU; 588834d3fecSHong Zhang A->assembled = PETSC_TRUE; 589ae844d54SHong Zhang PetscFunctionReturn(0); 590ae844d54SHong Zhang } 591ae844d54SHong Zhang 592d7c3f9d8SHong Zhang #undef __FUNCT__ 593d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental" 594d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 595d7c3f9d8SHong Zhang { 596d7c3f9d8SHong Zhang PetscErrorCode ierr; 597d7c3f9d8SHong Zhang 598d7c3f9d8SHong Zhang PetscFunctionBegin; 599d7c3f9d8SHong Zhang ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 600d7c3f9d8SHong Zhang ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr); 601d7c3f9d8SHong Zhang PetscFunctionReturn(0); 602d7c3f9d8SHong Zhang } 603d7c3f9d8SHong Zhang 604d7c3f9d8SHong Zhang #undef __FUNCT__ 605d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental" 606d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 607d7c3f9d8SHong Zhang { 608d7c3f9d8SHong Zhang PetscFunctionBegin; 609d7c3f9d8SHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 610d7c3f9d8SHong Zhang PetscFunctionReturn(0); 611d7c3f9d8SHong Zhang } 612d7c3f9d8SHong Zhang 61345cf121fSXuan Zhou #undef __FUNCT__ 61445cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental" 61545cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info) 61645cf121fSXuan Zhou { 61745cf121fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 618df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d; 61945cf121fSXuan Zhou 62045cf121fSXuan Zhou PetscFunctionBegin; 621c9fc186eSXuan Zhou elem::Cholesky(elem::UPPER,*a->emat); 622fc54b460SXuan Zhou A->factortype = MAT_FACTOR_CHOLESKY; 623834d3fecSHong Zhang A->assembled = PETSC_TRUE; 62445cf121fSXuan Zhou PetscFunctionReturn(0); 62545cf121fSXuan Zhou } 62645cf121fSXuan Zhou 62779673f7bSHong Zhang #undef __FUNCT__ 62879673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental" 62979673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 63079673f7bSHong Zhang { 631cb76c1d8SXuan Zhou PetscErrorCode ierr; 632cb76c1d8SXuan Zhou 633cb76c1d8SXuan Zhou PetscFunctionBegin; 634cb76c1d8SXuan Zhou ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 635cb76c1d8SXuan Zhou ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr); 636cb76c1d8SXuan Zhou PetscFunctionReturn(0); 63779673f7bSHong Zhang } 638cb76c1d8SXuan Zhou 63979673f7bSHong Zhang #undef __FUNCT__ 64079673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental" 64179673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info) 64279673f7bSHong Zhang { 64379673f7bSHong Zhang PetscFunctionBegin; 64479673f7bSHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 64579673f7bSHong Zhang PetscFunctionReturn(0); 64679673f7bSHong Zhang } 64779673f7bSHong Zhang 64815767789SHong Zhang EXTERN_C_BEGIN 6491293973dSHong Zhang #undef __FUNCT__ 65015767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental" 65115767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type) 65215767789SHong Zhang { 65315767789SHong Zhang PetscFunctionBegin; 65415767789SHong Zhang *type = MATSOLVERELEMENTAL; 65515767789SHong Zhang PetscFunctionReturn(0); 65615767789SHong Zhang } 65715767789SHong Zhang EXTERN_C_END 65815767789SHong Zhang 65915767789SHong Zhang EXTERN_C_BEGIN 66015767789SHong Zhang #undef __FUNCT__ 66115767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental" 66215767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F) 6631293973dSHong Zhang { 6641293973dSHong Zhang Mat B; 6651293973dSHong Zhang PetscErrorCode ierr; 6661293973dSHong Zhang 6671293973dSHong Zhang PetscFunctionBegin; 6681293973dSHong Zhang /* Create the factorization matrix */ 6691293973dSHong Zhang ierr = MatCreate(((PetscObject)A)->comm,&B);CHKERRQ(ierr); 6701293973dSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 6711293973dSHong Zhang ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr); 6721293973dSHong Zhang ierr = MatSetUp(B);CHKERRQ(ierr); 6731293973dSHong Zhang B->factortype = ftype; 67415767789SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatFactorGetSolverPackage_C","MatFactorGetSolverPackage_elemental_elemental",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr); 6751293973dSHong Zhang *F = B; 6761293973dSHong Zhang PetscFunctionReturn(0); 6771293973dSHong Zhang } 67815767789SHong Zhang EXTERN_C_END 6791293973dSHong Zhang 6801f881ff8SXuan Zhou #undef __FUNCT__ 6811f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental" 6821f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm) 6831f881ff8SXuan Zhou { 6841f881ff8SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 6851f881ff8SXuan Zhou 6861f881ff8SXuan Zhou PetscFunctionBegin; 6871f881ff8SXuan Zhou switch (type){ 6881f881ff8SXuan Zhou case NORM_1: 6891f881ff8SXuan Zhou *nrm = elem::Norm(*a->emat,elem::ONE_NORM); 6901f881ff8SXuan Zhou break; 6911f881ff8SXuan Zhou case NORM_FROBENIUS: 6921f881ff8SXuan Zhou *nrm = elem::Norm(*a->emat,elem::FROBENIUS_NORM); 6931f881ff8SXuan Zhou break; 6941f881ff8SXuan Zhou case NORM_INFINITY: 6951f881ff8SXuan Zhou *nrm = elem::Norm(*a->emat,elem::INFINITY_NORM); 6961f881ff8SXuan Zhou break; 6971f881ff8SXuan Zhou default: 6981f881ff8SXuan Zhou printf("Error: unsupported norm type!\n"); 6991f881ff8SXuan Zhou } 7001f881ff8SXuan Zhou PetscFunctionReturn(0); 7011f881ff8SXuan Zhou } 7021f881ff8SXuan Zhou 7035262d616SXuan Zhou #undef __FUNCT__ 7045262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental" 7055262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A) 7065262d616SXuan Zhou { 7075262d616SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 7085262d616SXuan Zhou 7095262d616SXuan Zhou PetscFunctionBegin; 7105262d616SXuan Zhou elem::Zero(*a->emat); 7115262d616SXuan Zhou PetscFunctionReturn(0); 7125262d616SXuan Zhou } 7135262d616SXuan Zhou 7141293973dSHong Zhang EXTERN_C_BEGIN 715e09a3074SHong Zhang #undef __FUNCT__ 716db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental" 717db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols) 718db31f6deSJed Brown { 719db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 720db31f6deSJed Brown PetscErrorCode ierr; 721db31f6deSJed Brown PetscInt i,m,shift,stride,*idx; 722db31f6deSJed Brown 723db31f6deSJed Brown PetscFunctionBegin; 724db31f6deSJed Brown if (rows) { 725db31f6deSJed Brown m = a->emat->LocalHeight(); 726db31f6deSJed Brown shift = a->emat->ColShift(); 727db31f6deSJed Brown stride = a->emat->ColStride(); 728db31f6deSJed Brown ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr); 729db31f6deSJed Brown for (i=0; i<m; i++) { 730db31f6deSJed Brown PetscInt rank,offset; 731db31f6deSJed Brown E2RO(A,0,shift+i*stride,&rank,&offset); 732db31f6deSJed Brown RO2P(A,0,rank,offset,&idx[i]); 733db31f6deSJed Brown } 734db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr); 735db31f6deSJed Brown } 736db31f6deSJed Brown if (cols) { 737db31f6deSJed Brown m = a->emat->LocalWidth(); 738db31f6deSJed Brown shift = a->emat->RowShift(); 739db31f6deSJed Brown stride = a->emat->RowStride(); 740db31f6deSJed Brown ierr = PetscMalloc(m*sizeof(PetscInt),&idx);CHKERRQ(ierr); 741db31f6deSJed Brown for (i=0; i<m; i++) { 742db31f6deSJed Brown PetscInt rank,offset; 743db31f6deSJed Brown E2RO(A,1,shift+i*stride,&rank,&offset); 744db31f6deSJed Brown RO2P(A,1,rank,offset,&idx[i]); 745db31f6deSJed Brown } 746db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr); 747db31f6deSJed Brown } 748db31f6deSJed Brown PetscFunctionReturn(0); 749db31f6deSJed Brown } 7501293973dSHong Zhang EXTERN_C_END 751db31f6deSJed Brown 752db31f6deSJed Brown #undef __FUNCT__ 7532ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense" 7542ef0cf24SXuan Zhou static PetscErrorCode MatConvert_Elemental_Dense(Mat A,const MatType newtype,MatReuse reuse,Mat *B) 755af295397SXuan Zhou { 7562ef0cf24SXuan Zhou Mat Bmpi; 757af295397SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 758af295397SXuan Zhou MPI_Comm comm=((PetscObject)A)->comm; 7592ef0cf24SXuan Zhou PetscErrorCode ierr; 7602ef0cf24SXuan Zhou PetscInt rrank,ridx,crank,cidx,nrows,ncols,i,j; 761df311e6cSXuan Zhou PetscElemScalar v; 762af295397SXuan Zhou 763af295397SXuan Zhou PetscFunctionBegin; 764c4ad791aSXuan Zhou if (strcmp(newtype,MATDENSE) && strcmp(newtype,MATSEQDENSE) && strcmp(newtype,MATMPIDENSE)) { 765c4ad791aSXuan Zhou SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE"); 766c4ad791aSXuan Zhou } 767af295397SXuan Zhou ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr); 768af295397SXuan Zhou ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 7692ef0cf24SXuan Zhou ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr); 770af295397SXuan Zhou ierr = MatSetUp(Bmpi);CHKERRQ(ierr); 7712ef0cf24SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 7722ef0cf24SXuan Zhou for (i=0; i<nrows; i++) { 7732ef0cf24SXuan Zhou PetscInt erow,ecol; 7742ef0cf24SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 7752ef0cf24SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 7762ef0cf24SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 7772ef0cf24SXuan Zhou for (j=0; j<ncols; j++) { 7782ef0cf24SXuan Zhou P2RO(A,1,j,&crank,&cidx); 7792ef0cf24SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 7802ef0cf24SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 7812ef0cf24SXuan Zhou v = a->emat->Get(erow,ecol); 782e6dea9dbSXuan Zhou ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 7832ef0cf24SXuan Zhou } 7842ef0cf24SXuan Zhou } 785af295397SXuan Zhou ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 786af295397SXuan Zhou ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 787c4ad791aSXuan Zhou if (reuse == MAT_REUSE_MATRIX) { 788c4ad791aSXuan Zhou ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr); 789c4ad791aSXuan Zhou } else { 790c4ad791aSXuan Zhou *B = Bmpi; 791c4ad791aSXuan Zhou } 792af295397SXuan Zhou PetscFunctionReturn(0); 793af295397SXuan Zhou } 794af295397SXuan Zhou 795af295397SXuan Zhou #undef __FUNCT__ 796db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental" 797db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A) 798db31f6deSJed Brown { 799db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 800db31f6deSJed Brown PetscErrorCode ierr; 8015e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 8025e9f5b67SHong Zhang PetscBool flg; 8035e9f5b67SHong Zhang MPI_Comm icomm; 804db31f6deSJed Brown 805db31f6deSJed Brown PetscFunctionBegin; 806c1ee1e62SHong Zhang a->interface->Detach(); 807aae2c449SHong Zhang delete a->interface; 808aae2c449SHong Zhang delete a->esubmat; 809db31f6deSJed Brown delete a->emat; 8105e9f5b67SHong Zhang 811180a43e4SHong Zhang elem::mpi::Comm cxxcomm(((PetscObject)A)->comm); 8125e9f5b67SHong Zhang ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr); 8135e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 8145e9f5b67SHong Zhang if (--commgrid->grid_refct == 0) { 8155e9f5b67SHong Zhang delete commgrid->grid; 8165e9f5b67SHong Zhang ierr = PetscFree(commgrid);CHKERRQ(ierr); 8175e9f5b67SHong Zhang } 8185e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 819db31f6deSJed Brown ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","",PETSC_NULL);CHKERRQ(ierr); 8201293973dSHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_petsc_C","",PETSC_NULL);CHKERRQ(ierr); 82115767789SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatFactorGetSolverPackage_C","",PETSC_NULL);CHKERRQ(ierr); 822db31f6deSJed Brown ierr = PetscFree(A->data);CHKERRQ(ierr); 823db31f6deSJed Brown PetscFunctionReturn(0); 824db31f6deSJed Brown } 825db31f6deSJed Brown 826db31f6deSJed Brown #undef __FUNCT__ 827db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental" 828db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A) 829db31f6deSJed Brown { 830db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 831db31f6deSJed Brown PetscErrorCode ierr; 832db31f6deSJed Brown PetscMPIInt rsize,csize; 833db31f6deSJed Brown 834db31f6deSJed Brown PetscFunctionBegin; 835db31f6deSJed Brown ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 836db31f6deSJed Brown ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 837db31f6deSJed Brown 838db31f6deSJed Brown a->emat->ResizeTo(A->rmap->N,A->cmap->N);CHKERRQ(ierr); 839db31f6deSJed Brown elem::Zero(*a->emat); 840db31f6deSJed Brown 841db31f6deSJed Brown ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr); 842db31f6deSJed Brown ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr); 843db31f6deSJed Brown if (csize != rsize) SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes"); 844db31f6deSJed Brown a->commsize = rsize; 845db31f6deSJed Brown a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize; 846db31f6deSJed Brown a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize; 847db31f6deSJed Brown a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize); 848db31f6deSJed Brown a->m[1] = A->cmap->N / csize + (a->mr[1] != csize); 849db31f6deSJed Brown PetscFunctionReturn(0); 850db31f6deSJed Brown } 851db31f6deSJed Brown 852aae2c449SHong Zhang #undef __FUNCT__ 853aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental" 854aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type) 855aae2c449SHong Zhang { 856aae2c449SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 857aae2c449SHong Zhang 858aae2c449SHong Zhang PetscFunctionBegin; 859aae2c449SHong Zhang a->interface->Detach(); 860aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 861aae2c449SHong Zhang PetscFunctionReturn(0); 862aae2c449SHong Zhang } 863aae2c449SHong Zhang 864aae2c449SHong Zhang #undef __FUNCT__ 865aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental" 866aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type) 867aae2c449SHong Zhang { 868aae2c449SHong Zhang PetscFunctionBegin; 869aae2c449SHong Zhang /* Currently does nothing */ 870aae2c449SHong Zhang PetscFunctionReturn(0); 871aae2c449SHong Zhang } 872aae2c449SHong Zhang 87340d92e34SHong Zhang /* -------------------------------------------------------------------*/ 87440d92e34SHong Zhang static struct _MatOps MatOps_Values = { 87540d92e34SHong Zhang MatSetValues_Elemental, 87640d92e34SHong Zhang 0, 87740d92e34SHong Zhang 0, 87840d92e34SHong Zhang MatMult_Elemental, 87940d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental, 8809426833fSXuan Zhou MatMultTranspose_Elemental, 881e883f9d5SXuan Zhou MatMultTransposeAdd_Elemental, 88240d92e34SHong Zhang MatSolve_Elemental, 883df311e6cSXuan Zhou MatSolveAdd_Elemental, 88440d92e34SHong Zhang 0, //MatSolveTranspose_Elemental, 88540d92e34SHong Zhang /*10*/ 0, //MatSolveTransposeAdd_Elemental, 88640d92e34SHong Zhang MatLUFactor_Elemental, 88740d92e34SHong Zhang MatCholeskyFactor_Elemental, 88840d92e34SHong Zhang 0, 88940d92e34SHong Zhang MatTranspose_Elemental, 89040d92e34SHong Zhang /*15*/ MatGetInfo_Elemental, 89140d92e34SHong Zhang 0, 892*2adf0be3SHong Zhang 0, //MatGetDiagonal_Elemental,? 893*2adf0be3SHong Zhang 0, //MatDiagonalScale_Elemental,? 89440d92e34SHong Zhang MatNorm_Elemental, 89540d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental, 89640d92e34SHong Zhang MatAssemblyEnd_Elemental, 89740d92e34SHong Zhang 0, //MatSetOption_Elemental, 89840d92e34SHong Zhang MatZeroEntries_Elemental, 89940d92e34SHong Zhang /*24*/ 0, 90040d92e34SHong Zhang MatLUFactorSymbolic_Elemental, 90140d92e34SHong Zhang MatLUFactorNumeric_Elemental, 90240d92e34SHong Zhang MatCholeskyFactorSymbolic_Elemental, 90340d92e34SHong Zhang MatCholeskyFactorNumeric_Elemental, 90440d92e34SHong Zhang /*29*/ MatSetUp_Elemental, 90540d92e34SHong Zhang 0, 90640d92e34SHong Zhang 0, 90740d92e34SHong Zhang 0, 90840d92e34SHong Zhang 0, 909df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental, 91040d92e34SHong Zhang 0, 91140d92e34SHong Zhang 0, 91240d92e34SHong Zhang 0, 91340d92e34SHong Zhang 0, 91440d92e34SHong Zhang /*39*/ MatAXPY_Elemental, 91540d92e34SHong Zhang 0, 91640d92e34SHong Zhang 0, 91740d92e34SHong Zhang 0, 91840d92e34SHong Zhang MatCopy_Elemental, 91940d92e34SHong Zhang /*44*/ 0, 92040d92e34SHong Zhang MatScale_Elemental, 92140d92e34SHong Zhang 0, 92240d92e34SHong Zhang 0, 92340d92e34SHong Zhang 0, 92440d92e34SHong Zhang /*49*/ 0, 92540d92e34SHong Zhang 0, 92640d92e34SHong Zhang 0, 92740d92e34SHong Zhang 0, 92840d92e34SHong Zhang 0, 92940d92e34SHong Zhang /*54*/ 0, 93040d92e34SHong Zhang 0, 93140d92e34SHong Zhang 0, 93240d92e34SHong Zhang 0, 93340d92e34SHong Zhang 0, 93440d92e34SHong Zhang /*59*/ 0, 93540d92e34SHong Zhang MatDestroy_Elemental, 93640d92e34SHong Zhang MatView_Elemental, 93740d92e34SHong Zhang 0, 93840d92e34SHong Zhang 0, 93940d92e34SHong Zhang /*64*/ 0, 94040d92e34SHong Zhang 0, 94140d92e34SHong Zhang 0, 94240d92e34SHong Zhang 0, 94340d92e34SHong Zhang 0, 94440d92e34SHong Zhang /*69*/ 0, 94540d92e34SHong Zhang 0, 9462ef0cf24SXuan Zhou MatConvert_Elemental_Dense, 94740d92e34SHong Zhang 0, 94840d92e34SHong Zhang 0, 94940d92e34SHong Zhang /*74*/ 0, 95040d92e34SHong Zhang 0, 95140d92e34SHong Zhang 0, 95240d92e34SHong Zhang 0, 95340d92e34SHong Zhang 0, 95440d92e34SHong Zhang /*79*/ 0, 95540d92e34SHong Zhang 0, 95640d92e34SHong Zhang 0, 95740d92e34SHong Zhang 0, 95840d92e34SHong Zhang 0, 95940d92e34SHong Zhang /*84*/ 0, 96040d92e34SHong Zhang 0, 96140d92e34SHong Zhang 0, 96240d92e34SHong Zhang 0, 96340d92e34SHong Zhang 0, 96440d92e34SHong Zhang /*89*/ MatMatMult_Elemental, 96540d92e34SHong Zhang MatMatMultSymbolic_Elemental, 96640d92e34SHong Zhang MatMatMultNumeric_Elemental, 96740d92e34SHong Zhang 0, 96840d92e34SHong Zhang 0, 96940d92e34SHong Zhang /*94*/ 0, 970df311e6cSXuan Zhou MatMatTransposeMult_Elemental, 971df311e6cSXuan Zhou MatMatTransposeMultSymbolic_Elemental, 972df311e6cSXuan Zhou MatMatTransposeMultNumeric_Elemental, 97340d92e34SHong Zhang 0, 97440d92e34SHong Zhang /*99*/ 0, 97540d92e34SHong Zhang 0, 97640d92e34SHong Zhang 0, 977dfcb0403SXuan Zhou MatConjugate_Elemental, 97840d92e34SHong Zhang 0, 97940d92e34SHong Zhang /*104*/0, 98040d92e34SHong Zhang 0, 98140d92e34SHong Zhang 0, 98240d92e34SHong Zhang 0, 98340d92e34SHong Zhang 0, 98440d92e34SHong Zhang /*109*/MatMatSolve_Elemental, 98540d92e34SHong Zhang 0, 98640d92e34SHong Zhang 0, 98740d92e34SHong Zhang 0, 98840d92e34SHong Zhang 0, 98940d92e34SHong Zhang /*114*/0, 99040d92e34SHong Zhang 0, 99140d92e34SHong Zhang 0, 99240d92e34SHong Zhang 0, 99340d92e34SHong Zhang 0, 99440d92e34SHong Zhang /*119*/0, 9954a29722dSXuan Zhou MatHermitianTranspose_Elemental, 99640d92e34SHong Zhang 0, 99740d92e34SHong Zhang 0, 99840d92e34SHong Zhang 0, 99940d92e34SHong Zhang /*124*/0, 100040d92e34SHong Zhang 0, 100140d92e34SHong Zhang 0, 100240d92e34SHong Zhang 0, 100340d92e34SHong Zhang 0, 100440d92e34SHong Zhang /*129*/0, 100540d92e34SHong Zhang 0, 100640d92e34SHong Zhang 0, 100740d92e34SHong Zhang 0, 100840d92e34SHong Zhang 0, 100940d92e34SHong Zhang /*134*/0, 101040d92e34SHong Zhang 0, 101140d92e34SHong Zhang 0, 101240d92e34SHong Zhang 0, 101340d92e34SHong Zhang 0 101440d92e34SHong Zhang }; 101540d92e34SHong Zhang 1016ed36708cSHong Zhang /*MC 1017ed36708cSHong Zhang MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package 1018ed36708cSHong Zhang 1019ed36708cSHong Zhang Options Database Keys: 1020ed36708cSHong Zhang . -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions() 1021ed36708cSHong Zhang 1022ed36708cSHong Zhang Level: beginner 1023ed36708cSHong Zhang 1024ed36708cSHong Zhang .seealso: MATDENSE,MatCreateElemental() 1025ed36708cSHong Zhang M*/ 10264a29722dSXuan Zhou 1027db31f6deSJed Brown #undef __FUNCT__ 1028db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental" 1029db31f6deSJed Brown PETSC_EXTERN_C PetscErrorCode MatCreate_Elemental(Mat A) 1030db31f6deSJed Brown { 1031db31f6deSJed Brown Mat_Elemental *a; 1032db31f6deSJed Brown PetscErrorCode ierr; 1033ed667823SXuan Zhou PetscBool flg,flg1,flg2; 10345e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 10355e9f5b67SHong Zhang MPI_Comm icomm; 1036ed667823SXuan Zhou PetscInt optv1,optv2; 1037db31f6deSJed Brown 1038db31f6deSJed Brown PetscFunctionBegin; 1039db31f6deSJed Brown ierr = PetscElementalInitializePackage(PETSC_NULL);CHKERRQ(ierr); 104040d92e34SHong Zhang ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 104140d92e34SHong Zhang A->insertmode = NOT_SET_VALUES; 1042db31f6deSJed Brown 1043db31f6deSJed Brown ierr = PetscNewLog(A,Mat_Elemental,&a);CHKERRQ(ierr); 1044db31f6deSJed Brown A->data = (void*)a; 1045db31f6deSJed Brown 1046db31f6deSJed Brown /* Set up the elemental matrix */ 1047db31f6deSJed Brown elem::mpi::Comm cxxcomm(((PetscObject)A)->comm); 1048ed667823SXuan Zhou ierr = PetscOptionsBegin(((PetscObject)A)->comm,((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 10495e9f5b67SHong Zhang 10505e9f5b67SHong Zhang /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */ 10515e9f5b67SHong Zhang if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) { 1052180a43e4SHong Zhang ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); 10535e9f5b67SHong Zhang } 10545e9f5b67SHong Zhang ierr = PetscCommDuplicate(cxxcomm,&icomm,PETSC_NULL);CHKERRQ(ierr); 10555e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 10565e9f5b67SHong Zhang if (!flg) { 10575e9f5b67SHong Zhang ierr = PetscNewLog(A,Mat_Elemental_Grid,&commgrid);CHKERRQ(ierr); 10582ef0cf24SXuan Zhou ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::CommSize(cxxcomm),&optv1,&flg1);CHKERRQ(ierr); 10592ef0cf24SXuan Zhou ierr = PetscOptionsInt("-mat_elemental_grid_width","Grid Width","None",1,&optv2,&flg2);CHKERRQ(ierr); 1060ed667823SXuan Zhou if (flg1 || flg2) { 1061ed667823SXuan Zhou if (optv1*optv2 != elem::mpi::CommSize(cxxcomm)) { 10622ef0cf24SXuan Zhou SETERRQ(((PetscObject)A)->comm,PETSC_ERR_ARG_INCOMP,"Grid Height times Grid Width must equal CommSize"); 1063ed667823SXuan Zhou } 1064*2adf0be3SHong Zhang commgrid->grid = new elem::Grid(cxxcomm,optv1,optv2); /* use user-provided grid sizes */ 10652ef0cf24SXuan Zhou } else { 1066*2adf0be3SHong Zhang commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */ 1067ed667823SXuan Zhou } 10685e9f5b67SHong Zhang commgrid->grid_refct = 1; 10695e9f5b67SHong Zhang ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr); 10705e9f5b67SHong Zhang } else { 10715e9f5b67SHong Zhang commgrid->grid_refct++; 10725e9f5b67SHong Zhang } 10735e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 10745e9f5b67SHong Zhang a->grid = commgrid->grid; 1075df311e6cSXuan Zhou a->emat = new elem::DistMatrix<PetscElemScalar>(*a->grid); 1076df311e6cSXuan Zhou a->esubmat = new elem::Matrix<PetscElemScalar>(1,1); 1077df311e6cSXuan Zhou a->interface = new elem::AxpyInterface<PetscElemScalar>; 10787c920d81SXuan Zhou a->pivot = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>; 1079db31f6deSJed Brown 1080db31f6deSJed Brown /* build cache for off array entries formed */ 1081aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 1082bafd5131SHong Zhang 1083db31f6deSJed Brown ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetOwnershipIS_C","MatGetOwnershipIS_Elemental",MatGetOwnershipIS_Elemental);CHKERRQ(ierr); 108415767789SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)A,"MatGetFactor_elemental_C","MatGetFactor_elemental_elemental",MatGetFactor_elemental_elemental);CHKERRQ(ierr); 1085db31f6deSJed Brown 1086db31f6deSJed Brown ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr); 1087ed667823SXuan Zhou PetscOptionsEnd(); 1088db31f6deSJed Brown PetscFunctionReturn(0); 1089db31f6deSJed Brown } 10904a29722dSXuan Zhou 1091