153cdbc3dSStefano Zampini /* TODOLIST 2eb97c9d2SStefano Zampini 3eb97c9d2SStefano Zampini Solvers 4a0d3c3abSStefano Zampini - Add support for cholesky for coarse solver (similar to local solvers) 5eb97c9d2SStefano Zampini - Propagate ksp prefixes for solvers to mat objects? 6eb97c9d2SStefano Zampini 7eb97c9d2SStefano Zampini User interface 80f202f7eSStefano Zampini - ** DM attached to pc? 9eb97c9d2SStefano Zampini 10eb97c9d2SStefano Zampini Debugging output 11b9b85e73SStefano Zampini - * Better management of verbosity levels of debugging output 12eb97c9d2SStefano Zampini 13eb97c9d2SStefano Zampini Extra 14b9b85e73SStefano Zampini - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)? 15eb97c9d2SStefano Zampini - BDDC with MG framework? 16eb97c9d2SStefano Zampini 17eb97c9d2SStefano Zampini FETIDP 18eb97c9d2SStefano Zampini - Move FETIDP code to its own classes 19eb97c9d2SStefano Zampini 20eb97c9d2SStefano Zampini MATIS related operations contained in BDDC code 21eb97c9d2SStefano Zampini - Provide general case for subassembling 22eb97c9d2SStefano Zampini 2353cdbc3dSStefano Zampini */ 240c7d97c5SJed Brown 25ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 26ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 273b03a366Sstefano_zampini #include <petscblaslapack.h> 28674ae819SStefano Zampini 290369aaf7SStefano Zampini /* temporarily declare it */ 300369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 310369aaf7SStefano Zampini 320c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 330c7d97c5SJed Brown #undef __FUNCT__ 340c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 358c34d3f5SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptions *PetscOptionsObject,PC pc) 360c7d97c5SJed Brown { 370c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 380c7d97c5SJed Brown PetscErrorCode ierr; 390c7d97c5SJed Brown 400c7d97c5SJed Brown PetscFunctionBegin; 41e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 428eeda7d8SStefano Zampini /* Verbose debugging */ 438eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 448eeda7d8SStefano Zampini /* Primal space cumstomization */ 4508a5cf49SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_local_mat_graph","Use or not adjacency graph of local mat for interface analysis","none",pcbddc->use_local_adj,&pcbddc->use_local_adj,NULL);CHKERRQ(ierr); 468eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_vertices","Use or not corner dofs in coarse space","none",pcbddc->use_vertices,&pcbddc->use_vertices,NULL);CHKERRQ(ierr); 478eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_edges","Use or not edge constraints in coarse space","none",pcbddc->use_edges,&pcbddc->use_edges,NULL);CHKERRQ(ierr); 488eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_faces","Use or not face constraints in coarse space","none",pcbddc->use_faces,&pcbddc->use_faces,NULL);CHKERRQ(ierr); 496661aa1dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_true_nnsp","Use near null space attached to the matrix without modifications","none",pcbddc->use_nnsp_true,&pcbddc->use_nnsp_true,NULL);CHKERRQ(ierr); 50ac632422SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr); 518eeda7d8SStefano Zampini /* Change of basis */ 52b9b85e73SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_change_of_basis","Use or not internal change of basis on local edge nodes","none",pcbddc->use_change_of_basis,&pcbddc->use_change_of_basis,NULL);CHKERRQ(ierr); 53b9b85e73SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_change_on_faces","Use or not internal change of basis on local face nodes","none",pcbddc->use_change_on_faces,&pcbddc->use_change_on_faces,NULL);CHKERRQ(ierr); 54674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 55674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 56674ae819SStefano Zampini } 578eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 588eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_switch_static","Switch on static condensation ops around the interface preconditioner","none",pcbddc->switch_static,&pcbddc->switch_static,NULL);CHKERRQ(ierr); 5974e2c79eSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_redistribute","Number of procs where to redistribute coarse problem","none",pcbddc->redistribute_coarse,&pcbddc->redistribute_coarse,NULL);CHKERRQ(ierr); 600298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 612b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 62323d395dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_coarse_estimates","Use estimated eigenvalues for coarse problem","none",pcbddc->use_coarse_estimates,&pcbddc->use_coarse_estimates,NULL);CHKERRQ(ierr); 63674ae819SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_deluxe_scaling","Use deluxe scaling for BDDC","none",pcbddc->use_deluxe_scaling,&pcbddc->use_deluxe_scaling,NULL);CHKERRQ(ierr); 64b96c3477SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_rebuild","Whether or not the interface graph for Schur principal minors has to be rebuilt (i.e. define the interface without any adjacency)","none",pcbddc->sub_schurs_rebuild,&pcbddc->sub_schurs_rebuild,NULL);CHKERRQ(ierr); 65b96c3477SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_schur_layers","Number of dofs' layers for the computation of principal minors (i.e. -1 uses all dofs)","none",pcbddc->sub_schurs_layers,&pcbddc->sub_schurs_layers,NULL);CHKERRQ(ierr); 66b96c3477SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_use_useradj","Whether or not the CSR graph specified by the user should be used for computing successive layers (default is to use adj of local mat)","none",pcbddc->sub_schurs_use_useradj,&pcbddc->sub_schurs_use_useradj,NULL);CHKERRQ(ierr); 67683d3df6SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_exact","Whether or not to use the exact Schur complement instead of the reduced one (which excludes size 1 cc)","none",pcbddc->sub_schurs_exact_schur,&pcbddc->sub_schurs_exact_schur,NULL);CHKERRQ(ierr); 68ac632422SStefano 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); 694c6709b3SStefano 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); 7008122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 7108122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 723301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 73b0c7d250SStefano 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); 7406a4e24aSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_trick","Apply the benign subspace trick to a class of saddle point problems","none",pcbddc->benign_saddle_point,&pcbddc->benign_saddle_point,NULL);CHKERRQ(ierr); 754f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 760c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 770c7d97c5SJed Brown PetscFunctionReturn(0); 780c7d97c5SJed Brown } 790c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 80674ae819SStefano Zampini #undef __FUNCT__ 81906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 82906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 83b9b85e73SStefano Zampini { 84b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 85b9b85e73SStefano Zampini PetscErrorCode ierr; 86b9b85e73SStefano Zampini 87b9b85e73SStefano Zampini PetscFunctionBegin; 88b9b85e73SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 89b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 90b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 91b9b85e73SStefano Zampini PetscFunctionReturn(0); 92b9b85e73SStefano Zampini } 93b9b85e73SStefano Zampini #undef __FUNCT__ 94906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 95b9b85e73SStefano Zampini /*@ 96906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 97b9b85e73SStefano Zampini 98b9b85e73SStefano Zampini Collective on PC 99b9b85e73SStefano Zampini 100b9b85e73SStefano Zampini Input Parameters: 101b9b85e73SStefano Zampini + pc - the preconditioning context 102906d46d4SStefano Zampini - change - the change of basis matrix 103b9b85e73SStefano Zampini 104b9b85e73SStefano Zampini Level: intermediate 105b9b85e73SStefano Zampini 106b9b85e73SStefano Zampini Notes: 107b9b85e73SStefano Zampini 108b9b85e73SStefano Zampini .seealso: PCBDDC 109b9b85e73SStefano Zampini @*/ 110906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 111b9b85e73SStefano Zampini { 112b9b85e73SStefano Zampini PetscErrorCode ierr; 113b9b85e73SStefano Zampini 114b9b85e73SStefano Zampini PetscFunctionBegin; 115b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 116b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 117906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 118906d46d4SStefano Zampini if (pc->mat) { 119906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 120906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 121906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 122906d46d4SStefano Zampini if (rows_c != rows) { 123906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 124906d46d4SStefano Zampini } 125906d46d4SStefano Zampini if (cols_c != cols) { 126906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 127906d46d4SStefano Zampini } 128906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 129906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 130906d46d4SStefano Zampini if (rows_c != rows) { 131906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 132906d46d4SStefano Zampini } 133906d46d4SStefano Zampini if (cols_c != cols) { 134906d46d4SStefano Zampini SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 135906d46d4SStefano Zampini } 136906d46d4SStefano Zampini } 137906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 138b9b85e73SStefano Zampini PetscFunctionReturn(0); 139b9b85e73SStefano Zampini } 140b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 141b9b85e73SStefano Zampini #undef __FUNCT__ 14230368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS_BDDC" 14330368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 14430368db7SStefano Zampini { 14530368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 14630368db7SStefano Zampini PetscErrorCode ierr; 14730368db7SStefano Zampini 14830368db7SStefano Zampini PetscFunctionBegin; 14930368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 15030368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 15130368db7SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 15230368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 15330368db7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 15430368db7SStefano Zampini PetscFunctionReturn(0); 15530368db7SStefano Zampini } 15630368db7SStefano Zampini #undef __FUNCT__ 15730368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS" 15830368db7SStefano Zampini /*@ 15930368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 16030368db7SStefano Zampini 16130368db7SStefano Zampini Collective 16230368db7SStefano Zampini 16330368db7SStefano Zampini Input Parameters: 16430368db7SStefano Zampini + pc - the preconditioning context 16530368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 16630368db7SStefano Zampini 16730368db7SStefano Zampini Level: intermediate 16830368db7SStefano Zampini 16930368db7SStefano Zampini Notes: 17030368db7SStefano Zampini Any process can list any global node 17130368db7SStefano Zampini 17230368db7SStefano Zampini .seealso: PCBDDC 17330368db7SStefano Zampini @*/ 17430368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 17530368db7SStefano Zampini { 17630368db7SStefano Zampini PetscErrorCode ierr; 17730368db7SStefano Zampini 17830368db7SStefano Zampini PetscFunctionBegin; 17930368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 18030368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 18130368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 18230368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 18330368db7SStefano Zampini PetscFunctionReturn(0); 18430368db7SStefano Zampini } 18530368db7SStefano Zampini /* -------------------------------------------------------------------------- */ 18630368db7SStefano Zampini #undef __FUNCT__ 187674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 188674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 189674ae819SStefano Zampini { 190674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 191674ae819SStefano Zampini PetscErrorCode ierr; 1921e6b0712SBarry Smith 193674ae819SStefano Zampini PetscFunctionBegin; 194674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 19530368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 196674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 19730368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 19830368db7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 199674ae819SStefano Zampini PetscFunctionReturn(0); 200674ae819SStefano Zampini } 201674ae819SStefano Zampini #undef __FUNCT__ 202674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 203674ae819SStefano Zampini /*@ 20428509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 205674ae819SStefano Zampini 20617eb1463SStefano Zampini Collective 207674ae819SStefano Zampini 208674ae819SStefano Zampini Input Parameters: 209674ae819SStefano Zampini + pc - the preconditioning context 21017eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 211674ae819SStefano Zampini 212674ae819SStefano Zampini Level: intermediate 213674ae819SStefano Zampini 214674ae819SStefano Zampini Notes: 215674ae819SStefano Zampini 216674ae819SStefano Zampini .seealso: PCBDDC 217674ae819SStefano Zampini @*/ 218674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 219674ae819SStefano Zampini { 220674ae819SStefano Zampini PetscErrorCode ierr; 221674ae819SStefano Zampini 222674ae819SStefano Zampini PetscFunctionBegin; 223674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 224674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 22517eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 226674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 227674ae819SStefano Zampini PetscFunctionReturn(0); 228674ae819SStefano Zampini } 229674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 2300c7d97c5SJed Brown #undef __FUNCT__ 2314fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 2324fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 2334fad6a16SStefano Zampini { 2344fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2354fad6a16SStefano Zampini 2364fad6a16SStefano Zampini PetscFunctionBegin; 2374fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 2384fad6a16SStefano Zampini PetscFunctionReturn(0); 2394fad6a16SStefano Zampini } 2401e6b0712SBarry Smith 2414fad6a16SStefano Zampini #undef __FUNCT__ 2424fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 2434fad6a16SStefano Zampini /*@ 24428509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 2454fad6a16SStefano Zampini 2464fad6a16SStefano Zampini Logically collective on PC 2474fad6a16SStefano Zampini 2484fad6a16SStefano Zampini Input Parameters: 2494fad6a16SStefano Zampini + pc - the preconditioning context 25028509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 2514fad6a16SStefano Zampini 2520f202f7eSStefano Zampini Options Database Keys: 2530f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 2544fad6a16SStefano Zampini 2554fad6a16SStefano Zampini Level: intermediate 2564fad6a16SStefano Zampini 2574fad6a16SStefano Zampini Notes: 2580f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 2594fad6a16SStefano Zampini 2600f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2614fad6a16SStefano Zampini @*/ 2624fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2634fad6a16SStefano Zampini { 2644fad6a16SStefano Zampini PetscErrorCode ierr; 2654fad6a16SStefano Zampini 2664fad6a16SStefano Zampini PetscFunctionBegin; 2674fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2682b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2694fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2704fad6a16SStefano Zampini PetscFunctionReturn(0); 2714fad6a16SStefano Zampini } 2722b510759SStefano Zampini 273b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2742b510759SStefano Zampini #undef __FUNCT__ 275b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 276b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 277b8ffe317SStefano Zampini { 278b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 279b8ffe317SStefano Zampini 280b8ffe317SStefano Zampini PetscFunctionBegin; 28185c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 282b8ffe317SStefano Zampini PetscFunctionReturn(0); 283b8ffe317SStefano Zampini } 284b8ffe317SStefano Zampini 285b8ffe317SStefano Zampini #undef __FUNCT__ 286b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 287b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 2882b510759SStefano Zampini { 2892b510759SStefano Zampini PetscErrorCode ierr; 2902b510759SStefano Zampini 2912b510759SStefano Zampini PetscFunctionBegin; 2922b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 293b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 294b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 2952b510759SStefano Zampini PetscFunctionReturn(0); 2962b510759SStefano Zampini } 2971e6b0712SBarry Smith 2984fad6a16SStefano Zampini #undef __FUNCT__ 2992b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 3002b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 3014fad6a16SStefano Zampini { 3024fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3034fad6a16SStefano Zampini 3044fad6a16SStefano Zampini PetscFunctionBegin; 3052b510759SStefano Zampini pcbddc->current_level = level; 3064fad6a16SStefano Zampini PetscFunctionReturn(0); 3074fad6a16SStefano Zampini } 3081e6b0712SBarry Smith 3094fad6a16SStefano Zampini #undef __FUNCT__ 310b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 311b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 312b8ffe317SStefano Zampini { 313b8ffe317SStefano Zampini PetscErrorCode ierr; 314b8ffe317SStefano Zampini 315b8ffe317SStefano Zampini PetscFunctionBegin; 316b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 317b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 318b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 319b8ffe317SStefano Zampini PetscFunctionReturn(0); 320b8ffe317SStefano Zampini } 321b8ffe317SStefano Zampini 322b8ffe317SStefano Zampini #undef __FUNCT__ 3232b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 3242b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 3252b510759SStefano Zampini { 3262b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3272b510759SStefano Zampini 3282b510759SStefano Zampini PetscFunctionBegin; 3292b510759SStefano Zampini pcbddc->max_levels = levels; 3302b510759SStefano Zampini PetscFunctionReturn(0); 3312b510759SStefano Zampini } 3322b510759SStefano Zampini 3332b510759SStefano Zampini #undef __FUNCT__ 3342b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 3354fad6a16SStefano Zampini /*@ 33628509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 3374fad6a16SStefano Zampini 3384fad6a16SStefano Zampini Logically collective on PC 3394fad6a16SStefano Zampini 3404fad6a16SStefano Zampini Input Parameters: 3414fad6a16SStefano Zampini + pc - the preconditioning context 34228509bceSStefano Zampini - levels - the maximum number of levels (max 9) 3434fad6a16SStefano Zampini 3440f202f7eSStefano Zampini Options Database Keys: 3450f202f7eSStefano Zampini . -pc_bddc_levels 3464fad6a16SStefano Zampini 3474fad6a16SStefano Zampini Level: intermediate 3484fad6a16SStefano Zampini 3494fad6a16SStefano Zampini Notes: 3500f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 3514fad6a16SStefano Zampini 3520f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 3534fad6a16SStefano Zampini @*/ 3542b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 3554fad6a16SStefano Zampini { 3564fad6a16SStefano Zampini PetscErrorCode ierr; 3574fad6a16SStefano Zampini 3584fad6a16SStefano Zampini PetscFunctionBegin; 3594fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 3602b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 361312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3622b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3634fad6a16SStefano Zampini PetscFunctionReturn(0); 3644fad6a16SStefano Zampini } 3654fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3661e6b0712SBarry Smith 3674fad6a16SStefano Zampini #undef __FUNCT__ 3680bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 3690bdf917eSStefano Zampini static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 3700bdf917eSStefano Zampini { 3710bdf917eSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3720bdf917eSStefano Zampini PetscErrorCode ierr; 3730bdf917eSStefano Zampini 3740bdf917eSStefano Zampini PetscFunctionBegin; 3750bdf917eSStefano Zampini ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 3760bdf917eSStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 3770bdf917eSStefano Zampini pcbddc->NullSpace = NullSpace; 3780bdf917eSStefano Zampini PetscFunctionReturn(0); 3790bdf917eSStefano Zampini } 3801e6b0712SBarry Smith 3810bdf917eSStefano Zampini #undef __FUNCT__ 3820bdf917eSStefano Zampini #define __FUNCT__ "PCBDDCSetNullSpace" 3830bdf917eSStefano Zampini /*@ 38428509bceSStefano Zampini PCBDDCSetNullSpace - Set nullspace for BDDC operator 3850bdf917eSStefano Zampini 3860bdf917eSStefano Zampini Logically collective on PC and MatNullSpace 3870bdf917eSStefano Zampini 3880bdf917eSStefano Zampini Input Parameters: 3890bdf917eSStefano Zampini + pc - the preconditioning context 39028509bceSStefano Zampini - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 3910bdf917eSStefano Zampini 3920bdf917eSStefano Zampini Level: intermediate 3930bdf917eSStefano Zampini 3940bdf917eSStefano Zampini Notes: 3950bdf917eSStefano Zampini 3960bdf917eSStefano Zampini .seealso: PCBDDC 3970bdf917eSStefano Zampini @*/ 3980bdf917eSStefano Zampini PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 3990bdf917eSStefano Zampini { 4000bdf917eSStefano Zampini PetscErrorCode ierr; 4010bdf917eSStefano Zampini 4020bdf917eSStefano Zampini PetscFunctionBegin; 4030bdf917eSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 404674ae819SStefano Zampini PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 4052b510759SStefano Zampini PetscCheckSameComm(pc,1,NullSpace,2); 4060bdf917eSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 4070bdf917eSStefano Zampini PetscFunctionReturn(0); 4080bdf917eSStefano Zampini } 4090bdf917eSStefano Zampini /* -------------------------------------------------------------------------- */ 4101e6b0712SBarry Smith 4110bdf917eSStefano Zampini #undef __FUNCT__ 4123b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 4133b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 4143b03a366Sstefano_zampini { 4153b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4163b03a366Sstefano_zampini PetscErrorCode ierr; 4173b03a366Sstefano_zampini 4183b03a366Sstefano_zampini PetscFunctionBegin; 419785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 420785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4213b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 42236e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 42336e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 424fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4253b03a366Sstefano_zampini PetscFunctionReturn(0); 4263b03a366Sstefano_zampini } 4271e6b0712SBarry Smith 4283b03a366Sstefano_zampini #undef __FUNCT__ 4293b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 4303b03a366Sstefano_zampini /*@ 43128509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 4323b03a366Sstefano_zampini 433785d1243SStefano Zampini Collective 4343b03a366Sstefano_zampini 4353b03a366Sstefano_zampini Input Parameters: 4363b03a366Sstefano_zampini + pc - the preconditioning context 437785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 4383b03a366Sstefano_zampini 4393b03a366Sstefano_zampini Level: intermediate 4403b03a366Sstefano_zampini 4410f202f7eSStefano Zampini Notes: 4420f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 4433b03a366Sstefano_zampini 4440f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 4453b03a366Sstefano_zampini @*/ 4463b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 4473b03a366Sstefano_zampini { 4483b03a366Sstefano_zampini PetscErrorCode ierr; 4493b03a366Sstefano_zampini 4503b03a366Sstefano_zampini PetscFunctionBegin; 4513b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 452674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 453785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4543b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4553b03a366Sstefano_zampini PetscFunctionReturn(0); 4563b03a366Sstefano_zampini } 4573b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4581e6b0712SBarry Smith 4593b03a366Sstefano_zampini #undef __FUNCT__ 46082ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 46182ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4620c7d97c5SJed Brown { 4630c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4640c7d97c5SJed Brown PetscErrorCode ierr; 4650c7d97c5SJed Brown 4660c7d97c5SJed Brown PetscFunctionBegin; 467785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 468785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4690c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 4700c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 471785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 4727d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4730c7d97c5SJed Brown PetscFunctionReturn(0); 4740c7d97c5SJed Brown } 4750c7d97c5SJed Brown 4760c7d97c5SJed Brown #undef __FUNCT__ 47782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4789c0446d6SStefano Zampini /*@ 47982ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4809c0446d6SStefano Zampini 481785d1243SStefano Zampini Collective 48253cdbc3dSStefano Zampini 48353cdbc3dSStefano Zampini Input Parameters: 48453cdbc3dSStefano Zampini + pc - the preconditioning context 48582ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 48653cdbc3dSStefano Zampini 48753cdbc3dSStefano Zampini Level: intermediate 48853cdbc3dSStefano Zampini 4899c0446d6SStefano Zampini Notes: 49053cdbc3dSStefano Zampini 4910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 49253cdbc3dSStefano Zampini @*/ 49382ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 4940c7d97c5SJed Brown { 4950c7d97c5SJed Brown PetscErrorCode ierr; 4960c7d97c5SJed Brown 4970c7d97c5SJed Brown PetscFunctionBegin; 4980c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4990c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 50082ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 50182ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 5020c7d97c5SJed Brown PetscFunctionReturn(0); 5030c7d97c5SJed Brown } 5040c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 5050c7d97c5SJed Brown 5060c7d97c5SJed Brown #undef __FUNCT__ 5070c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 50853cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 5090c7d97c5SJed Brown { 5100c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 51153cdbc3dSStefano Zampini PetscErrorCode ierr; 5120c7d97c5SJed Brown 5130c7d97c5SJed Brown PetscFunctionBegin; 514785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 515785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 51653cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 51736e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 51836e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 519fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5200c7d97c5SJed Brown PetscFunctionReturn(0); 5210c7d97c5SJed Brown } 5221e6b0712SBarry Smith 5230c7d97c5SJed Brown #undef __FUNCT__ 5240c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 52557527edcSJed Brown /*@ 52628509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 52757527edcSJed Brown 528785d1243SStefano Zampini Collective 52957527edcSJed Brown 53057527edcSJed Brown Input Parameters: 53157527edcSJed Brown + pc - the preconditioning context 532785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 53357527edcSJed Brown 53457527edcSJed Brown Level: intermediate 53557527edcSJed Brown 5360f202f7eSStefano Zampini Notes: 5370f202f7eSStefano Zampini Any process can list any global node 53857527edcSJed Brown 5390f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 54057527edcSJed Brown @*/ 54153cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 5420c7d97c5SJed Brown { 5430c7d97c5SJed Brown PetscErrorCode ierr; 5440c7d97c5SJed Brown 5450c7d97c5SJed Brown PetscFunctionBegin; 5460c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 547674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 548785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 54953cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 55053cdbc3dSStefano Zampini PetscFunctionReturn(0); 55153cdbc3dSStefano Zampini } 55253cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5531e6b0712SBarry Smith 55453cdbc3dSStefano Zampini #undef __FUNCT__ 55582ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 55682ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5570c7d97c5SJed Brown { 5580c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5590c7d97c5SJed Brown PetscErrorCode ierr; 5600c7d97c5SJed Brown 5610c7d97c5SJed Brown PetscFunctionBegin; 562785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 563785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5640c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 5650c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 566785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 5677d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5680c7d97c5SJed Brown PetscFunctionReturn(0); 5690c7d97c5SJed Brown } 5700c7d97c5SJed Brown 5710c7d97c5SJed Brown #undef __FUNCT__ 57282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5730c7d97c5SJed Brown /*@ 57482ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 5750c7d97c5SJed Brown 576785d1243SStefano Zampini Collective 5770c7d97c5SJed Brown 5780c7d97c5SJed Brown Input Parameters: 5790c7d97c5SJed Brown + pc - the preconditioning context 58082ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 5810c7d97c5SJed Brown 5820c7d97c5SJed Brown Level: intermediate 5830c7d97c5SJed Brown 5840c7d97c5SJed Brown Notes: 5850c7d97c5SJed Brown 5860f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 5870c7d97c5SJed Brown @*/ 58882ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 5890c7d97c5SJed Brown { 5900c7d97c5SJed Brown PetscErrorCode ierr; 5910c7d97c5SJed Brown 5920c7d97c5SJed Brown PetscFunctionBegin; 5930c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5940c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 59582ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 59682ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 59753cdbc3dSStefano Zampini PetscFunctionReturn(0); 59853cdbc3dSStefano Zampini } 59953cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 60053cdbc3dSStefano Zampini 60153cdbc3dSStefano Zampini #undef __FUNCT__ 602da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 603da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 604da1bb401SStefano Zampini { 605da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 606da1bb401SStefano Zampini 607da1bb401SStefano Zampini PetscFunctionBegin; 608da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 609da1bb401SStefano Zampini PetscFunctionReturn(0); 610da1bb401SStefano Zampini } 6111e6b0712SBarry Smith 612da1bb401SStefano Zampini #undef __FUNCT__ 613da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 614da1bb401SStefano Zampini /*@ 615785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 616da1bb401SStefano Zampini 617785d1243SStefano Zampini Collective 618785d1243SStefano Zampini 619785d1243SStefano Zampini Input Parameters: 620785d1243SStefano Zampini . pc - the preconditioning context 621785d1243SStefano Zampini 622785d1243SStefano Zampini Output Parameters: 623785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 624785d1243SStefano Zampini 625785d1243SStefano Zampini Level: intermediate 626785d1243SStefano Zampini 6270f202f7eSStefano Zampini Notes: 6280f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 629785d1243SStefano Zampini 630785d1243SStefano Zampini .seealso: PCBDDC 631785d1243SStefano Zampini @*/ 632785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 633785d1243SStefano Zampini { 634785d1243SStefano Zampini PetscErrorCode ierr; 635785d1243SStefano Zampini 636785d1243SStefano Zampini PetscFunctionBegin; 637785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 638785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 639785d1243SStefano Zampini PetscFunctionReturn(0); 640785d1243SStefano Zampini } 641785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 642785d1243SStefano Zampini 643785d1243SStefano Zampini #undef __FUNCT__ 644785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 645785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 646785d1243SStefano Zampini { 647785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 648785d1243SStefano Zampini 649785d1243SStefano Zampini PetscFunctionBegin; 650785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 651785d1243SStefano Zampini PetscFunctionReturn(0); 652785d1243SStefano Zampini } 653785d1243SStefano Zampini 654785d1243SStefano Zampini #undef __FUNCT__ 65582ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 656da1bb401SStefano Zampini /*@ 65782ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 658da1bb401SStefano Zampini 659785d1243SStefano Zampini Collective 660da1bb401SStefano Zampini 661da1bb401SStefano Zampini Input Parameters: 66228509bceSStefano Zampini . pc - the preconditioning context 663da1bb401SStefano Zampini 664da1bb401SStefano Zampini Output Parameters: 66528509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 666da1bb401SStefano Zampini 667da1bb401SStefano Zampini Level: intermediate 668da1bb401SStefano Zampini 669da1bb401SStefano Zampini Notes: 6700f202f7eSStefano 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). 6710f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 672da1bb401SStefano Zampini 673da1bb401SStefano Zampini .seealso: PCBDDC 674da1bb401SStefano Zampini @*/ 67582ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 676da1bb401SStefano Zampini { 677da1bb401SStefano Zampini PetscErrorCode ierr; 678da1bb401SStefano Zampini 679da1bb401SStefano Zampini PetscFunctionBegin; 680da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 68182ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 682da1bb401SStefano Zampini PetscFunctionReturn(0); 683da1bb401SStefano Zampini } 684da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 6851e6b0712SBarry Smith 686da1bb401SStefano Zampini #undef __FUNCT__ 68753cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 68853cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 68953cdbc3dSStefano Zampini { 69053cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 69153cdbc3dSStefano Zampini 69253cdbc3dSStefano Zampini PetscFunctionBegin; 69353cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 69453cdbc3dSStefano Zampini PetscFunctionReturn(0); 69553cdbc3dSStefano Zampini } 6961e6b0712SBarry Smith 69753cdbc3dSStefano Zampini #undef __FUNCT__ 69853cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 69953cdbc3dSStefano Zampini /*@ 700785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 70153cdbc3dSStefano Zampini 702785d1243SStefano Zampini Collective 703785d1243SStefano Zampini 704785d1243SStefano Zampini Input Parameters: 705785d1243SStefano Zampini . pc - the preconditioning context 706785d1243SStefano Zampini 707785d1243SStefano Zampini Output Parameters: 708785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 709785d1243SStefano Zampini 710785d1243SStefano Zampini Level: intermediate 711785d1243SStefano Zampini 7120f202f7eSStefano Zampini Notes: 7130f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 714785d1243SStefano Zampini 715785d1243SStefano Zampini .seealso: PCBDDC 716785d1243SStefano Zampini @*/ 717785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 718785d1243SStefano Zampini { 719785d1243SStefano Zampini PetscErrorCode ierr; 720785d1243SStefano Zampini 721785d1243SStefano Zampini PetscFunctionBegin; 722785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 723785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 724785d1243SStefano Zampini PetscFunctionReturn(0); 725785d1243SStefano Zampini } 726785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 727785d1243SStefano Zampini 728785d1243SStefano Zampini #undef __FUNCT__ 729785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 730785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 731785d1243SStefano Zampini { 732785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 733785d1243SStefano Zampini 734785d1243SStefano Zampini PetscFunctionBegin; 735785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 736785d1243SStefano Zampini PetscFunctionReturn(0); 737785d1243SStefano Zampini } 738785d1243SStefano Zampini 739785d1243SStefano Zampini #undef __FUNCT__ 74082ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 74153cdbc3dSStefano Zampini /*@ 74282ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 74353cdbc3dSStefano Zampini 744785d1243SStefano Zampini Collective 74553cdbc3dSStefano Zampini 74653cdbc3dSStefano Zampini Input Parameters: 74728509bceSStefano Zampini . pc - the preconditioning context 74853cdbc3dSStefano Zampini 74953cdbc3dSStefano Zampini Output Parameters: 75028509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 75153cdbc3dSStefano Zampini 75253cdbc3dSStefano Zampini Level: intermediate 75353cdbc3dSStefano Zampini 75453cdbc3dSStefano Zampini Notes: 7550f202f7eSStefano 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). 7560f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 75753cdbc3dSStefano Zampini 75853cdbc3dSStefano Zampini .seealso: PCBDDC 75953cdbc3dSStefano Zampini @*/ 76082ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 76153cdbc3dSStefano Zampini { 76253cdbc3dSStefano Zampini PetscErrorCode ierr; 76353cdbc3dSStefano Zampini 76453cdbc3dSStefano Zampini PetscFunctionBegin; 76553cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 76682ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7670c7d97c5SJed Brown PetscFunctionReturn(0); 7680c7d97c5SJed Brown } 76936e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7701e6b0712SBarry Smith 77136e030ebSStefano Zampini #undef __FUNCT__ 772da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7731a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 77436e030ebSStefano Zampini { 77536e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 776da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 777da1bb401SStefano Zampini PetscErrorCode ierr; 77836e030ebSStefano Zampini 77936e030ebSStefano Zampini PetscFunctionBegin; 780674ae819SStefano Zampini /* free old CSR */ 781674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 782fb223d50SStefano Zampini /* TODO: PCBDDCGraphSetAdjacency */ 783674ae819SStefano Zampini /* get CSR into graph structure */ 784da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 785854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 786785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 787674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 788674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 789da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 7901a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 7911a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 792674ae819SStefano Zampini } 793575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 79436e030ebSStefano Zampini PetscFunctionReturn(0); 79536e030ebSStefano Zampini } 7961e6b0712SBarry Smith 79736e030ebSStefano Zampini #undef __FUNCT__ 798da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 79936e030ebSStefano Zampini /*@ 8000f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 80136e030ebSStefano Zampini 80236e030ebSStefano Zampini Not collective 80336e030ebSStefano Zampini 80436e030ebSStefano Zampini Input Parameters: 80536e030ebSStefano Zampini + pc - the preconditioning context 8060f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 80728509bceSStefano Zampini . xadj, adjncy - the CSR graph 80828509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 80936e030ebSStefano Zampini 81036e030ebSStefano Zampini Level: intermediate 81136e030ebSStefano Zampini 81236e030ebSStefano Zampini Notes: 81336e030ebSStefano Zampini 81428509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 81536e030ebSStefano Zampini @*/ 8161a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 81736e030ebSStefano Zampini { 818575ad6abSStefano Zampini void (*f)(void) = 0; 81936e030ebSStefano Zampini PetscErrorCode ierr; 82036e030ebSStefano Zampini 82136e030ebSStefano Zampini PetscFunctionBegin; 82236e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 823674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 824d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 825674ae819SStefano Zampini if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) { 826674ae819SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__); 827da1bb401SStefano Zampini } 82836e030ebSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 829575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 830575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 831575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 832575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 833575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 83436e030ebSStefano Zampini } 83536e030ebSStefano Zampini PetscFunctionReturn(0); 83636e030ebSStefano Zampini } 8379c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 8381e6b0712SBarry Smith 8399c0446d6SStefano Zampini #undef __FUNCT__ 84063602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 84163602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 84263602bcaSStefano Zampini { 84363602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 84463602bcaSStefano Zampini PetscInt i; 84563602bcaSStefano Zampini PetscErrorCode ierr; 84663602bcaSStefano Zampini 84763602bcaSStefano Zampini PetscFunctionBegin; 84863602bcaSStefano Zampini /* Destroy ISes if they were already set */ 84963602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 85063602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 85163602bcaSStefano Zampini } 85263602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 85363602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 85463602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 85563602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 85663602bcaSStefano Zampini } 85763602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 85863602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 85963602bcaSStefano Zampini /* allocate space then set */ 860d02579f5SStefano Zampini if (n_is) { 861d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 862d02579f5SStefano Zampini } 86363602bcaSStefano Zampini for (i=0;i<n_is;i++) { 86463602bcaSStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 86563602bcaSStefano Zampini pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 86663602bcaSStefano Zampini } 86763602bcaSStefano Zampini pcbddc->n_ISForDofsLocal=n_is; 86863602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8691a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 87063602bcaSStefano Zampini PetscFunctionReturn(0); 87163602bcaSStefano Zampini } 87263602bcaSStefano Zampini 87363602bcaSStefano Zampini #undef __FUNCT__ 87463602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 87563602bcaSStefano Zampini /*@ 87663602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 87763602bcaSStefano Zampini 87863602bcaSStefano Zampini Collective 87963602bcaSStefano Zampini 88063602bcaSStefano Zampini Input Parameters: 88163602bcaSStefano Zampini + pc - the preconditioning context 8820f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8830f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 88463602bcaSStefano Zampini 88563602bcaSStefano Zampini Level: intermediate 88663602bcaSStefano Zampini 8870f202f7eSStefano Zampini Notes: 8880f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 88963602bcaSStefano Zampini 89063602bcaSStefano Zampini .seealso: PCBDDC 89163602bcaSStefano Zampini @*/ 89263602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 89363602bcaSStefano Zampini { 89463602bcaSStefano Zampini PetscInt i; 89563602bcaSStefano Zampini PetscErrorCode ierr; 89663602bcaSStefano Zampini 89763602bcaSStefano Zampini PetscFunctionBegin; 89863602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 89963602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 90063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 90163602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 90263602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 90363602bcaSStefano Zampini } 904e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 90563602bcaSStefano Zampini PetscFunctionReturn(0); 90663602bcaSStefano Zampini } 90763602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 90863602bcaSStefano Zampini 90963602bcaSStefano Zampini #undef __FUNCT__ 9109c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 9119c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 9129c0446d6SStefano Zampini { 9139c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 9149c0446d6SStefano Zampini PetscInt i; 9159c0446d6SStefano Zampini PetscErrorCode ierr; 9169c0446d6SStefano Zampini 9179c0446d6SStefano Zampini PetscFunctionBegin; 918da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 9199c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 9209c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 9219c0446d6SStefano Zampini } 922d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 92363602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 92463602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 92563602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 92663602bcaSStefano Zampini } 92763602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 92863602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 929da1bb401SStefano Zampini /* allocate space then set */ 930d02579f5SStefano Zampini if (n_is) { 931785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 932d02579f5SStefano Zampini } 9339c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 934da1bb401SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 935da1bb401SStefano Zampini pcbddc->ISForDofs[i]=ISForDofs[i]; 9369c0446d6SStefano Zampini } 9379c0446d6SStefano Zampini pcbddc->n_ISForDofs=n_is; 93863602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 9391a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 9409c0446d6SStefano Zampini PetscFunctionReturn(0); 9419c0446d6SStefano Zampini } 9421e6b0712SBarry Smith 9439c0446d6SStefano Zampini #undef __FUNCT__ 9449c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 9459c0446d6SStefano Zampini /*@ 94663602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 9479c0446d6SStefano Zampini 94863602bcaSStefano Zampini Collective 9499c0446d6SStefano Zampini 9509c0446d6SStefano Zampini Input Parameters: 9519c0446d6SStefano Zampini + pc - the preconditioning context 9520f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9530f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 9549c0446d6SStefano Zampini 9559c0446d6SStefano Zampini Level: intermediate 9569c0446d6SStefano Zampini 9570f202f7eSStefano Zampini Notes: 9580f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 9599c0446d6SStefano Zampini 9609c0446d6SStefano Zampini .seealso: PCBDDC 9619c0446d6SStefano Zampini @*/ 9629c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 9639c0446d6SStefano Zampini { 9642b510759SStefano Zampini PetscInt i; 9659c0446d6SStefano Zampini PetscErrorCode ierr; 9669c0446d6SStefano Zampini 9679c0446d6SStefano Zampini PetscFunctionBegin; 9689c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 96963602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 9702b510759SStefano Zampini for (i=0;i<n_is;i++) { 97163602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 97263602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 9732b510759SStefano Zampini } 9749c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 9759c0446d6SStefano Zampini PetscFunctionReturn(0); 9769c0446d6SStefano Zampini } 977906d46d4SStefano Zampini 978da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 979534831adSStefano Zampini #undef __FUNCT__ 980534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 981534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 982534831adSStefano Zampini /* 983534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 984534831adSStefano Zampini guess if a transformation of basis approach has been selected. 9859c0446d6SStefano Zampini 986534831adSStefano Zampini Input Parameter: 987534831adSStefano Zampini + pc - the preconditioner contex 988534831adSStefano Zampini 989534831adSStefano Zampini Application Interface Routine: PCPreSolve() 990534831adSStefano Zampini 991534831adSStefano Zampini Notes: 992534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 993534831adSStefano Zampini the user, but instead is called by KSPSolve(). 994534831adSStefano Zampini */ 995534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 996534831adSStefano Zampini { 997534831adSStefano Zampini PetscErrorCode ierr; 998534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 999534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 10003972b0daSStefano Zampini Vec used_vec; 10018d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 100251694757SStefano Zampini PetscBool benign_correction_is_zero = PETSC_FALSE; 1003a3df083aSStefano Zampini PetscBool iscg = PETSC_FALSE; 1004534831adSStefano Zampini 1005534831adSStefano Zampini PetscFunctionBegin; 100685c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 100785c4d303SStefano Zampini if (ksp) { 100885c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 100985c4d303SStefano Zampini if (!iscg) { 101085c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 101185c4d303SStefano Zampini } 101285c4d303SStefano Zampini } 101385c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 101462a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 101562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 101662a6ff1dSStefano Zampini } 101762a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 101862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 101962a6ff1dSStefano Zampini } 10208d00608fSStefano Zampini 10213972b0daSStefano Zampini if (x) { 10223972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 10233972b0daSStefano Zampini used_vec = x; 10248d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 10253972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 10263972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 10273972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 10283972b0daSStefano Zampini } 10298efcfb23SStefano Zampini 10308efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 10313972b0daSStefano Zampini if (ksp) { 1032a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 10338efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 10348efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 10353972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 10363972b0daSStefano Zampini } 10373972b0daSStefano Zampini } 10383308cffdSStefano Zampini 10398d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 10403972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 1041a07ea27aSStefano Zampini if (rhs) { 10423975b054SStefano Zampini IS dirIS = NULL; 10433975b054SStefano Zampini 1044a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 10453975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 10463975b054SStefano Zampini if (dirIS) { 1047906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1048785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 10492b095fd8SStefano Zampini PetscScalar *array_x; 10502b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1051785d1243SStefano Zampini 10523972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 10533972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1054e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1055e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1056e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1057e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 105882ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 10593972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 10602b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10613972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10622fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 10633972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10642b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10653972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1066e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1067e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10688d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10691b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 10708efcfb23SStefano Zampini } 1071a07ea27aSStefano Zampini } 1072b76ba322SStefano Zampini 10738efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 10748d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 10758d00608fSStefano Zampini /* store the original rhs */ 10768d00608fSStefano Zampini if (copy_rhs) { 10778d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10788d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10798d00608fSStefano Zampini } 10808d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10813972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10820369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,rhs,rhs);CHKERRQ(ierr); 10833972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10848efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 10857acc28cbSStefano Zampini if (ksp) { 10867acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 10877acc28cbSStefano Zampini } 10883308cffdSStefano Zampini } 10898efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1090b76ba322SStefano Zampini 10910369aaf7SStefano Zampini /* When using the benign trick: (TODO: what about FETI-DP?) 10920369aaf7SStefano Zampini - change rhs on pressures (iteration matrix is surely of type MATIS) 10930369aaf7SStefano Zampini - compute initial vector in benign space 10940369aaf7SStefano Zampini */ 10950369aaf7SStefano Zampini if (rhs && pcbddc->benign_saddle_point) { 109606f24817SStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 10974f1b2e48SStefano Zampini PetscInt i; 109806f24817SStefano Zampini 109906f24817SStefano Zampini /* store the original rhs */ 110006f24817SStefano Zampini if (copy_rhs) { 110106f24817SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 110206f24817SStefano Zampini copy_rhs = PETSC_FALSE; 110306f24817SStefano Zampini } 110406f24817SStefano Zampini 110506f24817SStefano Zampini /* now change (locally) the basis */ 110606f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 110706f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,rhs,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 110806f24817SStefano Zampini if (pcbddc->benign_change) { 110906f24817SStefano Zampini ierr = MatMultTranspose(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 111006f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 111106f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 11125cfd9691SStefano Zampini /* swap local iteration matrix (with the benign trick, amat == pmat) */ 1113a3df083aSStefano Zampini if (pcbddc->benign_n && !pcbddc->benign_change_explicit) { 1114a3df083aSStefano Zampini /* benign shell mat mult for matis->A and pcis->A_IB*/ 1115a3df083aSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 1116a3df083aSStefano Zampini } else { 11175cfd9691SStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_FALSE);CHKERRQ(ierr); 111806f24817SStefano Zampini pcbddc->benign_original_mat = matis->A; 111906f24817SStefano Zampini matis->A = pcbddc->local_mat; 1120a3df083aSStefano Zampini } 112106f24817SStefano Zampini } else { 112206f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 112306f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 112477a2de6eSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 112577a2de6eSStefano Zampini pcbddc->benign_original_mat = matis->A; 112677a2de6eSStefano Zampini matis->A = pcbddc->local_mat; 112777a2de6eSStefano Zampini } 112806f24817SStefano Zampini } 112906f24817SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 113006f24817SStefano Zampini 11310369aaf7SStefano Zampini /* compute u^*_h as in Xuemin Tu's thesis (see Section 4.8.1) */ 11320369aaf7SStefano Zampini /* TODO: what about Stokes? */ 11330369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 11340369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 11350369aaf7SStefano Zampini } 11364f1b2e48SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 11374f1b2e48SStefano Zampini if (pcbddc->benign_n) { 11380369aaf7SStefano Zampini const PetscScalar *array; 11390369aaf7SStefano Zampini 11400369aaf7SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr); 11414f1b2e48SStefano Zampini for (i=0;i<pcbddc->benign_n;i++) pcbddc->benign_p0[i] = array[pcbddc->benign_p0_lidx[i]]; 11420369aaf7SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array);CHKERRQ(ierr); 11430369aaf7SStefano Zampini } 114451694757SStefano Zampini if (pcbddc->benign_null && iscg) { /* this is a workaround, need to understand more */ 11454f1b2e48SStefano Zampini PetscBool iszero_l = PETSC_TRUE; 11464f1b2e48SStefano Zampini for (i=0;i<pcbddc->benign_n;i++) { 11474f1b2e48SStefano Zampini iszero_l = (iszero_l && (PetscAbsScalar(pcbddc->benign_p0[i]) < PETSC_SMALL ? PETSC_TRUE : PETSC_FALSE)); 11484f1b2e48SStefano Zampini } 114951694757SStefano Zampini ierr = MPI_Allreduce(&iszero_l,&benign_correction_is_zero,1,MPIU_BOOL,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 115051694757SStefano Zampini } 1151*4fee134fSStefano Zampini if (!benign_correction_is_zero) { /* use the coarse solver only */ 1152015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,pcis->vec1_global,PETSC_FALSE);CHKERRQ(ierr); 1153*4fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 11540369aaf7SStefano Zampini ierr = PCApply_BDDC(pc,pcis->vec1_global,pcbddc->benign_vec);CHKERRQ(ierr); 1155*4fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 11564f1b2e48SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 1157015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,pcbddc->benign_vec,PETSC_FALSE);CHKERRQ(ierr); 1158b76ba322SStefano Zampini } 115951694757SStefano Zampini } 1160b76ba322SStefano Zampini 11610369aaf7SStefano Zampini /* change rhs and iteration matrix if using the change of basis */ 1162b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1163906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1164906d46d4SStefano Zampini 1165906d46d4SStefano Zampini /* get change ctx */ 1166906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1167906d46d4SStefano Zampini 1168906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 116977a2de6eSStefano Zampini if (!pcbddc->benign_original_mat) { 1170906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1171906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1172906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1173906d46d4SStefano Zampini 1174906d46d4SStefano Zampini /* change iteration matrix */ 1175906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1176906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1177906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 11789bd2e3c7SStefano Zampini } 11798d00608fSStefano Zampini /* store the original rhs */ 11808d00608fSStefano Zampini if (copy_rhs) { 11818d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11828d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 11838d00608fSStefano Zampini } 11848d00608fSStefano Zampini 1185906d46d4SStefano Zampini /* change rhs */ 1186906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1187906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 11888d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 118992e3dcfbSStefano Zampini } 119092e3dcfbSStefano Zampini 11910369aaf7SStefano Zampini /* remove non-benign solution from the rhs */ 11920369aaf7SStefano Zampini if (pcbddc->benign_saddle_point) { 11930369aaf7SStefano Zampini /* store the original rhs */ 11940369aaf7SStefano Zampini if (copy_rhs) { 11950369aaf7SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 11960369aaf7SStefano Zampini copy_rhs = PETSC_FALSE; 11970369aaf7SStefano Zampini } 119851694757SStefano Zampini if (benign_correction_is_zero) { /* still need to understand why it works great */ 119951694757SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 120051694757SStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 120151694757SStefano Zampini } 12020369aaf7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 12030369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 12040369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12050369aaf7SStefano Zampini } 12060369aaf7SStefano Zampini 12070369aaf7SStefano Zampini /* set initial guess if using PCG */ 12080369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 12090369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 12100369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12110369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12120369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 12130369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12140369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12150369aaf7SStefano Zampini if (ksp) { 12160369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 12170369aaf7SStefano Zampini } 12180369aaf7SStefano Zampini } 1219b0f5fe93SStefano Zampini if (pcbddc->benign_null) { 1220b0f5fe93SStefano Zampini MatNullSpace null_space; 1221b0f5fe93SStefano Zampini Vec nullv; 1222b0f5fe93SStefano Zampini PetscBool isnull; 12234f1b2e48SStefano Zampini PetscInt i; 1224b0f5fe93SStefano Zampini 12254f1b2e48SStefano Zampini for (i=0;i<pcbddc->benign_n;i++) pcbddc->benign_p0[i] = 1.; 1226b0f5fe93SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&nullv);CHKERRQ(ierr); 1227b0f5fe93SStefano Zampini ierr = VecSet(nullv,0.);CHKERRQ(ierr); 1228b0f5fe93SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,nullv,PETSC_FALSE);CHKERRQ(ierr); 1229b0f5fe93SStefano Zampini ierr = VecNormalize(nullv,NULL);CHKERRQ(ierr); 1230b0f5fe93SStefano Zampini ierr = MatNullSpaceCreate(PetscObjectComm((PetscObject)pc),PETSC_FALSE,1,&nullv,&null_space);CHKERRQ(ierr); 1231b0f5fe93SStefano Zampini ierr = MatNullSpaceTest(null_space,pc->mat,&isnull);CHKERRQ(ierr); 12324f1b2e48SStefano Zampini if (isnull) { 1233b0f5fe93SStefano Zampini ierr = MatSetNullSpace(pc->mat,null_space);CHKERRQ(ierr); 12344f1b2e48SStefano Zampini } 1235b0f5fe93SStefano Zampini ierr = MatNullSpaceDestroy(&null_space);CHKERRQ(ierr); 1236b0f5fe93SStefano Zampini ierr = VecDestroy(&nullv);CHKERRQ(ierr); 1237b0f5fe93SStefano Zampini } 1238b0f5fe93SStefano Zampini 123992e3dcfbSStefano Zampini /* remove nullspace if present */ 12408efcfb23SStefano Zampini if (ksp && x && pcbddc->NullSpace) { 12418efcfb23SStefano Zampini ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr); 12428d00608fSStefano Zampini /* store the original rhs */ 12438d00608fSStefano Zampini if (copy_rhs) { 12448d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 12458d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 12468d00608fSStefano Zampini } 12478d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 1248d0195637SJed Brown ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr); 1249b76ba322SStefano Zampini } 1250534831adSStefano Zampini PetscFunctionReturn(0); 1251534831adSStefano Zampini } 1252906d46d4SStefano Zampini 1253534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1254534831adSStefano Zampini #undef __FUNCT__ 1255534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1256534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1257534831adSStefano Zampini /* 1258534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1259534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1260534831adSStefano Zampini 1261534831adSStefano Zampini Input Parameter: 1262534831adSStefano Zampini + pc - the preconditioner contex 1263534831adSStefano Zampini 1264534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1265534831adSStefano Zampini 1266534831adSStefano Zampini Notes: 1267534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1268534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1269534831adSStefano Zampini */ 1270534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1271534831adSStefano Zampini { 1272534831adSStefano Zampini PetscErrorCode ierr; 1273534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1274534831adSStefano Zampini 1275534831adSStefano Zampini PetscFunctionBegin; 127677a2de6eSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && !pcbddc->benign_original_mat) { 1277906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1278906d46d4SStefano Zampini 1279906d46d4SStefano Zampini /* get change ctx */ 1280906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1281906d46d4SStefano Zampini 1282906d46d4SStefano Zampini /* restore iteration matrix */ 1283906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1284906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1285906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 12869c6a02ceSStefano Zampini } 1287906d46d4SStefano Zampini 12888666afb4SStefano Zampini /* need to restore the local matrices */ 128977a2de6eSStefano Zampini if (pcbddc->benign_original_mat) { 12908666afb4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->mat->data; 12918666afb4SStefano Zampini 1292a3df083aSStefano Zampini if (pcbddc->benign_n && !pcbddc->benign_change_explicit) { 1293a3df083aSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1294a3df083aSStefano Zampini } else { 12955cfd9691SStefano Zampini pcbddc->local_mat = matis->A; 12968666afb4SStefano Zampini matis->A = pcbddc->benign_original_mat; 12975cfd9691SStefano Zampini pcbddc->benign_original_mat = NULL; 12988666afb4SStefano Zampini } 1299a3df083aSStefano Zampini } 13008666afb4SStefano Zampini 1301906d46d4SStefano Zampini /* get solution in original basis */ 1302906d46d4SStefano Zampini if (x) { 1303906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 130406f24817SStefano Zampini 13052d3346b4SStefano Zampini /* restore solution on pressures */ 130606a4e24aSStefano Zampini if (pcbddc->benign_saddle_point) { 13072d3346b4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->mat->data; 13082d3346b4SStefano Zampini 13090369aaf7SStefano Zampini /* add non-benign solution */ 13108666afb4SStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13110369aaf7SStefano Zampini 13120369aaf7SStefano Zampini /* change basis on pressures for x */ 131306f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 131406f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,x,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13152d3346b4SStefano Zampini if (pcbddc->benign_change) { 13162d3346b4SStefano Zampini ierr = MatMult(pcbddc->benign_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 131706f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 131806f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec2_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13192d3346b4SStefano Zampini } else { 132006f24817SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 132106f24817SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13222d3346b4SStefano Zampini } 13232d3346b4SStefano Zampini } 13249c6a02ceSStefano Zampini 13250369aaf7SStefano Zampini /* change basis on x */ 13269c6a02ceSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 13279c6a02ceSStefano Zampini PCBDDCChange_ctx change_ctx; 13289c6a02ceSStefano Zampini 13299c6a02ceSStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 13300369aaf7SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1331906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 13323425bc38SStefano Zampini } 1333534831adSStefano Zampini } 1334906d46d4SStefano Zampini 13353972b0daSStefano Zampini /* add solution removed in presolve */ 13366bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 13373425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 13383425bc38SStefano Zampini } 1339906d46d4SStefano Zampini 1340fb223d50SStefano Zampini /* restore rhs to its original state */ 13418d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1342fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1343fb223d50SStefano Zampini } 13448d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 13458efcfb23SStefano Zampini 13468efcfb23SStefano Zampini /* restore ksp guess state */ 13478efcfb23SStefano Zampini if (ksp) { 13488efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 13498efcfb23SStefano Zampini } 1350534831adSStefano Zampini PetscFunctionReturn(0); 1351534831adSStefano Zampini } 1352534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 135353cdbc3dSStefano Zampini #undef __FUNCT__ 135453cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 13550c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13560c7d97c5SJed Brown /* 13570c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 13580c7d97c5SJed Brown by setting data structures and options. 13590c7d97c5SJed Brown 13600c7d97c5SJed Brown Input Parameter: 136153cdbc3dSStefano Zampini + pc - the preconditioner context 13620c7d97c5SJed Brown 13630c7d97c5SJed Brown Application Interface Routine: PCSetUp() 13640c7d97c5SJed Brown 13650c7d97c5SJed Brown Notes: 13660c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 13670c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 13680c7d97c5SJed Brown */ 136953cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 13700c7d97c5SJed Brown { 13710c7d97c5SJed Brown PetscErrorCode ierr; 13720c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 13735e8657edSStefano Zampini Mat_IS* matis; 137408122e43SStefano Zampini MatNullSpace nearnullspace; 1375c9ed8603SStefano Zampini IS zerodiag = NULL; 137691e8d312SStefano Zampini PetscInt nrows,ncols; 137708122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 13788de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 13795e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 13800c7d97c5SJed Brown 13810c7d97c5SJed Brown PetscFunctionBegin; 13825e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 13835e8657edSStefano Zampini if (!ismatis) { 13845e8657edSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 13855e8657edSStefano Zampini } 138691e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 138791e8d312SStefano Zampini if (nrows != ncols) { 138891e8d312SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 138991e8d312SStefano Zampini } 13905e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1391f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 13923b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1393fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1394f4ddd8eeSStefano Zampini /* split work */ 1395674ae819SStefano Zampini if (pc->setupcalled) { 1396d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1397674ae819SStefano Zampini computetopography = PETSC_FALSE; 1398674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1399674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1400674ae819SStefano Zampini computetopography = PETSC_TRUE; 1401674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1402674ae819SStefano Zampini } 1403674ae819SStefano Zampini } else { 1404674ae819SStefano Zampini computetopography = PETSC_TRUE; 1405674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1406674ae819SStefano Zampini } 1407fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1408fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1409fb180af4SStefano Zampini } 14108de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1411b087196eSStefano Zampini 1412b087196eSStefano Zampini /* check parameters' compatibility */ 14135a95e1ceSStefano Zampini if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) { 14145a95e1ceSStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 14155a95e1ceSStefano Zampini } 14165a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1417862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1418862806e4SStefano Zampini 14195a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 14206816873aSStefano Zampini if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) { 14216816873aSStefano 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"); 14226816873aSStefano Zampini } 14232d3346b4SStefano Zampini 14242d3346b4SStefano Zampini /* check if the iteration matrix is of type MATIS in case the benign trick has been requested */ 14252d3346b4SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 142606a4e24aSStefano Zampini if (pcbddc->benign_saddle_point && !ismatis) { 14272d3346b4SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner with benign subspace trick requires the iteration matrix to be of type MATIS"); 14282d3346b4SStefano Zampini } 14295cfd9691SStefano Zampini if (pcbddc->benign_saddle_point && pc->mat != pc->pmat) { 14305cfd9691SStefano Zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner with benign subspace trick requires the Amat == Pmat"); 14315cfd9691SStefano Zampini } 14322d3346b4SStefano Zampini 1433f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 143470cf5478SStefano Zampini if (pcbddc->dbg_flag) { 143570cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 143658a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 14371575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1438f4ddd8eeSStefano Zampini } 143958a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1440f4ddd8eeSStefano Zampini } 1441f4ddd8eeSStefano Zampini 14425e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 14435e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 14445e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 14455e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 14465e8657edSStefano Zampini } else { 1447b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 14485e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 14495e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1450d16cbb6bSStefano Zampini } 1451d16cbb6bSStefano Zampini 14524f1b2e48SStefano Zampini /* detect local disconnected subdomains if requested and not done before */ 14534f1b2e48SStefano Zampini if (pcbddc->detect_disconnected && !pcbddc->n_local_subs) { 14544f1b2e48SStefano Zampini ierr = MatDetectDisconnectedComponents(pcbddc->local_mat,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 14554f1b2e48SStefano Zampini } 14564f1b2e48SStefano Zampini 14574f1b2e48SStefano Zampini /* 14584f1b2e48SStefano Zampini change basis on local pressures (aka zerodiag dofs) 14594f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 14604f1b2e48SStefano Zampini */ 1461d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 14629f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 14639f47a83aSStefano Zampini 1464a3df083aSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis) pcbddc->benign_change_explicit = PETSC_TRUE; 1465339f8db1SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat */ 1466339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1467a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1468c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 14699f47a83aSStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 1470a3df083aSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcbddc->benign_n && !pcbddc->benign_change_explicit && !pcbddc->dbg_flag) { 1471a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 1472a3df083aSStefano Zampini } else { 14739f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1474674ae819SStefano Zampini } 1475a3df083aSStefano Zampini } 1476f1a72664SStefano Zampini /* propagate relevant information */ 14774f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 14783301b35fSStefano Zampini if (matis->A->symmetric_set) { 14793301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 14803301b35fSStefano Zampini } 1481e496cd5dSStefano Zampini #endif 148206a4e24aSStefano Zampini if (matis->A->symmetric_set) { 148306a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 148406a4e24aSStefano Zampini } 148506a4e24aSStefano Zampini if (matis->A->spd_set) { 148606a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 148706a4e24aSStefano Zampini } 1488e496cd5dSStefano Zampini 14895e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 14905e8657edSStefano Zampini { 14915e8657edSStefano Zampini Mat temp_mat; 14925e8657edSStefano Zampini 14935e8657edSStefano Zampini temp_mat = matis->A; 14945e8657edSStefano Zampini matis->A = pcbddc->local_mat; 14955e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 14965e8657edSStefano Zampini pcbddc->local_mat = matis->A; 14975e8657edSStefano Zampini matis->A = temp_mat; 14985e8657edSStefano Zampini } 1499684f6988SStefano Zampini 150081d14e9dSStefano Zampini /* Analyze interface */ 1501674ae819SStefano Zampini if (computetopography) { 1502674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15038de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1504674ae819SStefano Zampini } 1505fb8d54d4SStefano Zampini 15065408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 15075408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 15085408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 15095408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 151009f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 15115408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 151209f581a4SStefano Zampini } 15135408967cSStefano Zampini #endif 15144f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 151506f24817SStefano Zampini 1516b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1517684f6988SStefano Zampini if (computesolvers) { 1518d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1519d5574798SStefano Zampini 1520d5574798SStefano Zampini if (computesubschurs && computetopography) { 152108122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1522b1b3d7a2SStefano Zampini } 1523df4d28bfSStefano Zampini if (sub_schurs->schur_explicit) { 15242070dbb6SStefano Zampini if (computesubschurs) { 152508122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 15262070dbb6SStefano Zampini } 1527d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1528d5574798SStefano Zampini } else { 1529d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 15302070dbb6SStefano Zampini if (computesubschurs) { 1531d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1532d5574798SStefano Zampini } 15332070dbb6SStefano Zampini } 153408122e43SStefano Zampini if (pcbddc->adaptive_selection) { 153508122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 15368de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1537b7eb3628SStefano Zampini } 1538b1b3d7a2SStefano Zampini } 1539684f6988SStefano Zampini 1540f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1541fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1542f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1543f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1544f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1545f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1546f4ddd8eeSStefano Zampini } else { 1547f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1548f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1549f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1550165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1551f4ddd8eeSStefano Zampini PetscInt i; 1552165b64e2SStefano Zampini const Vec *nearnullvecs; 1553165b64e2SStefano Zampini PetscObjectState state; 1554165b64e2SStefano Zampini PetscInt nnsp_size; 1555165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1556f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1557f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1558165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1559f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1560f4ddd8eeSStefano Zampini break; 1561f4ddd8eeSStefano Zampini } 1562f4ddd8eeSStefano Zampini } 1563f4ddd8eeSStefano Zampini } 1564f4ddd8eeSStefano Zampini } 1565f4ddd8eeSStefano Zampini } else { 1566f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1567f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1568f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1569f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1570f4ddd8eeSStefano Zampini } 1571f4ddd8eeSStefano Zampini } 1572f4ddd8eeSStefano Zampini 1573f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1574727cdba6SStefano Zampini /* reset primal space flags */ 1575f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1576727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 15778de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1578727cdba6SStefano Zampini /* It also sets the primal space flags */ 1579674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1580e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1581f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 15823975b054SStefano Zampini } 15835e8657edSStefano Zampini 15843975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 15855e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 15865e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1587c263805aSStefano Zampini Mat temp_mat = NULL; 15885e8657edSStefano Zampini 15894f1b2e48SStefano Zampini if (pcbddc->benign_change) { 1590c263805aSStefano Zampini /* insert B0 in pcbddc->local_mat */ 1591c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_FALSE);CHKERRQ(ierr); 1592f1a72664SStefano Zampini /* swap local matrices */ 1593f1a72664SStefano Zampini ierr = MatISGetLocalMat(pc->pmat,&temp_mat);CHKERRQ(ierr); 1594f1a72664SStefano Zampini ierr = PetscObjectReference((PetscObject)temp_mat);CHKERRQ(ierr); 1595f1a72664SStefano Zampini ierr = MatISSetLocalMat(pc->pmat,pcbddc->local_mat);CHKERRQ(ierr); 1596f1a72664SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 1597c263805aSStefano Zampini } 15985e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 15994f1b2e48SStefano Zampini if (pcbddc->benign_change) { 1600c263805aSStefano Zampini /* restore original matrix */ 1601f1a72664SStefano Zampini ierr = MatISSetLocalMat(pc->pmat,temp_mat);CHKERRQ(ierr); 1602f1a72664SStefano Zampini ierr = PetscObjectDereference((PetscObject)temp_mat);CHKERRQ(ierr); 1603c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1604c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1605c263805aSStefano Zampini } 16065e8657edSStefano Zampini /* get submatrices */ 16075e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16085e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16095e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16105e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16115e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16125e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16133975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16143975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16159c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1616b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16175e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16185e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16195e8657edSStefano Zampini } 1620b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 162199cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1622b96c3477SStefano Zampini /* SetUp Scaling operator */ 1623674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16240c7d97c5SJed Brown } 16250369aaf7SStefano Zampini 162658a03d70SStefano Zampini if (pcbddc->dbg_flag) { 162758a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 16282b510759SStefano Zampini } 16290c7d97c5SJed Brown PetscFunctionReturn(0); 16300c7d97c5SJed Brown } 16310c7d97c5SJed Brown 16320c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 16330c7d97c5SJed Brown /* 163450efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 16350c7d97c5SJed Brown 16360c7d97c5SJed Brown Input Parameters: 16370f202f7eSStefano Zampini + pc - the preconditioner context 16380f202f7eSStefano Zampini - r - input vector (global) 16390c7d97c5SJed Brown 16400c7d97c5SJed Brown Output Parameter: 16410c7d97c5SJed Brown . z - output vector (global) 16420c7d97c5SJed Brown 16430c7d97c5SJed Brown Application Interface Routine: PCApply() 16440c7d97c5SJed Brown */ 16450c7d97c5SJed Brown #undef __FUNCT__ 16460c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 164753cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 16480c7d97c5SJed Brown { 16490c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 16500c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1651b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 16520c7d97c5SJed Brown PetscErrorCode ierr; 16533b03a366Sstefano_zampini const PetscScalar one = 1.0; 16543b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 16552617d88aSStefano Zampini const PetscScalar zero = 0.0; 16560c7d97c5SJed Brown 16570c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 16580c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1659b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 16600c7d97c5SJed Brown 16610c7d97c5SJed Brown PetscFunctionBegin; 1662015636ebSStefano Zampini if (pcbddc->benign_saddle_point) { /* get p0 from r */ 1663015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1664efc2fbd9SStefano Zampini } 166585c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1666b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 16670c7d97c5SJed Brown /* First Dirichlet solve */ 16680c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16690c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16700c7d97c5SJed Brown /* 16710c7d97c5SJed Brown Assembling right hand side for BDDC operator 1672b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1673674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 16740c7d97c5SJed Brown */ 1675b097fa66SStefano Zampini if (n_D) { 1676b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 16770c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 16788eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1679b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1680b097fa66SStefano Zampini } else { 1681b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1682b097fa66SStefano Zampini } 16830c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 16840c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1685674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1686b76ba322SStefano Zampini } else { 1687*4fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1688b097fa66SStefano Zampini if (pcbddc->switch_static) { 16890bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1690b097fa66SStefano Zampini } 1691674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1692b76ba322SStefano Zampini } 1693*4fee134fSStefano Zampini } 1694b76ba322SStefano Zampini 16952617d88aSStefano Zampini /* Apply interface preconditioner 16962617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1697dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 16982617d88aSStefano Zampini 1699674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1700674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 17010c7d97c5SJed Brown 17023b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 17030c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17040c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1705b097fa66SStefano Zampini if (n_B) { 17060c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 17078eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1708b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1709b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1710b097fa66SStefano Zampini } 1711df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1712efc2fbd9SStefano Zampini 1713b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1714b097fa66SStefano Zampini if (pcbddc->switch_static) { 1715b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1716b097fa66SStefano Zampini } else { 1717b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1718b097fa66SStefano Zampini } 17190c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17200c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1721b097fa66SStefano Zampini } else { 1722b097fa66SStefano Zampini if (pcbddc->switch_static) { 1723b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1724b097fa66SStefano Zampini } else { 1725b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1726b097fa66SStefano Zampini } 1727b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1728b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1729b097fa66SStefano Zampini } 1730efc2fbd9SStefano Zampini 1731015636ebSStefano Zampini if (pcbddc->benign_saddle_point) { /* set p0 (computed in PCBDDCApplyInterface) */ 1732015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1733efc2fbd9SStefano Zampini } 17340c7d97c5SJed Brown PetscFunctionReturn(0); 17350c7d97c5SJed Brown } 173650efa1b5SStefano Zampini 173750efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 173850efa1b5SStefano Zampini /* 173950efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 174050efa1b5SStefano Zampini 174150efa1b5SStefano Zampini Input Parameters: 17420f202f7eSStefano Zampini + pc - the preconditioner context 17430f202f7eSStefano Zampini - r - input vector (global) 174450efa1b5SStefano Zampini 174550efa1b5SStefano Zampini Output Parameter: 174650efa1b5SStefano Zampini . z - output vector (global) 174750efa1b5SStefano Zampini 174850efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 174950efa1b5SStefano Zampini */ 175050efa1b5SStefano Zampini #undef __FUNCT__ 175150efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 175250efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 175350efa1b5SStefano Zampini { 175450efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 175550efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1756b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 175750efa1b5SStefano Zampini PetscErrorCode ierr; 175850efa1b5SStefano Zampini const PetscScalar one = 1.0; 175950efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 176050efa1b5SStefano Zampini const PetscScalar zero = 0.0; 176150efa1b5SStefano Zampini 176250efa1b5SStefano Zampini PetscFunctionBegin; 1763537c1cdfSStefano Zampini if (pcbddc->benign_saddle_point) { /* get p0 from r */ 1764537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1765537c1cdfSStefano Zampini } 176650efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1767b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 176850efa1b5SStefano Zampini /* First Dirichlet solve */ 176950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 177050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 177150efa1b5SStefano Zampini /* 177250efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1773b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 177450efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 177550efa1b5SStefano Zampini */ 1776b097fa66SStefano Zampini if (n_D) { 1777b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 177850efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 177950efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1780b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1781b097fa66SStefano Zampini } else { 1782b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1783b097fa66SStefano Zampini } 178450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 178550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 178650efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 178750efa1b5SStefano Zampini } else { 1788b097fa66SStefano Zampini if (pcbddc->switch_static) { 178950efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1790b097fa66SStefano Zampini } 179150efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 179250efa1b5SStefano Zampini } 179350efa1b5SStefano Zampini 179450efa1b5SStefano Zampini /* Apply interface preconditioner 179550efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1796dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 179750efa1b5SStefano Zampini 179850efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 179950efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 180050efa1b5SStefano Zampini 180150efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 180250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 180350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1804b097fa66SStefano Zampini if (n_B) { 180550efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 180650efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1807b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1808b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1809b097fa66SStefano Zampini } 1810b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1811b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1812b097fa66SStefano Zampini if (pcbddc->switch_static) { 1813b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1814b097fa66SStefano Zampini } else { 1815b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1816b097fa66SStefano Zampini } 181750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 181850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1819b097fa66SStefano Zampini } else { 1820b097fa66SStefano Zampini if (pcbddc->switch_static) { 1821b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1822b097fa66SStefano Zampini } else { 1823b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1824b097fa66SStefano Zampini } 1825b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1826b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1827b097fa66SStefano Zampini } 1828537c1cdfSStefano Zampini if (pcbddc->benign_saddle_point) { /* set p0 (computed in PCBDDCApplyInterface) */ 1829537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1830537c1cdfSStefano Zampini } 183150efa1b5SStefano Zampini PetscFunctionReturn(0); 183250efa1b5SStefano Zampini } 1833da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1834674ae819SStefano Zampini 1835da1bb401SStefano Zampini #undef __FUNCT__ 1836da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1837da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1838da1bb401SStefano Zampini { 1839da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1840da1bb401SStefano Zampini PetscErrorCode ierr; 1841da1bb401SStefano Zampini 1842da1bb401SStefano Zampini PetscFunctionBegin; 1843da1bb401SStefano Zampini /* free data created by PCIS */ 1844da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 1845674ae819SStefano Zampini /* free BDDC custom data */ 1846674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1847674ae819SStefano Zampini /* destroy objects related to topography */ 1848674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1849674ae819SStefano Zampini /* free allocated graph structure */ 1850da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1851b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1852b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 185334a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1854674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 185534a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1856674ae819SStefano Zampini /* free solvers stuff */ 1857674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 185862a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 185962a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 186062a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1861906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 1862906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 1863906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1864906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1865906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1866906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1867906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1868906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1869906d46d4SStefano Zampini } 1870906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 18713425bc38SStefano Zampini /* remove functions */ 1872906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1873674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 187430368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 1875bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 18762b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1877b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 18782b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1879bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 1880bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 188182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1882bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 188382ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1884bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 188582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1886bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1887785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1888bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 188963602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1890bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1891bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1892bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1893bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1894674ae819SStefano Zampini /* Free the private data structure */ 1895674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1896da1bb401SStefano Zampini PetscFunctionReturn(0); 1897da1bb401SStefano Zampini } 18983425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 18991e6b0712SBarry Smith 19003425bc38SStefano Zampini #undef __FUNCT__ 19013425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 19023425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 19033425bc38SStefano Zampini { 1904674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1905c08af4c6SStefano Zampini Vec copy_standard_rhs; 19063425bc38SStefano Zampini PC_IS* pcis; 19073425bc38SStefano Zampini PC_BDDC* pcbddc; 19083425bc38SStefano Zampini PetscErrorCode ierr; 19090c7d97c5SJed Brown 19103425bc38SStefano Zampini PetscFunctionBegin; 19113425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 19123425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 19133425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 19143425bc38SStefano Zampini 1915c08af4c6SStefano Zampini /* 1916c08af4c6SStefano Zampini change of basis for physical rhs if needed 1917c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1918c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1919c08af4c6SStefano Zampini */ 1920c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1921c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1922c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 19233425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1924c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1925c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1926fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1927fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1928c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1929c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1930674ae819SStefano Zampini /* Apply partition of unity */ 19313425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1932c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 19338eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 19343425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 19353425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 19363425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 19373425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1938c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1939c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1940c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1941c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1942c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1943c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19443425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 19453425bc38SStefano Zampini } 1946c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 19473425bc38SStefano Zampini /* BDDC rhs */ 19483425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 19498eeda7d8SStefano Zampini if (pcbddc->switch_static) { 19503425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 19513425bc38SStefano Zampini } 19523425bc38SStefano Zampini /* apply BDDC */ 1953dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 19543425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 19553425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 19563425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 19573425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19583425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19593425bc38SStefano Zampini PetscFunctionReturn(0); 19603425bc38SStefano Zampini } 19611e6b0712SBarry Smith 19623425bc38SStefano Zampini #undef __FUNCT__ 19633425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 19643425bc38SStefano Zampini /*@ 19650f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 19663425bc38SStefano Zampini 19673425bc38SStefano Zampini Collective 19683425bc38SStefano Zampini 19693425bc38SStefano Zampini Input Parameters: 19700f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 19710f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 19723425bc38SStefano Zampini 19733425bc38SStefano Zampini Output Parameters: 19740f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 19753425bc38SStefano Zampini 19763425bc38SStefano Zampini Level: developer 19773425bc38SStefano Zampini 19783425bc38SStefano Zampini Notes: 19793425bc38SStefano Zampini 19800f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 19813425bc38SStefano Zampini @*/ 19823425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 19833425bc38SStefano Zampini { 1984674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 19853425bc38SStefano Zampini PetscErrorCode ierr; 19863425bc38SStefano Zampini 19873425bc38SStefano Zampini PetscFunctionBegin; 19883425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 19893425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 19903425bc38SStefano Zampini PetscFunctionReturn(0); 19913425bc38SStefano Zampini } 19923425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 19931e6b0712SBarry Smith 19943425bc38SStefano Zampini #undef __FUNCT__ 19953425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 19963425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 19973425bc38SStefano Zampini { 1998674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 19993425bc38SStefano Zampini PC_IS* pcis; 20003425bc38SStefano Zampini PC_BDDC* pcbddc; 20013425bc38SStefano Zampini PetscErrorCode ierr; 20023425bc38SStefano Zampini 20033425bc38SStefano Zampini PetscFunctionBegin; 20043425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20053425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20063425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20073425bc38SStefano Zampini 20083425bc38SStefano Zampini /* apply B_delta^T */ 20093425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20103425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20113425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 20123425bc38SStefano Zampini /* compute rhs for BDDC application */ 20133425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 20148eeda7d8SStefano Zampini if (pcbddc->switch_static) { 20153425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20163425bc38SStefano Zampini } 20173425bc38SStefano Zampini /* apply BDDC */ 2018dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 20193425bc38SStefano Zampini /* put values into standard global vector */ 20203425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20213425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20228eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 20233425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 20243425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 20253425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 20263425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20273425bc38SStefano Zampini } 20283425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20293425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 20303425bc38SStefano Zampini /* final change of basis if needed 20313425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 20323308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 20333425bc38SStefano Zampini PetscFunctionReturn(0); 20343425bc38SStefano Zampini } 20351e6b0712SBarry Smith 20363425bc38SStefano Zampini #undef __FUNCT__ 20373425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 20383425bc38SStefano Zampini /*@ 20390f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 20403425bc38SStefano Zampini 20413425bc38SStefano Zampini Collective 20423425bc38SStefano Zampini 20433425bc38SStefano Zampini Input Parameters: 20440f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 20450f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 20463425bc38SStefano Zampini 20473425bc38SStefano Zampini Output Parameters: 20480f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 20493425bc38SStefano Zampini 20503425bc38SStefano Zampini Level: developer 20513425bc38SStefano Zampini 20523425bc38SStefano Zampini Notes: 20533425bc38SStefano Zampini 20540f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 20553425bc38SStefano Zampini @*/ 20563425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 20573425bc38SStefano Zampini { 2058674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 20593425bc38SStefano Zampini PetscErrorCode ierr; 20603425bc38SStefano Zampini 20613425bc38SStefano Zampini PetscFunctionBegin; 20623425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20633425bc38SStefano Zampini ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 20643425bc38SStefano Zampini PetscFunctionReturn(0); 20653425bc38SStefano Zampini } 20663425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20671e6b0712SBarry Smith 2068f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2069edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2070f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2071f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2072edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2073f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2074674ae819SStefano Zampini 20753425bc38SStefano Zampini #undef __FUNCT__ 20763425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 20773425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 20783425bc38SStefano Zampini { 2079674ae819SStefano Zampini 2080674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 20813425bc38SStefano Zampini Mat newmat; 2082674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 20833425bc38SStefano Zampini PC newpc; 2084ce94432eSBarry Smith MPI_Comm comm; 20853425bc38SStefano Zampini PetscErrorCode ierr; 20863425bc38SStefano Zampini 20873425bc38SStefano Zampini PetscFunctionBegin; 2088ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 20893425bc38SStefano Zampini /* FETIDP linear matrix */ 20903425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 20913425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 20923425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 20933425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2094edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 20953425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 20963425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 20973425bc38SStefano Zampini /* FETIDP preconditioner */ 20983425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 20993425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 21003425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 21013425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 21023425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 21033425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2104edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 21053425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 210623ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 21073425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 21083425bc38SStefano Zampini /* return pointers for objects created */ 21093425bc38SStefano Zampini *fetidp_mat=newmat; 21103425bc38SStefano Zampini *fetidp_pc=newpc; 21113425bc38SStefano Zampini PetscFunctionReturn(0); 21123425bc38SStefano Zampini } 21131e6b0712SBarry Smith 21143425bc38SStefano Zampini #undef __FUNCT__ 21153425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 21163425bc38SStefano Zampini /*@ 21170f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 21183425bc38SStefano Zampini 21193425bc38SStefano Zampini Collective 21203425bc38SStefano Zampini 21213425bc38SStefano Zampini Input Parameters: 21220f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 212328509bceSStefano Zampini 212428509bceSStefano Zampini Output Parameters: 21250f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 21260f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 212728509bceSStefano Zampini 212828509bceSStefano Zampini Options Database Keys: 21290f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 21303425bc38SStefano Zampini 21313425bc38SStefano Zampini Level: developer 21323425bc38SStefano Zampini 21333425bc38SStefano Zampini Notes: 21340f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 21353425bc38SStefano Zampini 21360f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 21373425bc38SStefano Zampini @*/ 21383425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 21393425bc38SStefano Zampini { 21403425bc38SStefano Zampini PetscErrorCode ierr; 21413425bc38SStefano Zampini 21423425bc38SStefano Zampini PetscFunctionBegin; 21433425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 21443425bc38SStefano Zampini if (pc->setupcalled) { 2145516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2146f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 21473425bc38SStefano Zampini PetscFunctionReturn(0); 21483425bc38SStefano Zampini } 21490c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2150da1bb401SStefano Zampini /*MC 2151da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 21520c7d97c5SJed Brown 215328509bceSStefano Zampini An implementation of the BDDC preconditioner based on 215428509bceSStefano Zampini 215528509bceSStefano Zampini .vb 215628509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 215728509bceSStefano 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 215828509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 21590f202f7eSStefano 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 216028509bceSStefano Zampini .ve 216128509bceSStefano Zampini 216228509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 216328509bceSStefano Zampini 21640f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 216528509bceSStefano Zampini 216628509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 216728509bceSStefano Zampini 2168b6fdb6dfSStefano 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. 2169b6fdb6dfSStefano Zampini 21700f202f7eSStefano Zampini Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 217128509bceSStefano Zampini 21720f202f7eSStefano 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() 217330368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 217428509bceSStefano Zampini 21750f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 217628509bceSStefano Zampini 21770f202f7eSStefano 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. 21780f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 217928509bceSStefano Zampini 21800f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 218128509bceSStefano Zampini 2182df4d28bfSStefano Zampini Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS or MKL_PARDISO are present. Future versions of the code will also consider using PASTIX. 218328509bceSStefano Zampini 21840f202f7eSStefano 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. 21850f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 21860f202f7eSStefano Zampini 21870f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 21880f202f7eSStefano Zampini 21890f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 21900f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 21910f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 21920f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 21930f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 21940f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 21950f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 219628509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 21970f202f7eSStefano 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) 21980f202f7eSStefano 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) 21990f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 22000f202f7eSStefano 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) 2201df4d28bfSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 220228509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 220328509bceSStefano Zampini 220428509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 220528509bceSStefano Zampini .vb 220628509bceSStefano Zampini -pc_bddc_dirichlet_ 220728509bceSStefano Zampini -pc_bddc_neumann_ 220828509bceSStefano Zampini -pc_bddc_coarse_ 220928509bceSStefano Zampini .ve 22100f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 221128509bceSStefano Zampini 22120f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 221328509bceSStefano Zampini .vb 2214312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2215312be037SStefano Zampini -pc_bddc_neumann_lN_ 2216312be037SStefano Zampini -pc_bddc_coarse_lN_ 221728509bceSStefano Zampini .ve 22180f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 22190f202f7eSStefano 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. 22200f202f7eSStefano Zampini .vb 22210f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 22220f202f7eSStefano Zampini .ve 22230f202f7eSStefano 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 2224da1bb401SStefano Zampini 2225da1bb401SStefano Zampini Level: intermediate 2226da1bb401SStefano Zampini 2227b6fdb6dfSStefano Zampini Developer notes: 2228da1bb401SStefano Zampini 2229da1bb401SStefano Zampini Contributed by Stefano Zampini 2230da1bb401SStefano Zampini 2231da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2232da1bb401SStefano Zampini M*/ 2233b2573a8aSBarry Smith 2234da1bb401SStefano Zampini #undef __FUNCT__ 2235da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 22368cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2237da1bb401SStefano Zampini { 2238da1bb401SStefano Zampini PetscErrorCode ierr; 2239da1bb401SStefano Zampini PC_BDDC *pcbddc; 2240da1bb401SStefano Zampini 2241da1bb401SStefano Zampini PetscFunctionBegin; 2242da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2243b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2244da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2245da1bb401SStefano Zampini 2246da1bb401SStefano Zampini /* create PCIS data structure */ 2247da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2248da1bb401SStefano Zampini 224947d04d0dSStefano Zampini /* BDDC customization */ 225008a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 225147d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 225247d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 225347d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 225447d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 225547d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 225647d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2257fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2258fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 22593301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 226006a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 226147d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2262b9d89cd5SStefano Zampini /* private */ 2263727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 22640e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 22650e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 22660e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2267e9189074SStefano Zampini pcbddc->n_vertices = 0; 2268727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2269fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 227068457ee5SStefano Zampini pcbddc->coarse_size = -1; 2271f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2272727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2273f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2274f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2275f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2276674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 227730368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 22780bdf917eSStefano Zampini pcbddc->NullSpace = 0; 22793972b0daSStefano Zampini pcbddc->temp_solution = 0; 2280534831adSStefano Zampini pcbddc->original_rhs = 0; 2281534831adSStefano Zampini pcbddc->local_mat = 0; 2282534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2283b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2284906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 2285da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2286da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2287da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2288da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 228915aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 229015aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2291da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2292da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2293da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2294da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2295da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2296da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2297da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2298da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2299da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2300da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2301785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2302785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2303785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 230460d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 230560d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 230663602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2307da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 230863602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2309da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 231085c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 231147d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 231247d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2313b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 23144fad6a16SStefano Zampini pcbddc->current_level = 0; 23152b510759SStefano Zampini pcbddc->max_levels = 0; 2316323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 231774e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2318f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2319323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 23204f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 23214f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 23224f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 232381d14e9dSStefano Zampini 232481d14e9dSStefano Zampini /* benign subspace trick */ 232581d14e9dSStefano Zampini pcbddc->benign_change = 0; 23260369aaf7SStefano Zampini pcbddc->benign_vec = 0; 23270369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 23280369aaf7SStefano Zampini pcbddc->benign_sf = 0; 23294f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 23304f1b2e48SStefano Zampini pcbddc->benign_n = 0; 23314f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 23324f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 23334f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2334b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 233581d14e9dSStefano Zampini 2336674ae819SStefano Zampini /* create local graph structure */ 2337674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2338674ae819SStefano Zampini 2339674ae819SStefano Zampini /* scaling */ 2340674ae819SStefano Zampini pcbddc->work_scaling = 0; 234134a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2342ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2343b96c3477SStefano Zampini 2344b96c3477SStefano Zampini /* create sub schurs structure */ 2345b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2346b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2347b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2348b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2349b96c3477SStefano Zampini 2350b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2351da1bb401SStefano Zampini 2352b7eb3628SStefano Zampini /* adaptivity */ 2353f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 235408122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2355f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2356b7eb3628SStefano Zampini 2357da1bb401SStefano Zampini /* function pointers */ 2358da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 235993bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2360da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2361da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2362da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2363da1bb401SStefano Zampini pc->ops->view = 0; 2364da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2365da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2366da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2367534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2368534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2369da1bb401SStefano Zampini 2370da1bb401SStefano Zampini /* composing function */ 2371906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2372674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 237330368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2374bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 23752b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2376b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 23772b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2378bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2379bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 238082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2381bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 238282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2383bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 238482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2385bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 238682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2387bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 238863602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2389bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2390bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2391bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2392bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2393da1bb401SStefano Zampini PetscFunctionReturn(0); 2394da1bb401SStefano Zampini } 23953425bc38SStefano Zampini 2396