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); 496661aa1dSStefano 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); 50ac632422SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr); 518eeda7d8SStefano Zampini /* Change of basis */ 52b9b85e73SStefano 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); 53b9b85e73SStefano 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); 54674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 55674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 56674ae819SStefano Zampini } 578eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 588eeda7d8SStefano 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); 5974e2c79eSStefano 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); 600298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 612b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 62323d395dSStefano 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); 63674ae819SStefano 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); 64b96c3477SStefano 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); 65b96c3477SStefano 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); 66b96c3477SStefano 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); 67683d3df6SStefano 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); 68bf3a8328SStefano 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); 69ac632422SStefano 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); 70bf3a8328SStefano 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); 714c6709b3SStefano 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); 7208122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 7308122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 743301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 75b0c7d250SStefano 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); 7606a4e24aSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_trick","Apply the benign subspace trick to a class of saddle point problems","none",pcbddc->benign_saddle_point,&pcbddc->benign_saddle_point,NULL);CHKERRQ(ierr); 774f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 780c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 790c7d97c5SJed Brown PetscFunctionReturn(0); 800c7d97c5SJed Brown } 810c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 82674ae819SStefano Zampini #undef __FUNCT__ 83906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 84906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 85b9b85e73SStefano Zampini { 86b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 87b9b85e73SStefano Zampini PetscErrorCode ierr; 88b9b85e73SStefano Zampini 89b9b85e73SStefano Zampini PetscFunctionBegin; 90b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 91*56282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 92b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 93b9b85e73SStefano Zampini PetscFunctionReturn(0); 94b9b85e73SStefano Zampini } 95b9b85e73SStefano Zampini #undef __FUNCT__ 96906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 97b9b85e73SStefano Zampini /*@ 98906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 99b9b85e73SStefano Zampini 100b9b85e73SStefano Zampini Collective on PC 101b9b85e73SStefano Zampini 102b9b85e73SStefano Zampini Input Parameters: 103b9b85e73SStefano Zampini + pc - the preconditioning context 104906d46d4SStefano Zampini - change - the change of basis matrix 105b9b85e73SStefano Zampini 106b9b85e73SStefano Zampini Level: intermediate 107b9b85e73SStefano Zampini 108b9b85e73SStefano Zampini Notes: 109b9b85e73SStefano Zampini 110b9b85e73SStefano Zampini .seealso: PCBDDC 111b9b85e73SStefano Zampini @*/ 112906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 113b9b85e73SStefano Zampini { 114b9b85e73SStefano Zampini PetscErrorCode ierr; 115b9b85e73SStefano Zampini 116b9b85e73SStefano Zampini PetscFunctionBegin; 117b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 118b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 119906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 120906d46d4SStefano Zampini if (pc->mat) { 121906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 122906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 123906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 124906d46d4SStefano Zampini if (rows_c != rows) { 125906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 126906d46d4SStefano Zampini } 127906d46d4SStefano Zampini if (cols_c != cols) { 128906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 129906d46d4SStefano Zampini } 130906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 131906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 132906d46d4SStefano Zampini if (rows_c != rows) { 133906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 134906d46d4SStefano Zampini } 135906d46d4SStefano Zampini if (cols_c != cols) { 136906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 137906d46d4SStefano Zampini } 138906d46d4SStefano Zampini } 139906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 140b9b85e73SStefano Zampini PetscFunctionReturn(0); 141b9b85e73SStefano Zampini } 142b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 143b9b85e73SStefano Zampini #undef __FUNCT__ 14430368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS_BDDC" 14530368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 14630368db7SStefano Zampini { 14730368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 148*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 14930368db7SStefano Zampini PetscErrorCode ierr; 15030368db7SStefano Zampini 15130368db7SStefano Zampini PetscFunctionBegin; 152*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 153*56282151SStefano Zampini if (pcbddc->user_primal_vertices) { 154*56282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 155*56282151SStefano Zampini } 15630368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 15730368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 15830368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 159*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 16030368db7SStefano Zampini PetscFunctionReturn(0); 16130368db7SStefano Zampini } 16230368db7SStefano Zampini #undef __FUNCT__ 16330368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS" 16430368db7SStefano Zampini /*@ 16530368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 16630368db7SStefano Zampini 16730368db7SStefano Zampini Collective 16830368db7SStefano Zampini 16930368db7SStefano Zampini Input Parameters: 17030368db7SStefano Zampini + pc - the preconditioning context 17130368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 17230368db7SStefano Zampini 17330368db7SStefano Zampini Level: intermediate 17430368db7SStefano Zampini 17530368db7SStefano Zampini Notes: 17630368db7SStefano Zampini Any process can list any global node 17730368db7SStefano Zampini 17830368db7SStefano Zampini .seealso: PCBDDC 17930368db7SStefano Zampini @*/ 18030368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 18130368db7SStefano Zampini { 18230368db7SStefano Zampini PetscErrorCode ierr; 18330368db7SStefano Zampini 18430368db7SStefano Zampini PetscFunctionBegin; 18530368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 18630368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 18730368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 18830368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 18930368db7SStefano Zampini PetscFunctionReturn(0); 19030368db7SStefano Zampini } 19130368db7SStefano Zampini /* -------------------------------------------------------------------------- */ 19230368db7SStefano Zampini #undef __FUNCT__ 193674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 194674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 195674ae819SStefano Zampini { 196674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 197*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 198674ae819SStefano Zampini PetscErrorCode ierr; 1991e6b0712SBarry Smith 200674ae819SStefano Zampini PetscFunctionBegin; 201*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 202*56282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 203*56282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 204*56282151SStefano Zampini } 205674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 20630368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 20730368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 208*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 209674ae819SStefano Zampini PetscFunctionReturn(0); 210674ae819SStefano Zampini } 211674ae819SStefano Zampini #undef __FUNCT__ 212674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 213674ae819SStefano Zampini /*@ 21428509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 215674ae819SStefano Zampini 21617eb1463SStefano Zampini Collective 217674ae819SStefano Zampini 218674ae819SStefano Zampini Input Parameters: 219674ae819SStefano Zampini + pc - the preconditioning context 22017eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 221674ae819SStefano Zampini 222674ae819SStefano Zampini Level: intermediate 223674ae819SStefano Zampini 224674ae819SStefano Zampini Notes: 225674ae819SStefano Zampini 226674ae819SStefano Zampini .seealso: PCBDDC 227674ae819SStefano Zampini @*/ 228674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 229674ae819SStefano Zampini { 230674ae819SStefano Zampini PetscErrorCode ierr; 231674ae819SStefano Zampini 232674ae819SStefano Zampini PetscFunctionBegin; 233674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 234674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 23517eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 236674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 237674ae819SStefano Zampini PetscFunctionReturn(0); 238674ae819SStefano Zampini } 239674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 2400c7d97c5SJed Brown #undef __FUNCT__ 2414fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 2424fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 2434fad6a16SStefano Zampini { 2444fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2454fad6a16SStefano Zampini 2464fad6a16SStefano Zampini PetscFunctionBegin; 2474fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 2484fad6a16SStefano Zampini PetscFunctionReturn(0); 2494fad6a16SStefano Zampini } 2501e6b0712SBarry Smith 2514fad6a16SStefano Zampini #undef __FUNCT__ 2524fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 2534fad6a16SStefano Zampini /*@ 25428509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 2554fad6a16SStefano Zampini 2564fad6a16SStefano Zampini Logically collective on PC 2574fad6a16SStefano Zampini 2584fad6a16SStefano Zampini Input Parameters: 2594fad6a16SStefano Zampini + pc - the preconditioning context 26028509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 2614fad6a16SStefano Zampini 2620f202f7eSStefano Zampini Options Database Keys: 2630f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 2644fad6a16SStefano Zampini 2654fad6a16SStefano Zampini Level: intermediate 2664fad6a16SStefano Zampini 2674fad6a16SStefano Zampini Notes: 2680f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 2694fad6a16SStefano Zampini 2700f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2714fad6a16SStefano Zampini @*/ 2724fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2734fad6a16SStefano Zampini { 2744fad6a16SStefano Zampini PetscErrorCode ierr; 2754fad6a16SStefano Zampini 2764fad6a16SStefano Zampini PetscFunctionBegin; 2774fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2782b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2794fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2804fad6a16SStefano Zampini PetscFunctionReturn(0); 2814fad6a16SStefano Zampini } 2822b510759SStefano Zampini 283b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2842b510759SStefano Zampini #undef __FUNCT__ 285b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 286b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 287b8ffe317SStefano Zampini { 288b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 289b8ffe317SStefano Zampini 290b8ffe317SStefano Zampini PetscFunctionBegin; 29185c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 292b8ffe317SStefano Zampini PetscFunctionReturn(0); 293b8ffe317SStefano Zampini } 294b8ffe317SStefano Zampini 295b8ffe317SStefano Zampini #undef __FUNCT__ 296b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 297b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 2982b510759SStefano Zampini { 2992b510759SStefano Zampini PetscErrorCode ierr; 3002b510759SStefano Zampini 3012b510759SStefano Zampini PetscFunctionBegin; 3022b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 303b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 304b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 3052b510759SStefano Zampini PetscFunctionReturn(0); 3062b510759SStefano Zampini } 3071e6b0712SBarry Smith 3084fad6a16SStefano Zampini #undef __FUNCT__ 3092b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 3102b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 3114fad6a16SStefano Zampini { 3124fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3134fad6a16SStefano Zampini 3144fad6a16SStefano Zampini PetscFunctionBegin; 3152b510759SStefano Zampini pcbddc->current_level = level; 3164fad6a16SStefano Zampini PetscFunctionReturn(0); 3174fad6a16SStefano Zampini } 3181e6b0712SBarry Smith 3194fad6a16SStefano Zampini #undef __FUNCT__ 320b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 321b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 322b8ffe317SStefano Zampini { 323b8ffe317SStefano Zampini PetscErrorCode ierr; 324b8ffe317SStefano Zampini 325b8ffe317SStefano Zampini PetscFunctionBegin; 326b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 327b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 328b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 329b8ffe317SStefano Zampini PetscFunctionReturn(0); 330b8ffe317SStefano Zampini } 331b8ffe317SStefano Zampini 332b8ffe317SStefano Zampini #undef __FUNCT__ 3332b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 3342b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 3352b510759SStefano Zampini { 3362b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3372b510759SStefano Zampini 3382b510759SStefano Zampini PetscFunctionBegin; 3392b510759SStefano Zampini pcbddc->max_levels = levels; 3402b510759SStefano Zampini PetscFunctionReturn(0); 3412b510759SStefano Zampini } 3422b510759SStefano Zampini 3432b510759SStefano Zampini #undef __FUNCT__ 3442b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 3454fad6a16SStefano Zampini /*@ 34628509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 3474fad6a16SStefano Zampini 3484fad6a16SStefano Zampini Logically collective on PC 3494fad6a16SStefano Zampini 3504fad6a16SStefano Zampini Input Parameters: 3514fad6a16SStefano Zampini + pc - the preconditioning context 35228509bceSStefano Zampini - levels - the maximum number of levels (max 9) 3534fad6a16SStefano Zampini 3540f202f7eSStefano Zampini Options Database Keys: 3550f202f7eSStefano Zampini . -pc_bddc_levels 3564fad6a16SStefano Zampini 3574fad6a16SStefano Zampini Level: intermediate 3584fad6a16SStefano Zampini 3594fad6a16SStefano Zampini Notes: 3600f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 3614fad6a16SStefano Zampini 3620f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 3634fad6a16SStefano Zampini @*/ 3642b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 3654fad6a16SStefano Zampini { 3664fad6a16SStefano Zampini PetscErrorCode ierr; 3674fad6a16SStefano Zampini 3684fad6a16SStefano Zampini PetscFunctionBegin; 3694fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 3702b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 371312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3722b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3734fad6a16SStefano Zampini PetscFunctionReturn(0); 3744fad6a16SStefano Zampini } 3754fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3761e6b0712SBarry Smith 3774fad6a16SStefano Zampini #undef __FUNCT__ 3780bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 3790bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 3800bdf917eSStefano Zampini { 3810bdf917eSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3820bdf917eSStefano Zampini PetscErrorCode ierr; 3830bdf917eSStefano Zampini 3840bdf917eSStefano Zampini PetscFunctionBegin; 3850bdf917eSStefano Zampini ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 3860bdf917eSStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 3870bdf917eSStefano Zampini pcbddc->NullSpace = NullSpace; 3880bdf917eSStefano Zampini PetscFunctionReturn(0); 3890bdf917eSStefano Zampini } 3901e6b0712SBarry Smith 3910bdf917eSStefano Zampini #undef __FUNCT__ 3920bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace" 3930bdf917eSStefano Zampini /*@ 39428509bceSStefano Zampini PCBDDCSetNullSpace - Set nullspace for BDDC operator 3950bdf917eSStefano Zampini 3960bdf917eSStefano Zampini Logically collective on PC and MatNullSpace 3970bdf917eSStefano Zampini 3980bdf917eSStefano Zampini Input Parameters: 3990bdf917eSStefano Zampini + pc - the preconditioning context 40028509bceSStefano Zampini - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 4010bdf917eSStefano Zampini 4020bdf917eSStefano Zampini Level: intermediate 4030bdf917eSStefano Zampini 4040bdf917eSStefano Zampini Notes: 4050bdf917eSStefano Zampini 4060bdf917eSStefano Zampini .seealso: PCBDDC 4070bdf917eSStefano Zampini @*/ 4080bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 4090bdf917eSStefano Zampini { 4100bdf917eSStefano Zampini PetscErrorCode ierr; 4110bdf917eSStefano Zampini 4120bdf917eSStefano Zampini PetscFunctionBegin; 4130bdf917eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 414674ae819SStefano Zampini PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 4152b510759SStefano Zampini PetscCheckSameComm(pc,1,NullSpace,2); 4160bdf917eSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 4170bdf917eSStefano Zampini PetscFunctionReturn(0); 4180bdf917eSStefano Zampini } 4190bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */ 4201e6b0712SBarry Smith 4210bdf917eSStefano Zampini #undef __FUNCT__ 4223b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 4233b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 4243b03a366Sstefano_zampini { 4253b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 426*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 4273b03a366Sstefano_zampini PetscErrorCode ierr; 4283b03a366Sstefano_zampini 4293b03a366Sstefano_zampini PetscFunctionBegin; 430*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 431*56282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 432*56282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 433*56282151SStefano Zampini } 434785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 435785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4363b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 43736e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 438*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 4393b03a366Sstefano_zampini PetscFunctionReturn(0); 4403b03a366Sstefano_zampini } 4411e6b0712SBarry Smith 4423b03a366Sstefano_zampini #undef __FUNCT__ 4433b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 4443b03a366Sstefano_zampini /*@ 44528509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 4463b03a366Sstefano_zampini 447785d1243SStefano Zampini Collective 4483b03a366Sstefano_zampini 4493b03a366Sstefano_zampini Input Parameters: 4503b03a366Sstefano_zampini + pc - the preconditioning context 451785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 4523b03a366Sstefano_zampini 4533b03a366Sstefano_zampini Level: intermediate 4543b03a366Sstefano_zampini 4550f202f7eSStefano Zampini Notes: 4560f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 4573b03a366Sstefano_zampini 4580f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 4593b03a366Sstefano_zampini @*/ 4603b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 4613b03a366Sstefano_zampini { 4623b03a366Sstefano_zampini PetscErrorCode ierr; 4633b03a366Sstefano_zampini 4643b03a366Sstefano_zampini PetscFunctionBegin; 4653b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 466674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 467785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4683b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4693b03a366Sstefano_zampini PetscFunctionReturn(0); 4703b03a366Sstefano_zampini } 4713b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4721e6b0712SBarry Smith 4733b03a366Sstefano_zampini #undef __FUNCT__ 47482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 47582ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4760c7d97c5SJed Brown { 4770c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 478*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 4790c7d97c5SJed Brown PetscErrorCode ierr; 4800c7d97c5SJed Brown 4810c7d97c5SJed Brown PetscFunctionBegin; 482*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 483*56282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 484*56282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 485*56282151SStefano Zampini } 486785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 487785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4880c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 489785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 490*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 4910c7d97c5SJed Brown PetscFunctionReturn(0); 4920c7d97c5SJed Brown } 4930c7d97c5SJed Brown 4940c7d97c5SJed Brown #undef __FUNCT__ 49582ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4969c0446d6SStefano Zampini /*@ 49782ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4989c0446d6SStefano Zampini 499785d1243SStefano Zampini Collective 50053cdbc3dSStefano Zampini 50153cdbc3dSStefano Zampini Input Parameters: 50253cdbc3dSStefano Zampini + pc - the preconditioning context 50382ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 50453cdbc3dSStefano Zampini 50553cdbc3dSStefano Zampini Level: intermediate 50653cdbc3dSStefano Zampini 5079c0446d6SStefano Zampini Notes: 50853cdbc3dSStefano Zampini 5090f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 51053cdbc3dSStefano Zampini @*/ 51182ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 5120c7d97c5SJed Brown { 5130c7d97c5SJed Brown PetscErrorCode ierr; 5140c7d97c5SJed Brown 5150c7d97c5SJed Brown PetscFunctionBegin; 5160c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5170c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 51882ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 51982ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 5200c7d97c5SJed Brown PetscFunctionReturn(0); 5210c7d97c5SJed Brown } 5220c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 5230c7d97c5SJed Brown 5240c7d97c5SJed Brown #undef __FUNCT__ 5250c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 52653cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 5270c7d97c5SJed Brown { 5280c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 529*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 53053cdbc3dSStefano Zampini PetscErrorCode ierr; 5310c7d97c5SJed Brown 5320c7d97c5SJed Brown PetscFunctionBegin; 533*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 534*56282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 535*56282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 536*56282151SStefano Zampini } 537785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 538785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 53953cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 54036e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 541*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 5420c7d97c5SJed Brown PetscFunctionReturn(0); 5430c7d97c5SJed Brown } 5441e6b0712SBarry Smith 5450c7d97c5SJed Brown #undef __FUNCT__ 5460c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 54757527edcSJed Brown /*@ 54828509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 54957527edcSJed Brown 550785d1243SStefano Zampini Collective 55157527edcSJed Brown 55257527edcSJed Brown Input Parameters: 55357527edcSJed Brown + pc - the preconditioning context 554785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 55557527edcSJed Brown 55657527edcSJed Brown Level: intermediate 55757527edcSJed Brown 5580f202f7eSStefano Zampini Notes: 5590f202f7eSStefano Zampini Any process can list any global node 56057527edcSJed Brown 5610f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 56257527edcSJed Brown @*/ 56353cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 5640c7d97c5SJed Brown { 5650c7d97c5SJed Brown PetscErrorCode ierr; 5660c7d97c5SJed Brown 5670c7d97c5SJed Brown PetscFunctionBegin; 5680c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 569674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 570785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 57153cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 57253cdbc3dSStefano Zampini PetscFunctionReturn(0); 57353cdbc3dSStefano Zampini } 57453cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5751e6b0712SBarry Smith 57653cdbc3dSStefano Zampini #undef __FUNCT__ 57782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 57882ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5790c7d97c5SJed Brown { 5800c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 581*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 5820c7d97c5SJed Brown PetscErrorCode ierr; 5830c7d97c5SJed Brown 5840c7d97c5SJed Brown PetscFunctionBegin; 585*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 586*56282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 587*56282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 588*56282151SStefano Zampini } 589785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 590785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5910c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 592785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 593*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 5940c7d97c5SJed Brown PetscFunctionReturn(0); 5950c7d97c5SJed Brown } 5960c7d97c5SJed Brown 5970c7d97c5SJed Brown #undef __FUNCT__ 59882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5990c7d97c5SJed Brown /*@ 60082ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 6010c7d97c5SJed Brown 602785d1243SStefano Zampini Collective 6030c7d97c5SJed Brown 6040c7d97c5SJed Brown Input Parameters: 6050c7d97c5SJed Brown + pc - the preconditioning context 60682ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 6070c7d97c5SJed Brown 6080c7d97c5SJed Brown Level: intermediate 6090c7d97c5SJed Brown 6100c7d97c5SJed Brown Notes: 6110c7d97c5SJed Brown 6120f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 6130c7d97c5SJed Brown @*/ 61482ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 6150c7d97c5SJed Brown { 6160c7d97c5SJed Brown PetscErrorCode ierr; 6170c7d97c5SJed Brown 6180c7d97c5SJed Brown PetscFunctionBegin; 6190c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6200c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 62182ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 62282ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 62353cdbc3dSStefano Zampini PetscFunctionReturn(0); 62453cdbc3dSStefano Zampini } 62553cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 62653cdbc3dSStefano Zampini 62753cdbc3dSStefano Zampini #undef __FUNCT__ 628da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 629da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 630da1bb401SStefano Zampini { 631da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 632da1bb401SStefano Zampini 633da1bb401SStefano Zampini PetscFunctionBegin; 634da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 635da1bb401SStefano Zampini PetscFunctionReturn(0); 636da1bb401SStefano Zampini } 6371e6b0712SBarry Smith 638da1bb401SStefano Zampini #undef __FUNCT__ 639da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 640da1bb401SStefano Zampini /*@ 641785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 642da1bb401SStefano Zampini 643785d1243SStefano Zampini Collective 644785d1243SStefano Zampini 645785d1243SStefano Zampini Input Parameters: 646785d1243SStefano Zampini . pc - the preconditioning context 647785d1243SStefano Zampini 648785d1243SStefano Zampini Output Parameters: 649785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 650785d1243SStefano Zampini 651785d1243SStefano Zampini Level: intermediate 652785d1243SStefano Zampini 6530f202f7eSStefano Zampini Notes: 6540f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 655785d1243SStefano Zampini 656785d1243SStefano Zampini .seealso: PCBDDC 657785d1243SStefano Zampini @*/ 658785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 659785d1243SStefano Zampini { 660785d1243SStefano Zampini PetscErrorCode ierr; 661785d1243SStefano Zampini 662785d1243SStefano Zampini PetscFunctionBegin; 663785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 664785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 665785d1243SStefano Zampini PetscFunctionReturn(0); 666785d1243SStefano Zampini } 667785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 668785d1243SStefano Zampini 669785d1243SStefano Zampini #undef __FUNCT__ 670785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 671785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 672785d1243SStefano Zampini { 673785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 674785d1243SStefano Zampini 675785d1243SStefano Zampini PetscFunctionBegin; 676785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 677785d1243SStefano Zampini PetscFunctionReturn(0); 678785d1243SStefano Zampini } 679785d1243SStefano Zampini 680785d1243SStefano Zampini #undef __FUNCT__ 68182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 682da1bb401SStefano Zampini /*@ 68382ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 684da1bb401SStefano Zampini 685785d1243SStefano Zampini Collective 686da1bb401SStefano Zampini 687da1bb401SStefano Zampini Input Parameters: 68828509bceSStefano Zampini . pc - the preconditioning context 689da1bb401SStefano Zampini 690da1bb401SStefano Zampini Output Parameters: 69128509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 692da1bb401SStefano Zampini 693da1bb401SStefano Zampini Level: intermediate 694da1bb401SStefano Zampini 695da1bb401SStefano Zampini Notes: 6960f202f7eSStefano 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). 6970f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 698da1bb401SStefano Zampini 699da1bb401SStefano Zampini .seealso: PCBDDC 700da1bb401SStefano Zampini @*/ 70182ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 702da1bb401SStefano Zampini { 703da1bb401SStefano Zampini PetscErrorCode ierr; 704da1bb401SStefano Zampini 705da1bb401SStefano Zampini PetscFunctionBegin; 706da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 70782ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 708da1bb401SStefano Zampini PetscFunctionReturn(0); 709da1bb401SStefano Zampini } 710da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 7111e6b0712SBarry Smith 712da1bb401SStefano Zampini #undef __FUNCT__ 71353cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 71453cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 71553cdbc3dSStefano Zampini { 71653cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 71753cdbc3dSStefano Zampini 71853cdbc3dSStefano Zampini PetscFunctionBegin; 71953cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 72053cdbc3dSStefano Zampini PetscFunctionReturn(0); 72153cdbc3dSStefano Zampini } 7221e6b0712SBarry Smith 72353cdbc3dSStefano Zampini #undef __FUNCT__ 72453cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 72553cdbc3dSStefano Zampini /*@ 726785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 72753cdbc3dSStefano Zampini 728785d1243SStefano Zampini Collective 729785d1243SStefano Zampini 730785d1243SStefano Zampini Input Parameters: 731785d1243SStefano Zampini . pc - the preconditioning context 732785d1243SStefano Zampini 733785d1243SStefano Zampini Output Parameters: 734785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 735785d1243SStefano Zampini 736785d1243SStefano Zampini Level: intermediate 737785d1243SStefano Zampini 7380f202f7eSStefano Zampini Notes: 7390f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 740785d1243SStefano Zampini 741785d1243SStefano Zampini .seealso: PCBDDC 742785d1243SStefano Zampini @*/ 743785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 744785d1243SStefano Zampini { 745785d1243SStefano Zampini PetscErrorCode ierr; 746785d1243SStefano Zampini 747785d1243SStefano Zampini PetscFunctionBegin; 748785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 749785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 750785d1243SStefano Zampini PetscFunctionReturn(0); 751785d1243SStefano Zampini } 752785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 753785d1243SStefano Zampini 754785d1243SStefano Zampini #undef __FUNCT__ 755785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 756785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 757785d1243SStefano Zampini { 758785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 759785d1243SStefano Zampini 760785d1243SStefano Zampini PetscFunctionBegin; 761785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 762785d1243SStefano Zampini PetscFunctionReturn(0); 763785d1243SStefano Zampini } 764785d1243SStefano Zampini 765785d1243SStefano Zampini #undef __FUNCT__ 76682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 76753cdbc3dSStefano Zampini /*@ 76882ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 76953cdbc3dSStefano Zampini 770785d1243SStefano Zampini Collective 77153cdbc3dSStefano Zampini 77253cdbc3dSStefano Zampini Input Parameters: 77328509bceSStefano Zampini . pc - the preconditioning context 77453cdbc3dSStefano Zampini 77553cdbc3dSStefano Zampini Output Parameters: 77628509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 77753cdbc3dSStefano Zampini 77853cdbc3dSStefano Zampini Level: intermediate 77953cdbc3dSStefano Zampini 78053cdbc3dSStefano Zampini Notes: 7810f202f7eSStefano 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). 7820f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 78353cdbc3dSStefano Zampini 78453cdbc3dSStefano Zampini .seealso: PCBDDC 78553cdbc3dSStefano Zampini @*/ 78682ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 78753cdbc3dSStefano Zampini { 78853cdbc3dSStefano Zampini PetscErrorCode ierr; 78953cdbc3dSStefano Zampini 79053cdbc3dSStefano Zampini PetscFunctionBegin; 79153cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 79282ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7930c7d97c5SJed Brown PetscFunctionReturn(0); 7940c7d97c5SJed Brown } 79536e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7961e6b0712SBarry Smith 79736e030ebSStefano Zampini #undef __FUNCT__ 798da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7991a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 80036e030ebSStefano Zampini { 80136e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 802da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 803*56282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 804da1bb401SStefano Zampini PetscErrorCode ierr; 80536e030ebSStefano Zampini 80636e030ebSStefano Zampini PetscFunctionBegin; 807*56282151SStefano Zampini if (mat_graph->nvtxs_csr == nvtxs) { 808*56282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 809*56282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 810*56282151SStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,nvtxs*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 811*56282151SStefano Zampini } 812*56282151SStefano Zampini } 813*56282151SStefano Zampini if (!same_data) { 814674ae819SStefano Zampini /* free old CSR */ 815674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 816674ae819SStefano Zampini /* get CSR into graph structure */ 817da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 818854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 819785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 820674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 821674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 822da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 8231a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 8241a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 825674ae819SStefano Zampini } 826575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 827*56282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 828*56282151SStefano Zampini } 82936e030ebSStefano Zampini PetscFunctionReturn(0); 83036e030ebSStefano Zampini } 8311e6b0712SBarry Smith 83236e030ebSStefano Zampini #undef __FUNCT__ 833da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 83436e030ebSStefano Zampini /*@ 8350f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 83636e030ebSStefano Zampini 83736e030ebSStefano Zampini Not collective 83836e030ebSStefano Zampini 83936e030ebSStefano Zampini Input Parameters: 84036e030ebSStefano Zampini + pc - the preconditioning context 8410f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 84228509bceSStefano Zampini . xadj, adjncy - the CSR graph 84328509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 84436e030ebSStefano Zampini 84536e030ebSStefano Zampini Level: intermediate 84636e030ebSStefano Zampini 84736e030ebSStefano Zampini Notes: 84836e030ebSStefano Zampini 84928509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 85036e030ebSStefano Zampini @*/ 8511a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 85236e030ebSStefano Zampini { 853575ad6abSStefano Zampini void (*f)(void) = 0; 85436e030ebSStefano Zampini PetscErrorCode ierr; 85536e030ebSStefano Zampini 85636e030ebSStefano Zampini PetscFunctionBegin; 85736e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 858674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 859d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 860674ae819SStefano Zampini if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) { 861674ae819SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__); 862da1bb401SStefano Zampini } 86336e030ebSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 864575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 865575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 866575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 867575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 868575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 86936e030ebSStefano Zampini } 87036e030ebSStefano Zampini PetscFunctionReturn(0); 87136e030ebSStefano Zampini } 8729c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 8731e6b0712SBarry Smith 8749c0446d6SStefano Zampini #undef __FUNCT__ 87563602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 87663602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 87763602bcaSStefano Zampini { 87863602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 87963602bcaSStefano Zampini PetscInt i; 880*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 88163602bcaSStefano Zampini PetscErrorCode ierr; 88263602bcaSStefano Zampini 88363602bcaSStefano Zampini PetscFunctionBegin; 884*56282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 885*56282151SStefano Zampini for (i=0;i<n_is;i++) { 886*56282151SStefano Zampini PetscBool isequalt; 887*56282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 888*56282151SStefano Zampini if (!isequalt) break; 889*56282151SStefano Zampini } 890*56282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 891*56282151SStefano Zampini } 892*56282151SStefano Zampini for (i=0;i<n_is;i++) { 893*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 894*56282151SStefano Zampini } 89563602bcaSStefano Zampini /* Destroy ISes if they were already set */ 89663602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 89763602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 89863602bcaSStefano Zampini } 89963602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 90063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 90163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 90263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 90363602bcaSStefano Zampini } 90463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 90563602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 90663602bcaSStefano Zampini /* allocate space then set */ 907d02579f5SStefano Zampini if (n_is) { 908d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 909d02579f5SStefano Zampini } 91063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 91163602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 91263602bcaSStefano Zampini } 91363602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 91463602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 915*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 91663602bcaSStefano Zampini PetscFunctionReturn(0); 91763602bcaSStefano Zampini } 91863602bcaSStefano Zampini 91963602bcaSStefano Zampini #undef __FUNCT__ 92063602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 92163602bcaSStefano Zampini /*@ 92263602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 92363602bcaSStefano Zampini 92463602bcaSStefano Zampini Collective 92563602bcaSStefano Zampini 92663602bcaSStefano Zampini Input Parameters: 92763602bcaSStefano Zampini + pc - the preconditioning context 9280f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9290f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 93063602bcaSStefano Zampini 93163602bcaSStefano Zampini Level: intermediate 93263602bcaSStefano Zampini 9330f202f7eSStefano Zampini Notes: 9340f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 93563602bcaSStefano Zampini 93663602bcaSStefano Zampini .seealso: PCBDDC 93763602bcaSStefano Zampini @*/ 93863602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 93963602bcaSStefano Zampini { 94063602bcaSStefano Zampini PetscInt i; 94163602bcaSStefano Zampini PetscErrorCode ierr; 94263602bcaSStefano Zampini 94363602bcaSStefano Zampini PetscFunctionBegin; 94463602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 94563602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 94663602bcaSStefano Zampini for (i=0;i<n_is;i++) { 94763602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 94863602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 94963602bcaSStefano Zampini } 950e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 95163602bcaSStefano Zampini PetscFunctionReturn(0); 95263602bcaSStefano Zampini } 95363602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 95463602bcaSStefano Zampini 95563602bcaSStefano Zampini #undef __FUNCT__ 9569c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 9579c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 9589c0446d6SStefano Zampini { 9599c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 9609c0446d6SStefano Zampini PetscInt i; 961*56282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 9629c0446d6SStefano Zampini PetscErrorCode ierr; 9639c0446d6SStefano Zampini 9649c0446d6SStefano Zampini PetscFunctionBegin; 965*56282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 966*56282151SStefano Zampini for (i=0;i<n_is;i++) { 967*56282151SStefano Zampini PetscBool isequalt; 968*56282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 969*56282151SStefano Zampini if (!isequalt) break; 970*56282151SStefano Zampini } 971*56282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 972*56282151SStefano Zampini } 973*56282151SStefano Zampini for (i=0;i<n_is;i++) { 974*56282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 975*56282151SStefano Zampini } 976da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 9779c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 9789c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 9799c0446d6SStefano Zampini } 980d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 98163602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 98263602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 98363602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 98463602bcaSStefano Zampini } 98563602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 98663602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 987da1bb401SStefano Zampini /* allocate space then set */ 988d02579f5SStefano Zampini if (n_is) { 989785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 990d02579f5SStefano Zampini } 9919c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 992da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 9939c0446d6SStefano Zampini } 9949c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 99563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 996*56282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 9979c0446d6SStefano Zampini PetscFunctionReturn(0); 9989c0446d6SStefano Zampini } 9991e6b0712SBarry Smith 10009c0446d6SStefano Zampini #undef __FUNCT__ 10019c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 10029c0446d6SStefano Zampini /*@ 100363602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 10049c0446d6SStefano Zampini 100563602bcaSStefano Zampini Collective 10069c0446d6SStefano Zampini 10079c0446d6SStefano Zampini Input Parameters: 10089c0446d6SStefano Zampini + pc - the preconditioning context 10090f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10100f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 10119c0446d6SStefano Zampini 10129c0446d6SStefano Zampini Level: intermediate 10139c0446d6SStefano Zampini 10140f202f7eSStefano Zampini Notes: 10150f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 10169c0446d6SStefano Zampini 10179c0446d6SStefano Zampini .seealso: PCBDDC 10189c0446d6SStefano Zampini @*/ 10199c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 10209c0446d6SStefano Zampini { 10212b510759SStefano Zampini PetscInt i; 10229c0446d6SStefano Zampini PetscErrorCode ierr; 10239c0446d6SStefano Zampini 10249c0446d6SStefano Zampini PetscFunctionBegin; 10259c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 102663602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 10272b510759SStefano Zampini for (i=0;i<n_is;i++) { 102863602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 102963602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 10302b510759SStefano Zampini } 10319c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 10329c0446d6SStefano Zampini PetscFunctionReturn(0); 10339c0446d6SStefano Zampini } 1034906d46d4SStefano Zampini 1035da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1036534831adSStefano Zampini #undef __FUNCT__ 1037534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 1038534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1039534831adSStefano Zampini /* 1040534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1041534831adSStefano Zampini guess if a transformation of basis approach has been selected. 10429c0446d6SStefano Zampini 1043534831adSStefano Zampini Input Parameter: 1044534831adSStefano Zampini + pc - the preconditioner contex 1045534831adSStefano Zampini 1046534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1047534831adSStefano Zampini 1048534831adSStefano Zampini Notes: 1049534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1050534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1051534831adSStefano Zampini */ 1052534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1053534831adSStefano Zampini { 1054534831adSStefano Zampini PetscErrorCode ierr; 1055534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1056534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 10573972b0daSStefano Zampini Vec used_vec; 10588d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 105951694757SStefano Zampini PetscBool benign_correction_is_zero = PETSC_FALSE; 1060a3df083aSStefano Zampini PetscBool iscg = PETSC_FALSE; 1061534831adSStefano Zampini 1062534831adSStefano Zampini PetscFunctionBegin; 106385c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 106485c4d303SStefano Zampini if (ksp) { 106585c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 106685c4d303SStefano Zampini if (!iscg) { 106785c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 106885c4d303SStefano Zampini } 106985c4d303SStefano Zampini } 107085c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 107162a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 107262a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 107362a6ff1dSStefano Zampini } 107462a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 107562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 107662a6ff1dSStefano Zampini } 10778d00608fSStefano Zampini 10783972b0daSStefano Zampini if (x) { 10793972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 10803972b0daSStefano Zampini used_vec = x; 10818d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 10823972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 10833972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 10843972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 10853972b0daSStefano Zampini } 10868efcfb23SStefano Zampini 10878efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 10883972b0daSStefano Zampini if (ksp) { 1089a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 10908efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 10918efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 10923972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 10933972b0daSStefano Zampini } 10943972b0daSStefano Zampini } 10953308cffdSStefano Zampini 10968d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 10973972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 1098a07ea27aSStefano Zampini if (rhs) { 10993975b054SStefano Zampini IS dirIS = NULL; 11003975b054SStefano Zampini 1101a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 11023975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 11033975b054SStefano Zampini if (dirIS) { 1104906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1105785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 11062b095fd8SStefano Zampini PetscScalar *array_x; 11072b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1108785d1243SStefano Zampini 11093972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 11103972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1111e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1112e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1113e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1114e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 111582ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 11163972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 11172b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 11183972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 11192fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 11203972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 11212b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 11223972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1123e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1124e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 11258d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 11261b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 11278efcfb23SStefano Zampini } 1128a07ea27aSStefano Zampini } 1129b76ba322SStefano Zampini 11308efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 11318d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 11328d00608fSStefano Zampini /* store the original rhs */ 11338d00608fSStefano Zampini if (copy_rhs) { 11348d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11358d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 11368d00608fSStefano Zampini } 11378d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 11383972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 11390369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,rhs,rhs);CHKERRQ(ierr); 11403972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 11418efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 11427acc28cbSStefano Zampini if (ksp) { 11437acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 11447acc28cbSStefano Zampini } 11453308cffdSStefano Zampini } 11468efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1147b76ba322SStefano Zampini 11480369aaf7SStefano Zampini /* When using the benign trick: (TODO: what about FETI-DP?) 11490369aaf7SStefano Zampini - change rhs on pressures (iteration matrix is surely of type MATIS) 11500369aaf7SStefano Zampini - compute initial vector in benign space 11510369aaf7SStefano Zampini */ 11520369aaf7SStefano Zampini if (rhs && pcbddc->benign_saddle_point) { 115306f24817SStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 11544f1b2e48SStefano Zampini PetscInt i; 115506f24817SStefano Zampini 115606f24817SStefano Zampini /* store the original rhs */ 115706f24817SStefano Zampini if (copy_rhs) { 115806f24817SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 115906f24817SStefano Zampini copy_rhs = PETSC_FALSE; 116006f24817SStefano Zampini } 116106f24817SStefano Zampini 116206f24817SStefano Zampini /* now change (locally) the basis */ 116306f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 116406f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 116506f24817SStefano Zampini if (pcbddc->benign_change) { 116606f24817SStefano Zampini ierr = MatMultTranspose(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 116706f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 116806f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 11695cfd9691SStefano Zampini /* swap local iteration matrix (with the benign trick, amat == pmat) */ 1170a3df083aSStefano Zampini if (pcbddc->benign_n && !pcbddc->benign_change_explicit) { 1171a3df083aSStefano Zampini /* benign shell mat mult for matis->A and pcis->A_IB*/ 1172a3df083aSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 1173a3df083aSStefano Zampini } else { 11745cfd9691SStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_FALSE);CHKERRQ(ierr); 117506f24817SStefano Zampini pcbddc->benign_original_mat = matis->A; 117606f24817SStefano Zampini matis->A = pcbddc->local_mat; 1177a3df083aSStefano Zampini } 117806f24817SStefano Zampini } else { 117906f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 118006f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 118177a2de6eSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 118277a2de6eSStefano Zampini pcbddc->benign_original_mat = matis->A; 118377a2de6eSStefano Zampini matis->A = pcbddc->local_mat; 118477a2de6eSStefano Zampini } 118506f24817SStefano Zampini } 118606f24817SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 118706f24817SStefano Zampini 11880369aaf7SStefano Zampini /* compute u^*_h as in Xuemin Tu's thesis (see Section 4.8.1) */ 11890369aaf7SStefano Zampini /* TODO: what about Stokes? */ 11900369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 11910369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 11920369aaf7SStefano Zampini } 11934f1b2e48SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 11944f1b2e48SStefano Zampini if (pcbddc->benign_n) { 11950369aaf7SStefano Zampini const PetscScalar *array; 11960369aaf7SStefano Zampini 11970369aaf7SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr); 11984f1b2e48SStefano Zampini for (i=0;i<pcbddc->benign_n;i++) pcbddc->benign_p0[i] = array[pcbddc->benign_p0_lidx[i]]; 11990369aaf7SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr); 12000369aaf7SStefano Zampini } 120151694757SStefano Zampini if (pcbddc->benign_null && iscg) { /* this is a workaround, need to understand more */ 12024f1b2e48SStefano Zampini PetscBool iszero_l = PETSC_TRUE; 12034f1b2e48SStefano Zampini for (i=0;i<pcbddc->benign_n;i++) { 12044f1b2e48SStefano Zampini iszero_l = (iszero_l && (PetscAbsScalar(pcbddc->benign_p0[i]) < PETSC_SMALL ? PETSC_TRUE : PETSC_FALSE)); 12054f1b2e48SStefano Zampini } 120651694757SStefano Zampini ierr = MPI_Allreduce(&iszero_l,&benign_correction_is_zero,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 120751694757SStefano Zampini } 12084fee134fSStefano Zampini if (!benign_correction_is_zero) { /* use the coarse solver only */ 1209015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr); 12104fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 12110369aaf7SStefano Zampini ierr = PCApply_BDDC(pc,pcis->vec1_global,pcbddc->benign_vec);CHKERRQ(ierr); 12124fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 12134f1b2e48SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 1214015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,pcbddc->benign_vec,PETSC_FALSE);CHKERRQ(ierr); 1215b76ba322SStefano Zampini } 121651694757SStefano Zampini } 1217b76ba322SStefano Zampini 12180369aaf7SStefano Zampini /* change rhs and iteration matrix if using the change of basis */ 1219b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1220906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1221906d46d4SStefano Zampini 1222906d46d4SStefano Zampini /* get change ctx */ 1223906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1224906d46d4SStefano Zampini 1225906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 122677a2de6eSStefano Zampini if (!pcbddc->benign_original_mat) { 1227906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1228906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1229906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1230906d46d4SStefano Zampini 1231906d46d4SStefano Zampini /* change iteration matrix */ 1232906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1233906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1234906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 12359bd2e3c7SStefano Zampini } 12368d00608fSStefano Zampini /* store the original rhs */ 12378d00608fSStefano Zampini if (copy_rhs) { 12388d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 12398d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 12408d00608fSStefano Zampini } 12418d00608fSStefano Zampini 1242906d46d4SStefano Zampini /* change rhs */ 1243906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1244906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 12458d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 124692e3dcfbSStefano Zampini } 124792e3dcfbSStefano Zampini 12480369aaf7SStefano Zampini /* remove non-benign solution from the rhs */ 12490369aaf7SStefano Zampini if (pcbddc->benign_saddle_point) { 12500369aaf7SStefano Zampini /* store the original rhs */ 12510369aaf7SStefano Zampini if (copy_rhs) { 12520369aaf7SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 12530369aaf7SStefano Zampini copy_rhs = PETSC_FALSE; 12540369aaf7SStefano Zampini } 125551694757SStefano Zampini if (benign_correction_is_zero) { /* still need to understand why it works great */ 125651694757SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 125751694757SStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 125851694757SStefano Zampini } 12590369aaf7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 12600369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 12610369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12620369aaf7SStefano Zampini } 12630369aaf7SStefano Zampini 12640369aaf7SStefano Zampini /* set initial guess if using PCG */ 12650369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 12660369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 12670369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12680369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12690369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 12700369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12710369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12720369aaf7SStefano Zampini if (ksp) { 12730369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 12740369aaf7SStefano Zampini } 12750369aaf7SStefano Zampini } 1276b0f5fe93SStefano Zampini if (pcbddc->benign_null) { 1277b0f5fe93SStefano Zampini MatNullSpace null_space; 1278b0f5fe93SStefano Zampini Vec nullv; 1279b0f5fe93SStefano Zampini PetscBool isnull; 12804f1b2e48SStefano Zampini PetscInt i; 1281b0f5fe93SStefano Zampini 12824f1b2e48SStefano Zampini for (i=0;i<pcbddc->benign_n;i++) pcbddc->benign_p0[i] = 1.; 1283b0f5fe93SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&nullv);CHKERRQ(ierr); 1284b0f5fe93SStefano Zampini ierr = VecSet(nullv,0.);CHKERRQ(ierr); 1285b0f5fe93SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,nullv,PETSC_FALSE);CHKERRQ(ierr); 1286b0f5fe93SStefano Zampini ierr = VecNormalize(nullv,NULL);CHKERRQ(ierr); 1287b0f5fe93SStefano Zampini ierr = MatNullSpaceCreate(PetscObjectComm((PetscObject)pc),PETSC_FALSE,1,&nullv,&null_space);CHKERRQ(ierr); 1288b0f5fe93SStefano Zampini ierr = MatNullSpaceTest(null_space,pc->mat,&isnull);CHKERRQ(ierr); 12894f1b2e48SStefano Zampini if (isnull) { 1290b0f5fe93SStefano Zampini ierr = MatSetNullSpace(pc->mat,null_space);CHKERRQ(ierr); 12914f1b2e48SStefano Zampini } 1292b0f5fe93SStefano Zampini ierr = MatNullSpaceDestroy(&null_space);CHKERRQ(ierr); 1293b0f5fe93SStefano Zampini ierr = VecDestroy(&nullv);CHKERRQ(ierr); 1294b0f5fe93SStefano Zampini } 1295b0f5fe93SStefano Zampini 129692e3dcfbSStefano Zampini /* remove nullspace if present */ 12978efcfb23SStefano Zampini if (ksp && x && pcbddc->NullSpace) { 12988efcfb23SStefano Zampini ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr); 12998d00608fSStefano Zampini /* store the original rhs */ 13008d00608fSStefano Zampini if (copy_rhs) { 13018d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 13028d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 13038d00608fSStefano Zampini } 13048d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 1305d0195637SJed Brown ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr); 1306b76ba322SStefano Zampini } 1307534831adSStefano Zampini PetscFunctionReturn(0); 1308534831adSStefano Zampini } 1309906d46d4SStefano Zampini 1310534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1311534831adSStefano Zampini #undef __FUNCT__ 1312534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1313534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1314534831adSStefano Zampini /* 1315534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1316534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1317534831adSStefano Zampini 1318534831adSStefano Zampini Input Parameter: 1319534831adSStefano Zampini + pc - the preconditioner contex 1320534831adSStefano Zampini 1321534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1322534831adSStefano Zampini 1323534831adSStefano Zampini Notes: 1324534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1325534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1326534831adSStefano Zampini */ 1327534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1328534831adSStefano Zampini { 1329534831adSStefano Zampini PetscErrorCode ierr; 1330534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1331534831adSStefano Zampini 1332534831adSStefano Zampini PetscFunctionBegin; 133377a2de6eSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && !pcbddc->benign_original_mat) { 1334906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1335906d46d4SStefano Zampini 1336906d46d4SStefano Zampini /* get change ctx */ 1337906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1338906d46d4SStefano Zampini 1339906d46d4SStefano Zampini /* restore iteration matrix */ 1340906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1341906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1342906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 13439c6a02ceSStefano Zampini } 1344906d46d4SStefano Zampini 13458666afb4SStefano Zampini /* need to restore the local matrices */ 134677a2de6eSStefano Zampini if (pcbddc->benign_original_mat) { 13478666afb4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->mat->data; 13488666afb4SStefano Zampini 1349a3df083aSStefano Zampini if (pcbddc->benign_n && !pcbddc->benign_change_explicit) { 1350a3df083aSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1351a3df083aSStefano Zampini } else { 13525cfd9691SStefano Zampini pcbddc->local_mat = matis->A; 13538666afb4SStefano Zampini matis->A = pcbddc->benign_original_mat; 13545cfd9691SStefano Zampini pcbddc->benign_original_mat = NULL; 13558666afb4SStefano Zampini } 1356a3df083aSStefano Zampini } 13578666afb4SStefano Zampini 1358906d46d4SStefano Zampini /* get solution in original basis */ 1359906d46d4SStefano Zampini if (x) { 1360906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 136106f24817SStefano Zampini 13622d3346b4SStefano Zampini /* restore solution on pressures */ 136306a4e24aSStefano Zampini if (pcbddc->benign_saddle_point) { 13642d3346b4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->mat->data; 13652d3346b4SStefano Zampini 13660369aaf7SStefano Zampini /* add non-benign solution */ 13678666afb4SStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13680369aaf7SStefano Zampini 13690369aaf7SStefano Zampini /* change basis on pressures for x */ 137006f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 137106f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13722d3346b4SStefano Zampini if (pcbddc->benign_change) { 13732d3346b4SStefano Zampini ierr = MatMult(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 137406f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 137506f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13762d3346b4SStefano Zampini } else { 137706f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 137806f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13792d3346b4SStefano Zampini } 13802d3346b4SStefano Zampini } 13819c6a02ceSStefano Zampini 13820369aaf7SStefano Zampini /* change basis on x */ 13839c6a02ceSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 13849c6a02ceSStefano Zampini PCBDDCChange_ctx change_ctx; 13859c6a02ceSStefano Zampini 13869c6a02ceSStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 13870369aaf7SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1388906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 13893425bc38SStefano Zampini } 1390534831adSStefano Zampini } 1391906d46d4SStefano Zampini 13923972b0daSStefano Zampini /* add solution removed in presolve */ 13936bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 13943425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 13953425bc38SStefano Zampini } 1396906d46d4SStefano Zampini 1397fb223d50SStefano Zampini /* restore rhs to its original state */ 13988d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1399fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1400fb223d50SStefano Zampini } 14018d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 14028efcfb23SStefano Zampini 14038efcfb23SStefano Zampini /* restore ksp guess state */ 14048efcfb23SStefano Zampini if (ksp) { 14058efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 14068efcfb23SStefano Zampini } 1407534831adSStefano Zampini PetscFunctionReturn(0); 1408534831adSStefano Zampini } 1409534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 141053cdbc3dSStefano Zampini #undef __FUNCT__ 141153cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 14120c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 14130c7d97c5SJed Brown /* 14140c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 14150c7d97c5SJed Brown by setting data structures and options. 14160c7d97c5SJed Brown 14170c7d97c5SJed Brown Input Parameter: 141853cdbc3dSStefano Zampini + pc - the preconditioner context 14190c7d97c5SJed Brown 14200c7d97c5SJed Brown Application Interface Routine: PCSetUp() 14210c7d97c5SJed Brown 14220c7d97c5SJed Brown Notes: 14230c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 14240c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 14250c7d97c5SJed Brown */ 142653cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 14270c7d97c5SJed Brown { 14280c7d97c5SJed Brown PetscErrorCode ierr; 14290c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 14305e8657edSStefano Zampini Mat_IS* matis; 143108122e43SStefano Zampini MatNullSpace nearnullspace; 1432c9ed8603SStefano Zampini IS zerodiag = NULL; 143391e8d312SStefano Zampini PetscInt nrows,ncols; 143408122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 14358de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 14365e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 14370c7d97c5SJed Brown 14380c7d97c5SJed Brown PetscFunctionBegin; 14395e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 14405e8657edSStefano Zampini if (!ismatis) { 14415e8657edSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 14425e8657edSStefano Zampini } 144391e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 144491e8d312SStefano Zampini if (nrows != ncols) { 144591e8d312SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 144691e8d312SStefano Zampini } 14475e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1448f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 14493b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1450fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1451f4ddd8eeSStefano Zampini /* split work */ 1452674ae819SStefano Zampini if (pc->setupcalled) { 1453d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1454674ae819SStefano Zampini computetopography = PETSC_FALSE; 1455674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1456674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1457674ae819SStefano Zampini computetopography = PETSC_TRUE; 1458674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1459674ae819SStefano Zampini } 1460674ae819SStefano Zampini } else { 1461674ae819SStefano Zampini computetopography = PETSC_TRUE; 1462674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1463674ae819SStefano Zampini } 1464fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1465fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1466fb180af4SStefano Zampini } 1467*56282151SStefano Zampini pcbddc->recompute_topography = computetopography; 14688de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1469b087196eSStefano Zampini 1470b087196eSStefano Zampini /* check parameters' compatibility */ 14715a95e1ceSStefano Zampini if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) { 14725a95e1ceSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 14735a95e1ceSStefano Zampini } 14745a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1475bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1476862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1477862806e4SStefano Zampini 14785a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 14796816873aSStefano Zampini if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) { 14806816873aSStefano 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"); 14816816873aSStefano Zampini } 14822d3346b4SStefano Zampini 14832d3346b4SStefano Zampini /* check if the iteration matrix is of type MATIS in case the benign trick has been requested */ 14842d3346b4SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 148506a4e24aSStefano Zampini if (pcbddc->benign_saddle_point && !ismatis) { 14862d3346b4SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner with benign subspace trick requires the iteration matrix to be of type MATIS"); 14872d3346b4SStefano Zampini } 14885cfd9691SStefano Zampini if (pcbddc->benign_saddle_point && pc->mat != pc->pmat) { 148916e386b8SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner with benign subspace trick requires Amat == Pmat"); 14905cfd9691SStefano Zampini } 14912d3346b4SStefano Zampini 1492f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 149370cf5478SStefano Zampini if (pcbddc->dbg_flag) { 149470cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 149558a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 14961575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1497f4ddd8eeSStefano Zampini } 149858a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1499f4ddd8eeSStefano Zampini } 1500f4ddd8eeSStefano Zampini 15015e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 15025e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 15035e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 15045e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 15055e8657edSStefano Zampini } else { 1506b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15075e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15085e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1509d16cbb6bSStefano Zampini } 1510d16cbb6bSStefano Zampini 15114f1b2e48SStefano Zampini /* detect local disconnected subdomains if requested and not done before */ 15124f1b2e48SStefano Zampini if (pcbddc->detect_disconnected && !pcbddc->n_local_subs) { 15134f1b2e48SStefano Zampini ierr = MatDetectDisconnectedComponents(pcbddc->local_mat,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 15144f1b2e48SStefano Zampini } 15154f1b2e48SStefano Zampini 15164f1b2e48SStefano Zampini /* 15174f1b2e48SStefano Zampini change basis on local pressures (aka zerodiag dofs) 15184f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 15194f1b2e48SStefano Zampini */ 1520d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 15219f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 15229f47a83aSStefano Zampini 1523a3df083aSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis) pcbddc->benign_change_explicit = PETSC_TRUE; 1524339f8db1SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat */ 1525339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1526a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1527c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 15289f47a83aSStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 1529a3df083aSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcbddc->benign_n && !pcbddc->benign_change_explicit && !pcbddc->dbg_flag) { 1530a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 1531a3df083aSStefano Zampini } else { 15329f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1533674ae819SStefano Zampini } 1534a3df083aSStefano Zampini } 1535f1a72664SStefano Zampini /* propagate relevant information */ 15364f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 15373301b35fSStefano Zampini if (matis->A->symmetric_set) { 15383301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 15393301b35fSStefano Zampini } 1540e496cd5dSStefano Zampini #endif 154106a4e24aSStefano Zampini if (matis->A->symmetric_set) { 154206a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 154306a4e24aSStefano Zampini } 154406a4e24aSStefano Zampini if (matis->A->spd_set) { 154506a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 154606a4e24aSStefano Zampini } 1547e496cd5dSStefano Zampini 15485e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 15495e8657edSStefano Zampini { 15505e8657edSStefano Zampini Mat temp_mat; 15515e8657edSStefano Zampini 15525e8657edSStefano Zampini temp_mat = matis->A; 15535e8657edSStefano Zampini matis->A = pcbddc->local_mat; 15545e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 15555e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15565e8657edSStefano Zampini matis->A = temp_mat; 15575e8657edSStefano Zampini } 1558684f6988SStefano Zampini 155981d14e9dSStefano Zampini /* Analyze interface */ 1560674ae819SStefano Zampini if (computetopography) { 1561674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15628de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1563674ae819SStefano Zampini } 1564fb8d54d4SStefano Zampini 15655408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 15665408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 15675408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 15685408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 156909f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 15705408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 157109f581a4SStefano Zampini } 15725408967cSStefano Zampini #endif 15734f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 157406f24817SStefano Zampini 1575b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1576684f6988SStefano Zampini if (computesolvers) { 1577d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1578d5574798SStefano Zampini 1579d5574798SStefano Zampini if (computesubschurs && computetopography) { 158008122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1581b1b3d7a2SStefano Zampini } 1582df4d28bfSStefano Zampini if (sub_schurs->schur_explicit) { 15832070dbb6SStefano Zampini if (computesubschurs) { 158408122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 15852070dbb6SStefano Zampini } 1586d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1587d5574798SStefano Zampini } else { 1588d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 15892070dbb6SStefano Zampini if (computesubschurs) { 1590d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1591d5574798SStefano Zampini } 15922070dbb6SStefano Zampini } 159308122e43SStefano Zampini if (pcbddc->adaptive_selection) { 159408122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 15958de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1596b7eb3628SStefano Zampini } 1597b1b3d7a2SStefano Zampini } 1598684f6988SStefano Zampini 1599f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1600fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1601f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1602f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1603f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1604f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1605f4ddd8eeSStefano Zampini } else { 1606f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1607f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1608f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1609165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1610f4ddd8eeSStefano Zampini PetscInt i; 1611165b64e2SStefano Zampini const Vec *nearnullvecs; 1612165b64e2SStefano Zampini PetscObjectState state; 1613165b64e2SStefano Zampini PetscInt nnsp_size; 1614165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1615f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1616f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1617165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1618f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1619f4ddd8eeSStefano Zampini break; 1620f4ddd8eeSStefano Zampini } 1621f4ddd8eeSStefano Zampini } 1622f4ddd8eeSStefano Zampini } 1623f4ddd8eeSStefano Zampini } 1624f4ddd8eeSStefano Zampini } else { 1625f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1626f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1627f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1628f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1629f4ddd8eeSStefano Zampini } 1630f4ddd8eeSStefano Zampini } 1631f4ddd8eeSStefano Zampini 1632f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1633727cdba6SStefano Zampini /* reset primal space flags */ 1634f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1635727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 16368de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1637727cdba6SStefano Zampini /* It also sets the primal space flags */ 1638674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1639e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1640f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 16413975b054SStefano Zampini } 16425e8657edSStefano Zampini 16433975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 16445e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16455e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1646c263805aSStefano Zampini Mat temp_mat = NULL; 16475e8657edSStefano Zampini 16484f1b2e48SStefano Zampini if (pcbddc->benign_change) { 1649c263805aSStefano Zampini /* insert B0 in pcbddc->local_mat */ 1650c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_FALSE);CHKERRQ(ierr); 1651f1a72664SStefano Zampini /* swap local matrices */ 1652f1a72664SStefano Zampini ierr = MatISGetLocalMat(pc->pmat,&temp_mat);CHKERRQ(ierr); 1653f1a72664SStefano Zampini ierr = PetscObjectReference((PetscObject)temp_mat);CHKERRQ(ierr); 1654f1a72664SStefano Zampini ierr = MatISSetLocalMat(pc->pmat,pcbddc->local_mat);CHKERRQ(ierr); 1655f1a72664SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 1656c263805aSStefano Zampini } 16575e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16584f1b2e48SStefano Zampini if (pcbddc->benign_change) { 1659c263805aSStefano Zampini /* restore original matrix */ 1660f1a72664SStefano Zampini ierr = MatISSetLocalMat(pc->pmat,temp_mat);CHKERRQ(ierr); 1661f1a72664SStefano Zampini ierr = PetscObjectDereference((PetscObject)temp_mat);CHKERRQ(ierr); 1662c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1663c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1664c263805aSStefano Zampini } 16655e8657edSStefano Zampini /* get submatrices */ 16665e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16675e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16685e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16695e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16705e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16715e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16723975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16733975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16749c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1675b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16765e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16775e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16785e8657edSStefano Zampini } 1679b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 168099cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1681b96c3477SStefano Zampini /* SetUp Scaling operator */ 1682674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16830c7d97c5SJed Brown } 1684*56282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 16850369aaf7SStefano Zampini 168658a03d70SStefano Zampini if (pcbddc->dbg_flag) { 168758a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 16882b510759SStefano Zampini } 16890c7d97c5SJed Brown PetscFunctionReturn(0); 16900c7d97c5SJed Brown } 16910c7d97c5SJed Brown 16920c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 16930c7d97c5SJed Brown /* 169450efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 16950c7d97c5SJed Brown 16960c7d97c5SJed Brown Input Parameters: 16970f202f7eSStefano Zampini + pc - the preconditioner context 16980f202f7eSStefano Zampini - r - input vector (global) 16990c7d97c5SJed Brown 17000c7d97c5SJed Brown Output Parameter: 17010c7d97c5SJed Brown . z - output vector (global) 17020c7d97c5SJed Brown 17030c7d97c5SJed Brown Application Interface Routine: PCApply() 17040c7d97c5SJed Brown */ 17050c7d97c5SJed Brown #undef __FUNCT__ 17060c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 170753cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 17080c7d97c5SJed Brown { 17090c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 17100c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1711b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 17120c7d97c5SJed Brown PetscErrorCode ierr; 17133b03a366Sstefano_zampini const PetscScalar one = 1.0; 17143b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 17152617d88aSStefano Zampini const PetscScalar zero = 0.0; 17160c7d97c5SJed Brown 17170c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 17180c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1719b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 17200c7d97c5SJed Brown 17210c7d97c5SJed Brown PetscFunctionBegin; 1722015636ebSStefano Zampini if (pcbddc->benign_saddle_point) { /* get p0 from r */ 1723015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1724efc2fbd9SStefano Zampini } 172585c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1726b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 17270c7d97c5SJed Brown /* First Dirichlet solve */ 17280c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17290c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17300c7d97c5SJed Brown /* 17310c7d97c5SJed Brown Assembling right hand side for BDDC operator 1732b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1733674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 17340c7d97c5SJed Brown */ 1735b097fa66SStefano Zampini if (n_D) { 1736b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 17370c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 17388eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1739b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1740b097fa66SStefano Zampini } else { 1741b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1742b097fa66SStefano Zampini } 17430c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17440c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1745674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1746b76ba322SStefano Zampini } else { 17474fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1748b097fa66SStefano Zampini if (pcbddc->switch_static) { 17490bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1750b097fa66SStefano Zampini } 1751674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1752b76ba322SStefano Zampini } 17534fee134fSStefano Zampini } 1754b76ba322SStefano Zampini 17552617d88aSStefano Zampini /* Apply interface preconditioner 17562617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1757dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 17582617d88aSStefano Zampini 1759674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1760674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 17610c7d97c5SJed Brown 17623b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 17630c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17640c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1765b097fa66SStefano Zampini if (n_B) { 17660c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 17678eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1768b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1769b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1770b097fa66SStefano Zampini } 1771df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1772efc2fbd9SStefano Zampini 1773b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1774b097fa66SStefano Zampini if (pcbddc->switch_static) { 1775b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1776b097fa66SStefano Zampini } else { 1777b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1778b097fa66SStefano Zampini } 17790c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17800c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1781b097fa66SStefano Zampini } else { 1782b097fa66SStefano Zampini if (pcbddc->switch_static) { 1783b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1784b097fa66SStefano Zampini } else { 1785b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1786b097fa66SStefano Zampini } 1787b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1788b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1789b097fa66SStefano Zampini } 1790efc2fbd9SStefano Zampini 1791015636ebSStefano Zampini if (pcbddc->benign_saddle_point) { /* set p0 (computed in PCBDDCApplyInterface) */ 1792015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1793efc2fbd9SStefano Zampini } 17940c7d97c5SJed Brown PetscFunctionReturn(0); 17950c7d97c5SJed Brown } 179650efa1b5SStefano Zampini 179750efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 179850efa1b5SStefano Zampini /* 179950efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 180050efa1b5SStefano Zampini 180150efa1b5SStefano Zampini Input Parameters: 18020f202f7eSStefano Zampini + pc - the preconditioner context 18030f202f7eSStefano Zampini - r - input vector (global) 180450efa1b5SStefano Zampini 180550efa1b5SStefano Zampini Output Parameter: 180650efa1b5SStefano Zampini . z - output vector (global) 180750efa1b5SStefano Zampini 180850efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 180950efa1b5SStefano Zampini */ 181050efa1b5SStefano Zampini #undef __FUNCT__ 181150efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 181250efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 181350efa1b5SStefano Zampini { 181450efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 181550efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1816b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 181750efa1b5SStefano Zampini PetscErrorCode ierr; 181850efa1b5SStefano Zampini const PetscScalar one = 1.0; 181950efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 182050efa1b5SStefano Zampini const PetscScalar zero = 0.0; 182150efa1b5SStefano Zampini 182250efa1b5SStefano Zampini PetscFunctionBegin; 1823537c1cdfSStefano Zampini if (pcbddc->benign_saddle_point) { /* get p0 from r */ 1824537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1825537c1cdfSStefano Zampini } 182650efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1827b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 182850efa1b5SStefano Zampini /* First Dirichlet solve */ 182950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183150efa1b5SStefano Zampini /* 183250efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1833b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 183450efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 183550efa1b5SStefano Zampini */ 1836b097fa66SStefano Zampini if (n_D) { 1837b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 183850efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 183950efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1840b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1841b097fa66SStefano Zampini } else { 1842b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1843b097fa66SStefano Zampini } 184450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 184550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 184650efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 184750efa1b5SStefano Zampini } else { 1848b097fa66SStefano Zampini if (pcbddc->switch_static) { 184950efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1850b097fa66SStefano Zampini } 185150efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 185250efa1b5SStefano Zampini } 185350efa1b5SStefano Zampini 185450efa1b5SStefano Zampini /* Apply interface preconditioner 185550efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1856dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 185750efa1b5SStefano Zampini 185850efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 185950efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 186050efa1b5SStefano Zampini 186150efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 186250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1864b097fa66SStefano Zampini if (n_B) { 186550efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 186650efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1867b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1868b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1869b097fa66SStefano Zampini } 1870b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1871b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1872b097fa66SStefano Zampini if (pcbddc->switch_static) { 1873b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1874b097fa66SStefano Zampini } else { 1875b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1876b097fa66SStefano Zampini } 187750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 187850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1879b097fa66SStefano Zampini } else { 1880b097fa66SStefano Zampini if (pcbddc->switch_static) { 1881b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1882b097fa66SStefano Zampini } else { 1883b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1884b097fa66SStefano Zampini } 1885b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1886b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1887b097fa66SStefano Zampini } 1888537c1cdfSStefano Zampini if (pcbddc->benign_saddle_point) { /* set p0 (computed in PCBDDCApplyInterface) */ 1889537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1890537c1cdfSStefano Zampini } 189150efa1b5SStefano Zampini PetscFunctionReturn(0); 189250efa1b5SStefano Zampini } 1893da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1894674ae819SStefano Zampini 1895da1bb401SStefano Zampini #undef __FUNCT__ 1896da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1897da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1898da1bb401SStefano Zampini { 1899da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1900da1bb401SStefano Zampini PetscErrorCode ierr; 1901da1bb401SStefano Zampini 1902da1bb401SStefano Zampini PetscFunctionBegin; 1903da1bb401SStefano Zampini /* free data created by PCIS */ 1904da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 1905674ae819SStefano Zampini /* free BDDC custom data */ 1906674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1907674ae819SStefano Zampini /* destroy objects related to topography */ 1908674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1909674ae819SStefano Zampini /* free allocated graph structure */ 1910da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1911b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1912b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 191334a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1914674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 191534a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1916674ae819SStefano Zampini /* free solvers stuff */ 1917674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 191862a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 191962a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 192062a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1921906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 1922906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 1923906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1924906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1925906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1926906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1927906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1928906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1929906d46d4SStefano Zampini } 1930906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 19313425bc38SStefano Zampini /* remove functions */ 1932906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1933674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 193430368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 1935bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 19362b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1937b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 19382b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1939bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 1940bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 194182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1942bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 194382ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1944bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 194582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1946bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1947785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1948bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 194963602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1950bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1951bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1952bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1953bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1954674ae819SStefano Zampini /* Free the private data structure */ 1955674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1956da1bb401SStefano Zampini PetscFunctionReturn(0); 1957da1bb401SStefano Zampini } 19583425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 19591e6b0712SBarry Smith 19603425bc38SStefano Zampini #undef __FUNCT__ 19613425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 19623425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 19633425bc38SStefano Zampini { 1964674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1965c08af4c6SStefano Zampini Vec copy_standard_rhs; 19663425bc38SStefano Zampini PC_IS* pcis; 19673425bc38SStefano Zampini PC_BDDC* pcbddc; 19683425bc38SStefano Zampini PetscErrorCode ierr; 19690c7d97c5SJed Brown 19703425bc38SStefano Zampini PetscFunctionBegin; 19713425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 19723425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 19733425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 19743425bc38SStefano Zampini 1975c08af4c6SStefano Zampini /* 1976c08af4c6SStefano Zampini change of basis for physical rhs if needed 1977c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1978c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1979c08af4c6SStefano Zampini */ 1980c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1981c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1982c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 19833425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1984c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1985c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1986fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1987fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1988c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1989c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1990674ae819SStefano Zampini /* Apply partition of unity */ 19913425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1992c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 19938eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 19943425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 19953425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 19963425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 19973425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1998c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1999c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2000c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2001c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2002c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2003c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 20043425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 20053425bc38SStefano Zampini } 2006c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 20073425bc38SStefano Zampini /* BDDC rhs */ 20083425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 20098eeda7d8SStefano Zampini if (pcbddc->switch_static) { 20103425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20113425bc38SStefano Zampini } 20123425bc38SStefano Zampini /* apply BDDC */ 2013dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 20143425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 20153425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 20163425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 20173425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 20183425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 20193425bc38SStefano Zampini PetscFunctionReturn(0); 20203425bc38SStefano Zampini } 20211e6b0712SBarry Smith 20223425bc38SStefano Zampini #undef __FUNCT__ 20233425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 20243425bc38SStefano Zampini /*@ 20250f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 20263425bc38SStefano Zampini 20273425bc38SStefano Zampini Collective 20283425bc38SStefano Zampini 20293425bc38SStefano Zampini Input Parameters: 20300f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 20310f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 20323425bc38SStefano Zampini 20333425bc38SStefano Zampini Output Parameters: 20340f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 20353425bc38SStefano Zampini 20363425bc38SStefano Zampini Level: developer 20373425bc38SStefano Zampini 20383425bc38SStefano Zampini Notes: 20393425bc38SStefano Zampini 20400f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 20413425bc38SStefano Zampini @*/ 20423425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 20433425bc38SStefano Zampini { 2044674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 20453425bc38SStefano Zampini PetscErrorCode ierr; 20463425bc38SStefano Zampini 20473425bc38SStefano Zampini PetscFunctionBegin; 20483425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20493425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 20503425bc38SStefano Zampini PetscFunctionReturn(0); 20513425bc38SStefano Zampini } 20523425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20531e6b0712SBarry Smith 20543425bc38SStefano Zampini #undef __FUNCT__ 20553425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 20563425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 20573425bc38SStefano Zampini { 2058674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 20593425bc38SStefano Zampini PC_IS* pcis; 20603425bc38SStefano Zampini PC_BDDC* pcbddc; 20613425bc38SStefano Zampini PetscErrorCode ierr; 20623425bc38SStefano Zampini 20633425bc38SStefano Zampini PetscFunctionBegin; 20643425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20653425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20663425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20673425bc38SStefano Zampini 20683425bc38SStefano Zampini /* apply B_delta^T */ 20693425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20703425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20713425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 20723425bc38SStefano Zampini /* compute rhs for BDDC application */ 20733425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 20748eeda7d8SStefano Zampini if (pcbddc->switch_static) { 20753425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20763425bc38SStefano Zampini } 20773425bc38SStefano Zampini /* apply BDDC */ 2078dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 20793425bc38SStefano Zampini /* put values into standard global vector */ 20803425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20813425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20828eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 20833425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 20843425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 20853425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 20863425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20873425bc38SStefano Zampini } 20883425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20893425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20903425bc38SStefano Zampini /* final change of basis if needed 20913425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 20923308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 20933425bc38SStefano Zampini PetscFunctionReturn(0); 20943425bc38SStefano Zampini } 20951e6b0712SBarry Smith 20963425bc38SStefano Zampini #undef __FUNCT__ 20973425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 20983425bc38SStefano Zampini /*@ 20990f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 21003425bc38SStefano Zampini 21013425bc38SStefano Zampini Collective 21023425bc38SStefano Zampini 21033425bc38SStefano Zampini Input Parameters: 21040f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 21050f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 21063425bc38SStefano Zampini 21073425bc38SStefano Zampini Output Parameters: 21080f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 21093425bc38SStefano Zampini 21103425bc38SStefano Zampini Level: developer 21113425bc38SStefano Zampini 21123425bc38SStefano Zampini Notes: 21133425bc38SStefano Zampini 21140f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 21153425bc38SStefano Zampini @*/ 21163425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 21173425bc38SStefano Zampini { 2118674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21193425bc38SStefano Zampini PetscErrorCode ierr; 21203425bc38SStefano Zampini 21213425bc38SStefano Zampini PetscFunctionBegin; 21223425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21233425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 21243425bc38SStefano Zampini PetscFunctionReturn(0); 21253425bc38SStefano Zampini } 21263425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 21271e6b0712SBarry Smith 2128f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2129edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2130f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2131f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2132edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2133f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2134674ae819SStefano Zampini 21353425bc38SStefano Zampini #undef __FUNCT__ 21363425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 21373425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 21383425bc38SStefano Zampini { 2139674ae819SStefano Zampini 2140674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 21413425bc38SStefano Zampini Mat newmat; 2142674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 21433425bc38SStefano Zampini PC newpc; 2144ce94432eSBarry Smith MPI_Comm comm; 21453425bc38SStefano Zampini PetscErrorCode ierr; 21463425bc38SStefano Zampini 21473425bc38SStefano Zampini PetscFunctionBegin; 2148ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 21493425bc38SStefano Zampini /* FETIDP linear matrix */ 21503425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 21513425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 21523425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 21533425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2154edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 21553425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 21563425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 21573425bc38SStefano Zampini /* FETIDP preconditioner */ 21583425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 21593425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 21603425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 21613425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 21623425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 21633425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2164edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 21653425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 216623ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 21673425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 21683425bc38SStefano Zampini /* return pointers for objects created */ 21693425bc38SStefano Zampini *fetidp_mat=newmat; 21703425bc38SStefano Zampini *fetidp_pc=newpc; 21713425bc38SStefano Zampini PetscFunctionReturn(0); 21723425bc38SStefano Zampini } 21731e6b0712SBarry Smith 21743425bc38SStefano Zampini #undef __FUNCT__ 21753425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 21763425bc38SStefano Zampini /*@ 21770f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 21783425bc38SStefano Zampini 21793425bc38SStefano Zampini Collective 21803425bc38SStefano Zampini 21813425bc38SStefano Zampini Input Parameters: 21820f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 218328509bceSStefano Zampini 218428509bceSStefano Zampini Output Parameters: 21850f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 21860f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 218728509bceSStefano Zampini 218828509bceSStefano Zampini Options Database Keys: 21890f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 21903425bc38SStefano Zampini 21913425bc38SStefano Zampini Level: developer 21923425bc38SStefano Zampini 21933425bc38SStefano Zampini Notes: 21940f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 21953425bc38SStefano Zampini 21960f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 21973425bc38SStefano Zampini @*/ 21983425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 21993425bc38SStefano Zampini { 22003425bc38SStefano Zampini PetscErrorCode ierr; 22013425bc38SStefano Zampini 22023425bc38SStefano Zampini PetscFunctionBegin; 22033425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 22043425bc38SStefano Zampini if (pc->setupcalled) { 2205516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2206f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 22073425bc38SStefano Zampini PetscFunctionReturn(0); 22083425bc38SStefano Zampini } 22090c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2210da1bb401SStefano Zampini /*MC 2211da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 22120c7d97c5SJed Brown 221328509bceSStefano Zampini An implementation of the BDDC preconditioner based on 221428509bceSStefano Zampini 221528509bceSStefano Zampini .vb 221628509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 221728509bceSStefano 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 221828509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 22190f202f7eSStefano 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 222028509bceSStefano Zampini .ve 222128509bceSStefano Zampini 222228509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 222328509bceSStefano Zampini 22240f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 222528509bceSStefano Zampini 222628509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 222728509bceSStefano Zampini 2228b6fdb6dfSStefano 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. 2229b6fdb6dfSStefano Zampini 22300f202f7eSStefano Zampini Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 223128509bceSStefano Zampini 22320f202f7eSStefano 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() 223330368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 223428509bceSStefano Zampini 22350f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 223628509bceSStefano Zampini 22370f202f7eSStefano 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. 22380f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 223928509bceSStefano Zampini 22400f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 224128509bceSStefano Zampini 2242df4d28bfSStefano 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. 224328509bceSStefano Zampini 22440f202f7eSStefano 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. 22450f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 22460f202f7eSStefano Zampini 22470f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 22480f202f7eSStefano Zampini 22490f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 22500f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 22510f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 22520f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 22530f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 22540f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 22550f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 225628509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 22570f202f7eSStefano 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) 22580f202f7eSStefano 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) 22590f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 22600f202f7eSStefano 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) 2261df4d28bfSStefano 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) 226228509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 226328509bceSStefano Zampini 226428509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 226528509bceSStefano Zampini .vb 226628509bceSStefano Zampini -pc_bddc_dirichlet_ 226728509bceSStefano Zampini -pc_bddc_neumann_ 226828509bceSStefano Zampini -pc_bddc_coarse_ 226928509bceSStefano Zampini .ve 22700f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 227128509bceSStefano Zampini 22720f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 227328509bceSStefano Zampini .vb 2274312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2275312be037SStefano Zampini -pc_bddc_neumann_lN_ 2276312be037SStefano Zampini -pc_bddc_coarse_lN_ 227728509bceSStefano Zampini .ve 22780f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 22790f202f7eSStefano 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. 22800f202f7eSStefano Zampini .vb 22810f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 22820f202f7eSStefano Zampini .ve 22830f202f7eSStefano 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 2284da1bb401SStefano Zampini 2285da1bb401SStefano Zampini Level: intermediate 2286da1bb401SStefano Zampini 2287b6fdb6dfSStefano Zampini Developer notes: 2288da1bb401SStefano Zampini 2289da1bb401SStefano Zampini Contributed by Stefano Zampini 2290da1bb401SStefano Zampini 2291da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2292da1bb401SStefano Zampini M*/ 2293b2573a8aSBarry Smith 2294da1bb401SStefano Zampini #undef __FUNCT__ 2295da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 22968cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2297da1bb401SStefano Zampini { 2298da1bb401SStefano Zampini PetscErrorCode ierr; 2299da1bb401SStefano Zampini PC_BDDC *pcbddc; 2300da1bb401SStefano Zampini 2301da1bb401SStefano Zampini PetscFunctionBegin; 2302da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2303b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2304da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2305da1bb401SStefano Zampini 2306da1bb401SStefano Zampini /* create PCIS data structure */ 2307da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2308da1bb401SStefano Zampini 230947d04d0dSStefano Zampini /* BDDC customization */ 231008a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 231147d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 231247d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 231347d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 231447d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 231547d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 231647d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2317fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2318fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 23193301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 232006a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 232147d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2322b9d89cd5SStefano Zampini /* private */ 2323727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 23240e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 23250e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 23260e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2327e9189074SStefano Zampini pcbddc->n_vertices = 0; 2328727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2329fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 233068457ee5SStefano Zampini pcbddc->coarse_size = -1; 2331f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2332727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2333f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2334f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2335f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2336674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 233730368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 23380bdf917eSStefano Zampini pcbddc->NullSpace = 0; 23393972b0daSStefano Zampini pcbddc->temp_solution = 0; 2340534831adSStefano Zampini pcbddc->original_rhs = 0; 2341534831adSStefano Zampini pcbddc->local_mat = 0; 2342534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2343b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2344906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 2345da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2346da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2347da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2348da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 234915aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 235015aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2351da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2352da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2353da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2354da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2355da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2356da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2357da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2358da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2359da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2360da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2361785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2362785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2363785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 236460d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 236560d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 236663602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2367da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 236863602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2369da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 237085c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 237147d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 237247d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2373b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 23744fad6a16SStefano Zampini pcbddc->current_level = 0; 23752b510759SStefano Zampini pcbddc->max_levels = 0; 2376323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 237774e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2378f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2379323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 23804f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 23814f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 23824f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 238381d14e9dSStefano Zampini 238481d14e9dSStefano Zampini /* benign subspace trick */ 238581d14e9dSStefano Zampini pcbddc->benign_change = 0; 23860369aaf7SStefano Zampini pcbddc->benign_vec = 0; 23870369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 23880369aaf7SStefano Zampini pcbddc->benign_sf = 0; 23894f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 23904f1b2e48SStefano Zampini pcbddc->benign_n = 0; 23914f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 23924f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 23934f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2394b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 239581d14e9dSStefano Zampini 2396674ae819SStefano Zampini /* create local graph structure */ 2397674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2398674ae819SStefano Zampini 2399674ae819SStefano Zampini /* scaling */ 2400674ae819SStefano Zampini pcbddc->work_scaling = 0; 240134a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2402ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2403b96c3477SStefano Zampini 2404b96c3477SStefano Zampini /* create sub schurs structure */ 2405b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2406b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2407b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2408b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2409b96c3477SStefano Zampini 2410b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2411da1bb401SStefano Zampini 2412b7eb3628SStefano Zampini /* adaptivity */ 2413f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 241408122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2415f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2416b7eb3628SStefano Zampini 2417da1bb401SStefano Zampini /* function pointers */ 2418da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 241993bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2420da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2421da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2422da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2423da1bb401SStefano Zampini pc->ops->view = 0; 2424da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2425da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2426da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2427534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2428534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2429da1bb401SStefano Zampini 2430da1bb401SStefano Zampini /* composing function */ 2431906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2432674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 243330368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2434bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 24352b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2436b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 24372b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2438bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2439bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 244082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2441bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 244282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2443bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 244482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2445bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 244682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2447bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 244863602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2449bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2450bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2451bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2452bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2453da1bb401SStefano Zampini PetscFunctionReturn(0); 2454da1bb401SStefano Zampini } 24553425bc38SStefano Zampini 2456