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 290c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 300c7d97c5SJed Brown #undef __FUNCT__ 310c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 328c34d3f5SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptions *PetscOptionsObject,PC pc) 330c7d97c5SJed Brown { 340c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 350c7d97c5SJed Brown PetscErrorCode ierr; 360c7d97c5SJed Brown 370c7d97c5SJed Brown PetscFunctionBegin; 38e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 398eeda7d8SStefano Zampini /* Verbose debugging */ 408eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 418eeda7d8SStefano Zampini /* Primal space cumstomization */ 4208a5cf49SStefano 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); 438eeda7d8SStefano 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); 448eeda7d8SStefano 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); 458eeda7d8SStefano 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); 466661aa1dSStefano 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); 47ac632422SStefano 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); 488eeda7d8SStefano Zampini /* Change of basis */ 49b9b85e73SStefano 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); 50b9b85e73SStefano 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); 51674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 52674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 53674ae819SStefano Zampini } 548eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 558eeda7d8SStefano 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); 5674e2c79eSStefano 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); 570298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 582b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 59323d395dSStefano 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); 60674ae819SStefano 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); 61b96c3477SStefano 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); 62b96c3477SStefano 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); 63b96c3477SStefano 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); 64ac632422SStefano 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); 654c6709b3SStefano 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); 6608122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 6708122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 683301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 69b0c7d250SStefano 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); 700c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 710c7d97c5SJed Brown PetscFunctionReturn(0); 720c7d97c5SJed Brown } 730c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 74674ae819SStefano Zampini #undef __FUNCT__ 75906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 76906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 77b9b85e73SStefano Zampini { 78b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79b9b85e73SStefano Zampini PetscErrorCode ierr; 80b9b85e73SStefano Zampini 81b9b85e73SStefano Zampini PetscFunctionBegin; 82b9b85e73SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 83b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 84b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 85b9b85e73SStefano Zampini PetscFunctionReturn(0); 86b9b85e73SStefano Zampini } 87b9b85e73SStefano Zampini #undef __FUNCT__ 88906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 89b9b85e73SStefano Zampini /*@ 90906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 91b9b85e73SStefano Zampini 92b9b85e73SStefano Zampini Collective on PC 93b9b85e73SStefano Zampini 94b9b85e73SStefano Zampini Input Parameters: 95b9b85e73SStefano Zampini + pc - the preconditioning context 96906d46d4SStefano Zampini - change - the change of basis matrix 97b9b85e73SStefano Zampini 98b9b85e73SStefano Zampini Level: intermediate 99b9b85e73SStefano Zampini 100b9b85e73SStefano Zampini Notes: 101b9b85e73SStefano Zampini 102b9b85e73SStefano Zampini .seealso: PCBDDC 103b9b85e73SStefano Zampini @*/ 104906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 105b9b85e73SStefano Zampini { 106b9b85e73SStefano Zampini PetscErrorCode ierr; 107b9b85e73SStefano Zampini 108b9b85e73SStefano Zampini PetscFunctionBegin; 109b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 110b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 111906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 112906d46d4SStefano Zampini if (pc->mat) { 113906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 114906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 115906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 116906d46d4SStefano Zampini if (rows_c != rows) { 117906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 118906d46d4SStefano Zampini } 119906d46d4SStefano Zampini if (cols_c != cols) { 120906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 121906d46d4SStefano Zampini } 122906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 123906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 124906d46d4SStefano Zampini if (rows_c != rows) { 125906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 126906d46d4SStefano Zampini } 127906d46d4SStefano Zampini if (cols_c != cols) { 128906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 129906d46d4SStefano Zampini } 130906d46d4SStefano Zampini } 131906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 132b9b85e73SStefano Zampini PetscFunctionReturn(0); 133b9b85e73SStefano Zampini } 134b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 135b9b85e73SStefano Zampini #undef __FUNCT__ 136674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 137674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 138674ae819SStefano Zampini { 139674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 140674ae819SStefano Zampini PetscErrorCode ierr; 1411e6b0712SBarry Smith 142674ae819SStefano Zampini PetscFunctionBegin; 143674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 144674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 145674ae819SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 146674ae819SStefano Zampini PetscFunctionReturn(0); 147674ae819SStefano Zampini } 148674ae819SStefano Zampini #undef __FUNCT__ 149674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 150674ae819SStefano Zampini /*@ 15128509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 152674ae819SStefano Zampini 15317eb1463SStefano Zampini Collective 154674ae819SStefano Zampini 155674ae819SStefano Zampini Input Parameters: 156674ae819SStefano Zampini + pc - the preconditioning context 15717eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 158674ae819SStefano Zampini 159674ae819SStefano Zampini Level: intermediate 160674ae819SStefano Zampini 161674ae819SStefano Zampini Notes: 162674ae819SStefano Zampini 163674ae819SStefano Zampini .seealso: PCBDDC 164674ae819SStefano Zampini @*/ 165674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 166674ae819SStefano Zampini { 167674ae819SStefano Zampini PetscErrorCode ierr; 168674ae819SStefano Zampini 169674ae819SStefano Zampini PetscFunctionBegin; 170674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 171674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 17217eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 173674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 174674ae819SStefano Zampini PetscFunctionReturn(0); 175674ae819SStefano Zampini } 176674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 1770c7d97c5SJed Brown #undef __FUNCT__ 1784fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 1794fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 1804fad6a16SStefano Zampini { 1814fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1824fad6a16SStefano Zampini 1834fad6a16SStefano Zampini PetscFunctionBegin; 1844fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 1854fad6a16SStefano Zampini PetscFunctionReturn(0); 1864fad6a16SStefano Zampini } 1871e6b0712SBarry Smith 1884fad6a16SStefano Zampini #undef __FUNCT__ 1894fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 1904fad6a16SStefano Zampini /*@ 19128509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 1924fad6a16SStefano Zampini 1934fad6a16SStefano Zampini Logically collective on PC 1944fad6a16SStefano Zampini 1954fad6a16SStefano Zampini Input Parameters: 1964fad6a16SStefano Zampini + pc - the preconditioning context 19728509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 1984fad6a16SStefano Zampini 1990f202f7eSStefano Zampini Options Database Keys: 2000f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 2014fad6a16SStefano Zampini 2024fad6a16SStefano Zampini Level: intermediate 2034fad6a16SStefano Zampini 2044fad6a16SStefano Zampini Notes: 2050f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 2064fad6a16SStefano Zampini 2070f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2084fad6a16SStefano Zampini @*/ 2094fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2104fad6a16SStefano Zampini { 2114fad6a16SStefano Zampini PetscErrorCode ierr; 2124fad6a16SStefano Zampini 2134fad6a16SStefano Zampini PetscFunctionBegin; 2144fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2152b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2164fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2174fad6a16SStefano Zampini PetscFunctionReturn(0); 2184fad6a16SStefano Zampini } 2192b510759SStefano Zampini 220b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2212b510759SStefano Zampini #undef __FUNCT__ 222b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 223b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 224b8ffe317SStefano Zampini { 225b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 226b8ffe317SStefano Zampini 227b8ffe317SStefano Zampini PetscFunctionBegin; 22885c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 229b8ffe317SStefano Zampini PetscFunctionReturn(0); 230b8ffe317SStefano Zampini } 231b8ffe317SStefano Zampini 232b8ffe317SStefano Zampini #undef __FUNCT__ 233b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 234b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 2352b510759SStefano Zampini { 2362b510759SStefano Zampini PetscErrorCode ierr; 2372b510759SStefano Zampini 2382b510759SStefano Zampini PetscFunctionBegin; 2392b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 240b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 241b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 2422b510759SStefano Zampini PetscFunctionReturn(0); 2432b510759SStefano Zampini } 2441e6b0712SBarry Smith 2454fad6a16SStefano Zampini #undef __FUNCT__ 2462b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 2472b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 2484fad6a16SStefano Zampini { 2494fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2504fad6a16SStefano Zampini 2514fad6a16SStefano Zampini PetscFunctionBegin; 2522b510759SStefano Zampini pcbddc->current_level = level; 2534fad6a16SStefano Zampini PetscFunctionReturn(0); 2544fad6a16SStefano Zampini } 2551e6b0712SBarry Smith 2564fad6a16SStefano Zampini #undef __FUNCT__ 257b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 258b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 259b8ffe317SStefano Zampini { 260b8ffe317SStefano Zampini PetscErrorCode ierr; 261b8ffe317SStefano Zampini 262b8ffe317SStefano Zampini PetscFunctionBegin; 263b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 264b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 265b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 266b8ffe317SStefano Zampini PetscFunctionReturn(0); 267b8ffe317SStefano Zampini } 268b8ffe317SStefano Zampini 269b8ffe317SStefano Zampini #undef __FUNCT__ 2702b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 2712b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 2722b510759SStefano Zampini { 2732b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2742b510759SStefano Zampini 2752b510759SStefano Zampini PetscFunctionBegin; 2762b510759SStefano Zampini pcbddc->max_levels = levels; 2772b510759SStefano Zampini PetscFunctionReturn(0); 2782b510759SStefano Zampini } 2792b510759SStefano Zampini 2802b510759SStefano Zampini #undef __FUNCT__ 2812b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 2824fad6a16SStefano Zampini /*@ 28328509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 2844fad6a16SStefano Zampini 2854fad6a16SStefano Zampini Logically collective on PC 2864fad6a16SStefano Zampini 2874fad6a16SStefano Zampini Input Parameters: 2884fad6a16SStefano Zampini + pc - the preconditioning context 28928509bceSStefano Zampini - levels - the maximum number of levels (max 9) 2904fad6a16SStefano Zampini 2910f202f7eSStefano Zampini Options Database Keys: 2920f202f7eSStefano Zampini . -pc_bddc_levels 2934fad6a16SStefano Zampini 2944fad6a16SStefano Zampini Level: intermediate 2954fad6a16SStefano Zampini 2964fad6a16SStefano Zampini Notes: 2970f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 2984fad6a16SStefano Zampini 2990f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 3004fad6a16SStefano Zampini @*/ 3012b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 3024fad6a16SStefano Zampini { 3034fad6a16SStefano Zampini PetscErrorCode ierr; 3044fad6a16SStefano Zampini 3054fad6a16SStefano Zampini PetscFunctionBegin; 3064fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 3072b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 308312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3092b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3104fad6a16SStefano Zampini PetscFunctionReturn(0); 3114fad6a16SStefano Zampini } 3124fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3131e6b0712SBarry Smith 3144fad6a16SStefano Zampini #undef __FUNCT__ 3150bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 3160bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 3170bdf917eSStefano Zampini { 3180bdf917eSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3190bdf917eSStefano Zampini PetscErrorCode ierr; 3200bdf917eSStefano Zampini 3210bdf917eSStefano Zampini PetscFunctionBegin; 3220bdf917eSStefano Zampini ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 3230bdf917eSStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 3240bdf917eSStefano Zampini pcbddc->NullSpace = NullSpace; 3250bdf917eSStefano Zampini PetscFunctionReturn(0); 3260bdf917eSStefano Zampini } 3271e6b0712SBarry Smith 3280bdf917eSStefano Zampini #undef __FUNCT__ 3290bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace" 3300bdf917eSStefano Zampini /*@ 33128509bceSStefano Zampini PCBDDCSetNullSpace - Set nullspace for BDDC operator 3320bdf917eSStefano Zampini 3330bdf917eSStefano Zampini Logically collective on PC and MatNullSpace 3340bdf917eSStefano Zampini 3350bdf917eSStefano Zampini Input Parameters: 3360bdf917eSStefano Zampini + pc - the preconditioning context 33728509bceSStefano Zampini - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 3380bdf917eSStefano Zampini 3390bdf917eSStefano Zampini Level: intermediate 3400bdf917eSStefano Zampini 3410bdf917eSStefano Zampini Notes: 3420bdf917eSStefano Zampini 3430bdf917eSStefano Zampini .seealso: PCBDDC 3440bdf917eSStefano Zampini @*/ 3450bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 3460bdf917eSStefano Zampini { 3470bdf917eSStefano Zampini PetscErrorCode ierr; 3480bdf917eSStefano Zampini 3490bdf917eSStefano Zampini PetscFunctionBegin; 3500bdf917eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 351674ae819SStefano Zampini PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 3522b510759SStefano Zampini PetscCheckSameComm(pc,1,NullSpace,2); 3530bdf917eSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 3540bdf917eSStefano Zampini PetscFunctionReturn(0); 3550bdf917eSStefano Zampini } 3560bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */ 3571e6b0712SBarry Smith 3580bdf917eSStefano Zampini #undef __FUNCT__ 3593b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 3603b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 3613b03a366Sstefano_zampini { 3623b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3633b03a366Sstefano_zampini PetscErrorCode ierr; 3643b03a366Sstefano_zampini 3653b03a366Sstefano_zampini PetscFunctionBegin; 366785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 367785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 3683b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 36936e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 37036e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 371fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 3723b03a366Sstefano_zampini PetscFunctionReturn(0); 3733b03a366Sstefano_zampini } 3741e6b0712SBarry Smith 3753b03a366Sstefano_zampini #undef __FUNCT__ 3763b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 3773b03a366Sstefano_zampini /*@ 37828509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 3793b03a366Sstefano_zampini 380785d1243SStefano Zampini Collective 3813b03a366Sstefano_zampini 3823b03a366Sstefano_zampini Input Parameters: 3833b03a366Sstefano_zampini + pc - the preconditioning context 384785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 3853b03a366Sstefano_zampini 3863b03a366Sstefano_zampini Level: intermediate 3873b03a366Sstefano_zampini 3880f202f7eSStefano Zampini Notes: 3890f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 3903b03a366Sstefano_zampini 3910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 3923b03a366Sstefano_zampini @*/ 3933b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 3943b03a366Sstefano_zampini { 3953b03a366Sstefano_zampini PetscErrorCode ierr; 3963b03a366Sstefano_zampini 3973b03a366Sstefano_zampini PetscFunctionBegin; 3983b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 399674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 400785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4013b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4023b03a366Sstefano_zampini PetscFunctionReturn(0); 4033b03a366Sstefano_zampini } 4043b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4051e6b0712SBarry Smith 4063b03a366Sstefano_zampini #undef __FUNCT__ 40782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 40882ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4090c7d97c5SJed Brown { 4100c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4110c7d97c5SJed Brown PetscErrorCode ierr; 4120c7d97c5SJed Brown 4130c7d97c5SJed Brown PetscFunctionBegin; 414785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 415785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4160c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 4170c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 418785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 4197d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4200c7d97c5SJed Brown PetscFunctionReturn(0); 4210c7d97c5SJed Brown } 4220c7d97c5SJed Brown 4230c7d97c5SJed Brown #undef __FUNCT__ 42482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4259c0446d6SStefano Zampini /*@ 42682ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4279c0446d6SStefano Zampini 428785d1243SStefano Zampini Collective 42953cdbc3dSStefano Zampini 43053cdbc3dSStefano Zampini Input Parameters: 43153cdbc3dSStefano Zampini + pc - the preconditioning context 43282ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 43353cdbc3dSStefano Zampini 43453cdbc3dSStefano Zampini Level: intermediate 43553cdbc3dSStefano Zampini 4369c0446d6SStefano Zampini Notes: 43753cdbc3dSStefano Zampini 4380f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 43953cdbc3dSStefano Zampini @*/ 44082ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 4410c7d97c5SJed Brown { 4420c7d97c5SJed Brown PetscErrorCode ierr; 4430c7d97c5SJed Brown 4440c7d97c5SJed Brown PetscFunctionBegin; 4450c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4460c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 44782ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 44882ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4490c7d97c5SJed Brown PetscFunctionReturn(0); 4500c7d97c5SJed Brown } 4510c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 4520c7d97c5SJed Brown 4530c7d97c5SJed Brown #undef __FUNCT__ 4540c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 45553cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 4560c7d97c5SJed Brown { 4570c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 45853cdbc3dSStefano Zampini PetscErrorCode ierr; 4590c7d97c5SJed Brown 4600c7d97c5SJed Brown PetscFunctionBegin; 461785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 462785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 46353cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 46436e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 46536e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 466fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4670c7d97c5SJed Brown PetscFunctionReturn(0); 4680c7d97c5SJed Brown } 4691e6b0712SBarry Smith 4700c7d97c5SJed Brown #undef __FUNCT__ 4710c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 47257527edcSJed Brown /*@ 47328509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 47457527edcSJed Brown 475785d1243SStefano Zampini Collective 47657527edcSJed Brown 47757527edcSJed Brown Input Parameters: 47857527edcSJed Brown + pc - the preconditioning context 479785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 48057527edcSJed Brown 48157527edcSJed Brown Level: intermediate 48257527edcSJed Brown 4830f202f7eSStefano Zampini Notes: 4840f202f7eSStefano Zampini Any process can list any global node 48557527edcSJed Brown 4860f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 48757527edcSJed Brown @*/ 48853cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 4890c7d97c5SJed Brown { 4900c7d97c5SJed Brown PetscErrorCode ierr; 4910c7d97c5SJed Brown 4920c7d97c5SJed Brown PetscFunctionBegin; 4930c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 494674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 495785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 49653cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 49753cdbc3dSStefano Zampini PetscFunctionReturn(0); 49853cdbc3dSStefano Zampini } 49953cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5001e6b0712SBarry Smith 50153cdbc3dSStefano Zampini #undef __FUNCT__ 50282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 50382ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5040c7d97c5SJed Brown { 5050c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5060c7d97c5SJed Brown PetscErrorCode ierr; 5070c7d97c5SJed Brown 5080c7d97c5SJed Brown PetscFunctionBegin; 509785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 510785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5110c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 5120c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 513785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 5147d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5150c7d97c5SJed Brown PetscFunctionReturn(0); 5160c7d97c5SJed Brown } 5170c7d97c5SJed Brown 5180c7d97c5SJed Brown #undef __FUNCT__ 51982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5200c7d97c5SJed Brown /*@ 52182ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 5220c7d97c5SJed Brown 523785d1243SStefano Zampini Collective 5240c7d97c5SJed Brown 5250c7d97c5SJed Brown Input Parameters: 5260c7d97c5SJed Brown + pc - the preconditioning context 52782ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 5280c7d97c5SJed Brown 5290c7d97c5SJed Brown Level: intermediate 5300c7d97c5SJed Brown 5310c7d97c5SJed Brown Notes: 5320c7d97c5SJed Brown 5330f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 5340c7d97c5SJed Brown @*/ 53582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 5360c7d97c5SJed Brown { 5370c7d97c5SJed Brown PetscErrorCode ierr; 5380c7d97c5SJed Brown 5390c7d97c5SJed Brown PetscFunctionBegin; 5400c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5410c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 54282ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 54382ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 54453cdbc3dSStefano Zampini PetscFunctionReturn(0); 54553cdbc3dSStefano Zampini } 54653cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 54753cdbc3dSStefano Zampini 54853cdbc3dSStefano Zampini #undef __FUNCT__ 549da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 550da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 551da1bb401SStefano Zampini { 552da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 553da1bb401SStefano Zampini 554da1bb401SStefano Zampini PetscFunctionBegin; 555da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 556da1bb401SStefano Zampini PetscFunctionReturn(0); 557da1bb401SStefano Zampini } 5581e6b0712SBarry Smith 559da1bb401SStefano Zampini #undef __FUNCT__ 560da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 561da1bb401SStefano Zampini /*@ 562785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 563da1bb401SStefano Zampini 564785d1243SStefano Zampini Collective 565785d1243SStefano Zampini 566785d1243SStefano Zampini Input Parameters: 567785d1243SStefano Zampini . pc - the preconditioning context 568785d1243SStefano Zampini 569785d1243SStefano Zampini Output Parameters: 570785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 571785d1243SStefano Zampini 572785d1243SStefano Zampini Level: intermediate 573785d1243SStefano Zampini 5740f202f7eSStefano Zampini Notes: 5750f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 576785d1243SStefano Zampini 577785d1243SStefano Zampini .seealso: PCBDDC 578785d1243SStefano Zampini @*/ 579785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 580785d1243SStefano Zampini { 581785d1243SStefano Zampini PetscErrorCode ierr; 582785d1243SStefano Zampini 583785d1243SStefano Zampini PetscFunctionBegin; 584785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 585785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 586785d1243SStefano Zampini PetscFunctionReturn(0); 587785d1243SStefano Zampini } 588785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 589785d1243SStefano Zampini 590785d1243SStefano Zampini #undef __FUNCT__ 591785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 592785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 593785d1243SStefano Zampini { 594785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 595785d1243SStefano Zampini 596785d1243SStefano Zampini PetscFunctionBegin; 597785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 598785d1243SStefano Zampini PetscFunctionReturn(0); 599785d1243SStefano Zampini } 600785d1243SStefano Zampini 601785d1243SStefano Zampini #undef __FUNCT__ 60282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 603da1bb401SStefano Zampini /*@ 60482ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 605da1bb401SStefano Zampini 606785d1243SStefano Zampini Collective 607da1bb401SStefano Zampini 608da1bb401SStefano Zampini Input Parameters: 60928509bceSStefano Zampini . pc - the preconditioning context 610da1bb401SStefano Zampini 611da1bb401SStefano Zampini Output Parameters: 61228509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 613da1bb401SStefano Zampini 614da1bb401SStefano Zampini Level: intermediate 615da1bb401SStefano Zampini 616da1bb401SStefano Zampini Notes: 6170f202f7eSStefano 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). 6180f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 619da1bb401SStefano Zampini 620da1bb401SStefano Zampini .seealso: PCBDDC 621da1bb401SStefano Zampini @*/ 62282ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 623da1bb401SStefano Zampini { 624da1bb401SStefano Zampini PetscErrorCode ierr; 625da1bb401SStefano Zampini 626da1bb401SStefano Zampini PetscFunctionBegin; 627da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 62882ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 629da1bb401SStefano Zampini PetscFunctionReturn(0); 630da1bb401SStefano Zampini } 631da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 6321e6b0712SBarry Smith 633da1bb401SStefano Zampini #undef __FUNCT__ 63453cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 63553cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 63653cdbc3dSStefano Zampini { 63753cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 63853cdbc3dSStefano Zampini 63953cdbc3dSStefano Zampini PetscFunctionBegin; 64053cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 64153cdbc3dSStefano Zampini PetscFunctionReturn(0); 64253cdbc3dSStefano Zampini } 6431e6b0712SBarry Smith 64453cdbc3dSStefano Zampini #undef __FUNCT__ 64553cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 64653cdbc3dSStefano Zampini /*@ 647785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 64853cdbc3dSStefano Zampini 649785d1243SStefano Zampini Collective 650785d1243SStefano Zampini 651785d1243SStefano Zampini Input Parameters: 652785d1243SStefano Zampini . pc - the preconditioning context 653785d1243SStefano Zampini 654785d1243SStefano Zampini Output Parameters: 655785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 656785d1243SStefano Zampini 657785d1243SStefano Zampini Level: intermediate 658785d1243SStefano Zampini 6590f202f7eSStefano Zampini Notes: 6600f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 661785d1243SStefano Zampini 662785d1243SStefano Zampini .seealso: PCBDDC 663785d1243SStefano Zampini @*/ 664785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 665785d1243SStefano Zampini { 666785d1243SStefano Zampini PetscErrorCode ierr; 667785d1243SStefano Zampini 668785d1243SStefano Zampini PetscFunctionBegin; 669785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 670785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 671785d1243SStefano Zampini PetscFunctionReturn(0); 672785d1243SStefano Zampini } 673785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 674785d1243SStefano Zampini 675785d1243SStefano Zampini #undef __FUNCT__ 676785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 677785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 678785d1243SStefano Zampini { 679785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 680785d1243SStefano Zampini 681785d1243SStefano Zampini PetscFunctionBegin; 682785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 683785d1243SStefano Zampini PetscFunctionReturn(0); 684785d1243SStefano Zampini } 685785d1243SStefano Zampini 686785d1243SStefano Zampini #undef __FUNCT__ 68782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 68853cdbc3dSStefano Zampini /*@ 68982ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 69053cdbc3dSStefano Zampini 691785d1243SStefano Zampini Collective 69253cdbc3dSStefano Zampini 69353cdbc3dSStefano Zampini Input Parameters: 69428509bceSStefano Zampini . pc - the preconditioning context 69553cdbc3dSStefano Zampini 69653cdbc3dSStefano Zampini Output Parameters: 69728509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 69853cdbc3dSStefano Zampini 69953cdbc3dSStefano Zampini Level: intermediate 70053cdbc3dSStefano Zampini 70153cdbc3dSStefano Zampini Notes: 7020f202f7eSStefano 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). 7030f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 70453cdbc3dSStefano Zampini 70553cdbc3dSStefano Zampini .seealso: PCBDDC 70653cdbc3dSStefano Zampini @*/ 70782ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 70853cdbc3dSStefano Zampini { 70953cdbc3dSStefano Zampini PetscErrorCode ierr; 71053cdbc3dSStefano Zampini 71153cdbc3dSStefano Zampini PetscFunctionBegin; 71253cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 71382ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7140c7d97c5SJed Brown PetscFunctionReturn(0); 7150c7d97c5SJed Brown } 71636e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7171e6b0712SBarry Smith 71836e030ebSStefano Zampini #undef __FUNCT__ 719da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7201a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 72136e030ebSStefano Zampini { 72236e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 723da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 724da1bb401SStefano Zampini PetscErrorCode ierr; 72536e030ebSStefano Zampini 72636e030ebSStefano Zampini PetscFunctionBegin; 727674ae819SStefano Zampini /* free old CSR */ 728674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 729fb223d50SStefano Zampini /* TODO: PCBDDCGraphSetAdjacency */ 730674ae819SStefano Zampini /* get CSR into graph structure */ 731da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 732854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 733785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 734674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 735674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 736da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 7371a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 7381a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 739674ae819SStefano Zampini } 740575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 74136e030ebSStefano Zampini PetscFunctionReturn(0); 74236e030ebSStefano Zampini } 7431e6b0712SBarry Smith 74436e030ebSStefano Zampini #undef __FUNCT__ 745da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 74636e030ebSStefano Zampini /*@ 7470f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 74836e030ebSStefano Zampini 74936e030ebSStefano Zampini Not collective 75036e030ebSStefano Zampini 75136e030ebSStefano Zampini Input Parameters: 75236e030ebSStefano Zampini + pc - the preconditioning context 7530f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 75428509bceSStefano Zampini . xadj, adjncy - the CSR graph 75528509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 75636e030ebSStefano Zampini 75736e030ebSStefano Zampini Level: intermediate 75836e030ebSStefano Zampini 75936e030ebSStefano Zampini Notes: 76036e030ebSStefano Zampini 76128509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 76236e030ebSStefano Zampini @*/ 7631a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 76436e030ebSStefano Zampini { 765575ad6abSStefano Zampini void (*f)(void) = 0; 76636e030ebSStefano Zampini PetscErrorCode ierr; 76736e030ebSStefano Zampini 76836e030ebSStefano Zampini PetscFunctionBegin; 76936e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 770674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 771d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 772674ae819SStefano Zampini if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) { 773674ae819SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__); 774da1bb401SStefano Zampini } 77536e030ebSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 776575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 777575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 778575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 779575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 780575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 78136e030ebSStefano Zampini } 78236e030ebSStefano Zampini PetscFunctionReturn(0); 78336e030ebSStefano Zampini } 7849c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 7851e6b0712SBarry Smith 7869c0446d6SStefano Zampini #undef __FUNCT__ 78763602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 78863602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 78963602bcaSStefano Zampini { 79063602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79163602bcaSStefano Zampini PetscInt i; 79263602bcaSStefano Zampini PetscErrorCode ierr; 79363602bcaSStefano Zampini 79463602bcaSStefano Zampini PetscFunctionBegin; 79563602bcaSStefano Zampini /* Destroy ISes if they were already set */ 79663602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 79763602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 79863602bcaSStefano Zampini } 79963602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 80063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 80163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 80263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 80363602bcaSStefano Zampini } 80463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 80563602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 80663602bcaSStefano Zampini /* allocate space then set */ 807d02579f5SStefano Zampini if (n_is) { 808d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 809d02579f5SStefano Zampini } 81063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 81163602bcaSStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 81263602bcaSStefano Zampini pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 81363602bcaSStefano Zampini } 81463602bcaSStefano Zampini pcbddc->n_ISForDofsLocal=n_is; 81563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8161a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 81763602bcaSStefano Zampini PetscFunctionReturn(0); 81863602bcaSStefano Zampini } 81963602bcaSStefano Zampini 82063602bcaSStefano Zampini #undef __FUNCT__ 82163602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 82263602bcaSStefano Zampini /*@ 82363602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 82463602bcaSStefano Zampini 82563602bcaSStefano Zampini Collective 82663602bcaSStefano Zampini 82763602bcaSStefano Zampini Input Parameters: 82863602bcaSStefano Zampini + pc - the preconditioning context 8290f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8300f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 83163602bcaSStefano Zampini 83263602bcaSStefano Zampini Level: intermediate 83363602bcaSStefano Zampini 8340f202f7eSStefano Zampini Notes: 8350f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 83663602bcaSStefano Zampini 83763602bcaSStefano Zampini .seealso: PCBDDC 83863602bcaSStefano Zampini @*/ 83963602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 84063602bcaSStefano Zampini { 84163602bcaSStefano Zampini PetscInt i; 84263602bcaSStefano Zampini PetscErrorCode ierr; 84363602bcaSStefano Zampini 84463602bcaSStefano Zampini PetscFunctionBegin; 84563602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 84663602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 84763602bcaSStefano Zampini for (i=0;i<n_is;i++) { 84863602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 84963602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 85063602bcaSStefano Zampini } 851e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 85263602bcaSStefano Zampini PetscFunctionReturn(0); 85363602bcaSStefano Zampini } 85463602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 85563602bcaSStefano Zampini 85663602bcaSStefano Zampini #undef __FUNCT__ 8579c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 8589c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 8599c0446d6SStefano Zampini { 8609c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 8619c0446d6SStefano Zampini PetscInt i; 8629c0446d6SStefano Zampini PetscErrorCode ierr; 8639c0446d6SStefano Zampini 8649c0446d6SStefano Zampini PetscFunctionBegin; 865da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 8669c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 8679c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 8689c0446d6SStefano Zampini } 869d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 87063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 87163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 87263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 87363602bcaSStefano Zampini } 87463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 87563602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 876da1bb401SStefano Zampini /* allocate space then set */ 877d02579f5SStefano Zampini if (n_is) { 878785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 879d02579f5SStefano Zampini } 8809c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 881da1bb401SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 882da1bb401SStefano Zampini pcbddc->ISForDofs[i]=ISForDofs[i]; 8839c0446d6SStefano Zampini } 8849c0446d6SStefano Zampini pcbddc->n_ISForDofs=n_is; 88563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8861a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 8879c0446d6SStefano Zampini PetscFunctionReturn(0); 8889c0446d6SStefano Zampini } 8891e6b0712SBarry Smith 8909c0446d6SStefano Zampini #undef __FUNCT__ 8919c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 8929c0446d6SStefano Zampini /*@ 89363602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 8949c0446d6SStefano Zampini 89563602bcaSStefano Zampini Collective 8969c0446d6SStefano Zampini 8979c0446d6SStefano Zampini Input Parameters: 8989c0446d6SStefano Zampini + pc - the preconditioning context 8990f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9000f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 9019c0446d6SStefano Zampini 9029c0446d6SStefano Zampini Level: intermediate 9039c0446d6SStefano Zampini 9040f202f7eSStefano Zampini Notes: 9050f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 9069c0446d6SStefano Zampini 9079c0446d6SStefano Zampini .seealso: PCBDDC 9089c0446d6SStefano Zampini @*/ 9099c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 9109c0446d6SStefano Zampini { 9112b510759SStefano Zampini PetscInt i; 9129c0446d6SStefano Zampini PetscErrorCode ierr; 9139c0446d6SStefano Zampini 9149c0446d6SStefano Zampini PetscFunctionBegin; 9159c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 91663602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 9172b510759SStefano Zampini for (i=0;i<n_is;i++) { 91863602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 91963602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 9202b510759SStefano Zampini } 9219c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 9229c0446d6SStefano Zampini PetscFunctionReturn(0); 9239c0446d6SStefano Zampini } 924906d46d4SStefano Zampini 925da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 926534831adSStefano Zampini #undef __FUNCT__ 927534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 928534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 929534831adSStefano Zampini /* 930534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 931534831adSStefano Zampini guess if a transformation of basis approach has been selected. 9329c0446d6SStefano Zampini 933534831adSStefano Zampini Input Parameter: 934534831adSStefano Zampini + pc - the preconditioner contex 935534831adSStefano Zampini 936534831adSStefano Zampini Application Interface Routine: PCPreSolve() 937534831adSStefano Zampini 938534831adSStefano Zampini Notes: 939534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 940534831adSStefano Zampini the user, but instead is called by KSPSolve(). 941534831adSStefano Zampini */ 942534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 943534831adSStefano Zampini { 944534831adSStefano Zampini PetscErrorCode ierr; 945534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 946534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 9473972b0daSStefano Zampini Vec used_vec; 9488d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 949534831adSStefano Zampini 950534831adSStefano Zampini PetscFunctionBegin; 95185c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 95285c4d303SStefano Zampini if (ksp) { 95385c4d303SStefano Zampini PetscBool iscg; 95485c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 95585c4d303SStefano Zampini if (!iscg) { 95685c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 95785c4d303SStefano Zampini } 95885c4d303SStefano Zampini } 95985c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 96062a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 96162a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 96262a6ff1dSStefano Zampini } 96362a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 96462a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 96562a6ff1dSStefano Zampini } 9668d00608fSStefano Zampini 9673972b0daSStefano Zampini if (x) { 9683972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 9693972b0daSStefano Zampini used_vec = x; 9708d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 9713972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 9723972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 9733972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9743972b0daSStefano Zampini } 9758efcfb23SStefano Zampini 9768efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 9773972b0daSStefano Zampini if (ksp) { 978a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 9798efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 9808efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 9813972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9823972b0daSStefano Zampini } 9833972b0daSStefano Zampini } 9843308cffdSStefano Zampini 9858d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 9863972b0daSStefano Zampini 9873972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 988a07ea27aSStefano Zampini if (rhs) { 9893975b054SStefano Zampini IS dirIS = NULL; 9903975b054SStefano Zampini 991a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 9923975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 9933975b054SStefano Zampini if (dirIS) { 994906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 995785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 9962b095fd8SStefano Zampini PetscScalar *array_x; 9972b095fd8SStefano Zampini const PetscScalar *array_diagonal; 998785d1243SStefano Zampini 9993972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 10003972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1001e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1002e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1003e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1004e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 100582ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 10063972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 10072b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10083972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10092fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 10103972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10112b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10123972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1013e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1014e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10158d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10161b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 10178efcfb23SStefano Zampini } 1018a07ea27aSStefano Zampini } 1019b76ba322SStefano Zampini 10208efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 10218d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 10228d00608fSStefano Zampini /* store the original rhs */ 10238d00608fSStefano Zampini if (copy_rhs) { 10248d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10258d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10268d00608fSStefano Zampini } 10278d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10283972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10293972b0daSStefano Zampini ierr = MatMultAdd(pc->pmat,used_vec,rhs,rhs);CHKERRQ(ierr); 10303972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10318efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 10327acc28cbSStefano Zampini if (ksp) { 10337acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 10347acc28cbSStefano Zampini } 10353308cffdSStefano Zampini } 10368efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1037b76ba322SStefano Zampini 1038b76ba322SStefano Zampini /* store partially computed solution and set initial guess */ 1039b097fa66SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 10408efcfb23SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 1041b76ba322SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1042b76ba322SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1043b76ba322SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 10448efcfb23SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10458efcfb23SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1046a0cb1b98SStefano Zampini if (ksp) { 1047b76ba322SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 1048b76ba322SStefano Zampini } 1049b76ba322SStefano Zampini } 1050b76ba322SStefano Zampini 1051b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1052906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1053906d46d4SStefano Zampini 1054906d46d4SStefano Zampini /* get change ctx */ 1055906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1056906d46d4SStefano Zampini 1057906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 1058906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1059906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1060906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1061906d46d4SStefano Zampini 1062906d46d4SStefano Zampini /* change iteration matrix */ 1063906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1064906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1065906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 1066906d46d4SStefano Zampini 10678d00608fSStefano Zampini /* store the original rhs */ 10688d00608fSStefano Zampini if (copy_rhs) { 10698d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10708d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10718d00608fSStefano Zampini } 10728d00608fSStefano Zampini 1073906d46d4SStefano Zampini /* change rhs */ 1074906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1075906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 10768d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 107792e3dcfbSStefano Zampini } 107892e3dcfbSStefano Zampini 107992e3dcfbSStefano Zampini /* remove nullspace if present */ 10808efcfb23SStefano Zampini if (ksp && x && pcbddc->NullSpace) { 10818efcfb23SStefano Zampini ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr); 10828d00608fSStefano Zampini /* store the original rhs */ 10838d00608fSStefano Zampini if (copy_rhs) { 10848d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10858d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10868d00608fSStefano Zampini } 10878d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 1088d0195637SJed Brown ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr); 1089b76ba322SStefano Zampini } 1090534831adSStefano Zampini PetscFunctionReturn(0); 1091534831adSStefano Zampini } 1092906d46d4SStefano Zampini 1093534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1094534831adSStefano Zampini #undef __FUNCT__ 1095534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1096534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1097534831adSStefano Zampini /* 1098534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1099534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1100534831adSStefano Zampini 1101534831adSStefano Zampini Input Parameter: 1102534831adSStefano Zampini + pc - the preconditioner contex 1103534831adSStefano Zampini 1104534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1105534831adSStefano Zampini 1106534831adSStefano Zampini Notes: 1107534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1108534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1109534831adSStefano Zampini */ 1110534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1111534831adSStefano Zampini { 1112534831adSStefano Zampini PetscErrorCode ierr; 1113534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1114534831adSStefano Zampini 1115534831adSStefano Zampini PetscFunctionBegin; 1116b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1117906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1118906d46d4SStefano Zampini 1119906d46d4SStefano Zampini /* get change ctx */ 1120906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1121906d46d4SStefano Zampini 1122906d46d4SStefano Zampini /* restore iteration matrix */ 1123906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1124906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1125906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 1126906d46d4SStefano Zampini 1127906d46d4SStefano Zampini /* get solution in original basis */ 1128906d46d4SStefano Zampini if (x) { 1129906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1130906d46d4SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1131906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 11323425bc38SStefano Zampini } 1133534831adSStefano Zampini } 1134906d46d4SStefano Zampini 11353972b0daSStefano Zampini /* add solution removed in presolve */ 11366bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 11373425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 11383425bc38SStefano Zampini } 1139906d46d4SStefano Zampini 1140fb223d50SStefano Zampini /* restore rhs to its original state */ 11418d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1142fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1143fb223d50SStefano Zampini } 11448d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11458efcfb23SStefano Zampini 11468efcfb23SStefano Zampini /* restore ksp guess state */ 11478efcfb23SStefano Zampini if (ksp) { 11488efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 11498efcfb23SStefano Zampini } 1150534831adSStefano Zampini PetscFunctionReturn(0); 1151534831adSStefano Zampini } 1152534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 115353cdbc3dSStefano Zampini #undef __FUNCT__ 115453cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 11550c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 11560c7d97c5SJed Brown /* 11570c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 11580c7d97c5SJed Brown by setting data structures and options. 11590c7d97c5SJed Brown 11600c7d97c5SJed Brown Input Parameter: 116153cdbc3dSStefano Zampini + pc - the preconditioner context 11620c7d97c5SJed Brown 11630c7d97c5SJed Brown Application Interface Routine: PCSetUp() 11640c7d97c5SJed Brown 11650c7d97c5SJed Brown Notes: 11660c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 11670c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 11680c7d97c5SJed Brown */ 116953cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 11700c7d97c5SJed Brown { 11710c7d97c5SJed Brown PetscErrorCode ierr; 11720c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 11735e8657edSStefano Zampini Mat_IS* matis; 117408122e43SStefano Zampini MatNullSpace nearnullspace; 1175*91e8d312SStefano Zampini PetscInt nrows,ncols; 117608122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 11778de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 11785e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 11790c7d97c5SJed Brown 11800c7d97c5SJed Brown PetscFunctionBegin; 11815e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 11825e8657edSStefano Zampini if (!ismatis) { 11835e8657edSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 11845e8657edSStefano Zampini } 1185*91e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 1186*91e8d312SStefano Zampini if (nrows != ncols) { 1187*91e8d312SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 1188*91e8d312SStefano Zampini } 11895e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1190f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 11913b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1192fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1193f4ddd8eeSStefano Zampini /* split work */ 1194674ae819SStefano Zampini if (pc->setupcalled) { 1195d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1196674ae819SStefano Zampini computetopography = PETSC_FALSE; 1197674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1198674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1199674ae819SStefano Zampini computetopography = PETSC_TRUE; 1200674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1201674ae819SStefano Zampini } 1202674ae819SStefano Zampini } else { 1203674ae819SStefano Zampini computetopography = PETSC_TRUE; 1204674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1205674ae819SStefano Zampini } 1206fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1207fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1208fb180af4SStefano Zampini } 12098de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 12105a95e1ceSStefano Zampini if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) { 12115a95e1ceSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 12125a95e1ceSStefano Zampini } 12135a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1214862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1215862806e4SStefano Zampini 12165a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 12176816873aSStefano Zampini if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) { 12186816873aSStefano 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"); 12196816873aSStefano Zampini } 1220f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 122170cf5478SStefano Zampini if (pcbddc->dbg_flag) { 122270cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 122358a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 1224f4ddd8eeSStefano Zampini ierr = PetscViewerASCIISynchronizedAllow(pcbddc->dbg_viewer,PETSC_TRUE);CHKERRQ(ierr); 1225f4ddd8eeSStefano Zampini } 122658a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1227f4ddd8eeSStefano Zampini } 1228f4ddd8eeSStefano Zampini 12295e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 12305e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 12315e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 12325e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 12335e8657edSStefano Zampini } else { 1234b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 12355e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 12365e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1237674ae819SStefano Zampini } 1238f4ddd8eeSStefano Zampini 1239e496cd5dSStefano Zampini /* workaround for reals */ 1240e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX) 12413301b35fSStefano Zampini if (matis->A->symmetric_set) { 12423301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 12433301b35fSStefano Zampini } 1244e496cd5dSStefano Zampini #endif 1245e496cd5dSStefano Zampini 12465e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 12475e8657edSStefano Zampini { 12485e8657edSStefano Zampini Mat temp_mat; 12495e8657edSStefano Zampini 12505e8657edSStefano Zampini temp_mat = matis->A; 12515e8657edSStefano Zampini matis->A = pcbddc->local_mat; 12525e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 12535e8657edSStefano Zampini pcbddc->local_mat = matis->A; 12545e8657edSStefano Zampini matis->A = temp_mat; 12555e8657edSStefano Zampini } 1256684f6988SStefano Zampini 1257b96c3477SStefano Zampini /* Analyze interface and setup sub_schurs data */ 1258674ae819SStefano Zampini if (computetopography) { 1259674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 12608de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1261674ae819SStefano Zampini } 1262fb8d54d4SStefano Zampini 1263b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1264684f6988SStefano Zampini if (computesolvers) { 1265d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1266d5574798SStefano Zampini 1267d5574798SStefano Zampini if (computesubschurs && computetopography) { 126808122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1269b1b3d7a2SStefano Zampini } 1270d5574798SStefano Zampini if (sub_schurs->use_mumps) { 12712070dbb6SStefano Zampini if (computesubschurs) { 127208122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 12732070dbb6SStefano Zampini } 1274d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1275d5574798SStefano Zampini } else { 1276d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 12772070dbb6SStefano Zampini if (computesubschurs) { 1278d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1279d5574798SStefano Zampini } 12802070dbb6SStefano Zampini } 128108122e43SStefano Zampini if (pcbddc->adaptive_selection) { 128208122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 12838de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1284b7eb3628SStefano Zampini } 1285b1b3d7a2SStefano Zampini } 1286684f6988SStefano Zampini 1287f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1288fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1289f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1290f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1291f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1292f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1293f4ddd8eeSStefano Zampini } else { 1294f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1295f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1296f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1297165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1298f4ddd8eeSStefano Zampini PetscInt i; 1299165b64e2SStefano Zampini const Vec *nearnullvecs; 1300165b64e2SStefano Zampini PetscObjectState state; 1301165b64e2SStefano Zampini PetscInt nnsp_size; 1302165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1303f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1304f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1305165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1306f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1307f4ddd8eeSStefano Zampini break; 1308f4ddd8eeSStefano Zampini } 1309f4ddd8eeSStefano Zampini } 1310f4ddd8eeSStefano Zampini } 1311f4ddd8eeSStefano Zampini } 1312f4ddd8eeSStefano Zampini } else { 1313f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1314f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1315f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1316f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1317f4ddd8eeSStefano Zampini } 1318f4ddd8eeSStefano Zampini } 1319f4ddd8eeSStefano Zampini 1320f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1321727cdba6SStefano Zampini /* reset primal space flags */ 1322f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1323727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 13248de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1325727cdba6SStefano Zampini /* It also sets the primal space flags */ 1326674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1327e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1328f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 13293975b054SStefano Zampini } 13305e8657edSStefano Zampini 13313975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 13325e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 13335e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 13345e8657edSStefano Zampini 13355e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 13365e8657edSStefano Zampini /* get submatrices */ 13375e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 13385e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 13395e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 13405e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 13415e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 13425e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 13433975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 13443975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 13455e8657edSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix) { 1346b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 13475e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 13485e8657edSStefano Zampini pcbddc->local_mat = matis->A; 13495e8657edSStefano Zampini } 1350b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 135199cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1352b96c3477SStefano Zampini /* SetUp Scaling operator */ 1353674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 13540c7d97c5SJed Brown } 135570cf5478SStefano Zampini 135658a03d70SStefano Zampini if (pcbddc->dbg_flag) { 135758a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 13582b510759SStefano Zampini } 13590c7d97c5SJed Brown PetscFunctionReturn(0); 13600c7d97c5SJed Brown } 13610c7d97c5SJed Brown 13620c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13630c7d97c5SJed Brown /* 136450efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 13650c7d97c5SJed Brown 13660c7d97c5SJed Brown Input Parameters: 13670f202f7eSStefano Zampini + pc - the preconditioner context 13680f202f7eSStefano Zampini - r - input vector (global) 13690c7d97c5SJed Brown 13700c7d97c5SJed Brown Output Parameter: 13710c7d97c5SJed Brown . z - output vector (global) 13720c7d97c5SJed Brown 13730c7d97c5SJed Brown Application Interface Routine: PCApply() 13740c7d97c5SJed Brown */ 13750c7d97c5SJed Brown #undef __FUNCT__ 13760c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 137753cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 13780c7d97c5SJed Brown { 13790c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 13800c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1381b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 13820c7d97c5SJed Brown PetscErrorCode ierr; 13833b03a366Sstefano_zampini const PetscScalar one = 1.0; 13843b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 13852617d88aSStefano Zampini const PetscScalar zero = 0.0; 13860c7d97c5SJed Brown 13870c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 13880c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1389b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 13900c7d97c5SJed Brown 13910c7d97c5SJed Brown PetscFunctionBegin; 139285c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1393b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 13940c7d97c5SJed Brown /* First Dirichlet solve */ 13950c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13960c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13970c7d97c5SJed Brown /* 13980c7d97c5SJed Brown Assembling right hand side for BDDC operator 1399b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1400674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 14010c7d97c5SJed Brown */ 1402b097fa66SStefano Zampini if (n_D) { 1403b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 14040c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 14058eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1406b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1407b097fa66SStefano Zampini } else { 1408b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1409b097fa66SStefano Zampini } 14100c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14110c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1412674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1413b76ba322SStefano Zampini } else { 1414b097fa66SStefano Zampini if (pcbddc->switch_static) { 14150bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1416b097fa66SStefano Zampini } 1417674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1418b76ba322SStefano Zampini } 1419b76ba322SStefano Zampini 14202617d88aSStefano Zampini /* Apply interface preconditioner 14212617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1422dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 14232617d88aSStefano Zampini 1424674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1425674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 14260c7d97c5SJed Brown 14273b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 14280c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14290c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1430b097fa66SStefano Zampini if (n_B) { 14310c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 14328eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1433b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1434b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1435b097fa66SStefano Zampini } 1436df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1437b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1438b097fa66SStefano Zampini if (pcbddc->switch_static) { 1439b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1440b097fa66SStefano Zampini } else { 1441b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1442b097fa66SStefano Zampini } 14430c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14440c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1445b097fa66SStefano Zampini } else { 1446b097fa66SStefano Zampini if (pcbddc->switch_static) { 1447b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1448b097fa66SStefano Zampini } else { 1449b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1450b097fa66SStefano Zampini } 1451b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1452b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1453b097fa66SStefano Zampini } 14540c7d97c5SJed Brown PetscFunctionReturn(0); 14550c7d97c5SJed Brown } 145650efa1b5SStefano Zampini 145750efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 145850efa1b5SStefano Zampini /* 145950efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 146050efa1b5SStefano Zampini 146150efa1b5SStefano Zampini Input Parameters: 14620f202f7eSStefano Zampini + pc - the preconditioner context 14630f202f7eSStefano Zampini - r - input vector (global) 146450efa1b5SStefano Zampini 146550efa1b5SStefano Zampini Output Parameter: 146650efa1b5SStefano Zampini . z - output vector (global) 146750efa1b5SStefano Zampini 146850efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 146950efa1b5SStefano Zampini */ 147050efa1b5SStefano Zampini #undef __FUNCT__ 147150efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 147250efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 147350efa1b5SStefano Zampini { 147450efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 147550efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1476b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 147750efa1b5SStefano Zampini PetscErrorCode ierr; 147850efa1b5SStefano Zampini const PetscScalar one = 1.0; 147950efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 148050efa1b5SStefano Zampini const PetscScalar zero = 0.0; 148150efa1b5SStefano Zampini 148250efa1b5SStefano Zampini PetscFunctionBegin; 148350efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1484b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 148550efa1b5SStefano Zampini /* First Dirichlet solve */ 148650efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 148750efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 148850efa1b5SStefano Zampini /* 148950efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1490b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 149150efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 149250efa1b5SStefano Zampini */ 1493b097fa66SStefano Zampini if (n_D) { 1494b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 149550efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 149650efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1497b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1498b097fa66SStefano Zampini } else { 1499b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1500b097fa66SStefano Zampini } 150150efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 150250efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 150350efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 150450efa1b5SStefano Zampini } else { 1505b097fa66SStefano Zampini if (pcbddc->switch_static) { 150650efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1507b097fa66SStefano Zampini } 150850efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 150950efa1b5SStefano Zampini } 151050efa1b5SStefano Zampini 151150efa1b5SStefano Zampini /* Apply interface preconditioner 151250efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1513dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 151450efa1b5SStefano Zampini 151550efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 151650efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 151750efa1b5SStefano Zampini 151850efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 151950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 152050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1521b097fa66SStefano Zampini if (n_B) { 152250efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 152350efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1524b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1525b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1526b097fa66SStefano Zampini } 1527b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1528b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1529b097fa66SStefano Zampini if (pcbddc->switch_static) { 1530b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1531b097fa66SStefano Zampini } else { 1532b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1533b097fa66SStefano Zampini } 153450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 153550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1536b097fa66SStefano Zampini } else { 1537b097fa66SStefano Zampini if (pcbddc->switch_static) { 1538b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1539b097fa66SStefano Zampini } else { 1540b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1541b097fa66SStefano Zampini } 1542b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1543b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1544b097fa66SStefano Zampini } 154550efa1b5SStefano Zampini PetscFunctionReturn(0); 154650efa1b5SStefano Zampini } 1547da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1548674ae819SStefano Zampini 1549da1bb401SStefano Zampini #undef __FUNCT__ 1550da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1551da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1552da1bb401SStefano Zampini { 1553da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1554da1bb401SStefano Zampini PetscErrorCode ierr; 1555da1bb401SStefano Zampini 1556da1bb401SStefano Zampini PetscFunctionBegin; 1557da1bb401SStefano Zampini /* free data created by PCIS */ 1558da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 1559674ae819SStefano Zampini /* free BDDC custom data */ 1560674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1561674ae819SStefano Zampini /* destroy objects related to topography */ 1562674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1563674ae819SStefano Zampini /* free allocated graph structure */ 1564da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1565b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1566b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 156734a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1568674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 156934a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1570674ae819SStefano Zampini /* free solvers stuff */ 1571674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 157262a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 157362a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 157462a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1575906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 1576906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 1577906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1578906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1579906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1580906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1581906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1582906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1583906d46d4SStefano Zampini } 1584906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 15853425bc38SStefano Zampini /* remove functions */ 1586906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1587674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 1588bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 15892b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1590b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 15912b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1592bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 1593bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 159482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1595bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 159682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1597bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 159882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1599bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1600785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1601bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 160263602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1603bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1604bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1605bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1606bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1607674ae819SStefano Zampini /* Free the private data structure */ 1608674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1609da1bb401SStefano Zampini PetscFunctionReturn(0); 1610da1bb401SStefano Zampini } 16113425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 16121e6b0712SBarry Smith 16133425bc38SStefano Zampini #undef __FUNCT__ 16143425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 16153425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16163425bc38SStefano Zampini { 1617674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1618c08af4c6SStefano Zampini Vec copy_standard_rhs; 16193425bc38SStefano Zampini PC_IS* pcis; 16203425bc38SStefano Zampini PC_BDDC* pcbddc; 16213425bc38SStefano Zampini PetscErrorCode ierr; 16220c7d97c5SJed Brown 16233425bc38SStefano Zampini PetscFunctionBegin; 16243425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 16253425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 16263425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 16273425bc38SStefano Zampini 1628c08af4c6SStefano Zampini /* 1629c08af4c6SStefano Zampini change of basis for physical rhs if needed 1630c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1631c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1632c08af4c6SStefano Zampini */ 1633c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1634c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1635c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 16363425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1637c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1638c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1639fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1640fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1641c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1642c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1643674ae819SStefano Zampini /* Apply partition of unity */ 16443425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1645c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 16468eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 16473425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 16483425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16493425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 16503425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1651c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1652c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1653c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1654c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1655c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1656c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16573425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 16583425bc38SStefano Zampini } 1659c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 16603425bc38SStefano Zampini /* BDDC rhs */ 16613425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 16628eeda7d8SStefano Zampini if (pcbddc->switch_static) { 16633425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16643425bc38SStefano Zampini } 16653425bc38SStefano Zampini /* apply BDDC */ 1666dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 16673425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 16683425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 16693425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 16703425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16713425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16723425bc38SStefano Zampini PetscFunctionReturn(0); 16733425bc38SStefano Zampini } 16741e6b0712SBarry Smith 16753425bc38SStefano Zampini #undef __FUNCT__ 16763425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 16773425bc38SStefano Zampini /*@ 16780f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 16793425bc38SStefano Zampini 16803425bc38SStefano Zampini Collective 16813425bc38SStefano Zampini 16823425bc38SStefano Zampini Input Parameters: 16830f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 16840f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 16853425bc38SStefano Zampini 16863425bc38SStefano Zampini Output Parameters: 16870f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 16883425bc38SStefano Zampini 16893425bc38SStefano Zampini Level: developer 16903425bc38SStefano Zampini 16913425bc38SStefano Zampini Notes: 16923425bc38SStefano Zampini 16930f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 16943425bc38SStefano Zampini @*/ 16953425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16963425bc38SStefano Zampini { 1697674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 16983425bc38SStefano Zampini PetscErrorCode ierr; 16993425bc38SStefano Zampini 17003425bc38SStefano Zampini PetscFunctionBegin; 17013425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17023425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 17033425bc38SStefano Zampini PetscFunctionReturn(0); 17043425bc38SStefano Zampini } 17053425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 17061e6b0712SBarry Smith 17073425bc38SStefano Zampini #undef __FUNCT__ 17083425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 17093425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 17103425bc38SStefano Zampini { 1711674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17123425bc38SStefano Zampini PC_IS* pcis; 17133425bc38SStefano Zampini PC_BDDC* pcbddc; 17143425bc38SStefano Zampini PetscErrorCode ierr; 17153425bc38SStefano Zampini 17163425bc38SStefano Zampini PetscFunctionBegin; 17173425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17183425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 17193425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 17203425bc38SStefano Zampini 17213425bc38SStefano Zampini /* apply B_delta^T */ 17223425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17233425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17243425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 17253425bc38SStefano Zampini /* compute rhs for BDDC application */ 17263425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 17278eeda7d8SStefano Zampini if (pcbddc->switch_static) { 17283425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17293425bc38SStefano Zampini } 17303425bc38SStefano Zampini /* apply BDDC */ 1731dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 17323425bc38SStefano Zampini /* put values into standard global vector */ 17333425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17343425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17358eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 17363425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 17373425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 17383425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 17393425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17403425bc38SStefano Zampini } 17413425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17423425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17433425bc38SStefano Zampini /* final change of basis if needed 17443425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 17453308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 17463425bc38SStefano Zampini PetscFunctionReturn(0); 17473425bc38SStefano Zampini } 17481e6b0712SBarry Smith 17493425bc38SStefano Zampini #undef __FUNCT__ 17503425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 17513425bc38SStefano Zampini /*@ 17520f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 17533425bc38SStefano Zampini 17543425bc38SStefano Zampini Collective 17553425bc38SStefano Zampini 17563425bc38SStefano Zampini Input Parameters: 17570f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 17580f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 17593425bc38SStefano Zampini 17603425bc38SStefano Zampini Output Parameters: 17610f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 17623425bc38SStefano Zampini 17633425bc38SStefano Zampini Level: developer 17643425bc38SStefano Zampini 17653425bc38SStefano Zampini Notes: 17663425bc38SStefano Zampini 17670f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 17683425bc38SStefano Zampini @*/ 17693425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 17703425bc38SStefano Zampini { 1771674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17723425bc38SStefano Zampini PetscErrorCode ierr; 17733425bc38SStefano Zampini 17743425bc38SStefano Zampini PetscFunctionBegin; 17753425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17763425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 17773425bc38SStefano Zampini PetscFunctionReturn(0); 17783425bc38SStefano Zampini } 17793425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 17801e6b0712SBarry Smith 1781f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 1782edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 1783f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 1784f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 1785edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 1786f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 1787674ae819SStefano Zampini 17883425bc38SStefano Zampini #undef __FUNCT__ 17893425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 17903425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 17913425bc38SStefano Zampini { 1792674ae819SStefano Zampini 1793674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 17943425bc38SStefano Zampini Mat newmat; 1795674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 17963425bc38SStefano Zampini PC newpc; 1797ce94432eSBarry Smith MPI_Comm comm; 17983425bc38SStefano Zampini PetscErrorCode ierr; 17993425bc38SStefano Zampini 18003425bc38SStefano Zampini PetscFunctionBegin; 1801ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 18023425bc38SStefano Zampini /* FETIDP linear matrix */ 18033425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 18043425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 18053425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 18063425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 1807edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 18083425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 18093425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 18103425bc38SStefano Zampini /* FETIDP preconditioner */ 18113425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 18123425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 18133425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 18143425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 18153425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 18163425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 1817edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 18183425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 181923ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 18203425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 18213425bc38SStefano Zampini /* return pointers for objects created */ 18223425bc38SStefano Zampini *fetidp_mat=newmat; 18233425bc38SStefano Zampini *fetidp_pc=newpc; 18243425bc38SStefano Zampini PetscFunctionReturn(0); 18253425bc38SStefano Zampini } 18261e6b0712SBarry Smith 18273425bc38SStefano Zampini #undef __FUNCT__ 18283425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 18293425bc38SStefano Zampini /*@ 18300f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 18313425bc38SStefano Zampini 18323425bc38SStefano Zampini Collective 18333425bc38SStefano Zampini 18343425bc38SStefano Zampini Input Parameters: 18350f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 183628509bceSStefano Zampini 183728509bceSStefano Zampini Output Parameters: 18380f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 18390f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 184028509bceSStefano Zampini 184128509bceSStefano Zampini Options Database Keys: 18420f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 18433425bc38SStefano Zampini 18443425bc38SStefano Zampini Level: developer 18453425bc38SStefano Zampini 18463425bc38SStefano Zampini Notes: 18470f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 18483425bc38SStefano Zampini 18490f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 18503425bc38SStefano Zampini @*/ 18513425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 18523425bc38SStefano Zampini { 18533425bc38SStefano Zampini PetscErrorCode ierr; 18543425bc38SStefano Zampini 18553425bc38SStefano Zampini PetscFunctionBegin; 18563425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 18573425bc38SStefano Zampini if (pc->setupcalled) { 1858516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 1859f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 18603425bc38SStefano Zampini PetscFunctionReturn(0); 18613425bc38SStefano Zampini } 18620c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 1863da1bb401SStefano Zampini /*MC 1864da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 18650c7d97c5SJed Brown 186628509bceSStefano Zampini An implementation of the BDDC preconditioner based on 186728509bceSStefano Zampini 186828509bceSStefano Zampini .vb 186928509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 187028509bceSStefano 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 187128509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 18720f202f7eSStefano 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 187328509bceSStefano Zampini .ve 187428509bceSStefano Zampini 187528509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 187628509bceSStefano Zampini 18770f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 187828509bceSStefano Zampini 187928509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 188028509bceSStefano Zampini 1881b6fdb6dfSStefano 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. 1882b6fdb6dfSStefano Zampini 18830f202f7eSStefano Zampini Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 188428509bceSStefano Zampini 18850f202f7eSStefano 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() 18860f202f7eSStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS() 188728509bceSStefano Zampini 18880f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 188928509bceSStefano Zampini 18900f202f7eSStefano 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. 18910f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 189228509bceSStefano Zampini 18930f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 189428509bceSStefano Zampini 18950f202f7eSStefano 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. 189628509bceSStefano Zampini 18970f202f7eSStefano 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. 18980f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 18990f202f7eSStefano Zampini 19000f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 19010f202f7eSStefano Zampini 19020f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 19030f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 19040f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 19050f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 19060f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 19070f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 19080f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 190928509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 19100f202f7eSStefano 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) 19110f202f7eSStefano 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) 19120f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 19130f202f7eSStefano 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) 19140f202f7eSStefano 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) 191528509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 191628509bceSStefano Zampini 191728509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 191828509bceSStefano Zampini .vb 191928509bceSStefano Zampini -pc_bddc_dirichlet_ 192028509bceSStefano Zampini -pc_bddc_neumann_ 192128509bceSStefano Zampini -pc_bddc_coarse_ 192228509bceSStefano Zampini .ve 19230f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 192428509bceSStefano Zampini 19250f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 192628509bceSStefano Zampini .vb 1927312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 1928312be037SStefano Zampini -pc_bddc_neumann_lN_ 1929312be037SStefano Zampini -pc_bddc_coarse_lN_ 193028509bceSStefano Zampini .ve 19310f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 19320f202f7eSStefano 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. 19330f202f7eSStefano Zampini .vb 19340f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 19350f202f7eSStefano Zampini .ve 19360f202f7eSStefano 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 1937da1bb401SStefano Zampini 1938da1bb401SStefano Zampini Level: intermediate 1939da1bb401SStefano Zampini 1940b6fdb6dfSStefano Zampini Developer notes: 1941da1bb401SStefano Zampini 1942da1bb401SStefano Zampini Contributed by Stefano Zampini 1943da1bb401SStefano Zampini 1944da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 1945da1bb401SStefano Zampini M*/ 1946b2573a8aSBarry Smith 1947da1bb401SStefano Zampini #undef __FUNCT__ 1948da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 19498cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 1950da1bb401SStefano Zampini { 1951da1bb401SStefano Zampini PetscErrorCode ierr; 1952da1bb401SStefano Zampini PC_BDDC *pcbddc; 1953da1bb401SStefano Zampini 1954da1bb401SStefano Zampini PetscFunctionBegin; 1955da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 1956b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 1957da1bb401SStefano Zampini pc->data = (void*)pcbddc; 1958da1bb401SStefano Zampini 1959da1bb401SStefano Zampini /* create PCIS data structure */ 1960da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 1961da1bb401SStefano Zampini 196247d04d0dSStefano Zampini /* BDDC customization */ 196308a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 196447d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 196547d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 196647d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 196747d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 196847d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 196947d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 1970fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 1971fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 19723301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 197347d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 1974b9d89cd5SStefano Zampini /* private */ 1975727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 19760e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 19770e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 19780e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 1979e9189074SStefano Zampini pcbddc->n_vertices = 0; 1980727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 1981fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 198268457ee5SStefano Zampini pcbddc->coarse_size = -1; 1983f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1984727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 1985f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 1986f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 1987f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 1988674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 19890bdf917eSStefano Zampini pcbddc->NullSpace = 0; 19903972b0daSStefano Zampini pcbddc->temp_solution = 0; 1991534831adSStefano Zampini pcbddc->original_rhs = 0; 1992534831adSStefano Zampini pcbddc->local_mat = 0; 1993534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 1994b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 1995906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 1996da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 1997da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 1998da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 1999da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 200015aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 200115aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2002da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2003da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2004da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2005da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2006da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2007da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2008da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2009da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2010da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2011da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2012785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2013785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2014785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 201560d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 201660d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 201763602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2018da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 201963602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2020da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 202185c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 202247d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 202347d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2024b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 20254fad6a16SStefano Zampini pcbddc->current_level = 0; 20262b510759SStefano Zampini pcbddc->max_levels = 0; 2027323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 202874e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2029f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2030323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 2031da1bb401SStefano Zampini 2032674ae819SStefano Zampini /* create local graph structure */ 2033674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2034674ae819SStefano Zampini 2035674ae819SStefano Zampini /* scaling */ 2036674ae819SStefano Zampini pcbddc->work_scaling = 0; 203734a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2038ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2039b96c3477SStefano Zampini 2040b96c3477SStefano Zampini /* create sub schurs structure */ 2041b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2042b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2043b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2044b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2045b96c3477SStefano Zampini 2046b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2047da1bb401SStefano Zampini 2048b7eb3628SStefano Zampini /* adaptivity */ 2049f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 205008122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2051f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2052b7eb3628SStefano Zampini 2053da1bb401SStefano Zampini /* function pointers */ 2054da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 205593bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2056da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2057da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2058da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2059da1bb401SStefano Zampini pc->ops->view = 0; 2060da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2061da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2062da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2063534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2064534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2065da1bb401SStefano Zampini 2066da1bb401SStefano Zampini /* composing function */ 2067906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2068674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 2069bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 20702b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2071b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 20722b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2073bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2074bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 207582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2076bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 207782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2078bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 207982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2080bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 208182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2082bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 208363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2084bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2085bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2086bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2087bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2088da1bb401SStefano Zampini PetscFunctionReturn(0); 2089da1bb401SStefano Zampini } 20903425bc38SStefano Zampini 2091