15b89ad90SMark F. Adams /* 25b89ad90SMark F. Adams GAMG geometric-algebric multiogrid PC - Mark Adams 2011 35b89ad90SMark F. Adams */ 45ef31b24SMark F. Adams #include <../src/ksp/pc/impls/gamg/gamg.h> 558471d46SMark F. Adams #include "private/matimpl.h" /*I "petscmat.h" I*/ 6f96513f1SMatthew G Knepley 7b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 8b4fbaa2aSMark F. Adams PetscLogEvent gamg_setup_events[NUM_SET]; 9b4fbaa2aSMark F. Adams #endif 10b4fbaa2aSMark F. Adams #define GAMG_MAXLEVELS 30 11b4fbaa2aSMark F. Adams 12b8fd24d8SMark F. Adams /*#define GAMG_STAGES*/ 13b4fbaa2aSMark F. Adams #if (defined PETSC_USE_LOG && defined GAMG_STAGES) 14b4fbaa2aSMark F. Adams static PetscLogStage gamg_stages[GAMG_MAXLEVELS]; 15b4fbaa2aSMark F. Adams #endif 16f96513f1SMatthew G Knepley 175b89ad90SMark F. Adams /* Private context for the GAMG preconditioner */ 185b89ad90SMark F. Adams typedef struct gamg_TAG { 195b89ad90SMark F. Adams PetscInt m_dim; 205b89ad90SMark F. Adams PetscInt m_Nlevels; 215b89ad90SMark F. Adams PetscInt m_data_sz; 22d3d6bff4SMark F. Adams PetscInt m_data_rows; 23d3d6bff4SMark F. Adams PetscInt m_data_cols; 24*03a628feSMark F. Adams PetscInt m_count; 25d3d6bff4SMark F. Adams PetscBool m_useSA; 265b89ad90SMark F. Adams PetscReal *m_data; /* blocked vector of vertex data on fine grid (coordinates) */ 275b89ad90SMark F. Adams } PC_GAMG; 285b89ad90SMark F. Adams 295b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */ 305b89ad90SMark F. Adams /* 315b89ad90SMark F. Adams PCSetCoordinates_GAMG 325b89ad90SMark F. Adams 335b89ad90SMark F. Adams Input Parameter: 345b89ad90SMark F. Adams . pc - the preconditioner context 355b89ad90SMark F. Adams */ 36a92563c5SMark F. Adams EXTERN_C_BEGIN 375b89ad90SMark F. Adams #undef __FUNCT__ 385b89ad90SMark F. Adams #define __FUNCT__ "PCSetCoordinates_GAMG" 39eb07cef2SMark F. Adams PetscErrorCode PCSetCoordinates_GAMG( PC a_pc, PetscInt a_ndm, PetscReal *a_coords ) 405b89ad90SMark F. Adams { 41eb07cef2SMark F. Adams PC_MG *mg = (PC_MG*)a_pc->data; 425b89ad90SMark F. Adams PC_GAMG *pc_gamg = (PC_GAMG*)mg->innerctx; 436c237d78SBarry Smith PetscErrorCode ierr; 44d3d6bff4SMark F. Adams PetscInt arrsz,bs,my0,kk,ii,jj,nloc,Iend; 45038e3b61SMark F. Adams Mat Amat = a_pc->pmat; 46d3d6bff4SMark F. Adams PetscBool flag; 47785cba28SMark F. Adams char str[64]; 485b89ad90SMark F. Adams 495b89ad90SMark F. Adams PetscFunctionBegin; 5058471d46SMark F. Adams PetscValidHeaderSpecific( Amat, MAT_CLASSID, 1 ); 51038e3b61SMark F. Adams ierr = MatGetBlockSize( Amat, &bs ); CHKERRQ( ierr ); 52d3d6bff4SMark F. Adams ierr = MatGetOwnershipRange( Amat, &my0, &Iend ); CHKERRQ(ierr); 53d3d6bff4SMark F. Adams nloc = (Iend-my0)/bs; 54d3d6bff4SMark F. Adams if((Iend-my0)%bs!=0) SETERRQ1(((PetscObject)Amat)->comm,PETSC_ERR_ARG_WRONG, "Bad local size %d.",nloc); 55038e3b61SMark F. Adams 56785cba28SMark F. Adams ierr = PetscOptionsGetString(PETSC_NULL,"-pc_gamg_type",str,64,&flag); CHKERRQ( ierr ); 57d3d6bff4SMark F. Adams pc_gamg->m_useSA = (PetscBool)(flag && strcmp(str,"sa") == 0); 58038e3b61SMark F. Adams 59d3d6bff4SMark F. Adams pc_gamg->m_data_rows = 1; 60d3d6bff4SMark F. Adams if(a_coords == 0) pc_gamg->m_useSA = PETSC_TRUE; /* use SA if no data */ 61d3d6bff4SMark F. Adams if( !pc_gamg->m_useSA ) pc_gamg->m_data_cols = a_ndm; /* coordinates */ 62038e3b61SMark F. Adams else{ /* SA: null space vectors */ 63d3d6bff4SMark F. Adams if(a_coords != 0 && bs==1 ) pc_gamg->m_data_cols = 1; /* scalar w/ coords and SA (not needed) */ 64d3d6bff4SMark F. Adams else if(a_coords != 0) pc_gamg->m_data_cols = (a_ndm==2 ? 3 : 6); /* elasticity */ 65d3d6bff4SMark F. Adams else pc_gamg->m_data_cols = bs; /* no data, force SA with constant null space vectors */ 66d3d6bff4SMark F. Adams pc_gamg->m_data_rows = bs; 67038e3b61SMark F. Adams } 68d3d6bff4SMark F. Adams arrsz = nloc*pc_gamg->m_data_rows*pc_gamg->m_data_cols; 695b89ad90SMark F. Adams 70038e3b61SMark F. Adams /* create data - syntactic sugar that should be refactored at some point */ 716e3e101aSMark F. Adams if (pc_gamg->m_data==0 || (pc_gamg->m_data_sz != arrsz)) { 725b89ad90SMark F. Adams ierr = PetscFree( pc_gamg->m_data ); CHKERRQ(ierr); 73eb07cef2SMark F. Adams ierr = PetscMalloc((arrsz+1)*sizeof(double), &pc_gamg->m_data ); CHKERRQ(ierr); 745b89ad90SMark F. Adams } 75038e3b61SMark F. Adams for(kk=0;kk<arrsz;kk++)pc_gamg->m_data[kk] = -999.; 76eb07cef2SMark F. Adams pc_gamg->m_data[arrsz] = -99.; 77038e3b61SMark F. Adams /* copy data in - column oriented */ 78d3d6bff4SMark F. Adams if( pc_gamg->m_useSA ) { 79d3d6bff4SMark F. Adams const PetscInt M = Iend - my0; 80038e3b61SMark F. Adams for(kk=0;kk<nloc;kk++){ 81038e3b61SMark F. Adams PetscReal *data = &pc_gamg->m_data[kk*bs]; 82d3d6bff4SMark F. Adams if( pc_gamg->m_data_cols==1 ) *data = 1.0; 83038e3b61SMark F. Adams else { 84038e3b61SMark F. Adams for(ii=0;ii<bs;ii++) 85038e3b61SMark F. Adams for(jj=0;jj<bs;jj++) 86038e3b61SMark F. Adams if(ii==jj)data[ii*M + jj] = 1.0; /* translational modes */ 87038e3b61SMark F. Adams else data[ii*M + jj] = 0.0; 88eb07cef2SMark F. Adams if( a_coords != 0 ) { 89038e3b61SMark F. Adams if( a_ndm == 2 ){ /* rotational modes */ 90038e3b61SMark F. Adams data += 2*M; 91038e3b61SMark F. Adams data[0] = -a_coords[2*kk+1]; 92038e3b61SMark F. Adams data[1] = a_coords[2*kk]; 935b89ad90SMark F. Adams } 94eb07cef2SMark F. Adams else { 95038e3b61SMark F. Adams data += 3*M; 96038e3b61SMark F. Adams data[0] = 0.0; data[M+0] = a_coords[3*kk+2]; data[2*M+0] = -a_coords[3*kk+1]; 97038e3b61SMark F. Adams data[1] = -a_coords[3*kk+2]; data[M+1] = 0.0; data[2*M+1] = a_coords[3*kk]; 98038e3b61SMark F. Adams data[2] = a_coords[3*kk+1]; data[M+2] = -a_coords[3*kk]; data[2*M+2] = 0.0; 99038e3b61SMark F. Adams } 100eb07cef2SMark F. Adams } 101eb07cef2SMark F. Adams } 102eb07cef2SMark F. Adams } 103eb07cef2SMark F. Adams } 104eb07cef2SMark F. Adams else { 105038e3b61SMark F. Adams for( kk = 0 ; kk < nloc ; kk++ ){ 106038e3b61SMark F. Adams for( ii = 0 ; ii < a_ndm ; ii++ ) { 107038e3b61SMark F. Adams pc_gamg->m_data[ii*nloc + kk] = a_coords[kk*a_ndm + ii]; 108eb07cef2SMark F. Adams } 109eb07cef2SMark F. Adams } 110038e3b61SMark F. Adams } 111eb07cef2SMark F. Adams assert(pc_gamg->m_data[arrsz] == -99.); 112038e3b61SMark F. Adams 1135b89ad90SMark F. Adams pc_gamg->m_data_sz = arrsz; 114eb07cef2SMark F. Adams pc_gamg->m_dim = a_ndm; 1155b89ad90SMark F. Adams 1165b89ad90SMark F. Adams PetscFunctionReturn(0); 1175b89ad90SMark F. Adams } 118a92563c5SMark F. Adams EXTERN_C_END 1195b89ad90SMark F. Adams 120d3d6bff4SMark F. Adams 121d3d6bff4SMark F. Adams /* -----------------------------------------------------------------------------*/ 122d3d6bff4SMark F. Adams #undef __FUNCT__ 123d3d6bff4SMark F. Adams #define __FUNCT__ "PCReset_GAMG" 124d3d6bff4SMark F. Adams PetscErrorCode PCReset_GAMG(PC pc) 125d3d6bff4SMark F. Adams { 126d3d6bff4SMark F. Adams PetscErrorCode ierr; 127d3d6bff4SMark F. Adams PC_MG *mg = (PC_MG*)pc->data; 128d3d6bff4SMark F. Adams PC_GAMG *pc_gamg = (PC_GAMG*)mg->innerctx; 129d3d6bff4SMark F. Adams 130d3d6bff4SMark F. Adams PetscFunctionBegin; 13158471d46SMark F. Adams if( pc_gamg->m_data != 0 ) { /* this should not happen, cleaned up in SetUp */ 132d3d6bff4SMark F. Adams ierr = PetscFree(pc_gamg->m_data); CHKERRQ(ierr); 13358471d46SMark F. Adams } 134d3d6bff4SMark F. Adams pc_gamg->m_data = 0; pc_gamg->m_data_sz = 0; 135d3d6bff4SMark F. Adams PetscFunctionReturn(0); 136d3d6bff4SMark F. Adams } 137d3d6bff4SMark F. Adams 1385b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */ 1395b89ad90SMark F. Adams /* 14011e60469SMark F. Adams partitionLevel 1415b89ad90SMark F. Adams 1425b89ad90SMark F. Adams Input Parameter: 1433530afc2SMark F. Adams . a_Amat_fine - matrix on this fine (k) level 144d3d6bff4SMark F. Adams . a_ndata_rows - size of data to move (coarse grid) 145d3d6bff4SMark F. Adams . a_ndata_cols - size of data to move (coarse grid) 1463530afc2SMark F. Adams In/Output Parameter: 1473530afc2SMark F. Adams . a_P_inout - prolongation operator to the next level (k-1) 148eb07cef2SMark F. Adams . a_coarse_data - data that need to be moved 149afc97cdcSMark F. Adams . a_nactive_proc - number of active procs 15011e60469SMark F. Adams Output Parameter: 1513530afc2SMark F. Adams . a_Amat_crs - coarse matrix that is created (k-1) 1525b89ad90SMark F. Adams */ 1535cb416c2SMark F. Adams 15458471d46SMark F. Adams #define MIN_EQ_PROC 600 155cc4ca5ccSMark F. Adams #define TOP_GRID_LIM 2*MIN_EQ_PROC /* this will happen anyway */ 15622063be5SMark F. Adams 1575b89ad90SMark F. Adams #undef __FUNCT__ 15811e60469SMark F. Adams #define __FUNCT__ "partitionLevel" 1593530afc2SMark F. Adams PetscErrorCode partitionLevel( Mat a_Amat_fine, 160d3d6bff4SMark F. Adams PetscInt a_ndata_rows, 161d3d6bff4SMark F. Adams PetscInt a_ndata_cols, 162038e3b61SMark F. Adams PetscInt a_cbs, 1633530afc2SMark F. Adams Mat *a_P_inout, 164eb07cef2SMark F. Adams PetscReal **a_coarse_data, 165afc97cdcSMark F. Adams PetscMPIInt *a_nactive_proc, 1663530afc2SMark F. Adams Mat *a_Amat_crs 16711e60469SMark F. Adams ) 1685b89ad90SMark F. Adams { 1695b89ad90SMark F. Adams PetscErrorCode ierr; 170038e3b61SMark F. Adams Mat Cmat,Pnew,Pold=*a_P_inout; 17111e60469SMark F. Adams IS new_indices,isnum; 1723530afc2SMark F. Adams MPI_Comm wcomm = ((PetscObject)a_Amat_fine)->comm; 173fa31c87dSMark F. Adams PetscMPIInt mype,npe,new_npe,nactive; 174fa31c87dSMark F. Adams PetscInt neq,NN,Istart,Iend,Istart0,Iend0,ncrs_new,ncrs0; 175737a81a9SMark F. Adams PetscBool flag = PETSC_FALSE; 1765b89ad90SMark F. Adams 1775b89ad90SMark F. Adams PetscFunctionBegin; 17811e60469SMark F. Adams ierr = MPI_Comm_rank( wcomm, &mype ); CHKERRQ(ierr); 17911e60469SMark F. Adams ierr = MPI_Comm_size( wcomm, &npe ); CHKERRQ(ierr); 18011e60469SMark F. Adams /* RAP */ 181038e3b61SMark F. Adams ierr = MatPtAP( a_Amat_fine, Pold, MAT_INITIAL_MATRIX, 2.0, &Cmat ); CHKERRQ(ierr); 182737a81a9SMark F. Adams 183038e3b61SMark F. Adams ierr = MatSetBlockSize( Cmat, a_cbs ); CHKERRQ(ierr); 184038e3b61SMark F. Adams ierr = MatGetOwnershipRange( Cmat, &Istart0, &Iend0 ); CHKERRQ(ierr); 185038e3b61SMark F. Adams ncrs0 = (Iend0-Istart0)/a_cbs; assert((Iend0-Istart0)%a_cbs == 0); 186038e3b61SMark F. Adams 187737a81a9SMark F. Adams ierr = PetscOptionsHasName(PETSC_NULL,"-pc_gamg_avoid_repartitioning",&flag); 188737a81a9SMark F. Adams CHKERRQ( ierr ); 18922063be5SMark F. Adams if( flag ) { 19022063be5SMark F. Adams *a_Amat_crs = Cmat; /* output */ 19122063be5SMark F. Adams } 19222063be5SMark F. Adams else { 19322063be5SMark F. Adams /* Repartition Cmat_{k} and move colums of P^{k}_{k-1} and coordinates accordingly */ 1945ef31b24SMark F. Adams Mat adj; 19522063be5SMark F. Adams const PetscInt *idx,data_sz=a_ndata_rows*a_ndata_cols; 196b4fbaa2aSMark F. Adams const PetscInt stride0=ncrs0*a_ndata_rows; 19792a756f0SMark F. Adams PetscInt is_sz,*isnewproc_idx,ii,jj,kk,strideNew,*tidx; 198c9a0b8beSMark F. Adams /* create sub communicator */ 199c9a0b8beSMark F. Adams MPI_Comm cm,new_comm; 200afc97cdcSMark F. Adams MPI_Group wg, g2; 201fd3c6afaSMark F. Adams PetscInt *counts,inpe; 202fd3c6afaSMark F. Adams PetscMPIInt *ranks; 20322063be5SMark F. Adams IS isscat; 20422063be5SMark F. Adams PetscScalar *array; 20522063be5SMark F. Adams Vec src_crd, dest_crd; 20622063be5SMark F. Adams PetscReal *data = *a_coarse_data; 20722063be5SMark F. Adams VecScatter vecscat; 208b4fbaa2aSMark F. Adams IS isnewproc; 209e33ef3b1SMark F. Adams 21022063be5SMark F. Adams /* get number of PEs to make active, reduce */ 21122063be5SMark F. Adams ierr = MatGetSize( Cmat, &neq, &NN );CHKERRQ(ierr); 21222063be5SMark F. Adams new_npe = neq/MIN_EQ_PROC; /* hardwire min. number of eq/proc */ 21322063be5SMark F. Adams if( new_npe == 0 || neq < TOP_GRID_LIM ) new_npe = 1; 21422063be5SMark F. Adams else if (new_npe >= *a_nactive_proc ) new_npe = *a_nactive_proc; /* no change, rare */ 21522063be5SMark F. Adams 21692a756f0SMark F. Adams ierr = PetscMalloc( npe*sizeof(PetscMPIInt), &ranks ); CHKERRQ(ierr); 217fd3c6afaSMark F. Adams ierr = PetscMalloc( npe*sizeof(PetscInt), &counts ); CHKERRQ(ierr); 21892a756f0SMark F. Adams 219fd3c6afaSMark F. Adams ierr = MPI_Allgather( &ncrs0, 1, MPIU_INT, counts, 1, MPIU_INT, wcomm ); CHKERRQ(ierr); 220afc97cdcSMark F. Adams assert(counts[mype]==ncrs0); 221afc97cdcSMark F. Adams /* count real active pes */ 22222063be5SMark F. Adams for( nactive = jj = 0 ; jj < npe ; jj++) { 223afc97cdcSMark F. Adams if( counts[jj] != 0 ) { 22422063be5SMark F. Adams ranks[nactive++] = jj; 225afc97cdcSMark F. Adams } 226afc97cdcSMark F. Adams } 22722063be5SMark F. Adams assert(nactive>=new_npe); 22822063be5SMark F. Adams 22958471d46SMark F. Adams #ifdef VERBOSE 23022063be5SMark F. Adams PetscPrintf(PETSC_COMM_WORLD,"\t[%d]%s npe (active): %d --> %d. new npe = %d, neq = %d\n",mype,__FUNCT__,*a_nactive_proc,nactive,new_npe,neq); 23158471d46SMark F. Adams #endif 23258471d46SMark F. Adams 23322063be5SMark F. Adams *a_nactive_proc = new_npe; /* output */ 2342f03bc48SMark F. Adams 235afc97cdcSMark F. Adams ierr = MPI_Comm_group( wcomm, &wg ); CHKERRQ(ierr); 23622063be5SMark F. Adams ierr = MPI_Group_incl( wg, nactive, ranks, &g2 ); CHKERRQ(ierr); 237afc97cdcSMark F. Adams ierr = MPI_Comm_create( wcomm, g2, &cm ); CHKERRQ(ierr); 238b4fbaa2aSMark F. Adams 239afc97cdcSMark F. Adams if( cm != MPI_COMM_NULL ) { 240b4fbaa2aSMark F. Adams assert(ncrs0 != 0); 241c9a0b8beSMark F. Adams ierr = PetscCommDuplicate( cm, &new_comm, PETSC_NULL ); CHKERRQ(ierr); 242c9a0b8beSMark F. Adams ierr = MPI_Comm_free( &cm ); CHKERRQ(ierr); 243afc97cdcSMark F. Adams } 244b4fbaa2aSMark F. Adams else assert(ncrs0 == 0); 245b4fbaa2aSMark F. Adams 246afc97cdcSMark F. Adams ierr = MPI_Group_free( &wg ); CHKERRQ(ierr); 247afc97cdcSMark F. Adams ierr = MPI_Group_free( &g2 ); CHKERRQ(ierr); 248c9a0b8beSMark F. Adams 2495ef31b24SMark F. Adams /* MatPartitioningApply call MatConvert, which is collective */ 250b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 251b4fbaa2aSMark F. Adams ierr = PetscLogEventBegin(gamg_setup_events[SET12],0,0,0,0);CHKERRQ(ierr); 252b4fbaa2aSMark F. Adams #endif 253038e3b61SMark F. Adams if( a_cbs == 1) { 254038e3b61SMark F. Adams ierr = MatConvert( Cmat, MATMPIADJ, MAT_INITIAL_MATRIX, &adj ); CHKERRQ(ierr); 255eb07cef2SMark F. Adams } 256eb07cef2SMark F. Adams else{ 257038e3b61SMark F. Adams /* make a scalar matrix to partition */ 258eb07cef2SMark F. Adams Mat tMat; 25958471d46SMark F. Adams PetscInt ncols,jj,Ii; 260b4fbaa2aSMark F. Adams const PetscScalar *vals; 261b4fbaa2aSMark F. Adams const PetscInt *idx; 26258471d46SMark F. Adams PetscInt *d_nnz; 263b4fbaa2aSMark F. Adams 26458471d46SMark F. Adams ierr = PetscMalloc( ncrs0*sizeof(PetscInt), &d_nnz ); CHKERRQ(ierr); 26558471d46SMark F. Adams for ( Ii = Istart0, jj = 0 ; Ii < Iend0 ; Ii += a_cbs, jj++ ) { 26658471d46SMark F. Adams ierr = MatGetRow(Cmat,Ii,&ncols,0,0); CHKERRQ(ierr); 26758471d46SMark F. Adams d_nnz[jj] = ncols/a_cbs; 26858471d46SMark F. Adams if( d_nnz[jj] > ncrs0 ) d_nnz[jj] = ncrs0; 26958471d46SMark F. Adams ierr = MatRestoreRow(Cmat,Ii,&ncols,0,0); CHKERRQ(ierr); 27058471d46SMark F. Adams } 2716876a03eSMark F. Adams 272eb07cef2SMark F. Adams ierr = MatCreateMPIAIJ( wcomm, ncrs0, ncrs0, 273eb07cef2SMark F. Adams PETSC_DETERMINE, PETSC_DETERMINE, 27458471d46SMark F. Adams 0, d_nnz, 0, d_nnz, 275eb07cef2SMark F. Adams &tMat ); 2766876a03eSMark F. Adams CHKERRQ(ierr); 27758471d46SMark F. Adams ierr = PetscFree( d_nnz ); CHKERRQ(ierr); 278eb07cef2SMark F. Adams 27922063be5SMark F. Adams for ( ii = Istart0; ii < Iend0; ii++ ) { 28022063be5SMark F. Adams PetscInt dest_row = ii/a_cbs; 28122063be5SMark F. Adams ierr = MatGetRow(Cmat,ii,&ncols,&idx,&vals); CHKERRQ(ierr); 282eb07cef2SMark F. Adams for( jj = 0 ; jj < ncols ; jj++ ){ 283038e3b61SMark F. Adams PetscInt dest_col = idx[jj]/a_cbs; 284eb07cef2SMark F. Adams PetscScalar v = 1.0; 285eb07cef2SMark F. Adams ierr = MatSetValues(tMat,1,&dest_row,1,&dest_col,&v,ADD_VALUES); CHKERRQ(ierr); 286eb07cef2SMark F. Adams } 28722063be5SMark F. Adams ierr = MatRestoreRow(Cmat,ii,&ncols,&idx,&vals); CHKERRQ(ierr); 288eb07cef2SMark F. Adams } 289eb07cef2SMark F. Adams ierr = MatAssemblyBegin(tMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 290eb07cef2SMark F. Adams ierr = MatAssemblyEnd(tMat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 291eb07cef2SMark F. Adams 292b4fbaa2aSMark F. Adams static int llev = 0; 293b4fbaa2aSMark F. Adams if( llev++ == -1 ) { 294b4fbaa2aSMark F. Adams PetscViewer viewer; char fname[32]; 295b4fbaa2aSMark F. Adams sprintf(fname,"part_mat_%d.mat",llev); 296b4fbaa2aSMark F. Adams PetscViewerBinaryOpen(wcomm,fname,FILE_MODE_WRITE,&viewer); 297b4fbaa2aSMark F. Adams ierr = MatView( tMat, viewer ); CHKERRQ(ierr); 298b4fbaa2aSMark F. Adams ierr = PetscViewerDestroy( &viewer ); 299b4fbaa2aSMark F. Adams } 300b4fbaa2aSMark F. Adams 301eb07cef2SMark F. Adams ierr = MatConvert( tMat, MATMPIADJ, MAT_INITIAL_MATRIX, &adj ); CHKERRQ(ierr); 302eb07cef2SMark F. Adams 303eb07cef2SMark F. Adams ierr = MatDestroy( &tMat ); CHKERRQ(ierr); 304eb07cef2SMark F. Adams } 305afc97cdcSMark F. Adams if( ncrs0 != 0 ){ 306b4fbaa2aSMark F. Adams const PetscInt *is_idx; 307b4fbaa2aSMark F. Adams MatPartitioning mpart; 3085ef31b24SMark F. Adams /* hack to fix global data that pmetis.c uses in 'adj' */ 30922063be5SMark F. Adams for( nactive = jj = 0 ; jj < npe ; jj++ ) { 310afc97cdcSMark F. Adams if( counts[jj] != 0 ) { 31122063be5SMark F. Adams adj->rmap->range[nactive++] = adj->rmap->range[jj]; 312afc97cdcSMark F. Adams } 3135ef31b24SMark F. Adams } 31422063be5SMark F. Adams adj->rmap->range[nactive] = adj->rmap->range[npe]; 3152f03bc48SMark F. Adams 3165ef31b24SMark F. Adams ierr = MatPartitioningCreate( new_comm, &mpart ); CHKERRQ(ierr); 3175ef31b24SMark F. Adams ierr = MatPartitioningSetAdjacency( mpart, adj ); CHKERRQ(ierr); 31811e60469SMark F. Adams ierr = MatPartitioningSetFromOptions( mpart ); CHKERRQ(ierr); 3195ef31b24SMark F. Adams ierr = MatPartitioningSetNParts( mpart, new_npe );CHKERRQ(ierr); 32011e60469SMark F. Adams ierr = MatPartitioningApply( mpart, &isnewproc ); CHKERRQ(ierr); 32111e60469SMark F. Adams ierr = MatPartitioningDestroy( &mpart ); CHKERRQ(ierr); 32222063be5SMark F. Adams 3235ef31b24SMark F. Adams /* collect IS info */ 3245ef31b24SMark F. Adams ierr = ISGetLocalSize( isnewproc, &is_sz ); CHKERRQ(ierr); 325038e3b61SMark F. Adams ierr = PetscMalloc( a_cbs*is_sz*sizeof(PetscInt), &isnewproc_idx ); CHKERRQ(ierr); 3265ef31b24SMark F. Adams ierr = ISGetIndices( isnewproc, &is_idx ); CHKERRQ(ierr); 327b4fbaa2aSMark F. Adams /* spread partitioning across machine - best way ??? */ 328ecd57171SMark F. Adams NN = 1; /*npe/new_npe;*/ 329b4fbaa2aSMark F. Adams for( kk = jj = 0 ; kk < is_sz ; kk++ ){ 330afc97cdcSMark F. Adams for( ii = 0 ; ii < a_cbs ; ii++, jj++ ) { 33122063be5SMark F. Adams isnewproc_idx[jj] = is_idx[kk] * NN; /* distribution */ 332eb07cef2SMark F. Adams } 3335ef31b24SMark F. Adams } 3345ef31b24SMark F. Adams ierr = ISRestoreIndices( isnewproc, &is_idx ); CHKERRQ(ierr); 3355ef31b24SMark F. Adams ierr = ISDestroy( &isnewproc ); CHKERRQ(ierr); 3364624a670SBarry Smith ierr = PetscCommDestroy( &new_comm ); CHKERRQ(ierr); 337b4fbaa2aSMark F. Adams 338b4fbaa2aSMark F. Adams is_sz *= a_cbs; 3395ef31b24SMark F. Adams } 3405ef31b24SMark F. Adams else{ 3415ef31b24SMark F. Adams isnewproc_idx = 0; 3425ef31b24SMark F. Adams is_sz = 0; 3435ef31b24SMark F. Adams } 3445ef31b24SMark F. Adams ierr = MatDestroy( &adj ); CHKERRQ(ierr); 3455ef31b24SMark F. Adams ierr = ISCreateGeneral( wcomm, is_sz, isnewproc_idx, PETSC_COPY_VALUES, &isnewproc ); 346afc97cdcSMark F. Adams if( isnewproc_idx != 0 ) { 3475ef31b24SMark F. Adams ierr = PetscFree( isnewproc_idx ); CHKERRQ(ierr); 3485ef31b24SMark F. Adams } 349e33ef3b1SMark F. Adams 35011e60469SMark F. Adams /* 35111e60469SMark F. Adams Create an index set from the isnewproc index set to indicate the mapping TO 35211e60469SMark F. Adams */ 35311e60469SMark F. Adams ierr = ISPartitioningToNumbering( isnewproc, &isnum ); CHKERRQ(ierr); 35411e60469SMark F. Adams /* 35511e60469SMark F. Adams Determine how many elements are assigned to each processor 35611e60469SMark F. Adams */ 357fd3c6afaSMark F. Adams inpe = npe; 358fd3c6afaSMark F. Adams ierr = ISPartitioningCount( isnewproc, inpe, counts ); CHKERRQ(ierr); 35911e60469SMark F. Adams ierr = ISDestroy( &isnewproc ); CHKERRQ(ierr); 360038e3b61SMark F. Adams ncrs_new = counts[mype]/a_cbs; 36122063be5SMark F. Adams strideNew = ncrs_new*a_ndata_rows; 362b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 363b4fbaa2aSMark F. Adams ierr = PetscLogEventEnd(gamg_setup_events[SET12],0,0,0,0); CHKERRQ(ierr); 364b4fbaa2aSMark F. Adams #endif 36522063be5SMark F. Adams /* Create a vector to contain the newly ordered element information */ 36611e60469SMark F. Adams ierr = VecCreate( wcomm, &dest_crd ); 367d3d6bff4SMark F. Adams ierr = VecSetSizes( dest_crd, data_sz*ncrs_new, PETSC_DECIDE ); CHKERRQ(ierr); 36811e60469SMark F. Adams ierr = VecSetFromOptions( dest_crd ); CHKERRQ(ierr); /*funny vector-get global options?*/ 36911e60469SMark F. Adams /* 370d3d6bff4SMark F. Adams There are 'a_ndata_rows*a_ndata_cols' data items per node, (one can think of the vectors of having 371d3d6bff4SMark F. Adams a block size of ...). Note, ISs are expanded into equation space by 'a_cbs'. 37211e60469SMark F. Adams */ 37392a756f0SMark F. Adams ierr = PetscMalloc( (ncrs0*data_sz)*sizeof(PetscInt), &tidx ); CHKERRQ(ierr); 37411e60469SMark F. Adams ierr = ISGetIndices( isnum, &idx ); CHKERRQ(ierr); 375038e3b61SMark F. Adams for(ii=0,jj=0; ii<ncrs0 ; ii++) { 376d3d6bff4SMark F. Adams PetscInt id = idx[ii*a_cbs]/a_cbs; /* get node back */ 377d3d6bff4SMark F. Adams for( kk=0; kk<data_sz ; kk++, jj++) tidx[jj] = id*data_sz + kk; 37811e60469SMark F. Adams } 379038e3b61SMark F. Adams ierr = ISRestoreIndices( isnum, &idx ); CHKERRQ(ierr); 380d3d6bff4SMark F. Adams ierr = ISCreateGeneral( wcomm, data_sz*ncrs0, tidx, PETSC_COPY_VALUES, &isscat ); 38111e60469SMark F. Adams CHKERRQ(ierr); 38292a756f0SMark F. Adams ierr = PetscFree( tidx ); CHKERRQ(ierr); 38311e60469SMark F. Adams /* 38411e60469SMark F. Adams Create a vector to contain the original vertex information for each element 38511e60469SMark F. Adams */ 386d3d6bff4SMark F. Adams ierr = VecCreateSeq( PETSC_COMM_SELF, data_sz*ncrs0, &src_crd ); CHKERRQ(ierr); 387d3d6bff4SMark F. Adams for( jj=0; jj<a_ndata_cols ; jj++ ) { 388d3d6bff4SMark F. Adams for( ii=0 ; ii<ncrs0 ; ii++) { 389d3d6bff4SMark F. Adams for( kk=0; kk<a_ndata_rows ; kk++ ) { 390d3d6bff4SMark F. Adams PetscInt ix = ii*a_ndata_rows + kk + jj*stride0, jx = ii*data_sz + kk*a_ndata_cols + jj; 391d3d6bff4SMark F. Adams ierr = VecSetValues( src_crd, 1, &jx, &data[ix], INSERT_VALUES ); CHKERRQ(ierr); 392d3d6bff4SMark F. Adams } 393038e3b61SMark F. Adams } 394eb07cef2SMark F. Adams } 395eb07cef2SMark F. Adams ierr = VecAssemblyBegin(src_crd); CHKERRQ(ierr); 396eb07cef2SMark F. Adams ierr = VecAssemblyEnd(src_crd); CHKERRQ(ierr); 39711e60469SMark F. Adams /* 39811e60469SMark F. Adams Scatter the element vertex information (still in the original vertex ordering) 39911e60469SMark F. Adams to the correct processor 40011e60469SMark F. Adams */ 40111e60469SMark F. Adams ierr = VecScatterCreate( src_crd, PETSC_NULL, dest_crd, isscat, &vecscat); 40211e60469SMark F. Adams CHKERRQ(ierr); 40311e60469SMark F. Adams ierr = ISDestroy( &isscat ); CHKERRQ(ierr); 40411e60469SMark F. Adams ierr = VecScatterBegin(vecscat,src_crd,dest_crd,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 40511e60469SMark F. Adams ierr = VecScatterEnd(vecscat,src_crd,dest_crd,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 40611e60469SMark F. Adams ierr = VecScatterDestroy( &vecscat ); CHKERRQ(ierr); 40711e60469SMark F. Adams ierr = VecDestroy( &src_crd ); CHKERRQ(ierr); 40811e60469SMark F. Adams /* 40911e60469SMark F. Adams Put the element vertex data into a new allocation of the gdata->ele 41011e60469SMark F. Adams */ 411eb07cef2SMark F. Adams ierr = PetscFree( *a_coarse_data ); CHKERRQ(ierr); 412d3d6bff4SMark F. Adams ierr = PetscMalloc( data_sz*ncrs_new*sizeof(PetscReal), a_coarse_data ); CHKERRQ(ierr); 413eb07cef2SMark F. Adams 41411e60469SMark F. Adams ierr = VecGetArray( dest_crd, &array ); CHKERRQ(ierr); 415eb07cef2SMark F. Adams data = *a_coarse_data; 416d3d6bff4SMark F. Adams for( jj=0; jj<a_ndata_cols ; jj++ ) { 417d3d6bff4SMark F. Adams for( ii=0 ; ii<ncrs_new ; ii++) { 418d3d6bff4SMark F. Adams for( kk=0; kk<a_ndata_rows ; kk++ ) { 419d3d6bff4SMark F. Adams PetscInt ix = ii*a_ndata_rows + kk + jj*strideNew, jx = ii*data_sz + kk*a_ndata_cols + jj; 420d3d6bff4SMark F. Adams data[ix] = PetscRealPart(array[jx]); 421d3d6bff4SMark F. Adams array[jx] = 1.e300; 422d3d6bff4SMark F. Adams } 423038e3b61SMark F. Adams } 424038e3b61SMark F. Adams } 42511e60469SMark F. Adams ierr = VecRestoreArray( dest_crd, &array ); CHKERRQ(ierr); 42611e60469SMark F. Adams ierr = VecDestroy( &dest_crd ); CHKERRQ(ierr); 42711e60469SMark F. Adams /* 42811e60469SMark F. Adams Invert for MatGetSubMatrix 42911e60469SMark F. Adams */ 430038e3b61SMark F. Adams ierr = ISInvertPermutation( isnum, ncrs_new*a_cbs, &new_indices ); CHKERRQ(ierr); 43111e60469SMark F. Adams ierr = ISSort( new_indices ); CHKERRQ(ierr); /* is this needed? */ 43211e60469SMark F. Adams ierr = ISDestroy( &isnum ); CHKERRQ(ierr); 43311e60469SMark F. Adams /* A_crs output */ 434038e3b61SMark F. Adams ierr = MatGetSubMatrix( Cmat, new_indices, new_indices, MAT_INITIAL_MATRIX, a_Amat_crs ); 43511e60469SMark F. Adams CHKERRQ(ierr); 436eb07cef2SMark F. Adams 437038e3b61SMark F. Adams ierr = MatDestroy( &Cmat ); CHKERRQ(ierr); 438e33ef3b1SMark F. Adams Cmat = *a_Amat_crs; /* output */ 439038e3b61SMark F. Adams ierr = MatSetBlockSize( Cmat, a_cbs ); CHKERRQ(ierr); 440eb07cef2SMark F. Adams 44111e60469SMark F. Adams /* prolongator */ 44211e60469SMark F. Adams ierr = MatGetOwnershipRange( Pold, &Istart, &Iend ); CHKERRQ(ierr); 44311e60469SMark F. Adams { 44411e60469SMark F. Adams IS findices; 44511e60469SMark F. Adams ierr = ISCreateStride(wcomm,Iend-Istart,Istart,1,&findices); CHKERRQ(ierr); 446d61a3a7dSMark F. Adams 44711e60469SMark F. Adams ierr = MatGetSubMatrix( Pold, findices, new_indices, MAT_INITIAL_MATRIX, &Pnew ); 44811e60469SMark F. Adams CHKERRQ(ierr); 449d61a3a7dSMark F. Adams 45011e60469SMark F. Adams ierr = ISDestroy( &findices ); CHKERRQ(ierr); 45111e60469SMark F. Adams } 452d61a3a7dSMark F. Adams 4533530afc2SMark F. Adams ierr = MatDestroy( a_P_inout ); CHKERRQ(ierr); 4543530afc2SMark F. Adams *a_P_inout = Pnew; /* output */ 45592a756f0SMark F. Adams 45611e60469SMark F. Adams ierr = ISDestroy( &new_indices ); CHKERRQ(ierr); 45792a756f0SMark F. Adams ierr = PetscFree( counts ); CHKERRQ(ierr); 45892a756f0SMark F. Adams ierr = PetscFree( ranks ); CHKERRQ(ierr); 459e33ef3b1SMark F. Adams } 4605b89ad90SMark F. Adams 4615b89ad90SMark F. Adams PetscFunctionReturn(0); 4625b89ad90SMark F. Adams } 4635b89ad90SMark F. Adams 4645b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */ 4655b89ad90SMark F. Adams /* 4665b89ad90SMark F. Adams PCSetUp_GAMG - Prepares for the use of the GAMG preconditioner 4675b89ad90SMark F. Adams by setting data structures and options. 4685b89ad90SMark F. Adams 4695b89ad90SMark F. Adams Input Parameter: 4705b89ad90SMark F. Adams . pc - the preconditioner context 4715b89ad90SMark F. Adams 4725b89ad90SMark F. Adams Application Interface Routine: PCSetUp() 4735b89ad90SMark F. Adams 4745b89ad90SMark F. Adams Notes: 4755b89ad90SMark F. Adams The interface routine PCSetUp() is not usually called directly by 4765b89ad90SMark F. Adams the user, but instead is called by PCApply() if necessary. 4775b89ad90SMark F. Adams */ 4785b89ad90SMark F. Adams #undef __FUNCT__ 4795b89ad90SMark F. Adams #define __FUNCT__ "PCSetUp_GAMG" 480eb07cef2SMark F. Adams PetscErrorCode PCSetUp_GAMG( PC a_pc ) 4815b89ad90SMark F. Adams { 4825b89ad90SMark F. Adams PetscErrorCode ierr; 483eb07cef2SMark F. Adams PC_MG *mg = (PC_MG*)a_pc->data; 4845b89ad90SMark F. Adams PC_GAMG *pc_gamg = (PC_GAMG*)mg->innerctx; 48558471d46SMark F. Adams PC_MG_Levels **mglevels = mg->levels; 486eb07cef2SMark F. Adams Mat Amat = a_pc->mat, Pmat = a_pc->pmat; 487d3d6bff4SMark F. Adams PetscInt fine_level, level, level1, M, N, bs, nloc, lidx, Istart, Iend; 488eb07cef2SMark F. Adams MPI_Comm wcomm = ((PetscObject)a_pc)->comm; 4893530afc2SMark F. Adams PetscMPIInt mype,npe,nactivepe; 490038e3b61SMark F. Adams PetscBool isOK; 491587fa25dSMark F. Adams Mat Aarr[GAMG_MAXLEVELS], Parr[GAMG_MAXLEVELS]; 492587fa25dSMark F. Adams PetscReal *coarse_data = 0, *data, emaxs[GAMG_MAXLEVELS]; 493737a81a9SMark F. Adams MatInfo info; 4945ef31b24SMark F. Adams 4955b89ad90SMark F. Adams PetscFunctionBegin; 496*03a628feSMark F. Adams pc_gamg->m_count++; 49758471d46SMark F. Adams if( a_pc->setupcalled > 0 ) { 498*03a628feSMark F. Adams /* just do Galerkin grids */ 49958471d46SMark F. Adams Mat B,dA,dB; 50058471d46SMark F. Adams 50158471d46SMark F. Adams /* PCSetUp_MG seems to insists on setting this to GMRES */ 50258471d46SMark F. Adams ierr = KSPSetType( mglevels[0]->smoothd, KSPPREONLY ); CHKERRQ(ierr); 50358471d46SMark F. Adams 50458471d46SMark F. Adams /* currently only handle case where mat and pmat are the same on coarser levels */ 50558471d46SMark F. Adams ierr = KSPGetOperators(mglevels[pc_gamg->m_Nlevels-1]->smoothd,&dA,&dB,PETSC_NULL);CHKERRQ(ierr); 50658471d46SMark F. Adams /* (re)set to get dirty flag */ 50758471d46SMark F. Adams ierr = KSPSetOperators(mglevels[pc_gamg->m_Nlevels-1]->smoothd,dA,dB,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 50858471d46SMark F. Adams ierr = KSPSetUp( mglevels[pc_gamg->m_Nlevels-1]->smoothd ); CHKERRQ(ierr); 50958471d46SMark F. Adams 51058471d46SMark F. Adams for (level=pc_gamg->m_Nlevels-2; level>-1; level--) { 51158471d46SMark F. Adams ierr = KSPGetOperators(mglevels[level]->smoothd,PETSC_NULL,&B,PETSC_NULL);CHKERRQ(ierr); 512*03a628feSMark F. Adams /* the first time through the matrix structure has changed from repartitioning */ 513*03a628feSMark F. Adams if( pc_gamg->m_count == 2 ) { 514*03a628feSMark F. Adams ierr = MatDestroy( &B ); CHKERRQ(ierr); 515*03a628feSMark F. Adams ierr = MatPtAP(dB,mglevels[level+1]->interpolate,MAT_INITIAL_MATRIX,1.0,&B);CHKERRQ(ierr); 516*03a628feSMark F. Adams mglevels[level]->A = B; 517*03a628feSMark F. Adams } 518*03a628feSMark F. Adams else { 51958471d46SMark F. Adams ierr = MatPtAP(dB,mglevels[level+1]->interpolate,MAT_REUSE_MATRIX,1.0,&B);CHKERRQ(ierr); 520*03a628feSMark F. Adams } 52158471d46SMark F. Adams ierr = KSPSetOperators(mglevels[level]->smoothd,B,B,SAME_NONZERO_PATTERN); CHKERRQ(ierr); 52258471d46SMark F. Adams dB = B; 52358471d46SMark F. Adams /* setup KSP/PC */ 52458471d46SMark F. Adams ierr = KSPSetUp( mglevels[level]->smoothd ); CHKERRQ(ierr); 52558471d46SMark F. Adams } 52658471d46SMark F. Adams 52758471d46SMark F. Adams #define PRINT_MATS !PETSC_TRUE 52858471d46SMark F. Adams /* plot levels - A */ 52958471d46SMark F. Adams if( PRINT_MATS ) { 53058471d46SMark F. Adams for (lidx=0, level=pc_gamg->m_Nlevels-1; level>0 ; level--,lidx++){ 53158471d46SMark F. Adams PetscViewer viewer; 53258471d46SMark F. Adams char fname[32]; KSP smoother; Mat Tmat, TTm; 53358471d46SMark F. Adams ierr = PCMGGetSmoother( a_pc, lidx, &smoother ); CHKERRQ(ierr); 53458471d46SMark F. Adams ierr = KSPGetOperators( smoother, &Tmat, &TTm, 0 ); CHKERRQ(ierr); 535*03a628feSMark F. Adams sprintf(fname,"Amat_%d_%d.m",(int)pc_gamg->m_count,(int)level); 53658471d46SMark F. Adams ierr = PetscViewerASCIIOpen( wcomm, fname, &viewer ); CHKERRQ(ierr); 53758471d46SMark F. Adams ierr = PetscViewerSetFormat( viewer, PETSC_VIEWER_ASCII_MATLAB); CHKERRQ(ierr); 53858471d46SMark F. Adams ierr = MatView( Tmat, viewer ); CHKERRQ(ierr); 53958471d46SMark F. Adams ierr = PetscViewerDestroy( &viewer ); 54058471d46SMark F. Adams } 54158471d46SMark F. Adams } 54258471d46SMark F. Adams 543*03a628feSMark F. Adams a_pc->setupcalled = 2; 544*03a628feSMark F. Adams 54558471d46SMark F. Adams PetscFunctionReturn(0); 546eb07cef2SMark F. Adams } 547baa4e9faSMark F. Adams ierr = MPI_Comm_rank(wcomm,&mype);CHKERRQ(ierr); 548baa4e9faSMark F. Adams ierr = MPI_Comm_size(wcomm,&npe);CHKERRQ(ierr); 5495b89ad90SMark F. Adams /* GAMG requires input of fine-grid matrix. It determines nlevels. */ 5505b89ad90SMark F. Adams ierr = MatGetBlockSize( Amat, &bs ); CHKERRQ(ierr); 5513530afc2SMark F. Adams ierr = MatGetSize( Amat, &M, &N );CHKERRQ(ierr); 552eb07cef2SMark F. Adams ierr = MatGetOwnershipRange( Amat, &Istart, &Iend ); CHKERRQ(ierr); 553eb07cef2SMark F. Adams nloc = (Iend-Istart)/bs; assert((Iend-Istart)%bs == 0); 554eb07cef2SMark F. Adams 555038e3b61SMark F. Adams /* get data of not around */ 556038e3b61SMark F. Adams if( pc_gamg->m_data == 0 && nloc > 0 ) { 557038e3b61SMark F. Adams ierr = PCSetCoordinates_GAMG( a_pc, -1, 0 ); CHKERRQ( ierr ); 558eb07cef2SMark F. Adams } 559eb07cef2SMark F. Adams data = pc_gamg->m_data; 560038e3b61SMark F. Adams 561eb07cef2SMark F. Adams /* Get A_i and R_i */ 562737a81a9SMark F. Adams ierr = MatGetInfo(Amat,MAT_GLOBAL_SUM,&info); CHKERRQ(ierr); 56358471d46SMark F. Adams #ifdef VERBOSE 564fd3c6afaSMark F. Adams PetscPrintf(PETSC_COMM_WORLD,"\t[%d]%s level %d N=%ld, n data rows=%d, n data cols=%d, nnz/row (ave)=%d, np=%d\n", 565fd3c6afaSMark F. Adams mype,__FUNCT__,0,(long long int)N,(int)pc_gamg->m_data_rows,(int)pc_gamg->m_data_cols, 566fd3c6afaSMark F. Adams (int)(info.nz_used/(PetscReal)N),(int)npe); 56758471d46SMark F. Adams #endif 5688f4b7eb5SMark F. Adams for ( level=0, Aarr[0] = Pmat, nactivepe = npe; /* hard wired stopping logic */ 569785cba28SMark F. Adams level < GAMG_MAXLEVELS-1 && (level==0 || M>TOP_GRID_LIM) && (npe==1 || nactivepe>1); 5700205a208SMark F. Adams level++ ){ 5715b89ad90SMark F. Adams level1 = level + 1; 572b4fbaa2aSMark F. Adams #if (defined PETSC_USE_LOG && defined GAMG_STAGES) 573b4fbaa2aSMark F. Adams ierr = PetscLogStagePush(gamg_stages[level]); CHKERRQ( ierr ); 574b4fbaa2aSMark F. Adams #endif 575b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 576b4fbaa2aSMark F. Adams ierr = PetscLogEventBegin(gamg_setup_events[SET1],0,0,0,0);CHKERRQ(ierr); 577b4fbaa2aSMark F. Adams #endif 578d3d6bff4SMark F. Adams ierr = createProlongation(Aarr[level], data, pc_gamg->m_dim, pc_gamg->m_data_cols, pc_gamg->m_useSA, 579785cba28SMark F. Adams level, &bs, &Parr[level1], &coarse_data, &isOK, &emaxs[level] ); 5805b89ad90SMark F. Adams CHKERRQ(ierr); 581d3d6bff4SMark F. Adams ierr = PetscFree( data ); CHKERRQ( ierr ); 582b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 583b4fbaa2aSMark F. Adams ierr = PetscLogEventEnd(gamg_setup_events[SET1],0,0,0,0);CHKERRQ(ierr); 584b4fbaa2aSMark F. Adams #endif 585baa4e9faSMark F. Adams if(level==0) Aarr[0] = Amat; /* use Pmat for finest level setup, but use mat for solver */ 586baa4e9faSMark F. Adams if( isOK ) { 587b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 588b4fbaa2aSMark F. Adams ierr = PetscLogEventBegin(gamg_setup_events[SET2],0,0,0,0);CHKERRQ(ierr); 589b4fbaa2aSMark F. Adams #endif 590d3d6bff4SMark F. Adams ierr = partitionLevel( Aarr[level], pc_gamg->m_useSA ? bs : 1, pc_gamg->m_data_cols, bs, 591eb07cef2SMark F. Adams &Parr[level1], &coarse_data, &nactivepe, &Aarr[level1] ); 5923530afc2SMark F. Adams CHKERRQ(ierr); 593b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 594b4fbaa2aSMark F. Adams ierr = PetscLogEventEnd(gamg_setup_events[SET2],0,0,0,0);CHKERRQ(ierr); 595b4fbaa2aSMark F. Adams #endif 5963530afc2SMark F. Adams ierr = MatGetSize( Aarr[level1], &M, &N );CHKERRQ(ierr); 597737a81a9SMark F. Adams ierr = MatGetInfo(Aarr[level1],MAT_GLOBAL_SUM,&info); CHKERRQ(ierr); 59858471d46SMark F. Adams #ifdef VERBOSE 59958471d46SMark F. Adams PetscPrintf(PETSC_COMM_WORLD,"\t\t[%d]%s %d) N=%d, n data cols=%d, nnz/row (ave)=%d, %d active pes\n", 60058471d46SMark F. Adams mype,__FUNCT__,(int)level1,(int)N,(int)pc_gamg->m_data_cols, 601fd3c6afaSMark F. Adams (int)(info.nz_used/(PetscReal)N),(int)nactivepe); 60258471d46SMark F. Adams #endif 603e33ef3b1SMark F. Adams /* coarse grids with SA can have zero row/cols from singleton aggregates */ 60458471d46SMark F. Adams /* aggregation method should gaurrentee this does not happen! */ 605737a81a9SMark F. Adams 60658471d46SMark F. Adams #ifdef VERBOSE 607e33ef3b1SMark F. Adams if( PETSC_TRUE ){ 608785cba28SMark F. Adams Vec diag; PetscScalar *data_arr,v; PetscInt Istart,Iend,kk,nloceq,id; 609785cba28SMark F. Adams v = 1.e-10; /* LU factor has hard wired numbers for small diags so this needs to match (yuk) */ 610e33ef3b1SMark F. Adams ierr = MatGetOwnershipRange(Aarr[level1], &Istart, &Iend); CHKERRQ(ierr); 611785cba28SMark F. Adams nloceq = Iend-Istart; 612e33ef3b1SMark F. Adams ierr = MatGetVecs( Aarr[level1], &diag, 0 ); CHKERRQ(ierr); 613e33ef3b1SMark F. Adams ierr = MatGetDiagonal( Aarr[level1], diag ); CHKERRQ(ierr); 614e33ef3b1SMark F. Adams ierr = VecGetArray( diag, &data_arr ); CHKERRQ(ierr); 615785cba28SMark F. Adams for(kk=0;kk<nloceq;kk++){ 616e33ef3b1SMark F. Adams if(data_arr[kk]==0.0) { 617785cba28SMark F. Adams id = kk + Istart; 618e33ef3b1SMark F. Adams ierr = MatSetValues(Aarr[level1],1,&id,1,&id,&v,INSERT_VALUES); 619e33ef3b1SMark F. Adams CHKERRQ(ierr); 620ecd57171SMark F. Adams PetscPrintf(PETSC_COMM_SELF,"\t[%d]%s warning: added zero to diag (%d) on level %d \n",mype,__FUNCT__,id,level1); 621e33ef3b1SMark F. Adams } 622e33ef3b1SMark F. Adams } 623e33ef3b1SMark F. Adams ierr = VecRestoreArray( diag, &data_arr ); CHKERRQ(ierr); 624e33ef3b1SMark F. Adams ierr = VecDestroy( &diag ); CHKERRQ(ierr); 625e33ef3b1SMark F. Adams ierr = MatAssemblyBegin(Aarr[level1],MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 626e33ef3b1SMark F. Adams ierr = MatAssemblyEnd(Aarr[level1],MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 627e33ef3b1SMark F. Adams } 62858471d46SMark F. Adams #endif 629baa4e9faSMark F. Adams } 630baa4e9faSMark F. Adams else{ 631be544d3cSMark F. Adams coarse_data = 0; 632baa4e9faSMark F. Adams break; 633baa4e9faSMark F. Adams } 634eb07cef2SMark F. Adams data = coarse_data; 635b4fbaa2aSMark F. Adams 636b4fbaa2aSMark F. Adams #if (defined PETSC_USE_LOG && defined GAMG_STAGES) 637b4fbaa2aSMark F. Adams ierr = PetscLogStagePop(); CHKERRQ( ierr ); 638b4fbaa2aSMark F. Adams #endif 6395b89ad90SMark F. Adams } 640be544d3cSMark F. Adams if( coarse_data ) { 641eb07cef2SMark F. Adams ierr = PetscFree( coarse_data ); CHKERRQ( ierr ); 642be544d3cSMark F. Adams } 64358471d46SMark F. Adams #ifdef VERBOSE 644baa4e9faSMark F. Adams PetscPrintf(PETSC_COMM_WORLD,"\t[%d]%s %d levels\n",0,__FUNCT__,level + 1); 64558471d46SMark F. Adams #endif 6465b89ad90SMark F. Adams pc_gamg->m_data = 0; /* destroyed coordinate data */ 6475b89ad90SMark F. Adams pc_gamg->m_Nlevels = level + 1; 6485b89ad90SMark F. Adams fine_level = level; 649eb07cef2SMark F. Adams ierr = PCMGSetLevels(a_pc,pc_gamg->m_Nlevels,PETSC_NULL);CHKERRQ(ierr); 6505b89ad90SMark F. Adams 651fc4362bfSMark F. Adams /* set default smoothers */ 652587fa25dSMark F. Adams for ( lidx=1, level = pc_gamg->m_Nlevels-2; 653587fa25dSMark F. Adams lidx <= fine_level; 654587fa25dSMark F. Adams lidx++, level--) { 6555745e0f5SMark F. Adams PetscReal emax, emin; 6565b89ad90SMark F. Adams KSP smoother; PC subpc; 657587fa25dSMark F. Adams ierr = PCMGGetSmoother( a_pc, lidx, &smoother ); CHKERRQ(ierr); 6585b89ad90SMark F. Adams ierr = KSPSetType( smoother, KSPCHEBYCHEV );CHKERRQ(ierr); 659038e3b61SMark F. Adams if( emaxs[level] > 0.0 ) emax=emaxs[level]; 660587fa25dSMark F. Adams else{ /* eigen estimate 'emax' */ 661587fa25dSMark F. Adams KSP eksp; Mat Lmat = Aarr[level]; 662fc4362bfSMark F. Adams Vec bb, xx; PC pc; 663038e3b61SMark F. Adams 6645745e0f5SMark F. Adams ierr = MatGetVecs( Lmat, &bb, 0 ); CHKERRQ(ierr); 6655745e0f5SMark F. Adams ierr = MatGetVecs( Lmat, &xx, 0 ); CHKERRQ(ierr); 666fc4362bfSMark F. Adams { 667fc4362bfSMark F. Adams PetscRandom rctx; 668fc4362bfSMark F. Adams ierr = PetscRandomCreate(wcomm,&rctx);CHKERRQ(ierr); 669fc4362bfSMark F. Adams ierr = PetscRandomSetFromOptions(rctx);CHKERRQ(ierr); 670fc4362bfSMark F. Adams ierr = VecSetRandom(bb,rctx);CHKERRQ(ierr); 671fc4362bfSMark F. Adams ierr = PetscRandomDestroy( &rctx ); CHKERRQ(ierr); 6725b89ad90SMark F. Adams } 673fc4362bfSMark F. Adams ierr = KSPCreate(wcomm,&eksp);CHKERRQ(ierr); 674e33ef3b1SMark F. Adams ierr = KSPSetType( eksp, KSPCG ); CHKERRQ(ierr); 675fc4362bfSMark F. Adams ierr = KSPSetInitialGuessNonzero( eksp, PETSC_FALSE ); CHKERRQ(ierr); 67658471d46SMark F. Adams ierr = KSPSetOperators( eksp, Lmat, Lmat, SAME_NONZERO_PATTERN ); CHKERRQ( ierr ); 677fc4362bfSMark F. Adams ierr = KSPGetPC( eksp, &pc );CHKERRQ( ierr ); 67858471d46SMark F. Adams ierr = PCSetType( pc, PETSC_GAMG_SMOOTHER ); CHKERRQ(ierr); /* should be same as eigen estimates op. */ 679038e3b61SMark F. Adams ierr = KSPSetTolerances( eksp, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT, 10 ); 680fc4362bfSMark F. Adams CHKERRQ(ierr); 681fc4362bfSMark F. Adams ierr = KSPSetNormType( eksp, KSP_NORM_NONE ); CHKERRQ(ierr); 682fc4362bfSMark F. Adams 683fc4362bfSMark F. Adams ierr = KSPSetComputeSingularValues( eksp,PETSC_TRUE ); CHKERRQ(ierr); 684fc4362bfSMark F. Adams ierr = KSPSolve( eksp, bb, xx ); CHKERRQ(ierr); 685fc4362bfSMark F. Adams ierr = KSPComputeExtremeSingularValues( eksp, &emax, &emin ); CHKERRQ(ierr); 686fc4362bfSMark F. Adams ierr = VecDestroy( &xx ); CHKERRQ(ierr); 687fc4362bfSMark F. Adams ierr = VecDestroy( &bb ); CHKERRQ(ierr); 688fc4362bfSMark F. Adams ierr = KSPDestroy( &eksp ); CHKERRQ(ierr); 68958471d46SMark F. Adams #ifdef VERBOSE 6901fddbf69SMark F. Adams PetscPrintf(PETSC_COMM_WORLD,"\t\t\t%s max eigen=%e min=%e PC=%s\n",__FUNCT__,emax,emin,PETSC_GAMG_SMOOTHER); 69158471d46SMark F. Adams #endif 692fc4362bfSMark F. Adams } 693038e3b61SMark F. Adams { 694038e3b61SMark F. Adams PetscInt N1, N0, tt; 695038e3b61SMark F. Adams ierr = MatGetSize( Aarr[level], &N1, &tt ); CHKERRQ(ierr); 696038e3b61SMark F. Adams ierr = MatGetSize( Aarr[level+1], &N0, &tt ); CHKERRQ(ierr); 697785cba28SMark F. Adams emin = 1.*emax/((PetscReal)N1/(PetscReal)N0); /* this should be about the coarsening rate */ 698038e3b61SMark F. Adams emax *= 1.05; 699038e3b61SMark F. Adams } 700038e3b61SMark F. Adams 70158471d46SMark F. Adams ierr = KSPSetOperators( smoother, Aarr[level], Aarr[level], SAME_NONZERO_PATTERN ); 702fc4362bfSMark F. Adams ierr = KSPChebychevSetEigenvalues( smoother, emax, emin );CHKERRQ(ierr); 7030e1b4bd6SMark F. Adams /* ierr = KSPSetTolerances(smoother,PETSC_DEFAULT,PETSC_DEFAULT,PETSC_DEFAULT,2); CHKERRQ(ierr); */ 7045745e0f5SMark F. Adams ierr = KSPGetPC( smoother, &subpc ); CHKERRQ(ierr); 705e33ef3b1SMark F. Adams ierr = PCSetType( subpc, PETSC_GAMG_SMOOTHER ); CHKERRQ(ierr); 7065745e0f5SMark F. Adams ierr = KSPSetNormType( smoother, KSP_NORM_NONE ); CHKERRQ(ierr); 7075745e0f5SMark F. Adams } 7085745e0f5SMark F. Adams { 709ecd57171SMark F. Adams /* coarse grid */ 710ecd57171SMark F. Adams KSP smoother,*k2; PC subpc,pc2; PetscInt ii,first; 7115745e0f5SMark F. Adams Mat Lmat = Aarr[pc_gamg->m_Nlevels-1]; 712eb07cef2SMark F. Adams ierr = PCMGGetSmoother( a_pc, 0, &smoother ); CHKERRQ(ierr); 71358471d46SMark F. Adams ierr = KSPSetOperators( smoother, Lmat, Lmat, SAME_NONZERO_PATTERN ); CHKERRQ(ierr); 7145745e0f5SMark F. Adams ierr = KSPSetNormType( smoother, KSP_NORM_NONE ); CHKERRQ(ierr); 715ecd57171SMark F. Adams ierr = KSPGetPC( smoother, &subpc ); CHKERRQ(ierr); 716ecd57171SMark F. Adams ierr = PCSetType( subpc, PCBJACOBI ); CHKERRQ(ierr); 717ecd57171SMark F. Adams ierr = PCSetUp( subpc ); CHKERRQ(ierr); 718ecd57171SMark F. Adams ierr = PCBJacobiGetSubKSP(subpc,&ii,&first,&k2);CHKERRQ(ierr); 719ecd57171SMark F. Adams assert(ii==1); 720ecd57171SMark F. Adams ierr = KSPGetPC(k2[0],&pc2);CHKERRQ(ierr); 721ecd57171SMark F. Adams ierr = PCSetType( pc2, PCLU ); CHKERRQ(ierr); 722fc4362bfSMark F. Adams } 723737a81a9SMark F. Adams 724fc4362bfSMark F. Adams /* should be called in PCSetFromOptions_GAMG(), but cannot be called prior to PCMGSetLevels() */ 725eb07cef2SMark F. Adams ierr = PCSetFromOptions_MG(a_pc); CHKERRQ(ierr); 7265b89ad90SMark F. Adams { 7275b89ad90SMark F. Adams PetscBool galerkin; 728eb07cef2SMark F. Adams ierr = PCMGGetGalerkin( a_pc, &galerkin); CHKERRQ(ierr); 7295b89ad90SMark F. Adams if(galerkin){ 7305b89ad90SMark F. Adams SETERRQ(wcomm,PETSC_ERR_ARG_WRONG, "GAMG does galerkin manually so it must not be used in PC_MG."); 7315b89ad90SMark F. Adams } 7325b89ad90SMark F. Adams } 7335745e0f5SMark F. Adams 73458471d46SMark F. Adams /* plot levels - R/P */ 73558471d46SMark F. Adams if( PRINT_MATS ) { 73658471d46SMark F. Adams for (level=pc_gamg->m_Nlevels-1;level>0;level--){ 73758471d46SMark F. Adams PetscViewer viewer; 73858471d46SMark F. Adams char fname[32]; 739*03a628feSMark F. Adams sprintf(fname,"Pmat_%d_%d.m",(int)pc_gamg->m_count,(int)level); 74011e60469SMark F. Adams ierr = PetscViewerASCIIOpen( wcomm, fname, &viewer ); CHKERRQ(ierr); 7415b89ad90SMark F. Adams ierr = PetscViewerSetFormat( viewer, PETSC_VIEWER_ASCII_MATLAB); CHKERRQ(ierr); 742038e3b61SMark F. Adams ierr = MatView( Parr[level], viewer ); CHKERRQ(ierr); 7435b89ad90SMark F. Adams ierr = PetscViewerDestroy( &viewer ); 744*03a628feSMark F. Adams sprintf(fname,"Amat_%d_%d.m",(int)pc_gamg->m_count,(int)level); 745e33ef3b1SMark F. Adams ierr = PetscViewerASCIIOpen( wcomm, fname, &viewer ); CHKERRQ(ierr); 746e33ef3b1SMark F. Adams ierr = PetscViewerSetFormat( viewer, PETSC_VIEWER_ASCII_MATLAB); CHKERRQ(ierr); 747e33ef3b1SMark F. Adams ierr = MatView( Aarr[level], viewer ); CHKERRQ(ierr); 748e33ef3b1SMark F. Adams ierr = PetscViewerDestroy( &viewer ); 7495b89ad90SMark F. Adams } 75058471d46SMark F. Adams } 75158471d46SMark F. Adams 75258471d46SMark F. Adams /* set interpolation between the levels, clean up */ 75358471d46SMark F. Adams for (lidx=0,level=pc_gamg->m_Nlevels-1; 75458471d46SMark F. Adams lidx<fine_level; 75558471d46SMark F. Adams lidx++, level--){ 75658471d46SMark F. Adams ierr = PCMGSetInterpolation( a_pc, lidx+1, Parr[level] );CHKERRQ(ierr); 757587fa25dSMark F. Adams ierr = MatDestroy( &Parr[level] ); CHKERRQ(ierr); 758587fa25dSMark F. Adams ierr = MatDestroy( &Aarr[level] ); CHKERRQ(ierr); 7595b89ad90SMark F. Adams } 7605745e0f5SMark F. Adams 7615b89ad90SMark F. Adams /* setupcalled is set to 0 so that MG is setup from scratch */ 762eb07cef2SMark F. Adams a_pc->setupcalled = 0; 763eb07cef2SMark F. Adams ierr = PCSetUp_MG( a_pc );CHKERRQ( ierr ); 764*03a628feSMark F. Adams a_pc->setupcalled = 1; /* use 1 as signal that this has not been re-setup */ 76558471d46SMark F. Adams 76658471d46SMark F. Adams { 76758471d46SMark F. Adams KSP smoother; /* PCSetUp_MG seems to insists on setting this to GMRES on coarse grid */ 76858471d46SMark F. Adams ierr = PCMGGetSmoother( a_pc, 0, &smoother ); CHKERRQ(ierr); 76958471d46SMark F. Adams ierr = KSPSetType( smoother, KSPPREONLY ); CHKERRQ(ierr); 77058471d46SMark F. Adams ierr = KSPSetUp( smoother ); CHKERRQ(ierr); 77158471d46SMark F. Adams } 772e33ef3b1SMark F. Adams 7735b89ad90SMark F. Adams PetscFunctionReturn(0); 7745b89ad90SMark F. Adams } 7755b89ad90SMark F. Adams 776eb07cef2SMark F. Adams /* ------------------------------------------------------------------------- */ 7775b89ad90SMark F. Adams /* 7785b89ad90SMark F. Adams PCDestroy_GAMG - Destroys the private context for the GAMG preconditioner 7795b89ad90SMark F. Adams that was created with PCCreate_GAMG(). 7805b89ad90SMark F. Adams 7815b89ad90SMark F. Adams Input Parameter: 7825b89ad90SMark F. Adams . pc - the preconditioner context 7835b89ad90SMark F. Adams 7845b89ad90SMark F. Adams Application Interface Routine: PCDestroy() 7855b89ad90SMark F. Adams */ 7865b89ad90SMark F. Adams #undef __FUNCT__ 7875b89ad90SMark F. Adams #define __FUNCT__ "PCDestroy_GAMG" 7885b89ad90SMark F. Adams PetscErrorCode PCDestroy_GAMG(PC pc) 7895b89ad90SMark F. Adams { 7905b89ad90SMark F. Adams PetscErrorCode ierr; 7915b89ad90SMark F. Adams PC_MG *mg = (PC_MG*)pc->data; 7925b89ad90SMark F. Adams PC_GAMG *pc_gamg= (PC_GAMG*)mg->innerctx; 7935b89ad90SMark F. Adams 7945b89ad90SMark F. Adams PetscFunctionBegin; 7955b89ad90SMark F. Adams ierr = PCReset_GAMG(pc);CHKERRQ(ierr); 7965b89ad90SMark F. Adams ierr = PetscFree(pc_gamg);CHKERRQ(ierr); 7975b89ad90SMark F. Adams ierr = PCDestroy_MG(pc);CHKERRQ(ierr); 7985b89ad90SMark F. Adams PetscFunctionReturn(0); 7995b89ad90SMark F. Adams } 8005b89ad90SMark F. Adams 8015b89ad90SMark F. Adams #undef __FUNCT__ 8025b89ad90SMark F. Adams #define __FUNCT__ "PCSetFromOptions_GAMG" 8035b89ad90SMark F. Adams PetscErrorCode PCSetFromOptions_GAMG(PC pc) 8045b89ad90SMark F. Adams { 8055b89ad90SMark F. Adams /* PetscErrorCode ierr; */ 8065b89ad90SMark F. Adams /* PC_MG *mg = (PC_MG*)pc->data; */ 8075b89ad90SMark F. Adams /* PC_GAMG *pc_gamg = (PC_GAMG*)mg->innerctx; */ 8085b89ad90SMark F. Adams /* MPI_Comm comm = ((PetscObject)pc)->comm; */ 8095b89ad90SMark F. Adams 8105b89ad90SMark F. Adams PetscFunctionBegin; 8115b89ad90SMark F. Adams PetscFunctionReturn(0); 8125b89ad90SMark F. Adams } 8135b89ad90SMark F. Adams 8145b89ad90SMark F. Adams /* -------------------------------------------------------------------------- */ 8155b89ad90SMark F. Adams /* 8165b89ad90SMark F. Adams PCCreate_GAMG - Creates a GAMG preconditioner context, PC_GAMG 8175b89ad90SMark F. Adams 8185b89ad90SMark F. Adams Input Parameter: 8195b89ad90SMark F. Adams . pc - the preconditioner context 8205b89ad90SMark F. Adams 8215b89ad90SMark F. Adams Application Interface Routine: PCCreate() 8225b89ad90SMark F. Adams 8235b89ad90SMark F. Adams */ 8245b89ad90SMark F. Adams /* MC 8255b89ad90SMark F. Adams PCGAMG - Use algebraic multigrid preconditioning. This preconditioner requires you provide 8265b89ad90SMark F. Adams fine grid discretization matrix and coordinates on the fine grid. 8275b89ad90SMark F. Adams 8285b89ad90SMark F. Adams Options Database Key: 8295b89ad90SMark F. Adams Multigrid options(inherited) 8305b89ad90SMark F. Adams + -pc_mg_cycles <1>: 1 for V cycle, 2 for W-cycle (MGSetCycles) 8315b89ad90SMark F. Adams . -pc_mg_smoothup <1>: Number of post-smoothing steps (MGSetNumberSmoothUp) 8325b89ad90SMark F. Adams . -pc_mg_smoothdown <1>: Number of pre-smoothing steps (MGSetNumberSmoothDown) 8335b89ad90SMark F. Adams -pc_mg_type <multiplicative>: (one of) additive multiplicative full cascade kascade 8345b89ad90SMark F. Adams GAMG options: 8355b89ad90SMark F. Adams 8365b89ad90SMark F. Adams Level: intermediate 8375b89ad90SMark F. Adams Concepts: multigrid 8385b89ad90SMark F. Adams 8395b89ad90SMark F. Adams .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, PCMGType, 8405b89ad90SMark F. Adams PCMGSetLevels(), PCMGGetLevels(), PCMGSetType(), MPSetCycles(), PCMGSetNumberSmoothDown(), 8415b89ad90SMark F. Adams PCMGSetNumberSmoothUp(), PCMGGetCoarseSolve(), PCMGSetResidual(), PCMGSetInterpolation(), 8425b89ad90SMark F. Adams PCMGSetRestriction(), PCMGGetSmoother(), PCMGGetSmootherUp(), PCMGGetSmootherDown(), 8435b89ad90SMark F. Adams PCMGSetCyclesOnLevel(), PCMGSetRhs(), PCMGSetX(), PCMGSetR() 8445b89ad90SMark F. Adams M */ 8455b89ad90SMark F. Adams 8465b89ad90SMark F. Adams EXTERN_C_BEGIN 8475b89ad90SMark F. Adams #undef __FUNCT__ 8485b89ad90SMark F. Adams #define __FUNCT__ "PCCreate_GAMG" 8495b89ad90SMark F. Adams PetscErrorCode PCCreate_GAMG(PC pc) 8505b89ad90SMark F. Adams { 8515b89ad90SMark F. Adams PetscErrorCode ierr; 8525b89ad90SMark F. Adams PC_GAMG *pc_gamg; 8535b89ad90SMark F. Adams PC_MG *mg; 8545ef31b24SMark F. Adams PetscClassId cookie; 8555b89ad90SMark F. Adams 8565b89ad90SMark F. Adams PetscFunctionBegin; 8575b89ad90SMark F. Adams /* PCGAMG is an inherited class of PCMG. Initialize pc as PCMG */ 8585b89ad90SMark F. Adams ierr = PCSetType(pc,PCMG);CHKERRQ(ierr); /* calls PCCreate_MG() and MGCreate_Private() */ 8595b89ad90SMark F. Adams ierr = PetscObjectChangeTypeName((PetscObject)pc,PCGAMG);CHKERRQ(ierr); 8605b89ad90SMark F. Adams 8615b89ad90SMark F. Adams /* create a supporting struct and attach it to pc */ 8625b89ad90SMark F. Adams ierr = PetscNewLog(pc,PC_GAMG,&pc_gamg);CHKERRQ(ierr); 863*03a628feSMark F. Adams pc_gamg->m_data_sz = 0; pc_gamg->m_data = 0; pc_gamg->m_count = 0; 8645b89ad90SMark F. Adams mg = (PC_MG*)pc->data; 8655b89ad90SMark F. Adams mg->innerctx = pc_gamg; 8665b89ad90SMark F. Adams 8675b89ad90SMark F. Adams pc_gamg->m_Nlevels = -1; 8685b89ad90SMark F. Adams 8695b89ad90SMark F. Adams /* overwrite the pointers of PCMG by the functions of PCGAMG */ 8705b89ad90SMark F. Adams pc->ops->setfromoptions = PCSetFromOptions_GAMG; 8715b89ad90SMark F. Adams pc->ops->setup = PCSetUp_GAMG; 8725b89ad90SMark F. Adams pc->ops->reset = PCReset_GAMG; 8735b89ad90SMark F. Adams pc->ops->destroy = PCDestroy_GAMG; 8745b89ad90SMark F. Adams 8755b89ad90SMark F. Adams ierr = PetscObjectComposeFunctionDynamic( (PetscObject)pc, 8765b89ad90SMark F. Adams "PCSetCoordinates_C", 8775b89ad90SMark F. Adams "PCSetCoordinates_GAMG", 8785b89ad90SMark F. Adams PCSetCoordinates_GAMG);CHKERRQ(ierr); 879b4fbaa2aSMark F. Adams #if defined PETSC_USE_LOG 880785cba28SMark F. Adams static int count = 0; 881785cba28SMark F. Adams if( count++ == 0 ) { 8825ef31b24SMark F. Adams PetscClassIdRegister("GAMG Setup",&cookie); 883b4fbaa2aSMark F. Adams PetscLogEventRegister("GAMG: createProl", cookie, &gamg_setup_events[SET1]); 884b4fbaa2aSMark F. Adams PetscLogEventRegister(" make graph", cookie, &gamg_setup_events[SET3]); 885b4fbaa2aSMark F. Adams PetscLogEventRegister(" MIS/Agg", cookie, &gamg_setup_events[SET4]); 886b4fbaa2aSMark F. Adams PetscLogEventRegister(" geo: growSupp", cookie, &gamg_setup_events[SET5]); 887b4fbaa2aSMark F. Adams PetscLogEventRegister(" geo: triangle", cookie, &gamg_setup_events[SET6]); 888b4fbaa2aSMark F. Adams PetscLogEventRegister(" search & set", cookie, &gamg_setup_events[FIND_V]); 889b4fbaa2aSMark F. Adams PetscLogEventRegister(" SA: init", cookie, &gamg_setup_events[SET7]); 890b4fbaa2aSMark F. Adams /* PetscLogEventRegister(" SA: frmProl0", cookie, &gamg_setup_events[SET8]); */ 891b4fbaa2aSMark F. Adams PetscLogEventRegister(" SA: smooth", cookie, &gamg_setup_events[SET9]); 892b4fbaa2aSMark F. Adams PetscLogEventRegister("GAMG: partLevel", cookie, &gamg_setup_events[SET2]); 893b4fbaa2aSMark F. Adams PetscLogEventRegister(" PL repartition", cookie, &gamg_setup_events[SET12]); 894b4fbaa2aSMark F. Adams /* PetscLogEventRegister(" PL move data", cookie, &gamg_setup_events[SET13]); */ 895b4fbaa2aSMark F. Adams /* PetscLogEventRegister("GAMG: fix", cookie, &gamg_setup_events[SET10]); */ 896b4fbaa2aSMark F. Adams /* PetscLogEventRegister("GAMG: set levels", cookie, &gamg_setup_events[SET11]); */ 8975ef31b24SMark F. Adams 898b4fbaa2aSMark F. Adams /* create timer stages */ 899b4fbaa2aSMark F. Adams #if defined GAMG_STAGES 900b4fbaa2aSMark F. Adams { 901b4fbaa2aSMark F. Adams char str[32]; 902b4fbaa2aSMark F. Adams sprintf(str,"MG Level %d (finest)",0); 903b4fbaa2aSMark F. Adams PetscLogStageRegister(str, &gamg_stages[0]); 904b4fbaa2aSMark F. Adams PetscInt lidx; 905b4fbaa2aSMark F. Adams for (lidx=1;lidx<9;lidx++){ 906b4fbaa2aSMark F. Adams sprintf(str,"MG Level %d",lidx); 907b4fbaa2aSMark F. Adams PetscLogStageRegister(str, &gamg_stages[lidx]); 908b4fbaa2aSMark F. Adams } 909b4fbaa2aSMark F. Adams } 910b4fbaa2aSMark F. Adams #endif 911b4fbaa2aSMark F. Adams } 912b4fbaa2aSMark F. Adams #endif 9135b89ad90SMark F. Adams PetscFunctionReturn(0); 9145b89ad90SMark F. Adams } 9155b89ad90SMark F. Adams EXTERN_C_END 916