153cdbc3dSStefano Zampini /* TODOLIST 2eb97c9d2SStefano Zampini 3eb97c9d2SStefano Zampini Solvers 4a0d3c3abSStefano Zampini - Add support for cholesky for coarse solver (similar to local solvers) 5eb97c9d2SStefano Zampini - Propagate ksp prefixes for solvers to mat objects? 6eb97c9d2SStefano Zampini 7eb97c9d2SStefano Zampini User interface 80f202f7eSStefano Zampini - ** DM attached to pc? 9eb97c9d2SStefano Zampini 10eb97c9d2SStefano Zampini Debugging output 11b9b85e73SStefano Zampini - * Better management of verbosity levels of debugging output 12eb97c9d2SStefano Zampini 13eb97c9d2SStefano Zampini Extra 14b9b85e73SStefano Zampini - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)? 15eb97c9d2SStefano Zampini - BDDC with MG framework? 16eb97c9d2SStefano Zampini 17eb97c9d2SStefano Zampini FETIDP 18eb97c9d2SStefano Zampini - Move FETIDP code to its own classes 19eb97c9d2SStefano Zampini 20eb97c9d2SStefano Zampini MATIS related operations contained in BDDC code 21eb97c9d2SStefano Zampini - Provide general case for subassembling 22eb97c9d2SStefano Zampini 2353cdbc3dSStefano Zampini */ 240c7d97c5SJed Brown 25ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 26ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 273b03a366Sstefano_zampini #include <petscblaslapack.h> 28674ae819SStefano Zampini 290c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 300c7d97c5SJed Brown #undef __FUNCT__ 310c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 324416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 330c7d97c5SJed Brown { 340c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 350c7d97c5SJed Brown PetscErrorCode ierr; 360c7d97c5SJed Brown 370c7d97c5SJed Brown PetscFunctionBegin; 38e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 398eeda7d8SStefano Zampini /* Verbose debugging */ 40*c7017625SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_dirichlet_approximate","Inform PCBDDC that we are using approximate Dirichlet solvers","none",pcbddc->NullSpace_corr[0],&pcbddc->NullSpace_corr[0],NULL);CHKERRQ(ierr); 41*c7017625SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_dirichlet_approximate_scale","Inform PCBDDC that we need to scale the Dirichlet solve","none",pcbddc->NullSpace_corr[1],&pcbddc->NullSpace_corr[1],NULL);CHKERRQ(ierr); 42*c7017625SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_neumann_approximate","Inform PCBDDC that we are using approximate Neumann solvers","none",pcbddc->NullSpace_corr[2],&pcbddc->NullSpace_corr[2],NULL);CHKERRQ(ierr); 43*c7017625SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_neumann_approximate_scale","Inform PCBDDC that we need to scale the Neumann solve","none",pcbddc->NullSpace_corr[3],&pcbddc->NullSpace_corr[3],NULL);CHKERRQ(ierr); 448eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 456b78500eSPatrick Sanan /* Primal space customization */ 4608a5cf49SStefano 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); 478eeda7d8SStefano 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); 488eeda7d8SStefano 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); 498eeda7d8SStefano 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); 506661aa1dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_true_nnsp","Use near null space attached to the matrix without modifications","none",pcbddc->use_nnsp_true,&pcbddc->use_nnsp_true,NULL);CHKERRQ(ierr); 51ac632422SStefano 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); 528eeda7d8SStefano Zampini /* Change of basis */ 53b9b85e73SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_change_of_basis","Use or not internal change of basis on local edge nodes","none",pcbddc->use_change_of_basis,&pcbddc->use_change_of_basis,NULL);CHKERRQ(ierr); 54b9b85e73SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_change_on_faces","Use or not internal change of basis on local face nodes","none",pcbddc->use_change_on_faces,&pcbddc->use_change_on_faces,NULL);CHKERRQ(ierr); 55674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 56674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 57674ae819SStefano Zampini } 588eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 598eeda7d8SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_switch_static","Switch on static condensation ops around the interface preconditioner","none",pcbddc->switch_static,&pcbddc->switch_static,NULL);CHKERRQ(ierr); 6074e2c79eSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_redistribute","Number of procs where to redistribute coarse problem","none",pcbddc->redistribute_coarse,&pcbddc->redistribute_coarse,NULL);CHKERRQ(ierr); 610298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 622b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 63323d395dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_coarse_estimates","Use estimated eigenvalues for coarse problem","none",pcbddc->use_coarse_estimates,&pcbddc->use_coarse_estimates,NULL);CHKERRQ(ierr); 64674ae819SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_deluxe_scaling","Use deluxe scaling for BDDC","none",pcbddc->use_deluxe_scaling,&pcbddc->use_deluxe_scaling,NULL);CHKERRQ(ierr); 65b96c3477SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_rebuild","Whether or not the interface graph for Schur principal minors has to be rebuilt (i.e. define the interface without any adjacency)","none",pcbddc->sub_schurs_rebuild,&pcbddc->sub_schurs_rebuild,NULL);CHKERRQ(ierr); 66b96c3477SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_schur_layers","Number of dofs' layers for the computation of principal minors (i.e. -1 uses all dofs)","none",pcbddc->sub_schurs_layers,&pcbddc->sub_schurs_layers,NULL);CHKERRQ(ierr); 67b96c3477SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_use_useradj","Whether or not the CSR graph specified by the user should be used for computing successive layers (default is to use adj of local mat)","none",pcbddc->sub_schurs_use_useradj,&pcbddc->sub_schurs_use_useradj,NULL);CHKERRQ(ierr); 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); 740c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 750c7d97c5SJed Brown PetscFunctionReturn(0); 760c7d97c5SJed Brown } 776b78500eSPatrick Sanan 786b78500eSPatrick Sanan /* -------------------------------------------------------------------------- */ 796b78500eSPatrick Sanan #undef __FUNCT__ 806b78500eSPatrick Sanan #define __FUNCT__ "PCView_BDDC" 816b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 826b78500eSPatrick Sanan { 836b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 846b78500eSPatrick Sanan PetscErrorCode ierr; 856b78500eSPatrick Sanan PetscBool isascii,isstring; 866b78500eSPatrick Sanan 876b78500eSPatrick Sanan PetscFunctionBegin; 886b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 896b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 906b78500eSPatrick Sanan 916b78500eSPatrick Sanan /* In a braindead way, print out anything which the user can control from the command line, 926b78500eSPatrick Sanan cribbing from PCSetFromOptions_BDDC */ 936b78500eSPatrick Sanan 946b78500eSPatrick Sanan /* Nothing printed for the String viewer */ 956b78500eSPatrick Sanan 966b78500eSPatrick Sanan /* ASCII viewer */ 976b78500eSPatrick Sanan if (isascii) { 986b78500eSPatrick Sanan /* Verbose debugging */ 996b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use verbose output: %d\n",pcbddc->dbg_flag);CHKERRQ(ierr); 1006b78500eSPatrick Sanan 1016b78500eSPatrick Sanan /* Primal space customization */ 1026b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use local mat graph: %d\n",pcbddc->use_local_adj);CHKERRQ(ierr); 1036b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use vertices: %d\n",pcbddc->use_vertices);CHKERRQ(ierr); 1046b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 1056b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 1066b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 1076b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 1086b78500eSPatrick Sanan 1096b78500eSPatrick Sanan /* Change of basis */ 1106b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 1116b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 1126b78500eSPatrick Sanan 1136b78500eSPatrick Sanan /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 1146b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 1156b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Coarse problem restribute procs: %d\n",pcbddc->redistribute_coarse);CHKERRQ(ierr); 1166b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 1176b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr); 1186b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 1196b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 1206b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 1216b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 1226b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 1236b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Fast deluxe scaling: %d\n",pcbddc->faster_deluxe);CHKERRQ(ierr); 1246b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Adaptive constraint selection threshold: %g\n",pcbddc->adaptive_threshold);CHKERRQ(ierr); 1256b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 1266b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 1276b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 1286b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 1296b78500eSPatrick Sanan } 1306b78500eSPatrick Sanan 1316b78500eSPatrick Sanan PetscFunctionReturn(0); 1326b78500eSPatrick Sanan } 1336b78500eSPatrick Sanan 1340c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 135674ae819SStefano Zampini #undef __FUNCT__ 136906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 137906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 138b9b85e73SStefano Zampini { 139b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 140b9b85e73SStefano Zampini PetscErrorCode ierr; 141b9b85e73SStefano Zampini 142b9b85e73SStefano Zampini PetscFunctionBegin; 143b9b85e73SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 144b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 145b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 146b9b85e73SStefano Zampini PetscFunctionReturn(0); 147b9b85e73SStefano Zampini } 148b9b85e73SStefano Zampini #undef __FUNCT__ 149906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 150b9b85e73SStefano Zampini /*@ 151906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 152b9b85e73SStefano Zampini 153b9b85e73SStefano Zampini Collective on PC 154b9b85e73SStefano Zampini 155b9b85e73SStefano Zampini Input Parameters: 156b9b85e73SStefano Zampini + pc - the preconditioning context 157906d46d4SStefano Zampini - change - the change of basis matrix 158b9b85e73SStefano Zampini 159b9b85e73SStefano Zampini Level: intermediate 160b9b85e73SStefano Zampini 161b9b85e73SStefano Zampini Notes: 162b9b85e73SStefano Zampini 163b9b85e73SStefano Zampini .seealso: PCBDDC 164b9b85e73SStefano Zampini @*/ 165906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 166b9b85e73SStefano Zampini { 167b9b85e73SStefano Zampini PetscErrorCode ierr; 168b9b85e73SStefano Zampini 169b9b85e73SStefano Zampini PetscFunctionBegin; 170b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 171b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 172906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 173906d46d4SStefano Zampini if (pc->mat) { 174906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 175906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 176906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 1776c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 1786c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 179906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 180906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 1816c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 1826c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 183906d46d4SStefano Zampini } 184906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 185b9b85e73SStefano Zampini PetscFunctionReturn(0); 186b9b85e73SStefano Zampini } 187b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 188b9b85e73SStefano Zampini #undef __FUNCT__ 189674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 190674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 191674ae819SStefano Zampini { 192674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 193674ae819SStefano Zampini PetscErrorCode ierr; 1941e6b0712SBarry Smith 195674ae819SStefano Zampini PetscFunctionBegin; 196674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 197674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 198674ae819SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 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__ 3683b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 3693b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 3703b03a366Sstefano_zampini { 3713b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3723b03a366Sstefano_zampini PetscErrorCode ierr; 3733b03a366Sstefano_zampini 3743b03a366Sstefano_zampini PetscFunctionBegin; 375785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 376785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 3773b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 37836e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 37936e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 380fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 3813b03a366Sstefano_zampini PetscFunctionReturn(0); 3823b03a366Sstefano_zampini } 3831e6b0712SBarry Smith 3843b03a366Sstefano_zampini #undef __FUNCT__ 3853b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 3863b03a366Sstefano_zampini /*@ 38728509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 3883b03a366Sstefano_zampini 389785d1243SStefano Zampini Collective 3903b03a366Sstefano_zampini 3913b03a366Sstefano_zampini Input Parameters: 3923b03a366Sstefano_zampini + pc - the preconditioning context 393785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 3943b03a366Sstefano_zampini 3953b03a366Sstefano_zampini Level: intermediate 3963b03a366Sstefano_zampini 3970f202f7eSStefano Zampini Notes: 3980f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 3993b03a366Sstefano_zampini 4000f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 4013b03a366Sstefano_zampini @*/ 4023b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 4033b03a366Sstefano_zampini { 4043b03a366Sstefano_zampini PetscErrorCode ierr; 4053b03a366Sstefano_zampini 4063b03a366Sstefano_zampini PetscFunctionBegin; 4073b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 408674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 409785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4103b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4113b03a366Sstefano_zampini PetscFunctionReturn(0); 4123b03a366Sstefano_zampini } 4133b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4141e6b0712SBarry Smith 4153b03a366Sstefano_zampini #undef __FUNCT__ 41682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 41782ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4180c7d97c5SJed Brown { 4190c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4200c7d97c5SJed Brown PetscErrorCode ierr; 4210c7d97c5SJed Brown 4220c7d97c5SJed Brown PetscFunctionBegin; 423785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 424785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4250c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 4260c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 427785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 4287d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4290c7d97c5SJed Brown PetscFunctionReturn(0); 4300c7d97c5SJed Brown } 4310c7d97c5SJed Brown 4320c7d97c5SJed Brown #undef __FUNCT__ 43382ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4349c0446d6SStefano Zampini /*@ 43582ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4369c0446d6SStefano Zampini 437785d1243SStefano Zampini Collective 43853cdbc3dSStefano Zampini 43953cdbc3dSStefano Zampini Input Parameters: 44053cdbc3dSStefano Zampini + pc - the preconditioning context 44182ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 44253cdbc3dSStefano Zampini 44353cdbc3dSStefano Zampini Level: intermediate 44453cdbc3dSStefano Zampini 4459c0446d6SStefano Zampini Notes: 44653cdbc3dSStefano Zampini 4470f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 44853cdbc3dSStefano Zampini @*/ 44982ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 4500c7d97c5SJed Brown { 4510c7d97c5SJed Brown PetscErrorCode ierr; 4520c7d97c5SJed Brown 4530c7d97c5SJed Brown PetscFunctionBegin; 4540c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4550c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 45682ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 45782ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4580c7d97c5SJed Brown PetscFunctionReturn(0); 4590c7d97c5SJed Brown } 4600c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 4610c7d97c5SJed Brown 4620c7d97c5SJed Brown #undef __FUNCT__ 4630c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 46453cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 4650c7d97c5SJed Brown { 4660c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 46753cdbc3dSStefano Zampini PetscErrorCode ierr; 4680c7d97c5SJed Brown 4690c7d97c5SJed Brown PetscFunctionBegin; 470785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 471785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 47253cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 47336e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 47436e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 475fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4760c7d97c5SJed Brown PetscFunctionReturn(0); 4770c7d97c5SJed Brown } 4781e6b0712SBarry Smith 4790c7d97c5SJed Brown #undef __FUNCT__ 4800c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 48157527edcSJed Brown /*@ 48228509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 48357527edcSJed Brown 484785d1243SStefano Zampini Collective 48557527edcSJed Brown 48657527edcSJed Brown Input Parameters: 48757527edcSJed Brown + pc - the preconditioning context 488785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 48957527edcSJed Brown 49057527edcSJed Brown Level: intermediate 49157527edcSJed Brown 4920f202f7eSStefano Zampini Notes: 4930f202f7eSStefano Zampini Any process can list any global node 49457527edcSJed Brown 4950f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 49657527edcSJed Brown @*/ 49753cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 4980c7d97c5SJed Brown { 4990c7d97c5SJed Brown PetscErrorCode ierr; 5000c7d97c5SJed Brown 5010c7d97c5SJed Brown PetscFunctionBegin; 5020c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 503674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 504785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 50553cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 50653cdbc3dSStefano Zampini PetscFunctionReturn(0); 50753cdbc3dSStefano Zampini } 50853cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5091e6b0712SBarry Smith 51053cdbc3dSStefano Zampini #undef __FUNCT__ 51182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 51282ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5130c7d97c5SJed Brown { 5140c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5150c7d97c5SJed Brown PetscErrorCode ierr; 5160c7d97c5SJed Brown 5170c7d97c5SJed Brown PetscFunctionBegin; 518785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 519785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5200c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 5210c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 522785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 5237d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5240c7d97c5SJed Brown PetscFunctionReturn(0); 5250c7d97c5SJed Brown } 5260c7d97c5SJed Brown 5270c7d97c5SJed Brown #undef __FUNCT__ 52882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5290c7d97c5SJed Brown /*@ 53082ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 5310c7d97c5SJed Brown 532785d1243SStefano Zampini Collective 5330c7d97c5SJed Brown 5340c7d97c5SJed Brown Input Parameters: 5350c7d97c5SJed Brown + pc - the preconditioning context 53682ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 5370c7d97c5SJed Brown 5380c7d97c5SJed Brown Level: intermediate 5390c7d97c5SJed Brown 5400c7d97c5SJed Brown Notes: 5410c7d97c5SJed Brown 5420f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 5430c7d97c5SJed Brown @*/ 54482ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 5450c7d97c5SJed Brown { 5460c7d97c5SJed Brown PetscErrorCode ierr; 5470c7d97c5SJed Brown 5480c7d97c5SJed Brown PetscFunctionBegin; 5490c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5500c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 55182ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 55282ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 55353cdbc3dSStefano Zampini PetscFunctionReturn(0); 55453cdbc3dSStefano Zampini } 55553cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 55653cdbc3dSStefano Zampini 55753cdbc3dSStefano Zampini #undef __FUNCT__ 558da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 559da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 560da1bb401SStefano Zampini { 561da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 562da1bb401SStefano Zampini 563da1bb401SStefano Zampini PetscFunctionBegin; 564da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 565da1bb401SStefano Zampini PetscFunctionReturn(0); 566da1bb401SStefano Zampini } 5671e6b0712SBarry Smith 568da1bb401SStefano Zampini #undef __FUNCT__ 569da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 570da1bb401SStefano Zampini /*@ 571785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 572da1bb401SStefano Zampini 573785d1243SStefano Zampini Collective 574785d1243SStefano Zampini 575785d1243SStefano Zampini Input Parameters: 576785d1243SStefano Zampini . pc - the preconditioning context 577785d1243SStefano Zampini 578785d1243SStefano Zampini Output Parameters: 579785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 580785d1243SStefano Zampini 581785d1243SStefano Zampini Level: intermediate 582785d1243SStefano Zampini 5830f202f7eSStefano Zampini Notes: 5840f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 585785d1243SStefano Zampini 586785d1243SStefano Zampini .seealso: PCBDDC 587785d1243SStefano Zampini @*/ 588785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 589785d1243SStefano Zampini { 590785d1243SStefano Zampini PetscErrorCode ierr; 591785d1243SStefano Zampini 592785d1243SStefano Zampini PetscFunctionBegin; 593785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 594785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 595785d1243SStefano Zampini PetscFunctionReturn(0); 596785d1243SStefano Zampini } 597785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 598785d1243SStefano Zampini 599785d1243SStefano Zampini #undef __FUNCT__ 600785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 601785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 602785d1243SStefano Zampini { 603785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 604785d1243SStefano Zampini 605785d1243SStefano Zampini PetscFunctionBegin; 606785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 607785d1243SStefano Zampini PetscFunctionReturn(0); 608785d1243SStefano Zampini } 609785d1243SStefano Zampini 610785d1243SStefano Zampini #undef __FUNCT__ 61182ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 612da1bb401SStefano Zampini /*@ 61382ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 614da1bb401SStefano Zampini 615785d1243SStefano Zampini Collective 616da1bb401SStefano Zampini 617da1bb401SStefano Zampini Input Parameters: 61828509bceSStefano Zampini . pc - the preconditioning context 619da1bb401SStefano Zampini 620da1bb401SStefano Zampini Output Parameters: 62128509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 622da1bb401SStefano Zampini 623da1bb401SStefano Zampini Level: intermediate 624da1bb401SStefano Zampini 625da1bb401SStefano Zampini Notes: 6260f202f7eSStefano 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). 6270f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 628da1bb401SStefano Zampini 629da1bb401SStefano Zampini .seealso: PCBDDC 630da1bb401SStefano Zampini @*/ 63182ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 632da1bb401SStefano Zampini { 633da1bb401SStefano Zampini PetscErrorCode ierr; 634da1bb401SStefano Zampini 635da1bb401SStefano Zampini PetscFunctionBegin; 636da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 63782ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 638da1bb401SStefano Zampini PetscFunctionReturn(0); 639da1bb401SStefano Zampini } 640da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 6411e6b0712SBarry Smith 642da1bb401SStefano Zampini #undef __FUNCT__ 64353cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 64453cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 64553cdbc3dSStefano Zampini { 64653cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 64753cdbc3dSStefano Zampini 64853cdbc3dSStefano Zampini PetscFunctionBegin; 64953cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 65053cdbc3dSStefano Zampini PetscFunctionReturn(0); 65153cdbc3dSStefano Zampini } 6521e6b0712SBarry Smith 65353cdbc3dSStefano Zampini #undef __FUNCT__ 65453cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 65553cdbc3dSStefano Zampini /*@ 656785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 65753cdbc3dSStefano Zampini 658785d1243SStefano Zampini Collective 659785d1243SStefano Zampini 660785d1243SStefano Zampini Input Parameters: 661785d1243SStefano Zampini . pc - the preconditioning context 662785d1243SStefano Zampini 663785d1243SStefano Zampini Output Parameters: 664785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 665785d1243SStefano Zampini 666785d1243SStefano Zampini Level: intermediate 667785d1243SStefano Zampini 6680f202f7eSStefano Zampini Notes: 6690f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 670785d1243SStefano Zampini 671785d1243SStefano Zampini .seealso: PCBDDC 672785d1243SStefano Zampini @*/ 673785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 674785d1243SStefano Zampini { 675785d1243SStefano Zampini PetscErrorCode ierr; 676785d1243SStefano Zampini 677785d1243SStefano Zampini PetscFunctionBegin; 678785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 679785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 680785d1243SStefano Zampini PetscFunctionReturn(0); 681785d1243SStefano Zampini } 682785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 683785d1243SStefano Zampini 684785d1243SStefano Zampini #undef __FUNCT__ 685785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 686785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 687785d1243SStefano Zampini { 688785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 689785d1243SStefano Zampini 690785d1243SStefano Zampini PetscFunctionBegin; 691785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 692785d1243SStefano Zampini PetscFunctionReturn(0); 693785d1243SStefano Zampini } 694785d1243SStefano Zampini 695785d1243SStefano Zampini #undef __FUNCT__ 69682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 69753cdbc3dSStefano Zampini /*@ 69882ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 69953cdbc3dSStefano Zampini 700785d1243SStefano Zampini Collective 70153cdbc3dSStefano Zampini 70253cdbc3dSStefano Zampini Input Parameters: 70328509bceSStefano Zampini . pc - the preconditioning context 70453cdbc3dSStefano Zampini 70553cdbc3dSStefano Zampini Output Parameters: 70628509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 70753cdbc3dSStefano Zampini 70853cdbc3dSStefano Zampini Level: intermediate 70953cdbc3dSStefano Zampini 71053cdbc3dSStefano Zampini Notes: 7110f202f7eSStefano 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). 7120f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 71353cdbc3dSStefano Zampini 71453cdbc3dSStefano Zampini .seealso: PCBDDC 71553cdbc3dSStefano Zampini @*/ 71682ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 71753cdbc3dSStefano Zampini { 71853cdbc3dSStefano Zampini PetscErrorCode ierr; 71953cdbc3dSStefano Zampini 72053cdbc3dSStefano Zampini PetscFunctionBegin; 72153cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 72282ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7230c7d97c5SJed Brown PetscFunctionReturn(0); 7240c7d97c5SJed Brown } 72536e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7261e6b0712SBarry Smith 72736e030ebSStefano Zampini #undef __FUNCT__ 728da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7291a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 73036e030ebSStefano Zampini { 73136e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 732da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 733da1bb401SStefano Zampini PetscErrorCode ierr; 73436e030ebSStefano Zampini 73536e030ebSStefano Zampini PetscFunctionBegin; 736674ae819SStefano Zampini /* free old CSR */ 737674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 738fb223d50SStefano Zampini /* TODO: PCBDDCGraphSetAdjacency */ 739674ae819SStefano Zampini /* get CSR into graph structure */ 740da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 741854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 742785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 743674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 744674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 745da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 7461a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 7471a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 748674ae819SStefano Zampini } 749575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 75036e030ebSStefano Zampini PetscFunctionReturn(0); 75136e030ebSStefano Zampini } 7521e6b0712SBarry Smith 75336e030ebSStefano Zampini #undef __FUNCT__ 754da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 75536e030ebSStefano Zampini /*@ 7560f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 75736e030ebSStefano Zampini 75836e030ebSStefano Zampini Not collective 75936e030ebSStefano Zampini 76036e030ebSStefano Zampini Input Parameters: 76136e030ebSStefano Zampini + pc - the preconditioning context 7620f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 76328509bceSStefano Zampini . xadj, adjncy - the CSR graph 76428509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 76536e030ebSStefano Zampini 76636e030ebSStefano Zampini Level: intermediate 76736e030ebSStefano Zampini 76836e030ebSStefano Zampini Notes: 76936e030ebSStefano Zampini 77028509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 77136e030ebSStefano Zampini @*/ 7721a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 77336e030ebSStefano Zampini { 774575ad6abSStefano Zampini void (*f)(void) = 0; 77536e030ebSStefano Zampini PetscErrorCode ierr; 77636e030ebSStefano Zampini 77736e030ebSStefano Zampini PetscFunctionBegin; 77836e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 779674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 780d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 7816c4ed002SBarry Smith if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode); 7821a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 783575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 784575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 785575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 786575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 787575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 788da1bb401SStefano Zampini } 78936e030ebSStefano Zampini PetscFunctionReturn(0); 79036e030ebSStefano Zampini } 7919c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 7921e6b0712SBarry Smith 7939c0446d6SStefano Zampini #undef __FUNCT__ 79463602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 79563602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 79663602bcaSStefano Zampini { 79763602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79863602bcaSStefano Zampini PetscInt i; 79963602bcaSStefano Zampini PetscErrorCode ierr; 80063602bcaSStefano Zampini 80163602bcaSStefano Zampini PetscFunctionBegin; 80263602bcaSStefano Zampini /* Destroy ISes if they were already set */ 80363602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 80463602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 80563602bcaSStefano Zampini } 80663602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 80763602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 80863602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 80963602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 81063602bcaSStefano Zampini } 81163602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 81263602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 81363602bcaSStefano Zampini /* allocate space then set */ 814d02579f5SStefano Zampini if (n_is) { 815d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 816d02579f5SStefano Zampini } 81763602bcaSStefano Zampini for (i=0;i<n_is;i++) { 81863602bcaSStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 81963602bcaSStefano Zampini pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 82063602bcaSStefano Zampini } 82163602bcaSStefano Zampini pcbddc->n_ISForDofsLocal=n_is; 82263602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8231a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 82463602bcaSStefano Zampini PetscFunctionReturn(0); 82563602bcaSStefano Zampini } 82663602bcaSStefano Zampini 82763602bcaSStefano Zampini #undef __FUNCT__ 82863602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 82963602bcaSStefano Zampini /*@ 83063602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 83163602bcaSStefano Zampini 83263602bcaSStefano Zampini Collective 83363602bcaSStefano Zampini 83463602bcaSStefano Zampini Input Parameters: 83563602bcaSStefano Zampini + pc - the preconditioning context 8360f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8370f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 83863602bcaSStefano Zampini 83963602bcaSStefano Zampini Level: intermediate 84063602bcaSStefano Zampini 8410f202f7eSStefano Zampini Notes: 8420f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 84363602bcaSStefano Zampini 84463602bcaSStefano Zampini .seealso: PCBDDC 84563602bcaSStefano Zampini @*/ 84663602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 84763602bcaSStefano Zampini { 84863602bcaSStefano Zampini PetscInt i; 84963602bcaSStefano Zampini PetscErrorCode ierr; 85063602bcaSStefano Zampini 85163602bcaSStefano Zampini PetscFunctionBegin; 85263602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 85363602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 85463602bcaSStefano Zampini for (i=0;i<n_is;i++) { 85563602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 85663602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 85763602bcaSStefano Zampini } 858e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 85963602bcaSStefano Zampini PetscFunctionReturn(0); 86063602bcaSStefano Zampini } 86163602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 86263602bcaSStefano Zampini 86363602bcaSStefano Zampini #undef __FUNCT__ 8649c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 8659c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 8669c0446d6SStefano Zampini { 8679c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 8689c0446d6SStefano Zampini PetscInt i; 8699c0446d6SStefano Zampini PetscErrorCode ierr; 8709c0446d6SStefano Zampini 8719c0446d6SStefano Zampini PetscFunctionBegin; 872da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 8739c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 8749c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 8759c0446d6SStefano Zampini } 876d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 87763602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 87863602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 87963602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 88063602bcaSStefano Zampini } 88163602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 88263602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 883da1bb401SStefano Zampini /* allocate space then set */ 884d02579f5SStefano Zampini if (n_is) { 885785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 886d02579f5SStefano Zampini } 8879c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 888da1bb401SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 889da1bb401SStefano Zampini pcbddc->ISForDofs[i]=ISForDofs[i]; 8909c0446d6SStefano Zampini } 8919c0446d6SStefano Zampini pcbddc->n_ISForDofs=n_is; 89263602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8931a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 8949c0446d6SStefano Zampini PetscFunctionReturn(0); 8959c0446d6SStefano Zampini } 8961e6b0712SBarry Smith 8979c0446d6SStefano Zampini #undef __FUNCT__ 8989c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 8999c0446d6SStefano Zampini /*@ 90063602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 9019c0446d6SStefano Zampini 90263602bcaSStefano Zampini Collective 9039c0446d6SStefano Zampini 9049c0446d6SStefano Zampini Input Parameters: 9059c0446d6SStefano Zampini + pc - the preconditioning context 9060f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9070f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 9089c0446d6SStefano Zampini 9099c0446d6SStefano Zampini Level: intermediate 9109c0446d6SStefano Zampini 9110f202f7eSStefano Zampini Notes: 9120f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 9139c0446d6SStefano Zampini 9149c0446d6SStefano Zampini .seealso: PCBDDC 9159c0446d6SStefano Zampini @*/ 9169c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 9179c0446d6SStefano Zampini { 9182b510759SStefano Zampini PetscInt i; 9199c0446d6SStefano Zampini PetscErrorCode ierr; 9209c0446d6SStefano Zampini 9219c0446d6SStefano Zampini PetscFunctionBegin; 9229c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 92363602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 9242b510759SStefano Zampini for (i=0;i<n_is;i++) { 92563602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 92663602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 9272b510759SStefano Zampini } 9289c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 9299c0446d6SStefano Zampini PetscFunctionReturn(0); 9309c0446d6SStefano Zampini } 931906d46d4SStefano Zampini 932da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 933534831adSStefano Zampini #undef __FUNCT__ 934534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 935534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 936534831adSStefano Zampini /* 937534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 938534831adSStefano Zampini guess if a transformation of basis approach has been selected. 9399c0446d6SStefano Zampini 940534831adSStefano Zampini Input Parameter: 941534831adSStefano Zampini + pc - the preconditioner contex 942534831adSStefano Zampini 943534831adSStefano Zampini Application Interface Routine: PCPreSolve() 944534831adSStefano Zampini 945534831adSStefano Zampini Notes: 946534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 947534831adSStefano Zampini the user, but instead is called by KSPSolve(). 948534831adSStefano Zampini */ 949534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 950534831adSStefano Zampini { 951534831adSStefano Zampini PetscErrorCode ierr; 952534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 953534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 9543972b0daSStefano Zampini Vec used_vec; 9558d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 956534831adSStefano Zampini 957534831adSStefano Zampini PetscFunctionBegin; 95885c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 95985c4d303SStefano Zampini if (ksp) { 96085c4d303SStefano Zampini PetscBool iscg; 96185c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 96285c4d303SStefano Zampini if (!iscg) { 96385c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 96485c4d303SStefano Zampini } 96585c4d303SStefano Zampini } 966*c7017625SStefano Zampini 96785c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 96862a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 96962a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 97062a6ff1dSStefano Zampini } 97162a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 97262a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 97362a6ff1dSStefano Zampini } 9748d00608fSStefano Zampini 9753972b0daSStefano Zampini if (x) { 9763972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 9773972b0daSStefano Zampini used_vec = x; 9788d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 9793972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 9803972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 9813972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9823972b0daSStefano Zampini } 9838efcfb23SStefano Zampini 9848efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 9853972b0daSStefano Zampini if (ksp) { 986a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 9878efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 9888efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 9893972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9903972b0daSStefano Zampini } 9913972b0daSStefano Zampini } 9923308cffdSStefano Zampini 9938d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 9943972b0daSStefano Zampini 9953972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 996a07ea27aSStefano Zampini if (rhs) { 9973975b054SStefano Zampini IS dirIS = NULL; 9983975b054SStefano Zampini 999a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 10003975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 10013975b054SStefano Zampini if (dirIS) { 1002906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1003785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 10042b095fd8SStefano Zampini PetscScalar *array_x; 10052b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1006785d1243SStefano Zampini 10073972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 10083972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1009e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1010e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1011e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1012e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 101382ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 10143972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 10152b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10163972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10172fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 10183972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10192b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10203972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1021e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1022e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10238d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10241b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 10258efcfb23SStefano Zampini } 1026a07ea27aSStefano Zampini } 1027b76ba322SStefano Zampini 10288efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 10298d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 10308d00608fSStefano Zampini /* store the original rhs */ 10318d00608fSStefano Zampini if (copy_rhs) { 10328d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10338d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10348d00608fSStefano Zampini } 10358d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10363972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10373972b0daSStefano Zampini ierr = MatMultAdd(pc->pmat,used_vec,rhs,rhs);CHKERRQ(ierr); 10383972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10398efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 10407acc28cbSStefano Zampini if (ksp) { 10417acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 10427acc28cbSStefano Zampini } 10433308cffdSStefano Zampini } 10448efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1045b76ba322SStefano Zampini 1046b76ba322SStefano Zampini /* store partially computed solution and set initial guess */ 1047b097fa66SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 10488efcfb23SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 1049b76ba322SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1050b76ba322SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1051b76ba322SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 10528efcfb23SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10538efcfb23SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1054a0cb1b98SStefano Zampini if (ksp) { 1055b76ba322SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 1056b76ba322SStefano Zampini } 1057b76ba322SStefano Zampini } 1058b76ba322SStefano Zampini 1059b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1060906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1061906d46d4SStefano Zampini 1062906d46d4SStefano Zampini /* get change ctx */ 1063906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1064906d46d4SStefano Zampini 1065906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 1066906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1067906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1068906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1069906d46d4SStefano Zampini 1070906d46d4SStefano Zampini /* change iteration matrix */ 1071906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1072906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1073906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 1074906d46d4SStefano Zampini 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 1081906d46d4SStefano Zampini /* change rhs */ 1082906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1083906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 10848d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 108592e3dcfbSStefano Zampini } 1086534831adSStefano Zampini PetscFunctionReturn(0); 1087534831adSStefano Zampini } 1088906d46d4SStefano Zampini 1089534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1090534831adSStefano Zampini #undef __FUNCT__ 1091534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1092534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1093534831adSStefano Zampini /* 1094534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1095534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1096534831adSStefano Zampini 1097534831adSStefano Zampini Input Parameter: 1098534831adSStefano Zampini + pc - the preconditioner contex 1099534831adSStefano Zampini 1100534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1101534831adSStefano Zampini 1102534831adSStefano Zampini Notes: 1103534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1104534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1105534831adSStefano Zampini */ 1106534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1107534831adSStefano Zampini { 1108534831adSStefano Zampini PetscErrorCode ierr; 1109534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1110534831adSStefano Zampini 1111534831adSStefano Zampini PetscFunctionBegin; 1112b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1113906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1114906d46d4SStefano Zampini 1115906d46d4SStefano Zampini /* get change ctx */ 1116906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1117906d46d4SStefano Zampini 1118906d46d4SStefano Zampini /* restore iteration matrix */ 1119906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1120906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1121906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 1122906d46d4SStefano Zampini 1123906d46d4SStefano Zampini /* get solution in original basis */ 1124906d46d4SStefano Zampini if (x) { 1125906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1126906d46d4SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1127906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 11283425bc38SStefano Zampini } 1129534831adSStefano Zampini } 1130906d46d4SStefano Zampini 11313972b0daSStefano Zampini /* add solution removed in presolve */ 11326bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 11333425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 11343425bc38SStefano Zampini } 1135906d46d4SStefano Zampini 1136fb223d50SStefano Zampini /* restore rhs to its original state */ 11378d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1138fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1139fb223d50SStefano Zampini } 11408d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11418efcfb23SStefano Zampini 11428efcfb23SStefano Zampini /* restore ksp guess state */ 11438efcfb23SStefano Zampini if (ksp) { 11448efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 11458efcfb23SStefano Zampini } 1146534831adSStefano Zampini PetscFunctionReturn(0); 1147534831adSStefano Zampini } 1148534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 114953cdbc3dSStefano Zampini #undef __FUNCT__ 115053cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 11510c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 11520c7d97c5SJed Brown /* 11530c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 11540c7d97c5SJed Brown by setting data structures and options. 11550c7d97c5SJed Brown 11560c7d97c5SJed Brown Input Parameter: 115753cdbc3dSStefano Zampini + pc - the preconditioner context 11580c7d97c5SJed Brown 11590c7d97c5SJed Brown Application Interface Routine: PCSetUp() 11600c7d97c5SJed Brown 11610c7d97c5SJed Brown Notes: 11620c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 11630c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 11640c7d97c5SJed Brown */ 116553cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 11660c7d97c5SJed Brown { 11670c7d97c5SJed Brown PetscErrorCode ierr; 11680c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 11695e8657edSStefano Zampini Mat_IS* matis; 117008122e43SStefano Zampini MatNullSpace nearnullspace; 117191e8d312SStefano Zampini PetscInt nrows,ncols; 117208122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 11738de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 11745e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 11750c7d97c5SJed Brown 11760c7d97c5SJed Brown PetscFunctionBegin; 11775e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 11786c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 117991e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 11806c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 11815e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1182f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 11833b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1184fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1185f4ddd8eeSStefano Zampini /* split work */ 1186674ae819SStefano Zampini if (pc->setupcalled) { 1187d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1188674ae819SStefano Zampini computetopography = PETSC_FALSE; 1189674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1190674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1191674ae819SStefano Zampini computetopography = PETSC_TRUE; 1192674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1193674ae819SStefano Zampini } 1194674ae819SStefano Zampini } else { 1195674ae819SStefano Zampini computetopography = PETSC_TRUE; 1196674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1197674ae819SStefano Zampini } 1198fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1199fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1200fb180af4SStefano Zampini } 12018de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 12026c4ed002SBarry Smith if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 12035a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1204862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1205862806e4SStefano Zampini 12065a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 12076c4ed002SBarry Smith if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) 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"); 12086c4ed002SBarry Smith 1209f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 121070cf5478SStefano Zampini if (pcbddc->dbg_flag) { 121170cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 121258a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 12131575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1214f4ddd8eeSStefano Zampini } 121558a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1216f4ddd8eeSStefano Zampini } 1217f4ddd8eeSStefano Zampini 12185e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 12195e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 12205e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 12215e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 12225e8657edSStefano Zampini } else { 1223b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 12245e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 12255e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1226674ae819SStefano Zampini } 1227f4ddd8eeSStefano Zampini 1228e496cd5dSStefano Zampini /* workaround for reals */ 1229e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX) 12303301b35fSStefano Zampini if (matis->A->symmetric_set) { 12313301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 12323301b35fSStefano Zampini } 1233e496cd5dSStefano Zampini #endif 1234e496cd5dSStefano Zampini 12355e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 12365e8657edSStefano Zampini { 12375e8657edSStefano Zampini Mat temp_mat; 12385e8657edSStefano Zampini 12395e8657edSStefano Zampini temp_mat = matis->A; 12405e8657edSStefano Zampini matis->A = pcbddc->local_mat; 12415e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 12425e8657edSStefano Zampini pcbddc->local_mat = matis->A; 12435e8657edSStefano Zampini matis->A = temp_mat; 12445e8657edSStefano Zampini } 1245684f6988SStefano Zampini 1246b96c3477SStefano Zampini /* Analyze interface and setup sub_schurs data */ 1247674ae819SStefano Zampini if (computetopography) { 1248674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 12498de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1250674ae819SStefano Zampini } 1251fb8d54d4SStefano Zampini 1252b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1253684f6988SStefano Zampini if (computesolvers) { 1254d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1255d5574798SStefano Zampini 1256d5574798SStefano Zampini if (computesubschurs && computetopography) { 125708122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1258b1b3d7a2SStefano Zampini } 1259d5574798SStefano Zampini if (sub_schurs->use_mumps) { 12602070dbb6SStefano Zampini if (computesubschurs) { 126108122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 12622070dbb6SStefano Zampini } 1263d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1264d5574798SStefano Zampini } else { 1265d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 12662070dbb6SStefano Zampini if (computesubschurs) { 1267d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1268d5574798SStefano Zampini } 12692070dbb6SStefano Zampini } 127008122e43SStefano Zampini if (pcbddc->adaptive_selection) { 127108122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 12728de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1273b7eb3628SStefano Zampini } 1274b1b3d7a2SStefano Zampini } 1275684f6988SStefano Zampini 1276f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1277fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1278f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1279f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1280f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1281f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1282f4ddd8eeSStefano Zampini } else { 1283f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1284f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1285f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1286165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1287f4ddd8eeSStefano Zampini PetscInt i; 1288165b64e2SStefano Zampini const Vec *nearnullvecs; 1289165b64e2SStefano Zampini PetscObjectState state; 1290165b64e2SStefano Zampini PetscInt nnsp_size; 1291165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1292f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1293f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1294165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1295f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1296f4ddd8eeSStefano Zampini break; 1297f4ddd8eeSStefano Zampini } 1298f4ddd8eeSStefano Zampini } 1299f4ddd8eeSStefano Zampini } 1300f4ddd8eeSStefano Zampini } 1301f4ddd8eeSStefano Zampini } else { 1302f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1303f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1304f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1305f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1306f4ddd8eeSStefano Zampini } 1307f4ddd8eeSStefano Zampini } 1308f4ddd8eeSStefano Zampini 1309f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1310727cdba6SStefano Zampini /* reset primal space flags */ 1311f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1312727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 13138de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1314727cdba6SStefano Zampini /* It also sets the primal space flags */ 1315674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1316e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1317f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 13183975b054SStefano Zampini } 13195e8657edSStefano Zampini 13203975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 13215e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 13225e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 13235e8657edSStefano Zampini 13245e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 13255e8657edSStefano Zampini /* get submatrices */ 13265e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 13275e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 13285e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 13295e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 13305e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 13315e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 13323975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 13333975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 13345e8657edSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix) { 1335b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 13365e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 13375e8657edSStefano Zampini pcbddc->local_mat = matis->A; 13385e8657edSStefano Zampini } 1339b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 134099cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1341b96c3477SStefano Zampini /* SetUp Scaling operator */ 1342674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 13430c7d97c5SJed Brown } 134470cf5478SStefano Zampini 134558a03d70SStefano Zampini if (pcbddc->dbg_flag) { 134658a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 13472b510759SStefano Zampini } 13480c7d97c5SJed Brown PetscFunctionReturn(0); 13490c7d97c5SJed Brown } 13500c7d97c5SJed Brown 13510c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13520c7d97c5SJed Brown /* 135350efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 13540c7d97c5SJed Brown 13550c7d97c5SJed Brown Input Parameters: 13560f202f7eSStefano Zampini + pc - the preconditioner context 13570f202f7eSStefano Zampini - r - input vector (global) 13580c7d97c5SJed Brown 13590c7d97c5SJed Brown Output Parameter: 13600c7d97c5SJed Brown . z - output vector (global) 13610c7d97c5SJed Brown 13620c7d97c5SJed Brown Application Interface Routine: PCApply() 13630c7d97c5SJed Brown */ 13640c7d97c5SJed Brown #undef __FUNCT__ 13650c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 136653cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 13670c7d97c5SJed Brown { 13680c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 13690c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1370b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 13710c7d97c5SJed Brown PetscErrorCode ierr; 13723b03a366Sstefano_zampini const PetscScalar one = 1.0; 13733b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 13742617d88aSStefano Zampini const PetscScalar zero = 0.0; 13750c7d97c5SJed Brown 13760c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 13770c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1378b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 13790c7d97c5SJed Brown 13800c7d97c5SJed Brown PetscFunctionBegin; 138185c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1382b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 13830c7d97c5SJed Brown /* First Dirichlet solve */ 13840c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13850c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13860c7d97c5SJed Brown /* 13870c7d97c5SJed Brown Assembling right hand side for BDDC operator 1388b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1389674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 13900c7d97c5SJed Brown */ 1391b097fa66SStefano Zampini if (n_D) { 1392b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13930c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 13948eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1395b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1396b097fa66SStefano Zampini } else { 1397b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1398b097fa66SStefano Zampini } 13990c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14000c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1401674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1402b76ba322SStefano Zampini } else { 1403b097fa66SStefano Zampini if (pcbddc->switch_static) { 14040bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1405b097fa66SStefano Zampini } 1406674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1407b76ba322SStefano Zampini } 1408b76ba322SStefano Zampini 14092617d88aSStefano Zampini /* Apply interface preconditioner 14102617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1411dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 14122617d88aSStefano Zampini 1413674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1414674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 14150c7d97c5SJed Brown 14163b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 14170c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14180c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1419b097fa66SStefano Zampini if (n_B) { 14200c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 14218eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1422b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1423b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1424b097fa66SStefano Zampini } 1425df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1426b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1427b097fa66SStefano Zampini if (pcbddc->switch_static) { 1428b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1429b097fa66SStefano Zampini } else { 1430b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1431b097fa66SStefano Zampini } 14320c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14330c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1434b097fa66SStefano Zampini } else { 1435b097fa66SStefano Zampini if (pcbddc->switch_static) { 1436b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1437b097fa66SStefano Zampini } else { 1438b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1439b097fa66SStefano Zampini } 1440b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1441b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1442b097fa66SStefano Zampini } 14430c7d97c5SJed Brown PetscFunctionReturn(0); 14440c7d97c5SJed Brown } 144550efa1b5SStefano Zampini 144650efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 144750efa1b5SStefano Zampini /* 144850efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 144950efa1b5SStefano Zampini 145050efa1b5SStefano Zampini Input Parameters: 14510f202f7eSStefano Zampini + pc - the preconditioner context 14520f202f7eSStefano Zampini - r - input vector (global) 145350efa1b5SStefano Zampini 145450efa1b5SStefano Zampini Output Parameter: 145550efa1b5SStefano Zampini . z - output vector (global) 145650efa1b5SStefano Zampini 145750efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 145850efa1b5SStefano Zampini */ 145950efa1b5SStefano Zampini #undef __FUNCT__ 146050efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 146150efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 146250efa1b5SStefano Zampini { 146350efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 146450efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1465b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 146650efa1b5SStefano Zampini PetscErrorCode ierr; 146750efa1b5SStefano Zampini const PetscScalar one = 1.0; 146850efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 146950efa1b5SStefano Zampini const PetscScalar zero = 0.0; 147050efa1b5SStefano Zampini 147150efa1b5SStefano Zampini PetscFunctionBegin; 147250efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1473b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 147450efa1b5SStefano Zampini /* First Dirichlet solve */ 147550efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 147650efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 147750efa1b5SStefano Zampini /* 147850efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1479b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 148050efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 148150efa1b5SStefano Zampini */ 1482b097fa66SStefano Zampini if (n_D) { 1483b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 148450efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 148550efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1486b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1487b097fa66SStefano Zampini } else { 1488b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1489b097fa66SStefano Zampini } 149050efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 149150efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 149250efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 149350efa1b5SStefano Zampini } else { 1494b097fa66SStefano Zampini if (pcbddc->switch_static) { 149550efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1496b097fa66SStefano Zampini } 149750efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 149850efa1b5SStefano Zampini } 149950efa1b5SStefano Zampini 150050efa1b5SStefano Zampini /* Apply interface preconditioner 150150efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1502dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 150350efa1b5SStefano Zampini 150450efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 150550efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 150650efa1b5SStefano Zampini 150750efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 150850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 150950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1510b097fa66SStefano Zampini if (n_B) { 151150efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 151250efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1513b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1514b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1515b097fa66SStefano Zampini } 1516b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1517b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1518b097fa66SStefano Zampini if (pcbddc->switch_static) { 1519b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1520b097fa66SStefano Zampini } else { 1521b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1522b097fa66SStefano Zampini } 152350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 152450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1525b097fa66SStefano Zampini } else { 1526b097fa66SStefano Zampini if (pcbddc->switch_static) { 1527b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1528b097fa66SStefano Zampini } else { 1529b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1530b097fa66SStefano Zampini } 1531b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1532b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1533b097fa66SStefano Zampini } 153450efa1b5SStefano Zampini PetscFunctionReturn(0); 153550efa1b5SStefano Zampini } 1536da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1537674ae819SStefano Zampini 1538da1bb401SStefano Zampini #undef __FUNCT__ 1539da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1540da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1541da1bb401SStefano Zampini { 1542da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1543da1bb401SStefano Zampini PetscErrorCode ierr; 1544da1bb401SStefano Zampini 1545da1bb401SStefano Zampini PetscFunctionBegin; 1546da1bb401SStefano Zampini /* free data created by PCIS */ 1547da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 1548674ae819SStefano Zampini /* free BDDC custom data */ 1549674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1550674ae819SStefano Zampini /* destroy objects related to topography */ 1551674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1552674ae819SStefano Zampini /* free allocated graph structure */ 1553da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1554b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1555b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 155634a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1557674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 155834a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1559674ae819SStefano Zampini /* free solvers stuff */ 1560674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 156162a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 156262a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 156362a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1564906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 1565906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 1566906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1567906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1568906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1569906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1570906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1571906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1572906d46d4SStefano Zampini } 1573906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 15743425bc38SStefano Zampini /* remove functions */ 1575906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1576674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 1577bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 15782b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1579b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 15802b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1581bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 158282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1583bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 158482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1585bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 158682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1587bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1588785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1589bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 159063602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1591bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1592bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1593bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1594bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1595674ae819SStefano Zampini /* Free the private data structure */ 1596674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1597da1bb401SStefano Zampini PetscFunctionReturn(0); 1598da1bb401SStefano Zampini } 15993425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 16001e6b0712SBarry Smith 16013425bc38SStefano Zampini #undef __FUNCT__ 16023425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 16033425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16043425bc38SStefano Zampini { 1605674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1606c08af4c6SStefano Zampini Vec copy_standard_rhs; 16073425bc38SStefano Zampini PC_IS* pcis; 16083425bc38SStefano Zampini PC_BDDC* pcbddc; 16093425bc38SStefano Zampini PetscErrorCode ierr; 16100c7d97c5SJed Brown 16113425bc38SStefano Zampini PetscFunctionBegin; 16123425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 16133425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 16143425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 16153425bc38SStefano Zampini 1616c08af4c6SStefano Zampini /* 1617c08af4c6SStefano Zampini change of basis for physical rhs if needed 1618c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1619c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1620c08af4c6SStefano Zampini */ 1621c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1622c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1623c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 16243425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1625c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1626c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1627fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1628fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1629c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1630c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1631674ae819SStefano Zampini /* Apply partition of unity */ 16323425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1633c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 16348eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 16353425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 16363425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16373425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 16383425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1639c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1640c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1641c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1642c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1643c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1644c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16453425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 16463425bc38SStefano Zampini } 1647c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 16483425bc38SStefano Zampini /* BDDC rhs */ 16493425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 16508eeda7d8SStefano Zampini if (pcbddc->switch_static) { 16513425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16523425bc38SStefano Zampini } 16533425bc38SStefano Zampini /* apply BDDC */ 1654dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 16553425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 16563425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 16573425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 16583425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16593425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16603425bc38SStefano Zampini PetscFunctionReturn(0); 16613425bc38SStefano Zampini } 16621e6b0712SBarry Smith 16633425bc38SStefano Zampini #undef __FUNCT__ 16643425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 16653425bc38SStefano Zampini /*@ 16660f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 16673425bc38SStefano Zampini 16683425bc38SStefano Zampini Collective 16693425bc38SStefano Zampini 16703425bc38SStefano Zampini Input Parameters: 16710f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 16720f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 16733425bc38SStefano Zampini 16743425bc38SStefano Zampini Output Parameters: 16750f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 16763425bc38SStefano Zampini 16773425bc38SStefano Zampini Level: developer 16783425bc38SStefano Zampini 16793425bc38SStefano Zampini Notes: 16803425bc38SStefano Zampini 16810f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 16823425bc38SStefano Zampini @*/ 16833425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16843425bc38SStefano Zampini { 1685674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 16863425bc38SStefano Zampini PetscErrorCode ierr; 16873425bc38SStefano Zampini 16883425bc38SStefano Zampini PetscFunctionBegin; 16893425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1690163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 16913425bc38SStefano Zampini PetscFunctionReturn(0); 16923425bc38SStefano Zampini } 16933425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 16941e6b0712SBarry Smith 16953425bc38SStefano Zampini #undef __FUNCT__ 16963425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 16973425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 16983425bc38SStefano Zampini { 1699674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17003425bc38SStefano Zampini PC_IS* pcis; 17013425bc38SStefano Zampini PC_BDDC* pcbddc; 17023425bc38SStefano Zampini PetscErrorCode ierr; 17033425bc38SStefano Zampini 17043425bc38SStefano Zampini PetscFunctionBegin; 17053425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17063425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 17073425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 17083425bc38SStefano Zampini 17093425bc38SStefano Zampini /* apply B_delta^T */ 17103425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17113425bc38SStefano Zampini ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17123425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 17133425bc38SStefano Zampini /* compute rhs for BDDC application */ 17143425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 17158eeda7d8SStefano Zampini if (pcbddc->switch_static) { 17163425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17173425bc38SStefano Zampini } 17183425bc38SStefano Zampini /* apply BDDC */ 1719dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 17203425bc38SStefano Zampini /* put values into standard global vector */ 17213425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17223425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17238eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 17243425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 17253425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 17263425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 17273425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17283425bc38SStefano Zampini } 17293425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17303425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17313425bc38SStefano Zampini /* final change of basis if needed 17323425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 17333308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 17343425bc38SStefano Zampini PetscFunctionReturn(0); 17353425bc38SStefano Zampini } 17361e6b0712SBarry Smith 17373425bc38SStefano Zampini #undef __FUNCT__ 17383425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 17393425bc38SStefano Zampini /*@ 17400f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 17413425bc38SStefano Zampini 17423425bc38SStefano Zampini Collective 17433425bc38SStefano Zampini 17443425bc38SStefano Zampini Input Parameters: 17450f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 17460f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 17473425bc38SStefano Zampini 17483425bc38SStefano Zampini Output Parameters: 17490f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 17503425bc38SStefano Zampini 17513425bc38SStefano Zampini Level: developer 17523425bc38SStefano Zampini 17533425bc38SStefano Zampini Notes: 17543425bc38SStefano Zampini 17550f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 17563425bc38SStefano Zampini @*/ 17573425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 17583425bc38SStefano Zampini { 1759674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17603425bc38SStefano Zampini PetscErrorCode ierr; 17613425bc38SStefano Zampini 17623425bc38SStefano Zampini PetscFunctionBegin; 17633425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1764163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 17653425bc38SStefano Zampini PetscFunctionReturn(0); 17663425bc38SStefano Zampini } 17673425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 17681e6b0712SBarry Smith 1769f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 1770edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 1771f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 1772f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 1773edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 1774f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 1775674ae819SStefano Zampini 17763425bc38SStefano Zampini #undef __FUNCT__ 17773425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 17783425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 17793425bc38SStefano Zampini { 1780674ae819SStefano Zampini 1781674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 17823425bc38SStefano Zampini Mat newmat; 1783674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 17843425bc38SStefano Zampini PC newpc; 1785ce94432eSBarry Smith MPI_Comm comm; 17863425bc38SStefano Zampini PetscErrorCode ierr; 17873425bc38SStefano Zampini 17883425bc38SStefano Zampini PetscFunctionBegin; 1789ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 17903425bc38SStefano Zampini /* FETIDP linear matrix */ 17913425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 17923425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 17933425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 17943425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 1795edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 17963425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 17973425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 17983425bc38SStefano Zampini /* FETIDP preconditioner */ 17993425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 18003425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 18013425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 18023425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 18033425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 18043425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 1805edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 18063425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 180723ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 18083425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 18093425bc38SStefano Zampini /* return pointers for objects created */ 18103425bc38SStefano Zampini *fetidp_mat=newmat; 18113425bc38SStefano Zampini *fetidp_pc=newpc; 18123425bc38SStefano Zampini PetscFunctionReturn(0); 18133425bc38SStefano Zampini } 18141e6b0712SBarry Smith 18153425bc38SStefano Zampini #undef __FUNCT__ 18163425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 18173425bc38SStefano Zampini /*@ 18180f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 18193425bc38SStefano Zampini 18203425bc38SStefano Zampini Collective 18213425bc38SStefano Zampini 18223425bc38SStefano Zampini Input Parameters: 18230f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 182428509bceSStefano Zampini 182528509bceSStefano Zampini Output Parameters: 18260f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 18270f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 182828509bceSStefano Zampini 182928509bceSStefano Zampini Options Database Keys: 18300f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 18313425bc38SStefano Zampini 18323425bc38SStefano Zampini Level: developer 18333425bc38SStefano Zampini 18343425bc38SStefano Zampini Notes: 18350f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 18363425bc38SStefano Zampini 18370f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 18383425bc38SStefano Zampini @*/ 18393425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 18403425bc38SStefano Zampini { 18413425bc38SStefano Zampini PetscErrorCode ierr; 18423425bc38SStefano Zampini 18433425bc38SStefano Zampini PetscFunctionBegin; 18443425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 18453425bc38SStefano Zampini if (pc->setupcalled) { 1846516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 1847f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 18483425bc38SStefano Zampini PetscFunctionReturn(0); 18493425bc38SStefano Zampini } 18500c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 1851da1bb401SStefano Zampini /*MC 1852da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 18530c7d97c5SJed Brown 185428509bceSStefano Zampini An implementation of the BDDC preconditioner based on 185528509bceSStefano Zampini 185628509bceSStefano Zampini .vb 185728509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 185828509bceSStefano 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 185928509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 18600f202f7eSStefano 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 186128509bceSStefano Zampini .ve 186228509bceSStefano Zampini 186328509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 186428509bceSStefano Zampini 18650f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 186628509bceSStefano Zampini 186728509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 186828509bceSStefano Zampini 1869b6fdb6dfSStefano 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. 1870b6fdb6dfSStefano Zampini 1871*c7017625SStefano Zampini Approximate local solvers are automatically adapted (see [1]) if the user has attached a nullspace object to the subdomain matrices, and informed BDDC of using approximate solvers (via the command line). 187228509bceSStefano Zampini 18730f202f7eSStefano 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() 18740f202f7eSStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS() 187528509bceSStefano Zampini 18760f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 187728509bceSStefano Zampini 18780f202f7eSStefano 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. 18790f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 188028509bceSStefano Zampini 18810f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 188228509bceSStefano Zampini 18830f202f7eSStefano Zampini Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS is present. Future versions of the code will also consider using MKL_PARDISO or PASTIX. 188428509bceSStefano Zampini 18850f202f7eSStefano 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. 18860f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 18870f202f7eSStefano Zampini 18880f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 18890f202f7eSStefano Zampini 18900f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 18910f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 18920f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 18930f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 18940f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 18950f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 18960f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 189728509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 18980f202f7eSStefano 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) 18990f202f7eSStefano 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) 19000f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 19010f202f7eSStefano 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) 19020f202f7eSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS installed) 190328509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 190428509bceSStefano Zampini 190528509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 190628509bceSStefano Zampini .vb 190728509bceSStefano Zampini -pc_bddc_dirichlet_ 190828509bceSStefano Zampini -pc_bddc_neumann_ 190928509bceSStefano Zampini -pc_bddc_coarse_ 191028509bceSStefano Zampini .ve 19110f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 191228509bceSStefano Zampini 19130f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 191428509bceSStefano Zampini .vb 1915312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 1916312be037SStefano Zampini -pc_bddc_neumann_lN_ 1917312be037SStefano Zampini -pc_bddc_coarse_lN_ 191828509bceSStefano Zampini .ve 19190f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 19200f202f7eSStefano 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. 19210f202f7eSStefano Zampini .vb 19220f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 19230f202f7eSStefano Zampini .ve 19240f202f7eSStefano 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 1925da1bb401SStefano Zampini 1926da1bb401SStefano Zampini Level: intermediate 1927da1bb401SStefano Zampini 1928b6fdb6dfSStefano Zampini Developer notes: 1929da1bb401SStefano Zampini 1930da1bb401SStefano Zampini Contributed by Stefano Zampini 1931da1bb401SStefano Zampini 1932da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 1933da1bb401SStefano Zampini M*/ 1934b2573a8aSBarry Smith 1935da1bb401SStefano Zampini #undef __FUNCT__ 1936da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 19378cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 1938da1bb401SStefano Zampini { 1939da1bb401SStefano Zampini PetscErrorCode ierr; 1940da1bb401SStefano Zampini PC_BDDC *pcbddc; 1941da1bb401SStefano Zampini 1942da1bb401SStefano Zampini PetscFunctionBegin; 1943da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 1944b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 1945da1bb401SStefano Zampini pc->data = (void*)pcbddc; 1946da1bb401SStefano Zampini 1947da1bb401SStefano Zampini /* create PCIS data structure */ 1948da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 1949da1bb401SStefano Zampini 195047d04d0dSStefano Zampini /* BDDC customization */ 195108a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 195247d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 195347d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 195447d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 195547d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 195647d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 195747d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 1958fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 1959fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 19603301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 196147d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 1962b9d89cd5SStefano Zampini /* private */ 1963727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 19640e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 19650e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 19660e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 1967e9189074SStefano Zampini pcbddc->n_vertices = 0; 1968727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 1969fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 197068457ee5SStefano Zampini pcbddc->coarse_size = -1; 1971f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1972727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 1973f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 1974f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 1975f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 1976674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 19773972b0daSStefano Zampini pcbddc->temp_solution = 0; 1978534831adSStefano Zampini pcbddc->original_rhs = 0; 1979534831adSStefano Zampini pcbddc->local_mat = 0; 1980534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 1981b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 1982906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 1983da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 1984da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 1985da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 1986da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 198715aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 198815aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 1989da1bb401SStefano Zampini pcbddc->vec1_P = 0; 1990da1bb401SStefano Zampini pcbddc->vec1_R = 0; 1991da1bb401SStefano Zampini pcbddc->vec2_R = 0; 1992da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 1993da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 1994da1bb401SStefano Zampini pcbddc->R_to_B = 0; 1995da1bb401SStefano Zampini pcbddc->R_to_D = 0; 1996da1bb401SStefano Zampini pcbddc->ksp_D = 0; 1997da1bb401SStefano Zampini pcbddc->ksp_R = 0; 1998da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 1999785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2000785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2001785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 200260d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 200360d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 200463602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2005da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 200663602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2007da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 200885c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 200947d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 201047d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2011b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 20124fad6a16SStefano Zampini pcbddc->current_level = 0; 20132b510759SStefano Zampini pcbddc->max_levels = 0; 2014323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 201574e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2016f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2017323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 2018da1bb401SStefano Zampini 2019674ae819SStefano Zampini /* create local graph structure */ 2020674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2021674ae819SStefano Zampini 2022674ae819SStefano Zampini /* scaling */ 2023674ae819SStefano Zampini pcbddc->work_scaling = 0; 202434a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2025ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2026b96c3477SStefano Zampini 2027b96c3477SStefano Zampini /* create sub schurs structure */ 2028b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2029b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2030b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2031b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2032b96c3477SStefano Zampini 2033b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2034da1bb401SStefano Zampini 2035b7eb3628SStefano Zampini /* adaptivity */ 2036f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 203708122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2038f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2039b7eb3628SStefano Zampini 2040da1bb401SStefano Zampini /* function pointers */ 2041da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 204293bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2043da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2044da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2045da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 20466b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2047da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2048da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2049da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2050534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2051534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2052da1bb401SStefano Zampini 2053da1bb401SStefano Zampini /* composing function */ 2054906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2055674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 2056bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 20572b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2058b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 20592b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2060bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 206182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2062bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 206382ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2064bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 206582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2066bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 206782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2068bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 206963602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2070bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2071bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2072bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2073bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2074da1bb401SStefano Zampini PetscFunctionReturn(0); 2075da1bb401SStefano Zampini } 20763425bc38SStefano Zampini 2077