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; 25e8146892SSatish Balay if (El::Initialized()) PetscFunctionReturn(0); 26e05a9d59SSatish Balay El::Initialize(); /* called by the 1st call of MatCreate_Elemental */ 27db31f6deSJed Brown ierr = PetscRegisterFinalize(PetscElementalFinalizePackage);CHKERRQ(ierr); 28db31f6deSJed Brown PetscFunctionReturn(0); 29db31f6deSJed Brown } 30db31f6deSJed Brown 31db31f6deSJed Brown #undef __FUNCT__ 32db31f6deSJed Brown #define __FUNCT__ "PetscElementalFinalizePackage" 33db31f6deSJed Brown /*@C 34db31f6deSJed Brown PetscElementalFinalizePackage - Finalize Elemental package 35db31f6deSJed Brown 36db31f6deSJed Brown Logically Collective 37db31f6deSJed Brown 38db31f6deSJed Brown Level: developer 39db31f6deSJed Brown 40db31f6deSJed Brown .seealso: MATELEMENTAL, PetscElementalInitializePackage() 41db31f6deSJed Brown @*/ 42db31f6deSJed Brown PetscErrorCode PetscElementalFinalizePackage(void) 43db31f6deSJed Brown { 44db31f6deSJed Brown PetscFunctionBegin; 45e8146892SSatish Balay El::Finalize(); /* called by PetscFinalize() */ 46db31f6deSJed Brown PetscFunctionReturn(0); 47db31f6deSJed Brown } 48db31f6deSJed Brown 49db31f6deSJed Brown #undef __FUNCT__ 50db31f6deSJed Brown #define __FUNCT__ "MatView_Elemental" 51db31f6deSJed Brown static PetscErrorCode MatView_Elemental(Mat A,PetscViewer viewer) 52db31f6deSJed Brown { 53db31f6deSJed Brown PetscErrorCode ierr; 54db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 55db31f6deSJed Brown PetscBool iascii; 56db31f6deSJed Brown 57db31f6deSJed Brown PetscFunctionBegin; 58db31f6deSJed Brown ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 59db31f6deSJed Brown if (iascii) { 60db31f6deSJed Brown PetscViewerFormat format; 61db31f6deSJed Brown ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 62db31f6deSJed Brown if (format == PETSC_VIEWER_ASCII_INFO) { 6379673f7bSHong Zhang /* call elemental viewing function */ 642d8adcc7SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"Elemental run parameters:\n");CHKERRQ(ierr); 65ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," allocated entries=%d\n",(*a->emat).AllocatedMemory());CHKERRQ(ierr); 66ed667823SXuan Zhou ierr = PetscViewerASCIIPrintf(viewer," grid height=%d, grid width=%d\n",(*a->emat).Grid().Height(),(*a->emat).Grid().Width());CHKERRQ(ierr); 674fe7bbcaSHong Zhang if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 6879673f7bSHong Zhang /* call elemental viewing function */ 69ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"test matview_elemental 2\n");CHKERRQ(ierr); 704fe7bbcaSHong Zhang } 7179673f7bSHong Zhang 72db31f6deSJed Brown } else if (format == PETSC_VIEWER_DEFAULT) { 73db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 74e8146892SSatish Balay El::Print( *a->emat, "Elemental matrix (cyclic ordering)" ); 75db31f6deSJed Brown ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 76834d3fecSHong Zhang if (A->factortype == MAT_FACTOR_NONE){ 7761119200SXuan Zhou Mat Adense; 78ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 7961119200SXuan Zhou ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr); 8061119200SXuan Zhou ierr = MatView(Adense,viewer);CHKERRQ(ierr); 8161119200SXuan Zhou ierr = MatDestroy(&Adense);CHKERRQ(ierr); 82834d3fecSHong Zhang } 83ce94432eSBarry Smith } else SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Format"); 84d2daa67eSHong Zhang } else { 855cb544a0SHong Zhang /* convert to dense format and call MatView() */ 8661119200SXuan Zhou Mat Adense; 87ce94432eSBarry Smith ierr = PetscPrintf(PetscObjectComm((PetscObject)viewer),"Elemental matrix (explicit ordering)\n");CHKERRQ(ierr); 8861119200SXuan Zhou ierr = MatConvert(A,MATDENSE,MAT_INITIAL_MATRIX,&Adense);CHKERRQ(ierr); 8961119200SXuan Zhou ierr = MatView(Adense,viewer);CHKERRQ(ierr); 9061119200SXuan Zhou ierr = MatDestroy(&Adense);CHKERRQ(ierr); 91d2daa67eSHong Zhang } 92db31f6deSJed Brown PetscFunctionReturn(0); 93db31f6deSJed Brown } 94db31f6deSJed Brown 95db31f6deSJed Brown #undef __FUNCT__ 96180a43e4SHong Zhang #define __FUNCT__ "MatGetInfo_Elemental" 9715767789SHong Zhang static PetscErrorCode MatGetInfo_Elemental(Mat A,MatInfoType flag,MatInfo *info) 98180a43e4SHong Zhang { 9915767789SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 10015767789SHong Zhang 101180a43e4SHong Zhang PetscFunctionBegin; 1025cb544a0SHong Zhang info->block_size = 1.0; 10315767789SHong Zhang 10415767789SHong Zhang if (flag == MAT_LOCAL) { 10515767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 10615767789SHong Zhang info->nz_used = info->nz_allocated; 10715767789SHong Zhang } else if (flag == MAT_GLOBAL_MAX) { 108b2566f29SBarry Smith //ierr = MPIU_Allreduce(isend,irecv,5,MPIU_REAL,MPIU_MAX,PetscObjectComm((PetscObject)matin));CHKERRQ(ierr); 10915767789SHong Zhang /* see MatGetInfo_MPIAIJ() for getting global info->nz_allocated! */ 11015767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_MAX not written yet"); 11115767789SHong Zhang } else if (flag == MAT_GLOBAL_SUM) { 11215767789SHong Zhang //SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP," MAT_GLOBAL_SUM not written yet"); 11315767789SHong Zhang info->nz_allocated = (double)(*a->emat).AllocatedMemory(); /* locally allocated */ 11415767789SHong Zhang info->nz_used = info->nz_allocated; /* assume Elemental does accurate allocation */ 115b2566f29SBarry Smith //ierr = MPIU_Allreduce(isend,irecv,1,MPIU_REAL,MPIU_SUM,PetscObjectComm((PetscObject)A));CHKERRQ(ierr); 11615767789SHong Zhang //PetscPrintf(PETSC_COMM_SELF," ... [%d] locally allocated %g\n",rank,info->nz_allocated); 11715767789SHong Zhang } 11815767789SHong Zhang 11915767789SHong Zhang info->nz_unneeded = 0.0; 12015767789SHong Zhang info->assemblies = (double)A->num_ass; 12115767789SHong Zhang info->mallocs = 0; 12215767789SHong Zhang info->memory = ((PetscObject)A)->mem; 12315767789SHong Zhang info->fill_ratio_given = 0; /* determined by Elemental */ 12415767789SHong Zhang info->fill_ratio_needed = 0; 12515767789SHong Zhang info->factor_mallocs = 0; 126180a43e4SHong Zhang PetscFunctionReturn(0); 127180a43e4SHong Zhang } 128180a43e4SHong Zhang 129180a43e4SHong Zhang #undef __FUNCT__ 13032d7a744SHong Zhang #define __FUNCT__ "MatSetOption_Elemental" 13132d7a744SHong Zhang PetscErrorCode MatSetOption_Elemental(Mat A,MatOption op,PetscBool flg) 13232d7a744SHong Zhang { 13332d7a744SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 13432d7a744SHong Zhang 13532d7a744SHong Zhang PetscFunctionBegin; 13632d7a744SHong Zhang switch (op) { 13732d7a744SHong Zhang case MAT_NEW_NONZERO_LOCATIONS: 13832d7a744SHong Zhang case MAT_NEW_NONZERO_LOCATION_ERR: 13932d7a744SHong Zhang case MAT_NEW_NONZERO_ALLOCATION_ERR: 14032d7a744SHong Zhang case MAT_ROW_ORIENTED: 14132d7a744SHong Zhang a->roworiented = flg; 14232d7a744SHong Zhang break; 143*5a8bcd62SHong Zhang case MAT_SYMMETRIC: 144*5a8bcd62SHong Zhang break; 14532d7a744SHong Zhang default: 14632d7a744SHong Zhang SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %s",MatOptions[op]); 14732d7a744SHong Zhang } 14832d7a744SHong Zhang PetscFunctionReturn(0); 14932d7a744SHong Zhang } 15032d7a744SHong Zhang 15132d7a744SHong Zhang #undef __FUNCT__ 152db31f6deSJed Brown #define __FUNCT__ "MatSetValues_Elemental" 153e6dea9dbSXuan Zhou static PetscErrorCode MatSetValues_Elemental(Mat A,PetscInt nr,const PetscInt *rows,PetscInt nc,const PetscInt *cols,const PetscScalar *vals,InsertMode imode) 154db31f6deSJed Brown { 155db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 156fbbcb730SJack Poulson PetscInt i,j,rrank,ridx,crank,cidx,erow,ecol,numQueues=0; 157db31f6deSJed Brown 158db31f6deSJed Brown PetscFunctionBegin; 159fbbcb730SJack Poulson // TODO: Initialize matrix to all zeros? 160fbbcb730SJack Poulson 161fbbcb730SJack Poulson // Count the number of queues from this process 16232d7a744SHong Zhang if (a->roworiented) { 163db31f6deSJed Brown for (i=0; i<nr; i++) { 164db31f6deSJed Brown if (rows[i] < 0) continue; 165db31f6deSJed Brown P2RO(A,0,rows[i],&rrank,&ridx); 166db31f6deSJed Brown RO2E(A,0,rrank,ridx,&erow); 167ce94432eSBarry Smith if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation"); 168db31f6deSJed Brown for (j=0; j<nc; j++) { 169db31f6deSJed Brown if (cols[j] < 0) continue; 170db31f6deSJed Brown P2RO(A,1,cols[j],&crank,&cidx); 171db31f6deSJed Brown RO2E(A,1,crank,cidx,&ecol); 172ce94432eSBarry Smith if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation"); 173fbbcb730SJack Poulson if (!a->emat->IsLocal(erow,ecol) ){ /* off-proc entry */ 174fbbcb730SJack Poulson /* printf("Will later remotely update (%d,%d)\n",erow,ecol); */ 175aae2c449SHong Zhang if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 176fbbcb730SJack Poulson ++numQueues; 177aae2c449SHong Zhang continue; 178ed36708cSHong Zhang } 179fbbcb730SJack Poulson /* printf("Locally updating (%d,%d)\n",erow,ecol); */ 180db31f6deSJed Brown switch (imode) { 181fbbcb730SJack Poulson case INSERT_VALUES: a->emat->Set(erow,ecol,(PetscElemScalar)vals[i*nc+j]); break; 182fbbcb730SJack Poulson case ADD_VALUES: a->emat->Update(erow,ecol,(PetscElemScalar)vals[i*nc+j]); break; 183ce94432eSBarry Smith default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 184db31f6deSJed Brown } 185db31f6deSJed Brown } 186db31f6deSJed Brown } 187fbbcb730SJack Poulson 188fbbcb730SJack Poulson /* printf("numQueues=%d\n",numQueues); */ 189fbbcb730SJack Poulson a->emat->Reserve( numQueues ); 190fbbcb730SJack Poulson for (i=0; i<nr; i++) { 191fbbcb730SJack Poulson if (rows[i] < 0) continue; 192fbbcb730SJack Poulson P2RO(A,0,rows[i],&rrank,&ridx); 193fbbcb730SJack Poulson RO2E(A,0,rrank,ridx,&erow); 194fbbcb730SJack Poulson for (j=0; j<nc; j++) { 195fbbcb730SJack Poulson if (cols[j] < 0) continue; 196fbbcb730SJack Poulson P2RO(A,1,cols[j],&crank,&cidx); 197fbbcb730SJack Poulson RO2E(A,1,crank,cidx,&ecol); 198fbbcb730SJack Poulson if ( !a->emat->IsLocal(erow,ecol) ) { /*off-proc entry*/ 199fbbcb730SJack Poulson /* printf("Queueing remotely update of (%d,%d)\n",erow,ecol); */ 200fbbcb730SJack Poulson a->emat->QueueUpdate( erow, ecol, vals[i*nc+j] ); 201fbbcb730SJack Poulson } 202fbbcb730SJack Poulson } 203fbbcb730SJack Poulson } 20432d7a744SHong Zhang } else { /* columnoriented */ 20532d7a744SHong Zhang for (j=0; j<nc; j++) { 20632d7a744SHong Zhang if (cols[j] < 0) continue; 20732d7a744SHong Zhang P2RO(A,1,cols[j],&crank,&cidx); 20832d7a744SHong Zhang RO2E(A,1,crank,cidx,&ecol); 20932d7a744SHong Zhang if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect col translation"); 21032d7a744SHong Zhang for (i=0; i<nr; i++) { 21132d7a744SHong Zhang if (rows[i] < 0) continue; 21232d7a744SHong Zhang P2RO(A,0,rows[i],&rrank,&ridx); 21332d7a744SHong Zhang RO2E(A,0,rrank,ridx,&erow); 21432d7a744SHong Zhang if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_PLIB,"Incorrect row translation"); 21532d7a744SHong Zhang if (!a->emat->IsLocal(erow,ecol) ){ /* off-proc entry */ 21632d7a744SHong Zhang /* printf("Will later remotely update (%d,%d)\n",erow,ecol); */ 21732d7a744SHong Zhang if (imode != ADD_VALUES) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only ADD_VALUES to off-processor entry is supported"); 21832d7a744SHong Zhang ++numQueues; 21932d7a744SHong Zhang continue; 22032d7a744SHong Zhang } 22132d7a744SHong Zhang /* printf("Locally updating (%d,%d)\n",erow,ecol); */ 22232d7a744SHong Zhang switch (imode) { 22332d7a744SHong Zhang case INSERT_VALUES: a->emat->Set(erow,ecol,(PetscElemScalar)vals[i+j*nr]); break; 22432d7a744SHong Zhang case ADD_VALUES: a->emat->Update(erow,ecol,(PetscElemScalar)vals[i+j*nr]); break; 22532d7a744SHong Zhang default: SETERRQ1(PetscObjectComm((PetscObject)A),PETSC_ERR_SUP,"No support for InsertMode %d",(int)imode); 22632d7a744SHong Zhang } 22732d7a744SHong Zhang } 22832d7a744SHong Zhang } 22932d7a744SHong Zhang 23032d7a744SHong Zhang /* printf("numQueues=%d\n",numQueues); */ 23132d7a744SHong Zhang a->emat->Reserve( numQueues ); 23232d7a744SHong Zhang for (j=0; j<nc; j++) { 23332d7a744SHong Zhang if (cols[j] < 0) continue; 23432d7a744SHong Zhang P2RO(A,1,cols[j],&crank,&cidx); 23532d7a744SHong Zhang RO2E(A,1,crank,cidx,&ecol); 23632d7a744SHong Zhang 23732d7a744SHong Zhang for (i=0; i<nr; i++) { 23832d7a744SHong Zhang if (rows[i] < 0) continue; 23932d7a744SHong Zhang P2RO(A,0,rows[i],&rrank,&ridx); 24032d7a744SHong Zhang RO2E(A,0,rrank,ridx,&erow); 24132d7a744SHong Zhang if ( !a->emat->IsLocal(erow,ecol) ) { /*off-proc entry*/ 24232d7a744SHong Zhang /* printf("Queueing remotely update of (%d,%d)\n",erow,ecol); */ 24332d7a744SHong Zhang a->emat->QueueUpdate( erow, ecol, vals[i+j*nr] ); 24432d7a744SHong Zhang } 24532d7a744SHong Zhang } 24632d7a744SHong Zhang } 24732d7a744SHong Zhang } 248db31f6deSJed Brown PetscFunctionReturn(0); 249db31f6deSJed Brown } 250db31f6deSJed Brown 251db31f6deSJed Brown #undef __FUNCT__ 252db31f6deSJed Brown #define __FUNCT__ "MatMult_Elemental" 253db31f6deSJed Brown static PetscErrorCode MatMult_Elemental(Mat A,Vec X,Vec Y) 254db31f6deSJed Brown { 255db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 256db31f6deSJed Brown PetscErrorCode ierr; 257e6dea9dbSXuan Zhou const PetscElemScalar *x; 258e6dea9dbSXuan Zhou PetscElemScalar *y; 259df311e6cSXuan Zhou PetscElemScalar one = 1,zero = 0; 260db31f6deSJed Brown 261db31f6deSJed Brown PetscFunctionBegin; 262e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 263e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 264db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 265e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ye; 2660c18141cSBarry Smith xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n); 2670c18141cSBarry Smith ye.Attach(A->rmap->N,1,*a->grid,0,0,y,A->rmap->n); 268e8146892SSatish Balay El::Gemv(El::NORMAL,one,*a->emat,xe,zero,ye); 269db31f6deSJed Brown } 270e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 271e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 272db31f6deSJed Brown PetscFunctionReturn(0); 273db31f6deSJed Brown } 274db31f6deSJed Brown 275db31f6deSJed Brown #undef __FUNCT__ 2769426833fSXuan Zhou #define __FUNCT__ "MatMultTranspose_Elemental" 2779426833fSXuan Zhou static PetscErrorCode MatMultTranspose_Elemental(Mat A,Vec X,Vec Y) 2789426833fSXuan Zhou { 2799426833fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 2809426833fSXuan Zhou PetscErrorCode ierr; 281df311e6cSXuan Zhou const PetscElemScalar *x; 282df311e6cSXuan Zhou PetscElemScalar *y; 283e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 2849426833fSXuan Zhou 2859426833fSXuan Zhou PetscFunctionBegin; 286e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 287e6dea9dbSXuan Zhou ierr = VecGetArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2889426833fSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 289e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ye; 2900c18141cSBarry Smith xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n); 2910c18141cSBarry Smith ye.Attach(A->cmap->N,1,*a->grid,0,0,y,A->cmap->n); 292e8146892SSatish Balay El::Gemv(El::TRANSPOSE,one,*a->emat,xe,zero,ye); 2939426833fSXuan Zhou } 294e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 295e6dea9dbSXuan Zhou ierr = VecRestoreArray(Y,(PetscScalar **)&y);CHKERRQ(ierr); 2969426833fSXuan Zhou PetscFunctionReturn(0); 2979426833fSXuan Zhou } 2989426833fSXuan Zhou 2999426833fSXuan Zhou #undef __FUNCT__ 300db31f6deSJed Brown #define __FUNCT__ "MatMultAdd_Elemental" 301db31f6deSJed Brown static PetscErrorCode MatMultAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 302db31f6deSJed Brown { 303db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 304db31f6deSJed Brown PetscErrorCode ierr; 305df311e6cSXuan Zhou const PetscElemScalar *x; 306df311e6cSXuan Zhou PetscElemScalar *z; 307e6dea9dbSXuan Zhou PetscElemScalar one = 1; 308db31f6deSJed Brown 309db31f6deSJed Brown PetscFunctionBegin; 310db31f6deSJed Brown if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 311e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 312e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 313db31f6deSJed Brown { /* Scoping so that constructor is called before pointer is returned */ 314e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ze; 3150c18141cSBarry Smith xe.LockedAttach(A->cmap->N,1,*a->grid,0,0,x,A->cmap->n); 3160c18141cSBarry Smith ze.Attach(A->rmap->N,1,*a->grid,0,0,z,A->rmap->n); 317e8146892SSatish Balay El::Gemv(El::NORMAL,one,*a->emat,xe,one,ze); 318db31f6deSJed Brown } 319e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 320e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 321db31f6deSJed Brown PetscFunctionReturn(0); 322db31f6deSJed Brown } 323db31f6deSJed Brown 324db31f6deSJed Brown #undef __FUNCT__ 325e883f9d5SXuan Zhou #define __FUNCT__ "MatMultTransposeAdd_Elemental" 326e883f9d5SXuan Zhou static PetscErrorCode MatMultTransposeAdd_Elemental(Mat A,Vec X,Vec Y,Vec Z) 327e883f9d5SXuan Zhou { 328e883f9d5SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 329e883f9d5SXuan Zhou PetscErrorCode ierr; 330df311e6cSXuan Zhou const PetscElemScalar *x; 331df311e6cSXuan Zhou PetscElemScalar *z; 332e6dea9dbSXuan Zhou PetscElemScalar one = 1; 333e883f9d5SXuan Zhou 334e883f9d5SXuan Zhou PetscFunctionBegin; 335e883f9d5SXuan Zhou if (Y != Z) {ierr = VecCopy(Y,Z);CHKERRQ(ierr);} 336e6dea9dbSXuan Zhou ierr = VecGetArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 337e6dea9dbSXuan Zhou ierr = VecGetArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 338e883f9d5SXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 339e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe, ze; 3400c18141cSBarry Smith xe.LockedAttach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n); 3410c18141cSBarry Smith ze.Attach(A->cmap->N,1,*a->grid,0,0,z,A->cmap->n); 342e8146892SSatish Balay El::Gemv(El::TRANSPOSE,one,*a->emat,xe,one,ze); 343e883f9d5SXuan Zhou } 344e6dea9dbSXuan Zhou ierr = VecRestoreArrayRead(X,(const PetscScalar **)&x);CHKERRQ(ierr); 345e6dea9dbSXuan Zhou ierr = VecRestoreArray(Z,(PetscScalar **)&z);CHKERRQ(ierr); 346e883f9d5SXuan Zhou PetscFunctionReturn(0); 347e883f9d5SXuan Zhou } 348e883f9d5SXuan Zhou 349e883f9d5SXuan Zhou #undef __FUNCT__ 3509a9e8502SHong Zhang #define __FUNCT__ "MatMatMultNumeric_Elemental" 3519a9e8502SHong Zhang static PetscErrorCode MatMatMultNumeric_Elemental(Mat A,Mat B,Mat C) 352c1d1b975SXuan Zhou { 353c1d1b975SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 354c1d1b975SXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 3559a9e8502SHong Zhang Mat_Elemental *c = (Mat_Elemental*)C->data; 356e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 357c1d1b975SXuan Zhou 358c1d1b975SXuan Zhou PetscFunctionBegin; 359aae2c449SHong Zhang { /* Scoping so that constructor is called before pointer is returned */ 360e8146892SSatish Balay El::Gemm(El::NORMAL,El::NORMAL,one,*a->emat,*b->emat,zero,*c->emat); 361aae2c449SHong Zhang } 3629a9e8502SHong Zhang C->assembled = PETSC_TRUE; 3639a9e8502SHong Zhang PetscFunctionReturn(0); 3649a9e8502SHong Zhang } 3659a9e8502SHong Zhang 3669a9e8502SHong Zhang #undef __FUNCT__ 3679a9e8502SHong Zhang #define __FUNCT__ "MatMatMultSymbolic_Elemental" 3689a9e8502SHong Zhang static PetscErrorCode MatMatMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 3699a9e8502SHong Zhang { 3709a9e8502SHong Zhang PetscErrorCode ierr; 3719a9e8502SHong Zhang Mat Ce; 372ce94432eSBarry Smith MPI_Comm comm; 3739a9e8502SHong Zhang 3749a9e8502SHong Zhang PetscFunctionBegin; 375ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 3769a9e8502SHong Zhang ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 3779a9e8502SHong Zhang ierr = MatSetSizes(Ce,A->rmap->n,B->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 3789a9e8502SHong Zhang ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 3799a9e8502SHong Zhang ierr = MatSetUp(Ce);CHKERRQ(ierr); 3809a9e8502SHong Zhang *C = Ce; 3819a9e8502SHong Zhang PetscFunctionReturn(0); 3829a9e8502SHong Zhang } 3839a9e8502SHong Zhang 3849a9e8502SHong Zhang #undef __FUNCT__ 3859a9e8502SHong Zhang #define __FUNCT__ "MatMatMult_Elemental" 3869a9e8502SHong Zhang static PetscErrorCode MatMatMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 3879a9e8502SHong Zhang { 3889a9e8502SHong Zhang PetscErrorCode ierr; 3899a9e8502SHong Zhang 3909a9e8502SHong Zhang PetscFunctionBegin; 3919a9e8502SHong Zhang if (scall == MAT_INITIAL_MATRIX){ 3923ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3939a9e8502SHong Zhang ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 3943ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr); 3959a9e8502SHong Zhang } 3963ff4c91cSHong Zhang ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 3979a9e8502SHong Zhang ierr = MatMatMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 3983ff4c91cSHong Zhang ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr); 399c1d1b975SXuan Zhou PetscFunctionReturn(0); 400c1d1b975SXuan Zhou } 401c1d1b975SXuan Zhou 402c1d1b975SXuan Zhou #undef __FUNCT__ 403df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultNumeric_Elemental" 404df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultNumeric_Elemental(Mat A,Mat B,Mat C) 405df311e6cSXuan Zhou { 406df311e6cSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 407df311e6cSXuan Zhou Mat_Elemental *b = (Mat_Elemental*)B->data; 408df311e6cSXuan Zhou Mat_Elemental *c = (Mat_Elemental*)C->data; 409e6dea9dbSXuan Zhou PetscElemScalar one = 1,zero = 0; 410df311e6cSXuan Zhou 411df311e6cSXuan Zhou PetscFunctionBegin; 412df311e6cSXuan Zhou { /* Scoping so that constructor is called before pointer is returned */ 413e8146892SSatish Balay El::Gemm(El::NORMAL,El::TRANSPOSE,one,*a->emat,*b->emat,zero,*c->emat); 414df311e6cSXuan Zhou } 415df311e6cSXuan Zhou C->assembled = PETSC_TRUE; 416df311e6cSXuan Zhou PetscFunctionReturn(0); 417df311e6cSXuan Zhou } 418df311e6cSXuan Zhou 419df311e6cSXuan Zhou #undef __FUNCT__ 420df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMultSymbolic_Elemental" 421df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMultSymbolic_Elemental(Mat A,Mat B,PetscReal fill,Mat *C) 422df311e6cSXuan Zhou { 423df311e6cSXuan Zhou PetscErrorCode ierr; 424df311e6cSXuan Zhou Mat Ce; 425ce94432eSBarry Smith MPI_Comm comm; 426df311e6cSXuan Zhou 427df311e6cSXuan Zhou PetscFunctionBegin; 428ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 429df311e6cSXuan Zhou ierr = MatCreate(comm,&Ce);CHKERRQ(ierr); 430df311e6cSXuan Zhou ierr = MatSetSizes(Ce,A->rmap->n,B->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 431df311e6cSXuan Zhou ierr = MatSetType(Ce,MATELEMENTAL);CHKERRQ(ierr); 432df311e6cSXuan Zhou ierr = MatSetUp(Ce);CHKERRQ(ierr); 433df311e6cSXuan Zhou *C = Ce; 434df311e6cSXuan Zhou PetscFunctionReturn(0); 435df311e6cSXuan Zhou } 436df311e6cSXuan Zhou 437df311e6cSXuan Zhou #undef __FUNCT__ 438df311e6cSXuan Zhou #define __FUNCT__ "MatMatTransposeMult_Elemental" 439df311e6cSXuan Zhou static PetscErrorCode MatMatTransposeMult_Elemental(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C) 440df311e6cSXuan Zhou { 441df311e6cSXuan Zhou PetscErrorCode ierr; 442df311e6cSXuan Zhou 443df311e6cSXuan Zhou PetscFunctionBegin; 444df311e6cSXuan Zhou if (scall == MAT_INITIAL_MATRIX){ 445df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 446df311e6cSXuan Zhou ierr = MatMatMultSymbolic_Elemental(A,B,1.0,C);CHKERRQ(ierr); 447df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultSymbolic,A,B,0,0);CHKERRQ(ierr); 448df311e6cSXuan Zhou } 449df311e6cSXuan Zhou ierr = PetscLogEventBegin(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 450df311e6cSXuan Zhou ierr = MatMatTransposeMultNumeric_Elemental(A,B,*C);CHKERRQ(ierr); 451df311e6cSXuan Zhou ierr = PetscLogEventEnd(MAT_MatTransposeMultNumeric,A,B,0,0);CHKERRQ(ierr); 452df311e6cSXuan Zhou PetscFunctionReturn(0); 453df311e6cSXuan Zhou } 454df311e6cSXuan Zhou 455df311e6cSXuan Zhou #undef __FUNCT__ 45661119200SXuan Zhou #define __FUNCT__ "MatGetDiagonal_Elemental" 457a9d89745SXuan Zhou static PetscErrorCode MatGetDiagonal_Elemental(Mat A,Vec D) 45861119200SXuan Zhou { 459a9d89745SXuan Zhou PetscInt i,nrows,ncols,nD,rrank,ridx,crank,cidx; 460a9d89745SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 46161119200SXuan Zhou PetscErrorCode ierr; 462a9d89745SXuan Zhou PetscElemScalar v; 463ce94432eSBarry Smith MPI_Comm comm; 46461119200SXuan Zhou 46561119200SXuan Zhou PetscFunctionBegin; 466ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 467a9d89745SXuan Zhou ierr = MatGetSize(A,&nrows,&ncols);CHKERRQ(ierr); 468a9d89745SXuan Zhou nD = nrows>ncols ? ncols : nrows; 469a9d89745SXuan Zhou for (i=0; i<nD; i++) { 470a9d89745SXuan Zhou PetscInt erow,ecol; 471a9d89745SXuan Zhou P2RO(A,0,i,&rrank,&ridx); 472a9d89745SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 473a9d89745SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 474a9d89745SXuan Zhou P2RO(A,1,i,&crank,&cidx); 475a9d89745SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 476a9d89745SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 477a9d89745SXuan Zhou v = a->emat->Get(erow,ecol); 4787c8b904fSSatish Balay ierr = VecSetValues(D,1,&i,(PetscScalar*)&v,INSERT_VALUES);CHKERRQ(ierr); 479ade3cc5eSXuan Zhou } 480a9d89745SXuan Zhou ierr = VecAssemblyBegin(D);CHKERRQ(ierr); 481a9d89745SXuan Zhou ierr = VecAssemblyEnd(D);CHKERRQ(ierr); 48261119200SXuan Zhou PetscFunctionReturn(0); 48361119200SXuan Zhou } 48461119200SXuan Zhou 48561119200SXuan Zhou #undef __FUNCT__ 486ade3cc5eSXuan Zhou #define __FUNCT__ "MatDiagonalScale_Elemental" 487ade3cc5eSXuan Zhou static PetscErrorCode MatDiagonalScale_Elemental(Mat X,Vec L,Vec R) 488ade3cc5eSXuan Zhou { 489ade3cc5eSXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 490ade3cc5eSXuan Zhou const PetscElemScalar *d; 491ade3cc5eSXuan Zhou PetscErrorCode ierr; 492ade3cc5eSXuan Zhou 493ade3cc5eSXuan Zhou PetscFunctionBegin; 4949065cd98SJed Brown if (R) { 495ade3cc5eSXuan Zhou ierr = VecGetArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr); 496e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> de; 4970c18141cSBarry Smith de.LockedAttach(X->cmap->N,1,*x->grid,0,0,d,X->cmap->n); 498e8146892SSatish Balay El::DiagonalScale(El::RIGHT,El::NORMAL,de,*x->emat); 499ade3cc5eSXuan Zhou ierr = VecRestoreArrayRead(R,(const PetscScalar **)&d);CHKERRQ(ierr); 5009065cd98SJed Brown } 5019065cd98SJed Brown if (L) { 502ade3cc5eSXuan Zhou ierr = VecGetArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr); 503e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> de; 5040c18141cSBarry Smith de.LockedAttach(X->rmap->N,1,*x->grid,0,0,d,X->rmap->n); 505e8146892SSatish Balay El::DiagonalScale(El::LEFT,El::NORMAL,de,*x->emat); 506ade3cc5eSXuan Zhou ierr = VecRestoreArrayRead(L,(const PetscScalar **)&d);CHKERRQ(ierr); 507ade3cc5eSXuan Zhou } 508ade3cc5eSXuan Zhou PetscFunctionReturn(0); 509ade3cc5eSXuan Zhou } 510ade3cc5eSXuan Zhou 511ade3cc5eSXuan Zhou #undef __FUNCT__ 5124ee44ac6SXuan Zhou #define __FUNCT__ "MatScale_Elemental" 513e6dea9dbSXuan Zhou static PetscErrorCode MatScale_Elemental(Mat X,PetscScalar a) 51465b78793SXuan Zhou { 51565b78793SXuan Zhou Mat_Elemental *x = (Mat_Elemental*)X->data; 51665b78793SXuan Zhou 51765b78793SXuan Zhou PetscFunctionBegin; 518e8146892SSatish Balay El::Scale((PetscElemScalar)a,*x->emat); 51965b78793SXuan Zhou PetscFunctionReturn(0); 52065b78793SXuan Zhou } 52165b78793SXuan Zhou 522f82baa17SHong Zhang /* 523f82baa17SHong Zhang MatAXPY - Computes Y = a*X + Y. 524f82baa17SHong Zhang */ 52565b78793SXuan Zhou #undef __FUNCT__ 5264ee44ac6SXuan Zhou #define __FUNCT__ "MatAXPY_Elemental" 527e6dea9dbSXuan Zhou static PetscErrorCode MatAXPY_Elemental(Mat Y,PetscScalar a,Mat X,MatStructure str) 528e09a3074SHong Zhang { 529e09a3074SHong Zhang Mat_Elemental *x = (Mat_Elemental*)X->data; 530e09a3074SHong Zhang Mat_Elemental *y = (Mat_Elemental*)Y->data; 53168446cf8SJed Brown PetscErrorCode ierr; 532e09a3074SHong Zhang 533e09a3074SHong Zhang PetscFunctionBegin; 534e8146892SSatish Balay El::Axpy((PetscElemScalar)a,*x->emat,*y->emat); 53521f6c9c4SJed Brown ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr); 536e09a3074SHong Zhang PetscFunctionReturn(0); 537e09a3074SHong Zhang } 538e09a3074SHong Zhang 539ae844d54SHong Zhang #undef __FUNCT__ 540d6223691SXuan Zhou #define __FUNCT__ "MatCopy_Elemental" 541d6223691SXuan Zhou static PetscErrorCode MatCopy_Elemental(Mat A,Mat B,MatStructure str) 542d6223691SXuan Zhou { 543d6223691SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 544d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 545d6223691SXuan Zhou 546d6223691SXuan Zhou PetscFunctionBegin; 547e8146892SSatish Balay El::Copy(*a->emat,*b->emat); 548d6223691SXuan Zhou PetscFunctionReturn(0); 549d6223691SXuan Zhou } 550d6223691SXuan Zhou 551d6223691SXuan Zhou #undef __FUNCT__ 552df311e6cSXuan Zhou #define __FUNCT__ "MatDuplicate_Elemental" 553df311e6cSXuan Zhou static PetscErrorCode MatDuplicate_Elemental(Mat A,MatDuplicateOption op,Mat *B) 554df311e6cSXuan Zhou { 555df311e6cSXuan Zhou Mat Be; 556ce94432eSBarry Smith MPI_Comm comm; 557df311e6cSXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 558df311e6cSXuan Zhou PetscErrorCode ierr; 559df311e6cSXuan Zhou 560df311e6cSXuan Zhou PetscFunctionBegin; 561ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 562df311e6cSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 563df311e6cSXuan Zhou ierr = MatSetSizes(Be,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 564df311e6cSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 565df311e6cSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 566df311e6cSXuan Zhou *B = Be; 567df311e6cSXuan Zhou if (op == MAT_COPY_VALUES) { 568df311e6cSXuan Zhou Mat_Elemental *b=(Mat_Elemental*)Be->data; 569e8146892SSatish Balay El::Copy(*a->emat,*b->emat); 570df311e6cSXuan Zhou } 571df311e6cSXuan Zhou Be->assembled = PETSC_TRUE; 572df311e6cSXuan Zhou PetscFunctionReturn(0); 573df311e6cSXuan Zhou } 574df311e6cSXuan Zhou 575df311e6cSXuan Zhou #undef __FUNCT__ 576d6223691SXuan Zhou #define __FUNCT__ "MatTranspose_Elemental" 577d6223691SXuan Zhou static PetscErrorCode MatTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 578d6223691SXuan Zhou { 579ec8cb81fSBarry Smith Mat Be = *B; 5805262d616SXuan Zhou PetscErrorCode ierr; 581ce94432eSBarry Smith MPI_Comm comm; 5824fe7bbcaSHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 583d6223691SXuan Zhou 584d6223691SXuan Zhou PetscFunctionBegin; 585ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 5865cb544a0SHong Zhang /* Only out-of-place supported */ 5875262d616SXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 5885262d616SXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 5895262d616SXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 5905262d616SXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 5915262d616SXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 5925262d616SXuan Zhou *B = Be; 5935262d616SXuan Zhou } 5944fe7bbcaSHong Zhang b = (Mat_Elemental*)Be->data; 595e8146892SSatish Balay El::Transpose(*a->emat,*b->emat); 5965262d616SXuan Zhou Be->assembled = PETSC_TRUE; 597d6223691SXuan Zhou PetscFunctionReturn(0); 598d6223691SXuan Zhou } 599d6223691SXuan Zhou 600d6223691SXuan Zhou #undef __FUNCT__ 601dfcb0403SXuan Zhou #define __FUNCT__ "MatConjugate_Elemental" 602dfcb0403SXuan Zhou static PetscErrorCode MatConjugate_Elemental(Mat A) 603dfcb0403SXuan Zhou { 604dfcb0403SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 605dfcb0403SXuan Zhou 606dfcb0403SXuan Zhou PetscFunctionBegin; 607e8146892SSatish Balay El::Conjugate(*a->emat); 608dfcb0403SXuan Zhou PetscFunctionReturn(0); 609dfcb0403SXuan Zhou } 610dfcb0403SXuan Zhou 611dfcb0403SXuan Zhou #undef __FUNCT__ 6124a29722dSXuan Zhou #define __FUNCT__ "MatHermitianTranspose_Elemental" 6134a29722dSXuan Zhou static PetscErrorCode MatHermitianTranspose_Elemental(Mat A,MatReuse reuse,Mat *B) 6144a29722dSXuan Zhou { 615ec8cb81fSBarry Smith Mat Be = *B; 6164a29722dSXuan Zhou PetscErrorCode ierr; 617ce94432eSBarry Smith MPI_Comm comm; 6184a29722dSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data, *b; 6194a29722dSXuan Zhou 6204a29722dSXuan Zhou PetscFunctionBegin; 621ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 6225cb544a0SHong Zhang /* Only out-of-place supported */ 6234a29722dSXuan Zhou if (reuse == MAT_INITIAL_MATRIX){ 6244a29722dSXuan Zhou ierr = MatCreate(comm,&Be);CHKERRQ(ierr); 6254a29722dSXuan Zhou ierr = MatSetSizes(Be,A->cmap->n,A->rmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 6264a29722dSXuan Zhou ierr = MatSetType(Be,MATELEMENTAL);CHKERRQ(ierr); 6274a29722dSXuan Zhou ierr = MatSetUp(Be);CHKERRQ(ierr); 6284a29722dSXuan Zhou *B = Be; 6294a29722dSXuan Zhou } 6304a29722dSXuan Zhou b = (Mat_Elemental*)Be->data; 631e8146892SSatish Balay El::Adjoint(*a->emat,*b->emat); 6324a29722dSXuan Zhou Be->assembled = PETSC_TRUE; 6334a29722dSXuan Zhou PetscFunctionReturn(0); 6344a29722dSXuan Zhou } 6354a29722dSXuan Zhou 6364a29722dSXuan Zhou #undef __FUNCT__ 6371f881ff8SXuan Zhou #define __FUNCT__ "MatSolve_Elemental" 6381f881ff8SXuan Zhou static PetscErrorCode MatSolve_Elemental(Mat A,Vec B,Vec X) 6391f881ff8SXuan Zhou { 6401f881ff8SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 6411f881ff8SXuan Zhou PetscErrorCode ierr; 642df311e6cSXuan Zhou PetscElemScalar *x; 6431f881ff8SXuan Zhou 6441f881ff8SXuan Zhou PetscFunctionBegin; 64545cf121fSXuan Zhou ierr = VecCopy(B,X);CHKERRQ(ierr); 646e6dea9dbSXuan Zhou ierr = VecGetArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 647e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::VC,El::STAR> xe; 6480c18141cSBarry Smith xe.Attach(A->rmap->N,1,*a->grid,0,0,x,A->rmap->n); 649e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::MC,El::MR> xer(xe); 650fc54b460SXuan Zhou switch (A->factortype) { 651fc54b460SXuan Zhou case MAT_FACTOR_LU: 6521f881ff8SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 653e8146892SSatish Balay El::lu::SolveAfter(El::NORMAL,*a->emat,*a->pivot,xer); 654e8146892SSatish Balay El::Copy(xer,xe); 655fc54b460SXuan Zhou } else { 656e8146892SSatish Balay El::lu::SolveAfter(El::NORMAL,*a->emat,xer); 657e8146892SSatish Balay El::Copy(xer,xe); 6581f881ff8SXuan Zhou } 659fc54b460SXuan Zhou break; 660fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 661e8146892SSatish Balay El::cholesky::SolveAfter(El::UPPER,El::NORMAL,*a->emat,xer); 662e8146892SSatish Balay El::Copy(xer,xe); 663fc54b460SXuan Zhou break; 664fc54b460SXuan Zhou default: 6654fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 666fc54b460SXuan Zhou break; 6671f881ff8SXuan Zhou } 668e6dea9dbSXuan Zhou ierr = VecRestoreArray(X,(PetscScalar **)&x);CHKERRQ(ierr); 6691f881ff8SXuan Zhou PetscFunctionReturn(0); 6701f881ff8SXuan Zhou } 6711f881ff8SXuan Zhou 6721f881ff8SXuan Zhou #undef __FUNCT__ 673df311e6cSXuan Zhou #define __FUNCT__ "MatSolveAdd_Elemental" 674df311e6cSXuan Zhou static PetscErrorCode MatSolveAdd_Elemental(Mat A,Vec B,Vec Y,Vec X) 675df311e6cSXuan Zhou { 676df311e6cSXuan Zhou PetscErrorCode ierr; 677df311e6cSXuan Zhou 678df311e6cSXuan Zhou PetscFunctionBegin; 679df311e6cSXuan Zhou ierr = MatSolve_Elemental(A,B,X);CHKERRQ(ierr); 6803d7f40dbSXuan Zhou ierr = VecAXPY(X,1,Y);CHKERRQ(ierr); 681df311e6cSXuan Zhou PetscFunctionReturn(0); 682df311e6cSXuan Zhou } 683df311e6cSXuan Zhou 684df311e6cSXuan Zhou #undef __FUNCT__ 685ae844d54SHong Zhang #define __FUNCT__ "MatMatSolve_Elemental" 686ae844d54SHong Zhang static PetscErrorCode MatMatSolve_Elemental(Mat A,Mat B,Mat X) 687ae844d54SHong Zhang { 6881f0e42cfSHong Zhang Mat_Elemental *a=(Mat_Elemental*)A->data; 689d6223691SXuan Zhou Mat_Elemental *b=(Mat_Elemental*)B->data; 6901f0e42cfSHong Zhang Mat_Elemental *x=(Mat_Elemental*)X->data; 6911f0e42cfSHong Zhang 692ae844d54SHong Zhang PetscFunctionBegin; 693e8146892SSatish Balay El::Copy(*b->emat,*x->emat); 694fc54b460SXuan Zhou switch (A->factortype) { 695fc54b460SXuan Zhou case MAT_FACTOR_LU: 696d6223691SXuan Zhou if ((*a->pivot).AllocatedMemory()) { 697e8146892SSatish Balay El::lu::SolveAfter(El::NORMAL,*a->emat,*a->pivot,*x->emat); 698fc54b460SXuan Zhou } else { 699e8146892SSatish Balay El::lu::SolveAfter(El::NORMAL,*a->emat,*x->emat); 700d6223691SXuan Zhou } 701fc54b460SXuan Zhou break; 702fc54b460SXuan Zhou case MAT_FACTOR_CHOLESKY: 703e8146892SSatish Balay El::cholesky::SolveAfter(El::UPPER,El::NORMAL,*a->emat,*x->emat); 704fc54b460SXuan Zhou break; 705fc54b460SXuan Zhou default: 7064fe7bbcaSHong Zhang SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unfactored Matrix or Unsupported MatFactorType"); 707fc54b460SXuan Zhou break; 708fc54b460SXuan Zhou } 709ae844d54SHong Zhang PetscFunctionReturn(0); 710ae844d54SHong Zhang } 711ae844d54SHong Zhang 712ae844d54SHong Zhang #undef __FUNCT__ 713ae844d54SHong Zhang #define __FUNCT__ "MatLUFactor_Elemental" 714ae844d54SHong Zhang static PetscErrorCode MatLUFactor_Elemental(Mat A,IS row,IS col,const MatFactorInfo *info) 715ae844d54SHong Zhang { 7167c920d81SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 7177c920d81SXuan Zhou 718ae844d54SHong Zhang PetscFunctionBegin; 719d6223691SXuan Zhou if (info->dtcol){ 720e8146892SSatish Balay El::LU(*a->emat,*a->pivot); 7211293973dSHong Zhang } else { 722e8146892SSatish Balay El::LU(*a->emat); 723d6223691SXuan Zhou } 7241293973dSHong Zhang A->factortype = MAT_FACTOR_LU; 725834d3fecSHong Zhang A->assembled = PETSC_TRUE; 726ae844d54SHong Zhang PetscFunctionReturn(0); 727ae844d54SHong Zhang } 728ae844d54SHong Zhang 729d7c3f9d8SHong Zhang #undef __FUNCT__ 730d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_Elemental" 731d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 732d7c3f9d8SHong Zhang { 733d7c3f9d8SHong Zhang PetscErrorCode ierr; 734d7c3f9d8SHong Zhang 735d7c3f9d8SHong Zhang PetscFunctionBegin; 736d7c3f9d8SHong Zhang ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 737d7c3f9d8SHong Zhang ierr = MatLUFactor_Elemental(F,0,0,info);CHKERRQ(ierr); 738d7c3f9d8SHong Zhang PetscFunctionReturn(0); 739d7c3f9d8SHong Zhang } 740d7c3f9d8SHong Zhang 741d7c3f9d8SHong Zhang #undef __FUNCT__ 742d7c3f9d8SHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_Elemental" 743d7c3f9d8SHong Zhang static PetscErrorCode MatLUFactorSymbolic_Elemental(Mat F,Mat A,IS r,IS c,const MatFactorInfo *info) 744d7c3f9d8SHong Zhang { 745d7c3f9d8SHong Zhang PetscFunctionBegin; 746d7c3f9d8SHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 747d7c3f9d8SHong Zhang PetscFunctionReturn(0); 748d7c3f9d8SHong Zhang } 749d7c3f9d8SHong Zhang 75045cf121fSXuan Zhou #undef __FUNCT__ 75145cf121fSXuan Zhou #define __FUNCT__ "MatCholeskyFactor_Elemental" 75245cf121fSXuan Zhou static PetscErrorCode MatCholeskyFactor_Elemental(Mat A,IS perm,const MatFactorInfo *info) 75345cf121fSXuan Zhou { 75445cf121fSXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 755e8146892SSatish Balay El::DistMatrix<PetscElemScalar,El::MC,El::STAR> d; 75645cf121fSXuan Zhou 75745cf121fSXuan Zhou PetscFunctionBegin; 758e8146892SSatish Balay El::Cholesky(El::UPPER,*a->emat); 759fc54b460SXuan Zhou A->factortype = MAT_FACTOR_CHOLESKY; 760834d3fecSHong Zhang A->assembled = PETSC_TRUE; 76145cf121fSXuan Zhou PetscFunctionReturn(0); 76245cf121fSXuan Zhou } 76345cf121fSXuan Zhou 76479673f7bSHong Zhang #undef __FUNCT__ 76579673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_Elemental" 76679673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorNumeric_Elemental(Mat F,Mat A,const MatFactorInfo *info) 76779673f7bSHong Zhang { 768cb76c1d8SXuan Zhou PetscErrorCode ierr; 769cb76c1d8SXuan Zhou 770cb76c1d8SXuan Zhou PetscFunctionBegin; 771cb76c1d8SXuan Zhou ierr = MatCopy(A,F,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 772cb76c1d8SXuan Zhou ierr = MatCholeskyFactor_Elemental(F,0,info);CHKERRQ(ierr); 773cb76c1d8SXuan Zhou PetscFunctionReturn(0); 77479673f7bSHong Zhang } 775cb76c1d8SXuan Zhou 77679673f7bSHong Zhang #undef __FUNCT__ 77779673f7bSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_Elemental" 77879673f7bSHong Zhang static PetscErrorCode MatCholeskyFactorSymbolic_Elemental(Mat F,Mat A,IS perm,const MatFactorInfo *info) 77979673f7bSHong Zhang { 78079673f7bSHong Zhang PetscFunctionBegin; 78179673f7bSHong Zhang /* F is create and allocated by MatGetFactor_elemental_petsc(), skip this routine. */ 78279673f7bSHong Zhang PetscFunctionReturn(0); 78379673f7bSHong Zhang } 78479673f7bSHong Zhang 7851293973dSHong Zhang #undef __FUNCT__ 78615767789SHong Zhang #define __FUNCT__ "MatFactorGetSolverPackage_elemental_elemental" 78715767789SHong Zhang PetscErrorCode MatFactorGetSolverPackage_elemental_elemental(Mat A,const MatSolverPackage *type) 78815767789SHong Zhang { 78915767789SHong Zhang PetscFunctionBegin; 79015767789SHong Zhang *type = MATSOLVERELEMENTAL; 79115767789SHong Zhang PetscFunctionReturn(0); 79215767789SHong Zhang } 79315767789SHong Zhang 79415767789SHong Zhang #undef __FUNCT__ 79515767789SHong Zhang #define __FUNCT__ "MatGetFactor_elemental_elemental" 79615767789SHong Zhang static PetscErrorCode MatGetFactor_elemental_elemental(Mat A,MatFactorType ftype,Mat *F) 7971293973dSHong Zhang { 7981293973dSHong Zhang Mat B; 7991293973dSHong Zhang PetscErrorCode ierr; 8001293973dSHong Zhang 8011293973dSHong Zhang PetscFunctionBegin; 8021293973dSHong Zhang /* Create the factorization matrix */ 803ce94432eSBarry Smith ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr); 8041293973dSHong Zhang ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 8051293973dSHong Zhang ierr = MatSetType(B,MATELEMENTAL);CHKERRQ(ierr); 8061293973dSHong Zhang ierr = MatSetUp(B);CHKERRQ(ierr); 8071293973dSHong Zhang B->factortype = ftype; 808bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)B,"MatFactorGetSolverPackage_C",MatFactorGetSolverPackage_elemental_elemental);CHKERRQ(ierr); 8091293973dSHong Zhang *F = B; 8101293973dSHong Zhang PetscFunctionReturn(0); 8111293973dSHong Zhang } 8121293973dSHong Zhang 8131f881ff8SXuan Zhou #undef __FUNCT__ 81442c9c57cSBarry Smith #define __FUNCT__ "MatSolverPackageRegister_Elemental" 81529b38603SBarry Smith PETSC_EXTERN PetscErrorCode MatSolverPackageRegister_Elemental(void) 81642c9c57cSBarry Smith { 81718713533SBarry Smith PetscErrorCode ierr; 81818713533SBarry Smith 81942c9c57cSBarry Smith PetscFunctionBegin; 82042c9c57cSBarry Smith ierr = MatSolverPackageRegister(MATSOLVERELEMENTAL,MATELEMENTAL, MAT_FACTOR_LU,MatGetFactor_elemental_elemental);CHKERRQ(ierr); 82142c9c57cSBarry Smith ierr = MatSolverPackageRegister(MATSOLVERELEMENTAL,MATELEMENTAL, MAT_FACTOR_CHOLESKY,MatGetFactor_elemental_elemental);CHKERRQ(ierr); 82242c9c57cSBarry Smith PetscFunctionReturn(0); 82342c9c57cSBarry Smith } 82442c9c57cSBarry Smith 82542c9c57cSBarry Smith #undef __FUNCT__ 8261f881ff8SXuan Zhou #define __FUNCT__ "MatNorm_Elemental" 8271f881ff8SXuan Zhou static PetscErrorCode MatNorm_Elemental(Mat A,NormType type,PetscReal *nrm) 8281f881ff8SXuan Zhou { 8291f881ff8SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 8301f881ff8SXuan Zhou 8311f881ff8SXuan Zhou PetscFunctionBegin; 8321f881ff8SXuan Zhou switch (type){ 8331f881ff8SXuan Zhou case NORM_1: 834e8146892SSatish Balay *nrm = El::OneNorm(*a->emat); 8351f881ff8SXuan Zhou break; 8361f881ff8SXuan Zhou case NORM_FROBENIUS: 837e8146892SSatish Balay *nrm = El::FrobeniusNorm(*a->emat); 8381f881ff8SXuan Zhou break; 8391f881ff8SXuan Zhou case NORM_INFINITY: 840e8146892SSatish Balay *nrm = El::InfinityNorm(*a->emat); 8411f881ff8SXuan Zhou break; 8421f881ff8SXuan Zhou default: 8431f881ff8SXuan Zhou printf("Error: unsupported norm type!\n"); 8441f881ff8SXuan Zhou } 8451f881ff8SXuan Zhou PetscFunctionReturn(0); 8461f881ff8SXuan Zhou } 8471f881ff8SXuan Zhou 8485262d616SXuan Zhou #undef __FUNCT__ 8495262d616SXuan Zhou #define __FUNCT__ "MatZeroEntries_Elemental" 8505262d616SXuan Zhou static PetscErrorCode MatZeroEntries_Elemental(Mat A) 8515262d616SXuan Zhou { 8525262d616SXuan Zhou Mat_Elemental *a=(Mat_Elemental*)A->data; 8535262d616SXuan Zhou 8545262d616SXuan Zhou PetscFunctionBegin; 855e8146892SSatish Balay El::Zero(*a->emat); 8565262d616SXuan Zhou PetscFunctionReturn(0); 8575262d616SXuan Zhou } 8585262d616SXuan Zhou 859e09a3074SHong Zhang #undef __FUNCT__ 860db31f6deSJed Brown #define __FUNCT__ "MatGetOwnershipIS_Elemental" 861db31f6deSJed Brown static PetscErrorCode MatGetOwnershipIS_Elemental(Mat A,IS *rows,IS *cols) 862db31f6deSJed Brown { 863db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 864db31f6deSJed Brown PetscErrorCode ierr; 865db31f6deSJed Brown PetscInt i,m,shift,stride,*idx; 866db31f6deSJed Brown 867db31f6deSJed Brown PetscFunctionBegin; 868db31f6deSJed Brown if (rows) { 869db31f6deSJed Brown m = a->emat->LocalHeight(); 870db31f6deSJed Brown shift = a->emat->ColShift(); 871db31f6deSJed Brown stride = a->emat->ColStride(); 872785e854fSJed Brown ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr); 873db31f6deSJed Brown for (i=0; i<m; i++) { 874db31f6deSJed Brown PetscInt rank,offset; 875db31f6deSJed Brown E2RO(A,0,shift+i*stride,&rank,&offset); 876db31f6deSJed Brown RO2P(A,0,rank,offset,&idx[i]); 877db31f6deSJed Brown } 878db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,rows);CHKERRQ(ierr); 879db31f6deSJed Brown } 880db31f6deSJed Brown if (cols) { 881db31f6deSJed Brown m = a->emat->LocalWidth(); 882db31f6deSJed Brown shift = a->emat->RowShift(); 883db31f6deSJed Brown stride = a->emat->RowStride(); 884785e854fSJed Brown ierr = PetscMalloc1(m,&idx);CHKERRQ(ierr); 885db31f6deSJed Brown for (i=0; i<m; i++) { 886db31f6deSJed Brown PetscInt rank,offset; 887db31f6deSJed Brown E2RO(A,1,shift+i*stride,&rank,&offset); 888db31f6deSJed Brown RO2P(A,1,rank,offset,&idx[i]); 889db31f6deSJed Brown } 890db31f6deSJed Brown ierr = ISCreateGeneral(PETSC_COMM_SELF,m,idx,PETSC_OWN_POINTER,cols);CHKERRQ(ierr); 891db31f6deSJed Brown } 892db31f6deSJed Brown PetscFunctionReturn(0); 893db31f6deSJed Brown } 894db31f6deSJed Brown 895db31f6deSJed Brown #undef __FUNCT__ 8962ef0cf24SXuan Zhou #define __FUNCT__ "MatConvert_Elemental_Dense" 89719fd82e9SBarry Smith static PetscErrorCode MatConvert_Elemental_Dense(Mat A,MatType newtype,MatReuse reuse,Mat *B) 898af295397SXuan Zhou { 8992ef0cf24SXuan Zhou Mat Bmpi; 900af295397SXuan Zhou Mat_Elemental *a = (Mat_Elemental*)A->data; 901ce94432eSBarry Smith MPI_Comm comm; 9022ef0cf24SXuan Zhou PetscErrorCode ierr; 903fa9e26c8SHong Zhang IS isrows,iscols; 904fa9e26c8SHong Zhang PetscInt rrank,ridx,crank,cidx,nrows,ncols,i,j,erow,ecol,elrow,elcol; 905fa9e26c8SHong Zhang const PetscInt *rows,*cols; 906df311e6cSXuan Zhou PetscElemScalar v; 907fa9e26c8SHong Zhang const El::Grid &grid = a->emat->Grid(); 908af295397SXuan Zhou 909af295397SXuan Zhou PetscFunctionBegin; 910ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 911a000e53bSHong Zhang 912de0a22f0SHong Zhang if (reuse == MAT_REUSE_MATRIX) { 913de0a22f0SHong Zhang Bmpi = *B; 914de0a22f0SHong Zhang } else { 915af295397SXuan Zhou ierr = MatCreate(comm,&Bmpi);CHKERRQ(ierr); 916af295397SXuan Zhou ierr = MatSetSizes(Bmpi,A->rmap->n,A->cmap->n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 9172ef0cf24SXuan Zhou ierr = MatSetType(Bmpi,MATDENSE);CHKERRQ(ierr); 918af295397SXuan Zhou ierr = MatSetUp(Bmpi);CHKERRQ(ierr); 919de0a22f0SHong Zhang } 920fa9e26c8SHong Zhang 921fa9e26c8SHong Zhang /* Get local entries of A */ 922fa9e26c8SHong Zhang ierr = MatGetOwnershipIS(A,&isrows,&iscols);CHKERRQ(ierr); 923fa9e26c8SHong Zhang ierr = ISGetLocalSize(isrows,&nrows);CHKERRQ(ierr); 924fa9e26c8SHong Zhang ierr = ISGetIndices(isrows,&rows);CHKERRQ(ierr); 925fa9e26c8SHong Zhang ierr = ISGetLocalSize(iscols,&ncols);CHKERRQ(ierr); 926fa9e26c8SHong Zhang ierr = ISGetIndices(iscols,&cols);CHKERRQ(ierr); 927fa9e26c8SHong Zhang 92857299f2fSHong Zhang if (a->roworiented) { 9292ef0cf24SXuan Zhou for (i=0; i<nrows; i++) { 930fa9e26c8SHong Zhang P2RO(A,0,rows[i],&rrank,&ridx); /* convert indices between PETSc <-> (Rank,Offset) <-> Elemental */ 9312ef0cf24SXuan Zhou RO2E(A,0,rrank,ridx,&erow); 9322ef0cf24SXuan Zhou if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 9332ef0cf24SXuan Zhou for (j=0; j<ncols; j++) { 934fa9e26c8SHong Zhang P2RO(A,1,cols[j],&crank,&cidx); 9352ef0cf24SXuan Zhou RO2E(A,1,crank,cidx,&ecol); 9362ef0cf24SXuan Zhou if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 937fa9e26c8SHong Zhang 938fa9e26c8SHong Zhang elrow = erow / grid.MCSize(); /* Elemental local row index */ 939fa9e26c8SHong Zhang elcol = ecol / grid.MRSize(); /* Elemental local column index */ 940fa9e26c8SHong Zhang v = a->emat->GetLocal(elrow,elcol); 941fa9e26c8SHong Zhang ierr = MatSetValues(Bmpi,1,&rows[i],1,&cols[j],(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 9422ef0cf24SXuan Zhou } 9432ef0cf24SXuan Zhou } 94457299f2fSHong Zhang } else { /* column-oriented */ 94557299f2fSHong Zhang for (j=0; j<ncols; j++) { 94657299f2fSHong Zhang P2RO(A,1,cols[j],&crank,&cidx); 94757299f2fSHong Zhang RO2E(A,1,crank,cidx,&ecol); 94857299f2fSHong Zhang if (crank < 0 || cidx < 0 || ecol < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect col translation"); 94957299f2fSHong Zhang for (i=0; i<nrows; i++) { 95057299f2fSHong Zhang P2RO(A,0,rows[i],&rrank,&ridx); /* convert indices between PETSc <-> (Rank,Offset) <-> Elemental */ 95157299f2fSHong Zhang RO2E(A,0,rrank,ridx,&erow); 95257299f2fSHong Zhang if (rrank < 0 || ridx < 0 || erow < 0) SETERRQ(comm,PETSC_ERR_PLIB,"Incorrect row translation"); 95357299f2fSHong Zhang 95457299f2fSHong Zhang elrow = erow / grid.MCSize(); /* Elemental local row index */ 95557299f2fSHong Zhang elcol = ecol / grid.MRSize(); /* Elemental local column index */ 95657299f2fSHong Zhang v = a->emat->GetLocal(elrow,elcol); 95757299f2fSHong Zhang ierr = MatSetValues(Bmpi,1,&rows[i],1,&cols[j],(PetscScalar *)&v,INSERT_VALUES);CHKERRQ(ierr); 95857299f2fSHong Zhang } 95957299f2fSHong Zhang } 96057299f2fSHong Zhang } 961af295397SXuan Zhou ierr = MatAssemblyBegin(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 962af295397SXuan Zhou ierr = MatAssemblyEnd(Bmpi,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 963511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 96428be2f97SBarry Smith ierr = MatHeaderReplace(A,&Bmpi);CHKERRQ(ierr); 965c4ad791aSXuan Zhou } else { 966c4ad791aSXuan Zhou *B = Bmpi; 967c4ad791aSXuan Zhou } 968fa9e26c8SHong Zhang ierr = ISDestroy(&isrows);CHKERRQ(ierr); 969fa9e26c8SHong Zhang ierr = ISDestroy(&iscols);CHKERRQ(ierr); 970af295397SXuan Zhou PetscFunctionReturn(0); 971af295397SXuan Zhou } 972af295397SXuan Zhou 973af295397SXuan Zhou #undef __FUNCT__ 974af8000cdSHong Zhang #define __FUNCT__ "MatConvert_SeqAIJ_Elemental" 975af8000cdSHong Zhang PETSC_EXTERN PetscErrorCode MatConvert_SeqAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 976af8000cdSHong Zhang { 977af8000cdSHong Zhang Mat mat_elemental; 978af8000cdSHong Zhang PetscErrorCode ierr; 979af8000cdSHong Zhang PetscInt M=A->rmap->N,N=A->cmap->N,row,ncols; 980af8000cdSHong Zhang const PetscInt *cols; 981af8000cdSHong Zhang const PetscScalar *vals; 982af8000cdSHong Zhang 983af8000cdSHong Zhang PetscFunctionBegin; 984af8000cdSHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr); 985af8000cdSHong Zhang ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); 986af8000cdSHong Zhang ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr); 987af8000cdSHong Zhang ierr = MatSetUp(mat_elemental);CHKERRQ(ierr); 988af8000cdSHong Zhang for (row=0; row<M; row++) { 989af8000cdSHong Zhang ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 990af8000cdSHong Zhang /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */ 991af8000cdSHong Zhang ierr = MatSetValues(mat_elemental,1,&row,ncols,cols,vals,ADD_VALUES);CHKERRQ(ierr); 992af8000cdSHong Zhang ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 993af8000cdSHong Zhang } 994af8000cdSHong Zhang ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 995af8000cdSHong Zhang ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 996af8000cdSHong Zhang 997511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 99828be2f97SBarry Smith ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr); 999af8000cdSHong Zhang } else { 1000af8000cdSHong Zhang *newmat = mat_elemental; 1001af8000cdSHong Zhang } 1002af8000cdSHong Zhang PetscFunctionReturn(0); 1003af8000cdSHong Zhang } 1004af8000cdSHong Zhang 1005af8000cdSHong Zhang #undef __FUNCT__ 10065d7652ecSHong Zhang #define __FUNCT__ "MatConvert_MPIAIJ_Elemental" 10075d7652ecSHong Zhang PETSC_EXTERN PetscErrorCode MatConvert_MPIAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 10085d7652ecSHong Zhang { 10095d7652ecSHong Zhang Mat mat_elemental; 10105d7652ecSHong Zhang PetscErrorCode ierr; 10115d7652ecSHong Zhang PetscInt row,ncols,rstart=A->rmap->rstart,rend=A->rmap->rend,j; 10125d7652ecSHong Zhang const PetscInt *cols; 10135d7652ecSHong Zhang const PetscScalar *vals; 10145d7652ecSHong Zhang 10155d7652ecSHong Zhang PetscFunctionBegin; 10165d7652ecSHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr); 10175d7652ecSHong Zhang ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,A->rmap->N,A->cmap->N);CHKERRQ(ierr); 10185d7652ecSHong Zhang ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr); 10195d7652ecSHong Zhang ierr = MatSetUp(mat_elemental);CHKERRQ(ierr); 10205d7652ecSHong Zhang for (row=rstart; row<rend; row++) { 10215d7652ecSHong Zhang ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 10225d7652ecSHong Zhang for (j=0; j<ncols; j++) { 10235d7652ecSHong Zhang /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */ 10245d7652ecSHong Zhang ierr = MatSetValues(mat_elemental,1,&row,1,&cols[j],&vals[j],ADD_VALUES);CHKERRQ(ierr); 10255d7652ecSHong Zhang } 10265d7652ecSHong Zhang ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 10275d7652ecSHong Zhang } 10285d7652ecSHong Zhang ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 10295d7652ecSHong Zhang ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 10305d7652ecSHong Zhang 1031511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 103228be2f97SBarry Smith ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr); 10335d7652ecSHong Zhang } else { 10345d7652ecSHong Zhang *newmat = mat_elemental; 10355d7652ecSHong Zhang } 10365d7652ecSHong Zhang PetscFunctionReturn(0); 10375d7652ecSHong Zhang } 10385d7652ecSHong Zhang 10395d7652ecSHong Zhang #undef __FUNCT__ 10406214f412SHong Zhang #define __FUNCT__ "MatConvert_SeqSBAIJ_Elemental" 10416214f412SHong Zhang PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 10426214f412SHong Zhang { 10436214f412SHong Zhang Mat mat_elemental; 10446214f412SHong Zhang PetscErrorCode ierr; 10456214f412SHong Zhang PetscInt M=A->rmap->N,N=A->cmap->N,row,ncols,j; 10466214f412SHong Zhang const PetscInt *cols; 10476214f412SHong Zhang const PetscScalar *vals; 10486214f412SHong Zhang 10496214f412SHong Zhang PetscFunctionBegin; 10506214f412SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr); 10516214f412SHong Zhang ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); 10526214f412SHong Zhang ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr); 10536214f412SHong Zhang ierr = MatSetUp(mat_elemental);CHKERRQ(ierr); 10546214f412SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 10556214f412SHong Zhang for (row=0; row<M; row++) { 10566214f412SHong Zhang ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 10576214f412SHong Zhang /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */ 10586214f412SHong Zhang ierr = MatSetValues(mat_elemental,1,&row,ncols,cols,vals,ADD_VALUES);CHKERRQ(ierr); 10596214f412SHong Zhang for (j=0; j<ncols; j++) { /* lower triangular part */ 10606214f412SHong Zhang if (cols[j] == row) continue; 10616214f412SHong Zhang ierr = MatSetValues(mat_elemental,1,&cols[j],1,&row,&vals[j],ADD_VALUES);CHKERRQ(ierr); 10626214f412SHong Zhang } 10636214f412SHong Zhang ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 10646214f412SHong Zhang } 10656214f412SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 10666214f412SHong Zhang ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 10676214f412SHong Zhang ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 10686214f412SHong Zhang 1069511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 107028be2f97SBarry Smith ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr); 10716214f412SHong Zhang } else { 10726214f412SHong Zhang *newmat = mat_elemental; 10736214f412SHong Zhang } 10746214f412SHong Zhang PetscFunctionReturn(0); 10756214f412SHong Zhang } 10766214f412SHong Zhang 10776214f412SHong Zhang #undef __FUNCT__ 10786214f412SHong Zhang #define __FUNCT__ "MatConvert_MPISBAIJ_Elemental" 10796214f412SHong Zhang PETSC_EXTERN PetscErrorCode MatConvert_MPISBAIJ_Elemental(Mat A, MatType newtype,MatReuse reuse,Mat *newmat) 10806214f412SHong Zhang { 10816214f412SHong Zhang Mat mat_elemental; 10826214f412SHong Zhang PetscErrorCode ierr; 10836214f412SHong Zhang PetscInt M=A->rmap->N,N=A->cmap->N,row,ncols,j,rstart=A->rmap->rstart,rend=A->rmap->rend; 10846214f412SHong Zhang const PetscInt *cols; 10856214f412SHong Zhang const PetscScalar *vals; 10866214f412SHong Zhang 10876214f412SHong Zhang PetscFunctionBegin; 10886214f412SHong Zhang ierr = MatCreate(PetscObjectComm((PetscObject)A), &mat_elemental);CHKERRQ(ierr); 10896214f412SHong Zhang ierr = MatSetSizes(mat_elemental,PETSC_DECIDE,PETSC_DECIDE,M,N);CHKERRQ(ierr); 10906214f412SHong Zhang ierr = MatSetType(mat_elemental,MATELEMENTAL);CHKERRQ(ierr); 10916214f412SHong Zhang ierr = MatSetUp(mat_elemental);CHKERRQ(ierr); 10926214f412SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 10936214f412SHong Zhang for (row=rstart; row<rend; row++) { 10946214f412SHong Zhang ierr = MatGetRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 10956214f412SHong Zhang /* PETSc-Elemental interaface uses axpy for setting off-processor entries, only ADD_VALUES is allowed */ 10966214f412SHong Zhang ierr = MatSetValues(mat_elemental,1,&row,ncols,cols,vals,ADD_VALUES);CHKERRQ(ierr); 10976214f412SHong Zhang for (j=0; j<ncols; j++) { /* lower triangular part */ 10986214f412SHong Zhang if (cols[j] == row) continue; 10996214f412SHong Zhang ierr = MatSetValues(mat_elemental,1,&cols[j],1,&row,&vals[j],ADD_VALUES);CHKERRQ(ierr); 11006214f412SHong Zhang } 11016214f412SHong Zhang ierr = MatRestoreRow(A,row,&ncols,&cols,&vals);CHKERRQ(ierr); 11026214f412SHong Zhang } 11036214f412SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11046214f412SHong Zhang ierr = MatAssemblyBegin(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11056214f412SHong Zhang ierr = MatAssemblyEnd(mat_elemental, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11066214f412SHong Zhang 1107511c6705SHong Zhang if (reuse == MAT_INPLACE_MATRIX) { 110828be2f97SBarry Smith ierr = MatHeaderReplace(A,&mat_elemental);CHKERRQ(ierr); 11096214f412SHong Zhang } else { 11106214f412SHong Zhang *newmat = mat_elemental; 11116214f412SHong Zhang } 11126214f412SHong Zhang PetscFunctionReturn(0); 11136214f412SHong Zhang } 11146214f412SHong Zhang 11156214f412SHong Zhang #undef __FUNCT__ 1116db31f6deSJed Brown #define __FUNCT__ "MatDestroy_Elemental" 1117db31f6deSJed Brown static PetscErrorCode MatDestroy_Elemental(Mat A) 1118db31f6deSJed Brown { 1119db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 1120db31f6deSJed Brown PetscErrorCode ierr; 11215e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 11225e9f5b67SHong Zhang PetscBool flg; 11235e9f5b67SHong Zhang MPI_Comm icomm; 1124db31f6deSJed Brown 1125db31f6deSJed Brown PetscFunctionBegin; 1126db31f6deSJed Brown delete a->emat; 1127da0640a4SHong Zhang delete a->pivot; 11285e9f5b67SHong Zhang 1129e8146892SSatish Balay El::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A)); 11300c18141cSBarry Smith ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr); 11315e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 11325e9f5b67SHong Zhang if (--commgrid->grid_refct == 0) { 11335e9f5b67SHong Zhang delete commgrid->grid; 11345e9f5b67SHong Zhang ierr = PetscFree(commgrid);CHKERRQ(ierr); 1135a000e53bSHong Zhang ierr = MPI_Keyval_free(&Petsc_Elemental_keyval);CHKERRQ(ierr); 11365e9f5b67SHong Zhang } 11375e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 1138bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",NULL);CHKERRQ(ierr); 1139bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatFactorGetSolverPackage_C",NULL);CHKERRQ(ierr); 1140d5560a0bSSatish Balay ierr = PetscObjectComposeFunction((PetscObject)A,"MatElementalHermitianGenDefEig_C",NULL);CHKERRQ(ierr); 1141db31f6deSJed Brown ierr = PetscFree(A->data);CHKERRQ(ierr); 1142db31f6deSJed Brown PetscFunctionReturn(0); 1143db31f6deSJed Brown } 1144db31f6deSJed Brown 1145db31f6deSJed Brown #undef __FUNCT__ 1146db31f6deSJed Brown #define __FUNCT__ "MatSetUp_Elemental" 1147db31f6deSJed Brown PetscErrorCode MatSetUp_Elemental(Mat A) 1148db31f6deSJed Brown { 1149db31f6deSJed Brown Mat_Elemental *a = (Mat_Elemental*)A->data; 1150db31f6deSJed Brown PetscErrorCode ierr; 1151db31f6deSJed Brown PetscMPIInt rsize,csize; 1152db31f6deSJed Brown 1153db31f6deSJed Brown PetscFunctionBegin; 1154db31f6deSJed Brown ierr = PetscLayoutSetUp(A->rmap);CHKERRQ(ierr); 1155db31f6deSJed Brown ierr = PetscLayoutSetUp(A->cmap);CHKERRQ(ierr); 1156db31f6deSJed Brown 1157efb79153SJack Poulson a->emat->Resize(A->rmap->N,A->cmap->N);CHKERRQ(ierr); 1158e8146892SSatish Balay El::Zero(*a->emat); 1159db31f6deSJed Brown 1160db31f6deSJed Brown ierr = MPI_Comm_size(A->rmap->comm,&rsize);CHKERRQ(ierr); 1161db31f6deSJed Brown ierr = MPI_Comm_size(A->cmap->comm,&csize);CHKERRQ(ierr); 1162ce94432eSBarry Smith if (csize != rsize) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Cannot use row and column communicators of different sizes"); 1163db31f6deSJed Brown a->commsize = rsize; 1164db31f6deSJed Brown a->mr[0] = A->rmap->N % rsize; if (!a->mr[0]) a->mr[0] = rsize; 1165db31f6deSJed Brown a->mr[1] = A->cmap->N % csize; if (!a->mr[1]) a->mr[1] = csize; 1166db31f6deSJed Brown a->m[0] = A->rmap->N / rsize + (a->mr[0] != rsize); 1167db31f6deSJed Brown a->m[1] = A->cmap->N / csize + (a->mr[1] != csize); 1168db31f6deSJed Brown PetscFunctionReturn(0); 1169db31f6deSJed Brown } 1170db31f6deSJed Brown 1171aae2c449SHong Zhang #undef __FUNCT__ 1172aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyBegin_Elemental" 1173aae2c449SHong Zhang PetscErrorCode MatAssemblyBegin_Elemental(Mat A, MatAssemblyType type) 1174aae2c449SHong Zhang { 1175aae2c449SHong Zhang Mat_Elemental *a = (Mat_Elemental*)A->data; 1176aae2c449SHong Zhang 1177aae2c449SHong Zhang PetscFunctionBegin; 1178fbbcb730SJack Poulson /* printf("Calling ProcessQueues\n"); */ 1179fbbcb730SJack Poulson a->emat->ProcessQueues(); 1180fbbcb730SJack Poulson /* printf("Finished ProcessQueues\n"); */ 1181aae2c449SHong Zhang PetscFunctionReturn(0); 1182aae2c449SHong Zhang } 1183aae2c449SHong Zhang 1184aae2c449SHong Zhang #undef __FUNCT__ 1185aae2c449SHong Zhang #define __FUNCT__ "MatAssemblyEnd_Elemental" 1186aae2c449SHong Zhang PetscErrorCode MatAssemblyEnd_Elemental(Mat A, MatAssemblyType type) 1187aae2c449SHong Zhang { 1188aae2c449SHong Zhang PetscFunctionBegin; 1189aae2c449SHong Zhang /* Currently does nothing */ 1190aae2c449SHong Zhang PetscFunctionReturn(0); 1191aae2c449SHong Zhang } 1192aae2c449SHong Zhang 11937b30e0f1SHong Zhang #undef __FUNCT__ 11947b30e0f1SHong Zhang #define __FUNCT__ "MatLoad_Elemental" 11957b30e0f1SHong Zhang PetscErrorCode MatLoad_Elemental(Mat newMat, PetscViewer viewer) 11967b30e0f1SHong Zhang { 11977b30e0f1SHong Zhang PetscErrorCode ierr; 11987b30e0f1SHong Zhang Mat Adense,Ae; 11997b30e0f1SHong Zhang MPI_Comm comm; 12007b30e0f1SHong Zhang 12017b30e0f1SHong Zhang PetscFunctionBegin; 12027b30e0f1SHong Zhang ierr = PetscObjectGetComm((PetscObject)newMat,&comm);CHKERRQ(ierr); 12037b30e0f1SHong Zhang ierr = MatCreate(comm,&Adense);CHKERRQ(ierr); 12047b30e0f1SHong Zhang ierr = MatSetType(Adense,MATDENSE);CHKERRQ(ierr); 12057b30e0f1SHong Zhang ierr = MatLoad(Adense,viewer);CHKERRQ(ierr); 12067b30e0f1SHong Zhang ierr = MatConvert(Adense, MATELEMENTAL, MAT_INITIAL_MATRIX,&Ae);CHKERRQ(ierr); 12077b30e0f1SHong Zhang ierr = MatDestroy(&Adense);CHKERRQ(ierr); 120828be2f97SBarry Smith ierr = MatHeaderReplace(newMat,&Ae);CHKERRQ(ierr); 12097b30e0f1SHong Zhang PetscFunctionReturn(0); 12107b30e0f1SHong Zhang } 12117b30e0f1SHong Zhang 12121d08bef3SHong Zhang #undef __FUNCT__ 1213d5560a0bSSatish Balay #define __FUNCT__ "MatElementalHermitianGenDefEig_Elemental" 12142ef15aa2SHong Zhang PetscErrorCode MatElementalHermitianGenDefEig_Elemental(El::Pencil eigtype,El::UpperOrLower uplo,Mat A,Mat B,Mat *evals,Mat *evec,El::SortType sort,El::HermitianEigSubset<PetscElemScalar> subset,const El::HermitianEigCtrl<PetscElemScalar> ctrl) 12151d08bef3SHong Zhang { 12165dd12673SHong Zhang PetscErrorCode ierr; 1217874e4880SHong Zhang Mat_Elemental *a=(Mat_Elemental*)A->data,*b=(Mat_Elemental*)B->data,*x; 1218874e4880SHong Zhang MPI_Comm comm; 1219874e4880SHong Zhang Mat EVAL; 12202ef15aa2SHong Zhang Mat_Elemental *e; 1221874e4880SHong Zhang 12221d08bef3SHong Zhang PetscFunctionBegin; 1223874e4880SHong Zhang /* Compute eigenvalues and eigenvectors */ 1224d5560a0bSSatish Balay El::DistMatrix<PetscElemScalar,El::VR,El::STAR> w( *a->grid ); /* holding eigenvalues */ 1225d5560a0bSSatish Balay El::DistMatrix<PetscElemScalar> X( *a->grid ); /* holding eigenvectors */ 1226935dd83bSHong Zhang El::HermitianGenDefEig(eigtype,uplo,*a->emat,*b->emat,w,X,sort,subset,ctrl); 1227d5560a0bSSatish Balay /* El::Print(w, "Eigenvalues"); */ 12285dd12673SHong Zhang 1229874e4880SHong Zhang /* Wrap w and X into PETSc's MATMATELEMENTAL matrices */ 1230d59fe0e1SHong Zhang ierr = PetscObjectGetComm((PetscObject)A,&comm);CHKERRQ(ierr); 1231d59fe0e1SHong Zhang ierr = MatCreate(comm,evec);CHKERRQ(ierr); 1232d59fe0e1SHong Zhang ierr = MatSetSizes(*evec,PETSC_DECIDE,PETSC_DECIDE,X.Height(),X.Width());CHKERRQ(ierr); 1233d59fe0e1SHong Zhang ierr = MatSetType(*evec,MATELEMENTAL);CHKERRQ(ierr); 1234d59fe0e1SHong Zhang ierr = MatSetFromOptions(*evec);CHKERRQ(ierr); 1235d59fe0e1SHong Zhang ierr = MatSetUp(*evec);CHKERRQ(ierr); 1236d59fe0e1SHong Zhang ierr = MatAssemblyBegin(*evec,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1237d59fe0e1SHong Zhang ierr = MatAssemblyEnd(*evec,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1238d59fe0e1SHong Zhang 1239874e4880SHong Zhang x = (Mat_Elemental*)(*evec)->data; 1240d59fe0e1SHong Zhang //delete x->emat; //-- memory leak??? 1241d59fe0e1SHong Zhang *x->emat = X; 1242d59fe0e1SHong Zhang 1243874e4880SHong Zhang ierr = MatCreate(comm,&EVAL);CHKERRQ(ierr); 1244874e4880SHong Zhang ierr = MatSetSizes(EVAL,PETSC_DECIDE,PETSC_DECIDE,w.Height(),w.Width());CHKERRQ(ierr); 1245874e4880SHong Zhang ierr = MatSetType(EVAL,MATELEMENTAL);CHKERRQ(ierr); 1246874e4880SHong Zhang ierr = MatSetFromOptions(EVAL);CHKERRQ(ierr); 1247874e4880SHong Zhang ierr = MatSetUp(EVAL);CHKERRQ(ierr); 1248874e4880SHong Zhang ierr = MatAssemblyBegin(EVAL,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1249874e4880SHong Zhang ierr = MatAssemblyEnd(EVAL,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 12502ef15aa2SHong Zhang e = (Mat_Elemental*)EVAL->data; 1251874e4880SHong Zhang *e->emat = w; //-- memory leak??? 1252874e4880SHong Zhang *evals = EVAL; 1253874e4880SHong Zhang 1254874e4880SHong Zhang #if defined(MV) 1255f82baa17SHong Zhang /* Test correctness norm = || - A*X + B*X*w || */ 1256d59fe0e1SHong Zhang { 1257f82baa17SHong Zhang PetscElemScalar alpha,beta; 1258d5560a0bSSatish Balay El::DistMatrix<PetscElemScalar> Y(*a->grid); //tmp matrix 1259f82baa17SHong Zhang alpha = 1.0; beta=0.0; 1260d5560a0bSSatish Balay El::Gemm(El::NORMAL,El::NORMAL,alpha,*b->emat,X,beta,Y); //Y = B*X 1261d5560a0bSSatish Balay El::DiagonalScale(El::RIGHT,El::NORMAL, w, Y); //Y = Y*w 1262f82baa17SHong Zhang alpha = -1.0; beta=1.0; 1263d5560a0bSSatish Balay El::Gemm(El::NORMAL,El::NORMAL,alpha,*a->emat,X,beta,Y); //Y = - A*X + B*X*w 1264f82baa17SHong Zhang 1265d5560a0bSSatish Balay PetscElemScalar norm = El::FrobeniusNorm(Y); 1266d59fe0e1SHong Zhang if ((*a->grid).Rank()==0) printf(" norm (- A*X + B*X*w) = %g\n",norm); 1267d59fe0e1SHong Zhang } 1268f82baa17SHong Zhang 1269874e4880SHong Zhang { 1270874e4880SHong Zhang PetscMPIInt rank; 1271874e4880SHong Zhang ierr = MPI_Comm_rank(comm,&rank); 1272874e4880SHong Zhang printf("w: [%d] [%d, %d %d] %d; X: %d %d\n",rank,w.DistRank(),w.ColRank(),w.RowRank(),w.LocalHeight(),X.LocalHeight(),X.LocalWidth()); 12735dd12673SHong Zhang } 1274874e4880SHong Zhang #endif 12751d08bef3SHong Zhang PetscFunctionReturn(0); 12761d08bef3SHong Zhang } 12771d08bef3SHong Zhang 12781d08bef3SHong Zhang #undef __FUNCT__ 1279d5560a0bSSatish Balay #define __FUNCT__ "MatElementalHermitianGenDefEig" 12801d08bef3SHong Zhang /*@ 12812ef15aa2SHong Zhang MatElementalHermitianGenDefEig - Compute the set of eigenvalues of the Hermitian-definite matrix pencil determined by the subset structure 12821d08bef3SHong Zhang 12831d08bef3SHong Zhang Logically Collective on Mat 12841d08bef3SHong Zhang 12851d08bef3SHong Zhang Level: beginner 12861d08bef3SHong Zhang 12872ef15aa2SHong Zhang References: Elemental Users' Guide 12881d08bef3SHong Zhang @*/ 12892ef15aa2SHong Zhang PetscErrorCode MatElementalHermitianGenDefEig(El::Pencil type,El::UpperOrLower uplo,Mat A,Mat B,Mat *evals,Mat *evec,El::SortType sort,El::HermitianEigSubset<PetscElemScalar> subset,const El::HermitianEigCtrl<PetscElemScalar> ctrl) 12901d08bef3SHong Zhang { 12911d08bef3SHong Zhang PetscErrorCode ierr; 12921d08bef3SHong Zhang 12931d08bef3SHong Zhang PetscFunctionBegin; 12942ef15aa2SHong Zhang ierr = PetscTryMethod(A,"MatElementalHermitianGenDefEig_C",(El::Pencil,El::UpperOrLower,Mat,Mat,Mat*,Mat*,El::SortType,El::HermitianEigSubset<PetscElemScalar>,const El::HermitianEigCtrl<PetscElemScalar>),(type,uplo,A,B,evals,evec,sort,subset,ctrl));CHKERRQ(ierr); 12951d08bef3SHong Zhang PetscFunctionReturn(0); 12961d08bef3SHong Zhang } 12971d08bef3SHong Zhang 129840d92e34SHong Zhang /* -------------------------------------------------------------------*/ 129940d92e34SHong Zhang static struct _MatOps MatOps_Values = { 130040d92e34SHong Zhang MatSetValues_Elemental, 130140d92e34SHong Zhang 0, 130240d92e34SHong Zhang 0, 130340d92e34SHong Zhang MatMult_Elemental, 130440d92e34SHong Zhang /* 4*/ MatMultAdd_Elemental, 13059426833fSXuan Zhou MatMultTranspose_Elemental, 1306e883f9d5SXuan Zhou MatMultTransposeAdd_Elemental, 130740d92e34SHong Zhang MatSolve_Elemental, 1308df311e6cSXuan Zhou MatSolveAdd_Elemental, 13092ef15aa2SHong Zhang 0, 13102ef15aa2SHong Zhang /*10*/ 0, 131140d92e34SHong Zhang MatLUFactor_Elemental, 131240d92e34SHong Zhang MatCholeskyFactor_Elemental, 131340d92e34SHong Zhang 0, 131440d92e34SHong Zhang MatTranspose_Elemental, 131540d92e34SHong Zhang /*15*/ MatGetInfo_Elemental, 131640d92e34SHong Zhang 0, 131761119200SXuan Zhou MatGetDiagonal_Elemental, 1318ade3cc5eSXuan Zhou MatDiagonalScale_Elemental, 131940d92e34SHong Zhang MatNorm_Elemental, 132040d92e34SHong Zhang /*20*/ MatAssemblyBegin_Elemental, 132140d92e34SHong Zhang MatAssemblyEnd_Elemental, 132232d7a744SHong Zhang MatSetOption_Elemental, 132340d92e34SHong Zhang MatZeroEntries_Elemental, 132440d92e34SHong Zhang /*24*/ 0, 132540d92e34SHong Zhang MatLUFactorSymbolic_Elemental, 132640d92e34SHong Zhang MatLUFactorNumeric_Elemental, 132740d92e34SHong Zhang MatCholeskyFactorSymbolic_Elemental, 132840d92e34SHong Zhang MatCholeskyFactorNumeric_Elemental, 132940d92e34SHong Zhang /*29*/ MatSetUp_Elemental, 133040d92e34SHong Zhang 0, 133140d92e34SHong Zhang 0, 133240d92e34SHong Zhang 0, 133340d92e34SHong Zhang 0, 1334df311e6cSXuan Zhou /*34*/ MatDuplicate_Elemental, 133540d92e34SHong Zhang 0, 133640d92e34SHong Zhang 0, 133740d92e34SHong Zhang 0, 133840d92e34SHong Zhang 0, 133940d92e34SHong Zhang /*39*/ MatAXPY_Elemental, 134040d92e34SHong Zhang 0, 134140d92e34SHong Zhang 0, 134240d92e34SHong Zhang 0, 134340d92e34SHong Zhang MatCopy_Elemental, 134440d92e34SHong Zhang /*44*/ 0, 134540d92e34SHong Zhang MatScale_Elemental, 13467d68702bSBarry Smith MatShift_Basic, 134740d92e34SHong Zhang 0, 134840d92e34SHong Zhang 0, 134940d92e34SHong Zhang /*49*/ 0, 135040d92e34SHong Zhang 0, 135140d92e34SHong Zhang 0, 135240d92e34SHong Zhang 0, 135340d92e34SHong Zhang 0, 135440d92e34SHong Zhang /*54*/ 0, 135540d92e34SHong Zhang 0, 135640d92e34SHong Zhang 0, 135740d92e34SHong Zhang 0, 135840d92e34SHong Zhang 0, 135940d92e34SHong Zhang /*59*/ 0, 136040d92e34SHong Zhang MatDestroy_Elemental, 136140d92e34SHong Zhang MatView_Elemental, 136240d92e34SHong Zhang 0, 136340d92e34SHong Zhang 0, 136440d92e34SHong Zhang /*64*/ 0, 136540d92e34SHong Zhang 0, 136640d92e34SHong Zhang 0, 136740d92e34SHong Zhang 0, 136840d92e34SHong Zhang 0, 136940d92e34SHong Zhang /*69*/ 0, 137040d92e34SHong Zhang 0, 13712ef0cf24SXuan Zhou MatConvert_Elemental_Dense, 137240d92e34SHong Zhang 0, 137340d92e34SHong Zhang 0, 137440d92e34SHong Zhang /*74*/ 0, 137540d92e34SHong Zhang 0, 137640d92e34SHong Zhang 0, 137740d92e34SHong Zhang 0, 137840d92e34SHong Zhang 0, 137940d92e34SHong Zhang /*79*/ 0, 138040d92e34SHong Zhang 0, 138140d92e34SHong Zhang 0, 138240d92e34SHong Zhang 0, 13837b30e0f1SHong Zhang MatLoad_Elemental, 138440d92e34SHong Zhang /*84*/ 0, 138540d92e34SHong Zhang 0, 138640d92e34SHong Zhang 0, 138740d92e34SHong Zhang 0, 138840d92e34SHong Zhang 0, 138940d92e34SHong Zhang /*89*/ MatMatMult_Elemental, 139040d92e34SHong Zhang MatMatMultSymbolic_Elemental, 139140d92e34SHong Zhang MatMatMultNumeric_Elemental, 139240d92e34SHong Zhang 0, 139340d92e34SHong Zhang 0, 139440d92e34SHong Zhang /*94*/ 0, 1395df311e6cSXuan Zhou MatMatTransposeMult_Elemental, 1396df311e6cSXuan Zhou MatMatTransposeMultSymbolic_Elemental, 1397df311e6cSXuan Zhou MatMatTransposeMultNumeric_Elemental, 139840d92e34SHong Zhang 0, 139940d92e34SHong Zhang /*99*/ 0, 140040d92e34SHong Zhang 0, 140140d92e34SHong Zhang 0, 1402dfcb0403SXuan Zhou MatConjugate_Elemental, 140340d92e34SHong Zhang 0, 140440d92e34SHong Zhang /*104*/0, 140540d92e34SHong Zhang 0, 140640d92e34SHong Zhang 0, 140740d92e34SHong Zhang 0, 140840d92e34SHong Zhang 0, 140940d92e34SHong Zhang /*109*/MatMatSolve_Elemental, 141040d92e34SHong Zhang 0, 141140d92e34SHong Zhang 0, 141240d92e34SHong Zhang 0, 141340d92e34SHong Zhang 0, 141440d92e34SHong Zhang /*114*/0, 141540d92e34SHong Zhang 0, 141640d92e34SHong Zhang 0, 141740d92e34SHong Zhang 0, 141840d92e34SHong Zhang 0, 141940d92e34SHong Zhang /*119*/0, 14204a29722dSXuan Zhou MatHermitianTranspose_Elemental, 142140d92e34SHong Zhang 0, 142240d92e34SHong Zhang 0, 142340d92e34SHong Zhang 0, 142440d92e34SHong Zhang /*124*/0, 142540d92e34SHong Zhang 0, 142640d92e34SHong Zhang 0, 142740d92e34SHong Zhang 0, 142840d92e34SHong Zhang 0, 142940d92e34SHong Zhang /*129*/0, 143040d92e34SHong Zhang 0, 143140d92e34SHong Zhang 0, 143240d92e34SHong Zhang 0, 143340d92e34SHong Zhang 0, 143440d92e34SHong Zhang /*134*/0, 143540d92e34SHong Zhang 0, 143640d92e34SHong Zhang 0, 143740d92e34SHong Zhang 0, 143840d92e34SHong Zhang 0 143940d92e34SHong Zhang }; 144040d92e34SHong Zhang 1441ed36708cSHong Zhang /*MC 1442ed36708cSHong Zhang MATELEMENTAL = "elemental" - A matrix type for dense matrices using the Elemental package 1443ed36708cSHong Zhang 1444c2b89b5dSBarry Smith Use ./configure --download-elemental to install PETSc to use Elemental 1445c2b89b5dSBarry Smith 1446c2b89b5dSBarry Smith Use -pc_type lu -pc_factor_mat_solver_package elemental to us this direct solver 1447c2b89b5dSBarry Smith 1448ed36708cSHong Zhang Options Database Keys: 14495cc86fc1SJed Brown + -mat_type elemental - sets the matrix type to "elemental" during a call to MatSetFromOptions() 14505cc86fc1SJed Brown - -mat_elemental_grid_height - sets Grid Height for 2D cyclic ordering of internal matrix 1451ed36708cSHong Zhang 1452ed36708cSHong Zhang Level: beginner 1453ed36708cSHong Zhang 14545cb544a0SHong Zhang .seealso: MATDENSE 1455ed36708cSHong Zhang M*/ 14564a29722dSXuan Zhou 1457db31f6deSJed Brown #undef __FUNCT__ 1458db31f6deSJed Brown #define __FUNCT__ "MatCreate_Elemental" 14598cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_Elemental(Mat A) 1460db31f6deSJed Brown { 1461db31f6deSJed Brown Mat_Elemental *a; 1462db31f6deSJed Brown PetscErrorCode ierr; 14635682a260SJack Poulson PetscBool flg,flg1; 14645e9f5b67SHong Zhang Mat_Elemental_Grid *commgrid; 14655e9f5b67SHong Zhang MPI_Comm icomm; 14665682a260SJack Poulson PetscInt optv1; 1467db31f6deSJed Brown 1468db31f6deSJed Brown PetscFunctionBegin; 1469607a6623SBarry Smith ierr = PetscElementalInitializePackage();CHKERRQ(ierr); 147040d92e34SHong Zhang ierr = PetscMemcpy(A->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 147140d92e34SHong Zhang A->insertmode = NOT_SET_VALUES; 1472db31f6deSJed Brown 1473b00a9115SJed Brown ierr = PetscNewLog(A,&a);CHKERRQ(ierr); 1474db31f6deSJed Brown A->data = (void*)a; 1475db31f6deSJed Brown 1476db31f6deSJed Brown /* Set up the elemental matrix */ 1477e8146892SSatish Balay El::mpi::Comm cxxcomm(PetscObjectComm((PetscObject)A)); 14785e9f5b67SHong Zhang 14795e9f5b67SHong Zhang /* Grid needs to be shared between multiple Mats on the same communicator, implement by attribute caching on the MPI_Comm */ 14805e9f5b67SHong Zhang if (Petsc_Elemental_keyval == MPI_KEYVAL_INVALID) { 1481180a43e4SHong Zhang ierr = MPI_Keyval_create(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); 1482a000e53bSHong Zhang /* ierr = MPI_Comm_create_Keyval(MPI_NULL_COPY_FN,MPI_NULL_DELETE_FN,&Petsc_Elemental_keyval,(void*)0); -- new version? */ 14835e9f5b67SHong Zhang } 14840c18141cSBarry Smith ierr = PetscCommDuplicate(cxxcomm.comm,&icomm,NULL);CHKERRQ(ierr); 14855e9f5b67SHong Zhang ierr = MPI_Attr_get(icomm,Petsc_Elemental_keyval,(void**)&commgrid,(int*)&flg);CHKERRQ(ierr); 14865e9f5b67SHong Zhang if (!flg) { 1487b00a9115SJed Brown ierr = PetscNewLog(A,&commgrid);CHKERRQ(ierr); 14885cb544a0SHong Zhang 1489ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)A),((PetscObject)A)->prefix,"Elemental Options","Mat");CHKERRQ(ierr); 1490e8146892SSatish Balay /* displayed default grid sizes (CommSize,1) are set by us arbitrarily until El::Grid() is called */ 1491e8146892SSatish Balay ierr = PetscOptionsInt("-mat_elemental_grid_height","Grid Height","None",El::mpi::Size(cxxcomm),&optv1,&flg1);CHKERRQ(ierr); 14925682a260SJack Poulson if (flg1) { 1493e8146892SSatish Balay if (El::mpi::Size(cxxcomm) % optv1 != 0) { 1494e8146892SSatish Balay SETERRQ2(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_INCOMP,"Grid Height %D must evenly divide CommSize %D",optv1,(PetscInt)El::mpi::Size(cxxcomm)); 1495ed667823SXuan Zhou } 1496e8146892SSatish Balay commgrid->grid = new El::Grid(cxxcomm,optv1); /* use user-provided grid height */ 14972ef0cf24SXuan Zhou } else { 1498e8146892SSatish Balay commgrid->grid = new El::Grid(cxxcomm); /* use Elemental default grid sizes */ 1499da0640a4SHong Zhang /* printf("new commgrid->grid = %p\n",commgrid->grid); -- memory leak revealed by valgrind? */ 1500ed667823SXuan Zhou } 15015e9f5b67SHong Zhang commgrid->grid_refct = 1; 15025e9f5b67SHong Zhang ierr = MPI_Attr_put(icomm,Petsc_Elemental_keyval,(void*)commgrid);CHKERRQ(ierr); 15035cb544a0SHong Zhang PetscOptionsEnd(); 15045e9f5b67SHong Zhang } else { 15055e9f5b67SHong Zhang commgrid->grid_refct++; 15065e9f5b67SHong Zhang } 15075e9f5b67SHong Zhang ierr = PetscCommDestroy(&icomm);CHKERRQ(ierr); 15085e9f5b67SHong Zhang a->grid = commgrid->grid; 1509e8146892SSatish Balay a->emat = new El::DistMatrix<PetscElemScalar>(*a->grid); 1510dc1b4460SJack Poulson a->pivot = new El::DistMatrix<PetscInt,El::VC,El::STAR>(*a->grid); 151132d7a744SHong Zhang a->roworiented = PETSC_TRUE; 1512db31f6deSJed Brown 1513bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)A,"MatGetOwnershipIS_C",MatGetOwnershipIS_Elemental);CHKERRQ(ierr); 1514d5560a0bSSatish Balay ierr = PetscObjectComposeFunction((PetscObject)A,"MatElementalHermitianGenDefEig_C",MatElementalHermitianGenDefEig_Elemental);CHKERRQ(ierr); 1515db31f6deSJed Brown 1516db31f6deSJed Brown ierr = PetscObjectChangeTypeName((PetscObject)A,MATELEMENTAL);CHKERRQ(ierr); 1517db31f6deSJed Brown PetscFunctionReturn(0); 1518db31f6deSJed Brown } 1519