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 MATIS related operations contained in BDDC code 18eb97c9d2SStefano Zampini - Provide general case for subassembling 19eb97c9d2SStefano Zampini 2053cdbc3dSStefano Zampini */ 210c7d97c5SJed Brown 22ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 23ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 243b03a366Sstefano_zampini #include <petscblaslapack.h> 25674ae819SStefano Zampini 260c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 270c7d97c5SJed Brown #undef __FUNCT__ 280c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 294416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 300c7d97c5SJed Brown { 310c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 320c7d97c5SJed Brown PetscErrorCode ierr; 330c7d97c5SJed Brown 340c7d97c5SJed Brown PetscFunctionBegin; 35e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 368eeda7d8SStefano Zampini /* Verbose debugging */ 37c7017625SStefano 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); 38c7017625SStefano 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); 39c7017625SStefano 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); 40c7017625SStefano 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); 418eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 426b78500eSPatrick Sanan /* Primal space customization */ 4308a5cf49SStefano 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); 448eeda7d8SStefano 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); 458eeda7d8SStefano 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); 468eeda7d8SStefano 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); 476661aa1dSStefano 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); 48ac632422SStefano 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); 498eeda7d8SStefano Zampini /* Change of basis */ 50b9b85e73SStefano 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); 51b9b85e73SStefano 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); 52674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 53674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 54674ae819SStefano Zampini } 558eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 568eeda7d8SStefano 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); 5774e2c79eSStefano 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); 580298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 592b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 60323d395dSStefano 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); 61674ae819SStefano 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); 62b96c3477SStefano 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); 63b96c3477SStefano 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); 64b96c3477SStefano 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); 65ac632422SStefano 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); 664c6709b3SStefano 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); 6708122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 6808122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 693301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 70b0c7d250SStefano 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); 710c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 720c7d97c5SJed Brown PetscFunctionReturn(0); 730c7d97c5SJed Brown } 746b78500eSPatrick Sanan 756b78500eSPatrick Sanan /* -------------------------------------------------------------------------- */ 766b78500eSPatrick Sanan #undef __FUNCT__ 776b78500eSPatrick Sanan #define __FUNCT__ "PCView_BDDC" 786b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 796b78500eSPatrick Sanan { 806b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 816b78500eSPatrick Sanan PetscErrorCode ierr; 826b78500eSPatrick Sanan PetscBool isascii,isstring; 836b78500eSPatrick Sanan 846b78500eSPatrick Sanan PetscFunctionBegin; 856b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 866b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 876b78500eSPatrick Sanan 886b78500eSPatrick Sanan /* In a braindead way, print out anything which the user can control from the command line, 896b78500eSPatrick Sanan cribbing from PCSetFromOptions_BDDC */ 906b78500eSPatrick Sanan 916b78500eSPatrick Sanan /* Nothing printed for the String viewer */ 926b78500eSPatrick Sanan 936b78500eSPatrick Sanan /* ASCII viewer */ 946b78500eSPatrick Sanan if (isascii) { 956b78500eSPatrick Sanan /* Verbose debugging */ 966b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use verbose output: %d\n",pcbddc->dbg_flag);CHKERRQ(ierr); 976b78500eSPatrick Sanan 986b78500eSPatrick Sanan /* Primal space customization */ 996b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use local mat graph: %d\n",pcbddc->use_local_adj);CHKERRQ(ierr); 1006b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use vertices: %d\n",pcbddc->use_vertices);CHKERRQ(ierr); 1016b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 1026b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 1036b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 1046b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 1056b78500eSPatrick Sanan 1066b78500eSPatrick Sanan /* Change of basis */ 1076b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 1086b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 1096b78500eSPatrick Sanan 1106b78500eSPatrick Sanan /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 1116b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 1126b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Coarse problem restribute procs: %d\n",pcbddc->redistribute_coarse);CHKERRQ(ierr); 1136b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 1146b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr); 1156b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 1166b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 1176b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 1186b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 1196b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 1206b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Fast deluxe scaling: %d\n",pcbddc->faster_deluxe);CHKERRQ(ierr); 1216b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Adaptive constraint selection threshold: %g\n",pcbddc->adaptive_threshold);CHKERRQ(ierr); 1226b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 1236b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 1246b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 1256b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer, " BDDC: Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 1266b78500eSPatrick Sanan } 1276b78500eSPatrick Sanan PetscFunctionReturn(0); 1286b78500eSPatrick Sanan } 1296b78500eSPatrick Sanan 1300c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 131674ae819SStefano Zampini #undef __FUNCT__ 132906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 133906d46d4SStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 134b9b85e73SStefano Zampini { 135b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 136b9b85e73SStefano Zampini PetscErrorCode ierr; 137b9b85e73SStefano Zampini 138b9b85e73SStefano Zampini PetscFunctionBegin; 139b9b85e73SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 140b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 141b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 142b9b85e73SStefano Zampini PetscFunctionReturn(0); 143b9b85e73SStefano Zampini } 144b9b85e73SStefano Zampini #undef __FUNCT__ 145906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 146b9b85e73SStefano Zampini /*@ 147906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 148b9b85e73SStefano Zampini 149b9b85e73SStefano Zampini Collective on PC 150b9b85e73SStefano Zampini 151b9b85e73SStefano Zampini Input Parameters: 152b9b85e73SStefano Zampini + pc - the preconditioning context 153906d46d4SStefano Zampini - change - the change of basis matrix 154b9b85e73SStefano Zampini 155b9b85e73SStefano Zampini Level: intermediate 156b9b85e73SStefano Zampini 157b9b85e73SStefano Zampini Notes: 158b9b85e73SStefano Zampini 159b9b85e73SStefano Zampini .seealso: PCBDDC 160b9b85e73SStefano Zampini @*/ 161906d46d4SStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 162b9b85e73SStefano Zampini { 163b9b85e73SStefano Zampini PetscErrorCode ierr; 164b9b85e73SStefano Zampini 165b9b85e73SStefano Zampini PetscFunctionBegin; 166b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 167b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 168906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 169906d46d4SStefano Zampini if (pc->mat) { 170906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 171906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 172906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 1736c4ed002SBarry 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); 1746c4ed002SBarry 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); 175906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 176906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 1776c4ed002SBarry 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); 1786c4ed002SBarry 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); 179906d46d4SStefano Zampini } 180906d46d4SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 181b9b85e73SStefano Zampini PetscFunctionReturn(0); 182b9b85e73SStefano Zampini } 183b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 184b9b85e73SStefano Zampini #undef __FUNCT__ 185674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 186674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 187674ae819SStefano Zampini { 188674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 189674ae819SStefano Zampini PetscErrorCode ierr; 1901e6b0712SBarry Smith 191674ae819SStefano Zampini PetscFunctionBegin; 192674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 193674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 194674ae819SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 195674ae819SStefano Zampini PetscFunctionReturn(0); 196674ae819SStefano Zampini } 197674ae819SStefano Zampini #undef __FUNCT__ 198674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 199674ae819SStefano Zampini /*@ 20028509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 201674ae819SStefano Zampini 20217eb1463SStefano Zampini Collective 203674ae819SStefano Zampini 204674ae819SStefano Zampini Input Parameters: 205674ae819SStefano Zampini + pc - the preconditioning context 20617eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 207674ae819SStefano Zampini 208674ae819SStefano Zampini Level: intermediate 209674ae819SStefano Zampini 210674ae819SStefano Zampini Notes: 211674ae819SStefano Zampini 212674ae819SStefano Zampini .seealso: PCBDDC 213674ae819SStefano Zampini @*/ 214674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 215674ae819SStefano Zampini { 216674ae819SStefano Zampini PetscErrorCode ierr; 217674ae819SStefano Zampini 218674ae819SStefano Zampini PetscFunctionBegin; 219674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 220674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 22117eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 222674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 223674ae819SStefano Zampini PetscFunctionReturn(0); 224674ae819SStefano Zampini } 225674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 2260c7d97c5SJed Brown #undef __FUNCT__ 2274fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 2284fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 2294fad6a16SStefano Zampini { 2304fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2314fad6a16SStefano Zampini 2324fad6a16SStefano Zampini PetscFunctionBegin; 2334fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 2344fad6a16SStefano Zampini PetscFunctionReturn(0); 2354fad6a16SStefano Zampini } 2361e6b0712SBarry Smith 2374fad6a16SStefano Zampini #undef __FUNCT__ 2384fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 2394fad6a16SStefano Zampini /*@ 24028509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 2414fad6a16SStefano Zampini 2424fad6a16SStefano Zampini Logically collective on PC 2434fad6a16SStefano Zampini 2444fad6a16SStefano Zampini Input Parameters: 2454fad6a16SStefano Zampini + pc - the preconditioning context 24628509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 2474fad6a16SStefano Zampini 2480f202f7eSStefano Zampini Options Database Keys: 2490f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 2504fad6a16SStefano Zampini 2514fad6a16SStefano Zampini Level: intermediate 2524fad6a16SStefano Zampini 2534fad6a16SStefano Zampini Notes: 2540f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 2554fad6a16SStefano Zampini 2560f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 2574fad6a16SStefano Zampini @*/ 2584fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 2594fad6a16SStefano Zampini { 2604fad6a16SStefano Zampini PetscErrorCode ierr; 2614fad6a16SStefano Zampini 2624fad6a16SStefano Zampini PetscFunctionBegin; 2634fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 2642b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 2654fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 2664fad6a16SStefano Zampini PetscFunctionReturn(0); 2674fad6a16SStefano Zampini } 2682b510759SStefano Zampini 269b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 2702b510759SStefano Zampini #undef __FUNCT__ 271b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 272b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 273b8ffe317SStefano Zampini { 274b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 275b8ffe317SStefano Zampini 276b8ffe317SStefano Zampini PetscFunctionBegin; 27785c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 278b8ffe317SStefano Zampini PetscFunctionReturn(0); 279b8ffe317SStefano Zampini } 280b8ffe317SStefano Zampini 281b8ffe317SStefano Zampini #undef __FUNCT__ 282b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 283b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 2842b510759SStefano Zampini { 2852b510759SStefano Zampini PetscErrorCode ierr; 2862b510759SStefano Zampini 2872b510759SStefano Zampini PetscFunctionBegin; 2882b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 289b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 290b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 2912b510759SStefano Zampini PetscFunctionReturn(0); 2922b510759SStefano Zampini } 2931e6b0712SBarry Smith 2944fad6a16SStefano Zampini #undef __FUNCT__ 2952b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 2962b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 2974fad6a16SStefano Zampini { 2984fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2994fad6a16SStefano Zampini 3004fad6a16SStefano Zampini PetscFunctionBegin; 3012b510759SStefano Zampini pcbddc->current_level = level; 3024fad6a16SStefano Zampini PetscFunctionReturn(0); 3034fad6a16SStefano Zampini } 3041e6b0712SBarry Smith 3054fad6a16SStefano Zampini #undef __FUNCT__ 306b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 307b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 308b8ffe317SStefano Zampini { 309b8ffe317SStefano Zampini PetscErrorCode ierr; 310b8ffe317SStefano Zampini 311b8ffe317SStefano Zampini PetscFunctionBegin; 312b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 313b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 314b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 315b8ffe317SStefano Zampini PetscFunctionReturn(0); 316b8ffe317SStefano Zampini } 317b8ffe317SStefano Zampini 318b8ffe317SStefano Zampini #undef __FUNCT__ 3192b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 3202b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 3212b510759SStefano Zampini { 3222b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3232b510759SStefano Zampini 3242b510759SStefano Zampini PetscFunctionBegin; 3252b510759SStefano Zampini pcbddc->max_levels = levels; 3262b510759SStefano Zampini PetscFunctionReturn(0); 3272b510759SStefano Zampini } 3282b510759SStefano Zampini 3292b510759SStefano Zampini #undef __FUNCT__ 3302b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 3314fad6a16SStefano Zampini /*@ 33228509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 3334fad6a16SStefano Zampini 3344fad6a16SStefano Zampini Logically collective on PC 3354fad6a16SStefano Zampini 3364fad6a16SStefano Zampini Input Parameters: 3374fad6a16SStefano Zampini + pc - the preconditioning context 33828509bceSStefano Zampini - levels - the maximum number of levels (max 9) 3394fad6a16SStefano Zampini 3400f202f7eSStefano Zampini Options Database Keys: 3410f202f7eSStefano Zampini . -pc_bddc_levels 3424fad6a16SStefano Zampini 3434fad6a16SStefano Zampini Level: intermediate 3444fad6a16SStefano Zampini 3454fad6a16SStefano Zampini Notes: 3460f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 3474fad6a16SStefano Zampini 3480f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 3494fad6a16SStefano Zampini @*/ 3502b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 3514fad6a16SStefano Zampini { 3524fad6a16SStefano Zampini PetscErrorCode ierr; 3534fad6a16SStefano Zampini 3544fad6a16SStefano Zampini PetscFunctionBegin; 3554fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 3562b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 357312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 3582b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 3594fad6a16SStefano Zampini PetscFunctionReturn(0); 3604fad6a16SStefano Zampini } 3614fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 3621e6b0712SBarry Smith 3634fad6a16SStefano Zampini #undef __FUNCT__ 3643b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 3653b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 3663b03a366Sstefano_zampini { 3673b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3683b03a366Sstefano_zampini PetscErrorCode ierr; 3693b03a366Sstefano_zampini 3703b03a366Sstefano_zampini PetscFunctionBegin; 371785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 372785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 3733b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 37436e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 37536e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 376fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 3773b03a366Sstefano_zampini PetscFunctionReturn(0); 3783b03a366Sstefano_zampini } 3791e6b0712SBarry Smith 3803b03a366Sstefano_zampini #undef __FUNCT__ 3813b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 3823b03a366Sstefano_zampini /*@ 38328509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 3843b03a366Sstefano_zampini 385785d1243SStefano Zampini Collective 3863b03a366Sstefano_zampini 3873b03a366Sstefano_zampini Input Parameters: 3883b03a366Sstefano_zampini + pc - the preconditioning context 389785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 3903b03a366Sstefano_zampini 3913b03a366Sstefano_zampini Level: intermediate 3923b03a366Sstefano_zampini 3930f202f7eSStefano Zampini Notes: 3940f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 3953b03a366Sstefano_zampini 3960f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 3973b03a366Sstefano_zampini @*/ 3983b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 3993b03a366Sstefano_zampini { 4003b03a366Sstefano_zampini PetscErrorCode ierr; 4013b03a366Sstefano_zampini 4023b03a366Sstefano_zampini PetscFunctionBegin; 4033b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 404674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 405785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 4063b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4073b03a366Sstefano_zampini PetscFunctionReturn(0); 4083b03a366Sstefano_zampini } 4093b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 4101e6b0712SBarry Smith 4113b03a366Sstefano_zampini #undef __FUNCT__ 41282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 41382ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 4140c7d97c5SJed Brown { 4150c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4160c7d97c5SJed Brown PetscErrorCode ierr; 4170c7d97c5SJed Brown 4180c7d97c5SJed Brown PetscFunctionBegin; 419785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 420785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 4210c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 4220c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 423785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 4247d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4250c7d97c5SJed Brown PetscFunctionReturn(0); 4260c7d97c5SJed Brown } 4270c7d97c5SJed Brown 4280c7d97c5SJed Brown #undef __FUNCT__ 42982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 4309c0446d6SStefano Zampini /*@ 43182ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 4329c0446d6SStefano Zampini 433785d1243SStefano Zampini Collective 43453cdbc3dSStefano Zampini 43553cdbc3dSStefano Zampini Input Parameters: 43653cdbc3dSStefano Zampini + pc - the preconditioning context 43782ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 43853cdbc3dSStefano Zampini 43953cdbc3dSStefano Zampini Level: intermediate 44053cdbc3dSStefano Zampini 4419c0446d6SStefano Zampini Notes: 44253cdbc3dSStefano Zampini 4430f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 44453cdbc3dSStefano Zampini @*/ 44582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 4460c7d97c5SJed Brown { 4470c7d97c5SJed Brown PetscErrorCode ierr; 4480c7d97c5SJed Brown 4490c7d97c5SJed Brown PetscFunctionBegin; 4500c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4510c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 45282ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 45382ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 4540c7d97c5SJed Brown PetscFunctionReturn(0); 4550c7d97c5SJed Brown } 4560c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 4570c7d97c5SJed Brown 4580c7d97c5SJed Brown #undef __FUNCT__ 4590c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 46053cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 4610c7d97c5SJed Brown { 4620c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 46353cdbc3dSStefano Zampini PetscErrorCode ierr; 4640c7d97c5SJed Brown 4650c7d97c5SJed Brown PetscFunctionBegin; 466785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 467785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 46853cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 46936e030ebSStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 47036e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 471fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 4720c7d97c5SJed Brown PetscFunctionReturn(0); 4730c7d97c5SJed Brown } 4741e6b0712SBarry Smith 4750c7d97c5SJed Brown #undef __FUNCT__ 4760c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 47757527edcSJed Brown /*@ 47828509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 47957527edcSJed Brown 480785d1243SStefano Zampini Collective 48157527edcSJed Brown 48257527edcSJed Brown Input Parameters: 48357527edcSJed Brown + pc - the preconditioning context 484785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 48557527edcSJed Brown 48657527edcSJed Brown Level: intermediate 48757527edcSJed Brown 4880f202f7eSStefano Zampini Notes: 4890f202f7eSStefano Zampini Any process can list any global node 49057527edcSJed Brown 4910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 49257527edcSJed Brown @*/ 49353cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 4940c7d97c5SJed Brown { 4950c7d97c5SJed Brown PetscErrorCode ierr; 4960c7d97c5SJed Brown 4970c7d97c5SJed Brown PetscFunctionBegin; 4980c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 499674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 500785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 50153cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 50253cdbc3dSStefano Zampini PetscFunctionReturn(0); 50353cdbc3dSStefano Zampini } 50453cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 5051e6b0712SBarry Smith 50653cdbc3dSStefano Zampini #undef __FUNCT__ 50782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 50882ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 5090c7d97c5SJed Brown { 5100c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5110c7d97c5SJed Brown PetscErrorCode ierr; 5120c7d97c5SJed Brown 5130c7d97c5SJed Brown PetscFunctionBegin; 514785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 515785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 5160c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 5170c7d97c5SJed Brown ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 518785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 5197d24d155SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 5200c7d97c5SJed Brown PetscFunctionReturn(0); 5210c7d97c5SJed Brown } 5220c7d97c5SJed Brown 5230c7d97c5SJed Brown #undef __FUNCT__ 52482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 5250c7d97c5SJed Brown /*@ 52682ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 5270c7d97c5SJed Brown 528785d1243SStefano Zampini Collective 5290c7d97c5SJed Brown 5300c7d97c5SJed Brown Input Parameters: 5310c7d97c5SJed Brown + pc - the preconditioning context 53282ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 5330c7d97c5SJed Brown 5340c7d97c5SJed Brown Level: intermediate 5350c7d97c5SJed Brown 5360c7d97c5SJed Brown Notes: 5370c7d97c5SJed Brown 5380f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 5390c7d97c5SJed Brown @*/ 54082ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 5410c7d97c5SJed Brown { 5420c7d97c5SJed Brown PetscErrorCode ierr; 5430c7d97c5SJed Brown 5440c7d97c5SJed Brown PetscFunctionBegin; 5450c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5460c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 54782ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 54882ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 54953cdbc3dSStefano Zampini PetscFunctionReturn(0); 55053cdbc3dSStefano Zampini } 55153cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 55253cdbc3dSStefano Zampini 55353cdbc3dSStefano Zampini #undef __FUNCT__ 554da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 555da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 556da1bb401SStefano Zampini { 557da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 558da1bb401SStefano Zampini 559da1bb401SStefano Zampini PetscFunctionBegin; 560da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 561da1bb401SStefano Zampini PetscFunctionReturn(0); 562da1bb401SStefano Zampini } 5631e6b0712SBarry Smith 564da1bb401SStefano Zampini #undef __FUNCT__ 565da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 566da1bb401SStefano Zampini /*@ 567785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 568da1bb401SStefano Zampini 569785d1243SStefano Zampini Collective 570785d1243SStefano Zampini 571785d1243SStefano Zampini Input Parameters: 572785d1243SStefano Zampini . pc - the preconditioning context 573785d1243SStefano Zampini 574785d1243SStefano Zampini Output Parameters: 575785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 576785d1243SStefano Zampini 577785d1243SStefano Zampini Level: intermediate 578785d1243SStefano Zampini 5790f202f7eSStefano Zampini Notes: 5800f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 581785d1243SStefano Zampini 582785d1243SStefano Zampini .seealso: PCBDDC 583785d1243SStefano Zampini @*/ 584785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 585785d1243SStefano Zampini { 586785d1243SStefano Zampini PetscErrorCode ierr; 587785d1243SStefano Zampini 588785d1243SStefano Zampini PetscFunctionBegin; 589785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 590785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 591785d1243SStefano Zampini PetscFunctionReturn(0); 592785d1243SStefano Zampini } 593785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 594785d1243SStefano Zampini 595785d1243SStefano Zampini #undef __FUNCT__ 596785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 597785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 598785d1243SStefano Zampini { 599785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 600785d1243SStefano Zampini 601785d1243SStefano Zampini PetscFunctionBegin; 602785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 603785d1243SStefano Zampini PetscFunctionReturn(0); 604785d1243SStefano Zampini } 605785d1243SStefano Zampini 606785d1243SStefano Zampini #undef __FUNCT__ 60782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 608da1bb401SStefano Zampini /*@ 60982ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 610da1bb401SStefano Zampini 611785d1243SStefano Zampini Collective 612da1bb401SStefano Zampini 613da1bb401SStefano Zampini Input Parameters: 61428509bceSStefano Zampini . pc - the preconditioning context 615da1bb401SStefano Zampini 616da1bb401SStefano Zampini Output Parameters: 61728509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 618da1bb401SStefano Zampini 619da1bb401SStefano Zampini Level: intermediate 620da1bb401SStefano Zampini 621da1bb401SStefano Zampini Notes: 6220f202f7eSStefano 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). 6230f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 624da1bb401SStefano Zampini 625da1bb401SStefano Zampini .seealso: PCBDDC 626da1bb401SStefano Zampini @*/ 62782ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 628da1bb401SStefano Zampini { 629da1bb401SStefano Zampini PetscErrorCode ierr; 630da1bb401SStefano Zampini 631da1bb401SStefano Zampini PetscFunctionBegin; 632da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 63382ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 634da1bb401SStefano Zampini PetscFunctionReturn(0); 635da1bb401SStefano Zampini } 636da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 6371e6b0712SBarry Smith 638da1bb401SStefano Zampini #undef __FUNCT__ 63953cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 64053cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 64153cdbc3dSStefano Zampini { 64253cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 64353cdbc3dSStefano Zampini 64453cdbc3dSStefano Zampini PetscFunctionBegin; 64553cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 64653cdbc3dSStefano Zampini PetscFunctionReturn(0); 64753cdbc3dSStefano Zampini } 6481e6b0712SBarry Smith 64953cdbc3dSStefano Zampini #undef __FUNCT__ 65053cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 65153cdbc3dSStefano Zampini /*@ 652785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 65353cdbc3dSStefano Zampini 654785d1243SStefano Zampini Collective 655785d1243SStefano Zampini 656785d1243SStefano Zampini Input Parameters: 657785d1243SStefano Zampini . pc - the preconditioning context 658785d1243SStefano Zampini 659785d1243SStefano Zampini Output Parameters: 660785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 661785d1243SStefano Zampini 662785d1243SStefano Zampini Level: intermediate 663785d1243SStefano Zampini 6640f202f7eSStefano Zampini Notes: 6650f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 666785d1243SStefano Zampini 667785d1243SStefano Zampini .seealso: PCBDDC 668785d1243SStefano Zampini @*/ 669785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 670785d1243SStefano Zampini { 671785d1243SStefano Zampini PetscErrorCode ierr; 672785d1243SStefano Zampini 673785d1243SStefano Zampini PetscFunctionBegin; 674785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 675785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 676785d1243SStefano Zampini PetscFunctionReturn(0); 677785d1243SStefano Zampini } 678785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 679785d1243SStefano Zampini 680785d1243SStefano Zampini #undef __FUNCT__ 681785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 682785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 683785d1243SStefano Zampini { 684785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 685785d1243SStefano Zampini 686785d1243SStefano Zampini PetscFunctionBegin; 687785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 688785d1243SStefano Zampini PetscFunctionReturn(0); 689785d1243SStefano Zampini } 690785d1243SStefano Zampini 691785d1243SStefano Zampini #undef __FUNCT__ 69282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 69353cdbc3dSStefano Zampini /*@ 69482ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 69553cdbc3dSStefano Zampini 696785d1243SStefano Zampini Collective 69753cdbc3dSStefano Zampini 69853cdbc3dSStefano Zampini Input Parameters: 69928509bceSStefano Zampini . pc - the preconditioning context 70053cdbc3dSStefano Zampini 70153cdbc3dSStefano Zampini Output Parameters: 70228509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 70353cdbc3dSStefano Zampini 70453cdbc3dSStefano Zampini Level: intermediate 70553cdbc3dSStefano Zampini 70653cdbc3dSStefano Zampini Notes: 7070f202f7eSStefano 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). 7080f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 70953cdbc3dSStefano Zampini 71053cdbc3dSStefano Zampini .seealso: PCBDDC 71153cdbc3dSStefano Zampini @*/ 71282ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 71353cdbc3dSStefano Zampini { 71453cdbc3dSStefano Zampini PetscErrorCode ierr; 71553cdbc3dSStefano Zampini 71653cdbc3dSStefano Zampini PetscFunctionBegin; 71753cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 71882ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 7190c7d97c5SJed Brown PetscFunctionReturn(0); 7200c7d97c5SJed Brown } 72136e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 7221e6b0712SBarry Smith 72336e030ebSStefano Zampini #undef __FUNCT__ 724da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 7251a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 72636e030ebSStefano Zampini { 72736e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 728da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 729da1bb401SStefano Zampini PetscErrorCode ierr; 73036e030ebSStefano Zampini 73136e030ebSStefano Zampini PetscFunctionBegin; 732674ae819SStefano Zampini /* free old CSR */ 733674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 734fb223d50SStefano Zampini /* TODO: PCBDDCGraphSetAdjacency */ 735674ae819SStefano Zampini /* get CSR into graph structure */ 736da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 737854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 738785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 739674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 740674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 741da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 7421a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 7431a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 744674ae819SStefano Zampini } 745575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 74636e030ebSStefano Zampini PetscFunctionReturn(0); 74736e030ebSStefano Zampini } 7481e6b0712SBarry Smith 74936e030ebSStefano Zampini #undef __FUNCT__ 750da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 75136e030ebSStefano Zampini /*@ 7520f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 75336e030ebSStefano Zampini 75436e030ebSStefano Zampini Not collective 75536e030ebSStefano Zampini 75636e030ebSStefano Zampini Input Parameters: 75736e030ebSStefano Zampini + pc - the preconditioning context 7580f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 75928509bceSStefano Zampini . xadj, adjncy - the CSR graph 76028509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 76136e030ebSStefano Zampini 76236e030ebSStefano Zampini Level: intermediate 76336e030ebSStefano Zampini 76436e030ebSStefano Zampini Notes: 76536e030ebSStefano Zampini 76628509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 76736e030ebSStefano Zampini @*/ 7681a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 76936e030ebSStefano Zampini { 770575ad6abSStefano Zampini void (*f)(void) = 0; 77136e030ebSStefano Zampini PetscErrorCode ierr; 77236e030ebSStefano Zampini 77336e030ebSStefano Zampini PetscFunctionBegin; 77436e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 775674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 776d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 7776c4ed002SBarry Smith if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode); 7781a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 779575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 780575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 781575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 782575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 783575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 784da1bb401SStefano Zampini } 78536e030ebSStefano Zampini PetscFunctionReturn(0); 78636e030ebSStefano Zampini } 7879c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 7881e6b0712SBarry Smith 7899c0446d6SStefano Zampini #undef __FUNCT__ 79063602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 79163602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 79263602bcaSStefano Zampini { 79363602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79463602bcaSStefano Zampini PetscInt i; 79563602bcaSStefano Zampini PetscErrorCode ierr; 79663602bcaSStefano Zampini 79763602bcaSStefano Zampini PetscFunctionBegin; 79863602bcaSStefano Zampini /* Destroy ISes if they were already set */ 79963602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 80063602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 80163602bcaSStefano Zampini } 80263602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 80363602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 80463602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 80563602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 80663602bcaSStefano Zampini } 80763602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 80863602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 80963602bcaSStefano Zampini /* allocate space then set */ 810d02579f5SStefano Zampini if (n_is) { 811d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 812d02579f5SStefano Zampini } 81363602bcaSStefano Zampini for (i=0;i<n_is;i++) { 81463602bcaSStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 81563602bcaSStefano Zampini pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 81663602bcaSStefano Zampini } 81763602bcaSStefano Zampini pcbddc->n_ISForDofsLocal=n_is; 81863602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8191a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 82063602bcaSStefano Zampini PetscFunctionReturn(0); 82163602bcaSStefano Zampini } 82263602bcaSStefano Zampini 82363602bcaSStefano Zampini #undef __FUNCT__ 82463602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 82563602bcaSStefano Zampini /*@ 82663602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 82763602bcaSStefano Zampini 82863602bcaSStefano Zampini Collective 82963602bcaSStefano Zampini 83063602bcaSStefano Zampini Input Parameters: 83163602bcaSStefano Zampini + pc - the preconditioning context 8320f202f7eSStefano Zampini . n_is - number of index sets defining the fields 8330f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 83463602bcaSStefano Zampini 83563602bcaSStefano Zampini Level: intermediate 83663602bcaSStefano Zampini 8370f202f7eSStefano Zampini Notes: 8380f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 83963602bcaSStefano Zampini 84063602bcaSStefano Zampini .seealso: PCBDDC 84163602bcaSStefano Zampini @*/ 84263602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 84363602bcaSStefano Zampini { 84463602bcaSStefano Zampini PetscInt i; 84563602bcaSStefano Zampini PetscErrorCode ierr; 84663602bcaSStefano Zampini 84763602bcaSStefano Zampini PetscFunctionBegin; 84863602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 84963602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 85063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 85163602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 85263602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 85363602bcaSStefano Zampini } 854e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 85563602bcaSStefano Zampini PetscFunctionReturn(0); 85663602bcaSStefano Zampini } 85763602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 85863602bcaSStefano Zampini 85963602bcaSStefano Zampini #undef __FUNCT__ 8609c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 8619c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 8629c0446d6SStefano Zampini { 8639c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 8649c0446d6SStefano Zampini PetscInt i; 8659c0446d6SStefano Zampini PetscErrorCode ierr; 8669c0446d6SStefano Zampini 8679c0446d6SStefano Zampini PetscFunctionBegin; 868da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 8699c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 8709c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 8719c0446d6SStefano Zampini } 872d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 87363602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 87463602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 87563602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 87663602bcaSStefano Zampini } 87763602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 87863602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 879da1bb401SStefano Zampini /* allocate space then set */ 880d02579f5SStefano Zampini if (n_is) { 881785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 882d02579f5SStefano Zampini } 8839c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 884da1bb401SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 885da1bb401SStefano Zampini pcbddc->ISForDofs[i]=ISForDofs[i]; 8869c0446d6SStefano Zampini } 8879c0446d6SStefano Zampini pcbddc->n_ISForDofs=n_is; 88863602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 8891a8a16d1SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 8909c0446d6SStefano Zampini PetscFunctionReturn(0); 8919c0446d6SStefano Zampini } 8921e6b0712SBarry Smith 8939c0446d6SStefano Zampini #undef __FUNCT__ 8949c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 8959c0446d6SStefano Zampini /*@ 89663602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 8979c0446d6SStefano Zampini 89863602bcaSStefano Zampini Collective 8999c0446d6SStefano Zampini 9009c0446d6SStefano Zampini Input Parameters: 9019c0446d6SStefano Zampini + pc - the preconditioning context 9020f202f7eSStefano Zampini . n_is - number of index sets defining the fields 9030f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 9049c0446d6SStefano Zampini 9059c0446d6SStefano Zampini Level: intermediate 9069c0446d6SStefano Zampini 9070f202f7eSStefano Zampini Notes: 9080f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 9099c0446d6SStefano Zampini 9109c0446d6SStefano Zampini .seealso: PCBDDC 9119c0446d6SStefano Zampini @*/ 9129c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 9139c0446d6SStefano Zampini { 9142b510759SStefano Zampini PetscInt i; 9159c0446d6SStefano Zampini PetscErrorCode ierr; 9169c0446d6SStefano Zampini 9179c0446d6SStefano Zampini PetscFunctionBegin; 9189c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 91963602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 9202b510759SStefano Zampini for (i=0;i<n_is;i++) { 92163602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 92263602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 9232b510759SStefano Zampini } 9249c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 9259c0446d6SStefano Zampini PetscFunctionReturn(0); 9269c0446d6SStefano Zampini } 927906d46d4SStefano Zampini 928da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 929534831adSStefano Zampini #undef __FUNCT__ 930534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 931534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 932534831adSStefano Zampini /* 933534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 934534831adSStefano Zampini guess if a transformation of basis approach has been selected. 9359c0446d6SStefano Zampini 936534831adSStefano Zampini Input Parameter: 937534831adSStefano Zampini + pc - the preconditioner contex 938534831adSStefano Zampini 939534831adSStefano Zampini Application Interface Routine: PCPreSolve() 940534831adSStefano Zampini 941534831adSStefano Zampini Notes: 942534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 943534831adSStefano Zampini the user, but instead is called by KSPSolve(). 944534831adSStefano Zampini */ 945534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 946534831adSStefano Zampini { 947534831adSStefano Zampini PetscErrorCode ierr; 948534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 949534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 9503972b0daSStefano Zampini Vec used_vec; 9518d00608fSStefano Zampini PetscBool copy_rhs = PETSC_TRUE; 952534831adSStefano Zampini 953534831adSStefano Zampini PetscFunctionBegin; 95485c4d303SStefano Zampini /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 95585c4d303SStefano Zampini if (ksp) { 95685c4d303SStefano Zampini PetscBool iscg; 95785c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 95885c4d303SStefano Zampini if (!iscg) { 95985c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 96085c4d303SStefano Zampini } 96185c4d303SStefano Zampini } 962c7017625SStefano Zampini 96385c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 96462a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 96562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 96662a6ff1dSStefano Zampini } 96762a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 96862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 96962a6ff1dSStefano Zampini } 9708d00608fSStefano Zampini 9713972b0daSStefano Zampini if (x) { 9723972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 9733972b0daSStefano Zampini used_vec = x; 9748d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 9753972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 9763972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 9773972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9783972b0daSStefano Zampini } 9798efcfb23SStefano Zampini 9808efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 9813972b0daSStefano Zampini if (ksp) { 982a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 9838efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 9848efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 9853972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 9863972b0daSStefano Zampini } 9873972b0daSStefano Zampini } 9883308cffdSStefano Zampini 9898d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 9903972b0daSStefano Zampini 9913972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 992a07ea27aSStefano Zampini if (rhs) { 9933975b054SStefano Zampini IS dirIS = NULL; 9943975b054SStefano Zampini 995a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 9963975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 9973975b054SStefano Zampini if (dirIS) { 998906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 999785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 10002b095fd8SStefano Zampini PetscScalar *array_x; 10012b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1002785d1243SStefano Zampini 10033972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 10043972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1005e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1006e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1007e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1008e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 100982ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 10103972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 10112b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10123972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10132fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 10143972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 10152b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 10163972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1017e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1018e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10198d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10201b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 10218efcfb23SStefano Zampini } 1022a07ea27aSStefano Zampini } 1023b76ba322SStefano Zampini 10248efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 10258d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 10268d00608fSStefano Zampini /* store the original rhs */ 10278d00608fSStefano Zampini if (copy_rhs) { 10288d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10298d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10308d00608fSStefano Zampini } 10318d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 10323972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10333972b0daSStefano Zampini ierr = MatMultAdd(pc->pmat,used_vec,rhs,rhs);CHKERRQ(ierr); 10343972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 10358efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 10367acc28cbSStefano Zampini if (ksp) { 10377acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 10387acc28cbSStefano Zampini } 10393308cffdSStefano Zampini } 10408efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1041b76ba322SStefano Zampini 1042b76ba322SStefano Zampini /* store partially computed solution and set initial guess */ 1043b097fa66SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 10448efcfb23SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 1045b76ba322SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1046b76ba322SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1047b76ba322SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 10488efcfb23SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 10498efcfb23SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1050a0cb1b98SStefano Zampini if (ksp) { 1051b76ba322SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 1052b76ba322SStefano Zampini } 1053b76ba322SStefano Zampini } 1054b76ba322SStefano Zampini 1055b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1056906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1057906d46d4SStefano Zampini 1058906d46d4SStefano Zampini /* get change ctx */ 1059906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1060906d46d4SStefano Zampini 1061906d46d4SStefano Zampini /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 1062906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1063906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1064906d46d4SStefano Zampini change_ctx->original_mat = pc->mat; 1065906d46d4SStefano Zampini 1066906d46d4SStefano Zampini /* change iteration matrix */ 1067906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1068906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1069906d46d4SStefano Zampini pc->mat = pcbddc->new_global_mat; 1070906d46d4SStefano Zampini 10718d00608fSStefano Zampini /* store the original rhs */ 10728d00608fSStefano Zampini if (copy_rhs) { 10738d00608fSStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 10748d00608fSStefano Zampini copy_rhs = PETSC_FALSE; 10758d00608fSStefano Zampini } 10768d00608fSStefano Zampini 1077906d46d4SStefano Zampini /* change rhs */ 1078906d46d4SStefano Zampini ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1079906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 10808d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 108192e3dcfbSStefano Zampini } 1082534831adSStefano Zampini PetscFunctionReturn(0); 1083534831adSStefano Zampini } 1084906d46d4SStefano Zampini 1085534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1086534831adSStefano Zampini #undef __FUNCT__ 1087534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1088534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1089534831adSStefano Zampini /* 1090534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1091534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1092534831adSStefano Zampini 1093534831adSStefano Zampini Input Parameter: 1094534831adSStefano Zampini + pc - the preconditioner contex 1095534831adSStefano Zampini 1096534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1097534831adSStefano Zampini 1098534831adSStefano Zampini Notes: 1099534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1100534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1101534831adSStefano Zampini */ 1102534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1103534831adSStefano Zampini { 1104534831adSStefano Zampini PetscErrorCode ierr; 1105534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1106534831adSStefano Zampini 1107534831adSStefano Zampini PetscFunctionBegin; 1108b9b85e73SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1109906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1110906d46d4SStefano Zampini 1111906d46d4SStefano Zampini /* get change ctx */ 1112906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1113906d46d4SStefano Zampini 1114906d46d4SStefano Zampini /* restore iteration matrix */ 1115906d46d4SStefano Zampini ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1116906d46d4SStefano Zampini ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1117906d46d4SStefano Zampini pc->mat = change_ctx->original_mat; 1118906d46d4SStefano Zampini 1119906d46d4SStefano Zampini /* get solution in original basis */ 1120906d46d4SStefano Zampini if (x) { 1121906d46d4SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 1122906d46d4SStefano Zampini ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1123906d46d4SStefano Zampini ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 11243425bc38SStefano Zampini } 1125534831adSStefano Zampini } 1126906d46d4SStefano Zampini 11273972b0daSStefano Zampini /* add solution removed in presolve */ 11286bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 11293425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 11303425bc38SStefano Zampini } 1131906d46d4SStefano Zampini 1132fb223d50SStefano Zampini /* restore rhs to its original state */ 11338d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 1134fb223d50SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1135fb223d50SStefano Zampini } 11368d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11378efcfb23SStefano Zampini 11388efcfb23SStefano Zampini /* restore ksp guess state */ 11398efcfb23SStefano Zampini if (ksp) { 11408efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 11418efcfb23SStefano Zampini } 1142534831adSStefano Zampini PetscFunctionReturn(0); 1143534831adSStefano Zampini } 1144534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 114553cdbc3dSStefano Zampini #undef __FUNCT__ 114653cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 11470c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 11480c7d97c5SJed Brown /* 11490c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 11500c7d97c5SJed Brown by setting data structures and options. 11510c7d97c5SJed Brown 11520c7d97c5SJed Brown Input Parameter: 115353cdbc3dSStefano Zampini + pc - the preconditioner context 11540c7d97c5SJed Brown 11550c7d97c5SJed Brown Application Interface Routine: PCSetUp() 11560c7d97c5SJed Brown 11570c7d97c5SJed Brown Notes: 11580c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 11590c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 11600c7d97c5SJed Brown */ 116153cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 11620c7d97c5SJed Brown { 11630c7d97c5SJed Brown PetscErrorCode ierr; 11640c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 11655e8657edSStefano Zampini Mat_IS* matis; 116608122e43SStefano Zampini MatNullSpace nearnullspace; 116791e8d312SStefano Zampini PetscInt nrows,ncols; 116808122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 11698de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 11705e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 11710c7d97c5SJed Brown 11720c7d97c5SJed Brown PetscFunctionBegin; 11735e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 11746c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 117591e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 11766c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 11775e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1178f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 11793b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1180fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1181f4ddd8eeSStefano Zampini /* split work */ 1182674ae819SStefano Zampini if (pc->setupcalled) { 1183d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1184674ae819SStefano Zampini computetopography = PETSC_FALSE; 1185674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1186674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1187674ae819SStefano Zampini computetopography = PETSC_TRUE; 1188674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1189674ae819SStefano Zampini } 1190674ae819SStefano Zampini } else { 1191674ae819SStefano Zampini computetopography = PETSC_TRUE; 1192674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1193674ae819SStefano Zampini } 1194fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1195fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1196fb180af4SStefano Zampini } 11978de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 11986c4ed002SBarry 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"); 11995a95e1ceSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1200862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1201862806e4SStefano Zampini 12025a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 12036c4ed002SBarry 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"); 12046c4ed002SBarry Smith 1205f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 120670cf5478SStefano Zampini if (pcbddc->dbg_flag) { 120770cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 120858a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 12091575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1210f4ddd8eeSStefano Zampini } 121158a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1212f4ddd8eeSStefano Zampini } 1213f4ddd8eeSStefano Zampini 12145e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 12155e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 12165e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 12175e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 12185e8657edSStefano Zampini } else { 1219b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 12205e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 12215e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1222674ae819SStefano Zampini } 1223f4ddd8eeSStefano Zampini 1224e496cd5dSStefano Zampini /* workaround for reals */ 1225e496cd5dSStefano Zampini #if !defined(PETSC_USE_COMPLEX) 12263301b35fSStefano Zampini if (matis->A->symmetric_set) { 12273301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 12283301b35fSStefano Zampini } 1229e496cd5dSStefano Zampini #endif 1230e496cd5dSStefano Zampini 12315e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 12325e8657edSStefano Zampini { 12335e8657edSStefano Zampini Mat temp_mat; 12345e8657edSStefano Zampini 12355e8657edSStefano Zampini temp_mat = matis->A; 12365e8657edSStefano Zampini matis->A = pcbddc->local_mat; 12375e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 12385e8657edSStefano Zampini pcbddc->local_mat = matis->A; 12395e8657edSStefano Zampini matis->A = temp_mat; 12405e8657edSStefano Zampini } 1241684f6988SStefano Zampini 1242b96c3477SStefano Zampini /* Analyze interface and setup sub_schurs data */ 1243674ae819SStefano Zampini if (computetopography) { 1244674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 12458de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1246674ae819SStefano Zampini } 1247fb8d54d4SStefano Zampini 1248b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1249684f6988SStefano Zampini if (computesolvers) { 1250d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1251d5574798SStefano Zampini 1252d5574798SStefano Zampini if (computesubschurs && computetopography) { 125308122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1254b1b3d7a2SStefano Zampini } 1255d5574798SStefano Zampini if (sub_schurs->use_mumps) { 12562070dbb6SStefano Zampini if (computesubschurs) { 125708122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 12582070dbb6SStefano Zampini } 1259d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1260d5574798SStefano Zampini } else { 1261d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 12622070dbb6SStefano Zampini if (computesubschurs) { 1263d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1264d5574798SStefano Zampini } 12652070dbb6SStefano Zampini } 126608122e43SStefano Zampini if (pcbddc->adaptive_selection) { 126708122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 12688de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1269b7eb3628SStefano Zampini } 1270b1b3d7a2SStefano Zampini } 1271684f6988SStefano Zampini 1272f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1273fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1274f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1275f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1276f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1277f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1278f4ddd8eeSStefano Zampini } else { 1279f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1280f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1281f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1282165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1283f4ddd8eeSStefano Zampini PetscInt i; 1284165b64e2SStefano Zampini const Vec *nearnullvecs; 1285165b64e2SStefano Zampini PetscObjectState state; 1286165b64e2SStefano Zampini PetscInt nnsp_size; 1287165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1288f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1289f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1290165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1291f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1292f4ddd8eeSStefano Zampini break; 1293f4ddd8eeSStefano Zampini } 1294f4ddd8eeSStefano Zampini } 1295f4ddd8eeSStefano Zampini } 1296f4ddd8eeSStefano Zampini } 1297f4ddd8eeSStefano Zampini } else { 1298f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1299f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1300f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1301f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1302f4ddd8eeSStefano Zampini } 1303f4ddd8eeSStefano Zampini } 1304f4ddd8eeSStefano Zampini 1305f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1306727cdba6SStefano Zampini /* reset primal space flags */ 1307f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1308727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 13098de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1310727cdba6SStefano Zampini /* It also sets the primal space flags */ 1311674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1312e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1313f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 13143975b054SStefano Zampini } 13155e8657edSStefano Zampini 13163975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 13175e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 13185e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 13195e8657edSStefano Zampini 13205e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 13215e8657edSStefano Zampini /* get submatrices */ 13225e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 13235e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 13245e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 13255e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 13265e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 13275e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 13283975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 13293975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 13305e8657edSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix) { 1331b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 13325e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 13335e8657edSStefano Zampini pcbddc->local_mat = matis->A; 13345e8657edSStefano Zampini } 1335b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 133699cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1337b96c3477SStefano Zampini /* SetUp Scaling operator */ 1338674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 13390c7d97c5SJed Brown } 134070cf5478SStefano Zampini 134158a03d70SStefano Zampini if (pcbddc->dbg_flag) { 134258a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 13432b510759SStefano Zampini } 13440c7d97c5SJed Brown PetscFunctionReturn(0); 13450c7d97c5SJed Brown } 13460c7d97c5SJed Brown 13470c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13480c7d97c5SJed Brown /* 134950efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 13500c7d97c5SJed Brown 13510c7d97c5SJed Brown Input Parameters: 13520f202f7eSStefano Zampini + pc - the preconditioner context 13530f202f7eSStefano Zampini - r - input vector (global) 13540c7d97c5SJed Brown 13550c7d97c5SJed Brown Output Parameter: 13560c7d97c5SJed Brown . z - output vector (global) 13570c7d97c5SJed Brown 13580c7d97c5SJed Brown Application Interface Routine: PCApply() 13590c7d97c5SJed Brown */ 13600c7d97c5SJed Brown #undef __FUNCT__ 13610c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 136253cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 13630c7d97c5SJed Brown { 13640c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 13650c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1366b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 13670c7d97c5SJed Brown PetscErrorCode ierr; 13683b03a366Sstefano_zampini const PetscScalar one = 1.0; 13693b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 13702617d88aSStefano Zampini const PetscScalar zero = 0.0; 13710c7d97c5SJed Brown 13720c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 13730c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1374b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 13750c7d97c5SJed Brown 13760c7d97c5SJed Brown PetscFunctionBegin; 137785c4d303SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1378b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 13790c7d97c5SJed Brown /* First Dirichlet solve */ 13800c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13810c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13820c7d97c5SJed Brown /* 13830c7d97c5SJed Brown Assembling right hand side for BDDC operator 1384b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1385674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 13860c7d97c5SJed Brown */ 1387b097fa66SStefano Zampini if (n_D) { 1388b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13890c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 13908eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1391b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1392b097fa66SStefano Zampini } else { 1393b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1394b097fa66SStefano Zampini } 13950c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13960c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1397674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1398b76ba322SStefano Zampini } else { 1399b097fa66SStefano Zampini if (pcbddc->switch_static) { 14000bdf917eSStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1401b097fa66SStefano Zampini } 1402674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1403b76ba322SStefano Zampini } 1404b76ba322SStefano Zampini 14052617d88aSStefano Zampini /* Apply interface preconditioner 14062617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1407dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 14082617d88aSStefano Zampini 1409674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1410674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 14110c7d97c5SJed Brown 14123b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 14130c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14140c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1415b097fa66SStefano Zampini if (n_B) { 14160c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 14178eeda7d8SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1418b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1419b097fa66SStefano Zampini ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1420b097fa66SStefano Zampini } 1421df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1422b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1423b097fa66SStefano Zampini if (pcbddc->switch_static) { 1424b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1425b097fa66SStefano Zampini } else { 1426b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1427b097fa66SStefano Zampini } 14280c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14290c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1430b097fa66SStefano Zampini } else { 1431b097fa66SStefano Zampini if (pcbddc->switch_static) { 1432b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1433b097fa66SStefano Zampini } else { 1434b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1435b097fa66SStefano Zampini } 1436b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1437b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1438b097fa66SStefano Zampini } 14390c7d97c5SJed Brown PetscFunctionReturn(0); 14400c7d97c5SJed Brown } 144150efa1b5SStefano Zampini 144250efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 144350efa1b5SStefano Zampini /* 144450efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 144550efa1b5SStefano Zampini 144650efa1b5SStefano Zampini Input Parameters: 14470f202f7eSStefano Zampini + pc - the preconditioner context 14480f202f7eSStefano Zampini - r - input vector (global) 144950efa1b5SStefano Zampini 145050efa1b5SStefano Zampini Output Parameter: 145150efa1b5SStefano Zampini . z - output vector (global) 145250efa1b5SStefano Zampini 145350efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 145450efa1b5SStefano Zampini */ 145550efa1b5SStefano Zampini #undef __FUNCT__ 145650efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 145750efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 145850efa1b5SStefano Zampini { 145950efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 146050efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1461b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 146250efa1b5SStefano Zampini PetscErrorCode ierr; 146350efa1b5SStefano Zampini const PetscScalar one = 1.0; 146450efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 146550efa1b5SStefano Zampini const PetscScalar zero = 0.0; 146650efa1b5SStefano Zampini 146750efa1b5SStefano Zampini PetscFunctionBegin; 146850efa1b5SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1469b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 147050efa1b5SStefano Zampini /* First Dirichlet solve */ 147150efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 147250efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 147350efa1b5SStefano Zampini /* 147450efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1475b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 147650efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 147750efa1b5SStefano Zampini */ 1478b097fa66SStefano Zampini if (n_D) { 1479b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 148050efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 148150efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1482b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1483b097fa66SStefano Zampini } else { 1484b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1485b097fa66SStefano Zampini } 148650efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 148750efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 148850efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 148950efa1b5SStefano Zampini } else { 1490b097fa66SStefano Zampini if (pcbddc->switch_static) { 149150efa1b5SStefano Zampini ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1492b097fa66SStefano Zampini } 149350efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 149450efa1b5SStefano Zampini } 149550efa1b5SStefano Zampini 149650efa1b5SStefano Zampini /* Apply interface preconditioner 149750efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1498dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 149950efa1b5SStefano Zampini 150050efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 150150efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 150250efa1b5SStefano Zampini 150350efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 150450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 150550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1506b097fa66SStefano Zampini if (n_B) { 150750efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 150850efa1b5SStefano Zampini if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1509b097fa66SStefano Zampini } else if (pcbddc->switch_static) { 1510b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1511b097fa66SStefano Zampini } 1512b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1513b097fa66SStefano Zampini if (!pcbddc->use_exact_dirichlet_trick) { 1514b097fa66SStefano Zampini if (pcbddc->switch_static) { 1515b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1516b097fa66SStefano Zampini } else { 1517b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1518b097fa66SStefano Zampini } 151950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 152050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1521b097fa66SStefano Zampini } else { 1522b097fa66SStefano Zampini if (pcbddc->switch_static) { 1523b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1524b097fa66SStefano Zampini } else { 1525b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1526b097fa66SStefano Zampini } 1527b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1528b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1529b097fa66SStefano Zampini } 153050efa1b5SStefano Zampini PetscFunctionReturn(0); 153150efa1b5SStefano Zampini } 1532da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1533674ae819SStefano Zampini 1534da1bb401SStefano Zampini #undef __FUNCT__ 1535da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1536da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1537da1bb401SStefano Zampini { 1538da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1539da1bb401SStefano Zampini PetscErrorCode ierr; 1540da1bb401SStefano Zampini 1541da1bb401SStefano Zampini PetscFunctionBegin; 1542da1bb401SStefano Zampini /* free data created by PCIS */ 1543da1bb401SStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 1544674ae819SStefano Zampini /* free BDDC custom data */ 1545674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1546674ae819SStefano Zampini /* destroy objects related to topography */ 1547674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1548674ae819SStefano Zampini /* free allocated graph structure */ 1549da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1550b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1551b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 155234a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1553674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 155434a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1555674ae819SStefano Zampini /* free solvers stuff */ 1556674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 155762a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 155862a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 155962a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1560906d46d4SStefano Zampini /* free stuff for change of basis hooks */ 1561906d46d4SStefano Zampini if (pcbddc->new_global_mat) { 1562906d46d4SStefano Zampini PCBDDCChange_ctx change_ctx; 1563906d46d4SStefano Zampini ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1564906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1565906d46d4SStefano Zampini ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1566906d46d4SStefano Zampini ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1567906d46d4SStefano Zampini ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1568906d46d4SStefano Zampini } 1569906d46d4SStefano Zampini ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 15703425bc38SStefano Zampini /* remove functions */ 1571906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1572674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 1573bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 15742b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1575b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 15762b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1577bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 157882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1579bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 158082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1581bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 158282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1583bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1584785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1585bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 158663602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1587bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1588bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1589bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1590bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1591a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 1592674ae819SStefano Zampini /* Free the private data structure */ 1593674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 1594da1bb401SStefano Zampini PetscFunctionReturn(0); 1595da1bb401SStefano Zampini } 15963425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 15971e6b0712SBarry Smith 15983425bc38SStefano Zampini #undef __FUNCT__ 1599a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC" 1600a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 1601a06fd7f2SStefano Zampini { 1602a06fd7f2SStefano Zampini PetscFunctionBegin; 1603a06fd7f2SStefano Zampini *change = PETSC_TRUE; 1604a06fd7f2SStefano Zampini PetscFunctionReturn(0); 1605a06fd7f2SStefano Zampini } 1606a06fd7f2SStefano Zampini 1607a06fd7f2SStefano Zampini #undef __FUNCT__ 16083425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 16093425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16103425bc38SStefano Zampini { 1611674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 1612c08af4c6SStefano Zampini Vec copy_standard_rhs; 16133425bc38SStefano Zampini PC_IS* pcis; 16143425bc38SStefano Zampini PC_BDDC* pcbddc; 16153425bc38SStefano Zampini PetscErrorCode ierr; 16160c7d97c5SJed Brown 16173425bc38SStefano Zampini PetscFunctionBegin; 16183425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 16193425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 16203425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 16213425bc38SStefano Zampini 1622c08af4c6SStefano Zampini /* 1623c08af4c6SStefano Zampini change of basis for physical rhs if needed 1624c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 1625c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 1626c08af4c6SStefano Zampini */ 1627c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1628c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1629c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 16303425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 1631c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1632c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1633fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 1634fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 1635c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1636c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1637674ae819SStefano Zampini /* Apply partition of unity */ 16383425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1639c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 16408eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 16413425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 16423425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16433425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 16443425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1645c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1646c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1647c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1648c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1649c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1650c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16513425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 16523425bc38SStefano Zampini } 1653c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 16543425bc38SStefano Zampini /* BDDC rhs */ 16553425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 16568eeda7d8SStefano Zampini if (pcbddc->switch_static) { 16573425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 16583425bc38SStefano Zampini } 16593425bc38SStefano Zampini /* apply BDDC */ 1660dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 16613425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 16623425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 16633425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 16643425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16653425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16663425bc38SStefano Zampini PetscFunctionReturn(0); 16673425bc38SStefano Zampini } 16681e6b0712SBarry Smith 16693425bc38SStefano Zampini #undef __FUNCT__ 16703425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 16713425bc38SStefano Zampini /*@ 16720f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 16733425bc38SStefano Zampini 16743425bc38SStefano Zampini Collective 16753425bc38SStefano Zampini 16763425bc38SStefano Zampini Input Parameters: 16770f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 16780f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 16793425bc38SStefano Zampini 16803425bc38SStefano Zampini Output Parameters: 16810f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 16823425bc38SStefano Zampini 16833425bc38SStefano Zampini Level: developer 16843425bc38SStefano Zampini 16853425bc38SStefano Zampini Notes: 16863425bc38SStefano Zampini 16870f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 16883425bc38SStefano Zampini @*/ 16893425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 16903425bc38SStefano Zampini { 1691674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 16923425bc38SStefano Zampini PetscErrorCode ierr; 16933425bc38SStefano Zampini 16943425bc38SStefano Zampini PetscFunctionBegin; 16953425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1696163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 16973425bc38SStefano Zampini PetscFunctionReturn(0); 16983425bc38SStefano Zampini } 16993425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 17001e6b0712SBarry Smith 17013425bc38SStefano Zampini #undef __FUNCT__ 17023425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 17033425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 17043425bc38SStefano Zampini { 1705674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17063425bc38SStefano Zampini PC_IS* pcis; 17073425bc38SStefano Zampini PC_BDDC* pcbddc; 17083425bc38SStefano Zampini PetscErrorCode ierr; 17093425bc38SStefano Zampini 17103425bc38SStefano Zampini PetscFunctionBegin; 17113425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 17123425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 17133425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 17143425bc38SStefano Zampini 17153425bc38SStefano Zampini /* apply B_delta^T */ 17163425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17173425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17183425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 17193425bc38SStefano Zampini /* compute rhs for BDDC application */ 17203425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 17218eeda7d8SStefano Zampini if (pcbddc->switch_static) { 17223425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 17233425bc38SStefano Zampini } 17243425bc38SStefano Zampini /* apply BDDC */ 1725dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 17263425bc38SStefano Zampini /* put values into standard global vector */ 17273425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17283425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17298eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 17303425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 17313425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 1732*00f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 1733*00f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 17343425bc38SStefano Zampini } 17353425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17363425bc38SStefano Zampini ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17373425bc38SStefano Zampini /* final change of basis if needed 17383425bc38SStefano Zampini Is also sums the dirichlet part removed during RHS assembling */ 17393308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 17403425bc38SStefano Zampini PetscFunctionReturn(0); 17413425bc38SStefano Zampini } 17421e6b0712SBarry Smith 17433425bc38SStefano Zampini #undef __FUNCT__ 17443425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 17453425bc38SStefano Zampini /*@ 17460f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 17473425bc38SStefano Zampini 17483425bc38SStefano Zampini Collective 17493425bc38SStefano Zampini 17503425bc38SStefano Zampini Input Parameters: 17510f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 17520f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 17533425bc38SStefano Zampini 17543425bc38SStefano Zampini Output Parameters: 17550f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 17563425bc38SStefano Zampini 17573425bc38SStefano Zampini Level: developer 17583425bc38SStefano Zampini 17593425bc38SStefano Zampini Notes: 17603425bc38SStefano Zampini 17610f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 17623425bc38SStefano Zampini @*/ 17633425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 17643425bc38SStefano Zampini { 1765674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 17663425bc38SStefano Zampini PetscErrorCode ierr; 17673425bc38SStefano Zampini 17683425bc38SStefano Zampini PetscFunctionBegin; 17693425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1770163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 17713425bc38SStefano Zampini PetscFunctionReturn(0); 17723425bc38SStefano Zampini } 17733425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 17741e6b0712SBarry Smith 1775f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 1776edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 1777f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 1778f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 1779edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 1780f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 1781674ae819SStefano Zampini 17823425bc38SStefano Zampini #undef __FUNCT__ 17833425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 17841720468bSStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, Mat *fetidp_mat, PC *fetidp_pc) 17853425bc38SStefano Zampini { 1786674ae819SStefano Zampini 1787674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 17883425bc38SStefano Zampini Mat newmat; 1789674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 17903425bc38SStefano Zampini PC newpc; 1791ce94432eSBarry Smith MPI_Comm comm; 17923425bc38SStefano Zampini PetscErrorCode ierr; 17933425bc38SStefano Zampini 17943425bc38SStefano Zampini PetscFunctionBegin; 1795ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 17963425bc38SStefano Zampini /* FETIDP linear matrix */ 17973425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 17981720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 17993425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 18003425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 18013425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 1802edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 18033425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 18043425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 18053425bc38SStefano Zampini /* FETIDP preconditioner */ 18063425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 18073425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 18083425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 18093425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 18103425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 18113425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 1812edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 18133425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 181423ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 18153425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 18163425bc38SStefano Zampini /* return pointers for objects created */ 18173425bc38SStefano Zampini *fetidp_mat=newmat; 18183425bc38SStefano Zampini *fetidp_pc=newpc; 18193425bc38SStefano Zampini PetscFunctionReturn(0); 18203425bc38SStefano Zampini } 18211e6b0712SBarry Smith 18223425bc38SStefano Zampini #undef __FUNCT__ 18233425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 18243425bc38SStefano Zampini /*@ 18250f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 18263425bc38SStefano Zampini 18273425bc38SStefano Zampini Collective 18283425bc38SStefano Zampini 18293425bc38SStefano Zampini Input Parameters: 18301720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 18311720468bSStefano Zampini - fully_redundant - true for a fully redundant set of Lagrange multipliers 183228509bceSStefano Zampini 183328509bceSStefano Zampini Output Parameters: 18340f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 18350f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 183628509bceSStefano Zampini 18373425bc38SStefano Zampini Level: developer 18383425bc38SStefano Zampini 18393425bc38SStefano Zampini Notes: 18400f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 18413425bc38SStefano Zampini 18420f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 18433425bc38SStefano Zampini @*/ 18441720468bSStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, Mat *fetidp_mat, PC *fetidp_pc) 18453425bc38SStefano Zampini { 18463425bc38SStefano Zampini PetscErrorCode ierr; 18473425bc38SStefano Zampini 18483425bc38SStefano Zampini PetscFunctionBegin; 18493425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 18503425bc38SStefano Zampini if (pc->setupcalled) { 18511720468bSStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,Mat*,PC*),(pc,fully_redundant,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 1852f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 18533425bc38SStefano Zampini PetscFunctionReturn(0); 18543425bc38SStefano Zampini } 18550c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 1856da1bb401SStefano Zampini /*MC 1857da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 18580c7d97c5SJed Brown 185928509bceSStefano Zampini An implementation of the BDDC preconditioner based on 186028509bceSStefano Zampini 186128509bceSStefano Zampini .vb 186228509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 186328509bceSStefano 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 186428509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 18650f202f7eSStefano 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 186628509bceSStefano Zampini .ve 186728509bceSStefano Zampini 186828509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 186928509bceSStefano Zampini 18700f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 187128509bceSStefano Zampini 187228509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 187328509bceSStefano Zampini 1874b6fdb6dfSStefano 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. 1875b6fdb6dfSStefano Zampini 1876c7017625SStefano 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). 187728509bceSStefano Zampini 18780f202f7eSStefano 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() 18790f202f7eSStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS() 188028509bceSStefano Zampini 18810f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 188228509bceSStefano Zampini 18830f202f7eSStefano 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. 18840f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 188528509bceSStefano Zampini 18860f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 188728509bceSStefano Zampini 18880f202f7eSStefano 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. 188928509bceSStefano Zampini 18900f202f7eSStefano 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. 18910f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 18920f202f7eSStefano Zampini 18930f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 18940f202f7eSStefano Zampini 18950f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 18960f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 18970f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 18980f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 18990f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 19000f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 19010f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 190228509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 19030f202f7eSStefano 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) 19040f202f7eSStefano 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) 19050f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 19060f202f7eSStefano 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) 19070f202f7eSStefano 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) 190828509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 190928509bceSStefano Zampini 191028509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 191128509bceSStefano Zampini .vb 191228509bceSStefano Zampini -pc_bddc_dirichlet_ 191328509bceSStefano Zampini -pc_bddc_neumann_ 191428509bceSStefano Zampini -pc_bddc_coarse_ 191528509bceSStefano Zampini .ve 19160f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 191728509bceSStefano Zampini 19180f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 191928509bceSStefano Zampini .vb 1920312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 1921312be037SStefano Zampini -pc_bddc_neumann_lN_ 1922312be037SStefano Zampini -pc_bddc_coarse_lN_ 192328509bceSStefano Zampini .ve 19240f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 19250f202f7eSStefano 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. 19260f202f7eSStefano Zampini .vb 19270f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 19280f202f7eSStefano Zampini .ve 19290f202f7eSStefano 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 1930da1bb401SStefano Zampini 1931da1bb401SStefano Zampini Level: intermediate 1932da1bb401SStefano Zampini 1933b6fdb6dfSStefano Zampini Developer notes: 1934da1bb401SStefano Zampini 1935da1bb401SStefano Zampini Contributed by Stefano Zampini 1936da1bb401SStefano Zampini 1937da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 1938da1bb401SStefano Zampini M*/ 1939b2573a8aSBarry Smith 1940da1bb401SStefano Zampini #undef __FUNCT__ 1941da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 19428cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 1943da1bb401SStefano Zampini { 1944da1bb401SStefano Zampini PetscErrorCode ierr; 1945da1bb401SStefano Zampini PC_BDDC *pcbddc; 1946da1bb401SStefano Zampini 1947da1bb401SStefano Zampini PetscFunctionBegin; 1948da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 1949b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 1950da1bb401SStefano Zampini pc->data = (void*)pcbddc; 1951da1bb401SStefano Zampini 1952da1bb401SStefano Zampini /* create PCIS data structure */ 1953da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 1954da1bb401SStefano Zampini 195547d04d0dSStefano Zampini /* BDDC customization */ 195608a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 195747d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 195847d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 195947d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 196047d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 196147d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 196247d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 1963fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 1964fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 19653301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 196647d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 1967b9d89cd5SStefano Zampini /* private */ 1968727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 19690e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 19700e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 19710e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 1972e9189074SStefano Zampini pcbddc->n_vertices = 0; 1973727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 1974fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 197568457ee5SStefano Zampini pcbddc->coarse_size = -1; 1976f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1977727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 1978f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 1979f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 1980f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 1981674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 19823972b0daSStefano Zampini pcbddc->temp_solution = 0; 1983534831adSStefano Zampini pcbddc->original_rhs = 0; 1984534831adSStefano Zampini pcbddc->local_mat = 0; 1985534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 1986b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 1987906d46d4SStefano Zampini pcbddc->new_global_mat = 0; 1988da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 1989da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 1990da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 1991da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 199215aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 199315aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 1994da1bb401SStefano Zampini pcbddc->vec1_P = 0; 1995da1bb401SStefano Zampini pcbddc->vec1_R = 0; 1996da1bb401SStefano Zampini pcbddc->vec2_R = 0; 1997da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 1998da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 1999da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2000da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2001da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2002da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2003da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2004785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2005785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2006785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 200760d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 200860d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 200963602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2010da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 201163602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2012da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 201385c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 201447d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 201547d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2016b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 20174fad6a16SStefano Zampini pcbddc->current_level = 0; 20182b510759SStefano Zampini pcbddc->max_levels = 0; 2019323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 202074e2c79eSStefano Zampini pcbddc->redistribute_coarse = 0; 2021f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 2022323d395dSStefano Zampini pcbddc->coarse_subassembling_init = 0; 2023da1bb401SStefano Zampini 2024674ae819SStefano Zampini /* create local graph structure */ 2025674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2026674ae819SStefano Zampini 2027674ae819SStefano Zampini /* scaling */ 2028674ae819SStefano Zampini pcbddc->work_scaling = 0; 202934a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2030ac632422SStefano Zampini pcbddc->faster_deluxe = PETSC_FALSE; 2031b96c3477SStefano Zampini 2032b96c3477SStefano Zampini /* create sub schurs structure */ 2033b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2034b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2035b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2036b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2037b96c3477SStefano Zampini 2038b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2039da1bb401SStefano Zampini 2040b7eb3628SStefano Zampini /* adaptivity */ 2041f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 204208122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2043f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2044b7eb3628SStefano Zampini 2045da1bb401SStefano Zampini /* function pointers */ 2046da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 204793bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2048da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2049da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2050da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 20516b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2052da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2053da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2054da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2055534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2056534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2057da1bb401SStefano Zampini 2058da1bb401SStefano Zampini /* composing function */ 2059906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2060674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 2061bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 20622b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2063b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 20642b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2065bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 206682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2067bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 206882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2069bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 207082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2071bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 207282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2073bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 207463602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2075bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2076bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2077bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2078bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2079a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2080da1bb401SStefano Zampini PetscFunctionReturn(0); 2081da1bb401SStefano Zampini } 20823425bc38SStefano Zampini 2083