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); 797a583510SJack Poulson elem::Print( *a->emat, "Elemental matrix (cyclic ordering)" ); 80db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 81834d3fecSHong Zhang if (A->factortype == MAT_FACTOR_NONE){ 8261119200SXuan Zhou Mat Adense; 83ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 8461119200SXuan Zhou ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr); 8561119200SXuan Zhou ierr = MatView(Adense,viewer);CHKERRQ(ierr); 8661119200SXuan Zhou ierr = MatDestroy(&Adense);CHKERRQ(ierr); 87834d3fecSHong Zhang } 88ce94432eSBarry Smith } else SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Format"); 89d2daa67eSHong Zhang } else { 905cb544a0SHong Zhang /* convert to dense format and call MatView() */ 9161119200SXuan Zhou Mat Adense; 92ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 9361119200SXuan Zhou ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr); 9461119200SXuan Zhou ierr = MatView(Adense,viewer);CHKERRQ(ierr); 9561119200SXuan Zhou ierr = MatDestroy(&Adense);CHKERRQ(ierr); 96d2daa67eSHong Zhang } 97db31f6deSJed Brown PetscFunctionReturn(0); 98db31f6deSJed Brown } 99db31f6deSJed Brown 100db31f6deSJed Brown #undef __FUNCT__ 101180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental" 10215767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info) 103180a43e4SHong Zhang { 10415767789SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 10515767789SHong Zhang PetscMPIInt rank; 10615767789SHong Zhang 107180a43e4SHong Zhang PetscFunctionBegin; 108ce94432eSBarry Smith MPI_Comm_rank(PetscObjectComm((PetscObject)A),&rank); 10915767789SHong Zhang 11015767789SHong Zhang /* if (!rank) printf(" .........MatGetInfo_Elemental ...\n"); */ 1115cb544a0SHong Zhang info->block_size = 1.0; 11215767789SHong Zhang 11315767789SHong Zhang if (flag == MAT_LOCAL) { 11415767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 11515767789SHong Zhang info->nz_used = info->nz_allocated; 11615767789SHong Zhang } else if (flag == MAT_GLOBAL_MAX) { 117ce94432eSBarry Smith //ierr = MPI_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr); 11815767789SHong Zhang /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */ 11915767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet"); 12015767789SHong Zhang } else if (flag == MAT_GLOBAL_SUM) { 12115767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet"); 12215767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 12315767789SHong Zhang info->nz_used = info->nz_allocated; /* assume Elemental does accurate allocation */ 124ce94432eSBarry Smith //ierr = MPI_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,PetscObjectComm((PetscObject)A));CHKERRQ(ierr); 12515767789SHong Zhang //PetscPrintf(PETSC_COMM_SELF," ... [%d] locally allocated %g\n",rank,info->nz_allocated); 12615767789SHong Zhang } 12715767789SHong Zhang 12815767789SHong Zhang info->nz_unneeded = 0.0; 12915767789SHong Zhang info->assemblies = (double)A->num_ass; 13015767789SHong Zhang info->mallocs = 0; 13115767789SHong Zhang info->memory = ((PetscObject)A)->mem; 13215767789SHong Zhang info->fill_ratio_given = 0; /* determined by Elemental */ 13315767789SHong Zhang info->fill_ratio_needed = 0; 13415767789SHong Zhang info->factor_mallocs = 0; 135180a43e4SHong Zhang PetscFunctionReturn(0); 136180a43e4SHong Zhang } 137180a43e4SHong Zhang 138180a43e4SHong Zhang #undef __FUNCT__ 139db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental" 140e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode) 141db31f6deSJed Brown { 142db31f6deSJed Brown PetscErrorCode ierr; 143db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 144db31f6deSJed Brown PetscMPIInt rank; 145db31f6deSJed Brown PetscInt i,j,rrank,ridx,crank,cidx; 146db31f6deSJed Brown 147db31f6deSJed Brown PetscFunctionBegin; 148ce94432eSBarry Smith ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)A),&rank);CHKERRQ(ierr); 149db31f6deSJed Brown 150db31f6deSJed Brown const elem::Grid &grid = a->emat->Grid(); 151db31f6deSJed Brown for (i=0; i<nr; i++) { 152db31f6deSJed Brown PetscInt erow,ecol,elrow,elcol; 153db31f6deSJed Brown if (rows[i] < 0) continue; 154db31f6deSJed Brown P2RO(A,0,rows[i],&rrank,&ridx); 155db31f6deSJed Brown RO2E(A,0,rrank,ridx,&erow); 156ce94432eSBarry Smith if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation"); 157db31f6deSJed Brown for (j=0; j<nc; j++) { 158db31f6deSJed Brown if (cols[j] < 0) continue; 159db31f6deSJed Brown P2RO(A,1,cols[j],&crank,&cidx); 160db31f6deSJed Brown RO2E(A,1,crank,cidx,&ecol); 161ce94432eSBarry Smith if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation"); 162aae2c449SHong Zhang if (erow % grid.MCSize() != grid.MCRank() || ecol % grid.MRSize() != grid.MRRank()){ /* off-proc entry */ 163aae2c449SHong Zhang if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 164aae2c449SHong 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); */ 165e6dea9dbSXuan Zhou a->esubmat->Set(0,0, (PetscElemScalar)vals[i*nc+j]); 166aae2c449SHong Zhang a->interface->Axpy(1.0,*(a->esubmat),erow,ecol); 167aae2c449SHong Zhang continue; 168ed36708cSHong Zhang } 169db31f6deSJed Brown elrow = erow / grid.MCSize(); 170db31f6deSJed Brown elcol = ecol / grid.MRSize(); 171db31f6deSJed Brown switch (imode) { 172e6dea9dbSXuan Zhou case INSERT_VALUES: a->emat->SetLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 173e6dea9dbSXuan Zhou case ADD_VALUES: a->emat->UpdateLocal(elrow,elcol,(PetscElemScalar)vals[i*nc+j]); break; 174ce94432eSBarry Smith default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 175db31f6deSJed Brown } 176db31f6deSJed Brown } 177db31f6deSJed Brown } 178db31f6deSJed Brown PetscFunctionReturn(0); 179db31f6deSJed Brown } 180db31f6deSJed Brown 181db31f6deSJed Brown #undef __FUNCT__ 182db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental" 183db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y) 184db31f6deSJed Brown { 185db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 186db31f6deSJed Brown PetscErrorCode ierr; 187e6dea9dbSXuan Zhou const PetscElemScalar *x; 188e6dea9dbSXuan Zhou PetscElemScalar *y; 189df311e6cSXuan Zhou PetscElemScalar one = 1,zero = 0; 190db31f6deSJed Brown 191db31f6deSJed Brown PetscFunctionBegin; 192e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 193e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 194db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 1950c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ye; 1960c18141cSBarry Smith xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n); 1970c18141cSBarry Smith ye.Attach(A->rmap->N,1,*a->grid,0,0,y,A->rmap->n); 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 */ 2190c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ye; 2200c18141cSBarry Smith xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n); 2210c18141cSBarry Smith ye.Attach(A->cmap->N,1,*a->grid,0,0,y,A->cmap->n); 2229426833fSXuan Zhou elem::Gemv(elem::TRANSPOSE,one,*a->emat,xe,zero,ye); 2239426833fSXuan Zhou } 224e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 225e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2269426833fSXuan Zhou PetscFunctionReturn(0); 2279426833fSXuan Zhou } 2289426833fSXuan Zhou 2299426833fSXuan Zhou #undef __FUNCT__ 230db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental" 231db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 232db31f6deSJed Brown { 233db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 234db31f6deSJed Brown PetscErrorCode ierr; 235df311e6cSXuan Zhou const PetscElemScalar *x; 236df311e6cSXuan Zhou PetscElemScalar *z; 237e6dea9dbSXuan Zhou PetscElemScalar one = 1; 238db31f6deSJed Brown 239db31f6deSJed Brown PetscFunctionBegin; 240db31f6deSJed Brown if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 241e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 242e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 243db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 2440c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ze; 2450c18141cSBarry Smith xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n); 2460c18141cSBarry Smith ze.Attach(A->rmap->N,1,*a->grid,0,0,z,A->rmap->n); 247db31f6deSJed Brown elem::Gemv(elem::NORMAL,one,*a->emat,xe,one,ze); 248db31f6deSJed Brown } 249e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 250e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 251db31f6deSJed Brown PetscFunctionReturn(0); 252db31f6deSJed Brown } 253db31f6deSJed Brown 254db31f6deSJed Brown #undef __FUNCT__ 255e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental" 256e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 257e883f9d5SXuan Zhou { 258e883f9d5SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 259e883f9d5SXuan Zhou PetscErrorCode ierr; 260df311e6cSXuan Zhou const PetscElemScalar *x; 261df311e6cSXuan Zhou PetscElemScalar *z; 262e6dea9dbSXuan Zhou PetscElemScalar one = 1; 263e883f9d5SXuan Zhou 264e883f9d5SXuan Zhou PetscFunctionBegin; 265e883f9d5SXuan Zhou if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 266e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 267e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 268e883f9d5SXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 2690c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe, ze; 2700c18141cSBarry Smith xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n); 2710c18141cSBarry Smith ze.Attach(A->cmap->N,1,*a->grid,0,0,z,A->cmap->n); 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; 302ce94432eSBarry Smith MPI_Comm comm; 3039a9e8502SHong Zhang 3049a9e8502SHong Zhang PetscFunctionBegin; 305ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 3069a9e8502SHong Zhang ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 3079a9e8502SHong Zhang ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 3089a9e8502SHong Zhang ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 3099a9e8502SHong Zhang ierr = MatSetUp(Ce);CHKERRQ(ierr); 3109a9e8502SHong Zhang *C = Ce; 3119a9e8502SHong Zhang PetscFunctionReturn(0); 3129a9e8502SHong Zhang } 3139a9e8502SHong Zhang 3149a9e8502SHong Zhang #undef __FUNCT__ 3159a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental" 3169a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 3179a9e8502SHong Zhang { 3189a9e8502SHong Zhang PetscErrorCode ierr; 3199a9e8502SHong Zhang 3209a9e8502SHong Zhang PetscFunctionBegin; 3219a9e8502SHong Zhang if (scall == MAT_INITIAL_MATRIX){ 3223ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3239a9e8502SHong Zhang ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 3243ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3259a9e8502SHong Zhang } 3263ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3279a9e8502SHong Zhang ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 3283ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 329c1d1b975SXuan Zhou PetscFunctionReturn(0); 330c1d1b975SXuan Zhou } 331c1d1b975SXuan Zhou 332c1d1b975SXuan Zhou #undef __FUNCT__ 333df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental" 334df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C) 335df311e6cSXuan Zhou { 336df311e6cSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 337df311e6cSXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 338df311e6cSXuan Zhou Mat_Elemental *c = (Mat_Elemental*)C->data; 339e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 340df311e6cSXuan Zhou 341df311e6cSXuan Zhou PetscFunctionBegin; 342df311e6cSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 343df311e6cSXuan Zhou elem::Gemm(elem::NORMAL,elem::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat); 344df311e6cSXuan Zhou } 345df311e6cSXuan Zhou C->assembled = PETSC_TRUE; 346df311e6cSXuan Zhou PetscFunctionReturn(0); 347df311e6cSXuan Zhou } 348df311e6cSXuan Zhou 349df311e6cSXuan Zhou #undef __FUNCT__ 350df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental" 351df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 352df311e6cSXuan Zhou { 353df311e6cSXuan Zhou PetscErrorCode ierr; 354df311e6cSXuan Zhou Mat Ce; 355ce94432eSBarry Smith MPI_Comm comm; 356df311e6cSXuan Zhou 357df311e6cSXuan Zhou PetscFunctionBegin; 358ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 359df311e6cSXuan Zhou ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 360df311e6cSXuan Zhou ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 361df311e6cSXuan Zhou ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 362df311e6cSXuan Zhou ierr = MatSetUp(Ce);CHKERRQ(ierr); 363df311e6cSXuan Zhou *C = Ce; 364df311e6cSXuan Zhou PetscFunctionReturn(0); 365df311e6cSXuan Zhou } 366df311e6cSXuan Zhou 367df311e6cSXuan Zhou #undef __FUNCT__ 368df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental" 369df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 370df311e6cSXuan Zhou { 371df311e6cSXuan Zhou PetscErrorCode ierr; 372df311e6cSXuan Zhou 373df311e6cSXuan Zhou PetscFunctionBegin; 374df311e6cSXuan Zhou if (scall == MAT_INITIAL_MATRIX){ 375df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 376df311e6cSXuan Zhou ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 377df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 378df311e6cSXuan Zhou } 379df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 380df311e6cSXuan Zhou ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 381df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 382df311e6cSXuan Zhou PetscFunctionReturn(0); 383df311e6cSXuan Zhou } 384df311e6cSXuan Zhou 385df311e6cSXuan Zhou #undef __FUNCT__ 38661119200SXuan Zhou #define __FUNCT__ "MatGetDiagonal_Elemental" 387a9d89745SXuan Zhou static PetscErrorCode MatGetDiagonal_Elemental(Mat A,Vec D) 38861119200SXuan Zhou { 389a9d89745SXuan Zhou PetscInt i,nrows,ncols,nD,rrank,ridx,crank,cidx; 390a9d89745SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 39161119200SXuan Zhou PetscErrorCode ierr; 392a9d89745SXuan Zhou PetscElemScalar v; 393ce94432eSBarry Smith MPI_Comm comm; 39461119200SXuan Zhou 39561119200SXuan Zhou PetscFunctionBegin; 396ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 397a9d89745SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 398a9d89745SXuan Zhou nD = nrows>ncols ? ncols : nrows; 399a9d89745SXuan Zhou for (i=0; i<nD; i++) { 400a9d89745SXuan Zhou PetscInt erow,ecol; 401a9d89745SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 402a9d89745SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 403a9d89745SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 404a9d89745SXuan Zhou P2RO(A,1,i,&crank,&cidx); 405a9d89745SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 406a9d89745SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 407a9d89745SXuan Zhou v = a->emat->Get(erow,ecol); 4087c8b904fSSatish Balay ierr = VecSetValues(D,1,&i,(PetscScalar*)&v,INSERT_VALUES);CHKERRQ(ierr); 409ade3cc5eSXuan Zhou } 410a9d89745SXuan Zhou ierr = VecAssemblyBegin(D);CHKERRQ(ierr); 411a9d89745SXuan Zhou ierr = VecAssemblyEnd(D);CHKERRQ(ierr); 41261119200SXuan Zhou PetscFunctionReturn(0); 41361119200SXuan Zhou } 41461119200SXuan Zhou 41561119200SXuan Zhou #undef __FUNCT__ 416ade3cc5eSXuan Zhou #define __FUNCT__ "MatDiagonalScale_Elemental" 417ade3cc5eSXuan Zhou static PetscErrorCode MatDiagonalScale_Elemental(Mat X,Vec L,Vec R) 418ade3cc5eSXuan Zhou { 419ade3cc5eSXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 420ade3cc5eSXuan Zhou const PetscElemScalar *d; 421ade3cc5eSXuan Zhou PetscErrorCode ierr; 422ade3cc5eSXuan Zhou 423ade3cc5eSXuan Zhou PetscFunctionBegin; 4249065cd98SJed Brown if (R) { 425ade3cc5eSXuan Zhou ierr = VecGetArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr); 4260c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> de; 4270c18141cSBarry Smith de.LockedAttach(X->cmap->N,1,*x->grid,0,0,d,X->cmap->n); 428ade3cc5eSXuan Zhou elem::DiagonalScale(elem::RIGHT,elem::NORMAL,de,*x->emat); 429ade3cc5eSXuan Zhou ierr = VecRestoreArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr); 4309065cd98SJed Brown } 4319065cd98SJed Brown if (L) { 432ade3cc5eSXuan Zhou ierr = VecGetArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr); 4330c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> de; 4340c18141cSBarry Smith de.LockedAttach(X->rmap->N,1,*x->grid,0,0,d,X->rmap->n); 435ade3cc5eSXuan Zhou elem::DiagonalScale(elem::LEFT,elem::NORMAL,de,*x->emat); 436ade3cc5eSXuan Zhou ierr = VecRestoreArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr); 437ade3cc5eSXuan Zhou } 438ade3cc5eSXuan Zhou PetscFunctionReturn(0); 439ade3cc5eSXuan Zhou } 440ade3cc5eSXuan Zhou 441ade3cc5eSXuan Zhou #undef __FUNCT__ 4424ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental" 443e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a) 44465b78793SXuan Zhou { 44565b78793SXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 44665b78793SXuan Zhou 44765b78793SXuan Zhou PetscFunctionBegin; 4480c18141cSBarry Smith elem::Scale((PetscElemScalar)a,*x->emat); 44965b78793SXuan Zhou PetscFunctionReturn(0); 45065b78793SXuan Zhou } 45165b78793SXuan Zhou 45265b78793SXuan Zhou #undef __FUNCT__ 4534ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental" 454e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str) 455e09a3074SHong Zhang { 456e09a3074SHong Zhang Mat_Elemental *x = (Mat_Elemental*)X->data; 457e09a3074SHong Zhang Mat_Elemental *y = (Mat_Elemental*)Y->data; 45868446cf8SJed Brown PetscErrorCode ierr; 459e09a3074SHong Zhang 460e09a3074SHong Zhang PetscFunctionBegin; 461e6dea9dbSXuan Zhou elem::Axpy((PetscElemScalar)a,*x->emat,*y->emat); 46221f6c9c4SJed Brown ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 463e09a3074SHong Zhang PetscFunctionReturn(0); 464e09a3074SHong Zhang } 465e09a3074SHong Zhang 466ae844d54SHong Zhang #undef __FUNCT__ 467d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental" 468d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str) 469d6223691SXuan Zhou { 470d6223691SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 471d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 472d6223691SXuan Zhou 473d6223691SXuan Zhou PetscFunctionBegin; 474d6223691SXuan Zhou elem::Copy(*a->emat,*b->emat); 475d6223691SXuan Zhou PetscFunctionReturn(0); 476d6223691SXuan Zhou } 477d6223691SXuan Zhou 478d6223691SXuan Zhou #undef __FUNCT__ 479df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental" 480df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B) 481df311e6cSXuan Zhou { 482df311e6cSXuan Zhou Mat Be; 483ce94432eSBarry Smith MPI_Comm comm; 484df311e6cSXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 485df311e6cSXuan Zhou PetscErrorCode ierr; 486df311e6cSXuan Zhou 487df311e6cSXuan Zhou PetscFunctionBegin; 488ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 489df311e6cSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 490df311e6cSXuan Zhou ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 491df311e6cSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 492df311e6cSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 493df311e6cSXuan Zhou *B = Be; 494df311e6cSXuan Zhou if (op == MAT_COPY_VALUES) { 495df311e6cSXuan Zhou Mat_Elemental *b=(Mat_Elemental*)Be->data; 496df311e6cSXuan Zhou elem::Copy(*a->emat,*b->emat); 497df311e6cSXuan Zhou } 498df311e6cSXuan Zhou Be->assembled = PETSC_TRUE; 499df311e6cSXuan Zhou PetscFunctionReturn(0); 500df311e6cSXuan Zhou } 501df311e6cSXuan Zhou 502df311e6cSXuan Zhou #undef __FUNCT__ 503d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental" 504d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 505d6223691SXuan Zhou { 506*ec8cb81fSBarry Smith Mat Be = *B; 5075262d616SXuan Zhou PetscErrorCode ierr; 508ce94432eSBarry Smith MPI_Comm comm; 5094fe7bbcaSHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 510d6223691SXuan Zhou 511d6223691SXuan Zhou PetscFunctionBegin; 512ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 5135cb544a0SHong Zhang /* Only out-of-place supported */ 5145262d616SXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 5155262d616SXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 5165262d616SXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 5175262d616SXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 5185262d616SXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 5195262d616SXuan Zhou *B = Be; 5205262d616SXuan Zhou } 5214fe7bbcaSHong Zhang b = (Mat_Elemental*)Be->data; 5225262d616SXuan Zhou elem::Transpose(*a->emat,*b->emat); 5235262d616SXuan Zhou Be->assembled = PETSC_TRUE; 524d6223691SXuan Zhou PetscFunctionReturn(0); 525d6223691SXuan Zhou } 526d6223691SXuan Zhou 527d6223691SXuan Zhou #undef __FUNCT__ 528dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental" 529dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A) 530dfcb0403SXuan Zhou { 531dfcb0403SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 532dfcb0403SXuan Zhou 533dfcb0403SXuan Zhou PetscFunctionBegin; 534dfcb0403SXuan Zhou elem::Conjugate(*a->emat); 535dfcb0403SXuan Zhou PetscFunctionReturn(0); 536dfcb0403SXuan Zhou } 537dfcb0403SXuan Zhou 538dfcb0403SXuan Zhou #undef __FUNCT__ 5394a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental" 5404a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 5414a29722dSXuan Zhou { 542*ec8cb81fSBarry Smith Mat Be = *B; 5434a29722dSXuan Zhou PetscErrorCode ierr; 544ce94432eSBarry Smith MPI_Comm comm; 5454a29722dSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 5464a29722dSXuan Zhou 5474a29722dSXuan Zhou PetscFunctionBegin; 548ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 5495cb544a0SHong Zhang /* Only out-of-place supported */ 5504a29722dSXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 5514a29722dSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 5524a29722dSXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 5534a29722dSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 5544a29722dSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 5554a29722dSXuan Zhou *B = Be; 5564a29722dSXuan Zhou } 5574a29722dSXuan Zhou b = (Mat_Elemental*)Be->data; 5584a29722dSXuan Zhou elem::Adjoint(*a->emat,*b->emat); 5594a29722dSXuan Zhou Be->assembled = PETSC_TRUE; 5604a29722dSXuan Zhou PetscFunctionReturn(0); 5614a29722dSXuan Zhou } 5624a29722dSXuan Zhou 5634a29722dSXuan Zhou #undef __FUNCT__ 5641f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental" 5651f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X) 5661f881ff8SXuan Zhou { 5671f881ff8SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 5681f881ff8SXuan Zhou PetscErrorCode ierr; 569df311e6cSXuan Zhou PetscElemScalar *x; 5701f881ff8SXuan Zhou 5711f881ff8SXuan Zhou PetscFunctionBegin; 57245cf121fSXuan Zhou ierr = VecCopy(B,X);CHKERRQ(ierr); 573e6dea9dbSXuan Zhou ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 5740c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::VC,elem::STAR> xe; 5750c18141cSBarry Smith xe.Attach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n); 5760c18141cSBarry Smith elem::DistMatrix<PetscElemScalar,elem::MC,elem::MR> xer(xe); 577fc54b460SXuan Zhou switch (A->factortype) { 578fc54b460SXuan Zhou case MAT_FACTOR_LU: 5791f881ff8SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 580324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,*a->pivot,xer); 58145cf121fSXuan Zhou elem::Copy(xer,xe); 582fc54b460SXuan Zhou } else { 583324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,xer); 58445cf121fSXuan Zhou elem::Copy(xer,xe); 5851f881ff8SXuan Zhou } 586fc54b460SXuan Zhou break; 587fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 588324645c7SJack Poulson elem::cholesky::SolveAfter(elem::UPPER,elem::NORMAL,*a->emat,xer); 589fc54b460SXuan Zhou elem::Copy(xer,xe); 590fc54b460SXuan Zhou break; 591fc54b460SXuan Zhou default: 5924fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 593fc54b460SXuan Zhou break; 5941f881ff8SXuan Zhou } 595e6dea9dbSXuan Zhou ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 5961f881ff8SXuan Zhou PetscFunctionReturn(0); 5971f881ff8SXuan Zhou } 5981f881ff8SXuan Zhou 5991f881ff8SXuan Zhou #undef __FUNCT__ 600df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental" 601df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X) 602df311e6cSXuan Zhou { 603df311e6cSXuan Zhou PetscErrorCode ierr; 604df311e6cSXuan Zhou 605df311e6cSXuan Zhou PetscFunctionBegin; 606df311e6cSXuan Zhou ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr); 6073d7f40dbSXuan Zhou ierr = VecAXPY(X,1,Y);CHKERRQ(ierr); 608df311e6cSXuan Zhou PetscFunctionReturn(0); 609df311e6cSXuan Zhou } 610df311e6cSXuan Zhou 611df311e6cSXuan Zhou #undef __FUNCT__ 612ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental" 613ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X) 614ae844d54SHong Zhang { 6151f0e42cfSHong Zhang Mat_Elemental *a=(Mat_Elemental*)A->data; 616d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 6171f0e42cfSHong Zhang Mat_Elemental *x=(Mat_Elemental*)X->data; 6181f0e42cfSHong Zhang 619ae844d54SHong Zhang PetscFunctionBegin; 620d6223691SXuan Zhou elem::Copy(*b->emat,*x->emat); 621fc54b460SXuan Zhou switch (A->factortype) { 622fc54b460SXuan Zhou case MAT_FACTOR_LU: 623d6223691SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 624324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,*a->pivot,*x->emat); 625fc54b460SXuan Zhou } else { 626324645c7SJack Poulson elem::lu::SolveAfter(elem::NORMAL,*a->emat,*x->emat); 627d6223691SXuan Zhou } 628fc54b460SXuan Zhou break; 629fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 630324645c7SJack Poulson elem::cholesky::SolveAfter(elem::UPPER,elem::NORMAL,*a->emat,*x->emat); 631fc54b460SXuan Zhou break; 632fc54b460SXuan Zhou default: 6334fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 634fc54b460SXuan Zhou break; 635fc54b460SXuan Zhou } 636ae844d54SHong Zhang PetscFunctionReturn(0); 637ae844d54SHong Zhang } 638ae844d54SHong Zhang 639ae844d54SHong Zhang #undef __FUNCT__ 640ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental" 641ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info) 642ae844d54SHong Zhang { 6437c920d81SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 6447c920d81SXuan Zhou 645ae844d54SHong Zhang PetscFunctionBegin; 646d6223691SXuan Zhou if (info->dtcol){ 6477c920d81SXuan Zhou elem::LU(*a->emat,*a->pivot); 6481293973dSHong Zhang } else { 649d6223691SXuan Zhou elem::LU(*a->emat); 650d6223691SXuan Zhou } 6511293973dSHong Zhang A->factortype = MAT_FACTOR_LU; 652834d3fecSHong Zhang A->assembled = PETSC_TRUE; 653ae844d54SHong Zhang PetscFunctionReturn(0); 654ae844d54SHong Zhang } 655ae844d54SHong Zhang 656d7c3f9d8SHong Zhang #undef __FUNCT__ 657d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental" 658d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 659d7c3f9d8SHong Zhang { 660d7c3f9d8SHong Zhang PetscErrorCode ierr; 661d7c3f9d8SHong Zhang 662d7c3f9d8SHong Zhang PetscFunctionBegin; 663d7c3f9d8SHong Zhang ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 664d7c3f9d8SHong Zhang ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr); 665d7c3f9d8SHong Zhang PetscFunctionReturn(0); 666d7c3f9d8SHong Zhang } 667d7c3f9d8SHong Zhang 668d7c3f9d8SHong Zhang #undef __FUNCT__ 669d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental" 670d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 671d7c3f9d8SHong Zhang { 672d7c3f9d8SHong Zhang PetscFunctionBegin; 673d7c3f9d8SHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 674d7c3f9d8SHong Zhang PetscFunctionReturn(0); 675d7c3f9d8SHong Zhang } 676d7c3f9d8SHong Zhang 67745cf121fSXuan Zhou #undef __FUNCT__ 67845cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental" 67945cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info) 68045cf121fSXuan Zhou { 68145cf121fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 682df311e6cSXuan Zhou elem::DistMatrix<PetscElemScalar,elem::MC,elem::STAR> d; 68345cf121fSXuan Zhou 68445cf121fSXuan Zhou PetscFunctionBegin; 685c9fc186eSXuan Zhou elem::Cholesky(elem::UPPER,*a->emat); 686fc54b460SXuan Zhou A->factortype = MAT_FACTOR_CHOLESKY; 687834d3fecSHong Zhang A->assembled = PETSC_TRUE; 68845cf121fSXuan Zhou PetscFunctionReturn(0); 68945cf121fSXuan Zhou } 69045cf121fSXuan Zhou 69179673f7bSHong Zhang #undef __FUNCT__ 69279673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental" 69379673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 69479673f7bSHong Zhang { 695cb76c1d8SXuan Zhou PetscErrorCode ierr; 696cb76c1d8SXuan Zhou 697cb76c1d8SXuan Zhou PetscFunctionBegin; 698cb76c1d8SXuan Zhou ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 699cb76c1d8SXuan Zhou ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr); 700cb76c1d8SXuan Zhou PetscFunctionReturn(0); 70179673f7bSHong Zhang } 702cb76c1d8SXuan Zhou 70379673f7bSHong Zhang #undef __FUNCT__ 70479673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental" 70579673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info) 70679673f7bSHong Zhang { 70779673f7bSHong Zhang PetscFunctionBegin; 70879673f7bSHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 70979673f7bSHong Zhang PetscFunctionReturn(0); 71079673f7bSHong Zhang } 71179673f7bSHong Zhang 7121293973dSHong Zhang #undef __FUNCT__ 71315767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental" 71415767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type) 71515767789SHong Zhang { 71615767789SHong Zhang PetscFunctionBegin; 71715767789SHong Zhang *type = MATSOLVERELEMENTAL; 71815767789SHong Zhang PetscFunctionReturn(0); 71915767789SHong Zhang } 72015767789SHong Zhang 72115767789SHong Zhang #undef __FUNCT__ 72215767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental" 72315767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F) 7241293973dSHong Zhang { 7251293973dSHong Zhang Mat B; 7261293973dSHong Zhang PetscErrorCode ierr; 7271293973dSHong Zhang 7281293973dSHong Zhang PetscFunctionBegin; 7291293973dSHong Zhang /* Create the factorization matrix */ 730ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 7311293973dSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 7321293973dSHong Zhang ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr); 7331293973dSHong Zhang ierr = MatSetUp(B);CHKERRQ(ierr); 7341293973dSHong Zhang B->factortype = ftype; 735bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr); 7361293973dSHong Zhang *F = B; 7371293973dSHong Zhang PetscFunctionReturn(0); 7381293973dSHong Zhang } 7391293973dSHong Zhang 7401f881ff8SXuan Zhou #undef __FUNCT__ 7411f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental" 7421f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm) 7431f881ff8SXuan Zhou { 7441f881ff8SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 7451f881ff8SXuan Zhou 7461f881ff8SXuan Zhou PetscFunctionBegin; 7471f881ff8SXuan Zhou switch (type){ 7481f881ff8SXuan Zhou case NORM_1: 749324645c7SJack Poulson *nrm = elem::OneNorm(*a->emat); 7501f881ff8SXuan Zhou break; 7511f881ff8SXuan Zhou case NORM_FROBENIUS: 752324645c7SJack Poulson *nrm = elem::FrobeniusNorm(*a->emat); 7531f881ff8SXuan Zhou break; 7541f881ff8SXuan Zhou case NORM_INFINITY: 755324645c7SJack Poulson *nrm = elem::InfinityNorm(*a->emat); 7561f881ff8SXuan Zhou break; 7571f881ff8SXuan Zhou default: 7581f881ff8SXuan Zhou printf("Error: unsupported norm type!\n"); 7591f881ff8SXuan Zhou } 7601f881ff8SXuan Zhou PetscFunctionReturn(0); 7611f881ff8SXuan Zhou } 7621f881ff8SXuan Zhou 7635262d616SXuan Zhou #undef __FUNCT__ 7645262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental" 7655262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A) 7665262d616SXuan Zhou { 7675262d616SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 7685262d616SXuan Zhou 7695262d616SXuan Zhou PetscFunctionBegin; 7705262d616SXuan Zhou elem::Zero(*a->emat); 7715262d616SXuan Zhou PetscFunctionReturn(0); 7725262d616SXuan Zhou } 7735262d616SXuan Zhou 774e09a3074SHong Zhang #undef __FUNCT__ 775db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental" 776db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols) 777db31f6deSJed Brown { 778db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 779db31f6deSJed Brown PetscErrorCode ierr; 780db31f6deSJed Brown PetscInt i,m,shift,stride,*idx; 781db31f6deSJed Brown 782db31f6deSJed Brown PetscFunctionBegin; 783db31f6deSJed Brown if (rows) { 784db31f6deSJed Brown m = a->emat->LocalHeight(); 785db31f6deSJed Brown shift = a->emat->ColShift(); 786db31f6deSJed Brown stride = a->emat->ColStride(); 787785e854fSJed Brown ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr); 788db31f6deSJed Brown for (i=0; i<m; i++) { 789db31f6deSJed Brown PetscInt rank,offset; 790db31f6deSJed Brown E2RO(A,0,shift+i*stride,&rank,&offset); 791db31f6deSJed Brown RO2P(A,0,rank,offset,&idx[i]); 792db31f6deSJed Brown } 793db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr); 794db31f6deSJed Brown } 795db31f6deSJed Brown if (cols) { 796db31f6deSJed Brown m = a->emat->LocalWidth(); 797db31f6deSJed Brown shift = a->emat->RowShift(); 798db31f6deSJed Brown stride = a->emat->RowStride(); 799785e854fSJed Brown ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr); 800db31f6deSJed Brown for (i=0; i<m; i++) { 801db31f6deSJed Brown PetscInt rank,offset; 802db31f6deSJed Brown E2RO(A,1,shift+i*stride,&rank,&offset); 803db31f6deSJed Brown RO2P(A,1,rank,offset,&idx[i]); 804db31f6deSJed Brown } 805db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr); 806db31f6deSJed Brown } 807db31f6deSJed Brown PetscFunctionReturn(0); 808db31f6deSJed Brown } 809db31f6deSJed Brown 810db31f6deSJed Brown #undef __FUNCT__ 8112ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense" 81219fd82e9SBarry Smith static PetscErrorCode MatConvert_Elemental_Dense(Mat A,MatType newtype,MatReuse reuse,Mat *B) 813af295397SXuan Zhou { 8142ef0cf24SXuan Zhou Mat Bmpi; 815af295397SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 816ce94432eSBarry Smith MPI_Comm comm; 8172ef0cf24SXuan Zhou PetscErrorCode ierr; 8182ef0cf24SXuan Zhou PetscInt rrank,ridx,crank,cidx,nrows,ncols,i,j; 819df311e6cSXuan Zhou PetscElemScalar v; 820573b0fb4SBarry Smith PetscBool s1,s2,s3; 821af295397SXuan Zhou 822af295397SXuan Zhou PetscFunctionBegin; 823ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 824573b0fb4SBarry Smith ierr = PetscStrcmp(newtype,MATDENSE,&s1);CHKERRQ(ierr); 825573b0fb4SBarry Smith ierr = PetscStrcmp(newtype,MATSEQDENSE,&s2);CHKERRQ(ierr); 826573b0fb4SBarry Smith ierr = PetscStrcmp(newtype,MATMPIDENSE,&s3);CHKERRQ(ierr); 827573b0fb4SBarry Smith if (!s1 && !s2 && !s3) SETERRQ(comm,PETSC_ERR_SUP,"Unsupported New MatType: must be MATDENSE, MATSEQDENSE or MATMPIDENSE"); 828af295397SXuan Zhou ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr); 829af295397SXuan Zhou ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 8302ef0cf24SXuan Zhou ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr); 831af295397SXuan Zhou ierr = MatSetUp(Bmpi);CHKERRQ(ierr); 8322ef0cf24SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 8332ef0cf24SXuan Zhou for (i=0; i<nrows; i++) { 8342ef0cf24SXuan Zhou PetscInt erow,ecol; 8352ef0cf24SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 8362ef0cf24SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 8372ef0cf24SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 8382ef0cf24SXuan Zhou for (j=0; j<ncols; j++) { 8392ef0cf24SXuan Zhou P2RO(A,1,j,&crank,&cidx); 8402ef0cf24SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 8412ef0cf24SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 8422ef0cf24SXuan Zhou v = a->emat->Get(erow,ecol); 843e6dea9dbSXuan Zhou ierr = MatSetValues(Bmpi,1,&i,1,&j,(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 8442ef0cf24SXuan Zhou } 8452ef0cf24SXuan Zhou } 846af295397SXuan Zhou ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 847af295397SXuan Zhou ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 848c4ad791aSXuan Zhou if (reuse == MAT_REUSE_MATRIX) { 849c4ad791aSXuan Zhou ierr = MatHeaderReplace(A,Bmpi);CHKERRQ(ierr); 850c4ad791aSXuan Zhou } else { 851c4ad791aSXuan Zhou *B = Bmpi; 852c4ad791aSXuan Zhou } 853af295397SXuan Zhou PetscFunctionReturn(0); 854af295397SXuan Zhou } 855af295397SXuan Zhou 856af295397SXuan Zhou #undef __FUNCT__ 857db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental" 858db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A) 859db31f6deSJed Brown { 860db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 861db31f6deSJed Brown PetscErrorCode ierr; 8625e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 8635e9f5b67SHong Zhang PetscBool flg; 8645e9f5b67SHong Zhang MPI_Comm icomm; 865db31f6deSJed Brown 866db31f6deSJed Brown PetscFunctionBegin; 867c1ee1e62SHong Zhang a->interface->Detach(); 868aae2c449SHong Zhang delete a->interface; 869aae2c449SHong Zhang delete a->esubmat; 870db31f6deSJed Brown delete a->emat; 8715e9f5b67SHong Zhang 872ce94432eSBarry Smith elem::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A)); 8730c18141cSBarry Smith ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr); 8745e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 8755e9f5b67SHong Zhang if (--commgrid->grid_refct == 0) { 8765e9f5b67SHong Zhang delete commgrid->grid; 8775e9f5b67SHong Zhang ierr = PetscFree(commgrid);CHKERRQ(ierr); 8785e9f5b67SHong Zhang } 8795e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 880bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",NULL);CHKERRQ(ierr); 881bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetFactor_petsc_C",NULL);CHKERRQ(ierr); 882bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSolverPackage_C",NULL);CHKERRQ(ierr); 883db31f6deSJed Brown ierr = PetscFree(A->data);CHKERRQ(ierr); 884db31f6deSJed Brown PetscFunctionReturn(0); 885db31f6deSJed Brown } 886db31f6deSJed Brown 887db31f6deSJed Brown #undef __FUNCT__ 888db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental" 889db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A) 890db31f6deSJed Brown { 891db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 892db31f6deSJed Brown PetscErrorCode ierr; 893db31f6deSJed Brown PetscMPIInt rsize,csize; 894db31f6deSJed Brown 895db31f6deSJed Brown PetscFunctionBegin; 896db31f6deSJed Brown ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 897db31f6deSJed Brown ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 898db31f6deSJed Brown 899efb79153SJack Poulson a->emat->Resize(A->rmap->N,A->cmap->N);CHKERRQ(ierr); 900db31f6deSJed Brown elem::Zero(*a->emat); 901db31f6deSJed Brown 902db31f6deSJed Brown ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr); 903db31f6deSJed Brown ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr); 904ce94432eSBarry Smith if (csize != rsize) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes"); 905db31f6deSJed Brown a->commsize = rsize; 906db31f6deSJed Brown a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize; 907db31f6deSJed Brown a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize; 908db31f6deSJed Brown a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize); 909db31f6deSJed Brown a->m[1] = A->cmap->N / csize + (a->mr[1] != csize); 910db31f6deSJed Brown PetscFunctionReturn(0); 911db31f6deSJed Brown } 912db31f6deSJed Brown 913aae2c449SHong Zhang #undef __FUNCT__ 914aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental" 915aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type) 916aae2c449SHong Zhang { 917aae2c449SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 918aae2c449SHong Zhang 919aae2c449SHong Zhang PetscFunctionBegin; 920aae2c449SHong Zhang a->interface->Detach(); 921aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 922aae2c449SHong Zhang PetscFunctionReturn(0); 923aae2c449SHong Zhang } 924aae2c449SHong Zhang 925aae2c449SHong Zhang #undef __FUNCT__ 926aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental" 927aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type) 928aae2c449SHong Zhang { 929aae2c449SHong Zhang PetscFunctionBegin; 930aae2c449SHong Zhang /* Currently does nothing */ 931aae2c449SHong Zhang PetscFunctionReturn(0); 932aae2c449SHong Zhang } 933aae2c449SHong Zhang 93440d92e34SHong Zhang /* -------------------------------------------------------------------*/ 93540d92e34SHong Zhang static struct _MatOps MatOps_Values = { 93640d92e34SHong Zhang MatSetValues_Elemental, 93740d92e34SHong Zhang 0, 93840d92e34SHong Zhang 0, 93940d92e34SHong Zhang MatMult_Elemental, 94040d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental, 9419426833fSXuan Zhou MatMultTranspose_Elemental, 942e883f9d5SXuan Zhou MatMultTransposeAdd_Elemental, 94340d92e34SHong Zhang MatSolve_Elemental, 944df311e6cSXuan Zhou MatSolveAdd_Elemental, 94540d92e34SHong Zhang 0, //MatSolveTranspose_Elemental, 94640d92e34SHong Zhang /*10*/ 0, //MatSolveTransposeAdd_Elemental, 94740d92e34SHong Zhang MatLUFactor_Elemental, 94840d92e34SHong Zhang MatCholeskyFactor_Elemental, 94940d92e34SHong Zhang 0, 95040d92e34SHong Zhang MatTranspose_Elemental, 95140d92e34SHong Zhang /*15*/ MatGetInfo_Elemental, 95240d92e34SHong Zhang 0, 95361119200SXuan Zhou MatGetDiagonal_Elemental, 954ade3cc5eSXuan Zhou MatDiagonalScale_Elemental, 95540d92e34SHong Zhang MatNorm_Elemental, 95640d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental, 95740d92e34SHong Zhang MatAssemblyEnd_Elemental, 95840d92e34SHong Zhang 0, //MatSetOption_Elemental, 95940d92e34SHong Zhang MatZeroEntries_Elemental, 96040d92e34SHong Zhang /*24*/ 0, 96140d92e34SHong Zhang MatLUFactorSymbolic_Elemental, 96240d92e34SHong Zhang MatLUFactorNumeric_Elemental, 96340d92e34SHong Zhang MatCholeskyFactorSymbolic_Elemental, 96440d92e34SHong Zhang MatCholeskyFactorNumeric_Elemental, 96540d92e34SHong Zhang /*29*/ MatSetUp_Elemental, 96640d92e34SHong Zhang 0, 96740d92e34SHong Zhang 0, 96840d92e34SHong Zhang 0, 96940d92e34SHong Zhang 0, 970df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental, 97140d92e34SHong Zhang 0, 97240d92e34SHong Zhang 0, 97340d92e34SHong Zhang 0, 97440d92e34SHong Zhang 0, 97540d92e34SHong Zhang /*39*/ MatAXPY_Elemental, 97640d92e34SHong Zhang 0, 97740d92e34SHong Zhang 0, 97840d92e34SHong Zhang 0, 97940d92e34SHong Zhang MatCopy_Elemental, 98040d92e34SHong Zhang /*44*/ 0, 98140d92e34SHong Zhang MatScale_Elemental, 98240d92e34SHong Zhang 0, 98340d92e34SHong Zhang 0, 98440d92e34SHong Zhang 0, 98540d92e34SHong Zhang /*49*/ 0, 98640d92e34SHong Zhang 0, 98740d92e34SHong Zhang 0, 98840d92e34SHong Zhang 0, 98940d92e34SHong Zhang 0, 99040d92e34SHong Zhang /*54*/ 0, 99140d92e34SHong Zhang 0, 99240d92e34SHong Zhang 0, 99340d92e34SHong Zhang 0, 99440d92e34SHong Zhang 0, 99540d92e34SHong Zhang /*59*/ 0, 99640d92e34SHong Zhang MatDestroy_Elemental, 99740d92e34SHong Zhang MatView_Elemental, 99840d92e34SHong Zhang 0, 99940d92e34SHong Zhang 0, 100040d92e34SHong Zhang /*64*/ 0, 100140d92e34SHong Zhang 0, 100240d92e34SHong Zhang 0, 100340d92e34SHong Zhang 0, 100440d92e34SHong Zhang 0, 100540d92e34SHong Zhang /*69*/ 0, 100640d92e34SHong Zhang 0, 10072ef0cf24SXuan Zhou MatConvert_Elemental_Dense, 100840d92e34SHong Zhang 0, 100940d92e34SHong Zhang 0, 101040d92e34SHong Zhang /*74*/ 0, 101140d92e34SHong Zhang 0, 101240d92e34SHong Zhang 0, 101340d92e34SHong Zhang 0, 101440d92e34SHong Zhang 0, 101540d92e34SHong Zhang /*79*/ 0, 101640d92e34SHong Zhang 0, 101740d92e34SHong Zhang 0, 101840d92e34SHong Zhang 0, 101940d92e34SHong Zhang 0, 102040d92e34SHong Zhang /*84*/ 0, 102140d92e34SHong Zhang 0, 102240d92e34SHong Zhang 0, 102340d92e34SHong Zhang 0, 102440d92e34SHong Zhang 0, 102540d92e34SHong Zhang /*89*/ MatMatMult_Elemental, 102640d92e34SHong Zhang MatMatMultSymbolic_Elemental, 102740d92e34SHong Zhang MatMatMultNumeric_Elemental, 102840d92e34SHong Zhang 0, 102940d92e34SHong Zhang 0, 103040d92e34SHong Zhang /*94*/ 0, 1031df311e6cSXuan Zhou MatMatTransposeMult_Elemental, 1032df311e6cSXuan Zhou MatMatTransposeMultSymbolic_Elemental, 1033df311e6cSXuan Zhou MatMatTransposeMultNumeric_Elemental, 103440d92e34SHong Zhang 0, 103540d92e34SHong Zhang /*99*/ 0, 103640d92e34SHong Zhang 0, 103740d92e34SHong Zhang 0, 1038dfcb0403SXuan Zhou MatConjugate_Elemental, 103940d92e34SHong Zhang 0, 104040d92e34SHong Zhang /*104*/0, 104140d92e34SHong Zhang 0, 104240d92e34SHong Zhang 0, 104340d92e34SHong Zhang 0, 104440d92e34SHong Zhang 0, 104540d92e34SHong Zhang /*109*/MatMatSolve_Elemental, 104640d92e34SHong Zhang 0, 104740d92e34SHong Zhang 0, 104840d92e34SHong Zhang 0, 104940d92e34SHong Zhang 0, 105040d92e34SHong Zhang /*114*/0, 105140d92e34SHong Zhang 0, 105240d92e34SHong Zhang 0, 105340d92e34SHong Zhang 0, 105440d92e34SHong Zhang 0, 105540d92e34SHong Zhang /*119*/0, 10564a29722dSXuan Zhou MatHermitianTranspose_Elemental, 105740d92e34SHong Zhang 0, 105840d92e34SHong Zhang 0, 105940d92e34SHong Zhang 0, 106040d92e34SHong Zhang /*124*/0, 106140d92e34SHong Zhang 0, 106240d92e34SHong Zhang 0, 106340d92e34SHong Zhang 0, 106440d92e34SHong Zhang 0, 106540d92e34SHong Zhang /*129*/0, 106640d92e34SHong Zhang 0, 106740d92e34SHong Zhang 0, 106840d92e34SHong Zhang 0, 106940d92e34SHong Zhang 0, 107040d92e34SHong Zhang /*134*/0, 107140d92e34SHong Zhang 0, 107240d92e34SHong Zhang 0, 107340d92e34SHong Zhang 0, 107440d92e34SHong Zhang 0 107540d92e34SHong Zhang }; 107640d92e34SHong Zhang 1077ed36708cSHong Zhang /*MC 1078ed36708cSHong Zhang MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package 1079ed36708cSHong Zhang 1080ed36708cSHong Zhang Options Database Keys: 10815cc86fc1SJed Brown + -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions() 10825cc86fc1SJed Brown - -mat_elemental_grid_height - sets Grid Height for 2D cyclic ordering of internal matrix 1083ed36708cSHong Zhang 1084ed36708cSHong Zhang Level: beginner 1085ed36708cSHong Zhang 10865cb544a0SHong Zhang .seealso: MATDENSE 1087ed36708cSHong Zhang M*/ 10884a29722dSXuan Zhou 1089db31f6deSJed Brown #undef __FUNCT__ 1090db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental" 10918cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_Elemental(Mat A) 1092db31f6deSJed Brown { 1093db31f6deSJed Brown Mat_Elemental *a; 1094db31f6deSJed Brown PetscErrorCode ierr; 10955682a260SJack Poulson PetscBool flg,flg1; 10965e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 10975e9f5b67SHong Zhang MPI_Comm icomm; 10985682a260SJack Poulson PetscInt optv1; 1099db31f6deSJed Brown 1100db31f6deSJed Brown PetscFunctionBegin; 1101607a6623SBarry Smith ierr = PetscElementalInitializePackage();CHKERRQ(ierr); 110240d92e34SHong Zhang ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 110340d92e34SHong Zhang A->insertmode = NOT_SET_VALUES; 1104db31f6deSJed Brown 1105b00a9115SJed Brown ierr = PetscNewLog(A,&a);CHKERRQ(ierr); 1106db31f6deSJed Brown A->data = (void*)a; 1107db31f6deSJed Brown 1108db31f6deSJed Brown /* Set up the elemental matrix */ 1109ce94432eSBarry Smith elem::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A)); 11105e9f5b67SHong Zhang 11115e9f5b67SHong Zhang /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */ 11125e9f5b67SHong Zhang if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) { 1113180a43e4SHong Zhang ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); 11145e9f5b67SHong Zhang } 11150c18141cSBarry Smith ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr); 11165e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 11175e9f5b67SHong Zhang if (!flg) { 1118b00a9115SJed Brown ierr = PetscNewLog(A,&commgrid);CHKERRQ(ierr); 11195cb544a0SHong Zhang 1120ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 11215cb544a0SHong Zhang /* displayed default grid sizes (CommSize,1) are set by us arbitrarily until elem::Grid() is called */ 11220c18141cSBarry Smith ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",elem::mpi::Size(cxxcomm),&optv1,&flg1);CHKERRQ(ierr); 11235682a260SJack Poulson if (flg1) { 11240c18141cSBarry Smith if (elem::mpi::Size(cxxcomm) % optv1 != 0) { 11250c18141cSBarry Smith SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Grid Height %D must evenly divide CommSize %D",optv1,(PetscInt)elem::mpi::Size(cxxcomm)); 1126ed667823SXuan Zhou } 11275682a260SJack Poulson commgrid->grid = new elem::Grid(cxxcomm,optv1); /* use user-provided grid height */ 11282ef0cf24SXuan Zhou } else { 11292adf0be3SHong Zhang commgrid->grid = new elem::Grid(cxxcomm); /* use Elemental default grid sizes */ 1130ed667823SXuan Zhou } 11315e9f5b67SHong Zhang commgrid->grid_refct = 1; 11325e9f5b67SHong Zhang ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr); 11335cb544a0SHong Zhang PetscOptionsEnd(); 11345e9f5b67SHong Zhang } else { 11355e9f5b67SHong Zhang commgrid->grid_refct++; 11365e9f5b67SHong Zhang } 11375e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 11385e9f5b67SHong Zhang a->grid = commgrid->grid; 1139df311e6cSXuan Zhou a->emat = new elem::DistMatrix<PetscElemScalar>(*a->grid); 1140df311e6cSXuan Zhou a->esubmat = new elem::Matrix<PetscElemScalar>(1,1); 1141df311e6cSXuan Zhou a->interface = new elem::AxpyInterface<PetscElemScalar>; 11427c920d81SXuan Zhou a->pivot = new elem::DistMatrix<PetscInt,elem::VC,elem::STAR>; 1143db31f6deSJed Brown 1144db31f6deSJed Brown /* build cache for off array entries formed */ 1145aae2c449SHong Zhang a->interface->Attach(elem::LOCAL_TO_GLOBAL,*(a->emat)); 1146bafd5131SHong Zhang 1147bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",MatGetOwnershipIS_Elemental);CHKERRQ(ierr); 1148bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetFactor_elemental_C",MatGetFactor_elemental_elemental);CHKERRQ(ierr); 1149db31f6deSJed Brown 1150db31f6deSJed Brown ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr); 1151db31f6deSJed Brown PetscFunctionReturn(0); 1152db31f6deSJed Brown } 11534a29722dSXuan Zhou 1154