116d9e3a6SLisandro Dalcin 216d9e3a6SLisandro Dalcin /* 316d9e3a6SLisandro Dalcin Provides an interface to the LLNL package hypre 416d9e3a6SLisandro Dalcin */ 50f1074feSSatish Balay 60f1074feSSatish Balay /* Must use hypre 2.0.0 or more recent. */ 70f1074feSSatish Balay 8af0996ceSBarry Smith #include <petsc/private/pcimpl.h> /*I "petscpc.h" I*/ 949a781f5SStefano Zampini /* this include is needed ONLY to allow access to the private data inside the Mat object specific to hypre */ 1049a781f5SStefano Zampini #include <petsc/private/matimpl.h> 1158968eb6SStefano Zampini #include <../src/vec/vec/impls/hypre/vhyp.h> 1249a781f5SStefano Zampini #include <../src/mat/impls/hypre/mhypre.h> 13c6db04a5SJed Brown #include <../src/dm/impls/da/hypre/mhyp.h> 144cb006feSStefano Zampini #include <_hypre_parcsr_ls.h> 1516d9e3a6SLisandro Dalcin 16dff31646SBarry Smith static PetscBool cite = PETSC_FALSE; 171f817a21SBarry Smith static const char hypreCitation[] = "@manual{hypre-web-page,\n title = {{\\sl hypre}: High Performance Preconditioners},\n organization = {Lawrence Livermore National Laboratory},\n note = {\\url{http://www.llnl.gov/CASC/hypre/}}\n}\n"; 181f817a21SBarry Smith 1916d9e3a6SLisandro Dalcin /* 2016d9e3a6SLisandro Dalcin Private context (data structure) for the preconditioner. 2116d9e3a6SLisandro Dalcin */ 2216d9e3a6SLisandro Dalcin typedef struct { 2316d9e3a6SLisandro Dalcin HYPRE_Solver hsolver; 2449a781f5SStefano Zampini Mat hpmat; /* MatHYPRE */ 2516d9e3a6SLisandro Dalcin 264ddd07fcSJed Brown HYPRE_Int (*destroy)(HYPRE_Solver); 274ddd07fcSJed Brown HYPRE_Int (*solve)(HYPRE_Solver,HYPRE_ParCSRMatrix,HYPRE_ParVector,HYPRE_ParVector); 284ddd07fcSJed Brown HYPRE_Int (*setup)(HYPRE_Solver,HYPRE_ParCSRMatrix,HYPRE_ParVector,HYPRE_ParVector); 2916d9e3a6SLisandro Dalcin 3016d9e3a6SLisandro Dalcin MPI_Comm comm_hypre; 3116d9e3a6SLisandro Dalcin char *hypre_type; 3216d9e3a6SLisandro Dalcin 3316d9e3a6SLisandro Dalcin /* options for Pilut and BoomerAMG*/ 344ddd07fcSJed Brown PetscInt maxiter; 3516d9e3a6SLisandro Dalcin double tol; 3616d9e3a6SLisandro Dalcin 3716d9e3a6SLisandro Dalcin /* options for Pilut */ 384ddd07fcSJed Brown PetscInt factorrowsize; 3916d9e3a6SLisandro Dalcin 4016d9e3a6SLisandro Dalcin /* options for ParaSails */ 414ddd07fcSJed Brown PetscInt nlevels; 4216d9e3a6SLisandro Dalcin double threshhold; 4316d9e3a6SLisandro Dalcin double filter; 444ddd07fcSJed Brown PetscInt sym; 4516d9e3a6SLisandro Dalcin double loadbal; 464ddd07fcSJed Brown PetscInt logging; 474ddd07fcSJed Brown PetscInt ruse; 484ddd07fcSJed Brown PetscInt symt; 4916d9e3a6SLisandro Dalcin 5022b6d1caSBarry Smith /* options for BoomerAMG */ 51ace3abfcSBarry Smith PetscBool printstatistics; 5216d9e3a6SLisandro Dalcin 5316d9e3a6SLisandro Dalcin /* options for BoomerAMG */ 544ddd07fcSJed Brown PetscInt cycletype; 554ddd07fcSJed Brown PetscInt maxlevels; 5616d9e3a6SLisandro Dalcin double strongthreshold; 5716d9e3a6SLisandro Dalcin double maxrowsum; 584ddd07fcSJed Brown PetscInt gridsweeps[3]; 594ddd07fcSJed Brown PetscInt coarsentype; 604ddd07fcSJed Brown PetscInt measuretype; 616a251517SEike Mueller PetscInt smoothtype; 628131ecf7SEike Mueller PetscInt smoothnumlevels; 63ec64516dSEike Mueller PetscInt eu_level; /* Number of levels for ILU(k) in Euclid */ 64ec64516dSEike Mueller double eu_droptolerance; /* Drop tolerance for ILU(k) in Euclid */ 65ec64516dSEike Mueller PetscInt eu_bj; /* Defines use of Block Jacobi ILU in Euclid */ 664ddd07fcSJed Brown PetscInt relaxtype[3]; 6716d9e3a6SLisandro Dalcin double relaxweight; 6816d9e3a6SLisandro Dalcin double outerrelaxweight; 694ddd07fcSJed Brown PetscInt relaxorder; 7016d9e3a6SLisandro Dalcin double truncfactor; 71ace3abfcSBarry Smith PetscBool applyrichardson; 724ddd07fcSJed Brown PetscInt pmax; 734ddd07fcSJed Brown PetscInt interptype; 744ddd07fcSJed Brown PetscInt agg_nl; 754ddd07fcSJed Brown PetscInt agg_num_paths; 764ddd07fcSJed Brown PetscInt nodal_coarsen; 77ace3abfcSBarry Smith PetscBool nodal_relax; 784ddd07fcSJed Brown PetscInt nodal_relax_levels; 794cb006feSStefano Zampini 805272c319SBarry Smith PetscInt nodal_coarsening; 815272c319SBarry Smith PetscInt vec_interp_variant; 825272c319SBarry Smith HYPRE_IJVector *hmnull; 835272c319SBarry Smith HYPRE_ParVector *phmnull; /* near null space passed to hypre */ 845272c319SBarry Smith PetscInt n_hmnull; 855272c319SBarry Smith Vec hmnull_constant; 8672827435SBarry Smith PetscScalar **hmnull_hypre_data_array; /* this is the space in hmnull that was allocated by hypre, it is restored to hypre just before freeing the phmnull vectors */ 875272c319SBarry Smith 88863406b8SStefano Zampini /* options for AS (Auxiliary Space preconditioners) */ 89863406b8SStefano Zampini PetscInt as_print; 90863406b8SStefano Zampini PetscInt as_max_iter; 91863406b8SStefano Zampini PetscReal as_tol; 92863406b8SStefano Zampini PetscInt as_relax_type; 93863406b8SStefano Zampini PetscInt as_relax_times; 94863406b8SStefano Zampini PetscReal as_relax_weight; 95863406b8SStefano Zampini PetscReal as_omega; 96863406b8SStefano Zampini PetscInt as_amg_alpha_opts[5]; /* AMG coarsen type, agg_levels, relax_type, interp_type, Pmax for vector Poisson (AMS) or Curl problem (ADS) */ 97863406b8SStefano Zampini PetscReal as_amg_alpha_theta; /* AMG strength for vector Poisson (AMS) or Curl problem (ADS) */ 98863406b8SStefano Zampini PetscInt as_amg_beta_opts[5]; /* AMG coarsen type, agg_levels, relax_type, interp_type, Pmax for scalar Poisson (AMS) or vector Poisson (ADS) */ 99863406b8SStefano Zampini PetscReal as_amg_beta_theta; /* AMG strength for scalar Poisson (AMS) or vector Poisson (ADS) */ 1004cb006feSStefano Zampini PetscInt ams_cycle_type; 101863406b8SStefano Zampini PetscInt ads_cycle_type; 1024cb006feSStefano Zampini 1034cb006feSStefano Zampini /* additional data */ 1045ac14e1cSStefano Zampini Mat G; /* MatHYPRE */ 1055ac14e1cSStefano Zampini Mat C; /* MatHYPRE */ 1065ac14e1cSStefano Zampini Mat alpha_Poisson; /* MatHYPRE */ 1075ac14e1cSStefano Zampini Mat beta_Poisson; /* MatHYPRE */ 1085ac14e1cSStefano Zampini 1095ac14e1cSStefano Zampini /* extra information for AMS */ 1105ac14e1cSStefano Zampini PetscInt dim; /* geometrical dimension */ 1114cb006feSStefano Zampini HYPRE_IJVector coords[3]; 1124cb006feSStefano Zampini HYPRE_IJVector constants[3]; 1136bf688a0SCe Qin Mat RT_PiFull, RT_Pi[3]; 1146bf688a0SCe Qin Mat ND_PiFull, ND_Pi[3]; 1154cb006feSStefano Zampini PetscBool ams_beta_is_zero; 11623df4f25SStefano Zampini PetscBool ams_beta_is_zero_part; 11723df4f25SStefano Zampini PetscInt ams_proj_freq; 11816d9e3a6SLisandro Dalcin } PC_HYPRE; 11916d9e3a6SLisandro Dalcin 120d2128fa2SBarry Smith PetscErrorCode PCHYPREGetSolver(PC pc,HYPRE_Solver *hsolver) 121d2128fa2SBarry Smith { 122d2128fa2SBarry Smith PC_HYPRE *jac = (PC_HYPRE*)pc->data; 123d2128fa2SBarry Smith 124d2128fa2SBarry Smith PetscFunctionBegin; 125d2128fa2SBarry Smith *hsolver = jac->hsolver; 126d2128fa2SBarry Smith PetscFunctionReturn(0); 127d2128fa2SBarry Smith } 12816d9e3a6SLisandro Dalcin 12916d9e3a6SLisandro Dalcin static PetscErrorCode PCSetUp_HYPRE(PC pc) 13016d9e3a6SLisandro Dalcin { 13116d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 13249a781f5SStefano Zampini Mat_HYPRE *hjac; 13316d9e3a6SLisandro Dalcin HYPRE_ParCSRMatrix hmat; 13416d9e3a6SLisandro Dalcin HYPRE_ParVector bv,xv; 13549a781f5SStefano Zampini PetscBool ishypre; 13649a781f5SStefano Zampini PetscErrorCode ierr; 13716d9e3a6SLisandro Dalcin 13816d9e3a6SLisandro Dalcin PetscFunctionBegin; 13916d9e3a6SLisandro Dalcin if (!jac->hypre_type) { 14002a17cd4SBarry Smith ierr = PCHYPRESetType(pc,"boomeramg");CHKERRQ(ierr); 14116d9e3a6SLisandro Dalcin } 1425f5c5b43SBarry Smith 14349a781f5SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATHYPRE,&ishypre);CHKERRQ(ierr); 14449a781f5SStefano Zampini if (!ishypre) { 1456bf688a0SCe Qin ierr = MatDestroy(&jac->hpmat);CHKERRQ(ierr); 1466bf688a0SCe Qin ierr = MatConvert(pc->pmat,MATHYPRE,MAT_INITIAL_MATRIX,&jac->hpmat);CHKERRQ(ierr); 14749a781f5SStefano Zampini } else { 14849a781f5SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->pmat);CHKERRQ(ierr); 14949a781f5SStefano Zampini ierr = MatDestroy(&jac->hpmat);CHKERRQ(ierr); 15049a781f5SStefano Zampini jac->hpmat = pc->pmat; 15116d9e3a6SLisandro Dalcin } 15249a781f5SStefano Zampini hjac = (Mat_HYPRE*)(jac->hpmat->data); 1535f5c5b43SBarry Smith 15416d9e3a6SLisandro Dalcin /* special case for BoomerAMG */ 15516d9e3a6SLisandro Dalcin if (jac->setup == HYPRE_BoomerAMGSetup) { 1565272c319SBarry Smith MatNullSpace mnull; 1575272c319SBarry Smith PetscBool has_const; 15849a781f5SStefano Zampini PetscInt bs,nvec,i; 1595272c319SBarry Smith const Vec *vecs; 16072827435SBarry Smith PetscScalar *petscvecarray; 1615272c319SBarry Smith 16216d9e3a6SLisandro Dalcin ierr = MatGetBlockSize(pc->pmat,&bs);CHKERRQ(ierr); 1632fa5cd67SKarl Rupp if (bs > 1) PetscStackCallStandard(HYPRE_BoomerAMGSetNumFunctions,(jac->hsolver,bs)); 1645272c319SBarry Smith ierr = MatGetNearNullSpace(pc->mat, &mnull);CHKERRQ(ierr); 1655272c319SBarry Smith if (mnull) { 1665272c319SBarry Smith ierr = MatNullSpaceGetVecs(mnull, &has_const, &nvec, &vecs);CHKERRQ(ierr); 1675272c319SBarry Smith ierr = PetscMalloc1(nvec+1,&jac->hmnull);CHKERRQ(ierr); 16872827435SBarry Smith ierr = PetscMalloc1(nvec+1,&jac->hmnull_hypre_data_array);CHKERRQ(ierr); 1695272c319SBarry Smith ierr = PetscMalloc1(nvec+1,&jac->phmnull);CHKERRQ(ierr); 1705272c319SBarry Smith for (i=0; i<nvec; i++) { 1715272c319SBarry Smith ierr = VecHYPRE_IJVectorCreate(vecs[i],&jac->hmnull[i]);CHKERRQ(ierr); 17272827435SBarry Smith ierr = VecGetArrayRead(vecs[i],(const PetscScalar **)&petscvecarray);CHKERRQ(ierr); 17358968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(jac->hmnull[i],petscvecarray,jac->hmnull_hypre_data_array[i]); 17472827435SBarry Smith ierr = VecRestoreArrayRead(vecs[i],(const PetscScalar **)&petscvecarray);CHKERRQ(ierr); 1755272c319SBarry Smith PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->hmnull[i],(void**)&jac->phmnull[i])); 1765272c319SBarry Smith } 1775272c319SBarry Smith if (has_const) { 1785272c319SBarry Smith ierr = MatCreateVecs(pc->pmat,&jac->hmnull_constant,NULL);CHKERRQ(ierr); 1795272c319SBarry Smith ierr = VecSet(jac->hmnull_constant,1);CHKERRQ(ierr); 1805272c319SBarry Smith ierr = VecNormalize(jac->hmnull_constant,NULL); 1815272c319SBarry Smith ierr = VecHYPRE_IJVectorCreate(jac->hmnull_constant,&jac->hmnull[nvec]);CHKERRQ(ierr); 18272827435SBarry Smith ierr = VecGetArrayRead(jac->hmnull_constant,(const PetscScalar **)&petscvecarray);CHKERRQ(ierr); 18358968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(jac->hmnull[nvec],petscvecarray,jac->hmnull_hypre_data_array[nvec]); 18472827435SBarry Smith ierr = VecRestoreArrayRead(jac->hmnull_constant,(const PetscScalar **)&petscvecarray);CHKERRQ(ierr); 1855272c319SBarry Smith PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->hmnull[nvec],(void**)&jac->phmnull[nvec])); 1865272c319SBarry Smith nvec++; 1875272c319SBarry Smith } 1885272c319SBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetInterpVectors,(jac->hsolver,nvec,jac->phmnull)); 1895272c319SBarry Smith jac->n_hmnull = nvec; 1905272c319SBarry Smith } 1914cb006feSStefano Zampini } 192863406b8SStefano Zampini 1934cb006feSStefano Zampini /* special case for AMS */ 1944cb006feSStefano Zampini if (jac->setup == HYPRE_AMSSetup) { 1955ac14e1cSStefano Zampini Mat_HYPRE *hm; 1965ac14e1cSStefano Zampini HYPRE_ParCSRMatrix parcsr; 1976bf688a0SCe Qin if (!jac->coords[0] && !jac->constants[0] && !(jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1]))) { 1986bf688a0SCe Qin SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"HYPRE AMS preconditioner needs either the coordinate vectors via PCSetCoordinates() or the edge constant vectors via PCHYPRESetEdgeConstantVectors() or the interpolation matrix via PCHYPRESetInterpolations"); 1996bf688a0SCe Qin } 2005ac14e1cSStefano Zampini if (jac->dim) { 2015ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetDimension,(jac->hsolver,jac->dim)); 2025ac14e1cSStefano Zampini } 2035ac14e1cSStefano Zampini if (jac->constants[0]) { 2045ac14e1cSStefano Zampini HYPRE_ParVector ozz,zoz,zzo = NULL; 2055ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->constants[0],(void**)(&ozz))); 2065ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->constants[1],(void**)(&zoz))); 2075ac14e1cSStefano Zampini if (jac->constants[2]) { 2085ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->constants[2],(void**)(&zzo))); 2095ac14e1cSStefano Zampini } 2105ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetEdgeConstantVectors,(jac->hsolver,ozz,zoz,zzo)); 2115ac14e1cSStefano Zampini } 2125ac14e1cSStefano Zampini if (jac->coords[0]) { 2135ac14e1cSStefano Zampini HYPRE_ParVector coords[3]; 2145ac14e1cSStefano Zampini coords[0] = NULL; 2155ac14e1cSStefano Zampini coords[1] = NULL; 2165ac14e1cSStefano Zampini coords[2] = NULL; 2175ac14e1cSStefano Zampini if (jac->coords[0]) PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->coords[0],(void**)(&coords[0]))); 2185ac14e1cSStefano Zampini if (jac->coords[1]) PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->coords[1],(void**)(&coords[1]))); 2195ac14e1cSStefano Zampini if (jac->coords[2]) PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->coords[2],(void**)(&coords[2]))); 2205ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetCoordinateVectors,(jac->hsolver,coords[0],coords[1],coords[2])); 2215ac14e1cSStefano Zampini } 22249a781f5SStefano Zampini if (!jac->G) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"HYPRE AMS preconditioner needs the discrete gradient operator via PCHYPRESetDiscreteGradient"); 2235ac14e1cSStefano Zampini hm = (Mat_HYPRE*)(jac->G->data); 2245ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&parcsr))); 2255ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetDiscreteGradient,(jac->hsolver,parcsr)); 2265ac14e1cSStefano Zampini if (jac->alpha_Poisson) { 2275ac14e1cSStefano Zampini hm = (Mat_HYPRE*)(jac->alpha_Poisson->data); 2285ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&parcsr))); 2295ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetAlphaPoissonMatrix,(jac->hsolver,parcsr)); 2305ac14e1cSStefano Zampini } 2315ac14e1cSStefano Zampini if (jac->ams_beta_is_zero) { 2325ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetBetaPoissonMatrix,(jac->hsolver,NULL)); 2335ac14e1cSStefano Zampini } else if (jac->beta_Poisson) { 2345ac14e1cSStefano Zampini hm = (Mat_HYPRE*)(jac->beta_Poisson->data); 2355ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&parcsr))); 2365ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetBetaPoissonMatrix,(jac->hsolver,parcsr)); 2375ac14e1cSStefano Zampini } 2386bf688a0SCe Qin if (jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1])) { 2396bf688a0SCe Qin PetscInt i; 2406bf688a0SCe Qin HYPRE_ParCSRMatrix nd_parcsrfull, nd_parcsr[3]; 2416bf688a0SCe Qin if (jac->ND_PiFull) { 2426bf688a0SCe Qin hm = (Mat_HYPRE*)(jac->ND_PiFull->data); 2436bf688a0SCe Qin PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&nd_parcsrfull))); 2446bf688a0SCe Qin } else { 2456bf688a0SCe Qin nd_parcsrfull = NULL; 2466bf688a0SCe Qin } 2476bf688a0SCe Qin for (i=0;i<3;++i) { 2486bf688a0SCe Qin if (jac->ND_Pi[i]) { 2496bf688a0SCe Qin hm = (Mat_HYPRE*)(jac->ND_Pi[i]->data); 2506bf688a0SCe Qin PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&nd_parcsr[i]))); 2516bf688a0SCe Qin } else { 2526bf688a0SCe Qin nd_parcsr[i] = NULL; 2536bf688a0SCe Qin } 2546bf688a0SCe Qin } 2556bf688a0SCe Qin PetscStackCallStandard(HYPRE_AMSSetInterpolations,(jac->hsolver,nd_parcsrfull,nd_parcsr[0],nd_parcsr[1],nd_parcsr[2])); 2566bf688a0SCe Qin } 2574cb006feSStefano Zampini } 258863406b8SStefano Zampini /* special case for ADS */ 259863406b8SStefano Zampini if (jac->setup == HYPRE_ADSSetup) { 2605ac14e1cSStefano Zampini Mat_HYPRE *hm; 2615ac14e1cSStefano Zampini HYPRE_ParCSRMatrix parcsr; 2626bf688a0SCe Qin if (!jac->coords[0] && !((jac->RT_PiFull || (jac->RT_Pi[0] && jac->RT_Pi[1])) && (jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1])))) { 2636bf688a0SCe Qin SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"HYPRE ADS preconditioner needs either the coordinate vectors via PCSetCoordinates() or the interpolation matrices via PCHYPRESetInterpolations"); 2646bf688a0SCe Qin } 26537096e45SBarry Smith else if (!jac->coords[1] || !jac->coords[2]) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"HYPRE ADS preconditioner has been designed for three dimensional problems! For two dimensional problems, use HYPRE AMS instead"); 26649a781f5SStefano Zampini if (!jac->G) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"HYPRE ADS preconditioner needs the discrete gradient operator via PCHYPRESetDiscreteGradient"); 26749a781f5SStefano Zampini if (!jac->C) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_USER,"HYPRE ADS preconditioner needs the discrete curl operator via PCHYPRESetDiscreteGradient"); 2685ac14e1cSStefano Zampini if (jac->coords[0]) { 2695ac14e1cSStefano Zampini HYPRE_ParVector coords[3]; 2705ac14e1cSStefano Zampini coords[0] = NULL; 2715ac14e1cSStefano Zampini coords[1] = NULL; 2725ac14e1cSStefano Zampini coords[2] = NULL; 2735ac14e1cSStefano Zampini if (jac->coords[0]) PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->coords[0],(void**)(&coords[0]))); 2745ac14e1cSStefano Zampini if (jac->coords[1]) PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->coords[1],(void**)(&coords[1]))); 2755ac14e1cSStefano Zampini if (jac->coords[2]) PetscStackCallStandard(HYPRE_IJVectorGetObject,(jac->coords[2],(void**)(&coords[2]))); 2765ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_ADSSetCoordinateVectors,(jac->hsolver,coords[0],coords[1],coords[2])); 2775ac14e1cSStefano Zampini } 2785ac14e1cSStefano Zampini hm = (Mat_HYPRE*)(jac->G->data); 2795ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&parcsr))); 2805ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_ADSSetDiscreteGradient,(jac->hsolver,parcsr)); 2815ac14e1cSStefano Zampini hm = (Mat_HYPRE*)(jac->C->data); 2825ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&parcsr))); 2835ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_ADSSetDiscreteCurl,(jac->hsolver,parcsr)); 2846bf688a0SCe Qin if ((jac->RT_PiFull || (jac->RT_Pi[0] && jac->RT_Pi[1])) && (jac->ND_PiFull || (jac->ND_Pi[0] && jac->ND_Pi[1]))) { 2856bf688a0SCe Qin PetscInt i; 2866bf688a0SCe Qin HYPRE_ParCSRMatrix rt_parcsrfull, rt_parcsr[3]; 2876bf688a0SCe Qin HYPRE_ParCSRMatrix nd_parcsrfull, nd_parcsr[3]; 2886bf688a0SCe Qin if (jac->RT_PiFull) { 2896bf688a0SCe Qin hm = (Mat_HYPRE*)(jac->RT_PiFull->data); 2906bf688a0SCe Qin PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&rt_parcsrfull))); 2916bf688a0SCe Qin } else { 2926bf688a0SCe Qin rt_parcsrfull = NULL; 2936bf688a0SCe Qin } 2946bf688a0SCe Qin for (i=0;i<3;++i) { 2956bf688a0SCe Qin if (jac->RT_Pi[i]) { 2966bf688a0SCe Qin hm = (Mat_HYPRE*)(jac->RT_Pi[i]->data); 2976bf688a0SCe Qin PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&rt_parcsr[i]))); 2986bf688a0SCe Qin } else { 2996bf688a0SCe Qin rt_parcsr[i] = NULL; 3006bf688a0SCe Qin } 3016bf688a0SCe Qin } 3026bf688a0SCe Qin if (jac->ND_PiFull) { 3036bf688a0SCe Qin hm = (Mat_HYPRE*)(jac->ND_PiFull->data); 3046bf688a0SCe Qin PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&nd_parcsrfull))); 3056bf688a0SCe Qin } else { 3066bf688a0SCe Qin nd_parcsrfull = NULL; 3076bf688a0SCe Qin } 3086bf688a0SCe Qin for (i=0;i<3;++i) { 3096bf688a0SCe Qin if (jac->ND_Pi[i]) { 3106bf688a0SCe Qin hm = (Mat_HYPRE*)(jac->ND_Pi[i]->data); 3116bf688a0SCe Qin PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hm->ij,(void**)(&nd_parcsr[i]))); 3126bf688a0SCe Qin } else { 3136bf688a0SCe Qin nd_parcsr[i] = NULL; 3146bf688a0SCe Qin } 3156bf688a0SCe Qin } 3166bf688a0SCe Qin PetscStackCallStandard(HYPRE_ADSSetInterpolations,(jac->hsolver,rt_parcsrfull,rt_parcsr[0],rt_parcsr[1],rt_parcsr[2],nd_parcsrfull,nd_parcsr[0],nd_parcsr[1],nd_parcsr[2])); 3176bf688a0SCe Qin } 318863406b8SStefano Zampini } 31949a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hjac->ij,(void**)&hmat)); 32049a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hjac->b,(void**)&bv)); 32149a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hjac->x,(void**)&xv)); 322fd3f9acdSBarry Smith PetscStackCall("HYPRE_SetupXXX",ierr = (*jac->setup)(jac->hsolver,hmat,bv,xv);CHKERRQ(ierr);); 32316d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 32416d9e3a6SLisandro Dalcin } 32516d9e3a6SLisandro Dalcin 32616d9e3a6SLisandro Dalcin static PetscErrorCode PCApply_HYPRE(PC pc,Vec b,Vec x) 32716d9e3a6SLisandro Dalcin { 32816d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 32949a781f5SStefano Zampini Mat_HYPRE *hjac = (Mat_HYPRE*)(jac->hpmat->data); 33016d9e3a6SLisandro Dalcin PetscErrorCode ierr; 33116d9e3a6SLisandro Dalcin HYPRE_ParCSRMatrix hmat; 332d9ca1df4SBarry Smith PetscScalar *xv; 333d9ca1df4SBarry Smith const PetscScalar *bv,*sbv; 33416d9e3a6SLisandro Dalcin HYPRE_ParVector jbv,jxv; 335d9ca1df4SBarry Smith PetscScalar *sxv; 3364ddd07fcSJed Brown PetscInt hierr; 33716d9e3a6SLisandro Dalcin 33816d9e3a6SLisandro Dalcin PetscFunctionBegin; 339dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 34016d9e3a6SLisandro Dalcin if (!jac->applyrichardson) {ierr = VecSet(x,0.0);CHKERRQ(ierr);} 341d9ca1df4SBarry Smith ierr = VecGetArrayRead(b,&bv);CHKERRQ(ierr); 34216d9e3a6SLisandro Dalcin ierr = VecGetArray(x,&xv);CHKERRQ(ierr); 34358968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->b,(PetscScalar*)bv,sbv); 34458968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->x,xv,sxv); 34516d9e3a6SLisandro Dalcin 34649a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hjac->ij,(void**)&hmat)); 34749a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hjac->b,(void**)&jbv)); 34849a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hjac->x,(void**)&jxv)); 349fd3f9acdSBarry Smith PetscStackCall("Hypre solve",hierr = (*jac->solve)(jac->hsolver,hmat,jbv,jxv); 35065e19b50SBarry Smith if (hierr && hierr != HYPRE_ERROR_CONV) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in HYPRE solver, error code %d",hierr); 351fd3f9acdSBarry Smith if (hierr) hypre__global_error = 0;); 35216d9e3a6SLisandro Dalcin 35323df4f25SStefano Zampini if (jac->setup == HYPRE_AMSSetup && jac->ams_beta_is_zero_part) { 3545ac14e1cSStefano Zampini PetscStackCallStandard(HYPRE_AMSProjectOutGradients,(jac->hsolver,jxv)); 35521df291bSStefano Zampini } 35658968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->b,(PetscScalar*)sbv,bv); 35758968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->x,sxv,xv); 35816d9e3a6SLisandro Dalcin ierr = VecRestoreArray(x,&xv);CHKERRQ(ierr); 359d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(b,&bv);CHKERRQ(ierr); 36016d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 36116d9e3a6SLisandro Dalcin } 36216d9e3a6SLisandro Dalcin 3638695de01SBarry Smith static PetscErrorCode PCReset_HYPRE(PC pc) 3648695de01SBarry Smith { 3658695de01SBarry Smith PC_HYPRE *jac = (PC_HYPRE*)pc->data; 3668695de01SBarry Smith PetscErrorCode ierr; 3678695de01SBarry Smith 3688695de01SBarry Smith PetscFunctionBegin; 36949a781f5SStefano Zampini ierr = MatDestroy(&jac->hpmat);CHKERRQ(ierr); 3705ac14e1cSStefano Zampini ierr = MatDestroy(&jac->G);CHKERRQ(ierr); 3715ac14e1cSStefano Zampini ierr = MatDestroy(&jac->C);CHKERRQ(ierr); 3725ac14e1cSStefano Zampini ierr = MatDestroy(&jac->alpha_Poisson);CHKERRQ(ierr); 3735ac14e1cSStefano Zampini ierr = MatDestroy(&jac->beta_Poisson);CHKERRQ(ierr); 3746bf688a0SCe Qin ierr = MatDestroy(&jac->RT_PiFull);CHKERRQ(ierr); 3756bf688a0SCe Qin ierr = MatDestroy(&jac->RT_Pi[0]);CHKERRQ(ierr); 3766bf688a0SCe Qin ierr = MatDestroy(&jac->RT_Pi[1]);CHKERRQ(ierr); 3776bf688a0SCe Qin ierr = MatDestroy(&jac->RT_Pi[2]);CHKERRQ(ierr); 3786bf688a0SCe Qin ierr = MatDestroy(&jac->ND_PiFull);CHKERRQ(ierr); 3796bf688a0SCe Qin ierr = MatDestroy(&jac->ND_Pi[0]);CHKERRQ(ierr); 3806bf688a0SCe Qin ierr = MatDestroy(&jac->ND_Pi[1]);CHKERRQ(ierr); 3816bf688a0SCe Qin ierr = MatDestroy(&jac->ND_Pi[2]);CHKERRQ(ierr); 3828695de01SBarry Smith if (jac->coords[0]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->coords[0])); jac->coords[0] = NULL; 3838695de01SBarry Smith if (jac->coords[1]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->coords[1])); jac->coords[1] = NULL; 3848695de01SBarry Smith if (jac->coords[2]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->coords[2])); jac->coords[2] = NULL; 3858695de01SBarry Smith if (jac->constants[0]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->constants[0])); jac->constants[0] = NULL; 3868695de01SBarry Smith if (jac->constants[1]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->constants[1])); jac->constants[1] = NULL; 3878695de01SBarry Smith if (jac->constants[2]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->constants[2])); jac->constants[2] = NULL; 388550a8b7dSBarry Smith if (jac->n_hmnull && jac->hmnull) { 3895272c319SBarry Smith PetscInt i; 390b1c1cd91SBarry Smith PETSC_UNUSED PetscScalar *petscvecarray; 3915272c319SBarry Smith 3925272c319SBarry Smith for (i=0; i<jac->n_hmnull; i++) { 39358968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(jac->hmnull[i],jac->hmnull_hypre_data_array[i],petscvecarray); 3945272c319SBarry Smith PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->hmnull[i])); 3955272c319SBarry Smith } 3965272c319SBarry Smith ierr = PetscFree(jac->hmnull);CHKERRQ(ierr); 39772827435SBarry Smith ierr = PetscFree(jac->hmnull_hypre_data_array);CHKERRQ(ierr); 3985272c319SBarry Smith ierr = PetscFree(jac->phmnull);CHKERRQ(ierr); 3995272c319SBarry Smith ierr = VecDestroy(&jac->hmnull_constant);CHKERRQ(ierr); 4005272c319SBarry Smith } 4015ac14e1cSStefano Zampini jac->ams_beta_is_zero = PETSC_FALSE; 4025ac14e1cSStefano Zampini jac->dim = 0; 4038695de01SBarry Smith PetscFunctionReturn(0); 4048695de01SBarry Smith } 4058695de01SBarry Smith 40616d9e3a6SLisandro Dalcin static PetscErrorCode PCDestroy_HYPRE(PC pc) 40716d9e3a6SLisandro Dalcin { 40816d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 40916d9e3a6SLisandro Dalcin PetscErrorCode ierr; 41016d9e3a6SLisandro Dalcin 41116d9e3a6SLisandro Dalcin PetscFunctionBegin; 4128695de01SBarry Smith ierr = PCReset_HYPRE(pc);CHKERRQ(ierr); 413226b0620SJed Brown if (jac->destroy) PetscStackCall("HYPRE_DestroyXXX",ierr = (*jac->destroy)(jac->hsolver);CHKERRQ(ierr);); 414503cfb0cSBarry Smith ierr = PetscFree(jac->hypre_type);CHKERRQ(ierr); 41516d9e3a6SLisandro Dalcin if (jac->comm_hypre != MPI_COMM_NULL) { ierr = MPI_Comm_free(&(jac->comm_hypre));CHKERRQ(ierr);} 416c31cb41cSBarry Smith ierr = PetscFree(pc->data);CHKERRQ(ierr); 41716d9e3a6SLisandro Dalcin 41816d9e3a6SLisandro Dalcin ierr = PetscObjectChangeTypeName((PetscObject)pc,0);CHKERRQ(ierr); 419bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetType_C",NULL);CHKERRQ(ierr); 420bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPREGetType_C",NULL);CHKERRQ(ierr); 4214cb006feSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetCoordinates_C",NULL);CHKERRQ(ierr); 4224cb006feSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetDiscreteGradient_C",NULL);CHKERRQ(ierr); 423863406b8SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetDiscreteCurl_C",NULL);CHKERRQ(ierr); 4246bf688a0SCe Qin ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetInterpolations_C",NULL);CHKERRQ(ierr); 4254cb006feSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetConstantEdgeVectors_C",NULL);CHKERRQ(ierr); 4265ac14e1cSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetPoissonMatrix_C",NULL);CHKERRQ(ierr); 42716d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 42816d9e3a6SLisandro Dalcin } 42916d9e3a6SLisandro Dalcin 43016d9e3a6SLisandro Dalcin /* --------------------------------------------------------------------------------------------*/ 4314416b707SBarry Smith static PetscErrorCode PCSetFromOptions_HYPRE_Pilut(PetscOptionItems *PetscOptionsObject,PC pc) 43216d9e3a6SLisandro Dalcin { 43316d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 43416d9e3a6SLisandro Dalcin PetscErrorCode ierr; 435ace3abfcSBarry Smith PetscBool flag; 43616d9e3a6SLisandro Dalcin 43716d9e3a6SLisandro Dalcin PetscFunctionBegin; 438e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"HYPRE Pilut Options");CHKERRQ(ierr); 43916d9e3a6SLisandro Dalcin ierr = PetscOptionsInt("-pc_hypre_pilut_maxiter","Number of iterations","None",jac->maxiter,&jac->maxiter,&flag);CHKERRQ(ierr); 440fd3f9acdSBarry Smith if (flag) PetscStackCallStandard(HYPRE_ParCSRPilutSetMaxIter,(jac->hsolver,jac->maxiter)); 44116d9e3a6SLisandro Dalcin ierr = PetscOptionsReal("-pc_hypre_pilut_tol","Drop tolerance","None",jac->tol,&jac->tol,&flag);CHKERRQ(ierr); 442fd3f9acdSBarry Smith if (flag) PetscStackCallStandard(HYPRE_ParCSRPilutSetDropTolerance,(jac->hsolver,jac->tol)); 44316d9e3a6SLisandro Dalcin ierr = PetscOptionsInt("-pc_hypre_pilut_factorrowsize","FactorRowSize","None",jac->factorrowsize,&jac->factorrowsize,&flag);CHKERRQ(ierr); 444fd3f9acdSBarry Smith if (flag) PetscStackCallStandard(HYPRE_ParCSRPilutSetFactorRowSize,(jac->hsolver,jac->factorrowsize)); 44516d9e3a6SLisandro Dalcin ierr = PetscOptionsTail();CHKERRQ(ierr); 44616d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 44716d9e3a6SLisandro Dalcin } 44816d9e3a6SLisandro Dalcin 44916d9e3a6SLisandro Dalcin static PetscErrorCode PCView_HYPRE_Pilut(PC pc,PetscViewer viewer) 45016d9e3a6SLisandro Dalcin { 45116d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 45216d9e3a6SLisandro Dalcin PetscErrorCode ierr; 453ace3abfcSBarry Smith PetscBool iascii; 45416d9e3a6SLisandro Dalcin 45516d9e3a6SLisandro Dalcin PetscFunctionBegin; 456251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 45716d9e3a6SLisandro Dalcin if (iascii) { 45816d9e3a6SLisandro Dalcin ierr = PetscViewerASCIIPrintf(viewer," HYPRE Pilut preconditioning\n");CHKERRQ(ierr); 45916d9e3a6SLisandro Dalcin if (jac->maxiter != PETSC_DEFAULT) { 460efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," maximum number of iterations %d\n",jac->maxiter);CHKERRQ(ierr); 46116d9e3a6SLisandro Dalcin } else { 462efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," default maximum number of iterations \n");CHKERRQ(ierr); 46316d9e3a6SLisandro Dalcin } 46416d9e3a6SLisandro Dalcin if (jac->tol != PETSC_DEFAULT) { 465efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," drop tolerance %g\n",(double)jac->tol);CHKERRQ(ierr); 46616d9e3a6SLisandro Dalcin } else { 467efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," default drop tolerance \n");CHKERRQ(ierr); 46816d9e3a6SLisandro Dalcin } 46916d9e3a6SLisandro Dalcin if (jac->factorrowsize != PETSC_DEFAULT) { 470efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," factor row size %d\n",jac->factorrowsize);CHKERRQ(ierr); 47116d9e3a6SLisandro Dalcin } else { 472efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," default factor row size \n");CHKERRQ(ierr); 47316d9e3a6SLisandro Dalcin } 47416d9e3a6SLisandro Dalcin } 47516d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 47616d9e3a6SLisandro Dalcin } 47716d9e3a6SLisandro Dalcin 47816d9e3a6SLisandro Dalcin /* --------------------------------------------------------------------------------------------*/ 47916d9e3a6SLisandro Dalcin 48016d9e3a6SLisandro Dalcin static PetscErrorCode PCApplyTranspose_HYPRE_BoomerAMG(PC pc,Vec b,Vec x) 48116d9e3a6SLisandro Dalcin { 48216d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 48349a781f5SStefano Zampini Mat_HYPRE *hjac = (Mat_HYPRE*)(jac->hpmat->data); 48416d9e3a6SLisandro Dalcin PetscErrorCode ierr; 48516d9e3a6SLisandro Dalcin HYPRE_ParCSRMatrix hmat; 486d9ca1df4SBarry Smith PetscScalar *xv; 487d9ca1df4SBarry Smith const PetscScalar *bv; 48816d9e3a6SLisandro Dalcin HYPRE_ParVector jbv,jxv; 48916d9e3a6SLisandro Dalcin PetscScalar *sbv,*sxv; 4904ddd07fcSJed Brown PetscInt hierr; 49116d9e3a6SLisandro Dalcin 49216d9e3a6SLisandro Dalcin PetscFunctionBegin; 493dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 49416d9e3a6SLisandro Dalcin ierr = VecSet(x,0.0);CHKERRQ(ierr); 495d9ca1df4SBarry Smith ierr = VecGetArrayRead(b,&bv);CHKERRQ(ierr); 49616d9e3a6SLisandro Dalcin ierr = VecGetArray(x,&xv);CHKERRQ(ierr); 49758968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->b,(PetscScalar*)bv,sbv); 49858968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->x,xv,sxv); 49916d9e3a6SLisandro Dalcin 50049a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJMatrixGetObject,(hjac->ij,(void**)&hmat)); 50149a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hjac->b,(void**)&jbv)); 50249a781f5SStefano Zampini PetscStackCallStandard(HYPRE_IJVectorGetObject,(hjac->x,(void**)&jxv)); 50316d9e3a6SLisandro Dalcin 50416d9e3a6SLisandro Dalcin hierr = HYPRE_BoomerAMGSolveT(jac->hsolver,hmat,jbv,jxv); 50516d9e3a6SLisandro Dalcin /* error code of 1 in BoomerAMG merely means convergence not achieved */ 506e32f2f54SBarry Smith if (hierr && (hierr != 1)) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in HYPRE solver, error code %d",hierr); 50716d9e3a6SLisandro Dalcin if (hierr) hypre__global_error = 0; 50816d9e3a6SLisandro Dalcin 50958968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->b,sbv,bv); 51058968eb6SStefano Zampini VecHYPRE_ParVectorReplacePointer(hjac->x,sxv,xv); 51116d9e3a6SLisandro Dalcin ierr = VecRestoreArray(x,&xv);CHKERRQ(ierr); 512d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(b,&bv);CHKERRQ(ierr); 51316d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 51416d9e3a6SLisandro Dalcin } 51516d9e3a6SLisandro Dalcin 516a669f990SJed Brown /* static array length */ 517a669f990SJed Brown #define ALEN(a) (sizeof(a)/sizeof((a)[0])) 518a669f990SJed Brown 51916d9e3a6SLisandro Dalcin static const char *HYPREBoomerAMGCycleType[] = {"","V","W"}; 5200f1074feSSatish Balay static const char *HYPREBoomerAMGCoarsenType[] = {"CLJP","Ruge-Stueben","","modifiedRuge-Stueben","","","Falgout", "", "PMIS", "", "HMIS"}; 52116d9e3a6SLisandro Dalcin static const char *HYPREBoomerAMGMeasureType[] = {"local","global"}; 52265de4495SJed Brown /* The following corresponds to HYPRE_BoomerAMGSetRelaxType which has many missing numbers in the enum */ 5236a251517SEike Mueller static const char *HYPREBoomerAMGSmoothType[] = {"Schwarz-smoothers","Pilut","ParaSails","Euclid"}; 52465de4495SJed Brown static const char *HYPREBoomerAMGRelaxType[] = {"Jacobi","sequential-Gauss-Seidel","seqboundary-Gauss-Seidel","SOR/Jacobi","backward-SOR/Jacobi", 52565de4495SJed Brown "" /* [5] hybrid chaotic Gauss-Seidel (works only with OpenMP) */,"symmetric-SOR/Jacobi", 52665de4495SJed Brown "" /* 7 */,"l1scaled-SOR/Jacobi","Gaussian-elimination", 52765de4495SJed Brown "" /* 10 */, "" /* 11 */, "" /* 12 */, "" /* 13 */, "" /* 14 */, 52865de4495SJed Brown "CG" /* non-stationary */,"Chebyshev","FCF-Jacobi","l1scaled-Jacobi"}; 5290f1074feSSatish Balay static const char *HYPREBoomerAMGInterpType[] = {"classical", "", "", "direct", "multipass", "multipass-wts", "ext+i", 530e2287abbSStefano Zampini "ext+i-cc", "standard", "standard-wts", "block", "block-wtd", "FF", "FF1"}; 5314416b707SBarry Smith static PetscErrorCode PCSetFromOptions_HYPRE_BoomerAMG(PetscOptionItems *PetscOptionsObject,PC pc) 53216d9e3a6SLisandro Dalcin { 53316d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 53416d9e3a6SLisandro Dalcin PetscErrorCode ierr; 5354ddd07fcSJed Brown PetscInt n,indx,level; 536ace3abfcSBarry Smith PetscBool flg, tmp_truth; 53716d9e3a6SLisandro Dalcin double tmpdbl, twodbl[2]; 53816d9e3a6SLisandro Dalcin 53916d9e3a6SLisandro Dalcin PetscFunctionBegin; 540e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"HYPRE BoomerAMG Options");CHKERRQ(ierr); 5414336a9eeSBarry Smith ierr = PetscOptionsEList("-pc_hypre_boomeramg_cycle_type","Cycle type","None",HYPREBoomerAMGCycleType+1,2,HYPREBoomerAMGCycleType[jac->cycletype],&indx,&flg);CHKERRQ(ierr); 54216d9e3a6SLisandro Dalcin if (flg) { 5434336a9eeSBarry Smith jac->cycletype = indx+1; 544fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleType,(jac->hsolver,jac->cycletype)); 54516d9e3a6SLisandro Dalcin } 54616d9e3a6SLisandro Dalcin ierr = PetscOptionsInt("-pc_hypre_boomeramg_max_levels","Number of levels (of grids) allowed","None",jac->maxlevels,&jac->maxlevels,&flg);CHKERRQ(ierr); 54716d9e3a6SLisandro Dalcin if (flg) { 548ce94432eSBarry Smith if (jac->maxlevels < 2) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Number of levels %d must be at least two",jac->maxlevels); 549fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxLevels,(jac->hsolver,jac->maxlevels)); 55016d9e3a6SLisandro Dalcin } 55116d9e3a6SLisandro Dalcin ierr = PetscOptionsInt("-pc_hypre_boomeramg_max_iter","Maximum iterations used PER hypre call","None",jac->maxiter,&jac->maxiter,&flg);CHKERRQ(ierr); 55216d9e3a6SLisandro Dalcin if (flg) { 553ce94432eSBarry Smith if (jac->maxiter < 1) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Number of iterations %d must be at least one",jac->maxiter); 554fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxIter,(jac->hsolver,jac->maxiter)); 55516d9e3a6SLisandro Dalcin } 5560f1074feSSatish Balay ierr = PetscOptionsScalar("-pc_hypre_boomeramg_tol","Convergence tolerance PER hypre call (0.0 = use a fixed number of iterations)","None",jac->tol,&jac->tol,&flg);CHKERRQ(ierr); 55716d9e3a6SLisandro Dalcin if (flg) { 55857622a8eSBarry Smith if (jac->tol < 0.0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Tolerance %g must be greater than or equal to zero",(double)jac->tol); 559fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetTol,(jac->hsolver,jac->tol)); 56016d9e3a6SLisandro Dalcin } 56116d9e3a6SLisandro Dalcin 5620f1074feSSatish Balay ierr = PetscOptionsScalar("-pc_hypre_boomeramg_truncfactor","Truncation factor for interpolation (0=no truncation)","None",jac->truncfactor,&jac->truncfactor,&flg);CHKERRQ(ierr); 56316d9e3a6SLisandro Dalcin if (flg) { 56457622a8eSBarry Smith if (jac->truncfactor < 0.0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Truncation factor %g must be great than or equal zero",(double)jac->truncfactor); 565fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetTruncFactor,(jac->hsolver,jac->truncfactor)); 56616d9e3a6SLisandro Dalcin } 56716d9e3a6SLisandro Dalcin 5680f1074feSSatish Balay ierr = PetscOptionsInt("-pc_hypre_boomeramg_P_max","Max elements per row for interpolation operator (0=unlimited)","None",jac->pmax,&jac->pmax,&flg);CHKERRQ(ierr); 5690f1074feSSatish Balay if (flg) { 57057622a8eSBarry Smith if (jac->pmax < 0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"P_max %g must be greater than or equal to zero",(double)jac->pmax); 571fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetPMaxElmts,(jac->hsolver,jac->pmax)); 5720f1074feSSatish Balay } 5730f1074feSSatish Balay 5740f1074feSSatish Balay ierr = PetscOptionsInt("-pc_hypre_boomeramg_agg_nl","Number of levels of aggressive coarsening","None",jac->agg_nl,&jac->agg_nl,&flg);CHKERRQ(ierr); 5750f1074feSSatish Balay if (flg) { 57657622a8eSBarry Smith if (jac->agg_nl < 0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Number of levels %g must be greater than or equal to zero",(double)jac->agg_nl); 5770f1074feSSatish Balay 578fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetAggNumLevels,(jac->hsolver,jac->agg_nl)); 5790f1074feSSatish Balay } 5800f1074feSSatish Balay 5810f1074feSSatish Balay 5820f1074feSSatish Balay ierr = PetscOptionsInt("-pc_hypre_boomeramg_agg_num_paths","Number of paths for aggressive coarsening","None",jac->agg_num_paths,&jac->agg_num_paths,&flg);CHKERRQ(ierr); 5830f1074feSSatish Balay if (flg) { 58457622a8eSBarry Smith if (jac->agg_num_paths < 1) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Number of paths %g must be greater than or equal to 1",(double)jac->agg_num_paths); 5850f1074feSSatish Balay 586fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetNumPaths,(jac->hsolver,jac->agg_num_paths)); 5870f1074feSSatish Balay } 5880f1074feSSatish Balay 5890f1074feSSatish Balay 59016d9e3a6SLisandro Dalcin ierr = PetscOptionsScalar("-pc_hypre_boomeramg_strong_threshold","Threshold for being strongly connected","None",jac->strongthreshold,&jac->strongthreshold,&flg);CHKERRQ(ierr); 59116d9e3a6SLisandro Dalcin if (flg) { 59257622a8eSBarry Smith if (jac->strongthreshold < 0.0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Strong threshold %g must be great than or equal zero",(double)jac->strongthreshold); 593fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetStrongThreshold,(jac->hsolver,jac->strongthreshold)); 59416d9e3a6SLisandro Dalcin } 59516d9e3a6SLisandro Dalcin ierr = PetscOptionsScalar("-pc_hypre_boomeramg_max_row_sum","Maximum row sum","None",jac->maxrowsum,&jac->maxrowsum,&flg);CHKERRQ(ierr); 59616d9e3a6SLisandro Dalcin if (flg) { 59757622a8eSBarry Smith if (jac->maxrowsum < 0.0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Maximum row sum %g must be greater than zero",(double)jac->maxrowsum); 59857622a8eSBarry Smith if (jac->maxrowsum > 1.0) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Maximum row sum %g must be less than or equal one",(double)jac->maxrowsum); 599fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxRowSum,(jac->hsolver,jac->maxrowsum)); 60016d9e3a6SLisandro Dalcin } 60116d9e3a6SLisandro Dalcin 60216d9e3a6SLisandro Dalcin /* Grid sweeps */ 6030f1074feSSatish Balay ierr = PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_all","Number of sweeps for the up and down grid levels","None",jac->gridsweeps[0],&indx,&flg);CHKERRQ(ierr); 60416d9e3a6SLisandro Dalcin if (flg) { 605fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetNumSweeps,(jac->hsolver,indx)); 60616d9e3a6SLisandro Dalcin /* modify the jac structure so we can view the updated options with PC_View */ 60716d9e3a6SLisandro Dalcin jac->gridsweeps[0] = indx; 6080f1074feSSatish Balay jac->gridsweeps[1] = indx; 6090f1074feSSatish Balay /*defaults coarse to 1 */ 6100f1074feSSatish Balay jac->gridsweeps[2] = 1; 61116d9e3a6SLisandro Dalcin } 6120f1074feSSatish Balay 6135272c319SBarry Smith ierr = PetscOptionsInt("-pc_hypre_boomeramg_nodal_coarsen","Use a nodal based coarsening 1-6","HYPRE_BoomerAMGSetNodal",jac->nodal_coarsening, &jac->nodal_coarsening,&flg);CHKERRQ(ierr); 6145272c319SBarry Smith if (flg) { 6155272c319SBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetNodal,(jac->hsolver,jac->nodal_coarsening)); 6165272c319SBarry Smith } 6175272c319SBarry Smith 618cbc39033SBarry Smith ierr = PetscOptionsInt("-pc_hypre_boomeramg_vec_interp_variant","Variant of algorithm 1-3","HYPRE_BoomerAMGSetInterpVecVariant",jac->vec_interp_variant, &jac->vec_interp_variant,&flg);CHKERRQ(ierr); 6195272c319SBarry Smith if (flg) { 6205272c319SBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetInterpVecVariant,(jac->hsolver,jac->vec_interp_variant)); 6215272c319SBarry Smith } 6225272c319SBarry Smith 6230f1074feSSatish Balay ierr = PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_down","Number of sweeps for the down cycles","None",jac->gridsweeps[0], &indx,&flg);CHKERRQ(ierr); 62416d9e3a6SLisandro Dalcin if (flg) { 625fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleNumSweeps,(jac->hsolver,indx, 1)); 6260f1074feSSatish Balay jac->gridsweeps[0] = indx; 62716d9e3a6SLisandro Dalcin } 62816d9e3a6SLisandro Dalcin ierr = PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_up","Number of sweeps for the up cycles","None",jac->gridsweeps[1],&indx,&flg);CHKERRQ(ierr); 62916d9e3a6SLisandro Dalcin if (flg) { 630fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleNumSweeps,(jac->hsolver,indx, 2)); 6310f1074feSSatish Balay jac->gridsweeps[1] = indx; 63216d9e3a6SLisandro Dalcin } 6330f1074feSSatish Balay ierr = PetscOptionsInt("-pc_hypre_boomeramg_grid_sweeps_coarse","Number of sweeps for the coarse level","None",jac->gridsweeps[2],&indx,&flg);CHKERRQ(ierr); 63416d9e3a6SLisandro Dalcin if (flg) { 635fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleNumSweeps,(jac->hsolver,indx, 3)); 6360f1074feSSatish Balay jac->gridsweeps[2] = indx; 63716d9e3a6SLisandro Dalcin } 63816d9e3a6SLisandro Dalcin 6396a251517SEike Mueller /* Smooth type */ 6406a251517SEike Mueller ierr = PetscOptionsEList("-pc_hypre_boomeramg_smooth_type","Enable more complex smoothers","None",HYPREBoomerAMGSmoothType,ALEN(HYPREBoomerAMGSmoothType),HYPREBoomerAMGSmoothType[0],&indx,&flg); 6416a251517SEike Mueller if (flg) { 6426a251517SEike Mueller jac->smoothtype = indx; 6436a251517SEike Mueller PetscStackCallStandard(HYPRE_BoomerAMGSetSmoothType,(jac->hsolver,indx+6)); 6448131ecf7SEike Mueller jac->smoothnumlevels = 25; 6458131ecf7SEike Mueller PetscStackCallStandard(HYPRE_BoomerAMGSetSmoothNumLevels,(jac->hsolver,25)); 6468131ecf7SEike Mueller } 6478131ecf7SEike Mueller 6488131ecf7SEike Mueller /* Number of smoothing levels */ 6498131ecf7SEike Mueller ierr = PetscOptionsInt("-pc_hypre_boomeramg_smooth_num_levels","Number of levels on which more complex smoothers are used","None",25,&indx,&flg);CHKERRQ(ierr); 6508131ecf7SEike Mueller if (flg && (jac->smoothtype != -1)) { 6518131ecf7SEike Mueller jac->smoothnumlevels = indx; 6528131ecf7SEike Mueller PetscStackCallStandard(HYPRE_BoomerAMGSetSmoothNumLevels,(jac->hsolver,indx)); 6536a251517SEike Mueller } 6546a251517SEike Mueller 6551810e44eSEike Mueller /* Number of levels for ILU(k) for Euclid */ 6561810e44eSEike Mueller ierr = PetscOptionsInt("-pc_hypre_boomeramg_eu_level","Number of levels for ILU(k) in Euclid smoother","None",0,&indx,&flg);CHKERRQ(ierr); 6571810e44eSEike Mueller if (flg && (jac->smoothtype == 3)) { 6581810e44eSEike Mueller jac->eu_level = indx; 6591810e44eSEike Mueller PetscStackCallStandard(HYPRE_BoomerAMGSetEuLevel,(jac->hsolver,indx)); 6601810e44eSEike Mueller } 6611810e44eSEike Mueller 6621810e44eSEike Mueller /* Filter for ILU(k) for Euclid */ 6631810e44eSEike Mueller double droptolerance; 6641810e44eSEike Mueller ierr = PetscOptionsScalar("-pc_hypre_boomeramg_eu_droptolerance","Drop tolerance for ILU(k) in Euclid smoother","None",0,&droptolerance,&flg);CHKERRQ(ierr); 6651810e44eSEike Mueller if (flg && (jac->smoothtype == 3)) { 6661810e44eSEike Mueller jac->eu_droptolerance = droptolerance; 6671810e44eSEike Mueller PetscStackCallStandard(HYPRE_BoomerAMGSetEuLevel,(jac->hsolver,droptolerance)); 6681810e44eSEike Mueller } 6691810e44eSEike Mueller 6701810e44eSEike Mueller /* Use Block Jacobi ILUT for Euclid */ 6711810e44eSEike Mueller ierr = PetscOptionsBool("-pc_hypre_boomeramg_eu_bj", "Use Block Jacobi for ILU in Euclid smoother?", "None", PETSC_FALSE, &tmp_truth, &flg);CHKERRQ(ierr); 6721810e44eSEike Mueller if (flg && (jac->smoothtype == 3)) { 6731810e44eSEike Mueller jac->eu_bj = tmp_truth; 674493fc9d9SEike Mueller PetscStackCallStandard(HYPRE_BoomerAMGSetEuBJ,(jac->hsolver,jac->eu_bj)); 6751810e44eSEike Mueller } 6761810e44eSEike Mueller 67716d9e3a6SLisandro Dalcin /* Relax type */ 678a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_relax_type_all","Relax type for the up and down cycles","None",HYPREBoomerAMGRelaxType,ALEN(HYPREBoomerAMGRelaxType),HYPREBoomerAMGRelaxType[6],&indx,&flg);CHKERRQ(ierr); 67916d9e3a6SLisandro Dalcin if (flg) { 6800f1074feSSatish Balay jac->relaxtype[0] = jac->relaxtype[1] = indx; 681fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetRelaxType,(jac->hsolver, indx)); 6820f1074feSSatish Balay /* by default, coarse type set to 9 */ 6830f1074feSSatish Balay jac->relaxtype[2] = 9; 6840f1074feSSatish Balay 68516d9e3a6SLisandro Dalcin } 686a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_relax_type_down","Relax type for the down cycles","None",HYPREBoomerAMGRelaxType,ALEN(HYPREBoomerAMGRelaxType),HYPREBoomerAMGRelaxType[6],&indx,&flg);CHKERRQ(ierr); 68716d9e3a6SLisandro Dalcin if (flg) { 68816d9e3a6SLisandro Dalcin jac->relaxtype[0] = indx; 689fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleRelaxType,(jac->hsolver, indx, 1)); 69016d9e3a6SLisandro Dalcin } 691a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_relax_type_up","Relax type for the up cycles","None",HYPREBoomerAMGRelaxType,ALEN(HYPREBoomerAMGRelaxType),HYPREBoomerAMGRelaxType[6],&indx,&flg);CHKERRQ(ierr); 69216d9e3a6SLisandro Dalcin if (flg) { 6930f1074feSSatish Balay jac->relaxtype[1] = indx; 694fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleRelaxType,(jac->hsolver, indx, 2)); 69516d9e3a6SLisandro Dalcin } 696a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_relax_type_coarse","Relax type on coarse grid","None",HYPREBoomerAMGRelaxType,ALEN(HYPREBoomerAMGRelaxType),HYPREBoomerAMGRelaxType[9],&indx,&flg);CHKERRQ(ierr); 69716d9e3a6SLisandro Dalcin if (flg) { 6980f1074feSSatish Balay jac->relaxtype[2] = indx; 699fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleRelaxType,(jac->hsolver, indx, 3)); 70016d9e3a6SLisandro Dalcin } 70116d9e3a6SLisandro Dalcin 70216d9e3a6SLisandro Dalcin /* Relaxation Weight */ 70316d9e3a6SLisandro Dalcin ierr = PetscOptionsReal("-pc_hypre_boomeramg_relax_weight_all","Relaxation weight for all levels (0 = hypre estimates, -k = determined with k CG steps)","None",jac->relaxweight, &tmpdbl,&flg);CHKERRQ(ierr); 70416d9e3a6SLisandro Dalcin if (flg) { 705fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetRelaxWt,(jac->hsolver,tmpdbl)); 70616d9e3a6SLisandro Dalcin jac->relaxweight = tmpdbl; 70716d9e3a6SLisandro Dalcin } 70816d9e3a6SLisandro Dalcin 70916d9e3a6SLisandro Dalcin n = 2; 71016d9e3a6SLisandro Dalcin twodbl[0] = twodbl[1] = 1.0; 71116d9e3a6SLisandro Dalcin ierr = PetscOptionsRealArray("-pc_hypre_boomeramg_relax_weight_level","Set the relaxation weight for a particular level (weight,level)","None",twodbl, &n, &flg);CHKERRQ(ierr); 71216d9e3a6SLisandro Dalcin if (flg) { 71316d9e3a6SLisandro Dalcin if (n == 2) { 71416d9e3a6SLisandro Dalcin indx = (int)PetscAbsReal(twodbl[1]); 715fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetLevelRelaxWt,(jac->hsolver,twodbl[0],indx)); 716ce94432eSBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Relax weight level: you must provide 2 values separated by a comma (and no space), you provided %d",n); 71716d9e3a6SLisandro Dalcin } 71816d9e3a6SLisandro Dalcin 71916d9e3a6SLisandro Dalcin /* Outer relaxation Weight */ 72016d9e3a6SLisandro Dalcin ierr = PetscOptionsReal("-pc_hypre_boomeramg_outer_relax_weight_all","Outer relaxation weight for all levels (-k = determined with k CG steps)","None",jac->outerrelaxweight, &tmpdbl,&flg);CHKERRQ(ierr); 72116d9e3a6SLisandro Dalcin if (flg) { 722fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetOuterWt,(jac->hsolver, tmpdbl)); 72316d9e3a6SLisandro Dalcin jac->outerrelaxweight = tmpdbl; 72416d9e3a6SLisandro Dalcin } 72516d9e3a6SLisandro Dalcin 72616d9e3a6SLisandro Dalcin n = 2; 72716d9e3a6SLisandro Dalcin twodbl[0] = twodbl[1] = 1.0; 72816d9e3a6SLisandro Dalcin ierr = PetscOptionsRealArray("-pc_hypre_boomeramg_outer_relax_weight_level","Set the outer relaxation weight for a particular level (weight,level)","None",twodbl, &n, &flg);CHKERRQ(ierr); 72916d9e3a6SLisandro Dalcin if (flg) { 73016d9e3a6SLisandro Dalcin if (n == 2) { 73116d9e3a6SLisandro Dalcin indx = (int)PetscAbsReal(twodbl[1]); 732fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetLevelOuterWt,(jac->hsolver, twodbl[0], indx)); 733ce94432eSBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_OUTOFRANGE,"Relax weight outer level: You must provide 2 values separated by a comma (and no space), you provided %d",n); 73416d9e3a6SLisandro Dalcin } 73516d9e3a6SLisandro Dalcin 73616d9e3a6SLisandro Dalcin /* the Relax Order */ 737acfcf0e5SJed Brown ierr = PetscOptionsBool("-pc_hypre_boomeramg_no_CF", "Do not use CF-relaxation", "None", PETSC_FALSE, &tmp_truth, &flg);CHKERRQ(ierr); 73816d9e3a6SLisandro Dalcin 7398afaa268SBarry Smith if (flg && tmp_truth) { 74016d9e3a6SLisandro Dalcin jac->relaxorder = 0; 741fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetRelaxOrder,(jac->hsolver, jac->relaxorder)); 74216d9e3a6SLisandro Dalcin } 743a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_measure_type","Measure type","None",HYPREBoomerAMGMeasureType,ALEN(HYPREBoomerAMGMeasureType),HYPREBoomerAMGMeasureType[0],&indx,&flg);CHKERRQ(ierr); 74416d9e3a6SLisandro Dalcin if (flg) { 74516d9e3a6SLisandro Dalcin jac->measuretype = indx; 746fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMeasureType,(jac->hsolver,jac->measuretype)); 74716d9e3a6SLisandro Dalcin } 7480f1074feSSatish Balay /* update list length 3/07 */ 749a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_coarsen_type","Coarsen type","None",HYPREBoomerAMGCoarsenType,ALEN(HYPREBoomerAMGCoarsenType),HYPREBoomerAMGCoarsenType[6],&indx,&flg);CHKERRQ(ierr); 75016d9e3a6SLisandro Dalcin if (flg) { 75116d9e3a6SLisandro Dalcin jac->coarsentype = indx; 752fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCoarsenType,(jac->hsolver,jac->coarsentype)); 75316d9e3a6SLisandro Dalcin } 7540f1074feSSatish Balay 7550f1074feSSatish Balay /* new 3/07 */ 756a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_boomeramg_interp_type","Interpolation type","None",HYPREBoomerAMGInterpType,ALEN(HYPREBoomerAMGInterpType),HYPREBoomerAMGInterpType[0],&indx,&flg);CHKERRQ(ierr); 7570f1074feSSatish Balay if (flg) { 7580f1074feSSatish Balay jac->interptype = indx; 759fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetInterpType,(jac->hsolver,jac->interptype)); 7600f1074feSSatish Balay } 7610f1074feSSatish Balay 762b96a4a96SBarry Smith ierr = PetscOptionsName("-pc_hypre_boomeramg_print_statistics","Print statistics","None",&flg);CHKERRQ(ierr); 76316d9e3a6SLisandro Dalcin if (flg) { 764b96a4a96SBarry Smith level = 3; 7650298fd71SBarry Smith ierr = PetscOptionsInt("-pc_hypre_boomeramg_print_statistics","Print statistics","None",level,&level,NULL);CHKERRQ(ierr); 7662fa5cd67SKarl Rupp 767b96a4a96SBarry Smith jac->printstatistics = PETSC_TRUE; 768fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetPrintLevel,(jac->hsolver,level)); 7692ae77aedSBarry Smith } 7702ae77aedSBarry Smith 771b96a4a96SBarry Smith ierr = PetscOptionsName("-pc_hypre_boomeramg_print_debug","Print debug information","None",&flg);CHKERRQ(ierr); 7722ae77aedSBarry Smith if (flg) { 773b96a4a96SBarry Smith level = 3; 7740298fd71SBarry Smith ierr = PetscOptionsInt("-pc_hypre_boomeramg_print_debug","Print debug information","None",level,&level,NULL);CHKERRQ(ierr); 7752fa5cd67SKarl Rupp 776b96a4a96SBarry Smith jac->printstatistics = PETSC_TRUE; 777fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetDebugFlag,(jac->hsolver,level)); 77816d9e3a6SLisandro Dalcin } 7798f87f92bSBarry Smith 780acfcf0e5SJed Brown ierr = PetscOptionsBool("-pc_hypre_boomeramg_nodal_relaxation", "Nodal relaxation via Schwarz", "None", PETSC_FALSE, &tmp_truth, &flg);CHKERRQ(ierr); 7818f87f92bSBarry Smith if (flg && tmp_truth) { 7828f87f92bSBarry Smith PetscInt tmp_int; 7838f87f92bSBarry Smith ierr = PetscOptionsInt("-pc_hypre_boomeramg_nodal_relaxation", "Nodal relaxation via Schwarz", "None",jac->nodal_relax_levels,&tmp_int,&flg);CHKERRQ(ierr); 7848f87f92bSBarry Smith if (flg) jac->nodal_relax_levels = tmp_int; 785fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetSmoothType,(jac->hsolver,6)); 786fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetDomainType,(jac->hsolver,1)); 787fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetOverlap,(jac->hsolver,0)); 788fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetSmoothNumLevels,(jac->hsolver,jac->nodal_relax_levels)); 7898f87f92bSBarry Smith } 7908f87f92bSBarry Smith 79116d9e3a6SLisandro Dalcin ierr = PetscOptionsTail();CHKERRQ(ierr); 79216d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 79316d9e3a6SLisandro Dalcin } 79416d9e3a6SLisandro Dalcin 795ace3abfcSBarry Smith static PetscErrorCode PCApplyRichardson_HYPRE_BoomerAMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol, PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt *outits,PCRichardsonConvergedReason *reason) 79616d9e3a6SLisandro Dalcin { 79716d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 79816d9e3a6SLisandro Dalcin PetscErrorCode ierr; 7994ddd07fcSJed Brown PetscInt oits; 80016d9e3a6SLisandro Dalcin 80116d9e3a6SLisandro Dalcin PetscFunctionBegin; 802dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 803fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxIter,(jac->hsolver,its*jac->maxiter)); 804fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetTol,(jac->hsolver,rtol)); 80516d9e3a6SLisandro Dalcin jac->applyrichardson = PETSC_TRUE; 80616d9e3a6SLisandro Dalcin ierr = PCApply_HYPRE(pc,b,y);CHKERRQ(ierr); 80716d9e3a6SLisandro Dalcin jac->applyrichardson = PETSC_FALSE; 8088b1f7689SBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGGetNumIterations,(jac->hsolver,(HYPRE_Int *)&oits)); 8094d0a8057SBarry Smith *outits = oits; 8104d0a8057SBarry Smith if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS; 8114d0a8057SBarry Smith else *reason = PCRICHARDSON_CONVERGED_RTOL; 812fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetTol,(jac->hsolver,jac->tol)); 813fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxIter,(jac->hsolver,jac->maxiter)); 81416d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 81516d9e3a6SLisandro Dalcin } 81616d9e3a6SLisandro Dalcin 81716d9e3a6SLisandro Dalcin 81816d9e3a6SLisandro Dalcin static PetscErrorCode PCView_HYPRE_BoomerAMG(PC pc,PetscViewer viewer) 81916d9e3a6SLisandro Dalcin { 82016d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 82116d9e3a6SLisandro Dalcin PetscErrorCode ierr; 822ace3abfcSBarry Smith PetscBool iascii; 82316d9e3a6SLisandro Dalcin 82416d9e3a6SLisandro Dalcin PetscFunctionBegin; 825251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 82616d9e3a6SLisandro Dalcin if (iascii) { 82716d9e3a6SLisandro Dalcin ierr = PetscViewerASCIIPrintf(viewer," HYPRE BoomerAMG preconditioning\n");CHKERRQ(ierr); 828efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Cycle type %s\n",HYPREBoomerAMGCycleType[jac->cycletype]);CHKERRQ(ierr); 829efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Maximum number of levels %d\n",jac->maxlevels);CHKERRQ(ierr); 830efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Maximum number of iterations PER hypre call %d\n",jac->maxiter);CHKERRQ(ierr); 831efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Convergence tolerance PER hypre call %g\n",(double)jac->tol);CHKERRQ(ierr); 832efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Threshold for strong coupling %g\n",(double)jac->strongthreshold);CHKERRQ(ierr); 833efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Interpolation truncation factor %g\n",(double)jac->truncfactor);CHKERRQ(ierr); 834efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Interpolation: max elements per row %d\n",jac->pmax);CHKERRQ(ierr); 835efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Number of levels of aggressive coarsening %d\n",jac->agg_nl);CHKERRQ(ierr); 836efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Number of paths for aggressive coarsening %d\n",jac->agg_num_paths);CHKERRQ(ierr); 8370f1074feSSatish Balay 838efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Maximum row sums %g\n",(double)jac->maxrowsum);CHKERRQ(ierr); 83916d9e3a6SLisandro Dalcin 840efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Sweeps down %d\n",jac->gridsweeps[0]);CHKERRQ(ierr); 841efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Sweeps up %d\n",jac->gridsweeps[1]);CHKERRQ(ierr); 842efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Sweeps on coarse %d\n",jac->gridsweeps[2]);CHKERRQ(ierr); 84316d9e3a6SLisandro Dalcin 844efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Relax down %s\n",HYPREBoomerAMGRelaxType[jac->relaxtype[0]]);CHKERRQ(ierr); 845efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Relax up %s\n",HYPREBoomerAMGRelaxType[jac->relaxtype[1]]);CHKERRQ(ierr); 846efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Relax on coarse %s\n",HYPREBoomerAMGRelaxType[jac->relaxtype[2]]);CHKERRQ(ierr); 84716d9e3a6SLisandro Dalcin 848efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Relax weight (all) %g\n",(double)jac->relaxweight);CHKERRQ(ierr); 849efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Outer relax weight (all) %g\n",(double)jac->outerrelaxweight);CHKERRQ(ierr); 85016d9e3a6SLisandro Dalcin 85116d9e3a6SLisandro Dalcin if (jac->relaxorder) { 852efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Using CF-relaxation\n");CHKERRQ(ierr); 85316d9e3a6SLisandro Dalcin } else { 854efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Not using CF-relaxation\n");CHKERRQ(ierr); 85516d9e3a6SLisandro Dalcin } 8566a251517SEike Mueller if (jac->smoothtype!=-1) { 857efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Smooth type %s\n",HYPREBoomerAMGSmoothType[jac->smoothtype]);CHKERRQ(ierr); 858efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Smooth num levels %d\n",jac->smoothnumlevels);CHKERRQ(ierr); 8597e352d70SEike Mueller } else { 860efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Not using more complex smoothers.\n");CHKERRQ(ierr); 8611810e44eSEike Mueller } 8621810e44eSEike Mueller if (jac->smoothtype==3) { 863efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Euclid ILU(k) levels %d\n",jac->eu_level);CHKERRQ(ierr); 864efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Euclid ILU(k) drop tolerance %g\n",jac->eu_droptolerance);CHKERRQ(ierr); 865efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Euclid ILU use Block-Jacobi? %d\n",jac->eu_bj);CHKERRQ(ierr); 8666a251517SEike Mueller } 867efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Measure type %s\n",HYPREBoomerAMGMeasureType[jac->measuretype]);CHKERRQ(ierr); 868efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Coarsen type %s\n",HYPREBoomerAMGCoarsenType[jac->coarsentype]);CHKERRQ(ierr); 869efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Interpolation type %s\n",HYPREBoomerAMGInterpType[jac->interptype]);CHKERRQ(ierr); 8705272c319SBarry Smith if (jac->nodal_coarsening) { 871efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Using nodal coarsening (with HYPRE_BOOMERAMGSetNodal() %D\n",jac->nodal_coarsening);CHKERRQ(ierr); 8725272c319SBarry Smith } 8735272c319SBarry Smith if (jac->vec_interp_variant) { 874efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," HYPRE_BoomerAMGSetInterpVecVariant() %D\n",jac->vec_interp_variant);CHKERRQ(ierr); 8758f87f92bSBarry Smith } 8768f87f92bSBarry Smith if (jac->nodal_relax) { 877efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Using nodal relaxation via Schwarz smoothing on levels %d\n",jac->nodal_relax_levels);CHKERRQ(ierr); 8788f87f92bSBarry Smith } 87916d9e3a6SLisandro Dalcin } 88016d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 88116d9e3a6SLisandro Dalcin } 88216d9e3a6SLisandro Dalcin 88316d9e3a6SLisandro Dalcin /* --------------------------------------------------------------------------------------------*/ 8844416b707SBarry Smith static PetscErrorCode PCSetFromOptions_HYPRE_ParaSails(PetscOptionItems *PetscOptionsObject,PC pc) 88516d9e3a6SLisandro Dalcin { 88616d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 88716d9e3a6SLisandro Dalcin PetscErrorCode ierr; 8884ddd07fcSJed Brown PetscInt indx; 889ace3abfcSBarry Smith PetscBool flag; 89016d9e3a6SLisandro Dalcin const char *symtlist[] = {"nonsymmetric","SPD","nonsymmetric,SPD"}; 89116d9e3a6SLisandro Dalcin 89216d9e3a6SLisandro Dalcin PetscFunctionBegin; 893e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"HYPRE ParaSails Options");CHKERRQ(ierr); 89416d9e3a6SLisandro Dalcin ierr = PetscOptionsInt("-pc_hypre_parasails_nlevels","Number of number of levels","None",jac->nlevels,&jac->nlevels,0);CHKERRQ(ierr); 89516d9e3a6SLisandro Dalcin ierr = PetscOptionsReal("-pc_hypre_parasails_thresh","Threshold","None",jac->threshhold,&jac->threshhold,&flag);CHKERRQ(ierr); 8962fa5cd67SKarl Rupp if (flag) PetscStackCallStandard(HYPRE_ParaSailsSetParams,(jac->hsolver,jac->threshhold,jac->nlevels)); 89716d9e3a6SLisandro Dalcin 89816d9e3a6SLisandro Dalcin ierr = PetscOptionsReal("-pc_hypre_parasails_filter","filter","None",jac->filter,&jac->filter,&flag);CHKERRQ(ierr); 8992fa5cd67SKarl Rupp if (flag) PetscStackCallStandard(HYPRE_ParaSailsSetFilter,(jac->hsolver,jac->filter)); 90016d9e3a6SLisandro Dalcin 90116d9e3a6SLisandro Dalcin ierr = PetscOptionsReal("-pc_hypre_parasails_loadbal","Load balance","None",jac->loadbal,&jac->loadbal,&flag);CHKERRQ(ierr); 9022fa5cd67SKarl Rupp if (flag) PetscStackCallStandard(HYPRE_ParaSailsSetLoadbal,(jac->hsolver,jac->loadbal)); 90316d9e3a6SLisandro Dalcin 904acfcf0e5SJed Brown ierr = PetscOptionsBool("-pc_hypre_parasails_logging","Print info to screen","None",(PetscBool)jac->logging,(PetscBool*)&jac->logging,&flag);CHKERRQ(ierr); 9052fa5cd67SKarl Rupp if (flag) PetscStackCallStandard(HYPRE_ParaSailsSetLogging,(jac->hsolver,jac->logging)); 90616d9e3a6SLisandro Dalcin 907acfcf0e5SJed Brown ierr = PetscOptionsBool("-pc_hypre_parasails_reuse","Reuse nonzero pattern in preconditioner","None",(PetscBool)jac->ruse,(PetscBool*)&jac->ruse,&flag);CHKERRQ(ierr); 9082fa5cd67SKarl Rupp if (flag) PetscStackCallStandard(HYPRE_ParaSailsSetReuse,(jac->hsolver,jac->ruse)); 90916d9e3a6SLisandro Dalcin 910a669f990SJed Brown ierr = PetscOptionsEList("-pc_hypre_parasails_sym","Symmetry of matrix and preconditioner","None",symtlist,ALEN(symtlist),symtlist[0],&indx,&flag);CHKERRQ(ierr); 91116d9e3a6SLisandro Dalcin if (flag) { 91216d9e3a6SLisandro Dalcin jac->symt = indx; 913fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetSym,(jac->hsolver,jac->symt)); 91416d9e3a6SLisandro Dalcin } 91516d9e3a6SLisandro Dalcin 91616d9e3a6SLisandro Dalcin ierr = PetscOptionsTail();CHKERRQ(ierr); 91716d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 91816d9e3a6SLisandro Dalcin } 91916d9e3a6SLisandro Dalcin 92016d9e3a6SLisandro Dalcin static PetscErrorCode PCView_HYPRE_ParaSails(PC pc,PetscViewer viewer) 92116d9e3a6SLisandro Dalcin { 92216d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 92316d9e3a6SLisandro Dalcin PetscErrorCode ierr; 924ace3abfcSBarry Smith PetscBool iascii; 92516d9e3a6SLisandro Dalcin const char *symt = 0;; 92616d9e3a6SLisandro Dalcin 92716d9e3a6SLisandro Dalcin PetscFunctionBegin; 928251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 92916d9e3a6SLisandro Dalcin if (iascii) { 93016d9e3a6SLisandro Dalcin ierr = PetscViewerASCIIPrintf(viewer," HYPRE ParaSails preconditioning\n");CHKERRQ(ierr); 931efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," nlevels %d\n",jac->nlevels);CHKERRQ(ierr); 932efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," threshold %g\n",(double)jac->threshhold);CHKERRQ(ierr); 933efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," filter %g\n",(double)jac->filter);CHKERRQ(ierr); 934efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," load balance %g\n",(double)jac->loadbal);CHKERRQ(ierr); 935efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," reuse nonzero structure %s\n",PetscBools[jac->ruse]);CHKERRQ(ierr); 936efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," print info to screen %s\n",PetscBools[jac->logging]);CHKERRQ(ierr); 9372fa5cd67SKarl Rupp if (!jac->symt) symt = "nonsymmetric matrix and preconditioner"; 9382fa5cd67SKarl Rupp else if (jac->symt == 1) symt = "SPD matrix and preconditioner"; 9392fa5cd67SKarl Rupp else if (jac->symt == 2) symt = "nonsymmetric matrix but SPD preconditioner"; 940ce94432eSBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Unknown HYPRE ParaSails symmetric option %d",jac->symt); 941efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %s\n",symt);CHKERRQ(ierr); 94216d9e3a6SLisandro Dalcin } 94316d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 94416d9e3a6SLisandro Dalcin } 9454cb006feSStefano Zampini /* --------------------------------------------------------------------------------------------*/ 9464416b707SBarry Smith static PetscErrorCode PCSetFromOptions_HYPRE_AMS(PetscOptionItems *PetscOptionsObject,PC pc) 9474cb006feSStefano Zampini { 9484cb006feSStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 9494cb006feSStefano Zampini PetscErrorCode ierr; 9504cb006feSStefano Zampini PetscInt n; 9514cb006feSStefano Zampini PetscBool flag,flag2,flag3,flag4; 9524cb006feSStefano Zampini 9534cb006feSStefano Zampini PetscFunctionBegin; 9549fa463a7SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"HYPRE AMS Options");CHKERRQ(ierr); 955863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ams_print_level","Debugging output level for AMS","None",jac->as_print,&jac->as_print,&flag);CHKERRQ(ierr); 956863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_AMSSetPrintLevel,(jac->hsolver,jac->as_print)); 957863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ams_max_iter","Maximum number of AMS multigrid iterations within PCApply","None",jac->as_max_iter,&jac->as_max_iter,&flag);CHKERRQ(ierr); 958863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_AMSSetMaxIter,(jac->hsolver,jac->as_max_iter)); 9594cb006feSStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ams_cycle_type","Cycle type for AMS multigrid","None",jac->ams_cycle_type,&jac->ams_cycle_type,&flag);CHKERRQ(ierr); 9604cb006feSStefano Zampini if (flag) PetscStackCallStandard(HYPRE_AMSSetCycleType,(jac->hsolver,jac->ams_cycle_type)); 961863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ams_tol","Error tolerance for AMS multigrid","None",jac->as_tol,&jac->as_tol,&flag);CHKERRQ(ierr); 962863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_AMSSetTol,(jac->hsolver,jac->as_tol)); 963863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ams_relax_type","Relaxation type for AMS smoother","None",jac->as_relax_type,&jac->as_relax_type,&flag);CHKERRQ(ierr); 964863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ams_relax_times","Number of relaxation steps for AMS smoother","None",jac->as_relax_times,&jac->as_relax_times,&flag2);CHKERRQ(ierr); 965863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ams_relax_weight","Relaxation weight for AMS smoother","None",jac->as_relax_weight,&jac->as_relax_weight,&flag3);CHKERRQ(ierr); 966863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ams_omega","SSOR coefficient for AMS smoother","None",jac->as_omega,&jac->as_omega,&flag4);CHKERRQ(ierr); 9674cb006feSStefano Zampini if (flag || flag2 || flag3 || flag4) { 968863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetSmoothingOptions,(jac->hsolver,jac->as_relax_type, 969863406b8SStefano Zampini jac->as_relax_times, 970863406b8SStefano Zampini jac->as_relax_weight, 971863406b8SStefano Zampini jac->as_omega)); 9724cb006feSStefano Zampini } 973863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ams_amg_alpha_theta","Threshold for strong coupling of vector Poisson AMG solver","None",jac->as_amg_alpha_theta,&jac->as_amg_alpha_theta,&flag);CHKERRQ(ierr); 9744cb006feSStefano Zampini n = 5; 975863406b8SStefano Zampini ierr = PetscOptionsIntArray("-pc_hypre_ams_amg_alpha_options","AMG options for vector Poisson","None",jac->as_amg_alpha_opts,&n,&flag2);CHKERRQ(ierr); 9764cb006feSStefano Zampini if (flag || flag2) { 977863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetAlphaAMGOptions,(jac->hsolver,jac->as_amg_alpha_opts[0], /* AMG coarsen type */ 978863406b8SStefano Zampini jac->as_amg_alpha_opts[1], /* AMG agg_levels */ 979863406b8SStefano Zampini jac->as_amg_alpha_opts[2], /* AMG relax_type */ 980863406b8SStefano Zampini jac->as_amg_alpha_theta, 981863406b8SStefano Zampini jac->as_amg_alpha_opts[3], /* AMG interp_type */ 982863406b8SStefano Zampini jac->as_amg_alpha_opts[4])); /* AMG Pmax */ 9834cb006feSStefano Zampini } 984863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ams_amg_beta_theta","Threshold for strong coupling of scalar Poisson AMG solver","None",jac->as_amg_beta_theta,&jac->as_amg_beta_theta,&flag);CHKERRQ(ierr); 9854cb006feSStefano Zampini n = 5; 986863406b8SStefano Zampini ierr = PetscOptionsIntArray("-pc_hypre_ams_amg_beta_options","AMG options for scalar Poisson solver","None",jac->as_amg_beta_opts,&n,&flag2);CHKERRQ(ierr); 9874cb006feSStefano Zampini if (flag || flag2) { 988863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetBetaAMGOptions,(jac->hsolver,jac->as_amg_beta_opts[0], /* AMG coarsen type */ 989863406b8SStefano Zampini jac->as_amg_beta_opts[1], /* AMG agg_levels */ 990863406b8SStefano Zampini jac->as_amg_beta_opts[2], /* AMG relax_type */ 991863406b8SStefano Zampini jac->as_amg_beta_theta, 992863406b8SStefano Zampini jac->as_amg_beta_opts[3], /* AMG interp_type */ 993863406b8SStefano Zampini jac->as_amg_beta_opts[4])); /* AMG Pmax */ 9944cb006feSStefano Zampini } 99523df4f25SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ams_projection_frequency","Frequency at which a projection onto the compatible subspace for problems with zero conductivity regions is performed","None",jac->ams_proj_freq,&jac->ams_proj_freq,&flag);CHKERRQ(ierr); 99623df4f25SStefano Zampini if (flag) { /* override HYPRE's default only if the options is used */ 99723df4f25SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetProjectionFrequency,(jac->hsolver,jac->ams_proj_freq)); 99823df4f25SStefano Zampini } 9994cb006feSStefano Zampini ierr = PetscOptionsTail();CHKERRQ(ierr); 10004cb006feSStefano Zampini PetscFunctionReturn(0); 10014cb006feSStefano Zampini } 10024cb006feSStefano Zampini 10034cb006feSStefano Zampini static PetscErrorCode PCView_HYPRE_AMS(PC pc,PetscViewer viewer) 10044cb006feSStefano Zampini { 10054cb006feSStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 10064cb006feSStefano Zampini PetscErrorCode ierr; 10074cb006feSStefano Zampini PetscBool iascii; 10084cb006feSStefano Zampini 10094cb006feSStefano Zampini PetscFunctionBegin; 10104cb006feSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 10114cb006feSStefano Zampini if (iascii) { 10124cb006feSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," HYPRE AMS preconditioning\n");CHKERRQ(ierr); 1013efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace iterations per application %d\n",jac->as_max_iter);CHKERRQ(ierr); 1014efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace cycle type %d\n",jac->ams_cycle_type);CHKERRQ(ierr); 1015efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace iteration tolerance %g\n",jac->as_tol);CHKERRQ(ierr); 1016efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," smoother type %d\n",jac->as_relax_type);CHKERRQ(ierr); 1017efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," number of smoothing steps %d\n",jac->as_relax_times);CHKERRQ(ierr); 1018efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," smoother weight %g\n",jac->as_relax_weight);CHKERRQ(ierr); 1019efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," smoother omega %g\n",jac->as_omega);CHKERRQ(ierr); 10204cb006feSStefano Zampini if (jac->alpha_Poisson) { 1021efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," vector Poisson solver (passed in by user)\n");CHKERRQ(ierr); 10224cb006feSStefano Zampini } else { 1023efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," vector Poisson solver (computed) \n");CHKERRQ(ierr); 10244cb006feSStefano Zampini } 1025efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG coarsening type %d\n",jac->as_amg_alpha_opts[0]);CHKERRQ(ierr); 1026efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG levels of aggressive coarsening %d\n",jac->as_amg_alpha_opts[1]);CHKERRQ(ierr); 1027efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG relaxation type %d\n",jac->as_amg_alpha_opts[2]);CHKERRQ(ierr); 1028efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG interpolation type %d\n",jac->as_amg_alpha_opts[3]);CHKERRQ(ierr); 1029efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG max nonzero elements in interpolation rows %d\n",jac->as_amg_alpha_opts[4]);CHKERRQ(ierr); 1030efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG strength threshold %g\n",jac->as_amg_alpha_theta);CHKERRQ(ierr); 10314cb006feSStefano Zampini if (!jac->ams_beta_is_zero) { 10324cb006feSStefano Zampini if (jac->beta_Poisson) { 1033efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," scalar Poisson solver (passed in by user)\n");CHKERRQ(ierr); 10344cb006feSStefano Zampini } else { 1035efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," scalar Poisson solver (computed) \n");CHKERRQ(ierr); 10364cb006feSStefano Zampini } 1037efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG coarsening type %d\n",jac->as_amg_beta_opts[0]);CHKERRQ(ierr); 1038efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG levels of aggressive coarsening %d\n",jac->as_amg_beta_opts[1]);CHKERRQ(ierr); 1039efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG relaxation type %d\n",jac->as_amg_beta_opts[2]);CHKERRQ(ierr); 1040efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG interpolation type %d\n",jac->as_amg_beta_opts[3]);CHKERRQ(ierr); 1041efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG max nonzero elements in interpolation rows %d\n",jac->as_amg_beta_opts[4]);CHKERRQ(ierr); 1042efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," boomerAMG strength threshold %g\n",jac->as_amg_beta_theta);CHKERRQ(ierr); 104323df4f25SStefano Zampini if (jac->ams_beta_is_zero_part) { 1044efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," compatible subspace projection frequency %d (-1 HYPRE uses default)\n",jac->ams_proj_freq);CHKERRQ(ierr); 104523df4f25SStefano Zampini } 104623df4f25SStefano Zampini } else { 1047efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," scalar Poisson solver not used (zero-conductivity everywhere) \n");CHKERRQ(ierr); 10484cb006feSStefano Zampini } 10494cb006feSStefano Zampini } 10504cb006feSStefano Zampini PetscFunctionReturn(0); 10514cb006feSStefano Zampini } 10524cb006feSStefano Zampini 10534416b707SBarry Smith static PetscErrorCode PCSetFromOptions_HYPRE_ADS(PetscOptionItems *PetscOptionsObject,PC pc) 1054863406b8SStefano Zampini { 1055863406b8SStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 1056863406b8SStefano Zampini PetscErrorCode ierr; 1057863406b8SStefano Zampini PetscInt n; 1058863406b8SStefano Zampini PetscBool flag,flag2,flag3,flag4; 1059863406b8SStefano Zampini 1060863406b8SStefano Zampini PetscFunctionBegin; 1061863406b8SStefano Zampini ierr = PetscOptionsHead(PetscOptionsObject,"HYPRE ADS Options");CHKERRQ(ierr); 1062863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ads_print_level","Debugging output level for ADS","None",jac->as_print,&jac->as_print,&flag);CHKERRQ(ierr); 1063863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_ADSSetPrintLevel,(jac->hsolver,jac->as_print)); 1064863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ads_max_iter","Maximum number of ADS multigrid iterations within PCApply","None",jac->as_max_iter,&jac->as_max_iter,&flag);CHKERRQ(ierr); 1065863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_ADSSetMaxIter,(jac->hsolver,jac->as_max_iter)); 1066863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ads_cycle_type","Cycle type for ADS multigrid","None",jac->ads_cycle_type,&jac->ads_cycle_type,&flag);CHKERRQ(ierr); 1067863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_ADSSetCycleType,(jac->hsolver,jac->ads_cycle_type)); 1068863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ads_tol","Error tolerance for ADS multigrid","None",jac->as_tol,&jac->as_tol,&flag);CHKERRQ(ierr); 1069863406b8SStefano Zampini if (flag) PetscStackCallStandard(HYPRE_ADSSetTol,(jac->hsolver,jac->as_tol)); 1070863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ads_relax_type","Relaxation type for ADS smoother","None",jac->as_relax_type,&jac->as_relax_type,&flag);CHKERRQ(ierr); 1071863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ads_relax_times","Number of relaxation steps for ADS smoother","None",jac->as_relax_times,&jac->as_relax_times,&flag2);CHKERRQ(ierr); 1072863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ads_relax_weight","Relaxation weight for ADS smoother","None",jac->as_relax_weight,&jac->as_relax_weight,&flag3);CHKERRQ(ierr); 1073863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ads_omega","SSOR coefficient for ADS smoother","None",jac->as_omega,&jac->as_omega,&flag4);CHKERRQ(ierr); 1074863406b8SStefano Zampini if (flag || flag2 || flag3 || flag4) { 1075863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetSmoothingOptions,(jac->hsolver,jac->as_relax_type, 1076863406b8SStefano Zampini jac->as_relax_times, 1077863406b8SStefano Zampini jac->as_relax_weight, 1078863406b8SStefano Zampini jac->as_omega)); 1079863406b8SStefano Zampini } 1080863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ads_ams_theta","Threshold for strong coupling of AMS solver inside ADS","None",jac->as_amg_alpha_theta,&jac->as_amg_alpha_theta,&flag);CHKERRQ(ierr); 1081863406b8SStefano Zampini n = 5; 1082863406b8SStefano Zampini ierr = PetscOptionsIntArray("-pc_hypre_ads_ams_options","AMG options for AMS solver inside ADS","None",jac->as_amg_alpha_opts,&n,&flag2);CHKERRQ(ierr); 1083863406b8SStefano Zampini ierr = PetscOptionsInt("-pc_hypre_ads_ams_cycle_type","Cycle type for AMS solver inside ADS","None",jac->ams_cycle_type,&jac->ams_cycle_type,&flag3);CHKERRQ(ierr); 1084863406b8SStefano Zampini if (flag || flag2 || flag3) { 1085863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetAMSOptions,(jac->hsolver,jac->ams_cycle_type, /* AMS cycle type */ 1086863406b8SStefano Zampini jac->as_amg_alpha_opts[0], /* AMG coarsen type */ 1087863406b8SStefano Zampini jac->as_amg_alpha_opts[1], /* AMG agg_levels */ 1088863406b8SStefano Zampini jac->as_amg_alpha_opts[2], /* AMG relax_type */ 1089863406b8SStefano Zampini jac->as_amg_alpha_theta, 1090863406b8SStefano Zampini jac->as_amg_alpha_opts[3], /* AMG interp_type */ 1091863406b8SStefano Zampini jac->as_amg_alpha_opts[4])); /* AMG Pmax */ 1092863406b8SStefano Zampini } 1093863406b8SStefano Zampini ierr = PetscOptionsReal("-pc_hypre_ads_amg_theta","Threshold for strong coupling of vector AMG solver inside ADS","None",jac->as_amg_beta_theta,&jac->as_amg_beta_theta,&flag);CHKERRQ(ierr); 1094863406b8SStefano Zampini n = 5; 1095863406b8SStefano Zampini ierr = PetscOptionsIntArray("-pc_hypre_ads_amg_options","AMG options for vector AMG solver inside ADS","None",jac->as_amg_beta_opts,&n,&flag2);CHKERRQ(ierr); 1096863406b8SStefano Zampini if (flag || flag2) { 1097863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetAMGOptions,(jac->hsolver,jac->as_amg_beta_opts[0], /* AMG coarsen type */ 1098863406b8SStefano Zampini jac->as_amg_beta_opts[1], /* AMG agg_levels */ 1099863406b8SStefano Zampini jac->as_amg_beta_opts[2], /* AMG relax_type */ 1100863406b8SStefano Zampini jac->as_amg_beta_theta, 1101863406b8SStefano Zampini jac->as_amg_beta_opts[3], /* AMG interp_type */ 1102863406b8SStefano Zampini jac->as_amg_beta_opts[4])); /* AMG Pmax */ 1103863406b8SStefano Zampini } 1104863406b8SStefano Zampini ierr = PetscOptionsTail();CHKERRQ(ierr); 1105863406b8SStefano Zampini PetscFunctionReturn(0); 1106863406b8SStefano Zampini } 1107863406b8SStefano Zampini 1108863406b8SStefano Zampini static PetscErrorCode PCView_HYPRE_ADS(PC pc,PetscViewer viewer) 1109863406b8SStefano Zampini { 1110863406b8SStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 1111863406b8SStefano Zampini PetscErrorCode ierr; 1112863406b8SStefano Zampini PetscBool iascii; 1113863406b8SStefano Zampini 1114863406b8SStefano Zampini PetscFunctionBegin; 1115863406b8SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 1116863406b8SStefano Zampini if (iascii) { 1117863406b8SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," HYPRE ADS preconditioning\n");CHKERRQ(ierr); 1118efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace iterations per application %d\n",jac->as_max_iter);CHKERRQ(ierr); 1119efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace cycle type %d\n",jac->ads_cycle_type);CHKERRQ(ierr); 1120efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace iteration tolerance %g\n",jac->as_tol);CHKERRQ(ierr); 1121efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," smoother type %d\n",jac->as_relax_type);CHKERRQ(ierr); 1122efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," number of smoothing steps %d\n",jac->as_relax_times);CHKERRQ(ierr); 1123efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," smoother weight %g\n",jac->as_relax_weight);CHKERRQ(ierr); 1124efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," smoother omega %g\n",jac->as_omega);CHKERRQ(ierr); 1125efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," AMS solver using boomerAMG\n");CHKERRQ(ierr); 1126efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," subspace cycle type %d\n",jac->ams_cycle_type);CHKERRQ(ierr); 1127efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," coarsening type %d\n",jac->as_amg_alpha_opts[0]);CHKERRQ(ierr); 1128efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," levels of aggressive coarsening %d\n",jac->as_amg_alpha_opts[1]);CHKERRQ(ierr); 1129efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," relaxation type %d\n",jac->as_amg_alpha_opts[2]);CHKERRQ(ierr); 1130efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," interpolation type %d\n",jac->as_amg_alpha_opts[3]);CHKERRQ(ierr); 1131efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," max nonzero elements in interpolation rows %d\n",jac->as_amg_alpha_opts[4]);CHKERRQ(ierr); 1132efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," strength threshold %g\n",jac->as_amg_alpha_theta);CHKERRQ(ierr); 1133efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," vector Poisson solver using boomerAMG\n");CHKERRQ(ierr); 1134efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," coarsening type %d\n",jac->as_amg_beta_opts[0]);CHKERRQ(ierr); 1135efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," levels of aggressive coarsening %d\n",jac->as_amg_beta_opts[1]);CHKERRQ(ierr); 1136efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," relaxation type %d\n",jac->as_amg_beta_opts[2]);CHKERRQ(ierr); 1137efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," interpolation type %d\n",jac->as_amg_beta_opts[3]);CHKERRQ(ierr); 1138efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," max nonzero elements in interpolation rows %d\n",jac->as_amg_beta_opts[4]);CHKERRQ(ierr); 1139efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," strength threshold %g\n",jac->as_amg_beta_theta);CHKERRQ(ierr); 1140863406b8SStefano Zampini } 1141863406b8SStefano Zampini PetscFunctionReturn(0); 1142863406b8SStefano Zampini } 1143863406b8SStefano Zampini 1144863406b8SStefano Zampini static PetscErrorCode PCHYPRESetDiscreteGradient_HYPRE(PC pc, Mat G) 11454cb006feSStefano Zampini { 11464cb006feSStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 11475ac14e1cSStefano Zampini PetscBool ishypre; 11484cb006feSStefano Zampini PetscErrorCode ierr; 11494cb006feSStefano Zampini 11504cb006feSStefano Zampini PetscFunctionBegin; 11515ac14e1cSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)G,MATHYPRE,&ishypre);CHKERRQ(ierr); 11525ac14e1cSStefano Zampini if (ishypre) { 11535ac14e1cSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 11545ac14e1cSStefano Zampini ierr = MatDestroy(&jac->G);CHKERRQ(ierr); 11555ac14e1cSStefano Zampini jac->G = G; 11565ac14e1cSStefano Zampini } else { 11576bf688a0SCe Qin ierr = MatDestroy(&jac->G);CHKERRQ(ierr); 11586bf688a0SCe Qin ierr = MatConvert(G,MATHYPRE,MAT_INITIAL_MATRIX,&jac->G);CHKERRQ(ierr); 11595ac14e1cSStefano Zampini } 11604cb006feSStefano Zampini PetscFunctionReturn(0); 11614cb006feSStefano Zampini } 11624cb006feSStefano Zampini 11634cb006feSStefano Zampini /*@ 11644cb006feSStefano Zampini PCHYPRESetDiscreteGradient - Set discrete gradient matrix 11654cb006feSStefano Zampini 11664cb006feSStefano Zampini Collective on PC 11674cb006feSStefano Zampini 11684cb006feSStefano Zampini Input Parameters: 11694cb006feSStefano Zampini + pc - the preconditioning context 11704cb006feSStefano Zampini - G - the discrete gradient 11714cb006feSStefano Zampini 11724cb006feSStefano Zampini Level: intermediate 11734cb006feSStefano Zampini 11744cb006feSStefano Zampini Notes: G should have as many rows as the number of edges and as many columns as the number of vertices in the mesh 1175863406b8SStefano Zampini Each row of G has 2 nonzeros, with column indexes being the global indexes of edge's endpoints: matrix entries are +1 and -1 depending on edge orientation 11764cb006feSStefano Zampini 11774cb006feSStefano Zampini .seealso: 11784cb006feSStefano Zampini @*/ 11794cb006feSStefano Zampini PetscErrorCode PCHYPRESetDiscreteGradient(PC pc, Mat G) 11804cb006feSStefano Zampini { 11814cb006feSStefano Zampini PetscErrorCode ierr; 11824cb006feSStefano Zampini 11834cb006feSStefano Zampini PetscFunctionBegin; 11844cb006feSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 11854cb006feSStefano Zampini PetscValidHeaderSpecific(G,MAT_CLASSID,2); 11864cb006feSStefano Zampini PetscCheckSameComm(pc,1,G,2); 11874cb006feSStefano Zampini ierr = PetscTryMethod(pc,"PCHYPRESetDiscreteGradient_C",(PC,Mat),(pc,G));CHKERRQ(ierr); 11884cb006feSStefano Zampini PetscFunctionReturn(0); 11894cb006feSStefano Zampini } 11904cb006feSStefano Zampini 1191863406b8SStefano Zampini static PetscErrorCode PCHYPRESetDiscreteCurl_HYPRE(PC pc, Mat C) 1192863406b8SStefano Zampini { 1193863406b8SStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 11945ac14e1cSStefano Zampini PetscBool ishypre; 1195863406b8SStefano Zampini PetscErrorCode ierr; 1196863406b8SStefano Zampini 1197863406b8SStefano Zampini PetscFunctionBegin; 11985ac14e1cSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)C,MATHYPRE,&ishypre);CHKERRQ(ierr); 11995ac14e1cSStefano Zampini if (ishypre) { 12005ac14e1cSStefano Zampini ierr = PetscObjectReference((PetscObject)C);CHKERRQ(ierr); 12015ac14e1cSStefano Zampini ierr = MatDestroy(&jac->C);CHKERRQ(ierr); 12025ac14e1cSStefano Zampini jac->C = C; 12035ac14e1cSStefano Zampini } else { 12046bf688a0SCe Qin ierr = MatDestroy(&jac->C);CHKERRQ(ierr); 12056bf688a0SCe Qin ierr = MatConvert(C,MATHYPRE,MAT_INITIAL_MATRIX,&jac->C);CHKERRQ(ierr); 12065ac14e1cSStefano Zampini } 1207863406b8SStefano Zampini PetscFunctionReturn(0); 1208863406b8SStefano Zampini } 1209863406b8SStefano Zampini 1210863406b8SStefano Zampini /*@ 1211863406b8SStefano Zampini PCHYPRESetDiscreteCurl - Set discrete curl matrix 1212863406b8SStefano Zampini 1213863406b8SStefano Zampini Collective on PC 1214863406b8SStefano Zampini 1215863406b8SStefano Zampini Input Parameters: 1216863406b8SStefano Zampini + pc - the preconditioning context 1217863406b8SStefano Zampini - C - the discrete curl 1218863406b8SStefano Zampini 1219863406b8SStefano Zampini Level: intermediate 1220863406b8SStefano Zampini 1221863406b8SStefano Zampini Notes: C should have as many rows as the number of faces and as many columns as the number of edges in the mesh 1222863406b8SStefano Zampini Each row of G has as many nonzeros as the number of edges of a face, with column indexes being the global indexes of the corresponding edge: matrix entries are +1 and -1 depending on edge orientation with respect to the face orientation 1223863406b8SStefano Zampini 1224863406b8SStefano Zampini .seealso: 1225863406b8SStefano Zampini @*/ 1226863406b8SStefano Zampini PetscErrorCode PCHYPRESetDiscreteCurl(PC pc, Mat C) 1227863406b8SStefano Zampini { 1228863406b8SStefano Zampini PetscErrorCode ierr; 1229863406b8SStefano Zampini 1230863406b8SStefano Zampini PetscFunctionBegin; 1231863406b8SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 1232863406b8SStefano Zampini PetscValidHeaderSpecific(C,MAT_CLASSID,2); 1233863406b8SStefano Zampini PetscCheckSameComm(pc,1,C,2); 1234863406b8SStefano Zampini ierr = PetscTryMethod(pc,"PCHYPRESetDiscreteCurl_C",(PC,Mat),(pc,C));CHKERRQ(ierr); 1235863406b8SStefano Zampini PetscFunctionReturn(0); 1236863406b8SStefano Zampini } 1237863406b8SStefano Zampini 12386bf688a0SCe Qin static PetscErrorCode PCHYPRESetInterpolations_HYPRE(PC pc, PetscInt dim, Mat RT_PiFull, Mat RT_Pi[], Mat ND_PiFull, Mat ND_Pi[]) 12396bf688a0SCe Qin { 12406bf688a0SCe Qin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 12416bf688a0SCe Qin PetscBool ishypre; 12426bf688a0SCe Qin PetscErrorCode ierr; 12436bf688a0SCe Qin PetscInt i; 12446bf688a0SCe Qin PetscFunctionBegin; 12456bf688a0SCe Qin 12466bf688a0SCe Qin ierr = MatDestroy(&jac->RT_PiFull);CHKERRQ(ierr); 12476bf688a0SCe Qin ierr = MatDestroy(&jac->ND_PiFull);CHKERRQ(ierr); 12486bf688a0SCe Qin for (i=0;i<3;++i) { 12496bf688a0SCe Qin ierr = MatDestroy(&jac->RT_Pi[i]);CHKERRQ(ierr); 12506bf688a0SCe Qin ierr = MatDestroy(&jac->ND_Pi[i]);CHKERRQ(ierr); 12516bf688a0SCe Qin } 12526bf688a0SCe Qin 12536bf688a0SCe Qin jac->dim = dim; 12546bf688a0SCe Qin if (RT_PiFull) { 12556bf688a0SCe Qin ierr = PetscObjectTypeCompare((PetscObject)RT_PiFull,MATHYPRE,&ishypre);CHKERRQ(ierr); 12566bf688a0SCe Qin if (ishypre) { 12576bf688a0SCe Qin ierr = PetscObjectReference((PetscObject)RT_PiFull);CHKERRQ(ierr); 12586bf688a0SCe Qin jac->RT_PiFull = RT_PiFull; 12596bf688a0SCe Qin } else { 12606bf688a0SCe Qin ierr = MatConvert(RT_PiFull,MATHYPRE,MAT_INITIAL_MATRIX,&jac->RT_PiFull);CHKERRQ(ierr); 12616bf688a0SCe Qin } 12626bf688a0SCe Qin } 12636bf688a0SCe Qin if (RT_Pi) { 12646bf688a0SCe Qin for (i=0;i<dim;++i) { 12656bf688a0SCe Qin if (RT_Pi[i]) { 12666bf688a0SCe Qin ierr = PetscObjectTypeCompare((PetscObject)RT_Pi[i],MATHYPRE,&ishypre);CHKERRQ(ierr); 12676bf688a0SCe Qin if (ishypre) { 12686bf688a0SCe Qin ierr = PetscObjectReference((PetscObject)RT_Pi[i]);CHKERRQ(ierr); 12696bf688a0SCe Qin jac->RT_Pi[i] = RT_Pi[i]; 12706bf688a0SCe Qin } else { 12716bf688a0SCe Qin ierr = MatConvert(RT_Pi[i],MATHYPRE,MAT_INITIAL_MATRIX,&jac->RT_Pi[i]);CHKERRQ(ierr); 12726bf688a0SCe Qin } 12736bf688a0SCe Qin } 12746bf688a0SCe Qin } 12756bf688a0SCe Qin } 12766bf688a0SCe Qin if (ND_PiFull) { 12776bf688a0SCe Qin ierr = PetscObjectTypeCompare((PetscObject)ND_PiFull,MATHYPRE,&ishypre);CHKERRQ(ierr); 12786bf688a0SCe Qin if (ishypre) { 12796bf688a0SCe Qin ierr = PetscObjectReference((PetscObject)ND_PiFull);CHKERRQ(ierr); 12806bf688a0SCe Qin jac->ND_PiFull = ND_PiFull; 12816bf688a0SCe Qin } else { 12826bf688a0SCe Qin ierr = MatConvert(ND_PiFull,MATHYPRE,MAT_INITIAL_MATRIX,&jac->ND_PiFull);CHKERRQ(ierr); 12836bf688a0SCe Qin } 12846bf688a0SCe Qin } 12856bf688a0SCe Qin if (ND_Pi) { 12866bf688a0SCe Qin for (i=0;i<dim;++i) { 12876bf688a0SCe Qin if (ND_Pi[i]) { 12886bf688a0SCe Qin ierr = PetscObjectTypeCompare((PetscObject)ND_Pi[i],MATHYPRE,&ishypre);CHKERRQ(ierr); 12896bf688a0SCe Qin if (ishypre) { 12906bf688a0SCe Qin ierr = PetscObjectReference((PetscObject)ND_Pi[i]);CHKERRQ(ierr); 12916bf688a0SCe Qin jac->ND_Pi[i] = ND_Pi[i]; 12926bf688a0SCe Qin } else { 12936bf688a0SCe Qin ierr = MatConvert(ND_Pi[i],MATHYPRE,MAT_INITIAL_MATRIX,&jac->ND_Pi[i]);CHKERRQ(ierr); 12946bf688a0SCe Qin } 12956bf688a0SCe Qin } 12966bf688a0SCe Qin } 12976bf688a0SCe Qin } 12986bf688a0SCe Qin 12996bf688a0SCe Qin PetscFunctionReturn(0); 13006bf688a0SCe Qin } 13016bf688a0SCe Qin 13026bf688a0SCe Qin /*@ 13036bf688a0SCe Qin PCHYPRESetInterpolations - Set interpolation matrices for AMS/ADS preconditioner 13046bf688a0SCe Qin 13056bf688a0SCe Qin Collective on PC 13066bf688a0SCe Qin 13076bf688a0SCe Qin Input Parameters: 13086bf688a0SCe Qin + pc - the preconditioning context 13096bf688a0SCe Qin - dim - the dimension of the problem, only used in AMS 13106bf688a0SCe Qin - RT_PiFull - Raviart-Thomas interpolation matrix 13116bf688a0SCe Qin - RT_Pi - x/y/z component of Raviart-Thomas interpolation matrix 13126bf688a0SCe Qin - ND_PiFull - Nedelec interpolation matrix 13136bf688a0SCe Qin - ND_Pi - x/y/z component of Nedelec interpolation matrix 13146bf688a0SCe Qin 13156bf688a0SCe Qin Notes: For AMS, only Nedelec interpolation matrices are needed, the Raviart-Thomas interpolation matrices can be set to NULL. 13166bf688a0SCe Qin For ADS, both type of interpolation matrices are needed. 13176bf688a0SCe Qin Level: intermediate 13186bf688a0SCe Qin 13196bf688a0SCe Qin .seealso: 13206bf688a0SCe Qin @*/ 13216bf688a0SCe Qin PetscErrorCode PCHYPRESetInterpolations(PC pc, PetscInt dim, Mat RT_PiFull, Mat RT_Pi[], Mat ND_PiFull, Mat ND_Pi[]) 13226bf688a0SCe Qin { 13236bf688a0SCe Qin PetscErrorCode ierr; 13246bf688a0SCe Qin PetscInt i; 13256bf688a0SCe Qin 13266bf688a0SCe Qin PetscFunctionBegin; 13276bf688a0SCe Qin PetscValidHeaderSpecific(pc,PC_CLASSID,1); 13286bf688a0SCe Qin if (RT_PiFull) { 13296bf688a0SCe Qin PetscValidHeaderSpecific(RT_PiFull,MAT_CLASSID,3); 13306bf688a0SCe Qin PetscCheckSameComm(pc,1,RT_PiFull,3); 13316bf688a0SCe Qin } 13326bf688a0SCe Qin if (RT_Pi) { 13336bf688a0SCe Qin PetscValidPointer(RT_Pi,4); 13346bf688a0SCe Qin for (i=0;i<dim;++i) { 13356bf688a0SCe Qin if (RT_Pi[i]) { 13366bf688a0SCe Qin PetscValidHeaderSpecific(RT_Pi[i],MAT_CLASSID,4); 13376bf688a0SCe Qin PetscCheckSameComm(pc,1,RT_Pi[i],4); 13386bf688a0SCe Qin } 13396bf688a0SCe Qin } 13406bf688a0SCe Qin } 13416bf688a0SCe Qin if (ND_PiFull) { 13426bf688a0SCe Qin PetscValidHeaderSpecific(ND_PiFull,MAT_CLASSID,5); 13436bf688a0SCe Qin PetscCheckSameComm(pc,1,ND_PiFull,5); 13446bf688a0SCe Qin } 13456bf688a0SCe Qin if (ND_Pi) { 13466bf688a0SCe Qin PetscValidPointer(ND_Pi,6); 13476bf688a0SCe Qin for (i=0;i<dim;++i) { 13486bf688a0SCe Qin if (ND_Pi[i]) { 13496bf688a0SCe Qin PetscValidHeaderSpecific(ND_Pi[i],MAT_CLASSID,6); 13506bf688a0SCe Qin PetscCheckSameComm(pc,1,ND_Pi[i],6); 13516bf688a0SCe Qin } 13526bf688a0SCe Qin } 13536bf688a0SCe Qin } 13546bf688a0SCe Qin ierr = PetscTryMethod(pc,"PCHYPRESetInterpolations_C",(PC,PetscInt,Mat,Mat[],Mat,Mat[]),(pc,dim,RT_PiFull,RT_Pi,ND_PiFull,ND_Pi));CHKERRQ(ierr); 13556bf688a0SCe Qin PetscFunctionReturn(0); 13566bf688a0SCe Qin } 13576bf688a0SCe Qin 13585ac14e1cSStefano Zampini static PetscErrorCode PCHYPRESetPoissonMatrix_HYPRE(PC pc, Mat A, PetscBool isalpha) 13594cb006feSStefano Zampini { 13604cb006feSStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 13615ac14e1cSStefano Zampini PetscBool ishypre; 13624cb006feSStefano Zampini PetscErrorCode ierr; 13634cb006feSStefano Zampini 13644cb006feSStefano Zampini PetscFunctionBegin; 13655ac14e1cSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)A,MATHYPRE,&ishypre);CHKERRQ(ierr); 13665ac14e1cSStefano Zampini if (ishypre) { 13675ac14e1cSStefano Zampini if (isalpha) { 13685ac14e1cSStefano Zampini ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 13695ac14e1cSStefano Zampini ierr = MatDestroy(&jac->alpha_Poisson);CHKERRQ(ierr); 13705ac14e1cSStefano Zampini jac->alpha_Poisson = A; 13715ac14e1cSStefano Zampini } else { 13725ac14e1cSStefano Zampini if (A) { 13735ac14e1cSStefano Zampini ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 13745ac14e1cSStefano Zampini } else { 13755ac14e1cSStefano Zampini jac->ams_beta_is_zero = PETSC_TRUE; 13765ac14e1cSStefano Zampini } 13775ac14e1cSStefano Zampini ierr = MatDestroy(&jac->beta_Poisson);CHKERRQ(ierr); 13785ac14e1cSStefano Zampini jac->beta_Poisson = A; 13795ac14e1cSStefano Zampini } 13805ac14e1cSStefano Zampini } else { 13815ac14e1cSStefano Zampini if (isalpha) { 13826bf688a0SCe Qin ierr = MatDestroy(&jac->alpha_Poisson);CHKERRQ(ierr); 13836bf688a0SCe Qin ierr = MatConvert(A,MATHYPRE,MAT_INITIAL_MATRIX,&jac->alpha_Poisson);CHKERRQ(ierr); 13845ac14e1cSStefano Zampini } else { 13855ac14e1cSStefano Zampini if (A) { 13866bf688a0SCe Qin ierr = MatDestroy(&jac->beta_Poisson);CHKERRQ(ierr); 13876bf688a0SCe Qin ierr = MatConvert(A,MATHYPRE,MAT_INITIAL_MATRIX,&jac->beta_Poisson);CHKERRQ(ierr); 13885ac14e1cSStefano Zampini } else { 13895ac14e1cSStefano Zampini ierr = MatDestroy(&jac->beta_Poisson);CHKERRQ(ierr); 13905ac14e1cSStefano Zampini jac->ams_beta_is_zero = PETSC_TRUE; 13915ac14e1cSStefano Zampini } 13925ac14e1cSStefano Zampini } 13935ac14e1cSStefano Zampini } 13944cb006feSStefano Zampini PetscFunctionReturn(0); 13954cb006feSStefano Zampini } 13964cb006feSStefano Zampini 13974cb006feSStefano Zampini /*@ 13984cb006feSStefano Zampini PCHYPRESetAlphaPoissonMatrix - Set vector Poisson matrix 13994cb006feSStefano Zampini 14004cb006feSStefano Zampini Collective on PC 14014cb006feSStefano Zampini 14024cb006feSStefano Zampini Input Parameters: 14034cb006feSStefano Zampini + pc - the preconditioning context 14044cb006feSStefano Zampini - A - the matrix 14054cb006feSStefano Zampini 14064cb006feSStefano Zampini Level: intermediate 14074cb006feSStefano Zampini 14084cb006feSStefano Zampini Notes: A should be obtained by discretizing the vector valued Poisson problem with linear finite elements 14094cb006feSStefano Zampini 14104cb006feSStefano Zampini .seealso: 14114cb006feSStefano Zampini @*/ 14124cb006feSStefano Zampini PetscErrorCode PCHYPRESetAlphaPoissonMatrix(PC pc, Mat A) 14134cb006feSStefano Zampini { 14144cb006feSStefano Zampini PetscErrorCode ierr; 14154cb006feSStefano Zampini 14164cb006feSStefano Zampini PetscFunctionBegin; 14174cb006feSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 14184cb006feSStefano Zampini PetscValidHeaderSpecific(A,MAT_CLASSID,2); 14194cb006feSStefano Zampini PetscCheckSameComm(pc,1,A,2); 14205ac14e1cSStefano Zampini ierr = PetscTryMethod(pc,"PCHYPRESetPoissonMatrix_C",(PC,Mat,PetscBool),(pc,A,PETSC_TRUE));CHKERRQ(ierr); 14214cb006feSStefano Zampini PetscFunctionReturn(0); 14224cb006feSStefano Zampini } 14234cb006feSStefano Zampini 14244cb006feSStefano Zampini /*@ 14254cb006feSStefano Zampini PCHYPRESetBetaPoissonMatrix - Set Poisson matrix 14264cb006feSStefano Zampini 14274cb006feSStefano Zampini Collective on PC 14284cb006feSStefano Zampini 14294cb006feSStefano Zampini Input Parameters: 14304cb006feSStefano Zampini + pc - the preconditioning context 14314cb006feSStefano Zampini - A - the matrix 14324cb006feSStefano Zampini 14334cb006feSStefano Zampini Level: intermediate 14344cb006feSStefano Zampini 14354cb006feSStefano Zampini Notes: A should be obtained by discretizing the Poisson problem with linear finite elements. 14364cb006feSStefano Zampini Following HYPRE convention, the scalar Poisson solver of AMS can be turned off by passing NULL. 14374cb006feSStefano Zampini 14384cb006feSStefano Zampini .seealso: 14394cb006feSStefano Zampini @*/ 14404cb006feSStefano Zampini PetscErrorCode PCHYPRESetBetaPoissonMatrix(PC pc, Mat A) 14414cb006feSStefano Zampini { 14424cb006feSStefano Zampini PetscErrorCode ierr; 14434cb006feSStefano Zampini 14444cb006feSStefano Zampini PetscFunctionBegin; 14454cb006feSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 14464cb006feSStefano Zampini if (A) { 14474cb006feSStefano Zampini PetscValidHeaderSpecific(A,MAT_CLASSID,2); 14484cb006feSStefano Zampini PetscCheckSameComm(pc,1,A,2); 14494cb006feSStefano Zampini } 14505ac14e1cSStefano Zampini ierr = PetscTryMethod(pc,"PCHYPRESetPoissonMatrix_C",(PC,Mat,PetscBool),(pc,A,PETSC_FALSE));CHKERRQ(ierr); 14514cb006feSStefano Zampini PetscFunctionReturn(0); 14524cb006feSStefano Zampini } 14534cb006feSStefano Zampini 14545ac14e1cSStefano Zampini static PetscErrorCode PCHYPRESetEdgeConstantVectors_HYPRE(PC pc,Vec ozz, Vec zoz, Vec zzo) 14554cb006feSStefano Zampini { 14564cb006feSStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 14574cb006feSStefano Zampini PetscErrorCode ierr; 14584cb006feSStefano Zampini 14594cb006feSStefano Zampini PetscFunctionBegin; 14604cb006feSStefano Zampini /* throw away any vector if already set */ 14614cb006feSStefano Zampini if (jac->constants[0]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->constants[0])); 14624cb006feSStefano Zampini if (jac->constants[1]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->constants[1])); 14634cb006feSStefano Zampini if (jac->constants[2]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->constants[2])); 14644cb006feSStefano Zampini jac->constants[0] = NULL; 14654cb006feSStefano Zampini jac->constants[1] = NULL; 14664cb006feSStefano Zampini jac->constants[2] = NULL; 14674cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCreate(ozz,&jac->constants[0]);CHKERRQ(ierr); 14684cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCopy(ozz,jac->constants[0]);CHKERRQ(ierr); 14694cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCreate(zoz,&jac->constants[1]);CHKERRQ(ierr); 14704cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCopy(zoz,jac->constants[1]);CHKERRQ(ierr); 14715ac14e1cSStefano Zampini jac->dim = 2; 14724cb006feSStefano Zampini if (zzo) { 14734cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCreate(zzo,&jac->constants[2]);CHKERRQ(ierr); 14744cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCopy(zzo,jac->constants[2]);CHKERRQ(ierr); 14755ac14e1cSStefano Zampini jac->dim++; 14764cb006feSStefano Zampini } 14774cb006feSStefano Zampini PetscFunctionReturn(0); 14784cb006feSStefano Zampini } 14794cb006feSStefano Zampini 14804cb006feSStefano Zampini /*@ 14814cb006feSStefano Zampini PCHYPRESetEdgeConstantVectors - Set the representation of the constant vector fields in edge element basis 14824cb006feSStefano Zampini 14834cb006feSStefano Zampini Collective on PC 14844cb006feSStefano Zampini 14854cb006feSStefano Zampini Input Parameters: 14864cb006feSStefano Zampini + pc - the preconditioning context 14874cb006feSStefano Zampini - ozz - vector representing (1,0,0) (or (1,0) in 2D) 14884cb006feSStefano Zampini - zoz - vector representing (0,1,0) (or (0,1) in 2D) 14894cb006feSStefano Zampini - zzo - vector representing (0,0,1) (use NULL in 2D) 14904cb006feSStefano Zampini 14914cb006feSStefano Zampini Level: intermediate 14924cb006feSStefano Zampini 14934cb006feSStefano Zampini Notes: 14944cb006feSStefano Zampini 14954cb006feSStefano Zampini .seealso: 14964cb006feSStefano Zampini @*/ 14974cb006feSStefano Zampini PetscErrorCode PCHYPRESetEdgeConstantVectors(PC pc, Vec ozz, Vec zoz, Vec zzo) 14984cb006feSStefano Zampini { 14994cb006feSStefano Zampini PetscErrorCode ierr; 15004cb006feSStefano Zampini 15014cb006feSStefano Zampini PetscFunctionBegin; 15024cb006feSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 15034cb006feSStefano Zampini PetscValidHeaderSpecific(ozz,VEC_CLASSID,2); 15044cb006feSStefano Zampini PetscValidHeaderSpecific(zoz,VEC_CLASSID,3); 15054cb006feSStefano Zampini if (zzo) PetscValidHeaderSpecific(zzo,VEC_CLASSID,4); 15064cb006feSStefano Zampini PetscCheckSameComm(pc,1,ozz,2); 15074cb006feSStefano Zampini PetscCheckSameComm(pc,1,zoz,3); 15084cb006feSStefano Zampini if (zzo) PetscCheckSameComm(pc,1,zzo,4); 15094cb006feSStefano Zampini ierr = PetscTryMethod(pc,"PCHYPRESetEdgeConstantVectors_C",(PC,Vec,Vec,Vec),(pc,ozz,zoz,zzo));CHKERRQ(ierr); 15104cb006feSStefano Zampini PetscFunctionReturn(0); 15114cb006feSStefano Zampini } 15124cb006feSStefano Zampini 1513863406b8SStefano Zampini static PetscErrorCode PCSetCoordinates_HYPRE(PC pc, PetscInt dim, PetscInt nloc, PetscReal *coords) 15144cb006feSStefano Zampini { 15154cb006feSStefano Zampini PC_HYPRE *jac = (PC_HYPRE*)pc->data; 15164cb006feSStefano Zampini Vec tv; 15174cb006feSStefano Zampini PetscInt i; 15184cb006feSStefano Zampini PetscErrorCode ierr; 15194cb006feSStefano Zampini 15204cb006feSStefano Zampini PetscFunctionBegin; 15214cb006feSStefano Zampini /* throw away any coordinate vector if already set */ 15224cb006feSStefano Zampini if (jac->coords[0]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->coords[0])); 15234cb006feSStefano Zampini if (jac->coords[1]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->coords[1])); 15244cb006feSStefano Zampini if (jac->coords[2]) PetscStackCallStandard(HYPRE_IJVectorDestroy,(jac->coords[2])); 15255ac14e1cSStefano Zampini jac->dim = dim; 15265ac14e1cSStefano Zampini 15274cb006feSStefano Zampini /* compute IJ vector for coordinates */ 15284cb006feSStefano Zampini ierr = VecCreate(PetscObjectComm((PetscObject)pc),&tv);CHKERRQ(ierr); 15294cb006feSStefano Zampini ierr = VecSetType(tv,VECSTANDARD);CHKERRQ(ierr); 15304cb006feSStefano Zampini ierr = VecSetSizes(tv,nloc,PETSC_DECIDE);CHKERRQ(ierr); 15314cb006feSStefano Zampini for (i=0;i<dim;i++) { 15324cb006feSStefano Zampini PetscScalar *array; 15334cb006feSStefano Zampini PetscInt j; 15344cb006feSStefano Zampini 15354cb006feSStefano Zampini ierr = VecHYPRE_IJVectorCreate(tv,&jac->coords[i]);CHKERRQ(ierr); 15364cb006feSStefano Zampini ierr = VecGetArray(tv,&array);CHKERRQ(ierr); 15374cb006feSStefano Zampini for (j=0;j<nloc;j++) { 15384cb006feSStefano Zampini array[j] = coords[j*dim+i]; 15394cb006feSStefano Zampini } 15404cb006feSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorSetValues,(jac->coords[i],nloc,NULL,array)); 15414cb006feSStefano Zampini PetscStackCallStandard(HYPRE_IJVectorAssemble,(jac->coords[i])); 15424cb006feSStefano Zampini ierr = VecRestoreArray(tv,&array);CHKERRQ(ierr); 15434cb006feSStefano Zampini } 15444cb006feSStefano Zampini ierr = VecDestroy(&tv);CHKERRQ(ierr); 15454cb006feSStefano Zampini PetscFunctionReturn(0); 15464cb006feSStefano Zampini } 15474cb006feSStefano Zampini 154816d9e3a6SLisandro Dalcin /* ---------------------------------------------------------------------------------*/ 154916d9e3a6SLisandro Dalcin 1550f7a08781SBarry Smith static PetscErrorCode PCHYPREGetType_HYPRE(PC pc,const char *name[]) 155116d9e3a6SLisandro Dalcin { 155216d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 155316d9e3a6SLisandro Dalcin 155416d9e3a6SLisandro Dalcin PetscFunctionBegin; 155516d9e3a6SLisandro Dalcin *name = jac->hypre_type; 155616d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 155716d9e3a6SLisandro Dalcin } 155816d9e3a6SLisandro Dalcin 1559f7a08781SBarry Smith static PetscErrorCode PCHYPRESetType_HYPRE(PC pc,const char name[]) 156016d9e3a6SLisandro Dalcin { 156116d9e3a6SLisandro Dalcin PC_HYPRE *jac = (PC_HYPRE*)pc->data; 156216d9e3a6SLisandro Dalcin PetscErrorCode ierr; 1563ace3abfcSBarry Smith PetscBool flag; 156416d9e3a6SLisandro Dalcin 156516d9e3a6SLisandro Dalcin PetscFunctionBegin; 156616d9e3a6SLisandro Dalcin if (jac->hypre_type) { 156716d9e3a6SLisandro Dalcin ierr = PetscStrcmp(jac->hypre_type,name,&flag);CHKERRQ(ierr); 1568ce94432eSBarry Smith if (!flag) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ORDER,"Cannot reset the HYPRE preconditioner type once it has been set"); 156916d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 157016d9e3a6SLisandro Dalcin } else { 157116d9e3a6SLisandro Dalcin ierr = PetscStrallocpy(name, &jac->hypre_type);CHKERRQ(ierr); 157216d9e3a6SLisandro Dalcin } 157316d9e3a6SLisandro Dalcin 157416d9e3a6SLisandro Dalcin jac->maxiter = PETSC_DEFAULT; 157516d9e3a6SLisandro Dalcin jac->tol = PETSC_DEFAULT; 157616d9e3a6SLisandro Dalcin jac->printstatistics = PetscLogPrintInfo; 157716d9e3a6SLisandro Dalcin 157816d9e3a6SLisandro Dalcin ierr = PetscStrcmp("pilut",jac->hypre_type,&flag);CHKERRQ(ierr); 157916d9e3a6SLisandro Dalcin if (flag) { 1580*572a0576SBarry Smith ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)pc),&(jac->comm_hypre));CHKERRQ(ierr); 1581fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParCSRPilutCreate,(jac->comm_hypre,&jac->hsolver)); 158216d9e3a6SLisandro Dalcin pc->ops->setfromoptions = PCSetFromOptions_HYPRE_Pilut; 158316d9e3a6SLisandro Dalcin pc->ops->view = PCView_HYPRE_Pilut; 158416d9e3a6SLisandro Dalcin jac->destroy = HYPRE_ParCSRPilutDestroy; 158516d9e3a6SLisandro Dalcin jac->setup = HYPRE_ParCSRPilutSetup; 158616d9e3a6SLisandro Dalcin jac->solve = HYPRE_ParCSRPilutSolve; 158716d9e3a6SLisandro Dalcin jac->factorrowsize = PETSC_DEFAULT; 158816d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 158916d9e3a6SLisandro Dalcin } 159016d9e3a6SLisandro Dalcin ierr = PetscStrcmp("parasails",jac->hypre_type,&flag);CHKERRQ(ierr); 159116d9e3a6SLisandro Dalcin if (flag) { 1592*572a0576SBarry Smith ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)pc),&(jac->comm_hypre));CHKERRQ(ierr); 1593fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsCreate,(jac->comm_hypre,&jac->hsolver)); 159416d9e3a6SLisandro Dalcin pc->ops->setfromoptions = PCSetFromOptions_HYPRE_ParaSails; 159516d9e3a6SLisandro Dalcin pc->ops->view = PCView_HYPRE_ParaSails; 159616d9e3a6SLisandro Dalcin jac->destroy = HYPRE_ParaSailsDestroy; 159716d9e3a6SLisandro Dalcin jac->setup = HYPRE_ParaSailsSetup; 159816d9e3a6SLisandro Dalcin jac->solve = HYPRE_ParaSailsSolve; 159916d9e3a6SLisandro Dalcin /* initialize */ 160016d9e3a6SLisandro Dalcin jac->nlevels = 1; 160116d9e3a6SLisandro Dalcin jac->threshhold = .1; 160216d9e3a6SLisandro Dalcin jac->filter = .1; 160316d9e3a6SLisandro Dalcin jac->loadbal = 0; 16042fa5cd67SKarl Rupp if (PetscLogPrintInfo) jac->logging = (int) PETSC_TRUE; 16052fa5cd67SKarl Rupp else jac->logging = (int) PETSC_FALSE; 16062fa5cd67SKarl Rupp 160716d9e3a6SLisandro Dalcin jac->ruse = (int) PETSC_FALSE; 160816d9e3a6SLisandro Dalcin jac->symt = 0; 1609fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetParams,(jac->hsolver,jac->threshhold,jac->nlevels)); 1610fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetFilter,(jac->hsolver,jac->filter)); 1611fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetLoadbal,(jac->hsolver,jac->loadbal)); 1612fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetLogging,(jac->hsolver,jac->logging)); 1613fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetReuse,(jac->hsolver,jac->ruse)); 1614fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_ParaSailsSetSym,(jac->hsolver,jac->symt)); 161516d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 161616d9e3a6SLisandro Dalcin } 161716d9e3a6SLisandro Dalcin ierr = PetscStrcmp("boomeramg",jac->hypre_type,&flag);CHKERRQ(ierr); 161816d9e3a6SLisandro Dalcin if (flag) { 161916d9e3a6SLisandro Dalcin ierr = HYPRE_BoomerAMGCreate(&jac->hsolver); 162016d9e3a6SLisandro Dalcin pc->ops->setfromoptions = PCSetFromOptions_HYPRE_BoomerAMG; 162116d9e3a6SLisandro Dalcin pc->ops->view = PCView_HYPRE_BoomerAMG; 162216d9e3a6SLisandro Dalcin pc->ops->applytranspose = PCApplyTranspose_HYPRE_BoomerAMG; 162316d9e3a6SLisandro Dalcin pc->ops->applyrichardson = PCApplyRichardson_HYPRE_BoomerAMG; 162416d9e3a6SLisandro Dalcin jac->destroy = HYPRE_BoomerAMGDestroy; 162516d9e3a6SLisandro Dalcin jac->setup = HYPRE_BoomerAMGSetup; 162616d9e3a6SLisandro Dalcin jac->solve = HYPRE_BoomerAMGSolve; 162716d9e3a6SLisandro Dalcin jac->applyrichardson = PETSC_FALSE; 162816d9e3a6SLisandro Dalcin /* these defaults match the hypre defaults */ 162916d9e3a6SLisandro Dalcin jac->cycletype = 1; 163016d9e3a6SLisandro Dalcin jac->maxlevels = 25; 163116d9e3a6SLisandro Dalcin jac->maxiter = 1; 16328f87f92bSBarry Smith jac->tol = 0.0; /* tolerance of zero indicates use as preconditioner (suppresses convergence errors) */ 163316d9e3a6SLisandro Dalcin jac->truncfactor = 0.0; 163416d9e3a6SLisandro Dalcin jac->strongthreshold = .25; 163516d9e3a6SLisandro Dalcin jac->maxrowsum = .9; 163616d9e3a6SLisandro Dalcin jac->coarsentype = 6; 163716d9e3a6SLisandro Dalcin jac->measuretype = 0; 16380f1074feSSatish Balay jac->gridsweeps[0] = jac->gridsweeps[1] = jac->gridsweeps[2] = 1; 16396a251517SEike Mueller jac->smoothtype = -1; /* Not set by default */ 1640b9eb5777SEike Mueller jac->smoothnumlevels = 25; 16411810e44eSEike Mueller jac->eu_level = 0; 16421810e44eSEike Mueller jac->eu_droptolerance = 0; 16431810e44eSEike Mueller jac->eu_bj = 0; 16448f87f92bSBarry Smith jac->relaxtype[0] = jac->relaxtype[1] = 6; /* Defaults to SYMMETRIC since in PETSc we are using a a PC - most likely with CG */ 16450f1074feSSatish Balay jac->relaxtype[2] = 9; /*G.E. */ 164616d9e3a6SLisandro Dalcin jac->relaxweight = 1.0; 164716d9e3a6SLisandro Dalcin jac->outerrelaxweight = 1.0; 164816d9e3a6SLisandro Dalcin jac->relaxorder = 1; 16490f1074feSSatish Balay jac->interptype = 0; 16500f1074feSSatish Balay jac->agg_nl = 0; 16510f1074feSSatish Balay jac->pmax = 0; 16520f1074feSSatish Balay jac->truncfactor = 0.0; 16530f1074feSSatish Balay jac->agg_num_paths = 1; 16548f87f92bSBarry Smith 16558f87f92bSBarry Smith jac->nodal_coarsen = 0; 16568f87f92bSBarry Smith jac->nodal_relax = PETSC_FALSE; 16578f87f92bSBarry Smith jac->nodal_relax_levels = 1; 1658fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCycleType,(jac->hsolver,jac->cycletype)); 1659fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxLevels,(jac->hsolver,jac->maxlevels)); 1660fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxIter,(jac->hsolver,jac->maxiter)); 1661fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetTol,(jac->hsolver,jac->tol)); 1662fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetTruncFactor,(jac->hsolver,jac->truncfactor)); 1663fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetStrongThreshold,(jac->hsolver,jac->strongthreshold)); 1664fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMaxRowSum,(jac->hsolver,jac->maxrowsum)); 1665fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetCoarsenType,(jac->hsolver,jac->coarsentype)); 1666fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetMeasureType,(jac->hsolver,jac->measuretype)); 1667fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetRelaxOrder,(jac->hsolver, jac->relaxorder)); 1668fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetInterpType,(jac->hsolver,jac->interptype)); 1669fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetAggNumLevels,(jac->hsolver,jac->agg_nl)); 1670fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetPMaxElmts,(jac->hsolver,jac->pmax)); 1671fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetNumPaths,(jac->hsolver,jac->agg_num_paths)); 1672fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetRelaxType,(jac->hsolver, jac->relaxtype[0])); /*defaults coarse to 9*/ 1673fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_BoomerAMGSetNumSweeps,(jac->hsolver, jac->gridsweeps[0])); /*defaults coarse to 1 */ 167416d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 167516d9e3a6SLisandro Dalcin } 16764cb006feSStefano Zampini ierr = PetscStrcmp("ams",jac->hypre_type,&flag);CHKERRQ(ierr); 16774cb006feSStefano Zampini if (flag) { 16784cb006feSStefano Zampini ierr = HYPRE_AMSCreate(&jac->hsolver); 16794cb006feSStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_HYPRE_AMS; 16804cb006feSStefano Zampini pc->ops->view = PCView_HYPRE_AMS; 16814cb006feSStefano Zampini jac->destroy = HYPRE_AMSDestroy; 16824cb006feSStefano Zampini jac->setup = HYPRE_AMSSetup; 16834cb006feSStefano Zampini jac->solve = HYPRE_AMSSolve; 16844cb006feSStefano Zampini jac->coords[0] = NULL; 16854cb006feSStefano Zampini jac->coords[1] = NULL; 16864cb006feSStefano Zampini jac->coords[2] = NULL; 16874cb006feSStefano Zampini /* solver parameters: these are borrowed from mfem package, and they are not the default values from HYPRE AMS */ 1688863406b8SStefano Zampini jac->as_print = 0; 1689863406b8SStefano Zampini jac->as_max_iter = 1; /* used as a preconditioner */ 1690863406b8SStefano Zampini jac->as_tol = 0.; /* used as a preconditioner */ 16914cb006feSStefano Zampini jac->ams_cycle_type = 13; 16924cb006feSStefano Zampini /* Smoothing options */ 1693863406b8SStefano Zampini jac->as_relax_type = 2; 1694863406b8SStefano Zampini jac->as_relax_times = 1; 1695863406b8SStefano Zampini jac->as_relax_weight = 1.0; 1696863406b8SStefano Zampini jac->as_omega = 1.0; 16974cb006feSStefano Zampini /* Vector valued Poisson AMG solver parameters: coarsen type, agg_levels, relax_type, interp_type, Pmax */ 1698863406b8SStefano Zampini jac->as_amg_alpha_opts[0] = 10; 1699863406b8SStefano Zampini jac->as_amg_alpha_opts[1] = 1; 17000bdd8552SBarry Smith jac->as_amg_alpha_opts[2] = 6; 1701863406b8SStefano Zampini jac->as_amg_alpha_opts[3] = 6; 1702863406b8SStefano Zampini jac->as_amg_alpha_opts[4] = 4; 1703863406b8SStefano Zampini jac->as_amg_alpha_theta = 0.25; 17044cb006feSStefano Zampini /* Scalar Poisson AMG solver parameters: coarsen type, agg_levels, relax_type, interp_type, Pmax */ 1705863406b8SStefano Zampini jac->as_amg_beta_opts[0] = 10; 1706863406b8SStefano Zampini jac->as_amg_beta_opts[1] = 1; 17070bdd8552SBarry Smith jac->as_amg_beta_opts[2] = 6; 1708863406b8SStefano Zampini jac->as_amg_beta_opts[3] = 6; 1709863406b8SStefano Zampini jac->as_amg_beta_opts[4] = 4; 1710863406b8SStefano Zampini jac->as_amg_beta_theta = 0.25; 1711863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetPrintLevel,(jac->hsolver,jac->as_print)); 1712863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetMaxIter,(jac->hsolver,jac->as_max_iter)); 17134cb006feSStefano Zampini PetscStackCallStandard(HYPRE_AMSSetCycleType,(jac->hsolver,jac->ams_cycle_type)); 1714863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetTol,(jac->hsolver,jac->as_tol)); 1715863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetSmoothingOptions,(jac->hsolver,jac->as_relax_type, 1716863406b8SStefano Zampini jac->as_relax_times, 1717863406b8SStefano Zampini jac->as_relax_weight, 1718863406b8SStefano Zampini jac->as_omega)); 1719863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetAlphaAMGOptions,(jac->hsolver,jac->as_amg_alpha_opts[0], /* AMG coarsen type */ 1720863406b8SStefano Zampini jac->as_amg_alpha_opts[1], /* AMG agg_levels */ 1721863406b8SStefano Zampini jac->as_amg_alpha_opts[2], /* AMG relax_type */ 1722863406b8SStefano Zampini jac->as_amg_alpha_theta, 1723863406b8SStefano Zampini jac->as_amg_alpha_opts[3], /* AMG interp_type */ 1724863406b8SStefano Zampini jac->as_amg_alpha_opts[4])); /* AMG Pmax */ 1725863406b8SStefano Zampini PetscStackCallStandard(HYPRE_AMSSetBetaAMGOptions,(jac->hsolver,jac->as_amg_beta_opts[0], /* AMG coarsen type */ 1726863406b8SStefano Zampini jac->as_amg_beta_opts[1], /* AMG agg_levels */ 1727863406b8SStefano Zampini jac->as_amg_beta_opts[2], /* AMG relax_type */ 1728863406b8SStefano Zampini jac->as_amg_beta_theta, 1729863406b8SStefano Zampini jac->as_amg_beta_opts[3], /* AMG interp_type */ 1730863406b8SStefano Zampini jac->as_amg_beta_opts[4])); /* AMG Pmax */ 173123df4f25SStefano Zampini /* Zero conductivity */ 173223df4f25SStefano Zampini jac->ams_beta_is_zero = PETSC_FALSE; 173323df4f25SStefano Zampini jac->ams_beta_is_zero_part = PETSC_FALSE; 17344cb006feSStefano Zampini PetscFunctionReturn(0); 17354cb006feSStefano Zampini } 1736863406b8SStefano Zampini ierr = PetscStrcmp("ads",jac->hypre_type,&flag);CHKERRQ(ierr); 1737863406b8SStefano Zampini if (flag) { 1738863406b8SStefano Zampini ierr = HYPRE_ADSCreate(&jac->hsolver); 1739863406b8SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_HYPRE_ADS; 1740863406b8SStefano Zampini pc->ops->view = PCView_HYPRE_ADS; 1741863406b8SStefano Zampini jac->destroy = HYPRE_ADSDestroy; 1742863406b8SStefano Zampini jac->setup = HYPRE_ADSSetup; 1743863406b8SStefano Zampini jac->solve = HYPRE_ADSSolve; 1744863406b8SStefano Zampini jac->coords[0] = NULL; 1745863406b8SStefano Zampini jac->coords[1] = NULL; 1746863406b8SStefano Zampini jac->coords[2] = NULL; 1747863406b8SStefano Zampini /* solver parameters: these are borrowed from mfem package, and they are not the default values from HYPRE ADS */ 1748863406b8SStefano Zampini jac->as_print = 0; 1749863406b8SStefano Zampini jac->as_max_iter = 1; /* used as a preconditioner */ 1750863406b8SStefano Zampini jac->as_tol = 0.; /* used as a preconditioner */ 1751863406b8SStefano Zampini jac->ads_cycle_type = 13; 1752863406b8SStefano Zampini /* Smoothing options */ 1753863406b8SStefano Zampini jac->as_relax_type = 2; 1754863406b8SStefano Zampini jac->as_relax_times = 1; 1755863406b8SStefano Zampini jac->as_relax_weight = 1.0; 1756863406b8SStefano Zampini jac->as_omega = 1.0; 1757863406b8SStefano Zampini /* AMS solver parameters: cycle_type, coarsen type, agg_levels, relax_type, interp_type, Pmax */ 1758863406b8SStefano Zampini jac->ams_cycle_type = 14; 1759863406b8SStefano Zampini jac->as_amg_alpha_opts[0] = 10; 1760863406b8SStefano Zampini jac->as_amg_alpha_opts[1] = 1; 1761863406b8SStefano Zampini jac->as_amg_alpha_opts[2] = 6; 1762863406b8SStefano Zampini jac->as_amg_alpha_opts[3] = 6; 1763863406b8SStefano Zampini jac->as_amg_alpha_opts[4] = 4; 1764863406b8SStefano Zampini jac->as_amg_alpha_theta = 0.25; 1765863406b8SStefano Zampini /* Vector Poisson AMG solver parameters: coarsen type, agg_levels, relax_type, interp_type, Pmax */ 1766863406b8SStefano Zampini jac->as_amg_beta_opts[0] = 10; 1767863406b8SStefano Zampini jac->as_amg_beta_opts[1] = 1; 1768863406b8SStefano Zampini jac->as_amg_beta_opts[2] = 6; 1769863406b8SStefano Zampini jac->as_amg_beta_opts[3] = 6; 1770863406b8SStefano Zampini jac->as_amg_beta_opts[4] = 4; 1771863406b8SStefano Zampini jac->as_amg_beta_theta = 0.25; 1772863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetPrintLevel,(jac->hsolver,jac->as_print)); 1773863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetMaxIter,(jac->hsolver,jac->as_max_iter)); 1774863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetCycleType,(jac->hsolver,jac->ams_cycle_type)); 1775863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetTol,(jac->hsolver,jac->as_tol)); 1776863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetSmoothingOptions,(jac->hsolver,jac->as_relax_type, 1777863406b8SStefano Zampini jac->as_relax_times, 1778863406b8SStefano Zampini jac->as_relax_weight, 1779863406b8SStefano Zampini jac->as_omega)); 1780863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetAMSOptions,(jac->hsolver,jac->ams_cycle_type, /* AMG coarsen type */ 1781863406b8SStefano Zampini jac->as_amg_alpha_opts[0], /* AMG coarsen type */ 1782863406b8SStefano Zampini jac->as_amg_alpha_opts[1], /* AMG agg_levels */ 1783863406b8SStefano Zampini jac->as_amg_alpha_opts[2], /* AMG relax_type */ 1784863406b8SStefano Zampini jac->as_amg_alpha_theta, 1785863406b8SStefano Zampini jac->as_amg_alpha_opts[3], /* AMG interp_type */ 1786863406b8SStefano Zampini jac->as_amg_alpha_opts[4])); /* AMG Pmax */ 1787863406b8SStefano Zampini PetscStackCallStandard(HYPRE_ADSSetAMGOptions,(jac->hsolver,jac->as_amg_beta_opts[0], /* AMG coarsen type */ 1788863406b8SStefano Zampini jac->as_amg_beta_opts[1], /* AMG agg_levels */ 1789863406b8SStefano Zampini jac->as_amg_beta_opts[2], /* AMG relax_type */ 1790863406b8SStefano Zampini jac->as_amg_beta_theta, 1791863406b8SStefano Zampini jac->as_amg_beta_opts[3], /* AMG interp_type */ 1792863406b8SStefano Zampini jac->as_amg_beta_opts[4])); /* AMG Pmax */ 1793863406b8SStefano Zampini PetscFunctionReturn(0); 1794863406b8SStefano Zampini } 1795503cfb0cSBarry Smith ierr = PetscFree(jac->hypre_type);CHKERRQ(ierr); 17962fa5cd67SKarl Rupp 17970298fd71SBarry Smith jac->hypre_type = NULL; 179833263987SBarry Smith SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_UNKNOWN_TYPE,"Unknown HYPRE preconditioner %s; Choices are pilut, parasails, boomeramg, ams",name); 179916d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 180016d9e3a6SLisandro Dalcin } 180116d9e3a6SLisandro Dalcin 180216d9e3a6SLisandro Dalcin /* 180316d9e3a6SLisandro Dalcin It only gets here if the HYPRE type has not been set before the call to 180416d9e3a6SLisandro Dalcin ...SetFromOptions() which actually is most of the time 180516d9e3a6SLisandro Dalcin */ 18064416b707SBarry Smith static PetscErrorCode PCSetFromOptions_HYPRE(PetscOptionItems *PetscOptionsObject,PC pc) 180716d9e3a6SLisandro Dalcin { 180816d9e3a6SLisandro Dalcin PetscErrorCode ierr; 18094ddd07fcSJed Brown PetscInt indx; 1810863406b8SStefano Zampini const char *type[] = {"pilut","parasails","boomeramg","ams","ads"}; 1811ace3abfcSBarry Smith PetscBool flg; 181216d9e3a6SLisandro Dalcin 181316d9e3a6SLisandro Dalcin PetscFunctionBegin; 18149fa463a7SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"HYPRE preconditioner options");CHKERRQ(ierr); 18159c81f712SBarry Smith ierr = PetscOptionsEList("-pc_hypre_type","HYPRE preconditioner type","PCHYPRESetType",type,4,"boomeramg",&indx,&flg);CHKERRQ(ierr); 181616d9e3a6SLisandro Dalcin if (flg) { 181716d9e3a6SLisandro Dalcin ierr = PCHYPRESetType_HYPRE(pc,type[indx]);CHKERRQ(ierr); 181802a17cd4SBarry Smith } else { 181902a17cd4SBarry Smith ierr = PCHYPRESetType_HYPRE(pc,"boomeramg");CHKERRQ(ierr); 182016d9e3a6SLisandro Dalcin } 182116d9e3a6SLisandro Dalcin if (pc->ops->setfromoptions) { 18223931853cSBarry Smith ierr = pc->ops->setfromoptions(PetscOptionsObject,pc);CHKERRQ(ierr); 182316d9e3a6SLisandro Dalcin } 182416d9e3a6SLisandro Dalcin ierr = PetscOptionsTail();CHKERRQ(ierr); 182516d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 182616d9e3a6SLisandro Dalcin } 182716d9e3a6SLisandro Dalcin 182816d9e3a6SLisandro Dalcin /*@C 182916d9e3a6SLisandro Dalcin PCHYPRESetType - Sets which hypre preconditioner you wish to use 183016d9e3a6SLisandro Dalcin 183116d9e3a6SLisandro Dalcin Input Parameters: 183216d9e3a6SLisandro Dalcin + pc - the preconditioner context 1833863406b8SStefano Zampini - name - either pilut, parasails, boomeramg, ams, ads 183416d9e3a6SLisandro Dalcin 183516d9e3a6SLisandro Dalcin Options Database Keys: 1836863406b8SStefano Zampini -pc_hypre_type - One of pilut, parasails, boomeramg, ams, ads 183716d9e3a6SLisandro Dalcin 183816d9e3a6SLisandro Dalcin Level: intermediate 183916d9e3a6SLisandro Dalcin 184016d9e3a6SLisandro Dalcin .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, 184116d9e3a6SLisandro Dalcin PCHYPRE 184216d9e3a6SLisandro Dalcin 184316d9e3a6SLisandro Dalcin @*/ 18447087cfbeSBarry Smith PetscErrorCode PCHYPRESetType(PC pc,const char name[]) 184516d9e3a6SLisandro Dalcin { 18464ac538c5SBarry Smith PetscErrorCode ierr; 184716d9e3a6SLisandro Dalcin 184816d9e3a6SLisandro Dalcin PetscFunctionBegin; 18490700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 185016d9e3a6SLisandro Dalcin PetscValidCharPointer(name,2); 18514ac538c5SBarry Smith ierr = PetscTryMethod(pc,"PCHYPRESetType_C",(PC,const char[]),(pc,name));CHKERRQ(ierr); 185216d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 185316d9e3a6SLisandro Dalcin } 185416d9e3a6SLisandro Dalcin 185516d9e3a6SLisandro Dalcin /*@C 185616d9e3a6SLisandro Dalcin PCHYPREGetType - Gets which hypre preconditioner you are using 185716d9e3a6SLisandro Dalcin 185816d9e3a6SLisandro Dalcin Input Parameter: 185916d9e3a6SLisandro Dalcin . pc - the preconditioner context 186016d9e3a6SLisandro Dalcin 186116d9e3a6SLisandro Dalcin Output Parameter: 1862863406b8SStefano Zampini . name - either pilut, parasails, boomeramg, ams, ads 186316d9e3a6SLisandro Dalcin 186416d9e3a6SLisandro Dalcin Level: intermediate 186516d9e3a6SLisandro Dalcin 186616d9e3a6SLisandro Dalcin .seealso: PCCreate(), PCHYPRESetType(), PCType (for list of available types), PC, 186716d9e3a6SLisandro Dalcin PCHYPRE 186816d9e3a6SLisandro Dalcin 186916d9e3a6SLisandro Dalcin @*/ 18707087cfbeSBarry Smith PetscErrorCode PCHYPREGetType(PC pc,const char *name[]) 187116d9e3a6SLisandro Dalcin { 18724ac538c5SBarry Smith PetscErrorCode ierr; 187316d9e3a6SLisandro Dalcin 187416d9e3a6SLisandro Dalcin PetscFunctionBegin; 18750700a824SBarry Smith PetscValidHeaderSpecific(pc,PC_CLASSID,1); 187616d9e3a6SLisandro Dalcin PetscValidPointer(name,2); 18774ac538c5SBarry Smith ierr = PetscTryMethod(pc,"PCHYPREGetType_C",(PC,const char*[]),(pc,name));CHKERRQ(ierr); 187816d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 187916d9e3a6SLisandro Dalcin } 188016d9e3a6SLisandro Dalcin 188116d9e3a6SLisandro Dalcin /*MC 188216d9e3a6SLisandro Dalcin PCHYPRE - Allows you to use the matrix element based preconditioners in the LLNL package hypre 188316d9e3a6SLisandro Dalcin 188416d9e3a6SLisandro Dalcin Options Database Keys: 1885863406b8SStefano Zampini + -pc_hypre_type - One of pilut, parasails, boomeramg, ams, ads 188616d9e3a6SLisandro Dalcin - Too many others to list, run with -pc_type hypre -pc_hypre_type XXX -help to see options for the XXX 188716d9e3a6SLisandro Dalcin preconditioner 188816d9e3a6SLisandro Dalcin 188916d9e3a6SLisandro Dalcin Level: intermediate 189016d9e3a6SLisandro Dalcin 189116d9e3a6SLisandro Dalcin Notes: Apart from pc_hypre_type (for which there is PCHYPRESetType()), 189216d9e3a6SLisandro Dalcin the many hypre options can ONLY be set via the options database (e.g. the command line 189316d9e3a6SLisandro Dalcin or with PetscOptionsSetValue(), there are no functions to set them) 189416d9e3a6SLisandro Dalcin 1895c231f9e3SBarryFSmith The options -pc_hypre_boomeramg_max_iter and -pc_hypre_boomeramg_tol refer to the number of iterations 18960f1074feSSatish Balay (V-cycles) and tolerance that boomeramg does EACH time it is called. So for example, if 18970f1074feSSatish Balay -pc_hypre_boomeramg_max_iter is set to 2 then 2-V-cycles are being used to define the preconditioner 1898c231f9e3SBarryFSmith (-pc_hypre_boomeramg_tol should be set to 0.0 - the default - to strictly use a fixed number of 18998f87f92bSBarry Smith iterations per hypre call). -ksp_max_it and -ksp_rtol STILL determine the total number of iterations 19000f1074feSSatish Balay and tolerance for the Krylov solver. For example, if -pc_hypre_boomeramg_max_iter is 2 and -ksp_max_it is 10 19010f1074feSSatish Balay then AT MOST twenty V-cycles of boomeramg will be called. 190216d9e3a6SLisandro Dalcin 19030f1074feSSatish Balay Note that the option -pc_hypre_boomeramg_relax_type_all defaults to symmetric relaxation 19040f1074feSSatish Balay (symmetric-SOR/Jacobi), which is required for Krylov solvers like CG that expect symmetry. 19050f1074feSSatish Balay Otherwise, you may want to use -pc_hypre_boomeramg_relax_type_all SOR/Jacobi. 190616d9e3a6SLisandro Dalcin If you wish to use BoomerAMG WITHOUT a Krylov method use -ksp_type richardson NOT -ksp_type preonly 190716d9e3a6SLisandro Dalcin and use -ksp_max_it to control the number of V-cycles. 190816d9e3a6SLisandro Dalcin (see the PETSc FAQ.html at the PETSc website under the Documentation tab). 190916d9e3a6SLisandro Dalcin 191016d9e3a6SLisandro Dalcin 2007-02-03 Using HYPRE-1.11.1b, the routine HYPRE_BoomerAMGSolveT and the option 191116d9e3a6SLisandro Dalcin -pc_hypre_parasails_reuse were failing with SIGSEGV. Dalcin L. 191216d9e3a6SLisandro Dalcin 19135272c319SBarry Smith MatSetNearNullSpace() - if you provide a near null space to your matrix it is ignored by hypre UNLESS you also use 19145272c319SBarry Smith the two options: 19155272c319SBarry Smith + -pc_hypre_boomeramg_nodal_coarsen <n> - where n is from 1 to 6 (see HYPRE_BOOMERAMGSetNodal()) 1916cbc39033SBarry Smith - -pc_hypre_boomeramg_vec_interp_variant <v> where v is from 1 to 3 (see HYPRE_BoomerAMGSetInterpVecVariant()) 19175272c319SBarry Smith 19185272c319SBarry Smith Depending on the linear system you may see the same or different convergence depending on the values you use. 19195272c319SBarry Smith 19209e5bc791SBarry Smith See PCPFMG for access to the hypre Struct PFMG solver 19219e5bc791SBarry Smith 192216d9e3a6SLisandro Dalcin .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, 19239e5bc791SBarry Smith PCHYPRESetType(), PCPFMG 192416d9e3a6SLisandro Dalcin 192516d9e3a6SLisandro Dalcin M*/ 192616d9e3a6SLisandro Dalcin 19278cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_HYPRE(PC pc) 192816d9e3a6SLisandro Dalcin { 192916d9e3a6SLisandro Dalcin PC_HYPRE *jac; 193016d9e3a6SLisandro Dalcin PetscErrorCode ierr; 193116d9e3a6SLisandro Dalcin 193216d9e3a6SLisandro Dalcin PetscFunctionBegin; 1933b00a9115SJed Brown ierr = PetscNewLog(pc,&jac);CHKERRQ(ierr); 19342fa5cd67SKarl Rupp 193516d9e3a6SLisandro Dalcin pc->data = jac; 19368695de01SBarry Smith pc->ops->reset = PCReset_HYPRE; 193716d9e3a6SLisandro Dalcin pc->ops->destroy = PCDestroy_HYPRE; 193816d9e3a6SLisandro Dalcin pc->ops->setfromoptions = PCSetFromOptions_HYPRE; 193916d9e3a6SLisandro Dalcin pc->ops->setup = PCSetUp_HYPRE; 194016d9e3a6SLisandro Dalcin pc->ops->apply = PCApply_HYPRE; 194116d9e3a6SLisandro Dalcin jac->comm_hypre = MPI_COMM_NULL; 1942bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetType_C",PCHYPRESetType_HYPRE);CHKERRQ(ierr); 1943bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPREGetType_C",PCHYPREGetType_HYPRE);CHKERRQ(ierr); 19445ac14e1cSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCSetCoordinates_C",PCSetCoordinates_HYPRE);CHKERRQ(ierr); 19455ac14e1cSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetDiscreteGradient_C",PCHYPRESetDiscreteGradient_HYPRE);CHKERRQ(ierr); 19465ac14e1cSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetDiscreteCurl_C",PCHYPRESetDiscreteCurl_HYPRE);CHKERRQ(ierr); 19476bf688a0SCe Qin ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetInterpolations_C",PCHYPRESetInterpolations_HYPRE);CHKERRQ(ierr); 19485ac14e1cSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetEdgeConstantVectors_C",PCHYPRESetEdgeConstantVectors_HYPRE);CHKERRQ(ierr); 19495ac14e1cSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCHYPRESetPoissonMatrix_C",PCHYPRESetPoissonMatrix_HYPRE);CHKERRQ(ierr); 195016d9e3a6SLisandro Dalcin PetscFunctionReturn(0); 195116d9e3a6SLisandro Dalcin } 1952ebc551c0SBarry Smith 1953f91d8e95SBarry Smith /* ---------------------------------------------------------------------------------------------------------------------------------*/ 1954f91d8e95SBarry Smith 1955ebc551c0SBarry Smith typedef struct { 195668326731SBarry Smith MPI_Comm hcomm; /* does not share comm with HYPRE_StructMatrix because need to create solver before getting matrix */ 1957f91d8e95SBarry Smith HYPRE_StructSolver hsolver; 19589e5bc791SBarry Smith 19599e5bc791SBarry Smith /* keep copy of PFMG options used so may view them */ 19604ddd07fcSJed Brown PetscInt its; 19619e5bc791SBarry Smith double tol; 19624ddd07fcSJed Brown PetscInt relax_type; 19634ddd07fcSJed Brown PetscInt rap_type; 19644ddd07fcSJed Brown PetscInt num_pre_relax,num_post_relax; 19654ddd07fcSJed Brown PetscInt max_levels; 1966ebc551c0SBarry Smith } PC_PFMG; 1967ebc551c0SBarry Smith 1968ebc551c0SBarry Smith PetscErrorCode PCDestroy_PFMG(PC pc) 1969ebc551c0SBarry Smith { 1970ebc551c0SBarry Smith PetscErrorCode ierr; 1971f91d8e95SBarry Smith PC_PFMG *ex = (PC_PFMG*) pc->data; 1972ebc551c0SBarry Smith 1973ebc551c0SBarry Smith PetscFunctionBegin; 19742fa5cd67SKarl Rupp if (ex->hsolver) PetscStackCallStandard(HYPRE_StructPFMGDestroy,(ex->hsolver)); 1975f91d8e95SBarry Smith ierr = MPI_Comm_free(&ex->hcomm);CHKERRQ(ierr); 1976c31cb41cSBarry Smith ierr = PetscFree(pc->data);CHKERRQ(ierr); 1977ebc551c0SBarry Smith PetscFunctionReturn(0); 1978ebc551c0SBarry Smith } 1979ebc551c0SBarry Smith 19809e5bc791SBarry Smith static const char *PFMGRelaxType[] = {"Jacobi","Weighted-Jacobi","symmetric-Red/Black-Gauss-Seidel","Red/Black-Gauss-Seidel"}; 19819e5bc791SBarry Smith static const char *PFMGRAPType[] = {"Galerkin","non-Galerkin"}; 19829e5bc791SBarry Smith 1983ebc551c0SBarry Smith PetscErrorCode PCView_PFMG(PC pc,PetscViewer viewer) 1984ebc551c0SBarry Smith { 1985ebc551c0SBarry Smith PetscErrorCode ierr; 1986ace3abfcSBarry Smith PetscBool iascii; 1987f91d8e95SBarry Smith PC_PFMG *ex = (PC_PFMG*) pc->data; 1988ebc551c0SBarry Smith 1989ebc551c0SBarry Smith PetscFunctionBegin; 1990251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 19919e5bc791SBarry Smith if (iascii) { 19929e5bc791SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," HYPRE PFMG preconditioning\n");CHKERRQ(ierr); 1993efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," max iterations %d\n",ex->its);CHKERRQ(ierr); 1994efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerance %g\n",ex->tol);CHKERRQ(ierr); 1995efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," relax type %s\n",PFMGRelaxType[ex->relax_type]);CHKERRQ(ierr); 1996efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," RAP type %s\n",PFMGRAPType[ex->rap_type]);CHKERRQ(ierr); 1997efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," number pre-relax %d post-relax %d\n",ex->num_pre_relax,ex->num_post_relax);CHKERRQ(ierr); 1998efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," max levels %d\n",ex->max_levels);CHKERRQ(ierr); 19999e5bc791SBarry Smith } 2000ebc551c0SBarry Smith PetscFunctionReturn(0); 2001ebc551c0SBarry Smith } 2002ebc551c0SBarry Smith 20034416b707SBarry Smith PetscErrorCode PCSetFromOptions_PFMG(PetscOptionItems *PetscOptionsObject,PC pc) 2004ebc551c0SBarry Smith { 2005ebc551c0SBarry Smith PetscErrorCode ierr; 2006f91d8e95SBarry Smith PC_PFMG *ex = (PC_PFMG*) pc->data; 2007ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 2008ebc551c0SBarry Smith 2009ebc551c0SBarry Smith PetscFunctionBegin; 2010e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"PFMG options");CHKERRQ(ierr); 20110298fd71SBarry Smith ierr = PetscOptionsBool("-pc_pfmg_print_statistics","Print statistics","HYPRE_StructPFMGSetPrintLevel",flg,&flg,NULL);CHKERRQ(ierr); 201268326731SBarry Smith if (flg) { 2013a0324ebeSBarry Smith int level=3; 2014fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetPrintLevel,(ex->hsolver,level)); 201568326731SBarry Smith } 20160298fd71SBarry Smith ierr = PetscOptionsInt("-pc_pfmg_its","Number of iterations of PFMG to use as preconditioner","HYPRE_StructPFMGSetMaxIter",ex->its,&ex->its,NULL);CHKERRQ(ierr); 2017fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetMaxIter,(ex->hsolver,ex->its)); 20180298fd71SBarry Smith ierr = PetscOptionsInt("-pc_pfmg_num_pre_relax","Number of smoothing steps before coarse grid","HYPRE_StructPFMGSetNumPreRelax",ex->num_pre_relax,&ex->num_pre_relax,NULL);CHKERRQ(ierr); 2019fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetNumPreRelax,(ex->hsolver,ex->num_pre_relax)); 20200298fd71SBarry Smith ierr = PetscOptionsInt("-pc_pfmg_num_post_relax","Number of smoothing steps after coarse grid","HYPRE_StructPFMGSetNumPostRelax",ex->num_post_relax,&ex->num_post_relax,NULL);CHKERRQ(ierr); 2021fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetNumPostRelax,(ex->hsolver,ex->num_post_relax)); 20229e5bc791SBarry Smith 20230298fd71SBarry Smith ierr = PetscOptionsInt("-pc_pfmg_max_levels","Max Levels for MG hierarchy","HYPRE_StructPFMGSetMaxLevels",ex->max_levels,&ex->max_levels,NULL);CHKERRQ(ierr); 2024fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetMaxLevels,(ex->hsolver,ex->max_levels)); 20253b46a515SGlenn Hammond 20260298fd71SBarry Smith ierr = PetscOptionsReal("-pc_pfmg_tol","Tolerance of PFMG","HYPRE_StructPFMGSetTol",ex->tol,&ex->tol,NULL);CHKERRQ(ierr); 2027fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetTol,(ex->hsolver,ex->tol)); 20280298fd71SBarry Smith ierr = PetscOptionsEList("-pc_pfmg_relax_type","Relax type for the up and down cycles","HYPRE_StructPFMGSetRelaxType",PFMGRelaxType,ALEN(PFMGRelaxType),PFMGRelaxType[ex->relax_type],&ex->relax_type,NULL);CHKERRQ(ierr); 2029fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetRelaxType,(ex->hsolver, ex->relax_type)); 20300298fd71SBarry Smith ierr = PetscOptionsEList("-pc_pfmg_rap_type","RAP type","HYPRE_StructPFMGSetRAPType",PFMGRAPType,ALEN(PFMGRAPType),PFMGRAPType[ex->rap_type],&ex->rap_type,NULL);CHKERRQ(ierr); 2031fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetRAPType,(ex->hsolver, ex->rap_type)); 2032ebc551c0SBarry Smith ierr = PetscOptionsTail();CHKERRQ(ierr); 2033ebc551c0SBarry Smith PetscFunctionReturn(0); 2034ebc551c0SBarry Smith } 2035ebc551c0SBarry Smith 2036f91d8e95SBarry Smith PetscErrorCode PCApply_PFMG(PC pc,Vec x,Vec y) 2037f91d8e95SBarry Smith { 2038f91d8e95SBarry Smith PetscErrorCode ierr; 2039f91d8e95SBarry Smith PC_PFMG *ex = (PC_PFMG*) pc->data; 2040d9ca1df4SBarry Smith PetscScalar *yy; 2041d9ca1df4SBarry Smith const PetscScalar *xx; 20424ddd07fcSJed Brown PetscInt ilower[3],iupper[3]; 204368326731SBarry Smith Mat_HYPREStruct *mx = (Mat_HYPREStruct*)(pc->pmat->data); 2044f91d8e95SBarry Smith 2045f91d8e95SBarry Smith PetscFunctionBegin; 2046dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 2047aa219208SBarry Smith ierr = DMDAGetCorners(mx->da,&ilower[0],&ilower[1],&ilower[2],&iupper[0],&iupper[1],&iupper[2]);CHKERRQ(ierr); 2048f91d8e95SBarry Smith iupper[0] += ilower[0] - 1; 2049f91d8e95SBarry Smith iupper[1] += ilower[1] - 1; 2050f91d8e95SBarry Smith iupper[2] += ilower[2] - 1; 2051f91d8e95SBarry Smith 2052f91d8e95SBarry Smith /* copy x values over to hypre */ 2053fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructVectorSetConstantValues,(mx->hb,0.0)); 2054d9ca1df4SBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 2055d9ca1df4SBarry Smith PetscStackCallStandard(HYPRE_StructVectorSetBoxValues,(mx->hb,(HYPRE_Int *)ilower,(HYPRE_Int *)iupper,(PetscScalar*)xx)); 2056d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 2057fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructVectorAssemble,(mx->hb)); 2058fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSolve,(ex->hsolver,mx->hmat,mx->hb,mx->hx)); 2059f91d8e95SBarry Smith 2060f91d8e95SBarry Smith /* copy solution values back to PETSc */ 2061f91d8e95SBarry Smith ierr = VecGetArray(y,&yy);CHKERRQ(ierr); 20628b1f7689SBarry Smith PetscStackCallStandard(HYPRE_StructVectorGetBoxValues,(mx->hx,(HYPRE_Int *)ilower,(HYPRE_Int *)iupper,yy)); 2063f91d8e95SBarry Smith ierr = VecRestoreArray(y,&yy);CHKERRQ(ierr); 2064f91d8e95SBarry Smith PetscFunctionReturn(0); 2065f91d8e95SBarry Smith } 2066f91d8e95SBarry Smith 2067ace3abfcSBarry Smith static PetscErrorCode PCApplyRichardson_PFMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol, PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt *outits,PCRichardsonConvergedReason *reason) 20689e5bc791SBarry Smith { 20699e5bc791SBarry Smith PC_PFMG *jac = (PC_PFMG*)pc->data; 20709e5bc791SBarry Smith PetscErrorCode ierr; 20714ddd07fcSJed Brown PetscInt oits; 20729e5bc791SBarry Smith 20739e5bc791SBarry Smith PetscFunctionBegin; 2074dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 2075fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetMaxIter,(jac->hsolver,its*jac->its)); 2076fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetTol,(jac->hsolver,rtol)); 20779e5bc791SBarry Smith 20789e5bc791SBarry Smith ierr = PCApply_PFMG(pc,b,y);CHKERRQ(ierr); 20798b1f7689SBarry Smith PetscStackCallStandard(HYPRE_StructPFMGGetNumIterations,(jac->hsolver,(HYPRE_Int *)&oits)); 20809e5bc791SBarry Smith *outits = oits; 20819e5bc791SBarry Smith if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS; 20829e5bc791SBarry Smith else *reason = PCRICHARDSON_CONVERGED_RTOL; 2083fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetTol,(jac->hsolver,jac->tol)); 2084fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetMaxIter,(jac->hsolver,jac->its)); 20859e5bc791SBarry Smith PetscFunctionReturn(0); 20869e5bc791SBarry Smith } 20879e5bc791SBarry Smith 20889e5bc791SBarry Smith 20893a32d3dbSGlenn Hammond PetscErrorCode PCSetUp_PFMG(PC pc) 20903a32d3dbSGlenn Hammond { 20913a32d3dbSGlenn Hammond PetscErrorCode ierr; 20923a32d3dbSGlenn Hammond PC_PFMG *ex = (PC_PFMG*) pc->data; 20933a32d3dbSGlenn Hammond Mat_HYPREStruct *mx = (Mat_HYPREStruct*)(pc->pmat->data); 2094ace3abfcSBarry Smith PetscBool flg; 20953a32d3dbSGlenn Hammond 20963a32d3dbSGlenn Hammond PetscFunctionBegin; 2097251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATHYPRESTRUCT,&flg);CHKERRQ(ierr); 2098ce94432eSBarry Smith if (!flg) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_INCOMP,"Must use MATHYPRESTRUCT with this preconditioner"); 20993a32d3dbSGlenn Hammond 21003a32d3dbSGlenn Hammond /* create the hypre solver object and set its information */ 21012fa5cd67SKarl Rupp if (ex->hsolver) PetscStackCallStandard(HYPRE_StructPFMGDestroy,(ex->hsolver)); 2102fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGCreate,(ex->hcomm,&ex->hsolver)); 2103fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetup,(ex->hsolver,mx->hmat,mx->hb,mx->hx)); 2104fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGSetZeroGuess,(ex->hsolver)); 21053a32d3dbSGlenn Hammond PetscFunctionReturn(0); 21063a32d3dbSGlenn Hammond } 21073a32d3dbSGlenn Hammond 2108ebc551c0SBarry Smith /*MC 2109ebc551c0SBarry Smith PCPFMG - the hypre PFMG multigrid solver 2110ebc551c0SBarry Smith 2111ebc551c0SBarry Smith Level: advanced 2112ebc551c0SBarry Smith 21139e5bc791SBarry Smith Options Database: 21149e5bc791SBarry Smith + -pc_pfmg_its <its> number of iterations of PFMG to use as preconditioner 21159e5bc791SBarry Smith . -pc_pfmg_num_pre_relax <steps> number of smoothing steps before coarse grid 21169e5bc791SBarry Smith . -pc_pfmg_num_post_relax <steps> number of smoothing steps after coarse grid 21179e5bc791SBarry Smith . -pc_pfmg_tol <tol> tolerance of PFMG 21189e5bc791SBarry Smith . -pc_pfmg_relax_type -relaxation type for the up and down cycles, one of Jacobi,Weighted-Jacobi,symmetric-Red/Black-Gauss-Seidel,Red/Black-Gauss-Seidel 21199e5bc791SBarry Smith - -pc_pfmg_rap_type - type of coarse matrix generation, one of Galerkin,non-Galerkin 2120f91d8e95SBarry Smith 21219e5bc791SBarry Smith Notes: This is for CELL-centered descretizations 21229e5bc791SBarry Smith 21238e395302SJed Brown This must be used with the MATHYPRESTRUCT matrix type. 2124aa219208SBarry Smith This is less general than in hypre, it supports only one block per process defined by a PETSc DMDA. 21259e5bc791SBarry Smith 21269e5bc791SBarry Smith .seealso: PCMG, MATHYPRESTRUCT 2127ebc551c0SBarry Smith M*/ 2128ebc551c0SBarry Smith 21298cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_PFMG(PC pc) 2130ebc551c0SBarry Smith { 2131ebc551c0SBarry Smith PetscErrorCode ierr; 2132ebc551c0SBarry Smith PC_PFMG *ex; 2133ebc551c0SBarry Smith 2134ebc551c0SBarry Smith PetscFunctionBegin; 2135b00a9115SJed Brown ierr = PetscNew(&ex);CHKERRQ(ierr); \ 213668326731SBarry Smith pc->data = ex; 2137ebc551c0SBarry Smith 21389e5bc791SBarry Smith ex->its = 1; 21399e5bc791SBarry Smith ex->tol = 1.e-8; 21409e5bc791SBarry Smith ex->relax_type = 1; 21419e5bc791SBarry Smith ex->rap_type = 0; 21429e5bc791SBarry Smith ex->num_pre_relax = 1; 21439e5bc791SBarry Smith ex->num_post_relax = 1; 21443b46a515SGlenn Hammond ex->max_levels = 0; 21459e5bc791SBarry Smith 2146ebc551c0SBarry Smith pc->ops->setfromoptions = PCSetFromOptions_PFMG; 2147ebc551c0SBarry Smith pc->ops->view = PCView_PFMG; 2148ebc551c0SBarry Smith pc->ops->destroy = PCDestroy_PFMG; 2149f91d8e95SBarry Smith pc->ops->apply = PCApply_PFMG; 21509e5bc791SBarry Smith pc->ops->applyrichardson = PCApplyRichardson_PFMG; 215168326731SBarry Smith pc->ops->setup = PCSetUp_PFMG; 21522fa5cd67SKarl Rupp 2153ce94432eSBarry Smith ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)pc),&(ex->hcomm));CHKERRQ(ierr); 2154fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_StructPFMGCreate,(ex->hcomm,&ex->hsolver)); 2155ebc551c0SBarry Smith PetscFunctionReturn(0); 2156ebc551c0SBarry Smith } 2157d851a50bSGlenn Hammond 2158325fc9f4SBarry Smith /* ---------------------------------------------------------------------------------------------------------------------------------------------------*/ 2159325fc9f4SBarry Smith 2160d851a50bSGlenn Hammond /* we know we are working with a HYPRE_SStructMatrix */ 2161d851a50bSGlenn Hammond typedef struct { 2162d851a50bSGlenn Hammond MPI_Comm hcomm; /* does not share comm with HYPRE_SStructMatrix because need to create solver before getting matrix */ 2163d851a50bSGlenn Hammond HYPRE_SStructSolver ss_solver; 2164d851a50bSGlenn Hammond 2165d851a50bSGlenn Hammond /* keep copy of SYSPFMG options used so may view them */ 21664ddd07fcSJed Brown PetscInt its; 2167d851a50bSGlenn Hammond double tol; 21684ddd07fcSJed Brown PetscInt relax_type; 21694ddd07fcSJed Brown PetscInt num_pre_relax,num_post_relax; 2170d851a50bSGlenn Hammond } PC_SysPFMG; 2171d851a50bSGlenn Hammond 2172d851a50bSGlenn Hammond PetscErrorCode PCDestroy_SysPFMG(PC pc) 2173d851a50bSGlenn Hammond { 2174d851a50bSGlenn Hammond PetscErrorCode ierr; 2175d851a50bSGlenn Hammond PC_SysPFMG *ex = (PC_SysPFMG*) pc->data; 2176d851a50bSGlenn Hammond 2177d851a50bSGlenn Hammond PetscFunctionBegin; 21782fa5cd67SKarl Rupp if (ex->ss_solver) PetscStackCallStandard(HYPRE_SStructSysPFMGDestroy,(ex->ss_solver)); 2179d851a50bSGlenn Hammond ierr = MPI_Comm_free(&ex->hcomm);CHKERRQ(ierr); 2180c31cb41cSBarry Smith ierr = PetscFree(pc->data);CHKERRQ(ierr); 2181d851a50bSGlenn Hammond PetscFunctionReturn(0); 2182d851a50bSGlenn Hammond } 2183d851a50bSGlenn Hammond 2184d851a50bSGlenn Hammond static const char *SysPFMGRelaxType[] = {"Weighted-Jacobi","Red/Black-Gauss-Seidel"}; 2185d851a50bSGlenn Hammond 2186d851a50bSGlenn Hammond PetscErrorCode PCView_SysPFMG(PC pc,PetscViewer viewer) 2187d851a50bSGlenn Hammond { 2188d851a50bSGlenn Hammond PetscErrorCode ierr; 2189ace3abfcSBarry Smith PetscBool iascii; 2190d851a50bSGlenn Hammond PC_SysPFMG *ex = (PC_SysPFMG*) pc->data; 2191d851a50bSGlenn Hammond 2192d851a50bSGlenn Hammond PetscFunctionBegin; 2193251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 2194d851a50bSGlenn Hammond if (iascii) { 2195d851a50bSGlenn Hammond ierr = PetscViewerASCIIPrintf(viewer," HYPRE SysPFMG preconditioning\n");CHKERRQ(ierr); 2196efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," max iterations %d\n",ex->its);CHKERRQ(ierr); 2197efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," tolerance %g\n",ex->tol);CHKERRQ(ierr); 2198efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," relax type %s\n",PFMGRelaxType[ex->relax_type]);CHKERRQ(ierr); 2199efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," number pre-relax %d post-relax %d\n",ex->num_pre_relax,ex->num_post_relax);CHKERRQ(ierr); 2200d851a50bSGlenn Hammond } 2201d851a50bSGlenn Hammond PetscFunctionReturn(0); 2202d851a50bSGlenn Hammond } 2203d851a50bSGlenn Hammond 22044416b707SBarry Smith PetscErrorCode PCSetFromOptions_SysPFMG(PetscOptionItems *PetscOptionsObject,PC pc) 2205d851a50bSGlenn Hammond { 2206d851a50bSGlenn Hammond PetscErrorCode ierr; 2207d851a50bSGlenn Hammond PC_SysPFMG *ex = (PC_SysPFMG*) pc->data; 2208ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 2209d851a50bSGlenn Hammond 2210d851a50bSGlenn Hammond PetscFunctionBegin; 2211e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"SysPFMG options");CHKERRQ(ierr); 22120298fd71SBarry Smith ierr = PetscOptionsBool("-pc_syspfmg_print_statistics","Print statistics","HYPRE_SStructSysPFMGSetPrintLevel",flg,&flg,NULL);CHKERRQ(ierr); 2213d851a50bSGlenn Hammond if (flg) { 2214d851a50bSGlenn Hammond int level=3; 2215fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetPrintLevel,(ex->ss_solver,level)); 2216d851a50bSGlenn Hammond } 22170298fd71SBarry Smith ierr = PetscOptionsInt("-pc_syspfmg_its","Number of iterations of SysPFMG to use as preconditioner","HYPRE_SStructSysPFMGSetMaxIter",ex->its,&ex->its,NULL);CHKERRQ(ierr); 2218fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetMaxIter,(ex->ss_solver,ex->its)); 22190298fd71SBarry Smith ierr = PetscOptionsInt("-pc_syspfmg_num_pre_relax","Number of smoothing steps before coarse grid","HYPRE_SStructSysPFMGSetNumPreRelax",ex->num_pre_relax,&ex->num_pre_relax,NULL);CHKERRQ(ierr); 2220fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetNumPreRelax,(ex->ss_solver,ex->num_pre_relax)); 22210298fd71SBarry Smith ierr = PetscOptionsInt("-pc_syspfmg_num_post_relax","Number of smoothing steps after coarse grid","HYPRE_SStructSysPFMGSetNumPostRelax",ex->num_post_relax,&ex->num_post_relax,NULL);CHKERRQ(ierr); 2222fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetNumPostRelax,(ex->ss_solver,ex->num_post_relax)); 2223d851a50bSGlenn Hammond 22240298fd71SBarry Smith ierr = PetscOptionsReal("-pc_syspfmg_tol","Tolerance of SysPFMG","HYPRE_SStructSysPFMGSetTol",ex->tol,&ex->tol,NULL);CHKERRQ(ierr); 2225fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetTol,(ex->ss_solver,ex->tol)); 222661710fbeSStefano Zampini ierr = PetscOptionsEList("-pc_syspfmg_relax_type","Relax type for the up and down cycles","HYPRE_SStructSysPFMGSetRelaxType",SysPFMGRelaxType,ALEN(SysPFMGRelaxType),SysPFMGRelaxType[ex->relax_type],&ex->relax_type,NULL);CHKERRQ(ierr); 2227fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetRelaxType,(ex->ss_solver, ex->relax_type)); 2228d851a50bSGlenn Hammond ierr = PetscOptionsTail();CHKERRQ(ierr); 2229d851a50bSGlenn Hammond PetscFunctionReturn(0); 2230d851a50bSGlenn Hammond } 2231d851a50bSGlenn Hammond 2232d851a50bSGlenn Hammond PetscErrorCode PCApply_SysPFMG(PC pc,Vec x,Vec y) 2233d851a50bSGlenn Hammond { 2234d851a50bSGlenn Hammond PetscErrorCode ierr; 2235d851a50bSGlenn Hammond PC_SysPFMG *ex = (PC_SysPFMG*) pc->data; 2236d9ca1df4SBarry Smith PetscScalar *yy; 2237d9ca1df4SBarry Smith const PetscScalar *xx; 22384ddd07fcSJed Brown PetscInt ilower[3],iupper[3]; 2239d851a50bSGlenn Hammond Mat_HYPRESStruct *mx = (Mat_HYPRESStruct*)(pc->pmat->data); 22404ddd07fcSJed Brown PetscInt ordering= mx->dofs_order; 22414ddd07fcSJed Brown PetscInt nvars = mx->nvars; 22424ddd07fcSJed Brown PetscInt part = 0; 22434ddd07fcSJed Brown PetscInt size; 22444ddd07fcSJed Brown PetscInt i; 2245d851a50bSGlenn Hammond 2246d851a50bSGlenn Hammond PetscFunctionBegin; 2247dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 2248aa219208SBarry Smith ierr = DMDAGetCorners(mx->da,&ilower[0],&ilower[1],&ilower[2],&iupper[0],&iupper[1],&iupper[2]);CHKERRQ(ierr); 2249d851a50bSGlenn Hammond iupper[0] += ilower[0] - 1; 2250d851a50bSGlenn Hammond iupper[1] += ilower[1] - 1; 2251d851a50bSGlenn Hammond iupper[2] += ilower[2] - 1; 2252d851a50bSGlenn Hammond 2253d851a50bSGlenn Hammond size = 1; 22542fa5cd67SKarl Rupp for (i= 0; i< 3; i++) size *= (iupper[i]-ilower[i]+1); 22552fa5cd67SKarl Rupp 2256d851a50bSGlenn Hammond /* copy x values over to hypre for variable ordering */ 2257d851a50bSGlenn Hammond if (ordering) { 2258fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructVectorSetConstantValues,(mx->ss_b,0.0)); 2259d9ca1df4SBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 2260d9ca1df4SBarry Smith for (i= 0; i< nvars; i++) PetscStackCallStandard(HYPRE_SStructVectorSetBoxValues,(mx->ss_b,part,(HYPRE_Int *)ilower,(HYPRE_Int *)iupper,i,(PetscScalar*)xx+(size*i))); 2261d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 2262fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructVectorAssemble,(mx->ss_b)); 2263fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructMatrixMatvec,(1.0,mx->ss_mat,mx->ss_b,0.0,mx->ss_x)); 2264fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSolve,(ex->ss_solver,mx->ss_mat,mx->ss_b,mx->ss_x)); 2265d851a50bSGlenn Hammond 2266d851a50bSGlenn Hammond /* copy solution values back to PETSc */ 2267d851a50bSGlenn Hammond ierr = VecGetArray(y,&yy);CHKERRQ(ierr); 22688b1f7689SBarry Smith for (i= 0; i< nvars; i++) PetscStackCallStandard(HYPRE_SStructVectorGetBoxValues,(mx->ss_x,part,(HYPRE_Int *)ilower,(HYPRE_Int *)iupper,i,yy+(size*i))); 2269d851a50bSGlenn Hammond ierr = VecRestoreArray(y,&yy);CHKERRQ(ierr); 2270a65764d7SBarry Smith } else { /* nodal ordering must be mapped to variable ordering for sys_pfmg */ 2271d851a50bSGlenn Hammond PetscScalar *z; 22724ddd07fcSJed Brown PetscInt j, k; 2273d851a50bSGlenn Hammond 2274785e854fSJed Brown ierr = PetscMalloc1(nvars*size,&z);CHKERRQ(ierr); 2275fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructVectorSetConstantValues,(mx->ss_b,0.0)); 2276d9ca1df4SBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 2277d851a50bSGlenn Hammond 2278d851a50bSGlenn Hammond /* transform nodal to hypre's variable ordering for sys_pfmg */ 2279d851a50bSGlenn Hammond for (i= 0; i< size; i++) { 2280d851a50bSGlenn Hammond k= i*nvars; 22812fa5cd67SKarl Rupp for (j= 0; j< nvars; j++) z[j*size+i]= xx[k+j]; 2282d851a50bSGlenn Hammond } 22838b1f7689SBarry Smith for (i= 0; i< nvars; i++) PetscStackCallStandard(HYPRE_SStructVectorSetBoxValues,(mx->ss_b,part,(HYPRE_Int *)ilower,(HYPRE_Int *)iupper,i,z+(size*i))); 2284d9ca1df4SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 2285fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructVectorAssemble,(mx->ss_b)); 2286fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSolve,(ex->ss_solver,mx->ss_mat,mx->ss_b,mx->ss_x)); 2287d851a50bSGlenn Hammond 2288d851a50bSGlenn Hammond /* copy solution values back to PETSc */ 2289d851a50bSGlenn Hammond ierr = VecGetArray(y,&yy);CHKERRQ(ierr); 22908b1f7689SBarry Smith for (i= 0; i< nvars; i++) PetscStackCallStandard(HYPRE_SStructVectorGetBoxValues,(mx->ss_x,part,(HYPRE_Int *)ilower,(HYPRE_Int *)iupper,i,z+(size*i))); 2291d851a50bSGlenn Hammond /* transform hypre's variable ordering for sys_pfmg to nodal ordering */ 2292d851a50bSGlenn Hammond for (i= 0; i< size; i++) { 2293d851a50bSGlenn Hammond k= i*nvars; 22942fa5cd67SKarl Rupp for (j= 0; j< nvars; j++) yy[k+j]= z[j*size+i]; 2295d851a50bSGlenn Hammond } 2296d851a50bSGlenn Hammond ierr = VecRestoreArray(y,&yy);CHKERRQ(ierr); 2297d851a50bSGlenn Hammond ierr = PetscFree(z);CHKERRQ(ierr); 2298d851a50bSGlenn Hammond } 2299d851a50bSGlenn Hammond PetscFunctionReturn(0); 2300d851a50bSGlenn Hammond } 2301d851a50bSGlenn Hammond 2302ace3abfcSBarry Smith static PetscErrorCode PCApplyRichardson_SysPFMG(PC pc,Vec b,Vec y,Vec w,PetscReal rtol,PetscReal abstol, PetscReal dtol,PetscInt its,PetscBool guesszero,PetscInt *outits,PCRichardsonConvergedReason *reason) 2303d851a50bSGlenn Hammond { 2304d851a50bSGlenn Hammond PC_SysPFMG *jac = (PC_SysPFMG*)pc->data; 2305d851a50bSGlenn Hammond PetscErrorCode ierr; 23064ddd07fcSJed Brown PetscInt oits; 2307d851a50bSGlenn Hammond 2308d851a50bSGlenn Hammond PetscFunctionBegin; 2309dff31646SBarry Smith ierr = PetscCitationsRegister(hypreCitation,&cite);CHKERRQ(ierr); 2310fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetMaxIter,(jac->ss_solver,its*jac->its)); 2311fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetTol,(jac->ss_solver,rtol)); 2312d851a50bSGlenn Hammond ierr = PCApply_SysPFMG(pc,b,y);CHKERRQ(ierr); 23138b1f7689SBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGGetNumIterations,(jac->ss_solver,(HYPRE_Int *)&oits)); 2314d851a50bSGlenn Hammond *outits = oits; 2315d851a50bSGlenn Hammond if (oits == its) *reason = PCRICHARDSON_CONVERGED_ITS; 2316d851a50bSGlenn Hammond else *reason = PCRICHARDSON_CONVERGED_RTOL; 2317fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetTol,(jac->ss_solver,jac->tol)); 2318fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetMaxIter,(jac->ss_solver,jac->its)); 2319d851a50bSGlenn Hammond PetscFunctionReturn(0); 2320d851a50bSGlenn Hammond } 2321d851a50bSGlenn Hammond 2322d851a50bSGlenn Hammond PetscErrorCode PCSetUp_SysPFMG(PC pc) 2323d851a50bSGlenn Hammond { 2324d851a50bSGlenn Hammond PetscErrorCode ierr; 2325d851a50bSGlenn Hammond PC_SysPFMG *ex = (PC_SysPFMG*) pc->data; 2326d851a50bSGlenn Hammond Mat_HYPRESStruct *mx = (Mat_HYPRESStruct*)(pc->pmat->data); 2327ace3abfcSBarry Smith PetscBool flg; 2328d851a50bSGlenn Hammond 2329d851a50bSGlenn Hammond PetscFunctionBegin; 2330251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATHYPRESSTRUCT,&flg);CHKERRQ(ierr); 2331ce94432eSBarry Smith if (!flg) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_INCOMP,"Must use MATHYPRESSTRUCT with this preconditioner"); 2332d851a50bSGlenn Hammond 2333d851a50bSGlenn Hammond /* create the hypre sstruct solver object and set its information */ 23342fa5cd67SKarl Rupp if (ex->ss_solver) PetscStackCallStandard(HYPRE_SStructSysPFMGDestroy,(ex->ss_solver)); 2335fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGCreate,(ex->hcomm,&ex->ss_solver)); 2336fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetZeroGuess,(ex->ss_solver)); 2337fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGSetup,(ex->ss_solver,mx->ss_mat,mx->ss_b,mx->ss_x)); 2338d851a50bSGlenn Hammond PetscFunctionReturn(0); 2339d851a50bSGlenn Hammond } 2340d851a50bSGlenn Hammond 2341d851a50bSGlenn Hammond /*MC 2342d851a50bSGlenn Hammond PCSysPFMG - the hypre SysPFMG multigrid solver 2343d851a50bSGlenn Hammond 2344d851a50bSGlenn Hammond Level: advanced 2345d851a50bSGlenn Hammond 2346d851a50bSGlenn Hammond Options Database: 2347d851a50bSGlenn Hammond + -pc_syspfmg_its <its> number of iterations of SysPFMG to use as preconditioner 2348d851a50bSGlenn Hammond . -pc_syspfmg_num_pre_relax <steps> number of smoothing steps before coarse grid 2349d851a50bSGlenn Hammond . -pc_syspfmg_num_post_relax <steps> number of smoothing steps after coarse grid 2350d851a50bSGlenn Hammond . -pc_syspfmg_tol <tol> tolerance of SysPFMG 2351d851a50bSGlenn Hammond . -pc_syspfmg_relax_type -relaxation type for the up and down cycles, one of Weighted-Jacobi,Red/Black-Gauss-Seidel 2352d851a50bSGlenn Hammond 2353d851a50bSGlenn Hammond Notes: This is for CELL-centered descretizations 2354d851a50bSGlenn Hammond 2355f6680f47SSatish Balay This must be used with the MATHYPRESSTRUCT matrix type. 2356aa219208SBarry Smith This is less general than in hypre, it supports only one part, and one block per process defined by a PETSc DMDA. 2357d851a50bSGlenn Hammond Also, only cell-centered variables. 2358d851a50bSGlenn Hammond 2359d851a50bSGlenn Hammond .seealso: PCMG, MATHYPRESSTRUCT 2360d851a50bSGlenn Hammond M*/ 2361d851a50bSGlenn Hammond 23628cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_SysPFMG(PC pc) 2363d851a50bSGlenn Hammond { 2364d851a50bSGlenn Hammond PetscErrorCode ierr; 2365d851a50bSGlenn Hammond PC_SysPFMG *ex; 2366d851a50bSGlenn Hammond 2367d851a50bSGlenn Hammond PetscFunctionBegin; 2368b00a9115SJed Brown ierr = PetscNew(&ex);CHKERRQ(ierr); \ 2369d851a50bSGlenn Hammond pc->data = ex; 2370d851a50bSGlenn Hammond 2371d851a50bSGlenn Hammond ex->its = 1; 2372d851a50bSGlenn Hammond ex->tol = 1.e-8; 2373d851a50bSGlenn Hammond ex->relax_type = 1; 2374d851a50bSGlenn Hammond ex->num_pre_relax = 1; 2375d851a50bSGlenn Hammond ex->num_post_relax = 1; 2376d851a50bSGlenn Hammond 2377d851a50bSGlenn Hammond pc->ops->setfromoptions = PCSetFromOptions_SysPFMG; 2378d851a50bSGlenn Hammond pc->ops->view = PCView_SysPFMG; 2379d851a50bSGlenn Hammond pc->ops->destroy = PCDestroy_SysPFMG; 2380d851a50bSGlenn Hammond pc->ops->apply = PCApply_SysPFMG; 2381d851a50bSGlenn Hammond pc->ops->applyrichardson = PCApplyRichardson_SysPFMG; 2382d851a50bSGlenn Hammond pc->ops->setup = PCSetUp_SysPFMG; 23832fa5cd67SKarl Rupp 2384ce94432eSBarry Smith ierr = MPI_Comm_dup(PetscObjectComm((PetscObject)pc),&(ex->hcomm));CHKERRQ(ierr); 2385fd3f9acdSBarry Smith PetscStackCallStandard(HYPRE_SStructSysPFMGCreate,(ex->hcomm,&ex->ss_solver)); 2386d851a50bSGlenn Hammond PetscFunctionReturn(0); 2387d851a50bSGlenn Hammond } 2388