153cdbc3dSStefano Zampini /* TODOLIST 2eb97c9d2SStefano Zampini 3eb97c9d2SStefano Zampini Solvers 4a0d3c3abSStefano Zampini - Add support for cholesky for coarse solver (similar to local solvers) 5eb97c9d2SStefano Zampini - Propagate ksp prefixes for solvers to mat objects? 6eb97c9d2SStefano Zampini 7eb97c9d2SStefano Zampini User interface 80f202f7eSStefano Zampini - ** DM attached to pc? 9eb97c9d2SStefano Zampini 10eb97c9d2SStefano Zampini Debugging output 11b9b85e73SStefano Zampini - * Better management of verbosity levels of debugging output 12eb97c9d2SStefano Zampini 13eb97c9d2SStefano Zampini Extra 14b9b85e73SStefano Zampini - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)? 15eb97c9d2SStefano Zampini - BDDC with MG framework? 16eb97c9d2SStefano Zampini 17eb97c9d2SStefano Zampini FETIDP 18eb97c9d2SStefano Zampini - Move FETIDP code to its own classes 19eb97c9d2SStefano Zampini 20eb97c9d2SStefano Zampini MATIS related operations contained in BDDC code 21eb97c9d2SStefano Zampini - Provide general case for subassembling 22eb97c9d2SStefano Zampini 2353cdbc3dSStefano Zampini */ 240c7d97c5SJed Brown 25ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 26ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 273b03a366Sstefano_zampini #include <petscblaslapack.h> 28674ae819SStefano Zampini 290369aaf7SStefano Zampini /* temporarily declare it */ 300369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 310369aaf7SStefano Zampini 320c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 330c7d97c5SJed Brown #undef __FUNCT__ 340c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 358c34d3f5SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptions *PetscOptionsObject,PC pc) 360c7d97c5SJed Brown { 370c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 380c7d97c5SJed Brown PetscErrorCode ierr; 390c7d97c5SJed Brown 400c7d97c5SJed Brown PetscFunctionBegin; 41e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 428eeda7d8SStefano Zampini /* Verbose debugging */ 438eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 448eeda7d8SStefano Zampini /* Primal space cumstomization */ 4508a5cf49SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_local_mat_graph","Use or not adjacency graph of local mat for interface analysis","none",pcbddc->use_local_adj,&pcbddc->use_local_adj,NULL);CHKERRQ(ierr); 468eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_vertices","Use or not corner dofs in coarse space","none",pcbddc->use_vertices,&pcbddc->use_vertices,NULL);CHKERRQ(ierr); 478eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_edges","Use or not edge constraints in coarse space","none",pcbddc->use_edges,&pcbddc->use_edges,NULL);CHKERRQ(ierr); 488eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_faces","Use or not face constraints in coarse space","none",pcbddc->use_faces,&pcbddc->use_faces,NULL);CHKERRQ(ierr); 4914f95afaSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_vertex_size","Connected components smaller or equal to vertex size will be considered as primal vertices","none",pcbddc->vertex_size,&pcbddc->vertex_size,NULL);CHKERRQ(ierr); 506661aa1dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_true_nnsp","Use near null space attached to the matrix without modifications","none",pcbddc->use_nnsp_true,&pcbddc->use_nnsp_true,NULL);CHKERRQ(ierr); 5114f95afaSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is always used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr); 528eeda7d8SStefano Zampini /* Change of basis */ 53b9b85e73SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_change_of_basis","Use or not internal change of basis on local edge nodes","none",pcbddc->use_change_of_basis,&pcbddc->use_change_of_basis,NULL);CHKERRQ(ierr); 54b9b85e73SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_change_on_faces","Use or not internal change of basis on local face nodes","none",pcbddc->use_change_on_faces,&pcbddc->use_change_on_faces,NULL);CHKERRQ(ierr); 55674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 56674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 57674ae819SStefano Zampini } 588eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 598eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_switch_static","Switch on static condensation ops around the interface preconditioner","none",pcbddc->switch_static,&pcbddc->switch_static,NULL);CHKERRQ(ierr); 6074e2c79eSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_redistribute","Number of procs where to redistribute coarse problem","none",pcbddc->redistribute_coarse,&pcbddc->redistribute_coarse,NULL);CHKERRQ(ierr); 610298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 622b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 63323d395dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_coarse_estimates","Use estimated eigenvalues for coarse problem","none",pcbddc->use_coarse_estimates,&pcbddc->use_coarse_estimates,NULL);CHKERRQ(ierr); 64674ae819SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_deluxe_scaling","Use deluxe scaling for BDDC","none",pcbddc->use_deluxe_scaling,&pcbddc->use_deluxe_scaling,NULL);CHKERRQ(ierr); 65b96c3477SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_rebuild","Whether or not the interface graph for Schur principal minors has to be rebuilt (i.e. define the interface without any adjacency)","none",pcbddc->sub_schurs_rebuild,&pcbddc->sub_schurs_rebuild,NULL);CHKERRQ(ierr); 66b96c3477SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_schur_layers","Number of dofs' layers for the computation of principal minors (i.e. -1 uses all dofs)","none",pcbddc->sub_schurs_layers,&pcbddc->sub_schurs_layers,NULL);CHKERRQ(ierr); 67b96c3477SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_use_useradj","Whether or not the CSR graph specified by the user should be used for computing successive layers (default is to use adj of local mat)","none",pcbddc->sub_schurs_use_useradj,&pcbddc->sub_schurs_use_useradj,NULL);CHKERRQ(ierr); 68683d3df6SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_exact","Whether or not to use the exact Schur complement instead of the reduced one (which excludes size 1 cc)","none",pcbddc->sub_schurs_exact_schur,&pcbddc->sub_schurs_exact_schur,NULL);CHKERRQ(ierr); 69bf3a8328SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_zerorows","Zero rows and columns of deluxe operators associated with primal dofs","none",pcbddc->deluxe_zerorows,&pcbddc->deluxe_zerorows,NULL);CHKERRQ(ierr); 70ac632422SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_faster","Faster application of deluxe scaling (requires extra work during setup)","none",pcbddc->faster_deluxe,&pcbddc->faster_deluxe,NULL);CHKERRQ(ierr); 71bf3a8328SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_adaptive_userdefined","Use user-defined constraints (should be attached via MatSetNearNullSpace to pmat) in addition to those adaptively generated","none",pcbddc->adaptive_userdefined,&pcbddc->adaptive_userdefined,NULL);CHKERRQ(ierr); 724c6709b3SStefano Zampini ierr = PetscOptionsReal("-pc_bddc_adaptive_threshold","Threshold to be used for adaptive selection of constraints","none",pcbddc->adaptive_threshold,&pcbddc->adaptive_threshold,NULL);CHKERRQ(ierr); 7308122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 7408122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 753301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 76b0c7d250SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_adj","Number of processors where to map the coarse adjacency list","none",pcbddc->coarse_adj_red,&pcbddc->coarse_adj_red,NULL);CHKERRQ(ierr); 77b3afcdbeSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_trick","Apply the benign subspace trick to saddle point problems with discontinuous pressures","none",pcbddc->benign_saddle_point,&pcbddc->benign_saddle_point,NULL);CHKERRQ(ierr); 78b3afcdbeSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->benign_compute_nonetflux,&pcbddc->benign_compute_nonetflux,NULL);CHKERRQ(ierr); 794f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 800c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 810c7d97c5SJed Brown PetscFunctionReturn(0); 820c7d97c5SJed Brown } 830c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 84674ae819SStefano Zampini #undef __FUNCT__ 85906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 861dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 87b9b85e73SStefano Zampini { 88b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 89b9b85e73SStefano Zampini PetscErrorCode ierr; 90b9b85e73SStefano Zampini 91b9b85e73SStefano Zampini PetscFunctionBegin; 92b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 9356282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 94b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 951dd7afcfSStefano Zampini pcbddc->change_interior = interior; 96b9b85e73SStefano Zampini PetscFunctionReturn(0); 97b9b85e73SStefano Zampini } 98b9b85e73SStefano Zampini #undef __FUNCT__ 99906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 100b9b85e73SStefano Zampini /*@ 101906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 102b9b85e73SStefano Zampini 103b9b85e73SStefano Zampini Collective on PC 104b9b85e73SStefano Zampini 105b9b85e73SStefano Zampini Input Parameters: 106b9b85e73SStefano Zampini + pc - the preconditioning context 1071dd7afcfSStefano Zampini . change - the change of basis matrix 1081dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 109b9b85e73SStefano Zampini 110b9b85e73SStefano Zampini Level: intermediate 111b9b85e73SStefano Zampini 112b9b85e73SStefano Zampini Notes: 113b9b85e73SStefano Zampini 114b9b85e73SStefano Zampini .seealso: PCBDDC 115b9b85e73SStefano Zampini @*/ 1161dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 117b9b85e73SStefano Zampini { 118b9b85e73SStefano Zampini PetscErrorCode ierr; 119b9b85e73SStefano Zampini 120b9b85e73SStefano Zampini PetscFunctionBegin; 121b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 122b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 123906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 124906d46d4SStefano Zampini if (pc->mat) { 125906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 126906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 127906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 128906d46d4SStefano Zampini if (rows_c != rows) { 129906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 130906d46d4SStefano Zampini } 131906d46d4SStefano Zampini if (cols_c != cols) { 132906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 133906d46d4SStefano Zampini } 134906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 135906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 136906d46d4SStefano Zampini if (rows_c != rows) { 137906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 138906d46d4SStefano Zampini } 139906d46d4SStefano Zampini if (cols_c != cols) { 140906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 141906d46d4SStefano Zampini } 142906d46d4SStefano Zampini } 1431dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 144b9b85e73SStefano Zampini PetscFunctionReturn(0); 145b9b85e73SStefano Zampini } 146b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 147b9b85e73SStefano Zampini #undef __FUNCT__ 14830368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS_BDDC" 14930368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 15030368db7SStefano Zampini { 15130368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 15256282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 15330368db7SStefano Zampini PetscErrorCode ierr; 15430368db7SStefano Zampini 15530368db7SStefano Zampini PetscFunctionBegin; 15656282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 15756282151SStefano Zampini if (pcbddc->user_primal_vertices) { 15856282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 15956282151SStefano Zampini } 16030368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 16130368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 16230368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 16356282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 16430368db7SStefano Zampini PetscFunctionReturn(0); 16530368db7SStefano Zampini } 16630368db7SStefano Zampini #undef __FUNCT__ 16730368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS" 16830368db7SStefano Zampini /*@ 16930368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 17030368db7SStefano Zampini 17130368db7SStefano Zampini Collective 17230368db7SStefano Zampini 17330368db7SStefano Zampini Input Parameters: 17430368db7SStefano Zampini + pc - the preconditioning context 17530368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 17630368db7SStefano Zampini 17730368db7SStefano Zampini Level: intermediate 17830368db7SStefano Zampini 17930368db7SStefano Zampini Notes: 18030368db7SStefano Zampini Any process can list any global node 18130368db7SStefano Zampini 18230368db7SStefano Zampini .seealso: PCBDDC 18330368db7SStefano Zampini @*/ 18430368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 18530368db7SStefano Zampini { 18630368db7SStefano Zampini PetscErrorCode ierr; 18730368db7SStefano Zampini 18830368db7SStefano Zampini PetscFunctionBegin; 18930368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 19030368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 19130368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 19230368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 19330368db7SStefano Zampini PetscFunctionReturn(0); 19430368db7SStefano Zampini } 19530368db7SStefano Zampini /* -------------------------------------------------------------------------- */ 19630368db7SStefano Zampini #undef __FUNCT__ 197674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 198674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 199674ae819SStefano Zampini { 200674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 20156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 202674ae819SStefano Zampini PetscErrorCode ierr; 2031e6b0712SBarry Smith 204674ae819SStefano Zampini PetscFunctionBegin; 20556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 20656282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 20756282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 20856282151SStefano Zampini } 209674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 21030368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 21130368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 21256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 213674ae819SStefano Zampini PetscFunctionReturn(0); 214674ae819SStefano Zampini } 215674ae819SStefano Zampini #undef __FUNCT__ 216674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 217674ae819SStefano Zampini /*@ 21828509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 219674ae819SStefano Zampini 22017eb1463SStefano Zampini Collective 221674ae819SStefano Zampini 222674ae819SStefano Zampini Input Parameters: 223674ae819SStefano Zampini + pc - the preconditioning context 22417eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 225674ae819SStefano Zampini 226674ae819SStefano Zampini Level: intermediate 227674ae819SStefano Zampini 228674ae819SStefano Zampini Notes: 229674ae819SStefano Zampini 230674ae819SStefano Zampini .seealso: PCBDDC 231674ae819SStefano Zampini @*/ 232674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 233674ae819SStefano Zampini { 234674ae819SStefano Zampini PetscErrorCode ierr; 235674ae819SStefano Zampini 236674ae819SStefano Zampini PetscFunctionBegin; 237674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 238674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 23917eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 240674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 241674ae819SStefano Zampini PetscFunctionReturn(0); 242674ae819SStefano Zampini } 243674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 2440c7d97c5SJed Brown #undef __FUNCT__ 2454fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 2464fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 2474fad6a16SStefano Zampini { 2484fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2494fad6a16SStefano Zampini 2504fad6a16SStefano Zampini PetscFunctionBegin; 2514fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 2524fad6a16SStefano Zampini PetscFunctionReturn(0); 2534fad6a16SStefano Zampini } 2541e6b0712SBarry Smith 2554fad6a16SStefano Zampini #undef __FUNCT__ 2564fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 2574fad6a16SStefano Zampini /*@ 25828509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 2594fad6a16SStefano Zampini 2604fad6a16SStefano Zampini Logically collective on PC 2614fad6a16SStefano Zampini 2624fad6a16SStefano Zampini Input Parameters: 2634fad6a16SStefano Zampini + pc - the preconditioning context 26428509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 2654fad6a16SStefano Zampini 2660f202f7eSStefano Zampini Options Database Keys: 2670f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 2684fad6a16SStefano Zampini 2694fad6a16SStefano Zampini Level: intermediate 2704fad6a16SStefano Zampini 2714fad6a16SStefano Zampini Notes: 2720f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 2734fad6a16SStefano Zampini 2740f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2754fad6a16SStefano Zampini @*/ 2764fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2774fad6a16SStefano Zampini { 2784fad6a16SStefano Zampini PetscErrorCode ierr; 2794fad6a16SStefano Zampini 2804fad6a16SStefano Zampini PetscFunctionBegin; 2814fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2822b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2834fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2844fad6a16SStefano Zampini PetscFunctionReturn(0); 2854fad6a16SStefano Zampini } 2862b510759SStefano Zampini 287b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2882b510759SStefano Zampini #undef __FUNCT__ 289b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 290b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 291b8ffe317SStefano Zampini { 292b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 293b8ffe317SStefano Zampini 294b8ffe317SStefano Zampini PetscFunctionBegin; 29585c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 296b8ffe317SStefano Zampini PetscFunctionReturn(0); 297b8ffe317SStefano Zampini } 298b8ffe317SStefano Zampini 299b8ffe317SStefano Zampini #undef __FUNCT__ 300b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 301b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 3022b510759SStefano Zampini { 3032b510759SStefano Zampini PetscErrorCode ierr; 3042b510759SStefano Zampini 3052b510759SStefano Zampini PetscFunctionBegin; 3062b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 307b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 308b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 3092b510759SStefano Zampini PetscFunctionReturn(0); 3102b510759SStefano Zampini } 3111e6b0712SBarry Smith 3124fad6a16SStefano Zampini #undef __FUNCT__ 3132b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 3142b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 3154fad6a16SStefano Zampini { 3164fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3174fad6a16SStefano Zampini 3184fad6a16SStefano Zampini PetscFunctionBegin; 3192b510759SStefano Zampini pcbddc->current_level = level; 3204fad6a16SStefano Zampini PetscFunctionReturn(0); 3214fad6a16SStefano Zampini } 3221e6b0712SBarry Smith 3234fad6a16SStefano Zampini #undef __FUNCT__ 324b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 325b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 326b8ffe317SStefano Zampini { 327b8ffe317SStefano Zampini PetscErrorCode ierr; 328b8ffe317SStefano Zampini 329b8ffe317SStefano Zampini PetscFunctionBegin; 330b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 331b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 332b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 333b8ffe317SStefano Zampini PetscFunctionReturn(0); 334b8ffe317SStefano Zampini } 335b8ffe317SStefano Zampini 336b8ffe317SStefano Zampini #undef __FUNCT__ 3372b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 3382b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 3392b510759SStefano Zampini { 3402b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3412b510759SStefano Zampini 3422b510759SStefano Zampini PetscFunctionBegin; 3432b510759SStefano Zampini pcbddc->max_levels = levels; 3442b510759SStefano Zampini PetscFunctionReturn(0); 3452b510759SStefano Zampini } 3462b510759SStefano Zampini 3472b510759SStefano Zampini #undef __FUNCT__ 3482b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 3494fad6a16SStefano Zampini /*@ 35028509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 3514fad6a16SStefano Zampini 3524fad6a16SStefano Zampini Logically collective on PC 3534fad6a16SStefano Zampini 3544fad6a16SStefano Zampini Input Parameters: 3554fad6a16SStefano Zampini + pc - the preconditioning context 35628509bceSStefano Zampini - levels - the maximum number of levels (max 9) 3574fad6a16SStefano Zampini 3580f202f7eSStefano Zampini Options Database Keys: 3590f202f7eSStefano Zampini . -pc_bddc_levels 3604fad6a16SStefano Zampini 3614fad6a16SStefano Zampini Level: intermediate 3624fad6a16SStefano Zampini 3634fad6a16SStefano Zampini Notes: 3640f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 3654fad6a16SStefano Zampini 3660f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 3674fad6a16SStefano Zampini @*/ 3682b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 3694fad6a16SStefano Zampini { 3704fad6a16SStefano Zampini PetscErrorCode ierr; 3714fad6a16SStefano Zampini 3724fad6a16SStefano Zampini PetscFunctionBegin; 3734fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 3742b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 375312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3762b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3774fad6a16SStefano Zampini PetscFunctionReturn(0); 3784fad6a16SStefano Zampini } 3794fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3801e6b0712SBarry Smith 3814fad6a16SStefano Zampini #undef __FUNCT__ 3820bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 3830bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 3840bdf917eSStefano Zampini { 3850bdf917eSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3860bdf917eSStefano Zampini PetscErrorCode ierr; 3870bdf917eSStefano Zampini 3880bdf917eSStefano Zampini PetscFunctionBegin; 3890bdf917eSStefano Zampini ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 3900bdf917eSStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 3910bdf917eSStefano Zampini pcbddc->NullSpace = NullSpace; 3920bdf917eSStefano Zampini PetscFunctionReturn(0); 3930bdf917eSStefano Zampini } 3941e6b0712SBarry Smith 3950bdf917eSStefano Zampini #undef __FUNCT__ 3960bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace" 3970bdf917eSStefano Zampini /*@ 39828509bceSStefano Zampini PCBDDCSetNullSpace - Set nullspace for BDDC operator 3990bdf917eSStefano Zampini 4000bdf917eSStefano Zampini Logically collective on PC and MatNullSpace 4010bdf917eSStefano Zampini 4020bdf917eSStefano Zampini Input Parameters: 4030bdf917eSStefano Zampini + pc - the preconditioning context 40428509bceSStefano Zampini - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 4050bdf917eSStefano Zampini 4060bdf917eSStefano Zampini Level: intermediate 4070bdf917eSStefano Zampini 4080bdf917eSStefano Zampini Notes: 4090bdf917eSStefano Zampini 4100bdf917eSStefano Zampini .seealso: PCBDDC 4110bdf917eSStefano Zampini @*/ 4120bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 4130bdf917eSStefano Zampini { 4140bdf917eSStefano Zampini PetscErrorCode ierr; 4150bdf917eSStefano Zampini 4160bdf917eSStefano Zampini PetscFunctionBegin; 4170bdf917eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 418674ae819SStefano Zampini PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 4192b510759SStefano Zampini PetscCheckSameComm(pc,1,NullSpace,2); 4200bdf917eSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 4210bdf917eSStefano Zampini PetscFunctionReturn(0); 4220bdf917eSStefano Zampini } 4230bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */ 4241e6b0712SBarry Smith 4250bdf917eSStefano Zampini #undef __FUNCT__ 4263b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 4273b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 4283b03a366Sstefano_zampini { 4293b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 43056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 4313b03a366Sstefano_zampini PetscErrorCode ierr; 4323b03a366Sstefano_zampini 4333b03a366Sstefano_zampini PetscFunctionBegin; 43456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 43556282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 43656282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 43756282151SStefano Zampini } 438785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 439785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4403b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 44136e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 44256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 4433b03a366Sstefano_zampini PetscFunctionReturn(0); 4443b03a366Sstefano_zampini } 4451e6b0712SBarry Smith 4463b03a366Sstefano_zampini #undef __FUNCT__ 4473b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 4483b03a366Sstefano_zampini /*@ 44928509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 4503b03a366Sstefano_zampini 451785d1243SStefano Zampini Collective 4523b03a366Sstefano_zampini 4533b03a366Sstefano_zampini Input Parameters: 4543b03a366Sstefano_zampini + pc - the preconditioning context 455785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 4563b03a366Sstefano_zampini 4573b03a366Sstefano_zampini Level: intermediate 4583b03a366Sstefano_zampini 4590f202f7eSStefano Zampini Notes: 4600f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 4613b03a366Sstefano_zampini 4620f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 4633b03a366Sstefano_zampini @*/ 4643b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 4653b03a366Sstefano_zampini { 4663b03a366Sstefano_zampini PetscErrorCode ierr; 4673b03a366Sstefano_zampini 4683b03a366Sstefano_zampini PetscFunctionBegin; 4693b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 470674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 471785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4723b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4733b03a366Sstefano_zampini PetscFunctionReturn(0); 4743b03a366Sstefano_zampini } 4753b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4761e6b0712SBarry Smith 4773b03a366Sstefano_zampini #undef __FUNCT__ 47882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 47982ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4800c7d97c5SJed Brown { 4810c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 48256282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 4830c7d97c5SJed Brown PetscErrorCode ierr; 4840c7d97c5SJed Brown 4850c7d97c5SJed Brown PetscFunctionBegin; 48656282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 48756282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 48856282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 48956282151SStefano Zampini } 490785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 491785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4920c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 493785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 49456282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 4950c7d97c5SJed Brown PetscFunctionReturn(0); 4960c7d97c5SJed Brown } 4970c7d97c5SJed Brown 4980c7d97c5SJed Brown #undef __FUNCT__ 49982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 5009c0446d6SStefano Zampini /*@ 50182ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 5029c0446d6SStefano Zampini 503785d1243SStefano Zampini Collective 50453cdbc3dSStefano Zampini 50553cdbc3dSStefano Zampini Input Parameters: 50653cdbc3dSStefano Zampini + pc - the preconditioning context 50782ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 50853cdbc3dSStefano Zampini 50953cdbc3dSStefano Zampini Level: intermediate 51053cdbc3dSStefano Zampini 5119c0446d6SStefano Zampini Notes: 51253cdbc3dSStefano Zampini 5130f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 51453cdbc3dSStefano Zampini @*/ 51582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 5160c7d97c5SJed Brown { 5170c7d97c5SJed Brown PetscErrorCode ierr; 5180c7d97c5SJed Brown 5190c7d97c5SJed Brown PetscFunctionBegin; 5200c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5210c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 52282ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 52382ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 5240c7d97c5SJed Brown PetscFunctionReturn(0); 5250c7d97c5SJed Brown } 5260c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 5270c7d97c5SJed Brown 5280c7d97c5SJed Brown #undef __FUNCT__ 5290c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 53053cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 5310c7d97c5SJed Brown { 5320c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 53356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 53453cdbc3dSStefano Zampini PetscErrorCode ierr; 5350c7d97c5SJed Brown 5360c7d97c5SJed Brown PetscFunctionBegin; 53756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 53856282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 53956282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 54056282151SStefano Zampini } 541785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 542785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 54353cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 54436e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 54556282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 5460c7d97c5SJed Brown PetscFunctionReturn(0); 5470c7d97c5SJed Brown } 5481e6b0712SBarry Smith 5490c7d97c5SJed Brown #undef __FUNCT__ 5500c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 55157527edcSJed Brown /*@ 55228509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 55357527edcSJed Brown 554785d1243SStefano Zampini Collective 55557527edcSJed Brown 55657527edcSJed Brown Input Parameters: 55757527edcSJed Brown + pc - the preconditioning context 558785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 55957527edcSJed Brown 56057527edcSJed Brown Level: intermediate 56157527edcSJed Brown 5620f202f7eSStefano Zampini Notes: 5630f202f7eSStefano Zampini Any process can list any global node 56457527edcSJed Brown 5650f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 56657527edcSJed Brown @*/ 56753cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 5680c7d97c5SJed Brown { 5690c7d97c5SJed Brown PetscErrorCode ierr; 5700c7d97c5SJed Brown 5710c7d97c5SJed Brown PetscFunctionBegin; 5720c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 573674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 574785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 57553cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 57653cdbc3dSStefano Zampini PetscFunctionReturn(0); 57753cdbc3dSStefano Zampini } 57853cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5791e6b0712SBarry Smith 58053cdbc3dSStefano Zampini #undef __FUNCT__ 58182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 58282ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5830c7d97c5SJed Brown { 5840c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 58556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 5860c7d97c5SJed Brown PetscErrorCode ierr; 5870c7d97c5SJed Brown 5880c7d97c5SJed Brown PetscFunctionBegin; 58956282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 59056282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 59156282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 59256282151SStefano Zampini } 593785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 594785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5950c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 596785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 59756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 5980c7d97c5SJed Brown PetscFunctionReturn(0); 5990c7d97c5SJed Brown } 6000c7d97c5SJed Brown 6010c7d97c5SJed Brown #undef __FUNCT__ 60282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 6030c7d97c5SJed Brown /*@ 60482ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 6050c7d97c5SJed Brown 606785d1243SStefano Zampini Collective 6070c7d97c5SJed Brown 6080c7d97c5SJed Brown Input Parameters: 6090c7d97c5SJed Brown + pc - the preconditioning context 61082ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 6110c7d97c5SJed Brown 6120c7d97c5SJed Brown Level: intermediate 6130c7d97c5SJed Brown 6140c7d97c5SJed Brown Notes: 6150c7d97c5SJed Brown 6160f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 6170c7d97c5SJed Brown @*/ 61882ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 6190c7d97c5SJed Brown { 6200c7d97c5SJed Brown PetscErrorCode ierr; 6210c7d97c5SJed Brown 6220c7d97c5SJed Brown PetscFunctionBegin; 6230c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6240c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 62582ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 62682ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 62753cdbc3dSStefano Zampini PetscFunctionReturn(0); 62853cdbc3dSStefano Zampini } 62953cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 63053cdbc3dSStefano Zampini 63153cdbc3dSStefano Zampini #undef __FUNCT__ 632da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 633da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 634da1bb401SStefano Zampini { 635da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 636da1bb401SStefano Zampini 637da1bb401SStefano Zampini PetscFunctionBegin; 638da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 639da1bb401SStefano Zampini PetscFunctionReturn(0); 640da1bb401SStefano Zampini } 6411e6b0712SBarry Smith 642da1bb401SStefano Zampini #undef __FUNCT__ 643da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 644da1bb401SStefano Zampini /*@ 645785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 646da1bb401SStefano Zampini 647785d1243SStefano Zampini Collective 648785d1243SStefano Zampini 649785d1243SStefano Zampini Input Parameters: 650785d1243SStefano Zampini . pc - the preconditioning context 651785d1243SStefano Zampini 652785d1243SStefano Zampini Output Parameters: 653785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 654785d1243SStefano Zampini 655785d1243SStefano Zampini Level: intermediate 656785d1243SStefano Zampini 6570f202f7eSStefano Zampini Notes: 6580f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 659785d1243SStefano Zampini 660785d1243SStefano Zampini .seealso: PCBDDC 661785d1243SStefano Zampini @*/ 662785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 663785d1243SStefano Zampini { 664785d1243SStefano Zampini PetscErrorCode ierr; 665785d1243SStefano Zampini 666785d1243SStefano Zampini PetscFunctionBegin; 667785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 668785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 669785d1243SStefano Zampini PetscFunctionReturn(0); 670785d1243SStefano Zampini } 671785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 672785d1243SStefano Zampini 673785d1243SStefano Zampini #undef __FUNCT__ 674785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 675785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 676785d1243SStefano Zampini { 677785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 678785d1243SStefano Zampini 679785d1243SStefano Zampini PetscFunctionBegin; 680785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 681785d1243SStefano Zampini PetscFunctionReturn(0); 682785d1243SStefano Zampini } 683785d1243SStefano Zampini 684785d1243SStefano Zampini #undef __FUNCT__ 68582ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 686da1bb401SStefano Zampini /*@ 68782ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 688da1bb401SStefano Zampini 689785d1243SStefano Zampini Collective 690da1bb401SStefano Zampini 691da1bb401SStefano Zampini Input Parameters: 69228509bceSStefano Zampini . pc - the preconditioning context 693da1bb401SStefano Zampini 694da1bb401SStefano Zampini Output Parameters: 69528509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 696da1bb401SStefano Zampini 697da1bb401SStefano Zampini Level: intermediate 698da1bb401SStefano Zampini 699da1bb401SStefano Zampini Notes: 7000f202f7eSStefano Zampini The IS returned could be the same passed in earlier by the user (if provided with PCBDDCSetDirichletBoundariesLocal) or a global-to-local map of the global IS (if provided with PCBDDCSetDirichletBoundaries). 7010f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 702da1bb401SStefano Zampini 703da1bb401SStefano Zampini .seealso: PCBDDC 704da1bb401SStefano Zampini @*/ 70582ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 706da1bb401SStefano Zampini { 707da1bb401SStefano Zampini PetscErrorCode ierr; 708da1bb401SStefano Zampini 709da1bb401SStefano Zampini PetscFunctionBegin; 710da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 71182ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 712da1bb401SStefano Zampini PetscFunctionReturn(0); 713da1bb401SStefano Zampini } 714da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 7151e6b0712SBarry Smith 716da1bb401SStefano Zampini #undef __FUNCT__ 71753cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 71853cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 71953cdbc3dSStefano Zampini { 72053cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 72153cdbc3dSStefano Zampini 72253cdbc3dSStefano Zampini PetscFunctionBegin; 72353cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 72453cdbc3dSStefano Zampini PetscFunctionReturn(0); 72553cdbc3dSStefano Zampini } 7261e6b0712SBarry Smith 72753cdbc3dSStefano Zampini #undef __FUNCT__ 72853cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 72953cdbc3dSStefano Zampini /*@ 730785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 73153cdbc3dSStefano Zampini 732785d1243SStefano Zampini Collective 733785d1243SStefano Zampini 734785d1243SStefano Zampini Input Parameters: 735785d1243SStefano Zampini . pc - the preconditioning context 736785d1243SStefano Zampini 737785d1243SStefano Zampini Output Parameters: 738785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 739785d1243SStefano Zampini 740785d1243SStefano Zampini Level: intermediate 741785d1243SStefano Zampini 7420f202f7eSStefano Zampini Notes: 7430f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 744785d1243SStefano Zampini 745785d1243SStefano Zampini .seealso: PCBDDC 746785d1243SStefano Zampini @*/ 747785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 748785d1243SStefano Zampini { 749785d1243SStefano Zampini PetscErrorCode ierr; 750785d1243SStefano Zampini 751785d1243SStefano Zampini PetscFunctionBegin; 752785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 753785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 754785d1243SStefano Zampini PetscFunctionReturn(0); 755785d1243SStefano Zampini } 756785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 757785d1243SStefano Zampini 758785d1243SStefano Zampini #undef __FUNCT__ 759785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 760785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 761785d1243SStefano Zampini { 762785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 763785d1243SStefano Zampini 764785d1243SStefano Zampini PetscFunctionBegin; 765785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 766785d1243SStefano Zampini PetscFunctionReturn(0); 767785d1243SStefano Zampini } 768785d1243SStefano Zampini 769785d1243SStefano Zampini #undef __FUNCT__ 77082ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 77153cdbc3dSStefano Zampini /*@ 77282ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 77353cdbc3dSStefano Zampini 774785d1243SStefano Zampini Collective 77553cdbc3dSStefano Zampini 77653cdbc3dSStefano Zampini Input Parameters: 77728509bceSStefano Zampini . pc - the preconditioning context 77853cdbc3dSStefano Zampini 77953cdbc3dSStefano Zampini Output Parameters: 78028509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 78153cdbc3dSStefano Zampini 78253cdbc3dSStefano Zampini Level: intermediate 78353cdbc3dSStefano Zampini 78453cdbc3dSStefano Zampini Notes: 7850f202f7eSStefano Zampini The IS returned could be the same passed in earlier by the user (if provided with PCBDDCSetNeumannBoundariesLocal) or a global-to-local map of the global IS (if provided with PCBDDCSetNeumannBoundaries). 7860f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 78753cdbc3dSStefano Zampini 78853cdbc3dSStefano Zampini .seealso: PCBDDC 78953cdbc3dSStefano Zampini @*/ 79082ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 79153cdbc3dSStefano Zampini { 79253cdbc3dSStefano Zampini PetscErrorCode ierr; 79353cdbc3dSStefano Zampini 79453cdbc3dSStefano Zampini PetscFunctionBegin; 79553cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 79682ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7970c7d97c5SJed Brown PetscFunctionReturn(0); 7980c7d97c5SJed Brown } 79936e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 8001e6b0712SBarry Smith 80136e030ebSStefano Zampini #undef __FUNCT__ 802da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 8031a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 80436e030ebSStefano Zampini { 80536e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 806da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 80756282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 808da1bb401SStefano Zampini PetscErrorCode ierr; 80936e030ebSStefano Zampini 81036e030ebSStefano Zampini PetscFunctionBegin; 81156282151SStefano Zampini if (mat_graph->nvtxs_csr == nvtxs) { 81256282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 81356282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 81456282151SStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,nvtxs*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 81556282151SStefano Zampini } 81656282151SStefano Zampini } 81756282151SStefano Zampini if (!same_data) { 818674ae819SStefano Zampini /* free old CSR */ 819674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 820674ae819SStefano Zampini /* get CSR into graph structure */ 821da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 822854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 823785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 824674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 825674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 826da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 8271a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 8281a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 829674ae819SStefano Zampini } 830575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 83156282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 83256282151SStefano Zampini } 83336e030ebSStefano Zampini PetscFunctionReturn(0); 83436e030ebSStefano Zampini } 8351e6b0712SBarry Smith 83636e030ebSStefano Zampini #undef __FUNCT__ 837da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 83836e030ebSStefano Zampini /*@ 8390f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 84036e030ebSStefano Zampini 84136e030ebSStefano Zampini Not collective 84236e030ebSStefano Zampini 84336e030ebSStefano Zampini Input Parameters: 84436e030ebSStefano Zampini + pc - the preconditioning context 8450f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 84628509bceSStefano Zampini . xadj, adjncy - the CSR graph 84728509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 84836e030ebSStefano Zampini 84936e030ebSStefano Zampini Level: intermediate 85036e030ebSStefano Zampini 85136e030ebSStefano Zampini Notes: 85236e030ebSStefano Zampini 85328509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 85436e030ebSStefano Zampini @*/ 8551a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 85636e030ebSStefano Zampini { 857575ad6abSStefano Zampini void (*f)(void) = 0; 85836e030ebSStefano Zampini PetscErrorCode ierr; 85936e030ebSStefano Zampini 86036e030ebSStefano Zampini PetscFunctionBegin; 86136e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 862674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 863d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 864674ae819SStefano Zampini if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) { 865674ae819SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__); 866da1bb401SStefano Zampini } 86736e030ebSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 868575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 869575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 870575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 871575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 872575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 87336e030ebSStefano Zampini } 87436e030ebSStefano Zampini PetscFunctionReturn(0); 87536e030ebSStefano Zampini } 8769c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 8771e6b0712SBarry Smith 8789c0446d6SStefano Zampini #undef __FUNCT__ 87963602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 88063602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 88163602bcaSStefano Zampini { 88263602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 88363602bcaSStefano Zampini PetscInt i; 88456282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 88563602bcaSStefano Zampini PetscErrorCode ierr; 88663602bcaSStefano Zampini 88763602bcaSStefano Zampini PetscFunctionBegin; 88856282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 88956282151SStefano Zampini for (i=0;i<n_is;i++) { 89056282151SStefano Zampini PetscBool isequalt; 89156282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 89256282151SStefano Zampini if (!isequalt) break; 89356282151SStefano Zampini } 89456282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 89556282151SStefano Zampini } 89656282151SStefano Zampini for (i=0;i<n_is;i++) { 89756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 89856282151SStefano Zampini } 89963602bcaSStefano Zampini /* Destroy ISes if they were already set */ 90063602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 90163602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 90263602bcaSStefano Zampini } 90363602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 90463602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 90563602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 90663602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 90763602bcaSStefano Zampini } 90863602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 90963602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 91063602bcaSStefano Zampini /* allocate space then set */ 911d02579f5SStefano Zampini if (n_is) { 912d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 913d02579f5SStefano Zampini } 91463602bcaSStefano Zampini for (i=0;i<n_is;i++) { 91563602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 91663602bcaSStefano Zampini } 91763602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 91863602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 91956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 92063602bcaSStefano Zampini PetscFunctionReturn(0); 92163602bcaSStefano Zampini } 92263602bcaSStefano Zampini 92363602bcaSStefano Zampini #undef __FUNCT__ 92463602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 92563602bcaSStefano Zampini /*@ 92663602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 92763602bcaSStefano Zampini 92863602bcaSStefano Zampini Collective 92963602bcaSStefano Zampini 93063602bcaSStefano Zampini Input Parameters: 93163602bcaSStefano Zampini + pc - the preconditioning context 9320f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9330f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 93463602bcaSStefano Zampini 93563602bcaSStefano Zampini Level: intermediate 93663602bcaSStefano Zampini 9370f202f7eSStefano Zampini Notes: 9380f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 93963602bcaSStefano Zampini 94063602bcaSStefano Zampini .seealso: PCBDDC 94163602bcaSStefano Zampini @*/ 94263602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 94363602bcaSStefano Zampini { 94463602bcaSStefano Zampini PetscInt i; 94563602bcaSStefano Zampini PetscErrorCode ierr; 94663602bcaSStefano Zampini 94763602bcaSStefano Zampini PetscFunctionBegin; 94863602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 94963602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 95063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 95163602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 95263602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 95363602bcaSStefano Zampini } 954e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 95563602bcaSStefano Zampini PetscFunctionReturn(0); 95663602bcaSStefano Zampini } 95763602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 95863602bcaSStefano Zampini 95963602bcaSStefano Zampini #undef __FUNCT__ 9609c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 9619c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 9629c0446d6SStefano Zampini { 9639c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 9649c0446d6SStefano Zampini PetscInt i; 96556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 9669c0446d6SStefano Zampini PetscErrorCode ierr; 9679c0446d6SStefano Zampini 9689c0446d6SStefano Zampini PetscFunctionBegin; 96956282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 97056282151SStefano Zampini for (i=0;i<n_is;i++) { 97156282151SStefano Zampini PetscBool isequalt; 97256282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 97356282151SStefano Zampini if (!isequalt) break; 97456282151SStefano Zampini } 97556282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 97656282151SStefano Zampini } 97756282151SStefano Zampini for (i=0;i<n_is;i++) { 97856282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 97956282151SStefano Zampini } 980da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 9819c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 9829c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 9839c0446d6SStefano Zampini } 984d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 98563602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 98663602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 98763602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 98863602bcaSStefano Zampini } 98963602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 99063602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 991da1bb401SStefano Zampini /* allocate space then set */ 992d02579f5SStefano Zampini if (n_is) { 993785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 994d02579f5SStefano Zampini } 9959c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 996da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 9979c0446d6SStefano Zampini } 9989c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 99963602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 100056282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 10019c0446d6SStefano Zampini PetscFunctionReturn(0); 10029c0446d6SStefano Zampini } 10031e6b0712SBarry Smith 10049c0446d6SStefano Zampini #undef __FUNCT__ 10059c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 10069c0446d6SStefano Zampini /*@ 100763602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 10089c0446d6SStefano Zampini 100963602bcaSStefano Zampini Collective 10109c0446d6SStefano Zampini 10119c0446d6SStefano Zampini Input Parameters: 10129c0446d6SStefano Zampini + pc - the preconditioning context 10130f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10140f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 10159c0446d6SStefano Zampini 10169c0446d6SStefano Zampini Level: intermediate 10179c0446d6SStefano Zampini 10180f202f7eSStefano Zampini Notes: 10190f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 10209c0446d6SStefano Zampini 10219c0446d6SStefano Zampini .seealso: PCBDDC 10229c0446d6SStefano Zampini @*/ 10239c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 10249c0446d6SStefano Zampini { 10252b510759SStefano Zampini PetscInt i; 10269c0446d6SStefano Zampini PetscErrorCode ierr; 10279c0446d6SStefano Zampini 10289c0446d6SStefano Zampini PetscFunctionBegin; 10299c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 103063602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 10312b510759SStefano Zampini for (i=0;i<n_is;i++) { 103263602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 103363602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 10342b510759SStefano Zampini } 10359c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 10369c0446d6SStefano Zampini PetscFunctionReturn(0); 10379c0446d6SStefano Zampini } 1038906d46d4SStefano Zampini 1039da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1040534831adSStefano Zampini #undef __FUNCT__ 1041534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 1042534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1043534831adSStefano Zampini /* 1044534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1045534831adSStefano Zampini guess if a transformation of basis approach has been selected. 10469c0446d6SStefano Zampini 1047534831adSStefano Zampini Input Parameter: 1048534831adSStefano Zampini + pc - the preconditioner contex 1049534831adSStefano Zampini 1050534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1051534831adSStefano Zampini 1052534831adSStefano Zampini Notes: 1053534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1054534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1055534831adSStefano Zampini */ 1056534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1057534831adSStefano Zampini { 1058534831adSStefano Zampini PetscErrorCode ierr; 1059534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1060534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 10613972b0daSStefano Zampini Vec used_vec; 10628d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 1063a3df083aSStefano Zampini PetscBool iscg = PETSC_FALSE; 1064534831adSStefano Zampini 1065534831adSStefano Zampini PetscFunctionBegin; 106685c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 106785c4d303SStefano Zampini if (ksp) { 106885c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 1069*16909a7fSStefano Zampini if (!iscg || pcbddc->switch_static) { 107085c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 107185c4d303SStefano Zampini } 107285c4d303SStefano Zampini } 107385c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 107462a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 107562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 107662a6ff1dSStefano Zampini } 107762a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 107862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 107962a6ff1dSStefano Zampini } 10808d00608fSStefano Zampini 10811dd7afcfSStefano Zampini ierr = VecSet(pcbddc->temp_solution,0.);CHKERRQ(ierr); 10823972b0daSStefano Zampini if (x) { 10833972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 10843972b0daSStefano Zampini used_vec = x; 10858d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 10863972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 10873972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 10883972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 10893972b0daSStefano Zampini } 10908efcfb23SStefano Zampini 10918efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 10923972b0daSStefano Zampini if (ksp) { 1093a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 10948efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 10958efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 10963972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 10973972b0daSStefano Zampini } 10983972b0daSStefano Zampini } 10993308cffdSStefano Zampini 11008d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11013972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 1102a07ea27aSStefano Zampini if (rhs) { 11033975b054SStefano Zampini IS dirIS = NULL; 11043975b054SStefano Zampini 1105a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 11063975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 11073975b054SStefano Zampini if (dirIS) { 1108906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1109785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 11102b095fd8SStefano Zampini PetscScalar *array_x; 11112b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1112785d1243SStefano Zampini 11133972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 11143972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1115e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1116e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1117e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1118e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 111982ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 11203972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 11212b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 11223972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 11232fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 11243972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 11252b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 11263972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1127e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1128e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 11298d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 11301b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 11318efcfb23SStefano Zampini } 1132a07ea27aSStefano Zampini } 1133b76ba322SStefano Zampini 11348efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 11358d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 11368d00608fSStefano Zampini /* store the original rhs */ 11378d00608fSStefano Zampini if (copy_rhs) { 11388d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11398d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 11408d00608fSStefano Zampini } 11418d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 11423972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 11430369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,rhs,rhs);CHKERRQ(ierr); 11443972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 11458efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 11467acc28cbSStefano Zampini if (ksp) { 11477acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 11487acc28cbSStefano Zampini } 11493308cffdSStefano Zampini } 11508efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1151b76ba322SStefano Zampini 11521dd7afcfSStefano Zampini /* compute initial vector in benign space (TODO: what about FETI-DP?)*/ 11530369aaf7SStefano Zampini if (rhs && pcbddc->benign_saddle_point) { 11540369aaf7SStefano Zampini /* compute u^*_h as in Xuemin Tu's thesis (see Section 4.8.1) */ 11550369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 11560369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 11570369aaf7SStefano Zampini } 11584fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 11591dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 11604fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 11611dd7afcfSStefano Zampini /* TODO: what about Stokes? */ 11621dd7afcfSStefano Zampini #if 0 11631dd7afcfSStefano Zampini ierr = VecNorm(pcbddc->benign_vec,NORM_INFINITY,&norm);CHKERRQ(ierr); 11641dd7afcfSStefano Zampini if (norm < PETSC_SMALL) benign_correction_is_zero = PETSC_TRUE; 11651dd7afcfSStefano Zampini #endif 116692e3dcfbSStefano Zampini } 116792e3dcfbSStefano Zampini 11680369aaf7SStefano Zampini /* remove non-benign solution from the rhs */ 11690369aaf7SStefano Zampini if (pcbddc->benign_saddle_point) { 11700369aaf7SStefano Zampini /* store the original rhs */ 11710369aaf7SStefano Zampini if (copy_rhs) { 11720369aaf7SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11730369aaf7SStefano Zampini copy_rhs = PETSC_FALSE; 11740369aaf7SStefano Zampini } 11751dd7afcfSStefano Zampini /* this code is disabled, until I test against incompressible Stokes */ 11761dd7afcfSStefano Zampini #if 0 117751694757SStefano Zampini if (benign_correction_is_zero) { /* still need to understand why it works great */ 117851694757SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 117951694757SStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 118051694757SStefano Zampini } 11811dd7afcfSStefano Zampini #endif 11820369aaf7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 11830369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 11840369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 11851dd7afcfSStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 11860369aaf7SStefano Zampini } 11870369aaf7SStefano Zampini 11880369aaf7SStefano Zampini /* set initial guess if using PCG */ 11890369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 11900369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 11911dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 11921dd7afcfSStefano Zampini if (pcbddc->benign_saddle_point) { /* we have already saved the changed rhs */ 11931dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 11941dd7afcfSStefano Zampini } else { 11951dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 11961dd7afcfSStefano Zampini } 11971dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 11981dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 11991dd7afcfSStefano Zampini } else { 12000369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12010369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12021dd7afcfSStefano Zampini } 12030369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 12041dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 12051dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 12061dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12071dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12081dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 12091dd7afcfSStefano Zampini } else { 12100369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12110369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12121dd7afcfSStefano Zampini } 12130369aaf7SStefano Zampini if (ksp) { 12140369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 12150369aaf7SStefano Zampini } 12160369aaf7SStefano Zampini } 1217534831adSStefano Zampini PetscFunctionReturn(0); 1218534831adSStefano Zampini } 1219906d46d4SStefano Zampini 1220534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1221534831adSStefano Zampini #undef __FUNCT__ 1222534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1223534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1224534831adSStefano Zampini /* 1225534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1226534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1227534831adSStefano Zampini 1228534831adSStefano Zampini Input Parameter: 1229534831adSStefano Zampini + pc - the preconditioner contex 1230534831adSStefano Zampini 1231534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1232534831adSStefano Zampini 1233534831adSStefano Zampini Notes: 1234534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1235534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1236534831adSStefano Zampini */ 1237534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1238534831adSStefano Zampini { 1239534831adSStefano Zampini PetscErrorCode ierr; 1240534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1241534831adSStefano Zampini 1242534831adSStefano Zampini PetscFunctionBegin; 12433972b0daSStefano Zampini /* add solution removed in presolve */ 12446bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 12453425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 12463425bc38SStefano Zampini } 1247fb223d50SStefano Zampini /* restore rhs to its original state */ 12488d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1249fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1250fb223d50SStefano Zampini } 12518d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12528efcfb23SStefano Zampini /* restore ksp guess state */ 12538efcfb23SStefano Zampini if (ksp) { 12548efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12558efcfb23SStefano Zampini } 1256534831adSStefano Zampini PetscFunctionReturn(0); 1257534831adSStefano Zampini } 1258534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 125953cdbc3dSStefano Zampini #undef __FUNCT__ 126053cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 12610c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 12620c7d97c5SJed Brown /* 12630c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 12640c7d97c5SJed Brown by setting data structures and options. 12650c7d97c5SJed Brown 12660c7d97c5SJed Brown Input Parameter: 126753cdbc3dSStefano Zampini + pc - the preconditioner context 12680c7d97c5SJed Brown 12690c7d97c5SJed Brown Application Interface Routine: PCSetUp() 12700c7d97c5SJed Brown 12710c7d97c5SJed Brown Notes: 12720c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 12730c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 12740c7d97c5SJed Brown */ 127553cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 12760c7d97c5SJed Brown { 12770c7d97c5SJed Brown PetscErrorCode ierr; 12780c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 12795e8657edSStefano Zampini Mat_IS* matis; 128008122e43SStefano Zampini MatNullSpace nearnullspace; 1281c9ed8603SStefano Zampini IS zerodiag = NULL; 128291e8d312SStefano Zampini PetscInt nrows,ncols; 128308122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 12848de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 12855e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 12860c7d97c5SJed Brown 12870c7d97c5SJed Brown PetscFunctionBegin; 12885e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 12895e8657edSStefano Zampini if (!ismatis) { 12905e8657edSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 12915e8657edSStefano Zampini } 129291e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 129391e8d312SStefano Zampini if (nrows != ncols) { 129491e8d312SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 129591e8d312SStefano Zampini } 12965e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1297f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 12983b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1299fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1300f4ddd8eeSStefano Zampini /* split work */ 1301674ae819SStefano Zampini if (pc->setupcalled) { 1302d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1303674ae819SStefano Zampini computetopography = PETSC_FALSE; 1304674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1305674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1306674ae819SStefano Zampini computetopography = PETSC_TRUE; 1307674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1308674ae819SStefano Zampini } 1309674ae819SStefano Zampini } else { 1310674ae819SStefano Zampini computetopography = PETSC_TRUE; 1311674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1312674ae819SStefano Zampini } 1313fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1314fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1315fb180af4SStefano Zampini } 131656282151SStefano Zampini pcbddc->recompute_topography = computetopography; 13178de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1318b087196eSStefano Zampini 1319b087196eSStefano Zampini /* check parameters' compatibility */ 13205a95e1ceSStefano Zampini if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) { 13215a95e1ceSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 13225a95e1ceSStefano Zampini } 13235a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1324bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1325862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1326862806e4SStefano Zampini 13275a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 13286816873aSStefano Zampini if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) { 13296816873aSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute faster deluxe if adaptivity and change of basis are both requested. Rerun with -pc_bddc_deluxe_faster false"); 13306816873aSStefano Zampini } 13312d3346b4SStefano Zampini 1332*16909a7fSStefano Zampini if (pcbddc->switch_static) { 1333*16909a7fSStefano Zampini PetscBool ismatis; 1334*16909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 1335*16909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 1336*16909a7fSStefano Zampini } 1337*16909a7fSStefano Zampini 1338f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 133970cf5478SStefano Zampini if (pcbddc->dbg_flag) { 134070cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 134158a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 13421575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1343f4ddd8eeSStefano Zampini } 134458a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1345f4ddd8eeSStefano Zampini } 1346f4ddd8eeSStefano Zampini 13475e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 13485e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 13495e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 13505e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 13515e8657edSStefano Zampini } else { 1352b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 13535e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 13545e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1355d16cbb6bSStefano Zampini } 1356d16cbb6bSStefano Zampini 13574f1b2e48SStefano Zampini /* detect local disconnected subdomains if requested and not done before */ 13584f1b2e48SStefano Zampini if (pcbddc->detect_disconnected && !pcbddc->n_local_subs) { 13594f1b2e48SStefano Zampini ierr = MatDetectDisconnectedComponents(pcbddc->local_mat,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 13604f1b2e48SStefano Zampini } 13614f1b2e48SStefano Zampini 13624f1b2e48SStefano Zampini /* 13631dd7afcfSStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) 13644f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 13654f1b2e48SStefano Zampini */ 13661dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1367d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 13689f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 13699f47a83aSStefano Zampini 137005b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1371339f8db1SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat */ 1372339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1373a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1374c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 13751dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 13761dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 13771dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 13781dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 13791dd7afcfSStefano Zampini } else { 1380a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 13811dd7afcfSStefano Zampini } 1382a3df083aSStefano Zampini } else { 13839f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1384674ae819SStefano Zampini } 1385a3df083aSStefano Zampini } 1386f1a72664SStefano Zampini /* propagate relevant information */ 13874f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 13883301b35fSStefano Zampini if (matis->A->symmetric_set) { 13893301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 13903301b35fSStefano Zampini } 1391e496cd5dSStefano Zampini #endif 139206a4e24aSStefano Zampini if (matis->A->symmetric_set) { 139306a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 139406a4e24aSStefano Zampini } 139506a4e24aSStefano Zampini if (matis->A->spd_set) { 139606a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 139706a4e24aSStefano Zampini } 1398e496cd5dSStefano Zampini 13995e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 14005e8657edSStefano Zampini { 14015e8657edSStefano Zampini Mat temp_mat; 14025e8657edSStefano Zampini 14035e8657edSStefano Zampini temp_mat = matis->A; 14045e8657edSStefano Zampini matis->A = pcbddc->local_mat; 14055e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 14065e8657edSStefano Zampini pcbddc->local_mat = matis->A; 14075e8657edSStefano Zampini matis->A = temp_mat; 14085e8657edSStefano Zampini } 1409684f6988SStefano Zampini 141081d14e9dSStefano Zampini /* Analyze interface */ 1411674ae819SStefano Zampini if (computetopography) { 1412674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 14138de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1414674ae819SStefano Zampini } 1415fb8d54d4SStefano Zampini 14165408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 14175408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 14185408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 14195408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 142009f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 14215408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 142209f581a4SStefano Zampini } 14235408967cSStefano Zampini #endif 14244f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 142506f24817SStefano Zampini 1426b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1427684f6988SStefano Zampini if (computesolvers) { 1428d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1429d5574798SStefano Zampini 1430d5574798SStefano Zampini if (computesubschurs && computetopography) { 143108122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1432b1b3d7a2SStefano Zampini } 1433df4d28bfSStefano Zampini if (sub_schurs->schur_explicit) { 14342070dbb6SStefano Zampini if (computesubschurs) { 143508122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 14362070dbb6SStefano Zampini } 1437d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1438d5574798SStefano Zampini } else { 1439d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 14402070dbb6SStefano Zampini if (computesubschurs) { 1441d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1442d5574798SStefano Zampini } 14432070dbb6SStefano Zampini } 144408122e43SStefano Zampini if (pcbddc->adaptive_selection) { 144508122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 14468de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1447b7eb3628SStefano Zampini } 1448b1b3d7a2SStefano Zampini } 1449684f6988SStefano Zampini 1450f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1451fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1452f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1453f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1454f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1455f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1456f4ddd8eeSStefano Zampini } else { 1457f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1458f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1459f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1460165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1461f4ddd8eeSStefano Zampini PetscInt i; 1462165b64e2SStefano Zampini const Vec *nearnullvecs; 1463165b64e2SStefano Zampini PetscObjectState state; 1464165b64e2SStefano Zampini PetscInt nnsp_size; 1465165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1466f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1467f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1468165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1469f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1470f4ddd8eeSStefano Zampini break; 1471f4ddd8eeSStefano Zampini } 1472f4ddd8eeSStefano Zampini } 1473f4ddd8eeSStefano Zampini } 1474f4ddd8eeSStefano Zampini } 1475f4ddd8eeSStefano Zampini } else { 1476f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1477f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1478f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1479f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1480f4ddd8eeSStefano Zampini } 1481f4ddd8eeSStefano Zampini } 1482f4ddd8eeSStefano Zampini 1483f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1484727cdba6SStefano Zampini /* reset primal space flags */ 1485f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1486727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 14878de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1488727cdba6SStefano Zampini /* It also sets the primal space flags */ 1489674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1490e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1491f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 14923975b054SStefano Zampini } 14935e8657edSStefano Zampini 14943975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 14955e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 14965e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 14975e8657edSStefano Zampini 14985e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 14994f1b2e48SStefano Zampini if (pcbddc->benign_change) { 15001dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1501c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1502c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1503c263805aSStefano Zampini } 15045e8657edSStefano Zampini /* get submatrices */ 15055e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 15065e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 15075e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 15085e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 15095e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 15105e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 15113975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 15123975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 15139c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1514b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15155e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15165e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15175e8657edSStefano Zampini } 1518b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 151999cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1520b96c3477SStefano Zampini /* SetUp Scaling operator */ 1521674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 15220c7d97c5SJed Brown } 15231dd7afcfSStefano Zampini /* mark topography as done */ 152456282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 15250369aaf7SStefano Zampini 15261dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 15271dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 15281dd7afcfSStefano Zampini 152958a03d70SStefano Zampini if (pcbddc->dbg_flag) { 153058a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 15312b510759SStefano Zampini } 15320c7d97c5SJed Brown PetscFunctionReturn(0); 15330c7d97c5SJed Brown } 15340c7d97c5SJed Brown 15350c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 15360c7d97c5SJed Brown /* 153750efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 15380c7d97c5SJed Brown 15390c7d97c5SJed Brown Input Parameters: 15400f202f7eSStefano Zampini + pc - the preconditioner context 15410f202f7eSStefano Zampini - r - input vector (global) 15420c7d97c5SJed Brown 15430c7d97c5SJed Brown Output Parameter: 15440c7d97c5SJed Brown . z - output vector (global) 15450c7d97c5SJed Brown 15460c7d97c5SJed Brown Application Interface Routine: PCApply() 15470c7d97c5SJed Brown */ 15480c7d97c5SJed Brown #undef __FUNCT__ 15490c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 155053cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 15510c7d97c5SJed Brown { 15520c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 15530c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1554b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 15550c7d97c5SJed Brown PetscErrorCode ierr; 15563b03a366Sstefano_zampini const PetscScalar one = 1.0; 15573b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 15582617d88aSStefano Zampini const PetscScalar zero = 0.0; 15590c7d97c5SJed Brown 15600c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 15610c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1562b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 15630c7d97c5SJed Brown 15640c7d97c5SJed Brown PetscFunctionBegin; 15651dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 15661dd7afcfSStefano Zampini Vec swap; 15671dd7afcfSStefano Zampini ierr = VecCopy(r,pcbddc->work_change);CHKERRQ(ierr); 15681dd7afcfSStefano Zampini swap = pcbddc->work_change; 15691dd7afcfSStefano Zampini pcbddc->work_change = r; 15701dd7afcfSStefano Zampini r = swap; 15711dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,r);CHKERRQ(ierr); 15721dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 15731dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 15741dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 15751dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 15761dd7afcfSStefano Zampini } 15771dd7afcfSStefano Zampini } 1578015636ebSStefano Zampini if (pcbddc->benign_saddle_point) { /* get p0 from r */ 1579015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1580efc2fbd9SStefano Zampini } 158185c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1582b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 15830c7d97c5SJed Brown /* First Dirichlet solve */ 15840c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 15850c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 15860c7d97c5SJed Brown /* 15870c7d97c5SJed Brown Assembling right hand side for BDDC operator 1588b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1589674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 15900c7d97c5SJed Brown */ 1591b097fa66SStefano Zampini if (n_D) { 1592b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 15930c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 1594*16909a7fSStefano Zampini if (pcbddc->switch_static) { 1595*16909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 1596*16909a7fSStefano Zampini 1597*16909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 1598*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1599*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1600*16909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 1601*16909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1602*16909a7fSStefano Zampini } else { 1603*16909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1604*16909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 1605*16909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1606*16909a7fSStefano Zampini } 1607*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1608*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1609*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1610*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1611*16909a7fSStefano Zampini } else { 1612b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1613*16909a7fSStefano Zampini } 1614b097fa66SStefano Zampini } else { 1615b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1616b097fa66SStefano Zampini } 16170c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 16180c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1619674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1620b76ba322SStefano Zampini } else { 16214fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1622674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1623b76ba322SStefano Zampini } 16244fee134fSStefano Zampini } 1625b76ba322SStefano Zampini 16262617d88aSStefano Zampini /* Apply interface preconditioner 16272617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1628dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 16292617d88aSStefano Zampini 1630674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1631674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 16320c7d97c5SJed Brown 16333b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 16340c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16350c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1636b097fa66SStefano Zampini if (n_B) { 1637*16909a7fSStefano Zampini if (pcbddc->switch_static) { 1638*16909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 1639*16909a7fSStefano Zampini 1640*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1641*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1642*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1643*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1644*16909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 1645*16909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1646*16909a7fSStefano Zampini } else { 1647*16909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1648*16909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 1649*16909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1650*16909a7fSStefano Zampini } 1651*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1652*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1653*16909a7fSStefano Zampini } else { 16540c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 1655*16909a7fSStefano Zampini } 1656*16909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 1657*16909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 1658*16909a7fSStefano Zampini 1659*16909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 1660*16909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1661*16909a7fSStefano Zampini } else { 1662*16909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 1663*16909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1664*16909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 1665*16909a7fSStefano Zampini } 1666b097fa66SStefano Zampini } 1667df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1668efc2fbd9SStefano Zampini 1669b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1670b097fa66SStefano Zampini if (pcbddc->switch_static) { 1671b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1672b097fa66SStefano Zampini } else { 1673b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1674b097fa66SStefano Zampini } 16750c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 16760c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1677b097fa66SStefano Zampini } else { 1678b097fa66SStefano Zampini if (pcbddc->switch_static) { 1679b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1680b097fa66SStefano Zampini } else { 1681b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1682b097fa66SStefano Zampini } 1683b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1684b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1685b097fa66SStefano Zampini } 1686efc2fbd9SStefano Zampini 1687015636ebSStefano Zampini if (pcbddc->benign_saddle_point) { /* set p0 (computed in PCBDDCApplyInterface) */ 16881dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 16891dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 16901dd7afcfSStefano Zampini } 1691015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1692efc2fbd9SStefano Zampini } 16931dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 16941dd7afcfSStefano Zampini Vec swap; 16951dd7afcfSStefano Zampini 16961dd7afcfSStefano Zampini swap = r; 16971dd7afcfSStefano Zampini r = pcbddc->work_change; 16981dd7afcfSStefano Zampini pcbddc->work_change = swap; 16991dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 17001dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 17011dd7afcfSStefano Zampini } 17020c7d97c5SJed Brown PetscFunctionReturn(0); 17030c7d97c5SJed Brown } 170450efa1b5SStefano Zampini 170550efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 170650efa1b5SStefano Zampini /* 170750efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 170850efa1b5SStefano Zampini 170950efa1b5SStefano Zampini Input Parameters: 17100f202f7eSStefano Zampini + pc - the preconditioner context 17110f202f7eSStefano Zampini - r - input vector (global) 171250efa1b5SStefano Zampini 171350efa1b5SStefano Zampini Output Parameter: 171450efa1b5SStefano Zampini . z - output vector (global) 171550efa1b5SStefano Zampini 171650efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 171750efa1b5SStefano Zampini */ 171850efa1b5SStefano Zampini #undef __FUNCT__ 171950efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 172050efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 172150efa1b5SStefano Zampini { 172250efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 172350efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1724b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 172550efa1b5SStefano Zampini PetscErrorCode ierr; 172650efa1b5SStefano Zampini const PetscScalar one = 1.0; 172750efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 172850efa1b5SStefano Zampini const PetscScalar zero = 0.0; 172950efa1b5SStefano Zampini 173050efa1b5SStefano Zampini PetscFunctionBegin; 17311dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 17321dd7afcfSStefano Zampini Vec swap; 17331dd7afcfSStefano Zampini ierr = VecCopy(r,pcbddc->work_change);CHKERRQ(ierr); 17341dd7afcfSStefano Zampini swap = pcbddc->work_change; 17351dd7afcfSStefano Zampini pcbddc->work_change = r; 17361dd7afcfSStefano Zampini r = swap; 17371dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,r);CHKERRQ(ierr); 17381dd7afcfSStefano Zampini } 1739537c1cdfSStefano Zampini if (pcbddc->benign_saddle_point) { /* get p0 from r */ 1740537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1741537c1cdfSStefano Zampini } 174250efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1743b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 174450efa1b5SStefano Zampini /* First Dirichlet solve */ 174550efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 174650efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 174750efa1b5SStefano Zampini /* 174850efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1749b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 175050efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 175150efa1b5SStefano Zampini */ 1752b097fa66SStefano Zampini if (n_D) { 1753b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 175450efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 1755*16909a7fSStefano Zampini if (pcbddc->switch_static) { 1756*16909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 1757*16909a7fSStefano Zampini 1758*16909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 1759*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1760*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1761*16909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 1762*16909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1763*16909a7fSStefano Zampini } else { 1764*16909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1765*16909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 1766*16909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1767*16909a7fSStefano Zampini } 1768*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1769*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1770*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1771*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1772*16909a7fSStefano Zampini } else { 1773b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1774*16909a7fSStefano Zampini } 1775b097fa66SStefano Zampini } else { 1776b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1777b097fa66SStefano Zampini } 177850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 177950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 178050efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 178150efa1b5SStefano Zampini } else { 178250efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 178350efa1b5SStefano Zampini } 178450efa1b5SStefano Zampini 178550efa1b5SStefano Zampini /* Apply interface preconditioner 178650efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1787dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 178850efa1b5SStefano Zampini 178950efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 179050efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 179150efa1b5SStefano Zampini 179250efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 179350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 179450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1795b097fa66SStefano Zampini if (n_B) { 1796*16909a7fSStefano Zampini if (pcbddc->switch_static) { 1797*16909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 1798*16909a7fSStefano Zampini 1799*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1800*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1801*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1802*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1803*16909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 1804*16909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1805*16909a7fSStefano Zampini } else { 1806*16909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1807*16909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 1808*16909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1809*16909a7fSStefano Zampini } 1810*16909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1811*16909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1812*16909a7fSStefano Zampini } else { 181350efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 1814*16909a7fSStefano Zampini } 1815*16909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 1816*16909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 1817*16909a7fSStefano Zampini 1818*16909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 1819*16909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1820*16909a7fSStefano Zampini } else { 1821*16909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 1822*16909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 1823*16909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 1824*16909a7fSStefano Zampini } 1825b097fa66SStefano Zampini } 1826b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1827b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1828b097fa66SStefano Zampini if (pcbddc->switch_static) { 1829b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1830b097fa66SStefano Zampini } else { 1831b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1832b097fa66SStefano Zampini } 183350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 183450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1835b097fa66SStefano Zampini } else { 1836b097fa66SStefano Zampini if (pcbddc->switch_static) { 1837b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1838b097fa66SStefano Zampini } else { 1839b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1840b097fa66SStefano Zampini } 1841b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1842b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1843b097fa66SStefano Zampini } 1844537c1cdfSStefano Zampini if (pcbddc->benign_saddle_point) { /* set p0 (computed in PCBDDCApplyInterface) */ 1845537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1846537c1cdfSStefano Zampini } 18471dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18481dd7afcfSStefano Zampini Vec swap; 18491dd7afcfSStefano Zampini 18501dd7afcfSStefano Zampini swap = r; 18511dd7afcfSStefano Zampini r = pcbddc->work_change; 18521dd7afcfSStefano Zampini pcbddc->work_change = swap; 18531dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 18541dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 18551dd7afcfSStefano Zampini } 185650efa1b5SStefano Zampini PetscFunctionReturn(0); 185750efa1b5SStefano Zampini } 1858da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1859674ae819SStefano Zampini 1860da1bb401SStefano Zampini #undef __FUNCT__ 1861da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1862da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1863da1bb401SStefano Zampini { 1864da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1865da1bb401SStefano Zampini PetscErrorCode ierr; 1866da1bb401SStefano Zampini 1867da1bb401SStefano Zampini PetscFunctionBegin; 1868674ae819SStefano Zampini /* free BDDC custom data */ 1869674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1870674ae819SStefano Zampini /* destroy objects related to topography */ 1871674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1872674ae819SStefano Zampini /* free allocated graph structure */ 1873da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1874b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1875b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 187634a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1877674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 187834a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1879674ae819SStefano Zampini /* free solvers stuff */ 1880674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 188162a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 188262a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 188362a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 18841dd7afcfSStefano Zampini /* free data created by PCIS */ 18851dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 18863425bc38SStefano Zampini /* remove functions */ 1887906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1888674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 188930368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 1890bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 18912b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1892b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 18932b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1894bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 1895bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 189682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1897bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 189882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1899bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 190082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1901bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1902785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1903bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 190463602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1905bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1906bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1907bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1908bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1909674ae819SStefano Zampini /* Free the private data structure */ 1910674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1911da1bb401SStefano Zampini PetscFunctionReturn(0); 1912da1bb401SStefano Zampini } 19133425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 19141e6b0712SBarry Smith 19153425bc38SStefano Zampini #undef __FUNCT__ 19163425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 19173425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 19183425bc38SStefano Zampini { 1919674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1920c08af4c6SStefano Zampini Vec copy_standard_rhs; 19213425bc38SStefano Zampini PC_IS* pcis; 19223425bc38SStefano Zampini PC_BDDC* pcbddc; 19233425bc38SStefano Zampini PetscErrorCode ierr; 19240c7d97c5SJed Brown 19253425bc38SStefano Zampini PetscFunctionBegin; 19263425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 19273425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 19283425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 19293425bc38SStefano Zampini 1930c08af4c6SStefano Zampini /* 1931c08af4c6SStefano Zampini change of basis for physical rhs if needed 1932c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1933c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1934c08af4c6SStefano Zampini */ 1935c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1936c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1937c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 19383425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1939c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1940c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1941fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1942fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1943c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1944c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1945674ae819SStefano Zampini /* Apply partition of unity */ 19463425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1947c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 19488eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 19493425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 19503425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 19513425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 19523425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1953c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1954c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1955c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1956c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1957c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1958c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19593425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 19603425bc38SStefano Zampini } 1961c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 19623425bc38SStefano Zampini /* BDDC rhs */ 19633425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 19648eeda7d8SStefano Zampini if (pcbddc->switch_static) { 19653425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 19663425bc38SStefano Zampini } 19673425bc38SStefano Zampini /* apply BDDC */ 1968dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 19693425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 19703425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 19713425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 19723425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19733425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19743425bc38SStefano Zampini PetscFunctionReturn(0); 19753425bc38SStefano Zampini } 19761e6b0712SBarry Smith 19773425bc38SStefano Zampini #undef __FUNCT__ 19783425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 19793425bc38SStefano Zampini /*@ 19800f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 19813425bc38SStefano Zampini 19823425bc38SStefano Zampini Collective 19833425bc38SStefano Zampini 19843425bc38SStefano Zampini Input Parameters: 19850f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 19860f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 19873425bc38SStefano Zampini 19883425bc38SStefano Zampini Output Parameters: 19890f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 19903425bc38SStefano Zampini 19913425bc38SStefano Zampini Level: developer 19923425bc38SStefano Zampini 19933425bc38SStefano Zampini Notes: 19943425bc38SStefano Zampini 19950f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 19963425bc38SStefano Zampini @*/ 19973425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 19983425bc38SStefano Zampini { 1999674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 20003425bc38SStefano Zampini PetscErrorCode ierr; 20013425bc38SStefano Zampini 20023425bc38SStefano Zampini PetscFunctionBegin; 20033425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20043425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 20053425bc38SStefano Zampini PetscFunctionReturn(0); 20063425bc38SStefano Zampini } 20073425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20081e6b0712SBarry Smith 20093425bc38SStefano Zampini #undef __FUNCT__ 20103425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 20113425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 20123425bc38SStefano Zampini { 2013674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 20143425bc38SStefano Zampini PC_IS* pcis; 20153425bc38SStefano Zampini PC_BDDC* pcbddc; 20163425bc38SStefano Zampini PetscErrorCode ierr; 20173425bc38SStefano Zampini 20183425bc38SStefano Zampini PetscFunctionBegin; 20193425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20203425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20213425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20223425bc38SStefano Zampini 20233425bc38SStefano Zampini /* apply B_delta^T */ 20243425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20253425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20263425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 20273425bc38SStefano Zampini /* compute rhs for BDDC application */ 20283425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 20298eeda7d8SStefano Zampini if (pcbddc->switch_static) { 20303425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20313425bc38SStefano Zampini } 20323425bc38SStefano Zampini /* apply BDDC */ 2033dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 20343425bc38SStefano Zampini /* put values into standard global vector */ 20353425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20363425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20378eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 20383425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 20393425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 20403425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 20413425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20423425bc38SStefano Zampini } 20433425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20443425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20453425bc38SStefano Zampini /* final change of basis if needed 20463425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 20473308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 20483425bc38SStefano Zampini PetscFunctionReturn(0); 20493425bc38SStefano Zampini } 20501e6b0712SBarry Smith 20513425bc38SStefano Zampini #undef __FUNCT__ 20523425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 20533425bc38SStefano Zampini /*@ 20540f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 20553425bc38SStefano Zampini 20563425bc38SStefano Zampini Collective 20573425bc38SStefano Zampini 20583425bc38SStefano Zampini Input Parameters: 20590f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 20600f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 20613425bc38SStefano Zampini 20623425bc38SStefano Zampini Output Parameters: 20630f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 20643425bc38SStefano Zampini 20653425bc38SStefano Zampini Level: developer 20663425bc38SStefano Zampini 20673425bc38SStefano Zampini Notes: 20683425bc38SStefano Zampini 20690f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 20703425bc38SStefano Zampini @*/ 20713425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 20723425bc38SStefano Zampini { 2073674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 20743425bc38SStefano Zampini PetscErrorCode ierr; 20753425bc38SStefano Zampini 20763425bc38SStefano Zampini PetscFunctionBegin; 20773425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20783425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 20793425bc38SStefano Zampini PetscFunctionReturn(0); 20803425bc38SStefano Zampini } 20813425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20821e6b0712SBarry Smith 2083f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2084edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2085f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2086f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2087edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2088f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2089674ae819SStefano Zampini 20903425bc38SStefano Zampini #undef __FUNCT__ 20913425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 20923425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 20933425bc38SStefano Zampini { 2094674ae819SStefano Zampini 2095674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 20963425bc38SStefano Zampini Mat newmat; 2097674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 20983425bc38SStefano Zampini PC newpc; 2099ce94432eSBarry Smith MPI_Comm comm; 21003425bc38SStefano Zampini PetscErrorCode ierr; 21013425bc38SStefano Zampini 21023425bc38SStefano Zampini PetscFunctionBegin; 2103ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 21043425bc38SStefano Zampini /* FETIDP linear matrix */ 21053425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 21063425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 21073425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 21083425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2109edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 21103425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 21113425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 21123425bc38SStefano Zampini /* FETIDP preconditioner */ 21133425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 21143425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 21153425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 21163425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 21173425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 21183425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2119edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 21203425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 212123ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 21223425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 21233425bc38SStefano Zampini /* return pointers for objects created */ 21243425bc38SStefano Zampini *fetidp_mat=newmat; 21253425bc38SStefano Zampini *fetidp_pc=newpc; 21263425bc38SStefano Zampini PetscFunctionReturn(0); 21273425bc38SStefano Zampini } 21281e6b0712SBarry Smith 21293425bc38SStefano Zampini #undef __FUNCT__ 21303425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 21313425bc38SStefano Zampini /*@ 21320f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 21333425bc38SStefano Zampini 21343425bc38SStefano Zampini Collective 21353425bc38SStefano Zampini 21363425bc38SStefano Zampini Input Parameters: 21370f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 213828509bceSStefano Zampini 213928509bceSStefano Zampini Output Parameters: 21400f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 21410f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 214228509bceSStefano Zampini 214328509bceSStefano Zampini Options Database Keys: 21440f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 21453425bc38SStefano Zampini 21463425bc38SStefano Zampini Level: developer 21473425bc38SStefano Zampini 21483425bc38SStefano Zampini Notes: 21490f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 21503425bc38SStefano Zampini 21510f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 21523425bc38SStefano Zampini @*/ 21533425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 21543425bc38SStefano Zampini { 21553425bc38SStefano Zampini PetscErrorCode ierr; 21563425bc38SStefano Zampini 21573425bc38SStefano Zampini PetscFunctionBegin; 21583425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 21593425bc38SStefano Zampini if (pc->setupcalled) { 2160516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2161f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 21623425bc38SStefano Zampini PetscFunctionReturn(0); 21633425bc38SStefano Zampini } 21640c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2165da1bb401SStefano Zampini /*MC 2166da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 21670c7d97c5SJed Brown 216828509bceSStefano Zampini An implementation of the BDDC preconditioner based on 216928509bceSStefano Zampini 217028509bceSStefano Zampini .vb 217128509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 217228509bceSStefano Zampini [2] A. Klawonn and O. B. Widlund. "Dual-Primal FETI Methods for Linear Elasticity", http://cs.nyu.edu/csweb/Research/TechReports/TR2004-855/TR2004-855.pdf 217328509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 21740f202f7eSStefano Zampini [4] C. Pechstein and C. R. Dohrmann. "Modern domain decomposition methods BDDC, deluxe scaling, and an algebraic approach", Seminar talk, Linz, December 2013, http://people.ricam.oeaw.ac.at/c.pechstein/pechstein-bddc2013.pdf 217528509bceSStefano Zampini .ve 217628509bceSStefano Zampini 217728509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 217828509bceSStefano Zampini 21790f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 218028509bceSStefano Zampini 218128509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 218228509bceSStefano Zampini 2183b6fdb6dfSStefano Zampini Unlike 'conventional' interface preconditioners, PCBDDC iterates over all degrees of freedom, not just those on the interface. This allows the use of approximate solvers on the subdomains. 2184b6fdb6dfSStefano Zampini 21850f202f7eSStefano Zampini Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 218628509bceSStefano Zampini 21870f202f7eSStefano Zampini Boundary nodes are split in vertices, edges and faces classes using information from the local to global mapping of dofs and the local connectivity graph of nodes. The latter can be customized by using PCBDDCSetLocalAdjacencyGraph() 218830368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 218928509bceSStefano Zampini 21900f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 219128509bceSStefano Zampini 21920f202f7eSStefano Zampini Change of basis is performed similarly to [2] when requested. When more than one constraint is present on a single connected component (i.e. an edge or a face), a robust method based on local QR factorizations is used. 21930f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 219428509bceSStefano Zampini 21950f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 219628509bceSStefano Zampini 2197df4d28bfSStefano Zampini Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS or MKL_PARDISO are present. Future versions of the code will also consider using PASTIX. 219828509bceSStefano Zampini 21990f202f7eSStefano Zampini An experimental interface to the FETI-DP method is available. FETI-DP operators could be created using PCBDDCCreateFETIDPOperators(). A stand-alone class for the FETI-DP method will be provided in the next releases. 22000f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 22010f202f7eSStefano Zampini 22020f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 22030f202f7eSStefano Zampini 22040f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 22050f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 22060f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 22070f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 22080f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 22090f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 22100f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 221128509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 22120f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio <8> - number of subdomains which will be aggregated together at the coarser level (e.g. H/h ratio at the coarser level, significative only in the multilevel case) 22130f202f7eSStefano Zampini . -pc_bddc_redistribute <0> - size of a subset of processors where the coarse problem will be remapped (the value is ignored if not at the coarsest level) 22140f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 22150f202f7eSStefano Zampini . -pc_bddc_schur_layers <-1> - select the economic version of deluxe scaling by specifying the number of layers (-1 corresponds to the original deluxe scaling) 2216df4d28bfSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 221728509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 221828509bceSStefano Zampini 221928509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 222028509bceSStefano Zampini .vb 222128509bceSStefano Zampini -pc_bddc_dirichlet_ 222228509bceSStefano Zampini -pc_bddc_neumann_ 222328509bceSStefano Zampini -pc_bddc_coarse_ 222428509bceSStefano Zampini .ve 22250f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 222628509bceSStefano Zampini 22270f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 222828509bceSStefano Zampini .vb 2229312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2230312be037SStefano Zampini -pc_bddc_neumann_lN_ 2231312be037SStefano Zampini -pc_bddc_coarse_lN_ 223228509bceSStefano Zampini .ve 22330f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 22340f202f7eSStefano Zampini In order to specify options for the BDDC operators at the coarser levels (and not for the solvers), prepend -pc_bddc_coarse_ or -pc_bddc_coarse_l to the option, e.g. 22350f202f7eSStefano Zampini .vb 22360f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 22370f202f7eSStefano Zampini .ve 22380f202f7eSStefano Zampini will use a threshold of 5 for constraints' selection at the first coarse level and will redistribute the coarse problem of the first coarse level on 3 processors 2239da1bb401SStefano Zampini 2240da1bb401SStefano Zampini Level: intermediate 2241da1bb401SStefano Zampini 2242b6fdb6dfSStefano Zampini Developer notes: 2243da1bb401SStefano Zampini 2244da1bb401SStefano Zampini Contributed by Stefano Zampini 2245da1bb401SStefano Zampini 2246da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2247da1bb401SStefano Zampini M*/ 2248b2573a8aSBarry Smith 2249da1bb401SStefano Zampini #undef __FUNCT__ 2250da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 22518cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2252da1bb401SStefano Zampini { 2253da1bb401SStefano Zampini PetscErrorCode ierr; 2254da1bb401SStefano Zampini PC_BDDC *pcbddc; 2255da1bb401SStefano Zampini 2256da1bb401SStefano Zampini PetscFunctionBegin; 2257da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2258b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2259da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2260da1bb401SStefano Zampini 2261da1bb401SStefano Zampini /* create PCIS data structure */ 2262da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2263da1bb401SStefano Zampini 226447d04d0dSStefano Zampini /* BDDC customization */ 226508a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 226647d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 226747d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 226847d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 226947d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 227047d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 227147d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2272fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2273fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 22743301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 227506a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 227614f95afaSStefano Zampini pcbddc->vertex_size = 1; 227747d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2278b9d89cd5SStefano Zampini /* private */ 2279727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 22800e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 22810e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 22820e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2283e9189074SStefano Zampini pcbddc->n_vertices = 0; 2284727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2285fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 228668457ee5SStefano Zampini pcbddc->coarse_size = -1; 2287f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2288727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2289f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2290f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2291f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2292674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 229330368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 22940bdf917eSStefano Zampini pcbddc->NullSpace = 0; 22953972b0daSStefano Zampini pcbddc->temp_solution = 0; 2296534831adSStefano Zampini pcbddc->original_rhs = 0; 2297534831adSStefano Zampini pcbddc->local_mat = 0; 2298534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2299b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2300da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2301da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2302da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2303da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 230415aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 230515aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2306da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2307da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2308da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2309da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2310da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2311da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2312da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2313da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2314da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2315da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2316785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2317785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2318785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 231960d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 232060d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 232163602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2322da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 232363602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2324da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 232585c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 232647d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 232747d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2328b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 23294fad6a16SStefano Zampini pcbddc->current_level = 0; 23302b510759SStefano Zampini pcbddc->max_levels = 0; 2331323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 233274e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2333f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2334323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 23354f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 23364f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 23374f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 233881d14e9dSStefano Zampini 233981d14e9dSStefano Zampini /* benign subspace trick */ 234081d14e9dSStefano Zampini pcbddc->benign_change = 0; 23410369aaf7SStefano Zampini pcbddc->benign_vec = 0; 23420369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 23430369aaf7SStefano Zampini pcbddc->benign_sf = 0; 23444f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 23454f1b2e48SStefano Zampini pcbddc->benign_n = 0; 23464f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 23474f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 23484f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2349b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 235081d14e9dSStefano Zampini 2351674ae819SStefano Zampini /* create local graph structure */ 2352674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2353674ae819SStefano Zampini 2354674ae819SStefano Zampini /* scaling */ 2355674ae819SStefano Zampini pcbddc->work_scaling = 0; 235634a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2357ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2358b96c3477SStefano Zampini 2359b96c3477SStefano Zampini /* create sub schurs structure */ 2360b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2361b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2362b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2363b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2364b96c3477SStefano Zampini 2365b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2366da1bb401SStefano Zampini 2367b7eb3628SStefano Zampini /* adaptivity */ 2368f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 236908122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2370f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2371b7eb3628SStefano Zampini 2372da1bb401SStefano Zampini /* function pointers */ 2373da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 237493bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2375da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2376da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2377da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2378da1bb401SStefano Zampini pc->ops->view = 0; 2379da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2380da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2381da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2382534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2383534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2384da1bb401SStefano Zampini 2385da1bb401SStefano Zampini /* composing function */ 2386906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2387674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 238830368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2389bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 23902b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2391b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 23922b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2393bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2394bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 239582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2396bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 239782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2398bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 239982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2400bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 240182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2402bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 240363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2404bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2405bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2406bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2407bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2408da1bb401SStefano Zampini PetscFunctionReturn(0); 2409da1bb401SStefano Zampini } 24103425bc38SStefano Zampini 2411