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 29*0369aaf7SStefano Zampini /* temporarily declare it */ 30*0369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 31*0369aaf7SStefano 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); 67ac632422SStefano 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); 684c6709b3SStefano 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); 6908122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 7008122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 713301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 72b0c7d250SStefano 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); 7306a4e24aSStefano 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); 740c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 750c7d97c5SJed Brown PetscFunctionReturn(0); 760c7d97c5SJed Brown } 770c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 78674ae819SStefano Zampini #undef __FUNCT__ 79906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 80906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 81b9b85e73SStefano Zampini { 82b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 83b9b85e73SStefano Zampini PetscErrorCode ierr; 84b9b85e73SStefano Zampini 85b9b85e73SStefano Zampini PetscFunctionBegin; 86b9b85e73SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 87b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 88b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 89b9b85e73SStefano Zampini PetscFunctionReturn(0); 90b9b85e73SStefano Zampini } 91b9b85e73SStefano Zampini #undef __FUNCT__ 92906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 93b9b85e73SStefano Zampini /*@ 94906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 95b9b85e73SStefano Zampini 96b9b85e73SStefano Zampini Collective on PC 97b9b85e73SStefano Zampini 98b9b85e73SStefano Zampini Input Parameters: 99b9b85e73SStefano Zampini + pc - the preconditioning context 100906d46d4SStefano Zampini - change - the change of basis matrix 101b9b85e73SStefano Zampini 102b9b85e73SStefano Zampini Level: intermediate 103b9b85e73SStefano Zampini 104b9b85e73SStefano Zampini Notes: 105b9b85e73SStefano Zampini 106b9b85e73SStefano Zampini .seealso: PCBDDC 107b9b85e73SStefano Zampini @*/ 108906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 109b9b85e73SStefano Zampini { 110b9b85e73SStefano Zampini PetscErrorCode ierr; 111b9b85e73SStefano Zampini 112b9b85e73SStefano Zampini PetscFunctionBegin; 113b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 114b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 115906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 116906d46d4SStefano Zampini if (pc->mat) { 117906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 118906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 119906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 120906d46d4SStefano Zampini if (rows_c != rows) { 121906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 122906d46d4SStefano Zampini } 123906d46d4SStefano Zampini if (cols_c != cols) { 124906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 125906d46d4SStefano Zampini } 126906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 127906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 128906d46d4SStefano Zampini if (rows_c != rows) { 129906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 130906d46d4SStefano Zampini } 131906d46d4SStefano Zampini if (cols_c != cols) { 132906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 133906d46d4SStefano Zampini } 134906d46d4SStefano Zampini } 135906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 136b9b85e73SStefano Zampini PetscFunctionReturn(0); 137b9b85e73SStefano Zampini } 138b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 139b9b85e73SStefano Zampini #undef __FUNCT__ 140674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 141674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 142674ae819SStefano Zampini { 143674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 144674ae819SStefano Zampini PetscErrorCode ierr; 1451e6b0712SBarry Smith 146674ae819SStefano Zampini PetscFunctionBegin; 147674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 148674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 149674ae819SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 150674ae819SStefano Zampini PetscFunctionReturn(0); 151674ae819SStefano Zampini } 152674ae819SStefano Zampini #undef __FUNCT__ 153674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 154674ae819SStefano Zampini /*@ 15528509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 156674ae819SStefano Zampini 15717eb1463SStefano Zampini Collective 158674ae819SStefano Zampini 159674ae819SStefano Zampini Input Parameters: 160674ae819SStefano Zampini + pc - the preconditioning context 16117eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 162674ae819SStefano Zampini 163674ae819SStefano Zampini Level: intermediate 164674ae819SStefano Zampini 165674ae819SStefano Zampini Notes: 166674ae819SStefano Zampini 167674ae819SStefano Zampini .seealso: PCBDDC 168674ae819SStefano Zampini @*/ 169674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 170674ae819SStefano Zampini { 171674ae819SStefano Zampini PetscErrorCode ierr; 172674ae819SStefano Zampini 173674ae819SStefano Zampini PetscFunctionBegin; 174674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 175674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 17617eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 177674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 178674ae819SStefano Zampini PetscFunctionReturn(0); 179674ae819SStefano Zampini } 180674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 1810c7d97c5SJed Brown #undef __FUNCT__ 1824fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 1834fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 1844fad6a16SStefano Zampini { 1854fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1864fad6a16SStefano Zampini 1874fad6a16SStefano Zampini PetscFunctionBegin; 1884fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 1894fad6a16SStefano Zampini PetscFunctionReturn(0); 1904fad6a16SStefano Zampini } 1911e6b0712SBarry Smith 1924fad6a16SStefano Zampini #undef __FUNCT__ 1934fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 1944fad6a16SStefano Zampini /*@ 19528509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 1964fad6a16SStefano Zampini 1974fad6a16SStefano Zampini Logically collective on PC 1984fad6a16SStefano Zampini 1994fad6a16SStefano Zampini Input Parameters: 2004fad6a16SStefano Zampini + pc - the preconditioning context 20128509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 2024fad6a16SStefano Zampini 2030f202f7eSStefano Zampini Options Database Keys: 2040f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 2054fad6a16SStefano Zampini 2064fad6a16SStefano Zampini Level: intermediate 2074fad6a16SStefano Zampini 2084fad6a16SStefano Zampini Notes: 2090f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 2104fad6a16SStefano Zampini 2110f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2124fad6a16SStefano Zampini @*/ 2134fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2144fad6a16SStefano Zampini { 2154fad6a16SStefano Zampini PetscErrorCode ierr; 2164fad6a16SStefano Zampini 2174fad6a16SStefano Zampini PetscFunctionBegin; 2184fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2192b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2204fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2214fad6a16SStefano Zampini PetscFunctionReturn(0); 2224fad6a16SStefano Zampini } 2232b510759SStefano Zampini 224b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2252b510759SStefano Zampini #undef __FUNCT__ 226b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 227b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 228b8ffe317SStefano Zampini { 229b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 230b8ffe317SStefano Zampini 231b8ffe317SStefano Zampini PetscFunctionBegin; 23285c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 233b8ffe317SStefano Zampini PetscFunctionReturn(0); 234b8ffe317SStefano Zampini } 235b8ffe317SStefano Zampini 236b8ffe317SStefano Zampini #undef __FUNCT__ 237b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 238b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 2392b510759SStefano Zampini { 2402b510759SStefano Zampini PetscErrorCode ierr; 2412b510759SStefano Zampini 2422b510759SStefano Zampini PetscFunctionBegin; 2432b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 244b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 245b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 2462b510759SStefano Zampini PetscFunctionReturn(0); 2472b510759SStefano Zampini } 2481e6b0712SBarry Smith 2494fad6a16SStefano Zampini #undef __FUNCT__ 2502b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 2512b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 2524fad6a16SStefano Zampini { 2534fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2544fad6a16SStefano Zampini 2554fad6a16SStefano Zampini PetscFunctionBegin; 2562b510759SStefano Zampini pcbddc->current_level = level; 2574fad6a16SStefano Zampini PetscFunctionReturn(0); 2584fad6a16SStefano Zampini } 2591e6b0712SBarry Smith 2604fad6a16SStefano Zampini #undef __FUNCT__ 261b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 262b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 263b8ffe317SStefano Zampini { 264b8ffe317SStefano Zampini PetscErrorCode ierr; 265b8ffe317SStefano Zampini 266b8ffe317SStefano Zampini PetscFunctionBegin; 267b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 268b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 269b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 270b8ffe317SStefano Zampini PetscFunctionReturn(0); 271b8ffe317SStefano Zampini } 272b8ffe317SStefano Zampini 273b8ffe317SStefano Zampini #undef __FUNCT__ 2742b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 2752b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 2762b510759SStefano Zampini { 2772b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2782b510759SStefano Zampini 2792b510759SStefano Zampini PetscFunctionBegin; 2802b510759SStefano Zampini pcbddc->max_levels = levels; 2812b510759SStefano Zampini PetscFunctionReturn(0); 2822b510759SStefano Zampini } 2832b510759SStefano Zampini 2842b510759SStefano Zampini #undef __FUNCT__ 2852b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 2864fad6a16SStefano Zampini /*@ 28728509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 2884fad6a16SStefano Zampini 2894fad6a16SStefano Zampini Logically collective on PC 2904fad6a16SStefano Zampini 2914fad6a16SStefano Zampini Input Parameters: 2924fad6a16SStefano Zampini + pc - the preconditioning context 29328509bceSStefano Zampini - levels - the maximum number of levels (max 9) 2944fad6a16SStefano Zampini 2950f202f7eSStefano Zampini Options Database Keys: 2960f202f7eSStefano Zampini . -pc_bddc_levels 2974fad6a16SStefano Zampini 2984fad6a16SStefano Zampini Level: intermediate 2994fad6a16SStefano Zampini 3004fad6a16SStefano Zampini Notes: 3010f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 3024fad6a16SStefano Zampini 3030f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 3044fad6a16SStefano Zampini @*/ 3052b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 3064fad6a16SStefano Zampini { 3074fad6a16SStefano Zampini PetscErrorCode ierr; 3084fad6a16SStefano Zampini 3094fad6a16SStefano Zampini PetscFunctionBegin; 3104fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 3112b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 312312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3132b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3144fad6a16SStefano Zampini PetscFunctionReturn(0); 3154fad6a16SStefano Zampini } 3164fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3171e6b0712SBarry Smith 3184fad6a16SStefano Zampini #undef __FUNCT__ 3190bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 3200bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 3210bdf917eSStefano Zampini { 3220bdf917eSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3230bdf917eSStefano Zampini PetscErrorCode ierr; 3240bdf917eSStefano Zampini 3250bdf917eSStefano Zampini PetscFunctionBegin; 3260bdf917eSStefano Zampini ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 3270bdf917eSStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 3280bdf917eSStefano Zampini pcbddc->NullSpace = NullSpace; 3290bdf917eSStefano Zampini PetscFunctionReturn(0); 3300bdf917eSStefano Zampini } 3311e6b0712SBarry Smith 3320bdf917eSStefano Zampini #undef __FUNCT__ 3330bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace" 3340bdf917eSStefano Zampini /*@ 33528509bceSStefano Zampini PCBDDCSetNullSpace - Set nullspace for BDDC operator 3360bdf917eSStefano Zampini 3370bdf917eSStefano Zampini Logically collective on PC and MatNullSpace 3380bdf917eSStefano Zampini 3390bdf917eSStefano Zampini Input Parameters: 3400bdf917eSStefano Zampini + pc - the preconditioning context 34128509bceSStefano Zampini - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 3420bdf917eSStefano Zampini 3430bdf917eSStefano Zampini Level: intermediate 3440bdf917eSStefano Zampini 3450bdf917eSStefano Zampini Notes: 3460bdf917eSStefano Zampini 3470bdf917eSStefano Zampini .seealso: PCBDDC 3480bdf917eSStefano Zampini @*/ 3490bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 3500bdf917eSStefano Zampini { 3510bdf917eSStefano Zampini PetscErrorCode ierr; 3520bdf917eSStefano Zampini 3530bdf917eSStefano Zampini PetscFunctionBegin; 3540bdf917eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 355674ae819SStefano Zampini PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 3562b510759SStefano Zampini PetscCheckSameComm(pc,1,NullSpace,2); 3570bdf917eSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 3580bdf917eSStefano Zampini PetscFunctionReturn(0); 3590bdf917eSStefano Zampini } 3600bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */ 3611e6b0712SBarry Smith 3620bdf917eSStefano Zampini #undef __FUNCT__ 3633b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 3643b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 3653b03a366Sstefano_zampini { 3663b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3673b03a366Sstefano_zampini PetscErrorCode ierr; 3683b03a366Sstefano_zampini 3693b03a366Sstefano_zampini PetscFunctionBegin; 370785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 371785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 3723b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 37336e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 37436e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 375fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 3763b03a366Sstefano_zampini PetscFunctionReturn(0); 3773b03a366Sstefano_zampini } 3781e6b0712SBarry Smith 3793b03a366Sstefano_zampini #undef __FUNCT__ 3803b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 3813b03a366Sstefano_zampini /*@ 38228509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 3833b03a366Sstefano_zampini 384785d1243SStefano Zampini Collective 3853b03a366Sstefano_zampini 3863b03a366Sstefano_zampini Input Parameters: 3873b03a366Sstefano_zampini + pc - the preconditioning context 388785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 3893b03a366Sstefano_zampini 3903b03a366Sstefano_zampini Level: intermediate 3913b03a366Sstefano_zampini 3920f202f7eSStefano Zampini Notes: 3930f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 3943b03a366Sstefano_zampini 3950f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 3963b03a366Sstefano_zampini @*/ 3973b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 3983b03a366Sstefano_zampini { 3993b03a366Sstefano_zampini PetscErrorCode ierr; 4003b03a366Sstefano_zampini 4013b03a366Sstefano_zampini PetscFunctionBegin; 4023b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 403674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 404785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4053b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4063b03a366Sstefano_zampini PetscFunctionReturn(0); 4073b03a366Sstefano_zampini } 4083b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4091e6b0712SBarry Smith 4103b03a366Sstefano_zampini #undef __FUNCT__ 41182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 41282ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4130c7d97c5SJed Brown { 4140c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4150c7d97c5SJed Brown PetscErrorCode ierr; 4160c7d97c5SJed Brown 4170c7d97c5SJed Brown PetscFunctionBegin; 418785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 419785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4200c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 4210c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 422785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 4237d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4240c7d97c5SJed Brown PetscFunctionReturn(0); 4250c7d97c5SJed Brown } 4260c7d97c5SJed Brown 4270c7d97c5SJed Brown #undef __FUNCT__ 42882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4299c0446d6SStefano Zampini /*@ 43082ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4319c0446d6SStefano Zampini 432785d1243SStefano Zampini Collective 43353cdbc3dSStefano Zampini 43453cdbc3dSStefano Zampini Input Parameters: 43553cdbc3dSStefano Zampini + pc - the preconditioning context 43682ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 43753cdbc3dSStefano Zampini 43853cdbc3dSStefano Zampini Level: intermediate 43953cdbc3dSStefano Zampini 4409c0446d6SStefano Zampini Notes: 44153cdbc3dSStefano Zampini 4420f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 44353cdbc3dSStefano Zampini @*/ 44482ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 4450c7d97c5SJed Brown { 4460c7d97c5SJed Brown PetscErrorCode ierr; 4470c7d97c5SJed Brown 4480c7d97c5SJed Brown PetscFunctionBegin; 4490c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4500c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 45182ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 45282ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4530c7d97c5SJed Brown PetscFunctionReturn(0); 4540c7d97c5SJed Brown } 4550c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 4560c7d97c5SJed Brown 4570c7d97c5SJed Brown #undef __FUNCT__ 4580c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 45953cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 4600c7d97c5SJed Brown { 4610c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 46253cdbc3dSStefano Zampini PetscErrorCode ierr; 4630c7d97c5SJed Brown 4640c7d97c5SJed Brown PetscFunctionBegin; 465785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 466785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 46753cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 46836e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 46936e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 470fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4710c7d97c5SJed Brown PetscFunctionReturn(0); 4720c7d97c5SJed Brown } 4731e6b0712SBarry Smith 4740c7d97c5SJed Brown #undef __FUNCT__ 4750c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 47657527edcSJed Brown /*@ 47728509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 47857527edcSJed Brown 479785d1243SStefano Zampini Collective 48057527edcSJed Brown 48157527edcSJed Brown Input Parameters: 48257527edcSJed Brown + pc - the preconditioning context 483785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 48457527edcSJed Brown 48557527edcSJed Brown Level: intermediate 48657527edcSJed Brown 4870f202f7eSStefano Zampini Notes: 4880f202f7eSStefano Zampini Any process can list any global node 48957527edcSJed Brown 4900f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 49157527edcSJed Brown @*/ 49253cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 4930c7d97c5SJed Brown { 4940c7d97c5SJed Brown PetscErrorCode ierr; 4950c7d97c5SJed Brown 4960c7d97c5SJed Brown PetscFunctionBegin; 4970c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 498674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 499785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 50053cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 50153cdbc3dSStefano Zampini PetscFunctionReturn(0); 50253cdbc3dSStefano Zampini } 50353cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5041e6b0712SBarry Smith 50553cdbc3dSStefano Zampini #undef __FUNCT__ 50682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 50782ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5080c7d97c5SJed Brown { 5090c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5100c7d97c5SJed Brown PetscErrorCode ierr; 5110c7d97c5SJed Brown 5120c7d97c5SJed Brown PetscFunctionBegin; 513785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 514785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5150c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 5160c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 517785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 5187d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5190c7d97c5SJed Brown PetscFunctionReturn(0); 5200c7d97c5SJed Brown } 5210c7d97c5SJed Brown 5220c7d97c5SJed Brown #undef __FUNCT__ 52382ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5240c7d97c5SJed Brown /*@ 52582ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 5260c7d97c5SJed Brown 527785d1243SStefano Zampini Collective 5280c7d97c5SJed Brown 5290c7d97c5SJed Brown Input Parameters: 5300c7d97c5SJed Brown + pc - the preconditioning context 53182ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 5320c7d97c5SJed Brown 5330c7d97c5SJed Brown Level: intermediate 5340c7d97c5SJed Brown 5350c7d97c5SJed Brown Notes: 5360c7d97c5SJed Brown 5370f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 5380c7d97c5SJed Brown @*/ 53982ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 5400c7d97c5SJed Brown { 5410c7d97c5SJed Brown PetscErrorCode ierr; 5420c7d97c5SJed Brown 5430c7d97c5SJed Brown PetscFunctionBegin; 5440c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5450c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 54682ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 54782ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 54853cdbc3dSStefano Zampini PetscFunctionReturn(0); 54953cdbc3dSStefano Zampini } 55053cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 55153cdbc3dSStefano Zampini 55253cdbc3dSStefano Zampini #undef __FUNCT__ 553da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 554da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 555da1bb401SStefano Zampini { 556da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 557da1bb401SStefano Zampini 558da1bb401SStefano Zampini PetscFunctionBegin; 559da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 560da1bb401SStefano Zampini PetscFunctionReturn(0); 561da1bb401SStefano Zampini } 5621e6b0712SBarry Smith 563da1bb401SStefano Zampini #undef __FUNCT__ 564da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 565da1bb401SStefano Zampini /*@ 566785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 567da1bb401SStefano Zampini 568785d1243SStefano Zampini Collective 569785d1243SStefano Zampini 570785d1243SStefano Zampini Input Parameters: 571785d1243SStefano Zampini . pc - the preconditioning context 572785d1243SStefano Zampini 573785d1243SStefano Zampini Output Parameters: 574785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 575785d1243SStefano Zampini 576785d1243SStefano Zampini Level: intermediate 577785d1243SStefano Zampini 5780f202f7eSStefano Zampini Notes: 5790f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 580785d1243SStefano Zampini 581785d1243SStefano Zampini .seealso: PCBDDC 582785d1243SStefano Zampini @*/ 583785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 584785d1243SStefano Zampini { 585785d1243SStefano Zampini PetscErrorCode ierr; 586785d1243SStefano Zampini 587785d1243SStefano Zampini PetscFunctionBegin; 588785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 589785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 590785d1243SStefano Zampini PetscFunctionReturn(0); 591785d1243SStefano Zampini } 592785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 593785d1243SStefano Zampini 594785d1243SStefano Zampini #undef __FUNCT__ 595785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 596785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 597785d1243SStefano Zampini { 598785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 599785d1243SStefano Zampini 600785d1243SStefano Zampini PetscFunctionBegin; 601785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 602785d1243SStefano Zampini PetscFunctionReturn(0); 603785d1243SStefano Zampini } 604785d1243SStefano Zampini 605785d1243SStefano Zampini #undef __FUNCT__ 60682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 607da1bb401SStefano Zampini /*@ 60882ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 609da1bb401SStefano Zampini 610785d1243SStefano Zampini Collective 611da1bb401SStefano Zampini 612da1bb401SStefano Zampini Input Parameters: 61328509bceSStefano Zampini . pc - the preconditioning context 614da1bb401SStefano Zampini 615da1bb401SStefano Zampini Output Parameters: 61628509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 617da1bb401SStefano Zampini 618da1bb401SStefano Zampini Level: intermediate 619da1bb401SStefano Zampini 620da1bb401SStefano Zampini Notes: 6210f202f7eSStefano 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). 6220f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 623da1bb401SStefano Zampini 624da1bb401SStefano Zampini .seealso: PCBDDC 625da1bb401SStefano Zampini @*/ 62682ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 627da1bb401SStefano Zampini { 628da1bb401SStefano Zampini PetscErrorCode ierr; 629da1bb401SStefano Zampini 630da1bb401SStefano Zampini PetscFunctionBegin; 631da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 63282ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 633da1bb401SStefano Zampini PetscFunctionReturn(0); 634da1bb401SStefano Zampini } 635da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 6361e6b0712SBarry Smith 637da1bb401SStefano Zampini #undef __FUNCT__ 63853cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 63953cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 64053cdbc3dSStefano Zampini { 64153cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 64253cdbc3dSStefano Zampini 64353cdbc3dSStefano Zampini PetscFunctionBegin; 64453cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 64553cdbc3dSStefano Zampini PetscFunctionReturn(0); 64653cdbc3dSStefano Zampini } 6471e6b0712SBarry Smith 64853cdbc3dSStefano Zampini #undef __FUNCT__ 64953cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 65053cdbc3dSStefano Zampini /*@ 651785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 65253cdbc3dSStefano Zampini 653785d1243SStefano Zampini Collective 654785d1243SStefano Zampini 655785d1243SStefano Zampini Input Parameters: 656785d1243SStefano Zampini . pc - the preconditioning context 657785d1243SStefano Zampini 658785d1243SStefano Zampini Output Parameters: 659785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 660785d1243SStefano Zampini 661785d1243SStefano Zampini Level: intermediate 662785d1243SStefano Zampini 6630f202f7eSStefano Zampini Notes: 6640f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 665785d1243SStefano Zampini 666785d1243SStefano Zampini .seealso: PCBDDC 667785d1243SStefano Zampini @*/ 668785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 669785d1243SStefano Zampini { 670785d1243SStefano Zampini PetscErrorCode ierr; 671785d1243SStefano Zampini 672785d1243SStefano Zampini PetscFunctionBegin; 673785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 674785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 675785d1243SStefano Zampini PetscFunctionReturn(0); 676785d1243SStefano Zampini } 677785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 678785d1243SStefano Zampini 679785d1243SStefano Zampini #undef __FUNCT__ 680785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 681785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 682785d1243SStefano Zampini { 683785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 684785d1243SStefano Zampini 685785d1243SStefano Zampini PetscFunctionBegin; 686785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 687785d1243SStefano Zampini PetscFunctionReturn(0); 688785d1243SStefano Zampini } 689785d1243SStefano Zampini 690785d1243SStefano Zampini #undef __FUNCT__ 69182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 69253cdbc3dSStefano Zampini /*@ 69382ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 69453cdbc3dSStefano Zampini 695785d1243SStefano Zampini Collective 69653cdbc3dSStefano Zampini 69753cdbc3dSStefano Zampini Input Parameters: 69828509bceSStefano Zampini . pc - the preconditioning context 69953cdbc3dSStefano Zampini 70053cdbc3dSStefano Zampini Output Parameters: 70128509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 70253cdbc3dSStefano Zampini 70353cdbc3dSStefano Zampini Level: intermediate 70453cdbc3dSStefano Zampini 70553cdbc3dSStefano Zampini Notes: 7060f202f7eSStefano 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). 7070f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 70853cdbc3dSStefano Zampini 70953cdbc3dSStefano Zampini .seealso: PCBDDC 71053cdbc3dSStefano Zampini @*/ 71182ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 71253cdbc3dSStefano Zampini { 71353cdbc3dSStefano Zampini PetscErrorCode ierr; 71453cdbc3dSStefano Zampini 71553cdbc3dSStefano Zampini PetscFunctionBegin; 71653cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 71782ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7180c7d97c5SJed Brown PetscFunctionReturn(0); 7190c7d97c5SJed Brown } 72036e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7211e6b0712SBarry Smith 72236e030ebSStefano Zampini #undef __FUNCT__ 723da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7241a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 72536e030ebSStefano Zampini { 72636e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 727da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 728da1bb401SStefano Zampini PetscErrorCode ierr; 72936e030ebSStefano Zampini 73036e030ebSStefano Zampini PetscFunctionBegin; 731674ae819SStefano Zampini /* free old CSR */ 732674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 733fb223d50SStefano Zampini /* TODO: PCBDDCGraphSetAdjacency */ 734674ae819SStefano Zampini /* get CSR into graph structure */ 735da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 736854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 737785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 738674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 739674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 740da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 7411a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 7421a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 743674ae819SStefano Zampini } 744575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 74536e030ebSStefano Zampini PetscFunctionReturn(0); 74636e030ebSStefano Zampini } 7471e6b0712SBarry Smith 74836e030ebSStefano Zampini #undef __FUNCT__ 749da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 75036e030ebSStefano Zampini /*@ 7510f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 75236e030ebSStefano Zampini 75336e030ebSStefano Zampini Not collective 75436e030ebSStefano Zampini 75536e030ebSStefano Zampini Input Parameters: 75636e030ebSStefano Zampini + pc - the preconditioning context 7570f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 75828509bceSStefano Zampini . xadj, adjncy - the CSR graph 75928509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 76036e030ebSStefano Zampini 76136e030ebSStefano Zampini Level: intermediate 76236e030ebSStefano Zampini 76336e030ebSStefano Zampini Notes: 76436e030ebSStefano Zampini 76528509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 76636e030ebSStefano Zampini @*/ 7671a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 76836e030ebSStefano Zampini { 769575ad6abSStefano Zampini void (*f)(void) = 0; 77036e030ebSStefano Zampini PetscErrorCode ierr; 77136e030ebSStefano Zampini 77236e030ebSStefano Zampini PetscFunctionBegin; 77336e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 774674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 775d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 776674ae819SStefano Zampini if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) { 777674ae819SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__); 778da1bb401SStefano Zampini } 77936e030ebSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 780575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 781575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 782575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 783575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 784575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 78536e030ebSStefano Zampini } 78636e030ebSStefano Zampini PetscFunctionReturn(0); 78736e030ebSStefano Zampini } 7889c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 7891e6b0712SBarry Smith 7909c0446d6SStefano Zampini #undef __FUNCT__ 79163602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 79263602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 79363602bcaSStefano Zampini { 79463602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79563602bcaSStefano Zampini PetscInt i; 79663602bcaSStefano Zampini PetscErrorCode ierr; 79763602bcaSStefano Zampini 79863602bcaSStefano Zampini PetscFunctionBegin; 79963602bcaSStefano Zampini /* Destroy ISes if they were already set */ 80063602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 80163602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 80263602bcaSStefano Zampini } 80363602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 80463602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 80563602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 80663602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 80763602bcaSStefano Zampini } 80863602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 80963602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 81063602bcaSStefano Zampini /* allocate space then set */ 811d02579f5SStefano Zampini if (n_is) { 812d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 813d02579f5SStefano Zampini } 81463602bcaSStefano Zampini for (i=0;i<n_is;i++) { 81563602bcaSStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 81663602bcaSStefano Zampini pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 81763602bcaSStefano Zampini } 81863602bcaSStefano Zampini pcbddc->n_ISForDofsLocal=n_is; 81963602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8201a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 82163602bcaSStefano Zampini PetscFunctionReturn(0); 82263602bcaSStefano Zampini } 82363602bcaSStefano Zampini 82463602bcaSStefano Zampini #undef __FUNCT__ 82563602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 82663602bcaSStefano Zampini /*@ 82763602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 82863602bcaSStefano Zampini 82963602bcaSStefano Zampini Collective 83063602bcaSStefano Zampini 83163602bcaSStefano Zampini Input Parameters: 83263602bcaSStefano Zampini + pc - the preconditioning context 8330f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8340f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 83563602bcaSStefano Zampini 83663602bcaSStefano Zampini Level: intermediate 83763602bcaSStefano Zampini 8380f202f7eSStefano Zampini Notes: 8390f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 84063602bcaSStefano Zampini 84163602bcaSStefano Zampini .seealso: PCBDDC 84263602bcaSStefano Zampini @*/ 84363602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 84463602bcaSStefano Zampini { 84563602bcaSStefano Zampini PetscInt i; 84663602bcaSStefano Zampini PetscErrorCode ierr; 84763602bcaSStefano Zampini 84863602bcaSStefano Zampini PetscFunctionBegin; 84963602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 85063602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 85163602bcaSStefano Zampini for (i=0;i<n_is;i++) { 85263602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 85363602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 85463602bcaSStefano Zampini } 855e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 85663602bcaSStefano Zampini PetscFunctionReturn(0); 85763602bcaSStefano Zampini } 85863602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 85963602bcaSStefano Zampini 86063602bcaSStefano Zampini #undef __FUNCT__ 8619c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 8629c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 8639c0446d6SStefano Zampini { 8649c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 8659c0446d6SStefano Zampini PetscInt i; 8669c0446d6SStefano Zampini PetscErrorCode ierr; 8679c0446d6SStefano Zampini 8689c0446d6SStefano Zampini PetscFunctionBegin; 869da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 8709c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 8719c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 8729c0446d6SStefano Zampini } 873d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 87463602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 87563602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 87663602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 87763602bcaSStefano Zampini } 87863602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 87963602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 880da1bb401SStefano Zampini /* allocate space then set */ 881d02579f5SStefano Zampini if (n_is) { 882785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 883d02579f5SStefano Zampini } 8849c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 885da1bb401SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 886da1bb401SStefano Zampini pcbddc->ISForDofs[i]=ISForDofs[i]; 8879c0446d6SStefano Zampini } 8889c0446d6SStefano Zampini pcbddc->n_ISForDofs=n_is; 88963602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8901a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 8919c0446d6SStefano Zampini PetscFunctionReturn(0); 8929c0446d6SStefano Zampini } 8931e6b0712SBarry Smith 8949c0446d6SStefano Zampini #undef __FUNCT__ 8959c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 8969c0446d6SStefano Zampini /*@ 89763602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 8989c0446d6SStefano Zampini 89963602bcaSStefano Zampini Collective 9009c0446d6SStefano Zampini 9019c0446d6SStefano Zampini Input Parameters: 9029c0446d6SStefano Zampini + pc - the preconditioning context 9030f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9040f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 9059c0446d6SStefano Zampini 9069c0446d6SStefano Zampini Level: intermediate 9079c0446d6SStefano Zampini 9080f202f7eSStefano Zampini Notes: 9090f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 9109c0446d6SStefano Zampini 9119c0446d6SStefano Zampini .seealso: PCBDDC 9129c0446d6SStefano Zampini @*/ 9139c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 9149c0446d6SStefano Zampini { 9152b510759SStefano Zampini PetscInt i; 9169c0446d6SStefano Zampini PetscErrorCode ierr; 9179c0446d6SStefano Zampini 9189c0446d6SStefano Zampini PetscFunctionBegin; 9199c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 92063602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 9212b510759SStefano Zampini for (i=0;i<n_is;i++) { 92263602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 92363602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 9242b510759SStefano Zampini } 9259c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 9269c0446d6SStefano Zampini PetscFunctionReturn(0); 9279c0446d6SStefano Zampini } 928906d46d4SStefano Zampini 929da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 930534831adSStefano Zampini #undef __FUNCT__ 931534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 932534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 933534831adSStefano Zampini /* 934534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 935534831adSStefano Zampini guess if a transformation of basis approach has been selected. 9369c0446d6SStefano Zampini 937534831adSStefano Zampini Input Parameter: 938534831adSStefano Zampini + pc - the preconditioner contex 939534831adSStefano Zampini 940534831adSStefano Zampini Application Interface Routine: PCPreSolve() 941534831adSStefano Zampini 942534831adSStefano Zampini Notes: 943534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 944534831adSStefano Zampini the user, but instead is called by KSPSolve(). 945534831adSStefano Zampini */ 946534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 947534831adSStefano Zampini { 948534831adSStefano Zampini PetscErrorCode ierr; 949534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 950534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 9513972b0daSStefano Zampini Vec used_vec; 9528d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 953534831adSStefano Zampini 954534831adSStefano Zampini PetscFunctionBegin; 95585c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 95685c4d303SStefano Zampini if (ksp) { 95785c4d303SStefano Zampini PetscBool iscg; 95885c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 95985c4d303SStefano Zampini if (!iscg) { 96085c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 96185c4d303SStefano Zampini } 96285c4d303SStefano Zampini } 96385c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 96462a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 96562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 96662a6ff1dSStefano Zampini } 96762a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 96862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 96962a6ff1dSStefano Zampini } 9708d00608fSStefano Zampini 9713972b0daSStefano Zampini if (x) { 9723972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 9733972b0daSStefano Zampini used_vec = x; 9748d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 9753972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 9763972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 9773972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9783972b0daSStefano Zampini } 9798efcfb23SStefano Zampini 9808efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 9813972b0daSStefano Zampini if (ksp) { 982a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 9838efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 9848efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 9853972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9863972b0daSStefano Zampini } 9873972b0daSStefano Zampini } 9883308cffdSStefano Zampini 9898d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 9903972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 991a07ea27aSStefano Zampini if (rhs) { 9923975b054SStefano Zampini IS dirIS = NULL; 9933975b054SStefano Zampini 994a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 9953975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 9963975b054SStefano Zampini if (dirIS) { 997906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 998785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 9992b095fd8SStefano Zampini PetscScalar *array_x; 10002b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1001785d1243SStefano Zampini 10023972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 10033972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1004e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1005e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1006e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1007e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 100882ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 10093972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 10102b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10113972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10122fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 10133972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10142b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10153972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1016e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1017e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10188d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10191b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 10208efcfb23SStefano Zampini } 1021a07ea27aSStefano Zampini } 1022b76ba322SStefano Zampini 10238efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 10248d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 10258d00608fSStefano Zampini /* store the original rhs */ 10268d00608fSStefano Zampini if (copy_rhs) { 10278d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10288d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10298d00608fSStefano Zampini } 10308d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10313972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 1032*0369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,rhs,rhs);CHKERRQ(ierr); 10333972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10348efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 10357acc28cbSStefano Zampini if (ksp) { 10367acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 10377acc28cbSStefano Zampini } 10383308cffdSStefano Zampini } 10398efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1040b76ba322SStefano Zampini 1041*0369aaf7SStefano Zampini /* When using the benign trick: (TODO: what about FETI-DP?) 1042*0369aaf7SStefano Zampini - change rhs on pressures (iteration matrix is surely of type MATIS) 1043*0369aaf7SStefano Zampini - compute initial vector in benign space 1044*0369aaf7SStefano Zampini */ 1045*0369aaf7SStefano Zampini if (rhs && pcbddc->benign_saddle_point) { 104606f24817SStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 104706f24817SStefano Zampini 104806f24817SStefano Zampini /* store the original rhs */ 104906f24817SStefano Zampini if (copy_rhs) { 105006f24817SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 105106f24817SStefano Zampini copy_rhs = PETSC_FALSE; 105206f24817SStefano Zampini } 105306f24817SStefano Zampini 105406f24817SStefano Zampini /* now change (locally) the basis */ 105506f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 105606f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 105706f24817SStefano Zampini if (pcbddc->benign_change) { 105806f24817SStefano Zampini ierr = MatMultTranspose(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 105906f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 106006f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 106106f24817SStefano Zampini /* change local iteration matrix */ 106206f24817SStefano Zampini ierr = MatPtAP(matis->A,pcbddc->benign_change,MAT_REUSE_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr); 106306f24817SStefano Zampini ierr = MatDestroy(&pcbddc->benign_original_mat);CHKERRQ(ierr); 106406f24817SStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 106506f24817SStefano Zampini pcbddc->benign_original_mat = matis->A; 106606f24817SStefano Zampini ierr = MatDestroy(&matis->A);CHKERRQ(ierr); 106706f24817SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->local_mat);CHKERRQ(ierr); 106806f24817SStefano Zampini matis->A = pcbddc->local_mat; 106906f24817SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 107006f24817SStefano Zampini } else { 107106f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 107206f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 107306f24817SStefano Zampini } 107406f24817SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 107506f24817SStefano Zampini 1076*0369aaf7SStefano Zampini /* compute u^*_h as in Xuemin Tu's thesis (see Section 4.8.1) */ 1077*0369aaf7SStefano Zampini /* TODO: what about Stokes? */ 1078*0369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 1079*0369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 1080*0369aaf7SStefano Zampini } 1081*0369aaf7SStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 1082*0369aaf7SStefano Zampini pcbddc->benign_p0 = 0.; 1083*0369aaf7SStefano Zampini if (pcbddc->benign_p0_lidx >= 0) { 1084*0369aaf7SStefano Zampini const PetscScalar *array; 1085*0369aaf7SStefano Zampini 1086*0369aaf7SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr); 1087*0369aaf7SStefano Zampini pcbddc->benign_p0 = array[pcbddc->benign_p0_lidx]; 1088*0369aaf7SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr); 1089*0369aaf7SStefano Zampini } 1090*0369aaf7SStefano Zampini ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] benign u* should be %1.4e\n",PetscGlobalRank,pcbddc->benign_p0);CHKERRQ(ierr); 1091*0369aaf7SStefano Zampini ierr = PCBDDCBenignPopOrPushP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr); 1092*0369aaf7SStefano Zampini ierr = PCApply_BDDC(pc,pcis->vec1_global,pcbddc->benign_vec);CHKERRQ(ierr); 1093*0369aaf7SStefano Zampini pcbddc->benign_p0 = 0.; 1094*0369aaf7SStefano Zampini ierr = PCBDDCBenignPopOrPushP0(pc,pcbddc->benign_vec,PETSC_FALSE);CHKERRQ(ierr); 1095*0369aaf7SStefano Zampini if (pcbddc->benign_saddle_point) { 1096*0369aaf7SStefano Zampini Mat_IS* matis = (Mat_IS*)(pc->mat->data); 1097*0369aaf7SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcbddc->benign_vec,matis->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1098*0369aaf7SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcbddc->benign_vec,matis->x,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1099*0369aaf7SStefano Zampini 1100*0369aaf7SStefano Zampini if (pcbddc->benign_p0_lidx >=0) { 1101*0369aaf7SStefano Zampini Mat B0; 1102*0369aaf7SStefano Zampini Vec dummy_vec; 1103*0369aaf7SStefano Zampini PetscScalar *array; 1104*0369aaf7SStefano Zampini PetscInt ii[2]; 1105*0369aaf7SStefano Zampini 1106*0369aaf7SStefano Zampini ii[0] = 0; 1107*0369aaf7SStefano Zampini ii[1] = pcbddc->B0_ncol; 1108*0369aaf7SStefano Zampini ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,1,pcis->n,ii,pcbddc->B0_cols,pcbddc->B0_vals,&B0);CHKERRQ(ierr); 1109*0369aaf7SStefano Zampini ierr = MatCreateVecs(B0,NULL,&dummy_vec);CHKERRQ(ierr); 1110*0369aaf7SStefano Zampini ierr = MatMult(B0,matis->x,dummy_vec);CHKERRQ(ierr); 1111*0369aaf7SStefano Zampini ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr); 1112*0369aaf7SStefano Zampini ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] benign u* is %1.4e\n",PetscGlobalRank,array[0]);CHKERRQ(ierr); 1113*0369aaf7SStefano Zampini ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr); 1114*0369aaf7SStefano Zampini ierr = MatDestroy(&B0);CHKERRQ(ierr); 1115*0369aaf7SStefano Zampini ierr = VecDestroy(&dummy_vec);CHKERRQ(ierr); 1116*0369aaf7SStefano Zampini } 1117b76ba322SStefano Zampini } 1118b76ba322SStefano Zampini } 1119b76ba322SStefano Zampini 1120*0369aaf7SStefano Zampini /* change rhs and iteration matrix if using the change of basis */ 1121b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1122906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1123906d46d4SStefano Zampini 1124906d46d4SStefano Zampini /* get change ctx */ 1125906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1126906d46d4SStefano Zampini 1127906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 1128906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1129906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1130906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1131906d46d4SStefano Zampini 1132906d46d4SStefano Zampini /* change iteration matrix */ 1133906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1134906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1135906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 1136906d46d4SStefano Zampini 11378d00608fSStefano Zampini /* store the original rhs */ 11388d00608fSStefano Zampini if (copy_rhs) { 11398d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11408d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 11418d00608fSStefano Zampini } 11428d00608fSStefano Zampini 1143906d46d4SStefano Zampini /* change rhs */ 1144906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1145906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 11468d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 114792e3dcfbSStefano Zampini } 114892e3dcfbSStefano Zampini 1149*0369aaf7SStefano Zampini /* remove non-benign solution from the rhs */ 1150*0369aaf7SStefano Zampini if (pcbddc->benign_saddle_point) { 1151*0369aaf7SStefano Zampini /* store the original rhs */ 1152*0369aaf7SStefano Zampini if (copy_rhs) { 1153*0369aaf7SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1154*0369aaf7SStefano Zampini copy_rhs = PETSC_FALSE; 1155*0369aaf7SStefano Zampini } 1156*0369aaf7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 1157*0369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 1158*0369aaf7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 1159*0369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 1160*0369aaf7SStefano Zampini } 1161*0369aaf7SStefano Zampini 1162*0369aaf7SStefano Zampini /* set initial guess if using PCG */ 1163*0369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 1164*0369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 1165*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1166*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1167*0369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 1168*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1169*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1170*0369aaf7SStefano Zampini if (ksp) { 1171*0369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 1172*0369aaf7SStefano Zampini } 1173*0369aaf7SStefano Zampini } 1174*0369aaf7SStefano Zampini 117592e3dcfbSStefano Zampini /* remove nullspace if present */ 11768efcfb23SStefano Zampini if (ksp && x && pcbddc->NullSpace) { 11778efcfb23SStefano Zampini ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr); 11788d00608fSStefano Zampini /* store the original rhs */ 11798d00608fSStefano Zampini if (copy_rhs) { 11808d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11818d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 11828d00608fSStefano Zampini } 11838d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 1184d0195637SJed Brown ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr); 1185b76ba322SStefano Zampini } 1186534831adSStefano Zampini PetscFunctionReturn(0); 1187534831adSStefano Zampini } 1188906d46d4SStefano Zampini 1189534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1190534831adSStefano Zampini #undef __FUNCT__ 1191534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1192534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1193534831adSStefano Zampini /* 1194534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1195534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1196534831adSStefano Zampini 1197534831adSStefano Zampini Input Parameter: 1198534831adSStefano Zampini + pc - the preconditioner contex 1199534831adSStefano Zampini 1200534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1201534831adSStefano Zampini 1202534831adSStefano Zampini Notes: 1203534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1204534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1205534831adSStefano Zampini */ 1206534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1207534831adSStefano Zampini { 1208534831adSStefano Zampini PetscErrorCode ierr; 1209534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1210534831adSStefano Zampini 1211534831adSStefano Zampini PetscFunctionBegin; 1212b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1213906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1214906d46d4SStefano Zampini 1215906d46d4SStefano Zampini /* get change ctx */ 1216906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1217906d46d4SStefano Zampini 1218906d46d4SStefano Zampini /* restore iteration matrix */ 1219906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1220906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1221906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 1222906d46d4SStefano Zampini 1223906d46d4SStefano Zampini /* get solution in original basis */ 1224906d46d4SStefano Zampini if (x) { 1225906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 122606f24817SStefano Zampini 12272d3346b4SStefano Zampini /* restore solution on pressures */ 122806a4e24aSStefano Zampini if (pcbddc->benign_saddle_point) { 12292d3346b4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->mat->data; 12302d3346b4SStefano Zampini 1231*0369aaf7SStefano Zampini /* add non-benign solution */ 1232*0369aaf7SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1233*0369aaf7SStefano Zampini 1234*0369aaf7SStefano Zampini /* change basis on pressures for x */ 123506f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 123606f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12372d3346b4SStefano Zampini if (pcbddc->benign_change) { 12382d3346b4SStefano Zampini 12392d3346b4SStefano Zampini ierr = MatMult(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 124006f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 124106f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12422d3346b4SStefano Zampini } else { 124306f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 124406f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12452d3346b4SStefano Zampini } 12462d3346b4SStefano Zampini } 1247*0369aaf7SStefano Zampini /* change basis on x */ 1248*0369aaf7SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1249906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 12503425bc38SStefano Zampini } 1251534831adSStefano Zampini } 1252906d46d4SStefano Zampini 12533972b0daSStefano Zampini /* add solution removed in presolve */ 12546bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 12553425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 12563425bc38SStefano Zampini } 1257906d46d4SStefano Zampini 1258fb223d50SStefano Zampini /* restore rhs to its original state */ 12598d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1260fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1261fb223d50SStefano Zampini } 12628d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12638efcfb23SStefano Zampini 12648efcfb23SStefano Zampini /* restore ksp guess state */ 12658efcfb23SStefano Zampini if (ksp) { 12668efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12678efcfb23SStefano Zampini } 1268534831adSStefano Zampini PetscFunctionReturn(0); 1269534831adSStefano Zampini } 1270534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 127153cdbc3dSStefano Zampini #undef __FUNCT__ 127253cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 12730c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 12740c7d97c5SJed Brown /* 12750c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 12760c7d97c5SJed Brown by setting data structures and options. 12770c7d97c5SJed Brown 12780c7d97c5SJed Brown Input Parameter: 127953cdbc3dSStefano Zampini + pc - the preconditioner context 12800c7d97c5SJed Brown 12810c7d97c5SJed Brown Application Interface Routine: PCSetUp() 12820c7d97c5SJed Brown 12830c7d97c5SJed Brown Notes: 12840c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 12850c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 12860c7d97c5SJed Brown */ 128753cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 12880c7d97c5SJed Brown { 12890c7d97c5SJed Brown PetscErrorCode ierr; 12900c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 12915e8657edSStefano Zampini Mat_IS* matis; 129208122e43SStefano Zampini MatNullSpace nearnullspace; 1293c9ed8603SStefano Zampini IS zerodiag = NULL; 129491e8d312SStefano Zampini PetscInt nrows,ncols; 129508122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 12968de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 12975e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 12980c7d97c5SJed Brown 12990c7d97c5SJed Brown PetscFunctionBegin; 13005e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 13015e8657edSStefano Zampini if (!ismatis) { 13025e8657edSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 13035e8657edSStefano Zampini } 130491e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 130591e8d312SStefano Zampini if (nrows != ncols) { 130691e8d312SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 130791e8d312SStefano Zampini } 13085e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1309f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 13103b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1311fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1312f4ddd8eeSStefano Zampini /* split work */ 1313674ae819SStefano Zampini if (pc->setupcalled) { 1314d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1315674ae819SStefano Zampini computetopography = PETSC_FALSE; 1316674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1317674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1318674ae819SStefano Zampini computetopography = PETSC_TRUE; 1319674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1320674ae819SStefano Zampini } 1321674ae819SStefano Zampini } else { 1322674ae819SStefano Zampini computetopography = PETSC_TRUE; 1323674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1324674ae819SStefano Zampini } 1325fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1326fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1327fb180af4SStefano Zampini } 13288de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 13295a95e1ceSStefano Zampini if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) { 13305a95e1ceSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 13315a95e1ceSStefano Zampini } 13325a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1333862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1334862806e4SStefano Zampini 13355a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 13366816873aSStefano Zampini if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) { 13376816873aSStefano 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"); 13386816873aSStefano Zampini } 13392d3346b4SStefano Zampini 13402d3346b4SStefano Zampini /* check if the iteration matrix is of type MATIS in case the benign trick has been requested */ 13412d3346b4SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 134206a4e24aSStefano Zampini if (pcbddc->benign_saddle_point && !ismatis) { 13432d3346b4SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner with benign subspace trick requires the iteration matrix to be of type MATIS"); 13442d3346b4SStefano Zampini } 13452d3346b4SStefano Zampini 1346f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 134770cf5478SStefano Zampini if (pcbddc->dbg_flag) { 134870cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 134958a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 1350f4ddd8eeSStefano Zampini ierr = PetscViewerASCIISynchronizedAllow(pcbddc->dbg_viewer,PETSC_TRUE);CHKERRQ(ierr); 1351f4ddd8eeSStefano Zampini } 135258a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1353f4ddd8eeSStefano Zampini } 1354f4ddd8eeSStefano Zampini 13555e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 13565e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 13575e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 13585e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 13595e8657edSStefano Zampini } else { 1360b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 136106a4e24aSStefano Zampini if (!pcbddc->benign_saddle_point) { 13625e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 13635e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1364c263805aSStefano Zampini } else { /* TODO: handle user change of basis */ 136581d14e9dSStefano Zampini PetscInt nz; 136681d14e9dSStefano Zampini PetscBool sorted; 1367f5fd1fbfSStefano Zampini 1368c9ed8603SStefano Zampini ierr = MatFindZeroDiagonals(matis->A,&zerodiag);CHKERRQ(ierr); 1369c9ed8603SStefano Zampini ierr = ISSorted(zerodiag,&sorted);CHKERRQ(ierr); 137081d14e9dSStefano Zampini if (!sorted) { 1371c9ed8603SStefano Zampini ierr = ISSort(zerodiag);CHKERRQ(ierr); 1372f5fd1fbfSStefano Zampini } 1373c9ed8603SStefano Zampini ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); 137481d14e9dSStefano Zampini if (nz) { 137581d14e9dSStefano Zampini IS zerodiagc; 137681d14e9dSStefano Zampini PetscScalar *array; 1377c263805aSStefano Zampini const PetscInt *idxs,*idxsc; 1378c263805aSStefano Zampini PetscInt i,n,*nnz; 137981d14e9dSStefano Zampini 138081d14e9dSStefano Zampini /* TODO: add check for shared dofs */ 138181d14e9dSStefano Zampini pcbddc->use_change_of_basis = PETSC_TRUE; 1382c9ed8603SStefano Zampini pcbddc->use_change_on_faces = PETSC_TRUE; 138381d14e9dSStefano Zampini ierr = MatGetLocalSize(matis->A,&n,NULL);CHKERRQ(ierr); 1384c9ed8603SStefano Zampini ierr = ISComplement(zerodiag,0,n,&zerodiagc);CHKERRQ(ierr); 1385c9ed8603SStefano Zampini ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr); 138681d14e9dSStefano Zampini ierr = ISGetIndices(zerodiagc,&idxsc);CHKERRQ(ierr); 138781d14e9dSStefano Zampini /* local change of basis for pressures */ 138881d14e9dSStefano Zampini ierr = MatDestroy(&pcbddc->benign_change);CHKERRQ(ierr); 138981d14e9dSStefano Zampini ierr = MatCreate(PetscObjectComm((PetscObject)matis->A),&pcbddc->benign_change);CHKERRQ(ierr); 139081d14e9dSStefano Zampini ierr = MatSetType(pcbddc->benign_change,MATAIJ);CHKERRQ(ierr); 139181d14e9dSStefano Zampini ierr = MatSetSizes(pcbddc->benign_change,n,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 139281d14e9dSStefano Zampini ierr = PetscMalloc1(n,&nnz);CHKERRQ(ierr); 139381d14e9dSStefano Zampini for (i=0;i<n-nz;i++) nnz[idxsc[i]] = 1; /* identity on velocities */ 139481d14e9dSStefano Zampini for (i=0;i<nz-1;i++) nnz[idxs[i]] = 2; /* change on pressures */ 139581d14e9dSStefano Zampini nnz[idxs[nz-1]] = nz; /* last local pressure dof: _0 set */ 139681d14e9dSStefano Zampini ierr = MatSeqAIJSetPreallocation(pcbddc->benign_change,0,nnz);CHKERRQ(ierr); 139781d14e9dSStefano Zampini ierr = PetscFree(nnz);CHKERRQ(ierr); 139881d14e9dSStefano Zampini /* set identity on velocities */ 139981d14e9dSStefano Zampini for (i=0;i<n-nz;i++) { 140081d14e9dSStefano Zampini ierr = MatSetValue(pcbddc->benign_change,idxsc[i],idxsc[i],1.,INSERT_VALUES);CHKERRQ(ierr); 140181d14e9dSStefano Zampini } 140281d14e9dSStefano Zampini /* set change on pressures */ 140381d14e9dSStefano Zampini for (i=0;i<nz-1;i++) { 140481d14e9dSStefano Zampini PetscScalar vals[2]; 140581d14e9dSStefano Zampini PetscInt cols[2]; 140681d14e9dSStefano Zampini 140781d14e9dSStefano Zampini /* TODO: add quadrature */ 140881d14e9dSStefano Zampini cols[0] = idxs[i]; 140981d14e9dSStefano Zampini cols[1] = idxs[nz-1]; 141081d14e9dSStefano Zampini vals[0] = 1.; 141181d14e9dSStefano Zampini vals[1] = 1./nz; 141281d14e9dSStefano Zampini ierr = MatSetValues(pcbddc->benign_change,1,idxs+i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 141381d14e9dSStefano Zampini } 141481d14e9dSStefano Zampini ierr = PetscMalloc1(nz,&array);CHKERRQ(ierr); 141581d14e9dSStefano Zampini for (i=0;i<nz-1;i++) array[i] = -1.; 141681d14e9dSStefano Zampini array[nz-1] = 1./nz; 141781d14e9dSStefano Zampini ierr = MatSetValues(pcbddc->benign_change,1,idxs+nz-1,nz,idxs,array,INSERT_VALUES);CHKERRQ(ierr); 141881d14e9dSStefano Zampini ierr = PetscFree(array);CHKERRQ(ierr); 141981d14e9dSStefano Zampini ierr = MatAssemblyBegin(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 142081d14e9dSStefano Zampini ierr = MatAssemblyEnd(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 142181d14e9dSStefano Zampini /* TODO: need optimization? */ 142281d14e9dSStefano Zampini ierr = MatPtAP(matis->A,pcbddc->benign_change,MAT_INITIAL_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr); 1423efc2fbd9SStefano Zampini /* store local and global idxs for p0 */ 1424efc2fbd9SStefano Zampini pcbddc->benign_p0_lidx = idxs[nz-1]; 1425efc2fbd9SStefano Zampini ierr = ISLocalToGlobalMappingApply(pc->pmat->rmap->mapping,1,&idxs[nz-1],&pcbddc->benign_p0_gidx);CHKERRQ(ierr); 1426c9ed8603SStefano Zampini ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr); 1427c263805aSStefano Zampini ierr = ISRestoreIndices(zerodiagc,&idxsc);CHKERRQ(ierr); 1428c263805aSStefano Zampini ierr = ISDestroy(&zerodiagc);CHKERRQ(ierr); 1429c263805aSStefano Zampini /* pop B0 mat from pcbddc->local_mat */ 1430c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1431019a44ceSStefano Zampini } else { /* this is unlikely to happen but, just in case, destroy the empty IS */ 1432c9ed8603SStefano Zampini pcbddc->benign_p0_lidx = -1; 1433c9ed8603SStefano Zampini pcbddc->benign_p0_gidx = -1; 1434c9ed8603SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 143581d14e9dSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 143681d14e9dSStefano Zampini pcbddc->local_mat = matis->A; 143781d14e9dSStefano Zampini } 1438f5fd1fbfSStefano Zampini } 1439674ae819SStefano Zampini } 1440f4ddd8eeSStefano Zampini 1441e496cd5dSStefano Zampini /* workaround for reals */ 1442e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX) 14433301b35fSStefano Zampini if (matis->A->symmetric_set) { 14443301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 14453301b35fSStefano Zampini } 1446e496cd5dSStefano Zampini #endif 144706a4e24aSStefano Zampini if (matis->A->symmetric_set) { 144806a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 144906a4e24aSStefano Zampini } 145006a4e24aSStefano Zampini if (matis->A->spd_set) { 145106a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 145206a4e24aSStefano Zampini } 1453e496cd5dSStefano Zampini 14545e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 14555e8657edSStefano Zampini { 14565e8657edSStefano Zampini Mat temp_mat; 14575e8657edSStefano Zampini 14585e8657edSStefano Zampini temp_mat = matis->A; 14595e8657edSStefano Zampini matis->A = pcbddc->local_mat; 14605e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 14615e8657edSStefano Zampini pcbddc->local_mat = matis->A; 14625e8657edSStefano Zampini matis->A = temp_mat; 14635e8657edSStefano Zampini } 1464684f6988SStefano Zampini 146581d14e9dSStefano Zampini /* Analyze interface */ 1466674ae819SStefano Zampini if (computetopography) { 1467674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 14688de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1469674ae819SStefano Zampini } 1470fb8d54d4SStefano Zampini 1471c9ed8603SStefano Zampini /* check b(v_I,p_0) = 0 for all v_I */ 1472c9ed8603SStefano Zampini if (pcbddc->dbg_flag && zerodiag) { 147306f24817SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 147406f24817SStefano Zampini IS dirIS = NULL; 147506f24817SStefano Zampini PetscScalar *vals; 147606f24817SStefano Zampini const PetscInt *idxs; 147706f24817SStefano Zampini PetscInt i,nz; 147806f24817SStefano Zampini 147906f24817SStefano Zampini /* p0 */ 148006f24817SStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 148106f24817SStefano Zampini ierr = PetscMalloc1(pcis->n,&vals);CHKERRQ(ierr); 1482c9ed8603SStefano Zampini ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); 1483c9ed8603SStefano Zampini ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr); 148406f24817SStefano Zampini for (i=0;i<nz;i++) vals[i] = 1.; 148506f24817SStefano Zampini ierr = VecSetValues(pcis->vec1_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 148606f24817SStefano Zampini ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); 148706f24817SStefano Zampini ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); 148806f24817SStefano Zampini /* v_I */ 148906f24817SStefano Zampini ierr = VecSetRandom(pcis->vec2_N,NULL);CHKERRQ(ierr); 149006f24817SStefano Zampini for (i=0;i<nz;i++) vals[i] = 0.; 149106f24817SStefano Zampini ierr = VecSetValues(pcis->vec2_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 1492c9ed8603SStefano Zampini ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr); 149306f24817SStefano Zampini ierr = ISGetIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); 149406f24817SStefano Zampini for (i=0;i<pcis->n_B;i++) vals[i] = 0.; 149506f24817SStefano Zampini ierr = VecSetValues(pcis->vec2_N,pcis->n_B,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 149606f24817SStefano Zampini ierr = ISRestoreIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); 149706f24817SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 149806f24817SStefano Zampini if (dirIS) { 149906f24817SStefano Zampini PetscInt n; 150006f24817SStefano Zampini 150106f24817SStefano Zampini ierr = ISGetLocalSize(dirIS,&n);CHKERRQ(ierr); 150206f24817SStefano Zampini ierr = ISGetIndices(dirIS,&idxs);CHKERRQ(ierr); 150306f24817SStefano Zampini for (i=0;i<n;i++) vals[i] = 0.; 150406f24817SStefano Zampini ierr = VecSetValues(pcis->vec2_N,n,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 150506f24817SStefano Zampini ierr = ISRestoreIndices(dirIS,&idxs);CHKERRQ(ierr); 150606f24817SStefano Zampini } 150706f24817SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 150806f24817SStefano Zampini ierr = VecAssemblyBegin(pcis->vec2_N);CHKERRQ(ierr); 150906f24817SStefano Zampini ierr = VecAssemblyEnd(pcis->vec2_N);CHKERRQ(ierr); 151006f24817SStefano Zampini ierr = VecSet(matis->x,0.);CHKERRQ(ierr); 151106f24817SStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,matis->x);CHKERRQ(ierr); 151206f24817SStefano Zampini ierr = VecDot(matis->x,pcis->vec2_N,&vals[0]);CHKERRQ(ierr); 151306f24817SStefano Zampini ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] check original mat: %1.4e\n",PetscGlobalRank,vals[0]);CHKERRQ(ierr); 151406f24817SStefano Zampini ierr = PetscFree(vals);CHKERRQ(ierr); 1515c9ed8603SStefano Zampini } 1516efc2fbd9SStefano Zampini 1517efc2fbd9SStefano Zampini /* check PCBDDCBenignPopOrPush */ 1518c9ed8603SStefano Zampini if (pcbddc->dbg_flag && pcbddc->benign_saddle_point) { 1519c9ed8603SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1520c9ed8603SStefano Zampini 1521efc2fbd9SStefano Zampini ierr = VecSetRandom(pcis->vec1_global,NULL);CHKERRQ(ierr); 1522efc2fbd9SStefano Zampini pcbddc->benign_p0 = -PetscGlobalRank; 1523efc2fbd9SStefano Zampini ierr = PCBDDCBenignPopOrPushP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr); 1524efc2fbd9SStefano Zampini pcbddc->benign_p0 = 1; 1525efc2fbd9SStefano Zampini ierr = PCBDDCBenignPopOrPushP0(pc,pcis->vec1_global,PETSC_TRUE);CHKERRQ(ierr); 1526c9ed8603SStefano Zampini if (pcbddc->benign_p0_gidx >=0 && pcbddc->benign_p0 != -PetscGlobalRank) { 1527efc2fbd9SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error testing PCBDDCBenignPopOrPushP0! Found %1.4e instead of %1.4e\n",pcbddc->benign_p0,-PetscGlobalRank);CHKERRQ(ierr); 1528efc2fbd9SStefano Zampini } 152906f24817SStefano Zampini } 153006f24817SStefano Zampini 1531b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1532684f6988SStefano Zampini if (computesolvers) { 1533d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1534d5574798SStefano Zampini 1535d5574798SStefano Zampini if (computesubschurs && computetopography) { 153608122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1537b1b3d7a2SStefano Zampini } 1538d5574798SStefano Zampini if (sub_schurs->use_mumps) { 15392070dbb6SStefano Zampini if (computesubschurs) { 154008122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 15412070dbb6SStefano Zampini } 1542d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1543d5574798SStefano Zampini } else { 1544d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 15452070dbb6SStefano Zampini if (computesubschurs) { 1546d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1547d5574798SStefano Zampini } 15482070dbb6SStefano Zampini } 154908122e43SStefano Zampini if (pcbddc->adaptive_selection) { 155008122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 15518de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1552b7eb3628SStefano Zampini } 1553b1b3d7a2SStefano Zampini } 1554684f6988SStefano Zampini 1555f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1556fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1557f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1558f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1559f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1560f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1561f4ddd8eeSStefano Zampini } else { 1562f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1563f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1564f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1565165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1566f4ddd8eeSStefano Zampini PetscInt i; 1567165b64e2SStefano Zampini const Vec *nearnullvecs; 1568165b64e2SStefano Zampini PetscObjectState state; 1569165b64e2SStefano Zampini PetscInt nnsp_size; 1570165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1571f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1572f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1573165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1574f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1575f4ddd8eeSStefano Zampini break; 1576f4ddd8eeSStefano Zampini } 1577f4ddd8eeSStefano Zampini } 1578f4ddd8eeSStefano Zampini } 1579f4ddd8eeSStefano Zampini } 1580f4ddd8eeSStefano Zampini } else { 1581f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1582f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1583f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1584f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1585f4ddd8eeSStefano Zampini } 1586f4ddd8eeSStefano Zampini } 1587f4ddd8eeSStefano Zampini 1588f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1589727cdba6SStefano Zampini /* reset primal space flags */ 1590f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1591727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 15928de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1593727cdba6SStefano Zampini /* It also sets the primal space flags */ 1594674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1595e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1596f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 15973975b054SStefano Zampini } 15985e8657edSStefano Zampini 15993975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 16005e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16015e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1602c263805aSStefano Zampini Mat temp_mat = NULL; 16035e8657edSStefano Zampini 1604c9ed8603SStefano Zampini if (zerodiag) { 1605c263805aSStefano Zampini /* insert B0 in pcbddc->local_mat */ 1606c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_FALSE);CHKERRQ(ierr); 160706f24817SStefano Zampini if (pcbddc->dbg_flag) { 160806f24817SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 160906f24817SStefano Zampini IS dirIS = NULL; 161006f24817SStefano Zampini PetscScalar *vals; 161106f24817SStefano Zampini const PetscInt *idxs; 161206f24817SStefano Zampini PetscInt i,nz; 161306f24817SStefano Zampini 161406f24817SStefano Zampini /* p0 */ 161506f24817SStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 161606f24817SStefano Zampini ierr = PetscMalloc1(pcis->n,&vals);CHKERRQ(ierr); 1617c9ed8603SStefano Zampini ierr = ISGetLocalSize(zerodiag,&nz);CHKERRQ(ierr); 1618c9ed8603SStefano Zampini ierr = ISGetIndices(zerodiag,&idxs);CHKERRQ(ierr); 161906f24817SStefano Zampini vals[0] = 1.; 162006f24817SStefano Zampini ierr = VecSetValues(pcis->vec1_N,1,&idxs[nz-1],vals,INSERT_VALUES);CHKERRQ(ierr); 162106f24817SStefano Zampini ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); 162206f24817SStefano Zampini ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); 162306f24817SStefano Zampini /* v_I */ 162406f24817SStefano Zampini ierr = VecSetRandom(pcis->vec2_N,NULL);CHKERRQ(ierr); 162506f24817SStefano Zampini for (i=0;i<nz;i++) vals[i] = 0.; 162606f24817SStefano Zampini ierr = VecSetValues(pcis->vec2_N,nz,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 1627c9ed8603SStefano Zampini ierr = ISRestoreIndices(zerodiag,&idxs);CHKERRQ(ierr); 162806f24817SStefano Zampini ierr = ISGetIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); 162906f24817SStefano Zampini for (i=0;i<pcis->n_B;i++) vals[i] = 0.; 163006f24817SStefano Zampini ierr = VecSetValues(pcis->vec2_N,pcis->n_B,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 163106f24817SStefano Zampini ierr = ISRestoreIndices(pcis->is_B_local,&idxs);CHKERRQ(ierr); 163206f24817SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 163306f24817SStefano Zampini if (dirIS) { 163406f24817SStefano Zampini PetscInt n; 163506f24817SStefano Zampini 163606f24817SStefano Zampini ierr = ISGetLocalSize(dirIS,&n);CHKERRQ(ierr); 163706f24817SStefano Zampini ierr = ISGetIndices(dirIS,&idxs);CHKERRQ(ierr); 163806f24817SStefano Zampini for (i=0;i<n;i++) vals[i] = 0.; 163906f24817SStefano Zampini ierr = VecSetValues(pcis->vec2_N,n,idxs,vals,INSERT_VALUES);CHKERRQ(ierr); 164006f24817SStefano Zampini ierr = ISRestoreIndices(dirIS,&idxs);CHKERRQ(ierr); 164106f24817SStefano Zampini } 164206f24817SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 164306f24817SStefano Zampini ierr = VecAssemblyBegin(pcis->vec2_N);CHKERRQ(ierr); 164406f24817SStefano Zampini ierr = VecAssemblyEnd(pcis->vec2_N);CHKERRQ(ierr); 164506f24817SStefano Zampini ierr = VecSet(matis->x,0.);CHKERRQ(ierr); 164606f24817SStefano Zampini ierr = MatMult(pcbddc->local_mat,pcis->vec1_N,matis->x);CHKERRQ(ierr); 164706f24817SStefano Zampini ierr = VecDot(matis->x,pcis->vec2_N,&vals[0]);CHKERRQ(ierr); 164806f24817SStefano Zampini ierr = PetscPrintf(PETSC_COMM_SELF,"[%d] check new mat: %1.4e\n",PetscGlobalRank,vals[0]);CHKERRQ(ierr); 164906f24817SStefano Zampini ierr = PetscFree(vals);CHKERRQ(ierr); 165006f24817SStefano Zampini } 1651c263805aSStefano Zampini /* hack: swap pointers */ 1652c263805aSStefano Zampini temp_mat = matis->A; 1653c263805aSStefano Zampini matis->A = pcbddc->local_mat; 1654c263805aSStefano Zampini pcbddc->local_mat = NULL; 1655c263805aSStefano Zampini } 16565e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 1657c9ed8603SStefano Zampini if (zerodiag) { 1658c263805aSStefano Zampini /* restore original matrix */ 1659c263805aSStefano Zampini ierr = MatDestroy(&matis->A);CHKERRQ(ierr); 1660c263805aSStefano Zampini matis->A = temp_mat; 1661c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1662c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1663c263805aSStefano Zampini } 16645e8657edSStefano Zampini /* get submatrices */ 16655e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16665e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16675e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16685e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16695e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16705e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16713975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16723975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16735e8657edSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix) { 1674b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16755e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16765e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16775e8657edSStefano Zampini } 1678b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 167999cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1680b96c3477SStefano Zampini /* SetUp Scaling operator */ 1681674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16820c7d97c5SJed Brown } 168370cf5478SStefano Zampini 1684*0369aaf7SStefano Zampini /* check BDDC operator */ 1685*0369aaf7SStefano Zampini if (pcbddc->dbg_flag && ( 1686*0369aaf7SStefano Zampini (pcbddc->n_vertices == pcbddc->local_primal_size) || pcbddc->benign_saddle_point) ) { 1687*0369aaf7SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1688*0369aaf7SStefano Zampini Mat S_j,B0=NULL,B0_B=NULL; 1689*0369aaf7SStefano Zampini Vec dummy_vec=NULL,vec_check_B,vec_scale_P; 1690*0369aaf7SStefano Zampini PetscScalar norm,p0_check,*array,*array2; 1691*0369aaf7SStefano Zampini PetscInt i; 1692*0369aaf7SStefano Zampini 1693*0369aaf7SStefano Zampini /* B0 and B0_B */ 1694*0369aaf7SStefano Zampini if (zerodiag) { 1695*0369aaf7SStefano Zampini IS dummy; 1696*0369aaf7SStefano Zampini PetscInt ii[2]; 1697*0369aaf7SStefano Zampini 1698*0369aaf7SStefano Zampini ii[0] = 0; 1699*0369aaf7SStefano Zampini ii[1] = pcbddc->B0_ncol; 1700*0369aaf7SStefano Zampini ierr = MatCreateSeqAIJWithArrays(PETSC_COMM_SELF,1,pcis->n,ii,pcbddc->B0_cols,pcbddc->B0_vals,&B0);CHKERRQ(ierr); 1701*0369aaf7SStefano Zampini ierr = ISCreateStride(PETSC_COMM_SELF,1,0,1,&dummy);CHKERRQ(ierr); 1702*0369aaf7SStefano Zampini ierr = MatGetSubMatrix(B0,dummy,pcis->is_B_local,MAT_INITIAL_MATRIX,&B0_B);CHKERRQ(ierr); 1703*0369aaf7SStefano Zampini ierr = MatCreateVecs(B0_B,NULL,&dummy_vec);CHKERRQ(ierr); 1704*0369aaf7SStefano Zampini ierr = ISDestroy(&dummy);CHKERRQ(ierr); 1705*0369aaf7SStefano Zampini } 1706*0369aaf7SStefano Zampini /* I need a primal vector to scale primal nodes since BDDC sums contibutions */ 1707*0369aaf7SStefano Zampini ierr = VecDuplicate(pcbddc->vec1_P,&vec_scale_P);CHKERRQ(ierr); 1708*0369aaf7SStefano Zampini ierr = VecSet(pcbddc->vec1_P,1.0);CHKERRQ(ierr); 1709*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1710*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1711*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,vec_scale_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1712*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,vec_scale_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1713*0369aaf7SStefano Zampini ierr = VecReciprocal(vec_scale_P);CHKERRQ(ierr); 1714*0369aaf7SStefano Zampini /* S_j */ 1715*0369aaf7SStefano Zampini ierr = MatCreateSchurComplement(pcis->A_II,pcis->A_II,pcis->A_IB,pcis->A_BI,pcis->A_BB,&S_j);CHKERRQ(ierr); 1716*0369aaf7SStefano Zampini ierr = MatSchurComplementSetKSP(S_j,pcbddc->ksp_D);CHKERRQ(ierr); 1717*0369aaf7SStefano Zampini 1718*0369aaf7SStefano Zampini /* mimic vector in \widetilde{W}_\Gamma */ 1719*0369aaf7SStefano Zampini ierr = VecSetRandom(pcis->vec1_N,NULL);CHKERRQ(ierr); 1720*0369aaf7SStefano Zampini /* continuous in primal space */ 1721*0369aaf7SStefano Zampini ierr = VecSetRandom(pcbddc->coarse_vec,NULL);CHKERRQ(ierr); 1722*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1723*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1724*0369aaf7SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 1725*0369aaf7SStefano Zampini if (zerodiag) { 1726*0369aaf7SStefano Zampini p0_check = array[pcbddc->local_primal_size-1]; 1727*0369aaf7SStefano Zampini } else { 1728*0369aaf7SStefano Zampini p0_check = 0; 1729*0369aaf7SStefano Zampini } 1730*0369aaf7SStefano Zampini ierr = VecSetValues(pcis->vec1_N,pcbddc->local_primal_size,pcbddc->local_primal_ref_node,array,INSERT_VALUES);CHKERRQ(ierr); 1731*0369aaf7SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 1732*0369aaf7SStefano Zampini ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); 1733*0369aaf7SStefano Zampini ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); 1734*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_N,pcis->vec2_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1735*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_N,pcis->vec2_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1736*0369aaf7SStefano Zampini ierr = VecDuplicate(pcis->vec2_B,&vec_check_B);CHKERRQ(ierr); 1737*0369aaf7SStefano Zampini ierr = VecCopy(pcis->vec2_B,vec_check_B);CHKERRQ(ierr); 1738*0369aaf7SStefano Zampini 1739*0369aaf7SStefano Zampini /* assemble rhs for coarse problem */ 1740*0369aaf7SStefano Zampini /* widetilde{S}_\Gamma w_\Gamma + \widetilde{B0}^T_B p0 */ 1741*0369aaf7SStefano Zampini /* local with Schur */ 1742*0369aaf7SStefano Zampini ierr = MatMult(S_j,pcis->vec2_B,pcis->vec1_B);CHKERRQ(ierr); 1743*0369aaf7SStefano Zampini if (zerodiag) { 1744*0369aaf7SStefano Zampini ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr); 1745*0369aaf7SStefano Zampini array[0] = p0_check; 1746*0369aaf7SStefano Zampini ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr); 1747*0369aaf7SStefano Zampini ierr = MatMultTransposeAdd(B0_B,dummy_vec,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 1748*0369aaf7SStefano Zampini } 1749*0369aaf7SStefano Zampini /* sum on primal nodes the local contributions */ 1750*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1751*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1752*0369aaf7SStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); 1753*0369aaf7SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); 1754*0369aaf7SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) array2[i] = array[pcbddc->local_primal_ref_node[i]]; 1755*0369aaf7SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); 1756*0369aaf7SStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); 1757*0369aaf7SStefano Zampini ierr = VecSet(pcbddc->coarse_vec,0.);CHKERRQ(ierr); 1758*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1759*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->vec1_P,pcbddc->coarse_vec,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1760*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1761*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1762*0369aaf7SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 1763*0369aaf7SStefano Zampini /* scale primal nodes (BDDC sums contibutions) */ 1764*0369aaf7SStefano Zampini ierr = VecPointwiseMult(pcbddc->vec1_P,vec_scale_P,pcbddc->vec1_P);CHKERRQ(ierr); 1765*0369aaf7SStefano Zampini ierr = VecSetValues(pcis->vec1_N,pcbddc->local_primal_size,pcbddc->local_primal_ref_node,array,INSERT_VALUES);CHKERRQ(ierr); 1766*0369aaf7SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 1767*0369aaf7SStefano Zampini ierr = VecAssemblyBegin(pcis->vec1_N);CHKERRQ(ierr); 1768*0369aaf7SStefano Zampini ierr = VecAssemblyEnd(pcis->vec1_N);CHKERRQ(ierr); 1769*0369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1770*0369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1771*0369aaf7SStefano Zampini /* global: \widetilde{B0}_B w_\Gamma */ 1772*0369aaf7SStefano Zampini if (zerodiag) { 1773*0369aaf7SStefano Zampini ierr = MatMult(B0_B,pcis->vec2_B,dummy_vec);CHKERRQ(ierr); 1774*0369aaf7SStefano Zampini ierr = VecGetArray(dummy_vec,&array);CHKERRQ(ierr); 1775*0369aaf7SStefano Zampini pcbddc->benign_p0 = array[0]; 1776*0369aaf7SStefano Zampini ierr = VecRestoreArray(dummy_vec,&array);CHKERRQ(ierr); 1777*0369aaf7SStefano Zampini } else { 1778*0369aaf7SStefano Zampini pcbddc->benign_p0 = 0.; 1779*0369aaf7SStefano Zampini } 1780*0369aaf7SStefano Zampini /* BDDC */ 1781*0369aaf7SStefano Zampini ierr = VecSet(pcis->vec1_D,0.);CHKERRQ(ierr); 1782*0369aaf7SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 1783*0369aaf7SStefano Zampini 1784*0369aaf7SStefano Zampini ierr = VecCopy(pcis->vec1_B,pcis->vec2_B);CHKERRQ(ierr); 1785*0369aaf7SStefano Zampini ierr = VecAXPY(pcis->vec1_B,-1.0,vec_check_B);CHKERRQ(ierr); 1786*0369aaf7SStefano Zampini ierr = VecNorm(pcis->vec1_B,NORM_INFINITY,&norm);CHKERRQ(ierr); 1787*0369aaf7SStefano Zampini PetscPrintf(PETSC_COMM_SELF,"[%d] BDDC local error is %1.4e\n",PetscGlobalRank,norm); 1788*0369aaf7SStefano Zampini if (pcbddc->benign_p0_lidx >= 0) { 1789*0369aaf7SStefano Zampini PetscPrintf(PETSC_COMM_SELF,"[%d] BDDC p0 error is %1.4e\n",PetscGlobalRank,PetscAbsScalar(pcbddc->benign_p0-p0_check)); 1790*0369aaf7SStefano Zampini } 1791*0369aaf7SStefano Zampini 1792*0369aaf7SStefano Zampini ierr = VecDestroy(&vec_scale_P);CHKERRQ(ierr); 1793*0369aaf7SStefano Zampini ierr = VecDestroy(&vec_check_B);CHKERRQ(ierr); 1794*0369aaf7SStefano Zampini ierr = VecDestroy(&dummy_vec);CHKERRQ(ierr); 1795*0369aaf7SStefano Zampini ierr = MatDestroy(&S_j);CHKERRQ(ierr); 1796*0369aaf7SStefano Zampini ierr = MatDestroy(&B0);CHKERRQ(ierr); 1797*0369aaf7SStefano Zampini ierr = MatDestroy(&B0_B);CHKERRQ(ierr); 1798*0369aaf7SStefano Zampini } 1799*0369aaf7SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 1800*0369aaf7SStefano Zampini 180158a03d70SStefano Zampini if (pcbddc->dbg_flag) { 180258a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 18032b510759SStefano Zampini } 18040c7d97c5SJed Brown PetscFunctionReturn(0); 18050c7d97c5SJed Brown } 18060c7d97c5SJed Brown 18070c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 18080c7d97c5SJed Brown /* 180950efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 18100c7d97c5SJed Brown 18110c7d97c5SJed Brown Input Parameters: 18120f202f7eSStefano Zampini + pc - the preconditioner context 18130f202f7eSStefano Zampini - r - input vector (global) 18140c7d97c5SJed Brown 18150c7d97c5SJed Brown Output Parameter: 18160c7d97c5SJed Brown . z - output vector (global) 18170c7d97c5SJed Brown 18180c7d97c5SJed Brown Application Interface Routine: PCApply() 18190c7d97c5SJed Brown */ 18200c7d97c5SJed Brown #undef __FUNCT__ 18210c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 182253cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 18230c7d97c5SJed Brown { 18240c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 18250c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1826b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 18270c7d97c5SJed Brown PetscErrorCode ierr; 18283b03a366Sstefano_zampini const PetscScalar one = 1.0; 18293b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 18302617d88aSStefano Zampini const PetscScalar zero = 0.0; 18310c7d97c5SJed Brown 18320c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 18330c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1834b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 18350c7d97c5SJed Brown 18360c7d97c5SJed Brown PetscFunctionBegin; 1837efc2fbd9SStefano Zampini if (pcbddc->benign_saddle_point) { /* extract p0 from r */ 1838efc2fbd9SStefano Zampini ierr = PCBDDCBenignPopOrPushP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1839efc2fbd9SStefano Zampini } 184085c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1841b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 18420c7d97c5SJed Brown /* First Dirichlet solve */ 18430c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18440c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18450c7d97c5SJed Brown /* 18460c7d97c5SJed Brown Assembling right hand side for BDDC operator 1847b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1848674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 18490c7d97c5SJed Brown */ 1850b097fa66SStefano Zampini if (n_D) { 1851b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 18520c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 18538eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1854b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1855b097fa66SStefano Zampini } else { 1856b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1857b097fa66SStefano Zampini } 18580c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18590c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1860674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1861b76ba322SStefano Zampini } else { 1862b097fa66SStefano Zampini if (pcbddc->switch_static) { 18630bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1864b097fa66SStefano Zampini } 1865674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1866b76ba322SStefano Zampini } 1867b76ba322SStefano Zampini 18682617d88aSStefano Zampini /* Apply interface preconditioner 18692617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1870dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 18712617d88aSStefano Zampini 1872674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1873674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 18740c7d97c5SJed Brown 18753b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 18760c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18770c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1878b097fa66SStefano Zampini if (n_B) { 18790c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 18808eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1881b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1882b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1883b097fa66SStefano Zampini } 1884df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1885efc2fbd9SStefano Zampini 1886b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1887b097fa66SStefano Zampini if (pcbddc->switch_static) { 1888b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1889b097fa66SStefano Zampini } else { 1890b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1891b097fa66SStefano Zampini } 18920c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18930c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1894b097fa66SStefano Zampini } else { 1895b097fa66SStefano Zampini if (pcbddc->switch_static) { 1896b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1897b097fa66SStefano Zampini } else { 1898b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1899b097fa66SStefano Zampini } 1900b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1901b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1902b097fa66SStefano Zampini } 1903efc2fbd9SStefano Zampini 1904efc2fbd9SStefano Zampini if (pcbddc->benign_saddle_point) { /* push p0 (computed in PCBDDCApplyInterface) */ 1905efc2fbd9SStefano Zampini ierr = PCBDDCBenignPopOrPushP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1906efc2fbd9SStefano Zampini } 19070c7d97c5SJed Brown PetscFunctionReturn(0); 19080c7d97c5SJed Brown } 190950efa1b5SStefano Zampini 191050efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 191150efa1b5SStefano Zampini /* 191250efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 191350efa1b5SStefano Zampini 191450efa1b5SStefano Zampini Input Parameters: 19150f202f7eSStefano Zampini + pc - the preconditioner context 19160f202f7eSStefano Zampini - r - input vector (global) 191750efa1b5SStefano Zampini 191850efa1b5SStefano Zampini Output Parameter: 191950efa1b5SStefano Zampini . z - output vector (global) 192050efa1b5SStefano Zampini 192150efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 192250efa1b5SStefano Zampini */ 192350efa1b5SStefano Zampini #undef __FUNCT__ 192450efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 192550efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 192650efa1b5SStefano Zampini { 192750efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 192850efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1929b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 193050efa1b5SStefano Zampini PetscErrorCode ierr; 193150efa1b5SStefano Zampini const PetscScalar one = 1.0; 193250efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 193350efa1b5SStefano Zampini const PetscScalar zero = 0.0; 193450efa1b5SStefano Zampini 193550efa1b5SStefano Zampini PetscFunctionBegin; 193650efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1937b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 193850efa1b5SStefano Zampini /* First Dirichlet solve */ 193950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 194050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 194150efa1b5SStefano Zampini /* 194250efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1943b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 194450efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 194550efa1b5SStefano Zampini */ 1946b097fa66SStefano Zampini if (n_D) { 1947b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 194850efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 194950efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1950b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1951b097fa66SStefano Zampini } else { 1952b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1953b097fa66SStefano Zampini } 195450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 195550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 195650efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 195750efa1b5SStefano Zampini } else { 1958b097fa66SStefano Zampini if (pcbddc->switch_static) { 195950efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1960b097fa66SStefano Zampini } 196150efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 196250efa1b5SStefano Zampini } 196350efa1b5SStefano Zampini 196450efa1b5SStefano Zampini /* Apply interface preconditioner 196550efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1966dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 196750efa1b5SStefano Zampini 196850efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 196950efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 197050efa1b5SStefano Zampini 197150efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 197250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 197350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1974b097fa66SStefano Zampini if (n_B) { 197550efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 197650efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1977b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1978b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1979b097fa66SStefano Zampini } 1980b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1981b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1982b097fa66SStefano Zampini if (pcbddc->switch_static) { 1983b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1984b097fa66SStefano Zampini } else { 1985b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1986b097fa66SStefano Zampini } 198750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 198850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1989b097fa66SStefano Zampini } else { 1990b097fa66SStefano Zampini if (pcbddc->switch_static) { 1991b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1992b097fa66SStefano Zampini } else { 1993b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1994b097fa66SStefano Zampini } 1995b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1996b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1997b097fa66SStefano Zampini } 199850efa1b5SStefano Zampini PetscFunctionReturn(0); 199950efa1b5SStefano Zampini } 2000da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 2001674ae819SStefano Zampini 2002da1bb401SStefano Zampini #undef __FUNCT__ 2003da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 2004da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 2005da1bb401SStefano Zampini { 2006da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2007da1bb401SStefano Zampini PetscErrorCode ierr; 2008da1bb401SStefano Zampini 2009da1bb401SStefano Zampini PetscFunctionBegin; 2010da1bb401SStefano Zampini /* free data created by PCIS */ 2011da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 2012674ae819SStefano Zampini /* free BDDC custom data */ 2013674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2014674ae819SStefano Zampini /* destroy objects related to topography */ 2015674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 2016674ae819SStefano Zampini /* free allocated graph structure */ 2017da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 2018b96c3477SStefano Zampini /* free allocated sub schurs structure */ 2019b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 202034a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2021674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 202234a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 2023674ae819SStefano Zampini /* free solvers stuff */ 2024674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 202562a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 202662a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 202762a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 2028906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 2029906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 2030906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 2031906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 2032906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 2033906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 2034906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 2035906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 2036906d46d4SStefano Zampini } 2037906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 20383425bc38SStefano Zampini /* remove functions */ 2039906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2040674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 2041bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 20422b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2043b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 20442b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2045bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 2046bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 204782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2048bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 204982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2050bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 205182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2052bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2053785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2054bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 205563602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2056bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2057bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2058bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2059bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2060674ae819SStefano Zampini /* Free the private data structure */ 2061674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2062da1bb401SStefano Zampini PetscFunctionReturn(0); 2063da1bb401SStefano Zampini } 20643425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20651e6b0712SBarry Smith 20663425bc38SStefano Zampini #undef __FUNCT__ 20673425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 20683425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 20693425bc38SStefano Zampini { 2070674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2071c08af4c6SStefano Zampini Vec copy_standard_rhs; 20723425bc38SStefano Zampini PC_IS* pcis; 20733425bc38SStefano Zampini PC_BDDC* pcbddc; 20743425bc38SStefano Zampini PetscErrorCode ierr; 20750c7d97c5SJed Brown 20763425bc38SStefano Zampini PetscFunctionBegin; 20773425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20783425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20793425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20803425bc38SStefano Zampini 2081c08af4c6SStefano Zampini /* 2082c08af4c6SStefano Zampini change of basis for physical rhs if needed 2083c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2084c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 2085c08af4c6SStefano Zampini */ 2086c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 2087c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 2088c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 20893425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2090c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2091c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2092fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2093fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2094c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2095c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2096674ae819SStefano Zampini /* Apply partition of unity */ 20973425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2098c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 20998eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 21003425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 21013425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21023425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 21033425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2104c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 2105c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2106c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2107c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2108c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2109c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21103425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21113425bc38SStefano Zampini } 2112c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 21133425bc38SStefano Zampini /* BDDC rhs */ 21143425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 21158eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21163425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21173425bc38SStefano Zampini } 21183425bc38SStefano Zampini /* apply BDDC */ 2119dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 21203425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 21213425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 21223425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 21233425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21243425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21253425bc38SStefano Zampini PetscFunctionReturn(0); 21263425bc38SStefano Zampini } 21271e6b0712SBarry Smith 21283425bc38SStefano Zampini #undef __FUNCT__ 21293425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 21303425bc38SStefano Zampini /*@ 21310f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 21323425bc38SStefano Zampini 21333425bc38SStefano Zampini Collective 21343425bc38SStefano Zampini 21353425bc38SStefano Zampini Input Parameters: 21360f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 21370f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 21383425bc38SStefano Zampini 21393425bc38SStefano Zampini Output Parameters: 21400f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 21413425bc38SStefano Zampini 21423425bc38SStefano Zampini Level: developer 21433425bc38SStefano Zampini 21443425bc38SStefano Zampini Notes: 21453425bc38SStefano Zampini 21460f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 21473425bc38SStefano Zampini @*/ 21483425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21493425bc38SStefano Zampini { 2150674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21513425bc38SStefano Zampini PetscErrorCode ierr; 21523425bc38SStefano Zampini 21533425bc38SStefano Zampini PetscFunctionBegin; 21543425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21553425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 21563425bc38SStefano Zampini PetscFunctionReturn(0); 21573425bc38SStefano Zampini } 21583425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 21591e6b0712SBarry Smith 21603425bc38SStefano Zampini #undef __FUNCT__ 21613425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 21623425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 21633425bc38SStefano Zampini { 2164674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21653425bc38SStefano Zampini PC_IS* pcis; 21663425bc38SStefano Zampini PC_BDDC* pcbddc; 21673425bc38SStefano Zampini PetscErrorCode ierr; 21683425bc38SStefano Zampini 21693425bc38SStefano Zampini PetscFunctionBegin; 21703425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21713425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21723425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21733425bc38SStefano Zampini 21743425bc38SStefano Zampini /* apply B_delta^T */ 21753425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21763425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21773425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 21783425bc38SStefano Zampini /* compute rhs for BDDC application */ 21793425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21808eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21813425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21823425bc38SStefano Zampini } 21833425bc38SStefano Zampini /* apply BDDC */ 2184dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 21853425bc38SStefano Zampini /* put values into standard global vector */ 21863425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21873425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21888eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 21893425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 21903425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 21913425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 21923425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21933425bc38SStefano Zampini } 21943425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21953425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21963425bc38SStefano Zampini /* final change of basis if needed 21973425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 21983308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 21993425bc38SStefano Zampini PetscFunctionReturn(0); 22003425bc38SStefano Zampini } 22011e6b0712SBarry Smith 22023425bc38SStefano Zampini #undef __FUNCT__ 22033425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 22043425bc38SStefano Zampini /*@ 22050f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 22063425bc38SStefano Zampini 22073425bc38SStefano Zampini Collective 22083425bc38SStefano Zampini 22093425bc38SStefano Zampini Input Parameters: 22100f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 22110f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 22123425bc38SStefano Zampini 22133425bc38SStefano Zampini Output Parameters: 22140f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 22153425bc38SStefano Zampini 22163425bc38SStefano Zampini Level: developer 22173425bc38SStefano Zampini 22183425bc38SStefano Zampini Notes: 22193425bc38SStefano Zampini 22200f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 22213425bc38SStefano Zampini @*/ 22223425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22233425bc38SStefano Zampini { 2224674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22253425bc38SStefano Zampini PetscErrorCode ierr; 22263425bc38SStefano Zampini 22273425bc38SStefano Zampini PetscFunctionBegin; 22283425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 22293425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 22303425bc38SStefano Zampini PetscFunctionReturn(0); 22313425bc38SStefano Zampini } 22323425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 22331e6b0712SBarry Smith 2234f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2235edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2236f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2237f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2238edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2239f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2240674ae819SStefano Zampini 22413425bc38SStefano Zampini #undef __FUNCT__ 22423425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 22433425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 22443425bc38SStefano Zampini { 2245674ae819SStefano Zampini 2246674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 22473425bc38SStefano Zampini Mat newmat; 2248674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 22493425bc38SStefano Zampini PC newpc; 2250ce94432eSBarry Smith MPI_Comm comm; 22513425bc38SStefano Zampini PetscErrorCode ierr; 22523425bc38SStefano Zampini 22533425bc38SStefano Zampini PetscFunctionBegin; 2254ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 22553425bc38SStefano Zampini /* FETIDP linear matrix */ 22563425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 22573425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 22583425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 22593425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2260edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 22613425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 22623425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 22633425bc38SStefano Zampini /* FETIDP preconditioner */ 22643425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 22653425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 22663425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 22673425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 22683425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 22693425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2270edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 22713425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 227223ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 22733425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 22743425bc38SStefano Zampini /* return pointers for objects created */ 22753425bc38SStefano Zampini *fetidp_mat=newmat; 22763425bc38SStefano Zampini *fetidp_pc=newpc; 22773425bc38SStefano Zampini PetscFunctionReturn(0); 22783425bc38SStefano Zampini } 22791e6b0712SBarry Smith 22803425bc38SStefano Zampini #undef __FUNCT__ 22813425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 22823425bc38SStefano Zampini /*@ 22830f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 22843425bc38SStefano Zampini 22853425bc38SStefano Zampini Collective 22863425bc38SStefano Zampini 22873425bc38SStefano Zampini Input Parameters: 22880f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 228928509bceSStefano Zampini 229028509bceSStefano Zampini Output Parameters: 22910f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 22920f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 229328509bceSStefano Zampini 229428509bceSStefano Zampini Options Database Keys: 22950f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 22963425bc38SStefano Zampini 22973425bc38SStefano Zampini Level: developer 22983425bc38SStefano Zampini 22993425bc38SStefano Zampini Notes: 23000f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 23013425bc38SStefano Zampini 23020f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 23033425bc38SStefano Zampini @*/ 23043425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 23053425bc38SStefano Zampini { 23063425bc38SStefano Zampini PetscErrorCode ierr; 23073425bc38SStefano Zampini 23083425bc38SStefano Zampini PetscFunctionBegin; 23093425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 23103425bc38SStefano Zampini if (pc->setupcalled) { 2311516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2312f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 23133425bc38SStefano Zampini PetscFunctionReturn(0); 23143425bc38SStefano Zampini } 23150c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2316da1bb401SStefano Zampini /*MC 2317da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 23180c7d97c5SJed Brown 231928509bceSStefano Zampini An implementation of the BDDC preconditioner based on 232028509bceSStefano Zampini 232128509bceSStefano Zampini .vb 232228509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 232328509bceSStefano 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 232428509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 23250f202f7eSStefano 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 232628509bceSStefano Zampini .ve 232728509bceSStefano Zampini 232828509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 232928509bceSStefano Zampini 23300f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 233128509bceSStefano Zampini 233228509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 233328509bceSStefano Zampini 2334b6fdb6dfSStefano 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. 2335b6fdb6dfSStefano Zampini 23360f202f7eSStefano Zampini Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 233728509bceSStefano Zampini 23380f202f7eSStefano 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() 23390f202f7eSStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS() 234028509bceSStefano Zampini 23410f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 234228509bceSStefano Zampini 23430f202f7eSStefano 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. 23440f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 234528509bceSStefano Zampini 23460f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 234728509bceSStefano Zampini 23480f202f7eSStefano Zampini Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS is present. Future versions of the code will also consider using MKL_PARDISO or PASTIX. 234928509bceSStefano Zampini 23500f202f7eSStefano 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. 23510f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 23520f202f7eSStefano Zampini 23530f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 23540f202f7eSStefano Zampini 23550f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 23560f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 23570f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 23580f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 23590f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 23600f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 23610f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 236228509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 23630f202f7eSStefano 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) 23640f202f7eSStefano 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) 23650f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 23660f202f7eSStefano 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) 23670f202f7eSStefano 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 installed) 236828509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 236928509bceSStefano Zampini 237028509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 237128509bceSStefano Zampini .vb 237228509bceSStefano Zampini -pc_bddc_dirichlet_ 237328509bceSStefano Zampini -pc_bddc_neumann_ 237428509bceSStefano Zampini -pc_bddc_coarse_ 237528509bceSStefano Zampini .ve 23760f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 237728509bceSStefano Zampini 23780f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 237928509bceSStefano Zampini .vb 2380312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2381312be037SStefano Zampini -pc_bddc_neumann_lN_ 2382312be037SStefano Zampini -pc_bddc_coarse_lN_ 238328509bceSStefano Zampini .ve 23840f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 23850f202f7eSStefano 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. 23860f202f7eSStefano Zampini .vb 23870f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 23880f202f7eSStefano Zampini .ve 23890f202f7eSStefano 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 2390da1bb401SStefano Zampini 2391da1bb401SStefano Zampini Level: intermediate 2392da1bb401SStefano Zampini 2393b6fdb6dfSStefano Zampini Developer notes: 2394da1bb401SStefano Zampini 2395da1bb401SStefano Zampini Contributed by Stefano Zampini 2396da1bb401SStefano Zampini 2397da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2398da1bb401SStefano Zampini M*/ 2399b2573a8aSBarry Smith 2400da1bb401SStefano Zampini #undef __FUNCT__ 2401da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 24028cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2403da1bb401SStefano Zampini { 2404da1bb401SStefano Zampini PetscErrorCode ierr; 2405da1bb401SStefano Zampini PC_BDDC *pcbddc; 2406da1bb401SStefano Zampini 2407da1bb401SStefano Zampini PetscFunctionBegin; 2408da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2409b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2410da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2411da1bb401SStefano Zampini 2412da1bb401SStefano Zampini /* create PCIS data structure */ 2413da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2414da1bb401SStefano Zampini 241547d04d0dSStefano Zampini /* BDDC customization */ 241608a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 241747d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 241847d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 241947d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 242047d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 242147d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 242247d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2423fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2424fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 24253301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 242606a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 242747d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2428b9d89cd5SStefano Zampini /* private */ 2429727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 24300e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 24310e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 24320e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2433e9189074SStefano Zampini pcbddc->n_vertices = 0; 2434727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2435fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 243668457ee5SStefano Zampini pcbddc->coarse_size = -1; 2437f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2438727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2439f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2440f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2441f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2442674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 24430bdf917eSStefano Zampini pcbddc->NullSpace = 0; 24443972b0daSStefano Zampini pcbddc->temp_solution = 0; 2445534831adSStefano Zampini pcbddc->original_rhs = 0; 2446534831adSStefano Zampini pcbddc->local_mat = 0; 2447534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2448b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2449906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 2450da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2451da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2452da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2453da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 245415aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 245515aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2456da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2457da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2458da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2459da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2460da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2461da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2462da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2463da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2464da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2465da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2466785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2467785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2468785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 246960d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 247060d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 247163602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2472da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 247363602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2474da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 247585c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 247647d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 247747d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2478b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 24794fad6a16SStefano Zampini pcbddc->current_level = 0; 24802b510759SStefano Zampini pcbddc->max_levels = 0; 2481323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 248274e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2483f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2484323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 248581d14e9dSStefano Zampini 248681d14e9dSStefano Zampini /* benign subspace trick */ 248781d14e9dSStefano Zampini pcbddc->B0_ncol = 0; 248881d14e9dSStefano Zampini pcbddc->B0_cols = NULL; 248981d14e9dSStefano Zampini pcbddc->B0_vals = NULL; 249081d14e9dSStefano Zampini pcbddc->benign_change = 0; 2491*0369aaf7SStefano Zampini pcbddc->benign_vec = 0; 2492*0369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 2493*0369aaf7SStefano Zampini pcbddc->benign_sf = 0; 249481d14e9dSStefano Zampini 2495674ae819SStefano Zampini /* create local graph structure */ 2496674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2497674ae819SStefano Zampini 2498674ae819SStefano Zampini /* scaling */ 2499674ae819SStefano Zampini pcbddc->work_scaling = 0; 250034a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2501ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2502b96c3477SStefano Zampini 2503b96c3477SStefano Zampini /* create sub schurs structure */ 2504b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2505b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2506b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2507b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2508b96c3477SStefano Zampini 2509b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2510da1bb401SStefano Zampini 2511b7eb3628SStefano Zampini /* adaptivity */ 2512f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 251308122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2514f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2515b7eb3628SStefano Zampini 2516da1bb401SStefano Zampini /* function pointers */ 2517da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 251893bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2519da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2520da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2521da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2522da1bb401SStefano Zampini pc->ops->view = 0; 2523da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2524da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2525da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2526534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2527534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2528da1bb401SStefano Zampini 2529da1bb401SStefano Zampini /* composing function */ 2530906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2531674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 2532bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 25332b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2534b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 25352b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2536bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2537bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 253882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2539bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 254082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2541bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 254282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2543bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 254482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2545bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 254663602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2547bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2548bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2549bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2550bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2551da1bb401SStefano Zampini PetscFunctionReturn(0); 2552da1bb401SStefano Zampini } 25533425bc38SStefano Zampini 2554