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 26f3d41395Sstefano_zampini static PetscBool cited = PETSC_FALSE; 27f3d41395Sstefano_zampini static const char citation[] = 28f3d41395Sstefano_zampini "@article{ZampiniPCBDDC,\n" 29f3d41395Sstefano_zampini "author = {Stefano Zampini},\n" 30f3d41395Sstefano_zampini "title = {{PCBDDC}: A Class of Robust Dual-Primal Methods in {PETS}c},\n" 31f3d41395Sstefano_zampini "journal = {SIAM Journal on Scientific Computing},\n" 32f3d41395Sstefano_zampini "volume = {38},\n" 33f3d41395Sstefano_zampini "number = {5},\n" 34f3d41395Sstefano_zampini "pages = {S282-S306},\n" 35f3d41395Sstefano_zampini "year = {2016},\n" 36f3d41395Sstefano_zampini "doi = {10.1137/15M1025785},\n" 37f3d41395Sstefano_zampini "URL = {http://dx.doi.org/10.1137/15M1025785},\n" 38f3d41395Sstefano_zampini "eprint = {http://dx.doi.org/10.1137/15M1025785}\n" 39f3d41395Sstefano_zampini "}\n"; 40f3d41395Sstefano_zampini 410369aaf7SStefano Zampini /* temporarily declare it */ 420369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 430369aaf7SStefano Zampini 444416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 450c7d97c5SJed Brown { 460c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 470c7d97c5SJed Brown PetscErrorCode ierr; 480c7d97c5SJed Brown 490c7d97c5SJed Brown PetscFunctionBegin; 50e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 518eeda7d8SStefano Zampini /* Verbose debugging */ 52a13144ffSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 53a13144ffSStefano Zampini /* Approximate solvers */ 54c7017625SStefano 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); 55c7017625SStefano 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); 56c7017625SStefano 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); 57c7017625SStefano 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); 586b78500eSPatrick Sanan /* Primal space customization */ 5908a5cf49SStefano 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); 60be12c134Sstefano_zampini ierr = PetscOptionsInt("-pc_bddc_graph_maxcount","Maximum number of shared subdomains for a connected component","none",pcbddc->graphmaxcount,&pcbddc->graphmaxcount,NULL);CHKERRQ(ierr); 618eeda7d8SStefano 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); 628eeda7d8SStefano 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); 638eeda7d8SStefano 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); 6414f95afaSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_vertex_size","Connected components smaller or equal to vertex size will be considered as primal vertices","none",pcbddc->vertex_size,&pcbddc->vertex_size,NULL);CHKERRQ(ierr); 656661aa1dSStefano 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); 6614f95afaSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is always used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr); 678eeda7d8SStefano Zampini /* Change of basis */ 68b9b85e73SStefano 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); 69b9b85e73SStefano 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); 70674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 71674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 72674ae819SStefano Zampini } 738eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 748eeda7d8SStefano 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); 7557de7509SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_eqs_per_proc","Number of equations per process for coarse problem redistribution (significant only at the coarsest level)","none",pcbddc->coarse_eqs_per_proc,&pcbddc->coarse_eqs_per_proc,NULL);CHKERRQ(ierr); 760298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 772b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 78323d395dSStefano 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); 79674ae819SStefano 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); 80b96c3477SStefano 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); 81b96c3477SStefano 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); 82b96c3477SStefano 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); 83683d3df6SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_schur_exact","Whether or not to use the exact Schur complement instead of the reduced one (which excludes size 1 cc)","none",pcbddc->sub_schurs_exact_schur,&pcbddc->sub_schurs_exact_schur,NULL);CHKERRQ(ierr); 84bf3a8328SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_zerorows","Zero rows and columns of deluxe operators associated with primal dofs","none",pcbddc->deluxe_zerorows,&pcbddc->deluxe_zerorows,NULL);CHKERRQ(ierr); 85839e9adbSstefano_zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_singlemat","Collapse deluxe operators","none",pcbddc->deluxe_singlemat,&pcbddc->deluxe_singlemat,NULL);CHKERRQ(ierr); 86bf3a8328SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_adaptive_userdefined","Use user-defined constraints (should be attached via MatSetNearNullSpace to pmat) in addition to those adaptively generated","none",pcbddc->adaptive_userdefined,&pcbddc->adaptive_userdefined,NULL);CHKERRQ(ierr); 874c6709b3SStefano 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); 8808122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 8908122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 903301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 91b0c7d250SStefano 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); 92b3afcdbeSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_trick","Apply the benign subspace trick to saddle point problems with discontinuous pressures","none",pcbddc->benign_saddle_point,&pcbddc->benign_saddle_point,NULL);CHKERRQ(ierr); 93e9627c49SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_change","Compute the pressure change of basis explicitly","none",pcbddc->benign_change_explicit,&pcbddc->benign_change_explicit,NULL);CHKERRQ(ierr); 9427b6a85dSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_compute_correction","Compute the benign correction during PreSolve","none",pcbddc->benign_compute_correction,&pcbddc->benign_compute_correction,NULL);CHKERRQ(ierr); 95a198735bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->compute_nonetflux,&pcbddc->compute_nonetflux,NULL);CHKERRQ(ierr); 964f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 9770c64980SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_eliminate_dirichlet","Whether or not we want to eliminate dirichlet dofs during presolve","none",pcbddc->eliminate_dirdofs,&pcbddc->eliminate_dirdofs,NULL);CHKERRQ(ierr); 980c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 990c7d97c5SJed Brown PetscFunctionReturn(0); 1000c7d97c5SJed Brown } 1016b78500eSPatrick Sanan 1026b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 1036b78500eSPatrick Sanan { 1046b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 105e9627c49SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 1066b78500eSPatrick Sanan PetscErrorCode ierr; 1076b78500eSPatrick Sanan PetscBool isascii,isstring; 108e9627c49SStefano Zampini PetscSubcomm subcomm; 109e9627c49SStefano Zampini PetscViewer subviewer; 1106b78500eSPatrick Sanan 1116b78500eSPatrick Sanan PetscFunctionBegin; 1126b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 1136b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 1146b78500eSPatrick Sanan /* Nothing printed for the String viewer */ 1156b78500eSPatrick Sanan /* ASCII viewer */ 1166b78500eSPatrick Sanan if (isascii) { 1174b2aedd3SStefano Zampini PetscMPIInt color,rank,size; 118fbad9177SStefano Zampini PetscInt64 loc[7],gsum[6],gmax[6],gmin[6],totbenign; 119e9627c49SStefano Zampini PetscScalar interface_size; 120e9627c49SStefano Zampini PetscReal ratio1=0.,ratio2=0.; 121e9627c49SStefano Zampini Vec counter; 1226b78500eSPatrick Sanan 123b74ba07aSstefano_zampini if (!pc->setupcalled) { 124b74ba07aSstefano_zampini ierr = PetscViewerASCIIPrintf(viewer," Partial information available: preconditioner has not been setup yet\n");CHKERRQ(ierr); 125b74ba07aSstefano_zampini } 126efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use verbose output: %D\n",pcbddc->dbg_flag);CHKERRQ(ierr); 127efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use user-defined CSR: %D\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 128efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use local mat graph: %D\n",pcbddc->use_local_adj && !pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 129e9627c49SStefano Zampini if (pcbddc->mat_graph->twodim) { 130efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Connectivity graph topological dimension: 2\n");CHKERRQ(ierr); 131e9627c49SStefano Zampini } else { 132efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Connectivity graph topological dimension: 3\n");CHKERRQ(ierr); 133e9627c49SStefano Zampini } 134efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Graph max count: %D\n",pcbddc->graphmaxcount);CHKERRQ(ierr); 135efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use vertices: %d (vertex size %d)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 136efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 137efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 138efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 139efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 140efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 141efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 142efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 143efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 144efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 145efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 146efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 147efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr); 148efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 149efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 150efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 151efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 152efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr); 153efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 154efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 155efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 156efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_threshold,pcbddc->adaptive_selection,pcbddc->adaptive_userdefined);CHKERRQ(ierr); 157efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 158efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 159efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 160efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 161efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 162efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Coarse eqs per proc (significant at the coarsest level): %d\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 163efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr); 164efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 165efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr); 166efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr); 167b74ba07aSstefano_zampini if (!pc->setupcalled) PetscFunctionReturn(0); 1686b78500eSPatrick Sanan 169fbad9177SStefano Zampini /* compute interface size */ 170e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 171e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 172e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 173e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 174e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 176e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 177fbad9177SStefano Zampini 178fbad9177SStefano Zampini /* compute some statistics on the domain decomposition */ 179e9627c49SStefano Zampini gsum[0] = 1; 180fbad9177SStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0; 181e9627c49SStefano Zampini loc[0] = !!pcis->n; 182e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 183e9627c49SStefano Zampini loc[2] = pcis->n_B; 184e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 185345ecf6cSStefano Zampini loc[4] = pcis->n; 186fbad9177SStefano Zampini loc[5] = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0); 187fbad9177SStefano Zampini loc[6] = pcbddc->benign_n; 188fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 189fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1; 190fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 191fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT; 192fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 193fbad9177SStefano Zampini ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 194e9627c49SStefano Zampini if (pcbddc->coarse_size) { 195e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 196e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 197e9627c49SStefano Zampini } 198efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 199efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Global dofs sizes: all %d interface %d coarse %d\n",pc->pmat->rmap->N,(PetscInt)PetscRealPart(interface_size),pcbddc->coarse_size);CHKERRQ(ierr); 200efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Coarsening ratios: all/coarse %d interface/coarse %d\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 201efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Active processes : %d\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 202efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Total subdomains : %d\n",(PetscInt)gsum[5]);CHKERRQ(ierr); 203345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 204efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subs : %d\n",(PetscInt)totbenign);CHKERRQ(ierr); 205345ecf6cSStefano Zampini } 206efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Dofs type :\tMIN\tMAX\tMEAN\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 207efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Interior dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 208efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Interface dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[2],(PetscInt)gmax[2],(PetscInt)(gsum[2]/gsum[0]));CHKERRQ(ierr); 209efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Primal dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[3],(PetscInt)gmax[3],(PetscInt)(gsum[3]/gsum[0]));CHKERRQ(ierr); 210efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Local dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[4],(PetscInt)gmax[4],(PetscInt)(gsum[4]/gsum[0]));CHKERRQ(ierr); 211efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Local subs :\t%d\t%d\n",(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr); 212efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ********************************** COARSE PROBLEM DETAILS *********************************\n",pcbddc->current_level);CHKERRQ(ierr); 21327b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 214e9627c49SStefano Zampini 215fbad9177SStefano Zampini /* the coarse problem can be handled by a different communicator */ 216e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 217e9627c49SStefano Zampini else color = 0; 218e9627c49SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 2194b2aedd3SStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 220e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 2214b2aedd3SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr); 222e9627c49SStefano Zampini ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr); 223e9627c49SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 224e9627c49SStefano Zampini if (color == 1) { 225e9627c49SStefano Zampini ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr); 226e9627c49SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 227e9627c49SStefano Zampini } 228e9627c49SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 229e9627c49SStefano Zampini ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr); 230e9627c49SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 231e9627c49SStefano Zampini } 2326b78500eSPatrick Sanan PetscFunctionReturn(0); 2336b78500eSPatrick Sanan } 234a13144ffSStefano Zampini 2351e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 236a13144ffSStefano Zampini { 237a13144ffSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 238a13144ffSStefano Zampini PetscErrorCode ierr; 239a13144ffSStefano Zampini 240a13144ffSStefano Zampini PetscFunctionBegin; 241a13144ffSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 242a13144ffSStefano Zampini ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); 243a13144ffSStefano Zampini pcbddc->discretegradient = G; 244a13144ffSStefano Zampini pcbddc->nedorder = order > 0 ? order : -order; 245495a2a07SStefano Zampini pcbddc->nedfield = field; 2461e0482f5SStefano Zampini pcbddc->nedglobal = global; 2471e0482f5SStefano Zampini pcbddc->conforming = conforming; 248a13144ffSStefano Zampini PetscFunctionReturn(0); 249a13144ffSStefano Zampini } 250a13144ffSStefano Zampini 251a13144ffSStefano Zampini /*@ 252a13144ffSStefano Zampini PCBDDCSetDiscreteGradient - Sets the discrete gradient 253a13144ffSStefano Zampini 254a13144ffSStefano Zampini Collective on PC 255a13144ffSStefano Zampini 256a13144ffSStefano Zampini Input Parameters: 257a13144ffSStefano Zampini + pc - the preconditioning context 258a13144ffSStefano Zampini . G - the discrete gradient matrix (should be in AIJ format) 259a13144ffSStefano Zampini . order - the order of the Nedelec space (1 for the lowest order) 260495a2a07SStefano Zampini . field - the field id of the Nedelec dofs (not used if the fields have not been specified) 2611e0482f5SStefano Zampini . global - the type of global ordering for the rows of G 262a13144ffSStefano Zampini - conforming - whether the mesh is conforming or not 263a13144ffSStefano Zampini 264a13144ffSStefano Zampini Level: advanced 265a13144ffSStefano Zampini 266a13144ffSStefano Zampini Notes: The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry. 267495a2a07SStefano Zampini For variable order spaces, the order should be set to zero. 2681e0482f5SStefano Zampini If global is true, the rows of G should be given in global ordering for the whole dofs; 2691e0482f5SStefano Zampini if false, the ordering should be global for the Nedelec field. 2701e0482f5SStefano Zampini In the latter case, it should hold gid[i] < gid[j] iff geid[i] < geid[j], with gid the global orderding for all the dofs 2711e0482f5SStefano Zampini and geid the one for the Nedelec field. 272a13144ffSStefano Zampini 273495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal() 274a13144ffSStefano Zampini @*/ 2751e0482f5SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 276a13144ffSStefano Zampini { 277a13144ffSStefano Zampini PetscErrorCode ierr; 278a13144ffSStefano Zampini 279a13144ffSStefano Zampini PetscFunctionBegin; 280a13144ffSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 281a13144ffSStefano Zampini PetscValidHeaderSpecific(G,MAT_CLASSID,2); 282a13144ffSStefano Zampini PetscValidLogicalCollectiveInt(pc,order,3); 2831e0482f5SStefano Zampini PetscValidLogicalCollectiveInt(pc,field,4); 2841e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,global,5); 2851e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,conforming,6); 2861e0482f5SStefano Zampini PetscCheckSameComm(pc,1,G,2); 2871e0482f5SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool,PetscBool),(pc,G,order,field,global,conforming));CHKERRQ(ierr); 288a13144ffSStefano Zampini PetscFunctionReturn(0); 289a13144ffSStefano Zampini } 290a13144ffSStefano Zampini 2918ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 292a198735bSStefano Zampini { 293a198735bSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 294a198735bSStefano Zampini PetscErrorCode ierr; 2956b78500eSPatrick Sanan 296a198735bSStefano Zampini PetscFunctionBegin; 297a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr); 298a198735bSStefano Zampini ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); 299a198735bSStefano Zampini pcbddc->divudotp = divudotp; 3008ae0ca82SStefano Zampini pcbddc->divudotp_trans = trans; 301a198735bSStefano Zampini pcbddc->compute_nonetflux = PETSC_TRUE; 302a198735bSStefano Zampini if (vl2l) { 303a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr); 304fa23a32eSStefano Zampini ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); 305a198735bSStefano Zampini pcbddc->divudotp_vl2l = vl2l; 306a198735bSStefano Zampini } 307a198735bSStefano Zampini PetscFunctionReturn(0); 308a198735bSStefano Zampini } 3093d996552SStefano Zampini 310a198735bSStefano Zampini /*@ 311a198735bSStefano Zampini PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx 312a198735bSStefano Zampini 313a198735bSStefano Zampini Collective on PC 314a198735bSStefano Zampini 315a198735bSStefano Zampini Input Parameters: 316a198735bSStefano Zampini + pc - the preconditioning context 317a198735bSStefano Zampini . divudotp - the matrix (must be of type MATIS) 3188ae0ca82SStefano Zampini . trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space. 319bb05f991SStefano Zampini - vl2l - optional IS describing the local (wrt the local mat in divudotp) to local (wrt the local mat in pc->pmat) map for the velocities 320a198735bSStefano Zampini 321a198735bSStefano Zampini Level: advanced 322a198735bSStefano Zampini 3238ae0ca82SStefano Zampini Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries 324a198735bSStefano Zampini 325a198735bSStefano Zampini .seealso: PCBDDC 326a198735bSStefano Zampini @*/ 3278ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 328a198735bSStefano Zampini { 329a198735bSStefano Zampini PetscBool ismatis; 330a198735bSStefano Zampini PetscErrorCode ierr; 331a198735bSStefano Zampini 332a198735bSStefano Zampini PetscFunctionBegin; 333a198735bSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 334a198735bSStefano Zampini PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2); 335a198735bSStefano Zampini PetscCheckSameComm(pc,1,divudotp,2); 3368ae0ca82SStefano Zampini PetscValidLogicalCollectiveBool(pc,trans,3); 3378ae0ca82SStefano Zampini if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4); 338a198735bSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr); 339a198735bSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS"); 3408ae0ca82SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr); 341a198735bSStefano Zampini PetscFunctionReturn(0); 342a198735bSStefano Zampini } 3432d505d7fSStefano Zampini 3441dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 345b9b85e73SStefano Zampini { 346b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 347b9b85e73SStefano Zampini PetscErrorCode ierr; 348b9b85e73SStefano Zampini 349b9b85e73SStefano Zampini PetscFunctionBegin; 350b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 35156282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 352b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 3531dd7afcfSStefano Zampini pcbddc->change_interior = interior; 354b9b85e73SStefano Zampini PetscFunctionReturn(0); 355b9b85e73SStefano Zampini } 356b9b85e73SStefano Zampini /*@ 357906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 358b9b85e73SStefano Zampini 359b9b85e73SStefano Zampini Collective on PC 360b9b85e73SStefano Zampini 361b9b85e73SStefano Zampini Input Parameters: 362b9b85e73SStefano Zampini + pc - the preconditioning context 3631dd7afcfSStefano Zampini . change - the change of basis matrix 3641dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 365b9b85e73SStefano Zampini 366b9b85e73SStefano Zampini Level: intermediate 367b9b85e73SStefano Zampini 368b9b85e73SStefano Zampini Notes: 369b9b85e73SStefano Zampini 370b9b85e73SStefano Zampini .seealso: PCBDDC 371b9b85e73SStefano Zampini @*/ 3721dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 373b9b85e73SStefano Zampini { 374b9b85e73SStefano Zampini PetscErrorCode ierr; 375b9b85e73SStefano Zampini 376b9b85e73SStefano Zampini PetscFunctionBegin; 377b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 378b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 379906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 380906d46d4SStefano Zampini if (pc->mat) { 381906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 382906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 383906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 384e0fe2d75SToby Isaac 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); 385e0fe2d75SToby Isaac 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); 386906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 387906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 388e0fe2d75SToby Isaac 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); 389e0fe2d75SToby Isaac 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); 390906d46d4SStefano Zampini } 3911dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 392b9b85e73SStefano Zampini PetscFunctionReturn(0); 393b9b85e73SStefano Zampini } 3942d505d7fSStefano Zampini 39530368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 39630368db7SStefano Zampini { 39730368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 39856282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 39930368db7SStefano Zampini PetscErrorCode ierr; 40030368db7SStefano Zampini 40130368db7SStefano Zampini PetscFunctionBegin; 40256282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 40356282151SStefano Zampini if (pcbddc->user_primal_vertices) { 40456282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 40556282151SStefano Zampini } 40630368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 40730368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 40830368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 40956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 41030368db7SStefano Zampini PetscFunctionReturn(0); 41130368db7SStefano Zampini } 41230368db7SStefano Zampini /*@ 41330368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 41430368db7SStefano Zampini 41530368db7SStefano Zampini Collective 41630368db7SStefano Zampini 41730368db7SStefano Zampini Input Parameters: 41830368db7SStefano Zampini + pc - the preconditioning context 41930368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 42030368db7SStefano Zampini 42130368db7SStefano Zampini Level: intermediate 42230368db7SStefano Zampini 42330368db7SStefano Zampini Notes: 42430368db7SStefano Zampini Any process can list any global node 42530368db7SStefano Zampini 42630368db7SStefano Zampini .seealso: PCBDDC 42730368db7SStefano Zampini @*/ 42830368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 42930368db7SStefano Zampini { 43030368db7SStefano Zampini PetscErrorCode ierr; 43130368db7SStefano Zampini 43230368db7SStefano Zampini PetscFunctionBegin; 43330368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 43430368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 43530368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 43630368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 43730368db7SStefano Zampini PetscFunctionReturn(0); 43830368db7SStefano Zampini } 4392d505d7fSStefano Zampini 440674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 441674ae819SStefano Zampini { 442674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 44356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 444674ae819SStefano Zampini PetscErrorCode ierr; 4451e6b0712SBarry Smith 446674ae819SStefano Zampini PetscFunctionBegin; 44756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 44856282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 44956282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 45056282151SStefano Zampini } 451674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 45230368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 45330368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 45456282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 455674ae819SStefano Zampini PetscFunctionReturn(0); 456674ae819SStefano Zampini } 457674ae819SStefano Zampini /*@ 45828509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 459674ae819SStefano Zampini 46017eb1463SStefano Zampini Collective 461674ae819SStefano Zampini 462674ae819SStefano Zampini Input Parameters: 463674ae819SStefano Zampini + pc - the preconditioning context 46417eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 465674ae819SStefano Zampini 466674ae819SStefano Zampini Level: intermediate 467674ae819SStefano Zampini 468674ae819SStefano Zampini Notes: 469674ae819SStefano Zampini 470674ae819SStefano Zampini .seealso: PCBDDC 471674ae819SStefano Zampini @*/ 472674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 473674ae819SStefano Zampini { 474674ae819SStefano Zampini PetscErrorCode ierr; 475674ae819SStefano Zampini 476674ae819SStefano Zampini PetscFunctionBegin; 477674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 478674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 47917eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 480674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 481674ae819SStefano Zampini PetscFunctionReturn(0); 482674ae819SStefano Zampini } 4832d505d7fSStefano Zampini 4844fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 4854fad6a16SStefano Zampini { 4864fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4874fad6a16SStefano Zampini 4884fad6a16SStefano Zampini PetscFunctionBegin; 4894fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 4904fad6a16SStefano Zampini PetscFunctionReturn(0); 4914fad6a16SStefano Zampini } 4921e6b0712SBarry Smith 4934fad6a16SStefano Zampini /*@ 49428509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 4954fad6a16SStefano Zampini 4964fad6a16SStefano Zampini Logically collective on PC 4974fad6a16SStefano Zampini 4984fad6a16SStefano Zampini Input Parameters: 4994fad6a16SStefano Zampini + pc - the preconditioning context 50028509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 5014fad6a16SStefano Zampini 5020f202f7eSStefano Zampini Options Database Keys: 5030f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 5044fad6a16SStefano Zampini 5054fad6a16SStefano Zampini Level: intermediate 5064fad6a16SStefano Zampini 5074fad6a16SStefano Zampini Notes: 5080f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 5094fad6a16SStefano Zampini 5100f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 5114fad6a16SStefano Zampini @*/ 5124fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 5134fad6a16SStefano Zampini { 5144fad6a16SStefano Zampini PetscErrorCode ierr; 5154fad6a16SStefano Zampini 5164fad6a16SStefano Zampini PetscFunctionBegin; 5174fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5182b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 5194fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 5204fad6a16SStefano Zampini PetscFunctionReturn(0); 5214fad6a16SStefano Zampini } 5222b510759SStefano Zampini 523b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 524b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 525b8ffe317SStefano Zampini { 526b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 527b8ffe317SStefano Zampini 528b8ffe317SStefano Zampini PetscFunctionBegin; 52985c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 530b8ffe317SStefano Zampini PetscFunctionReturn(0); 531b8ffe317SStefano Zampini } 532b8ffe317SStefano Zampini 533b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 5342b510759SStefano Zampini { 5352b510759SStefano Zampini PetscErrorCode ierr; 5362b510759SStefano Zampini 5372b510759SStefano Zampini PetscFunctionBegin; 5382b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 539b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 540b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 5412b510759SStefano Zampini PetscFunctionReturn(0); 5422b510759SStefano Zampini } 5431e6b0712SBarry Smith 5442b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 5454fad6a16SStefano Zampini { 5464fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5474fad6a16SStefano Zampini 5484fad6a16SStefano Zampini PetscFunctionBegin; 5492b510759SStefano Zampini pcbddc->current_level = level; 5504fad6a16SStefano Zampini PetscFunctionReturn(0); 5514fad6a16SStefano Zampini } 5521e6b0712SBarry Smith 553b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 554b8ffe317SStefano Zampini { 555b8ffe317SStefano Zampini PetscErrorCode ierr; 556b8ffe317SStefano Zampini 557b8ffe317SStefano Zampini PetscFunctionBegin; 558b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 559b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 560b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 561b8ffe317SStefano Zampini PetscFunctionReturn(0); 562b8ffe317SStefano Zampini } 563b8ffe317SStefano Zampini 5642b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 5652b510759SStefano Zampini { 5662b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5672b510759SStefano Zampini 5682b510759SStefano Zampini PetscFunctionBegin; 5692b510759SStefano Zampini pcbddc->max_levels = levels; 5702b510759SStefano Zampini PetscFunctionReturn(0); 5712b510759SStefano Zampini } 5722b510759SStefano Zampini 5734fad6a16SStefano Zampini /*@ 57428509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 5754fad6a16SStefano Zampini 5764fad6a16SStefano Zampini Logically collective on PC 5774fad6a16SStefano Zampini 5784fad6a16SStefano Zampini Input Parameters: 5794fad6a16SStefano Zampini + pc - the preconditioning context 58028509bceSStefano Zampini - levels - the maximum number of levels (max 9) 5814fad6a16SStefano Zampini 5820f202f7eSStefano Zampini Options Database Keys: 5830f202f7eSStefano Zampini . -pc_bddc_levels 5844fad6a16SStefano Zampini 5854fad6a16SStefano Zampini Level: intermediate 5864fad6a16SStefano Zampini 5874fad6a16SStefano Zampini Notes: 5880f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 5894fad6a16SStefano Zampini 5900f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 5914fad6a16SStefano Zampini @*/ 5922b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 5934fad6a16SStefano Zampini { 5944fad6a16SStefano Zampini PetscErrorCode ierr; 5954fad6a16SStefano Zampini 5964fad6a16SStefano Zampini PetscFunctionBegin; 5974fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5982b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 599312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 6002b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 6014fad6a16SStefano Zampini PetscFunctionReturn(0); 6024fad6a16SStefano Zampini } 6031e6b0712SBarry Smith 6043b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 6053b03a366Sstefano_zampini { 6063b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 60756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6083b03a366Sstefano_zampini PetscErrorCode ierr; 6093b03a366Sstefano_zampini 6103b03a366Sstefano_zampini PetscFunctionBegin; 61156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 61256282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 61356282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 61456282151SStefano Zampini } 615785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 616785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6173b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 61836e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 61956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6203b03a366Sstefano_zampini PetscFunctionReturn(0); 6213b03a366Sstefano_zampini } 6221e6b0712SBarry Smith 6233b03a366Sstefano_zampini /*@ 62428509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 6253b03a366Sstefano_zampini 626785d1243SStefano Zampini Collective 6273b03a366Sstefano_zampini 6283b03a366Sstefano_zampini Input Parameters: 6293b03a366Sstefano_zampini + pc - the preconditioning context 630785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 6313b03a366Sstefano_zampini 6323b03a366Sstefano_zampini Level: intermediate 6333b03a366Sstefano_zampini 6340f202f7eSStefano Zampini Notes: 6350f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 6363b03a366Sstefano_zampini 6370f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 6383b03a366Sstefano_zampini @*/ 6393b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 6403b03a366Sstefano_zampini { 6413b03a366Sstefano_zampini PetscErrorCode ierr; 6423b03a366Sstefano_zampini 6433b03a366Sstefano_zampini PetscFunctionBegin; 6443b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 645674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 646785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 6473b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6483b03a366Sstefano_zampini PetscFunctionReturn(0); 6493b03a366Sstefano_zampini } 6501e6b0712SBarry Smith 65182ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 6523b03a366Sstefano_zampini { 6533b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 65456282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6553b03a366Sstefano_zampini PetscErrorCode ierr; 6563b03a366Sstefano_zampini 6573b03a366Sstefano_zampini PetscFunctionBegin; 65856282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 65956282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 66056282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 66156282151SStefano Zampini } 662785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 663785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6643b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 665785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 66656282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6673b03a366Sstefano_zampini PetscFunctionReturn(0); 6683b03a366Sstefano_zampini } 6693b03a366Sstefano_zampini 6703b03a366Sstefano_zampini /*@ 67182ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 6723b03a366Sstefano_zampini 673785d1243SStefano Zampini Collective 6743b03a366Sstefano_zampini 6753b03a366Sstefano_zampini Input Parameters: 6763b03a366Sstefano_zampini + pc - the preconditioning context 67782ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 6783b03a366Sstefano_zampini 6793b03a366Sstefano_zampini Level: intermediate 6803b03a366Sstefano_zampini 6813b03a366Sstefano_zampini Notes: 6823b03a366Sstefano_zampini 6830f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 6843b03a366Sstefano_zampini @*/ 68582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 6863b03a366Sstefano_zampini { 6873b03a366Sstefano_zampini PetscErrorCode ierr; 6883b03a366Sstefano_zampini 6893b03a366Sstefano_zampini PetscFunctionBegin; 6903b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6913b03a366Sstefano_zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 69282ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 69382ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6943b03a366Sstefano_zampini PetscFunctionReturn(0); 6953b03a366Sstefano_zampini } 6963b03a366Sstefano_zampini 69753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 6980c7d97c5SJed Brown { 6990c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 70056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 70153cdbc3dSStefano Zampini PetscErrorCode ierr; 7020c7d97c5SJed Brown 7030c7d97c5SJed Brown PetscFunctionBegin; 70456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 70556282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 70656282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 70756282151SStefano Zampini } 708785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 709785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 71053cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 71136e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 71256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7130c7d97c5SJed Brown PetscFunctionReturn(0); 7140c7d97c5SJed Brown } 7151e6b0712SBarry Smith 71657527edcSJed Brown /*@ 71728509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 71857527edcSJed Brown 719785d1243SStefano Zampini Collective 72057527edcSJed Brown 72157527edcSJed Brown Input Parameters: 72257527edcSJed Brown + pc - the preconditioning context 723785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 72457527edcSJed Brown 72557527edcSJed Brown Level: intermediate 72657527edcSJed Brown 7270f202f7eSStefano Zampini Notes: 7280f202f7eSStefano Zampini Any process can list any global node 72957527edcSJed Brown 7300f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 73157527edcSJed Brown @*/ 73253cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 7330c7d97c5SJed Brown { 7340c7d97c5SJed Brown PetscErrorCode ierr; 7350c7d97c5SJed Brown 7360c7d97c5SJed Brown PetscFunctionBegin; 7370c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 738674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 739785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 74053cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 74153cdbc3dSStefano Zampini PetscFunctionReturn(0); 74253cdbc3dSStefano Zampini } 7431e6b0712SBarry Smith 74482ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 7450c7d97c5SJed Brown { 7460c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 74756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7480c7d97c5SJed Brown PetscErrorCode ierr; 7490c7d97c5SJed Brown 7500c7d97c5SJed Brown PetscFunctionBegin; 75156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 75256282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 75356282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 75456282151SStefano Zampini } 755785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 756785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 7570c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 758785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 75956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7600c7d97c5SJed Brown PetscFunctionReturn(0); 7610c7d97c5SJed Brown } 7620c7d97c5SJed Brown 7630c7d97c5SJed Brown /*@ 76482ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 7650c7d97c5SJed Brown 766785d1243SStefano Zampini Collective 7670c7d97c5SJed Brown 7680c7d97c5SJed Brown Input Parameters: 7690c7d97c5SJed Brown + pc - the preconditioning context 77082ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 7710c7d97c5SJed Brown 7720c7d97c5SJed Brown Level: intermediate 7730c7d97c5SJed Brown 7740c7d97c5SJed Brown Notes: 7750c7d97c5SJed Brown 7760f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 7770c7d97c5SJed Brown @*/ 77882ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 7790c7d97c5SJed Brown { 7800c7d97c5SJed Brown PetscErrorCode ierr; 7810c7d97c5SJed Brown 7820c7d97c5SJed Brown PetscFunctionBegin; 7830c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7840c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 78582ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 78682ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 78753cdbc3dSStefano Zampini PetscFunctionReturn(0); 78853cdbc3dSStefano Zampini } 78953cdbc3dSStefano Zampini 790da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 791da1bb401SStefano Zampini { 792da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 793da1bb401SStefano Zampini 794da1bb401SStefano Zampini PetscFunctionBegin; 795da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 796da1bb401SStefano Zampini PetscFunctionReturn(0); 797da1bb401SStefano Zampini } 7981e6b0712SBarry Smith 799da1bb401SStefano Zampini /*@ 800785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 801da1bb401SStefano Zampini 802785d1243SStefano Zampini Collective 803785d1243SStefano Zampini 804785d1243SStefano Zampini Input Parameters: 805785d1243SStefano Zampini . pc - the preconditioning context 806785d1243SStefano Zampini 807785d1243SStefano Zampini Output Parameters: 808785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 809785d1243SStefano Zampini 810785d1243SStefano Zampini Level: intermediate 811785d1243SStefano Zampini 8120f202f7eSStefano Zampini Notes: 8130f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 814785d1243SStefano Zampini 815785d1243SStefano Zampini .seealso: PCBDDC 816785d1243SStefano Zampini @*/ 817785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 818785d1243SStefano Zampini { 819785d1243SStefano Zampini PetscErrorCode ierr; 820785d1243SStefano Zampini 821785d1243SStefano Zampini PetscFunctionBegin; 822785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 823785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 824785d1243SStefano Zampini PetscFunctionReturn(0); 825785d1243SStefano Zampini } 826785d1243SStefano Zampini 827785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 828785d1243SStefano Zampini { 829785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 830785d1243SStefano Zampini 831785d1243SStefano Zampini PetscFunctionBegin; 832785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 833785d1243SStefano Zampini PetscFunctionReturn(0); 834785d1243SStefano Zampini } 835785d1243SStefano Zampini 836da1bb401SStefano Zampini /*@ 83782ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 838da1bb401SStefano Zampini 839785d1243SStefano Zampini Collective 840da1bb401SStefano Zampini 841da1bb401SStefano Zampini Input Parameters: 84228509bceSStefano Zampini . pc - the preconditioning context 843da1bb401SStefano Zampini 844da1bb401SStefano Zampini Output Parameters: 84528509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 846da1bb401SStefano Zampini 847da1bb401SStefano Zampini Level: intermediate 848da1bb401SStefano Zampini 849da1bb401SStefano Zampini Notes: 8500f202f7eSStefano 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). 8510f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 852da1bb401SStefano Zampini 853da1bb401SStefano Zampini .seealso: PCBDDC 854da1bb401SStefano Zampini @*/ 85582ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 856da1bb401SStefano Zampini { 857da1bb401SStefano Zampini PetscErrorCode ierr; 858da1bb401SStefano Zampini 859da1bb401SStefano Zampini PetscFunctionBegin; 860da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 86182ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 862da1bb401SStefano Zampini PetscFunctionReturn(0); 863da1bb401SStefano Zampini } 8641e6b0712SBarry Smith 86553cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 86653cdbc3dSStefano Zampini { 86753cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 86853cdbc3dSStefano Zampini 86953cdbc3dSStefano Zampini PetscFunctionBegin; 87053cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 87153cdbc3dSStefano Zampini PetscFunctionReturn(0); 87253cdbc3dSStefano Zampini } 8731e6b0712SBarry Smith 87453cdbc3dSStefano Zampini /*@ 875785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 87653cdbc3dSStefano Zampini 877785d1243SStefano Zampini Collective 878785d1243SStefano Zampini 879785d1243SStefano Zampini Input Parameters: 880785d1243SStefano Zampini . pc - the preconditioning context 881785d1243SStefano Zampini 882785d1243SStefano Zampini Output Parameters: 883785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 884785d1243SStefano Zampini 885785d1243SStefano Zampini Level: intermediate 886785d1243SStefano Zampini 8870f202f7eSStefano Zampini Notes: 8880f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 889785d1243SStefano Zampini 890785d1243SStefano Zampini .seealso: PCBDDC 891785d1243SStefano Zampini @*/ 892785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 893785d1243SStefano Zampini { 894785d1243SStefano Zampini PetscErrorCode ierr; 895785d1243SStefano Zampini 896785d1243SStefano Zampini PetscFunctionBegin; 897785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 898785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 899785d1243SStefano Zampini PetscFunctionReturn(0); 900785d1243SStefano Zampini } 901785d1243SStefano Zampini 902785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 903785d1243SStefano Zampini { 904785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 905785d1243SStefano Zampini 906785d1243SStefano Zampini PetscFunctionBegin; 907785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 908785d1243SStefano Zampini PetscFunctionReturn(0); 909785d1243SStefano Zampini } 910785d1243SStefano Zampini 91153cdbc3dSStefano Zampini /*@ 91282ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 91353cdbc3dSStefano Zampini 914785d1243SStefano Zampini Collective 91553cdbc3dSStefano Zampini 91653cdbc3dSStefano Zampini Input Parameters: 91728509bceSStefano Zampini . pc - the preconditioning context 91853cdbc3dSStefano Zampini 91953cdbc3dSStefano Zampini Output Parameters: 92028509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 92153cdbc3dSStefano Zampini 92253cdbc3dSStefano Zampini Level: intermediate 92353cdbc3dSStefano Zampini 92453cdbc3dSStefano Zampini Notes: 9250f202f7eSStefano 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). 9260f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 92753cdbc3dSStefano Zampini 92853cdbc3dSStefano Zampini .seealso: PCBDDC 92953cdbc3dSStefano Zampini @*/ 93082ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 93153cdbc3dSStefano Zampini { 93253cdbc3dSStefano Zampini PetscErrorCode ierr; 93353cdbc3dSStefano Zampini 93453cdbc3dSStefano Zampini PetscFunctionBegin; 93553cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 93682ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 9370c7d97c5SJed Brown PetscFunctionReturn(0); 9380c7d97c5SJed Brown } 9391e6b0712SBarry Smith 9401a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 94136e030ebSStefano Zampini { 94236e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 943da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 94456282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 945da1bb401SStefano Zampini PetscErrorCode ierr; 94636e030ebSStefano Zampini 94736e030ebSStefano Zampini PetscFunctionBegin; 9488687889aSStefano Zampini if (!nvtxs) { 94904194a47SStefano Zampini if (copymode == PETSC_OWN_POINTER) { 95004194a47SStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 95104194a47SStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 95204194a47SStefano Zampini } 9538687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 9548687889aSStefano Zampini PetscFunctionReturn(0); 9558687889aSStefano Zampini } 95666da6bd7Sstefano_zampini if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */ 95756282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 95856282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 9592d505d7fSStefano Zampini ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9602d505d7fSStefano Zampini if (same_data) { 9612d505d7fSStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9622d505d7fSStefano Zampini } 96356282151SStefano Zampini } 96456282151SStefano Zampini } 96556282151SStefano Zampini if (!same_data) { 966674ae819SStefano Zampini /* free old CSR */ 967674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 968674ae819SStefano Zampini /* get CSR into graph structure */ 969da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 970854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 971785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 972674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 973674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 974a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 975da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 9761a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 9771a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 978a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 979a1dbd327SStefano Zampini } else if (copymode == PETSC_USE_POINTER) { 980a1dbd327SStefano Zampini mat_graph->xadj = (PetscInt*)xadj; 981a1dbd327SStefano Zampini mat_graph->adjncy = (PetscInt*)adjncy; 982a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_FALSE; 983e0fe2d75SToby Isaac } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %D",copymode); 984575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 98556282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 98656282151SStefano Zampini } 98736e030ebSStefano Zampini PetscFunctionReturn(0); 98836e030ebSStefano Zampini } 9891e6b0712SBarry Smith 99036e030ebSStefano Zampini /*@ 99154fffbccSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom. 99236e030ebSStefano Zampini 99336e030ebSStefano Zampini Not collective 99436e030ebSStefano Zampini 99536e030ebSStefano Zampini Input Parameters: 99654fffbccSStefano Zampini + pc - the preconditioning context. 99754fffbccSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the number of local dofs). 99854fffbccSStefano Zampini . xadj, adjncy - the connectivity of the dofs in CSR format. 99954fffbccSStefano Zampini - copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER. 100036e030ebSStefano Zampini 100136e030ebSStefano Zampini Level: intermediate 100236e030ebSStefano Zampini 100354fffbccSStefano Zampini Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative. 100436e030ebSStefano Zampini 100528509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 100636e030ebSStefano Zampini @*/ 10071a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 100836e030ebSStefano Zampini { 1009575ad6abSStefano Zampini void (*f)(void) = 0; 101036e030ebSStefano Zampini PetscErrorCode ierr; 101136e030ebSStefano Zampini 101236e030ebSStefano Zampini PetscFunctionBegin; 101336e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 10148687889aSStefano Zampini if (nvtxs) { 1015674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 10161633d1f0SStefano Zampini if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4); 10178687889aSStefano Zampini } 10181a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 1019575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 1020575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 1021575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 1022575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 1023575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 1024da1bb401SStefano Zampini } 102536e030ebSStefano Zampini PetscFunctionReturn(0); 102636e030ebSStefano Zampini } 10271e6b0712SBarry Smith 102863602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 102963602bcaSStefano Zampini { 103063602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 103163602bcaSStefano Zampini PetscInt i; 103256282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 103363602bcaSStefano Zampini PetscErrorCode ierr; 103463602bcaSStefano Zampini 103563602bcaSStefano Zampini PetscFunctionBegin; 103656282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 103756282151SStefano Zampini for (i=0;i<n_is;i++) { 103856282151SStefano Zampini PetscBool isequalt; 103956282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 104056282151SStefano Zampini if (!isequalt) break; 104156282151SStefano Zampini } 104256282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 104356282151SStefano Zampini } 104456282151SStefano Zampini for (i=0;i<n_is;i++) { 104556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 104656282151SStefano Zampini } 104763602bcaSStefano Zampini /* Destroy ISes if they were already set */ 104863602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 104963602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 105063602bcaSStefano Zampini } 105163602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 105263602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 105363602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 105463602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 105563602bcaSStefano Zampini } 105663602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 105763602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 105863602bcaSStefano Zampini /* allocate space then set */ 1059d02579f5SStefano Zampini if (n_is) { 1060d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 1061d02579f5SStefano Zampini } 106263602bcaSStefano Zampini for (i=0;i<n_is;i++) { 106363602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 106463602bcaSStefano Zampini } 106563602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 106663602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 106756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 106863602bcaSStefano Zampini PetscFunctionReturn(0); 106963602bcaSStefano Zampini } 107063602bcaSStefano Zampini 107163602bcaSStefano Zampini /*@ 107263602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 107363602bcaSStefano Zampini 107463602bcaSStefano Zampini Collective 107563602bcaSStefano Zampini 107663602bcaSStefano Zampini Input Parameters: 107763602bcaSStefano Zampini + pc - the preconditioning context 10780f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10790f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 108063602bcaSStefano Zampini 108163602bcaSStefano Zampini Level: intermediate 108263602bcaSStefano Zampini 10830f202f7eSStefano Zampini Notes: 10840f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 108563602bcaSStefano Zampini 108663602bcaSStefano Zampini .seealso: PCBDDC 108763602bcaSStefano Zampini @*/ 108863602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 108963602bcaSStefano Zampini { 109063602bcaSStefano Zampini PetscInt i; 109163602bcaSStefano Zampini PetscErrorCode ierr; 109263602bcaSStefano Zampini 109363602bcaSStefano Zampini PetscFunctionBegin; 109463602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 109563602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 109663602bcaSStefano Zampini for (i=0;i<n_is;i++) { 109763602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 109863602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 109963602bcaSStefano Zampini } 1100e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 110163602bcaSStefano Zampini PetscFunctionReturn(0); 110263602bcaSStefano Zampini } 110363602bcaSStefano Zampini 11049c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 11059c0446d6SStefano Zampini { 11069c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 11079c0446d6SStefano Zampini PetscInt i; 110856282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 11099c0446d6SStefano Zampini PetscErrorCode ierr; 11109c0446d6SStefano Zampini 11119c0446d6SStefano Zampini PetscFunctionBegin; 111256282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 111356282151SStefano Zampini for (i=0;i<n_is;i++) { 111456282151SStefano Zampini PetscBool isequalt; 111556282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 111656282151SStefano Zampini if (!isequalt) break; 111756282151SStefano Zampini } 111856282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 111956282151SStefano Zampini } 112056282151SStefano Zampini for (i=0;i<n_is;i++) { 112156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 112256282151SStefano Zampini } 1123da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 11249c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 11259c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 11269c0446d6SStefano Zampini } 1127d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 112863602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 112963602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 113063602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 113163602bcaSStefano Zampini } 113263602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 113363602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1134da1bb401SStefano Zampini /* allocate space then set */ 1135d02579f5SStefano Zampini if (n_is) { 1136785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1137d02579f5SStefano Zampini } 11389c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1139da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 11409c0446d6SStefano Zampini } 11419c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 114263602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 114356282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 11449c0446d6SStefano Zampini PetscFunctionReturn(0); 11459c0446d6SStefano Zampini } 11461e6b0712SBarry Smith 11479c0446d6SStefano Zampini /*@ 114863602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 11499c0446d6SStefano Zampini 115063602bcaSStefano Zampini Collective 11519c0446d6SStefano Zampini 11529c0446d6SStefano Zampini Input Parameters: 11539c0446d6SStefano Zampini + pc - the preconditioning context 11540f202f7eSStefano Zampini . n_is - number of index sets defining the fields 11550f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 11569c0446d6SStefano Zampini 11579c0446d6SStefano Zampini Level: intermediate 11589c0446d6SStefano Zampini 11590f202f7eSStefano Zampini Notes: 11600f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 11619c0446d6SStefano Zampini 11629c0446d6SStefano Zampini .seealso: PCBDDC 11639c0446d6SStefano Zampini @*/ 11649c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 11659c0446d6SStefano Zampini { 11662b510759SStefano Zampini PetscInt i; 11679c0446d6SStefano Zampini PetscErrorCode ierr; 11689c0446d6SStefano Zampini 11699c0446d6SStefano Zampini PetscFunctionBegin; 11709c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 117163602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 11722b510759SStefano Zampini for (i=0;i<n_is;i++) { 117363602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 1174a011d5a7Sstefano_zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 11752b510759SStefano Zampini } 11769c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 11779c0446d6SStefano Zampini PetscFunctionReturn(0); 11789c0446d6SStefano Zampini } 1179906d46d4SStefano Zampini 1180534831adSStefano Zampini /* 1181534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1182534831adSStefano Zampini guess if a transformation of basis approach has been selected. 11839c0446d6SStefano Zampini 1184534831adSStefano Zampini Input Parameter: 1185534831adSStefano Zampini + pc - the preconditioner contex 1186534831adSStefano Zampini 1187534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1188534831adSStefano Zampini 1189534831adSStefano Zampini Notes: 1190534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1191534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1192534831adSStefano Zampini */ 1193534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1194534831adSStefano Zampini { 1195534831adSStefano Zampini PetscErrorCode ierr; 1196534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1197534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 11983972b0daSStefano Zampini Vec used_vec; 119927b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1200534831adSStefano Zampini 1201534831adSStefano Zampini PetscFunctionBegin; 12021f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 120385c4d303SStefano Zampini if (ksp) { 12041f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 120585c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 120627b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 120727b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1208f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 12091f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 121085c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 121185c4d303SStefano Zampini } 121285c4d303SStefano Zampini } 1213fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1214fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1215fc17d649SStefano Zampini } 12161f4df5f7SStefano Zampini 121785c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 121862a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 121962a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 122062a6ff1dSStefano Zampini } 122162a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 122262a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 122362a6ff1dSStefano Zampini } 12248d00608fSStefano Zampini 122527b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 12263972b0daSStefano Zampini if (x) { 12273972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 12283972b0daSStefano Zampini used_vec = x; 12298d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 12303972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 12313972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 12323972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 123327b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1234266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1235266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1236266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 12373972b0daSStefano Zampini } 12388efcfb23SStefano Zampini 12398efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 12403972b0daSStefano Zampini if (ksp) { 1241a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 12428efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12438efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 12443972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 12453972b0daSStefano Zampini } 12463972b0daSStefano Zampini } 12473308cffdSStefano Zampini 12488d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12493972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 125070c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 12513975b054SStefano Zampini IS dirIS = NULL; 12523975b054SStefano Zampini 1253a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 12543975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 12553975b054SStefano Zampini if (dirIS) { 1256906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1257785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 12582b095fd8SStefano Zampini PetscScalar *array_x; 12592b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1260785d1243SStefano Zampini 12613972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 12623972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1263e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1264e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1265e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1266e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 126782ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 12683972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 12692b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12703972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12712fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 12723972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12732b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12743972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1275e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1276e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12778d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12781b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 12798efcfb23SStefano Zampini } 1280a07ea27aSStefano Zampini } 1281b76ba322SStefano Zampini 12828efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 12838d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 128427b6a85dSStefano Zampini /* save the original rhs */ 128527b6a85dSStefano Zampini if (save_rhs) { 128627b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 128727b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 12888d00608fSStefano Zampini } 12898d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12903972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 129127b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 12923972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 12938efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 129427b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 12957acc28cbSStefano Zampini if (ksp) { 12967acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 12977acc28cbSStefano Zampini } 12983308cffdSStefano Zampini } 12998efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1300b76ba322SStefano Zampini 1301fc17d649SStefano Zampini /* compute initial vector in benign space if needed 130227b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 130327b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 130427b6a85dSStefano Zampini if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) { 13051f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 13061f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 13070369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 13080369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 13090369aaf7SStefano Zampini } 13104fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 131127b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 13121dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 13133bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 13141f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 13151f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13161f4df5f7SStefano Zampini } 13171f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 131827b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 131927b6a85dSStefano Zampini if (save_rhs) { 132027b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 132192e3dcfbSStefano Zampini } 132227b6a85dSStefano Zampini if (pcbddc->rhs_change) { 13230369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 132427b6a85dSStefano Zampini } else { 132527b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 132627b6a85dSStefano Zampini } 13270369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 132827b6a85dSStefano Zampini } 132927b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 13300369aaf7SStefano Zampini } 13312d4c4fecSStefano Zampini 13322d4c4fecSStefano Zampini /* dbg output */ 1333a198735bSStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 13341f4df5f7SStefano Zampini Vec v; 13351f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 13361f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 13371f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1338e0fe2d75SToby Isaac ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %D: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr); 13392d4c4fecSStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr); 13401f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 13411f4df5f7SStefano Zampini } 13420369aaf7SStefano Zampini 13430369aaf7SStefano Zampini /* set initial guess if using PCG */ 13448ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 13450369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 13460369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 13471dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 134827b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 13491dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13501dd7afcfSStefano Zampini } else { 13511dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 13521dd7afcfSStefano Zampini } 13531dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13541dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13551dd7afcfSStefano Zampini } else { 13560369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13570369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13581dd7afcfSStefano Zampini } 13590369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13601dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 13611dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 13621dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13631dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13641dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 13651dd7afcfSStefano Zampini } else { 13660369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13670369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13681dd7afcfSStefano Zampini } 13690369aaf7SStefano Zampini if (ksp) { 13700369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 13710369aaf7SStefano Zampini } 13728ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_TRUE; 1373266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1374266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13750369aaf7SStefano Zampini } 1376534831adSStefano Zampini PetscFunctionReturn(0); 1377534831adSStefano Zampini } 1378906d46d4SStefano Zampini 1379534831adSStefano Zampini /* 1380534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1381534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1382534831adSStefano Zampini 1383534831adSStefano Zampini Input Parameter: 1384534831adSStefano Zampini + pc - the preconditioner contex 1385534831adSStefano Zampini 1386534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1387534831adSStefano Zampini 1388534831adSStefano Zampini Notes: 1389534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1390534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1391534831adSStefano Zampini */ 1392534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1393534831adSStefano Zampini { 1394534831adSStefano Zampini PetscErrorCode ierr; 1395534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1396534831adSStefano Zampini 1397534831adSStefano Zampini PetscFunctionBegin; 13983972b0daSStefano Zampini /* add solution removed in presolve */ 13996bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 140027b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 14013425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1402af140850Sstefano_zampini } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) { 140327b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14043425bc38SStefano Zampini } 1405af140850Sstefano_zampini /* restore to original state (not for FETI-DP) */ 1406af140850Sstefano_zampini if (ksp) pcbddc->temp_solution_used = PETSC_FALSE; 140727b6a85dSStefano Zampini } 140827b6a85dSStefano Zampini 1409266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 14108d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 141127b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 14128d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 1413af140850Sstefano_zampini } 14148efcfb23SStefano Zampini /* restore ksp guess state */ 14158efcfb23SStefano Zampini if (ksp) { 14168efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 14178ae0ca82SStefano Zampini /* reset flag for exact dirichlet trick */ 14188ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 1419af140850Sstefano_zampini } 1420534831adSStefano Zampini PetscFunctionReturn(0); 1421534831adSStefano Zampini } 1422af140850Sstefano_zampini 14230c7d97c5SJed Brown /* 14240c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 14250c7d97c5SJed Brown by setting data structures and options. 14260c7d97c5SJed Brown 14270c7d97c5SJed Brown Input Parameter: 142853cdbc3dSStefano Zampini + pc - the preconditioner context 14290c7d97c5SJed Brown 14300c7d97c5SJed Brown Application Interface Routine: PCSetUp() 14310c7d97c5SJed Brown 14320c7d97c5SJed Brown Notes: 14330c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 14340c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 14350c7d97c5SJed Brown */ 143653cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 14370c7d97c5SJed Brown { 14380c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1439c703fcc7SStefano Zampini PCBDDCSubSchurs sub_schurs; 14405e8657edSStefano Zampini Mat_IS* matis; 144108122e43SStefano Zampini MatNullSpace nearnullspace; 144235509ce9Sstefano_zampini Mat lA; 144335509ce9Sstefano_zampini IS lP,zerodiag = NULL; 144491e8d312SStefano Zampini PetscInt nrows,ncols; 1445c703fcc7SStefano Zampini PetscBool computesubschurs; 14468de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 14475e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 1448c703fcc7SStefano Zampini PetscErrorCode ierr; 14490c7d97c5SJed Brown 14500c7d97c5SJed Brown PetscFunctionBegin; 14515e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 14526c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 145391e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 14546c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 14555e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1456f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 14573b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 145871582508SStefano Zampini Also, BDDC builds its own KSP for the Dirichlet problem */ 1459a485753aSstefano_zampini if (!pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE; 1460c83e1ba7SStefano Zampini if (pcbddc->recompute_topography) { 1461c83e1ba7SStefano Zampini pcbddc->graphanalyzed = PETSC_FALSE; 1462c83e1ba7SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1463c83e1ba7SStefano Zampini } else { 14648de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1465c83e1ba7SStefano Zampini } 1466b087196eSStefano Zampini 1467b087196eSStefano Zampini /* check parameters' compatibility */ 1468b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 14699d54b7f4SStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0); 1470bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1471862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1472862806e4SStefano Zampini 14735a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 147416909a7fSStefano Zampini if (pcbddc->switch_static) { 147516909a7fSStefano Zampini PetscBool ismatis; 147616909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 147716909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 147816909a7fSStefano Zampini } 147916909a7fSStefano Zampini 148071582508SStefano Zampini /* activate all connected components if the netflux has been requested */ 1481bb05f991SStefano Zampini if (pcbddc->compute_nonetflux) { 1482bb05f991SStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 1483bb05f991SStefano Zampini pcbddc->use_edges = PETSC_TRUE; 1484bb05f991SStefano Zampini pcbddc->use_faces = PETSC_TRUE; 1485bb05f991SStefano Zampini } 1486bb05f991SStefano Zampini 1487f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 148870cf5478SStefano Zampini if (pcbddc->dbg_flag) { 148970cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 149058a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 14911575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1492f4ddd8eeSStefano Zampini } 149358a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1494f4ddd8eeSStefano Zampini } 1495f4ddd8eeSStefano Zampini 1496c703fcc7SStefano Zampini /* process topology information */ 149771582508SStefano Zampini if (pcbddc->recompute_topography) { 149871582508SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 1499c83e1ba7SStefano Zampini /* detect local disconnected subdomains if requested (use matis->A) */ 1500c83e1ba7SStefano Zampini if (pcbddc->detect_disconnected) { 1501c80a6c00SStefano Zampini IS primalv; 1502c83e1ba7SStefano Zampini PetscInt i; 1503c80a6c00SStefano Zampini 1504c83e1ba7SStefano Zampini for (i=0;i<pcbddc->n_local_subs;i++) { 1505c83e1ba7SStefano Zampini ierr = ISDestroy(&pcbddc->local_subs[i]);CHKERRQ(ierr); 1506c83e1ba7SStefano Zampini } 1507c83e1ba7SStefano Zampini ierr = PetscFree(pcbddc->local_subs);CHKERRQ(ierr); 1508c80a6c00SStefano Zampini ierr = PCBDDCDetectDisconnectedComponents(pc,&pcbddc->n_local_subs,&pcbddc->local_subs,&primalv);CHKERRQ(ierr); 1509c80a6c00SStefano Zampini if (primalv) { 1510c80a6c00SStefano Zampini if (pcbddc->user_primal_vertices_local) { 1511*956e2312SStefano Zampini IS list[2], newp; 1512c80a6c00SStefano Zampini 1513*956e2312SStefano Zampini list[0] = primalv; 1514*956e2312SStefano Zampini list[1] = pcbddc->user_primal_vertices_local; 1515c80a6c00SStefano Zampini ierr = ISConcatenate(PetscObjectComm((PetscObject)pc),2,list,&newp);CHKERRQ(ierr); 1516c80a6c00SStefano Zampini ierr = ISDestroy(&list[0]);CHKERRQ(ierr); 1517c80a6c00SStefano Zampini ierr = ISDestroy(&list[1]);CHKERRQ(ierr); 1518c80a6c00SStefano Zampini pcbddc->user_primal_vertices_local = newp; 1519c80a6c00SStefano Zampini } else { 1520c80a6c00SStefano Zampini pcbddc->user_primal_vertices_local = primalv; 1521c80a6c00SStefano Zampini } 1522c80a6c00SStefano Zampini } 1523c83e1ba7SStefano Zampini } 1524c703fcc7SStefano Zampini if (pcbddc->discretegradient) { 1525a13144ffSStefano Zampini ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr); 1526a13144ffSStefano Zampini } 1527c703fcc7SStefano Zampini } 1528a13144ffSStefano Zampini 1529c703fcc7SStefano Zampini /* change basis if requested by the user */ 15305e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 15315e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 15325e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 15335e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 15345e8657edSStefano Zampini } else { 1535b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15365e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15375e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1538d16cbb6bSStefano Zampini } 1539d16cbb6bSStefano Zampini 15404f1b2e48SStefano Zampini /* 1541c703fcc7SStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick 15424f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 15434f1b2e48SStefano Zampini */ 15441dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1545d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 15469f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 15479f47a83aSStefano Zampini 154805b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1549669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 1550339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1551a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1552c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 15531dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 15541dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 15551dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 15561dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 15571dd7afcfSStefano Zampini } else { 1558a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 15591dd7afcfSStefano Zampini } 1560a3df083aSStefano Zampini } else { 15619f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1562674ae819SStefano Zampini } 1563a3df083aSStefano Zampini } 156427b6a85dSStefano Zampini 1565c703fcc7SStefano Zampini /* propagate relevant information -> TODO remove*/ 15664f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 15673301b35fSStefano Zampini if (matis->A->symmetric_set) { 15683301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 15693301b35fSStefano Zampini } 1570e496cd5dSStefano Zampini #endif 157106a4e24aSStefano Zampini if (matis->A->symmetric_set) { 157206a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 157306a4e24aSStefano Zampini } 157406a4e24aSStefano Zampini if (matis->A->spd_set) { 157506a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 157606a4e24aSStefano Zampini } 1577e496cd5dSStefano Zampini 15785e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 15795e8657edSStefano Zampini { 15805e8657edSStefano Zampini Mat temp_mat; 15815e8657edSStefano Zampini 15825e8657edSStefano Zampini temp_mat = matis->A; 15835e8657edSStefano Zampini matis->A = pcbddc->local_mat; 15845e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 15855e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15865e8657edSStefano Zampini matis->A = temp_mat; 15875e8657edSStefano Zampini } 1588684f6988SStefano Zampini 158981d14e9dSStefano Zampini /* Analyze interface */ 159064ac59b8SStefano Zampini if (!pcbddc->graphanalyzed) { 1591674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15928de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1593345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 15944247aa23Sstefano_zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1595345ecf6cSStefano Zampini } 1596a198735bSStefano Zampini if (pcbddc->compute_nonetflux) { 1597669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1598669cc0f4SStefano Zampini 159921ef3d20SStefano Zampini if (!pcbddc->divudotp) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Missing divudotp operator"); 16008ae0ca82SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 160171582508SStefano Zampini /* TODO what if a nearnullspace is already attached? */ 1602669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1603669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1604669cc0f4SStefano Zampini } 1605674ae819SStefano Zampini } 1606fb8d54d4SStefano Zampini 16075408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 16085408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 16095408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 1610ff1f7e73Sstefano_zampini if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) { 16115408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 161209f581a4SStefano Zampini } 16134f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 161406f24817SStefano Zampini 1615b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1616c703fcc7SStefano Zampini if (computesubschurs && pcbddc->recompute_topography) { 161708122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1618b1b3d7a2SStefano Zampini } 16199d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 16209d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 16219d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16229d54b7f4SStefano Zampini } 1623c703fcc7SStefano Zampini 1624c703fcc7SStefano Zampini /* finish setup solvers and do adaptive selection of constraints */ 1625b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1626b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 16272070dbb6SStefano Zampini if (computesubschurs) { 162808122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 16292070dbb6SStefano Zampini } 1630d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1631d5574798SStefano Zampini } else { 1632d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 16332070dbb6SStefano Zampini if (computesubschurs) { 1634d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1635d5574798SStefano Zampini } 16362070dbb6SStefano Zampini } 163708122e43SStefano Zampini if (pcbddc->adaptive_selection) { 163808122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 16398de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1640b7eb3628SStefano Zampini } 1641684f6988SStefano Zampini 1642f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1643fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1644f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1645f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1646f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1647f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1648f4ddd8eeSStefano Zampini } else { 1649f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1650f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1651f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1652165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1653f4ddd8eeSStefano Zampini PetscInt i; 1654165b64e2SStefano Zampini const Vec *nearnullvecs; 1655165b64e2SStefano Zampini PetscObjectState state; 1656165b64e2SStefano Zampini PetscInt nnsp_size; 1657165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1658f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1659f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1660165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1661f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1662f4ddd8eeSStefano Zampini break; 1663f4ddd8eeSStefano Zampini } 1664f4ddd8eeSStefano Zampini } 1665f4ddd8eeSStefano Zampini } 1666f4ddd8eeSStefano Zampini } 1667f4ddd8eeSStefano Zampini } else { 1668f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1669f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1670f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1671f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1672f4ddd8eeSStefano Zampini } 1673f4ddd8eeSStefano Zampini } 1674f4ddd8eeSStefano Zampini 1675f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1676727cdba6SStefano Zampini /* reset primal space flags */ 1677f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1678727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 16798de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1680727cdba6SStefano Zampini /* It also sets the primal space flags */ 1681674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 16829543d0ffSStefano Zampini } 1683e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1684f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 16855e8657edSStefano Zampini 16865e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16875e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 16885e8657edSStefano Zampini 16895e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16904f1b2e48SStefano Zampini if (pcbddc->benign_change) { 16911dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1692c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1693c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1694c263805aSStefano Zampini } 16955e8657edSStefano Zampini /* get submatrices */ 16965e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16975e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16985e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16997dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 17007dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 17017dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 17023975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 17033975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 17049c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1705b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 17065e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 17075e8657edSStefano Zampini pcbddc->local_mat = matis->A; 17085e8657edSStefano Zampini } 170935509ce9Sstefano_zampini 171035509ce9Sstefano_zampini /* interface pressure block row for B_C */ 171135509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr); 171235509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr); 171335509ce9Sstefano_zampini if (lA && lP) { 171435509ce9Sstefano_zampini PC_IS* pcis = (PC_IS*)pc->data; 171535509ce9Sstefano_zampini Mat B_BI,B_BB,Bt_BI,Bt_BB; 171635509ce9Sstefano_zampini PetscBool issym; 171735509ce9Sstefano_zampini ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr); 17186cc1294bSstefano_zampini if (issym) { 17197dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17207dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 172135509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr); 172235509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr); 172335509ce9Sstefano_zampini } else { 17247dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17257dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 17267dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr); 17277dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr); 172835509ce9Sstefano_zampini } 172935509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr); 173035509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr); 173135509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr); 173235509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr); 173335509ce9Sstefano_zampini ierr = MatDestroy(&B_BI);CHKERRQ(ierr); 173435509ce9Sstefano_zampini ierr = MatDestroy(&B_BB);CHKERRQ(ierr); 173535509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr); 173635509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr); 173735509ce9Sstefano_zampini } 173835509ce9Sstefano_zampini 1739b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 174099cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1741b96c3477SStefano Zampini /* SetUp Scaling operator */ 17429d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1743674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 17440c7d97c5SJed Brown } 1745c703fcc7SStefano Zampini 17461dd7afcfSStefano Zampini /* mark topography as done */ 174756282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 17480369aaf7SStefano Zampini 17491dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 17501dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 17511dd7afcfSStefano Zampini 175258a03d70SStefano Zampini if (pcbddc->dbg_flag) { 175358a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 17542b510759SStefano Zampini } 17550c7d97c5SJed Brown PetscFunctionReturn(0); 17560c7d97c5SJed Brown } 17570c7d97c5SJed Brown 17580c7d97c5SJed Brown /* 175950efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 17600c7d97c5SJed Brown 17610c7d97c5SJed Brown Input Parameters: 17620f202f7eSStefano Zampini + pc - the preconditioner context 17630f202f7eSStefano Zampini - r - input vector (global) 17640c7d97c5SJed Brown 17650c7d97c5SJed Brown Output Parameter: 17660c7d97c5SJed Brown . z - output vector (global) 17670c7d97c5SJed Brown 17680c7d97c5SJed Brown Application Interface Routine: PCApply() 17690c7d97c5SJed Brown */ 177053cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 17710c7d97c5SJed Brown { 17720c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 17730c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1774b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 17750c7d97c5SJed Brown PetscErrorCode ierr; 17763b03a366Sstefano_zampini const PetscScalar one = 1.0; 17773b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 17782617d88aSStefano Zampini const PetscScalar zero = 0.0; 17790c7d97c5SJed Brown 17800c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 17810c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1782b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 17830c7d97c5SJed Brown 17840c7d97c5SJed Brown PetscFunctionBegin; 1785f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 17861dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 17871dd7afcfSStefano Zampini Vec swap; 178827b6a85dSStefano Zampini 178927b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 17901dd7afcfSStefano Zampini swap = pcbddc->work_change; 17911dd7afcfSStefano Zampini pcbddc->work_change = r; 17921dd7afcfSStefano Zampini r = swap; 17931dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 17949cc2a9b1Sstefano_zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 17951dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 17961dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 17971dd7afcfSStefano Zampini } 17981dd7afcfSStefano Zampini } 179927b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1800015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1801efc2fbd9SStefano Zampini } 18028ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1803b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 18040c7d97c5SJed Brown /* First Dirichlet solve */ 18050c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18060c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18070c7d97c5SJed Brown /* 18080c7d97c5SJed Brown Assembling right hand side for BDDC operator 1809b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1810674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 18110c7d97c5SJed Brown */ 1812b097fa66SStefano Zampini if (n_D) { 1813b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 18140c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 181516909a7fSStefano Zampini if (pcbddc->switch_static) { 181616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 181716909a7fSStefano Zampini 181816909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 181916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 182016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 182116909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 182216909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182316909a7fSStefano Zampini } else { 182416909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182516909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 182616909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182716909a7fSStefano Zampini } 182816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 182916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183216909a7fSStefano Zampini } else { 1833b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 183416909a7fSStefano Zampini } 1835b097fa66SStefano Zampini } else { 1836b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1837b097fa66SStefano Zampini } 18380c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18390c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1840674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1841b76ba322SStefano Zampini } else { 18424fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1843674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1844b76ba322SStefano Zampini } 18454fee134fSStefano Zampini } 1846b76ba322SStefano Zampini 18472617d88aSStefano Zampini /* Apply interface preconditioner 18482617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1849dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 18502617d88aSStefano Zampini 1851674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1852674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 18530c7d97c5SJed Brown 18543b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 18550c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18560c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1857b097fa66SStefano Zampini if (n_B) { 185816909a7fSStefano Zampini if (pcbddc->switch_static) { 185916909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 186016909a7fSStefano Zampini 186116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186516909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 186616909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186716909a7fSStefano Zampini } else { 186816909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186916909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 187016909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 187116909a7fSStefano Zampini } 187216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187416909a7fSStefano Zampini } else { 18750c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 187616909a7fSStefano Zampini } 187716909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 187816909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 187916909a7fSStefano Zampini 188016909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 188116909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 188216909a7fSStefano Zampini } else { 188316909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 188416909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 188516909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 188616909a7fSStefano Zampini } 1887b097fa66SStefano Zampini } 1888df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1889efc2fbd9SStefano Zampini 18908ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1891b097fa66SStefano Zampini if (pcbddc->switch_static) { 1892b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1893b097fa66SStefano Zampini } else { 1894b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1895b097fa66SStefano Zampini } 18960c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18970c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1898b097fa66SStefano Zampini } else { 1899b097fa66SStefano Zampini if (pcbddc->switch_static) { 1900b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1901b097fa66SStefano Zampini } else { 1902b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1903b097fa66SStefano Zampini } 1904b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1905b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1906b097fa66SStefano Zampini } 190727b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 19081dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 19091dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 19101dd7afcfSStefano Zampini } 1911015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1912efc2fbd9SStefano Zampini } 19131f4df5f7SStefano Zampini 19141dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1915f913dca9SStefano Zampini pcbddc->work_change = r; 19161dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19171dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19181dd7afcfSStefano Zampini } 19190c7d97c5SJed Brown PetscFunctionReturn(0); 19200c7d97c5SJed Brown } 192150efa1b5SStefano Zampini 192250efa1b5SStefano Zampini /* 192350efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 192450efa1b5SStefano Zampini 192550efa1b5SStefano Zampini Input Parameters: 19260f202f7eSStefano Zampini + pc - the preconditioner context 19270f202f7eSStefano Zampini - r - input vector (global) 192850efa1b5SStefano Zampini 192950efa1b5SStefano Zampini Output Parameter: 193050efa1b5SStefano Zampini . z - output vector (global) 193150efa1b5SStefano Zampini 193250efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 193350efa1b5SStefano Zampini */ 193450efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 193550efa1b5SStefano Zampini { 193650efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 193750efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1938b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 193950efa1b5SStefano Zampini PetscErrorCode ierr; 194050efa1b5SStefano Zampini const PetscScalar one = 1.0; 194150efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 194250efa1b5SStefano Zampini const PetscScalar zero = 0.0; 194350efa1b5SStefano Zampini 194450efa1b5SStefano Zampini PetscFunctionBegin; 1945f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 19461dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19471dd7afcfSStefano Zampini Vec swap; 194827b6a85dSStefano Zampini 194927b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 19501dd7afcfSStefano Zampini swap = pcbddc->work_change; 19511dd7afcfSStefano Zampini pcbddc->work_change = r; 19521dd7afcfSStefano Zampini r = swap; 195327b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 19548ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 195527b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 195627b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 19571dd7afcfSStefano Zampini } 195827b6a85dSStefano Zampini } 195927b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1960537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1961537c1cdfSStefano Zampini } 19628ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1963b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 196450efa1b5SStefano Zampini /* First Dirichlet solve */ 196550efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196650efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196750efa1b5SStefano Zampini /* 196850efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1969b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 197050efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 197150efa1b5SStefano Zampini */ 1972b097fa66SStefano Zampini if (n_D) { 1973b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 197450efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 197516909a7fSStefano Zampini if (pcbddc->switch_static) { 197616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 197716909a7fSStefano Zampini 197816909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 197916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 198016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 198116909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 198216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 198316909a7fSStefano Zampini } else { 198416909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 198516909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 198616909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 198716909a7fSStefano Zampini } 198816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 198916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199216909a7fSStefano Zampini } else { 1993b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 199416909a7fSStefano Zampini } 1995b097fa66SStefano Zampini } else { 1996b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1997b097fa66SStefano Zampini } 199850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 199950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 200050efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 200150efa1b5SStefano Zampini } else { 200250efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 200350efa1b5SStefano Zampini } 200450efa1b5SStefano Zampini 200550efa1b5SStefano Zampini /* Apply interface preconditioner 200650efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 2007dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 200850efa1b5SStefano Zampini 200950efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 201050efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 201150efa1b5SStefano Zampini 201250efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 201350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 201450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2015b097fa66SStefano Zampini if (n_B) { 201616909a7fSStefano Zampini if (pcbddc->switch_static) { 201716909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 201816909a7fSStefano Zampini 201916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 202416909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202516909a7fSStefano Zampini } else { 202616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202716909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 202816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202916909a7fSStefano Zampini } 203016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 203116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 203216909a7fSStefano Zampini } else { 203350efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 203416909a7fSStefano Zampini } 203516909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 203616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 203716909a7fSStefano Zampini 203816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 203916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 204016909a7fSStefano Zampini } else { 204116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 204216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 204316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 204416909a7fSStefano Zampini } 2045b097fa66SStefano Zampini } 2046b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 20478ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2048b097fa66SStefano Zampini if (pcbddc->switch_static) { 2049b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 2050b097fa66SStefano Zampini } else { 2051b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 2052b097fa66SStefano Zampini } 205350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 205450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2055b097fa66SStefano Zampini } else { 2056b097fa66SStefano Zampini if (pcbddc->switch_static) { 2057b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 2058b097fa66SStefano Zampini } else { 2059b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 2060b097fa66SStefano Zampini } 2061b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2062b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2063b097fa66SStefano Zampini } 206427b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 2065537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 2066537c1cdfSStefano Zampini } 20671dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2068f913dca9SStefano Zampini pcbddc->work_change = r; 20691dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 20701dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 20711dd7afcfSStefano Zampini } 207250efa1b5SStefano Zampini PetscFunctionReturn(0); 207350efa1b5SStefano Zampini } 2074674ae819SStefano Zampini 20759326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc) 2076da1bb401SStefano Zampini { 2077da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 20789326c5c6Sstefano_zampini PC_IS *pcis = (PC_IS*)pc->data; 20799326c5c6Sstefano_zampini KSP kspD,kspR,kspC; 2080da1bb401SStefano Zampini PetscErrorCode ierr; 2081da1bb401SStefano Zampini 2082da1bb401SStefano Zampini PetscFunctionBegin; 2083674ae819SStefano Zampini /* free BDDC custom data */ 2084674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2085674ae819SStefano Zampini /* destroy objects related to topography */ 2086674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 208734a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2088674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 2089674ae819SStefano Zampini /* free solvers stuff */ 2090674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 209162a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 209262a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 209362a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 20941dd7afcfSStefano Zampini /* free data created by PCIS */ 20951dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 20969326c5c6Sstefano_zampini 20979326c5c6Sstefano_zampini /* restore defaults */ 20989326c5c6Sstefano_zampini kspD = pcbddc->ksp_D; 20999326c5c6Sstefano_zampini kspR = pcbddc->ksp_R; 21009326c5c6Sstefano_zampini kspC = pcbddc->coarse_ksp; 21019326c5c6Sstefano_zampini ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr); 21029326c5c6Sstefano_zampini pcis->n_neigh = -1; 21039326c5c6Sstefano_zampini pcis->scaling_factor = 1.0; 21049326c5c6Sstefano_zampini pcis->reusesubmatrices = PETSC_TRUE; 21059326c5c6Sstefano_zampini pcbddc->use_local_adj = PETSC_TRUE; 21069326c5c6Sstefano_zampini pcbddc->use_vertices = PETSC_TRUE; 21079326c5c6Sstefano_zampini pcbddc->use_edges = PETSC_TRUE; 21089326c5c6Sstefano_zampini pcbddc->symmetric_primal = PETSC_TRUE; 21099326c5c6Sstefano_zampini pcbddc->vertex_size = 1; 21109326c5c6Sstefano_zampini pcbddc->recompute_topography = PETSC_TRUE; 21119326c5c6Sstefano_zampini pcbddc->coarse_size = -1; 21129326c5c6Sstefano_zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 21139326c5c6Sstefano_zampini pcbddc->coarsening_ratio = 8; 21149326c5c6Sstefano_zampini pcbddc->coarse_eqs_per_proc = 1; 21159326c5c6Sstefano_zampini pcbddc->benign_compute_correction = PETSC_TRUE; 21169326c5c6Sstefano_zampini pcbddc->nedfield = -1; 21179326c5c6Sstefano_zampini pcbddc->nedglobal = PETSC_TRUE; 21189326c5c6Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 21199326c5c6Sstefano_zampini pcbddc->sub_schurs_layers = -1; 21209326c5c6Sstefano_zampini pcbddc->ksp_D = kspD; 21219326c5c6Sstefano_zampini pcbddc->ksp_R = kspR; 21229326c5c6Sstefano_zampini pcbddc->coarse_ksp = kspC; 21239326c5c6Sstefano_zampini PetscFunctionReturn(0); 21249326c5c6Sstefano_zampini } 21259326c5c6Sstefano_zampini 21269326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc) 21279326c5c6Sstefano_zampini { 21289326c5c6Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 21299326c5c6Sstefano_zampini PetscErrorCode ierr; 21309326c5c6Sstefano_zampini 21319326c5c6Sstefano_zampini PetscFunctionBegin; 21329326c5c6Sstefano_zampini ierr = PCReset_BDDC(pc);CHKERRQ(ierr); 21339326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr); 21349326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr); 21359326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr); 2136a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr); 2137a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr); 2138906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2139674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 214030368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2141bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 21422b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2143b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 21442b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2145bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 214682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2147bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 214882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2149bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 215082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2151bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2152785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2153bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 215463602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2155bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2156bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2157bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2158bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2159a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2160674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2161da1bb401SStefano Zampini PetscFunctionReturn(0); 2162da1bb401SStefano Zampini } 21631e6b0712SBarry Smith 2164a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2165a06fd7f2SStefano Zampini { 2166a06fd7f2SStefano Zampini PetscFunctionBegin; 2167a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2168a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2169a06fd7f2SStefano Zampini } 2170a06fd7f2SStefano Zampini 21713425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21723425bc38SStefano Zampini { 2173674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2174266e20e9SStefano Zampini Vec work; 21753425bc38SStefano Zampini PC_IS* pcis; 21763425bc38SStefano Zampini PC_BDDC* pcbddc; 21773425bc38SStefano Zampini PetscErrorCode ierr; 21780c7d97c5SJed Brown 21793425bc38SStefano Zampini PetscFunctionBegin; 21803425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21813425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21823425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21833425bc38SStefano Zampini 2184229984c5Sstefano_zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 2185229984c5Sstefano_zampini /* copy rhs since we may change it during PCPreSolve_BDDC */ 2186229984c5Sstefano_zampini if (!pcbddc->original_rhs) { 2187229984c5Sstefano_zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 2188229984c5Sstefano_zampini } 21896cc1294bSstefano_zampini if (mat_ctx->rhs_flip) { 21906cc1294bSstefano_zampini ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr); 21916cc1294bSstefano_zampini } else { 2192229984c5Sstefano_zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 21936cc1294bSstefano_zampini } 2194af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2195229984c5Sstefano_zampini /* interface pressure rhs */ 2196022d8d2bSstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2197022d8d2bSstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2198229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2199229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22006cc1294bSstefano_zampini if (!mat_ctx->rhs_flip) { 2201229984c5Sstefano_zampini ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr); 2202229984c5Sstefano_zampini } 22036cc1294bSstefano_zampini } 2204c08af4c6SStefano Zampini /* 2205c08af4c6SStefano Zampini change of basis for physical rhs if needed 2206c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2207c08af4c6SStefano Zampini */ 22083738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2209fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 22103738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 22113738a8e6SStefano Zampini work = pcbddc->work_change; 2212fc17d649SStefano Zampini } else { 22133738a8e6SStefano Zampini work = pcbddc->original_rhs; 2214fc17d649SStefano Zampini } 22153425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2216266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2217266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2218fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2219fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2220266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2221266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2222674ae819SStefano Zampini /* Apply partition of unity */ 22233425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2224266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 22258eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 22263425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 22273425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22283425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 22293425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2230266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2231266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2232266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2233266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2234266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2235266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22363425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 22373425bc38SStefano Zampini } 22383425bc38SStefano Zampini /* BDDC rhs */ 22393425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 22408eeda7d8SStefano Zampini if (pcbddc->switch_static) { 22413425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22423425bc38SStefano Zampini } 22433425bc38SStefano Zampini /* apply BDDC */ 2244229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2245dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2246266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2247229984c5Sstefano_zampini 22483425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 22493425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 22503425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22513425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2252229984c5Sstefano_zampini /* Add contribution to interface pressures */ 2253229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2254229984c5Sstefano_zampini ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr); 2255229984c5Sstefano_zampini if (pcbddc->switch_static) { 2256229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr); 2257229984c5Sstefano_zampini } 2258229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2259229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2260229984c5Sstefano_zampini } 22613425bc38SStefano Zampini PetscFunctionReturn(0); 22623425bc38SStefano Zampini } 22631e6b0712SBarry Smith 22643425bc38SStefano Zampini /*@ 22650f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 22663425bc38SStefano Zampini 22673425bc38SStefano Zampini Collective 22683425bc38SStefano Zampini 22693425bc38SStefano Zampini Input Parameters: 22700f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 22710f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 22723425bc38SStefano Zampini 22733425bc38SStefano Zampini Output Parameters: 22740f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 22753425bc38SStefano Zampini 22763425bc38SStefano Zampini Level: developer 22773425bc38SStefano Zampini 22783425bc38SStefano Zampini Notes: 22793425bc38SStefano Zampini 22800f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 22813425bc38SStefano Zampini @*/ 22823425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 22833425bc38SStefano Zampini { 2284674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22853425bc38SStefano Zampini PetscErrorCode ierr; 22863425bc38SStefano Zampini 22873425bc38SStefano Zampini PetscFunctionBegin; 2288266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2289266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2290266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 22913425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2292163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 22933425bc38SStefano Zampini PetscFunctionReturn(0); 22943425bc38SStefano Zampini } 22951e6b0712SBarry Smith 22963425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22973425bc38SStefano Zampini { 2298674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22993425bc38SStefano Zampini PC_IS* pcis; 23003425bc38SStefano Zampini PC_BDDC* pcbddc; 23013425bc38SStefano Zampini PetscErrorCode ierr; 2302229984c5Sstefano_zampini Vec work; 23033425bc38SStefano Zampini 23043425bc38SStefano Zampini PetscFunctionBegin; 23053425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 23063425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 23073425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 23083425bc38SStefano Zampini 23093425bc38SStefano Zampini /* apply B_delta^T */ 2310af140850Sstefano_zampini ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr); 23113425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23123425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23133425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 2314229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2315229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2316229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2317229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 2318229984c5Sstefano_zampini } 2319229984c5Sstefano_zampini 23203425bc38SStefano Zampini /* compute rhs for BDDC application */ 23213425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 23228eeda7d8SStefano Zampini if (pcbddc->switch_static) { 23233425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 2324229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2325229984c5Sstefano_zampini ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr); 2326229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23273425bc38SStefano Zampini } 2328229984c5Sstefano_zampini } 2329229984c5Sstefano_zampini 23303425bc38SStefano Zampini /* apply BDDC */ 2331229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2332dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2333229984c5Sstefano_zampini 2334229984c5Sstefano_zampini /* put values into global vector */ 2335af140850Sstefano_zampini if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change; 2336af140850Sstefano_zampini else work = standard_sol; 2337229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2338229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23398eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 23403425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 23413425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 234200f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 234300f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23443425bc38SStefano Zampini } 2345229984c5Sstefano_zampini 2346229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2347229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2348266e20e9SStefano Zampini /* add p0 solution to final solution */ 2349229984c5Sstefano_zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr); 2350fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2351af140850Sstefano_zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr); 2352fc17d649SStefano Zampini } 2353af140850Sstefano_zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 2354af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2355229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2356229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2357229984c5Sstefano_zampini } 23583425bc38SStefano Zampini PetscFunctionReturn(0); 23593425bc38SStefano Zampini } 23601e6b0712SBarry Smith 23613425bc38SStefano Zampini /*@ 23620f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 23633425bc38SStefano Zampini 23643425bc38SStefano Zampini Collective 23653425bc38SStefano Zampini 23663425bc38SStefano Zampini Input Parameters: 23670f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 23680f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 23693425bc38SStefano Zampini 23703425bc38SStefano Zampini Output Parameters: 23710f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 23723425bc38SStefano Zampini 23733425bc38SStefano Zampini Level: developer 23743425bc38SStefano Zampini 23753425bc38SStefano Zampini Notes: 23763425bc38SStefano Zampini 23770f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 23783425bc38SStefano Zampini @*/ 23793425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23803425bc38SStefano Zampini { 2381674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23823425bc38SStefano Zampini PetscErrorCode ierr; 23833425bc38SStefano Zampini 23843425bc38SStefano Zampini PetscFunctionBegin; 2385266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2386266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2387266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 23883425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2389163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 23903425bc38SStefano Zampini PetscFunctionReturn(0); 23913425bc38SStefano Zampini } 23921e6b0712SBarry Smith 2393547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc) 23943425bc38SStefano Zampini { 2395674ae819SStefano Zampini 2396674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 23973425bc38SStefano Zampini Mat newmat; 2398674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 23993425bc38SStefano Zampini PC newpc; 2400ce94432eSBarry Smith MPI_Comm comm; 24013425bc38SStefano Zampini PetscErrorCode ierr; 24023425bc38SStefano Zampini 24033425bc38SStefano Zampini PetscFunctionBegin; 2404ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 24053425bc38SStefano Zampini /* FETIDP linear matrix */ 24063425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 24071720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 24083425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 2409a5bb87b3Sstefano_zampini ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 24103425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2411edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 24123425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 2413547c9a8eSstefano_zampini ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr); 2414547c9a8eSstefano_zampini ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr); 24153425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 24163425bc38SStefano Zampini /* FETIDP preconditioner */ 24173425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 24183425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 24193425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 2420e1214c54Sstefano_zampini ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 2421e1214c54Sstefano_zampini ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr); 2422e1214c54Sstefano_zampini ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr); 2423e1214c54Sstefano_zampini if (!fetidpmat_ctx->l2g_lambda_only) { 24243425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 24253425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 24263425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2427edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2428c45b8d2dSstefano_zampini ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr); 24293425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2430e1214c54Sstefano_zampini } else { 2431e1214c54Sstefano_zampini KSP *ksps; 2432e1214c54Sstefano_zampini PC lagpc; 2433e1214c54Sstefano_zampini Mat M,AM,PM; 2434e1214c54Sstefano_zampini PetscInt nn; 2435e1214c54Sstefano_zampini 2436e1214c54Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_PPmat",(PetscObject*)&M);CHKERRQ(ierr); 2437e1214c54Sstefano_zampini ierr = PCSetType(newpc,PCFIELDSPLIT);CHKERRQ(ierr); 2438e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"lag",fetidpmat_ctx->lagrange);CHKERRQ(ierr); 2439e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"p",fetidpmat_ctx->pressure);CHKERRQ(ierr); 2440e1214c54Sstefano_zampini ierr = PCFieldSplitSetType(newpc,PC_COMPOSITE_SCHUR);CHKERRQ(ierr); 244140c75d76SStefano Zampini ierr = PCFieldSplitSetSchurFactType(newpc,PC_FIELDSPLIT_SCHUR_FACT_DIAG);CHKERRQ(ierr); 2442e1214c54Sstefano_zampini ierr = PCFieldSplitSetSchurPre(newpc,PC_FIELDSPLIT_SCHUR_PRE_USER,M);CHKERRQ(ierr); 2443c096484dSStefano Zampini ierr = PCFieldSplitSetSchurScale(newpc,1.0);CHKERRQ(ierr); 2444e1214c54Sstefano_zampini ierr = PCSetFromOptions(newpc);CHKERRQ(ierr); 2445e1214c54Sstefano_zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 2446e1214c54Sstefano_zampini 2447e1214c54Sstefano_zampini /* set the solver for the (0,0) block */ 244840c75d76SStefano Zampini ierr = PCFieldSplitGetSubKSP(newpc,&nn,&ksps);CHKERRQ(ierr); 2449e1214c54Sstefano_zampini ierr = PCCreate(comm,&lagpc);CHKERRQ(ierr); 2450e1214c54Sstefano_zampini ierr = PCSetType(lagpc,PCSHELL);CHKERRQ(ierr); 2451e1214c54Sstefano_zampini ierr = KSPGetOperators(ksps[0],&AM,&PM);CHKERRQ(ierr); 2452e1214c54Sstefano_zampini ierr = PCSetOperators(lagpc,AM,PM);CHKERRQ(ierr); 2453e1214c54Sstefano_zampini ierr = PCShellSetContext(lagpc,fetidppc_ctx);CHKERRQ(ierr); 2454e1214c54Sstefano_zampini ierr = PCShellSetApply(lagpc,FETIDPPCApply);CHKERRQ(ierr); 2455e1214c54Sstefano_zampini ierr = PCShellSetApplyTranspose(lagpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2456e1214c54Sstefano_zampini ierr = PCShellSetView(lagpc,FETIDPPCView);CHKERRQ(ierr); 2457e1214c54Sstefano_zampini ierr = PCShellSetDestroy(lagpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2458e1214c54Sstefano_zampini ierr = KSPSetPC(ksps[0],lagpc);CHKERRQ(ierr); 2459e1214c54Sstefano_zampini ierr = PCDestroy(&lagpc);CHKERRQ(ierr); 2460e1214c54Sstefano_zampini ierr = PetscFree(ksps);CHKERRQ(ierr); 2461e1214c54Sstefano_zampini } 24623425bc38SStefano Zampini /* return pointers for objects created */ 24633425bc38SStefano Zampini *fetidp_mat = newmat; 24643425bc38SStefano Zampini *fetidp_pc = newpc; 24653425bc38SStefano Zampini PetscFunctionReturn(0); 24663425bc38SStefano Zampini } 24671e6b0712SBarry Smith 246894ef8ddeSSatish Balay /*@C 24690f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 24703425bc38SStefano Zampini 24713425bc38SStefano Zampini Collective 24723425bc38SStefano Zampini 24733425bc38SStefano Zampini Input Parameters: 24741720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 2475547c9a8eSstefano_zampini . fully_redundant - true for a fully redundant set of Lagrange multipliers 2476547c9a8eSstefano_zampini - prefix - optional options database prefix for the objects to be created (can be NULL) 247728509bceSStefano Zampini 247828509bceSStefano Zampini Output Parameters: 24790f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 24800f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 248128509bceSStefano Zampini 24823425bc38SStefano Zampini Level: developer 24833425bc38SStefano Zampini 24843425bc38SStefano Zampini Notes: 24850f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 24863425bc38SStefano Zampini 24870f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 24883425bc38SStefano Zampini @*/ 2489547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc) 24903425bc38SStefano Zampini { 24913425bc38SStefano Zampini PetscErrorCode ierr; 24923425bc38SStefano Zampini 24933425bc38SStefano Zampini PetscFunctionBegin; 24943425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 24953425bc38SStefano Zampini if (pc->setupcalled) { 2496547c9a8eSstefano_zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 24974247aa23Sstefano_zampini } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 24983425bc38SStefano Zampini PetscFunctionReturn(0); 24993425bc38SStefano Zampini } 25000c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2501da1bb401SStefano Zampini /*MC 2502da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 25030c7d97c5SJed Brown 250428509bceSStefano Zampini An implementation of the BDDC preconditioner based on 250528509bceSStefano Zampini 250628509bceSStefano Zampini .vb 250728509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 250828509bceSStefano 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 250928509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 25100f202f7eSStefano 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 251128509bceSStefano Zampini .ve 251228509bceSStefano Zampini 251328509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 251428509bceSStefano Zampini 25150f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 251628509bceSStefano Zampini 251728509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 251828509bceSStefano Zampini 2519b6fdb6dfSStefano 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. 2520b6fdb6dfSStefano Zampini 2521c7017625SStefano 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). 252228509bceSStefano Zampini 25230f202f7eSStefano 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() 252430368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 252528509bceSStefano Zampini 25260f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 252728509bceSStefano Zampini 25280f202f7eSStefano 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. 25290f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 253028509bceSStefano Zampini 25310f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 253228509bceSStefano Zampini 2533df4d28bfSStefano Zampini Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS or MKL_PARDISO are present. Future versions of the code will also consider using PASTIX. 253428509bceSStefano Zampini 25350f202f7eSStefano 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. 25360f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 25370f202f7eSStefano Zampini 25380f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 25390f202f7eSStefano Zampini 25400f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 25410f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 25420f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 25430f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 25440f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 25450f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 25460f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 254728509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 25480f202f7eSStefano 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) 25490f202f7eSStefano 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) 25500f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 25510f202f7eSStefano 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) 2552df4d28bfSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 255328509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 255428509bceSStefano Zampini 255528509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 255628509bceSStefano Zampini .vb 255728509bceSStefano Zampini -pc_bddc_dirichlet_ 255828509bceSStefano Zampini -pc_bddc_neumann_ 255928509bceSStefano Zampini -pc_bddc_coarse_ 256028509bceSStefano Zampini .ve 25610f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 256228509bceSStefano Zampini 25630f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 256428509bceSStefano Zampini .vb 2565312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2566312be037SStefano Zampini -pc_bddc_neumann_lN_ 2567312be037SStefano Zampini -pc_bddc_coarse_lN_ 256828509bceSStefano Zampini .ve 25690f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 25700f202f7eSStefano 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. 25710f202f7eSStefano Zampini .vb 25720f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 25730f202f7eSStefano Zampini .ve 25740f202f7eSStefano 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 2575da1bb401SStefano Zampini 2576da1bb401SStefano Zampini Level: intermediate 2577da1bb401SStefano Zampini 2578b6fdb6dfSStefano Zampini Developer notes: 2579da1bb401SStefano Zampini 2580da1bb401SStefano Zampini Contributed by Stefano Zampini 2581da1bb401SStefano Zampini 2582da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2583da1bb401SStefano Zampini M*/ 2584b2573a8aSBarry Smith 25858cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2586da1bb401SStefano Zampini { 2587da1bb401SStefano Zampini PetscErrorCode ierr; 2588da1bb401SStefano Zampini PC_BDDC *pcbddc; 2589da1bb401SStefano Zampini 2590da1bb401SStefano Zampini PetscFunctionBegin; 2591b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2592da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2593da1bb401SStefano Zampini 2594da1bb401SStefano Zampini /* create PCIS data structure */ 2595da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2596da1bb401SStefano Zampini 25979326c5c6Sstefano_zampini /* create local graph structure */ 25989326c5c6Sstefano_zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 25999326c5c6Sstefano_zampini 26009326c5c6Sstefano_zampini /* BDDC nonzero defaults */ 260108a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 260247d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 260347d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 26043301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 260514f95afaSStefano Zampini pcbddc->vertex_size = 1; 2606c703fcc7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 260768457ee5SStefano Zampini pcbddc->coarse_size = -1; 260885c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 260947d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 261057de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 261127b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 26121e0482f5SStefano Zampini pcbddc->nedfield = -1; 26131e0482f5SStefano Zampini pcbddc->nedglobal = PETSC_TRUE; 2614be12c134Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 2615b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2616b7eb3628SStefano Zampini 2617da1bb401SStefano Zampini /* function pointers */ 2618da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 261993bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2620da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2621da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2622da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 26236b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2624da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2625da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2626da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2627534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2628534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 26299326c5c6Sstefano_zampini pc->ops->reset = PCReset_BDDC; 2630da1bb401SStefano Zampini 2631da1bb401SStefano Zampini /* composing function */ 2632a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr); 2633a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr); 2634906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2635674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 263630368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2637bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 26382b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2639b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 26402b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2641bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 264282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2643bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 264482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2645bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 264682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2647bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 264882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2649bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 265063602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2651bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2652bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2653bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2654bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2655a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2656da1bb401SStefano Zampini PetscFunctionReturn(0); 2657da1bb401SStefano Zampini } 26583425bc38SStefano Zampini 2659