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); 1276f0c0a6aSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use user-defined CSR: %d\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 1286f0c0a6aSStefano Zampini 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 } 134aefa1729SStefano Zampini if (pcbddc->graphmaxcount != PETSC_MAX_INT) { 135efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Graph max count: %D\n",pcbddc->graphmaxcount);CHKERRQ(ierr); 136aefa1729SStefano Zampini } 13750e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use vertices: %d (vertex size %D)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 138efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 139efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 140efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 141efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 142efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 143efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 144efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 145efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 146efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 147efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 148efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 14950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Multilevel max levels: %D\n",pcbddc->max_levels);CHKERRQ(ierr); 15050e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Multilevel coarsening ratio: %D\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 151efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 152efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 153efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 154efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr); 155efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 15650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Number of dofs' layers for the computation of principal minors: %D\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 157efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 15818032b1bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_selection,pcbddc->adaptive_userdefined,pcbddc->adaptive_threshold);CHKERRQ(ierr); 15950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Min constraints / connected component: %D\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 16050e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Max constraints / connected component: %D\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 161efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 162efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 16350e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Num. Procs. to map coarse adjacency list: %D\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 16450e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Coarse eqs per proc (significant at the coarsest level): %D\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 165efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr); 166efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 167efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr); 168efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr); 169b74ba07aSstefano_zampini if (!pc->setupcalled) PetscFunctionReturn(0); 1706b78500eSPatrick Sanan 171fbad9177SStefano Zampini /* compute interface size */ 172e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 173e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 174e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 175e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 176e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 177e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 178e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 179fbad9177SStefano Zampini 180fbad9177SStefano Zampini /* compute some statistics on the domain decomposition */ 181e9627c49SStefano Zampini gsum[0] = 1; 182fbad9177SStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0; 183e9627c49SStefano Zampini loc[0] = !!pcis->n; 184e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 185e9627c49SStefano Zampini loc[2] = pcis->n_B; 186e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 187345ecf6cSStefano Zampini loc[4] = pcis->n; 188fbad9177SStefano Zampini loc[5] = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0); 189fbad9177SStefano Zampini loc[6] = pcbddc->benign_n; 190fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 191fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1; 192fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 193fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT; 194fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 195fbad9177SStefano Zampini ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 196e9627c49SStefano Zampini if (pcbddc->coarse_size) { 197e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 198e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 199e9627c49SStefano Zampini } 200efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 20150e0721cSStefano Zampini 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); 20250e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Coarsening ratios: all/coarse %D interface/coarse %D\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 20350e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Active processes : %D\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 20450e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Total subdomains : %D\n",(PetscInt)gsum[5]);CHKERRQ(ierr); 205345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 20650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Benign subs : %D\n",(PetscInt)totbenign);CHKERRQ(ierr); 207345ecf6cSStefano Zampini } 20850e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Dofs type :\tMIN\tMAX\tMEAN\n");CHKERRQ(ierr); 20950e0721cSStefano Zampini 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); 21050e0721cSStefano Zampini 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); 21150e0721cSStefano Zampini 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); 21250e0721cSStefano Zampini 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); 21350e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Local subs :\t%D\t%D\n" ,(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr); 21450e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," ********************************** COARSE PROBLEM DETAILS *********************************\n");CHKERRQ(ierr); 21527b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 216e9627c49SStefano Zampini 217fbad9177SStefano Zampini /* the coarse problem can be handled by a different communicator */ 218e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 219e9627c49SStefano Zampini else color = 0; 220e9627c49SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 2214b2aedd3SStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 222e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 2234b2aedd3SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr); 224e9627c49SStefano Zampini ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr); 225e9627c49SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 226e9627c49SStefano Zampini if (color == 1) { 227e9627c49SStefano Zampini ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr); 228e9627c49SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 229e9627c49SStefano Zampini } 230e9627c49SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 231e9627c49SStefano Zampini ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr); 232e9627c49SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 233e9627c49SStefano Zampini } 2346b78500eSPatrick Sanan PetscFunctionReturn(0); 2356b78500eSPatrick Sanan } 236a13144ffSStefano Zampini 2371e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 238a13144ffSStefano Zampini { 239a13144ffSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 240a13144ffSStefano Zampini PetscErrorCode ierr; 241a13144ffSStefano Zampini 242a13144ffSStefano Zampini PetscFunctionBegin; 243a13144ffSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 244a13144ffSStefano Zampini ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); 245a13144ffSStefano Zampini pcbddc->discretegradient = G; 246a13144ffSStefano Zampini pcbddc->nedorder = order > 0 ? order : -order; 247495a2a07SStefano Zampini pcbddc->nedfield = field; 2481e0482f5SStefano Zampini pcbddc->nedglobal = global; 2491e0482f5SStefano Zampini pcbddc->conforming = conforming; 250a13144ffSStefano Zampini PetscFunctionReturn(0); 251a13144ffSStefano Zampini } 252a13144ffSStefano Zampini 253a13144ffSStefano Zampini /*@ 254a13144ffSStefano Zampini PCBDDCSetDiscreteGradient - Sets the discrete gradient 255a13144ffSStefano Zampini 256a13144ffSStefano Zampini Collective on PC 257a13144ffSStefano Zampini 258a13144ffSStefano Zampini Input Parameters: 259a13144ffSStefano Zampini + pc - the preconditioning context 260a13144ffSStefano Zampini . G - the discrete gradient matrix (should be in AIJ format) 261a13144ffSStefano Zampini . order - the order of the Nedelec space (1 for the lowest order) 262495a2a07SStefano Zampini . field - the field id of the Nedelec dofs (not used if the fields have not been specified) 2631e0482f5SStefano Zampini . global - the type of global ordering for the rows of G 264a13144ffSStefano Zampini - conforming - whether the mesh is conforming or not 265a13144ffSStefano Zampini 266a13144ffSStefano Zampini Level: advanced 267a13144ffSStefano Zampini 268a13144ffSStefano Zampini Notes: The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry. 269495a2a07SStefano Zampini For variable order spaces, the order should be set to zero. 2701e0482f5SStefano Zampini If global is true, the rows of G should be given in global ordering for the whole dofs; 2711e0482f5SStefano Zampini if false, the ordering should be global for the Nedelec field. 2721e0482f5SStefano 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 2731e0482f5SStefano Zampini and geid the one for the Nedelec field. 274a13144ffSStefano Zampini 275495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal() 276a13144ffSStefano Zampini @*/ 2771e0482f5SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 278a13144ffSStefano Zampini { 279a13144ffSStefano Zampini PetscErrorCode ierr; 280a13144ffSStefano Zampini 281a13144ffSStefano Zampini PetscFunctionBegin; 282a13144ffSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 283a13144ffSStefano Zampini PetscValidHeaderSpecific(G,MAT_CLASSID,2); 284a13144ffSStefano Zampini PetscValidLogicalCollectiveInt(pc,order,3); 2851e0482f5SStefano Zampini PetscValidLogicalCollectiveInt(pc,field,4); 2861e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,global,5); 2871e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,conforming,6); 2881e0482f5SStefano Zampini PetscCheckSameComm(pc,1,G,2); 2891e0482f5SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool,PetscBool),(pc,G,order,field,global,conforming));CHKERRQ(ierr); 290a13144ffSStefano Zampini PetscFunctionReturn(0); 291a13144ffSStefano Zampini } 292a13144ffSStefano Zampini 2938ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 294a198735bSStefano Zampini { 295a198735bSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 296a198735bSStefano Zampini PetscErrorCode ierr; 2976b78500eSPatrick Sanan 298a198735bSStefano Zampini PetscFunctionBegin; 299a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr); 300a198735bSStefano Zampini ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); 301a198735bSStefano Zampini pcbddc->divudotp = divudotp; 3028ae0ca82SStefano Zampini pcbddc->divudotp_trans = trans; 303a198735bSStefano Zampini pcbddc->compute_nonetflux = PETSC_TRUE; 304a198735bSStefano Zampini if (vl2l) { 305a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr); 306fa23a32eSStefano Zampini ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); 307a198735bSStefano Zampini pcbddc->divudotp_vl2l = vl2l; 308a198735bSStefano Zampini } 309a198735bSStefano Zampini PetscFunctionReturn(0); 310a198735bSStefano Zampini } 3113d996552SStefano Zampini 312a198735bSStefano Zampini /*@ 313a198735bSStefano Zampini PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx 314a198735bSStefano Zampini 315a198735bSStefano Zampini Collective on PC 316a198735bSStefano Zampini 317a198735bSStefano Zampini Input Parameters: 318a198735bSStefano Zampini + pc - the preconditioning context 319a198735bSStefano Zampini . divudotp - the matrix (must be of type MATIS) 3208ae0ca82SStefano Zampini . trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space. 321*05a3bf82SStefano Zampini - vl2l - optional index set describing the local (wrt the local matrix in divudotp) to local (wrt the local matrix in the preconditioning matrix) map for the velocities 322a198735bSStefano Zampini 323a198735bSStefano Zampini Level: advanced 324a198735bSStefano Zampini 3258ae0ca82SStefano Zampini Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries 326*05a3bf82SStefano Zampini If vl2l is NULL, the local ordering for velocities in divudotp should match that of the preconditioning matrix 327a198735bSStefano Zampini 328a198735bSStefano Zampini .seealso: PCBDDC 329a198735bSStefano Zampini @*/ 3308ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 331a198735bSStefano Zampini { 332a198735bSStefano Zampini PetscBool ismatis; 333a198735bSStefano Zampini PetscErrorCode ierr; 334a198735bSStefano Zampini 335a198735bSStefano Zampini PetscFunctionBegin; 336a198735bSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 337a198735bSStefano Zampini PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2); 338a198735bSStefano Zampini PetscCheckSameComm(pc,1,divudotp,2); 3398ae0ca82SStefano Zampini PetscValidLogicalCollectiveBool(pc,trans,3); 3408ae0ca82SStefano Zampini if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4); 341a198735bSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr); 342a198735bSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS"); 3438ae0ca82SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr); 344a198735bSStefano Zampini PetscFunctionReturn(0); 345a198735bSStefano Zampini } 3462d505d7fSStefano Zampini 3471dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 348b9b85e73SStefano Zampini { 349b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 350b9b85e73SStefano Zampini PetscErrorCode ierr; 351b9b85e73SStefano Zampini 352b9b85e73SStefano Zampini PetscFunctionBegin; 353b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 35456282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 355b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 3561dd7afcfSStefano Zampini pcbddc->change_interior = interior; 357b9b85e73SStefano Zampini PetscFunctionReturn(0); 358b9b85e73SStefano Zampini } 359b9b85e73SStefano Zampini /*@ 360906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 361b9b85e73SStefano Zampini 362b9b85e73SStefano Zampini Collective on PC 363b9b85e73SStefano Zampini 364b9b85e73SStefano Zampini Input Parameters: 365b9b85e73SStefano Zampini + pc - the preconditioning context 3661dd7afcfSStefano Zampini . change - the change of basis matrix 3671dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 368b9b85e73SStefano Zampini 369b9b85e73SStefano Zampini Level: intermediate 370b9b85e73SStefano Zampini 371b9b85e73SStefano Zampini Notes: 372b9b85e73SStefano Zampini 373b9b85e73SStefano Zampini .seealso: PCBDDC 374b9b85e73SStefano Zampini @*/ 3751dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 376b9b85e73SStefano Zampini { 377b9b85e73SStefano Zampini PetscErrorCode ierr; 378b9b85e73SStefano Zampini 379b9b85e73SStefano Zampini PetscFunctionBegin; 380b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 381b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 382906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 383906d46d4SStefano Zampini if (pc->mat) { 384906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 385906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 386906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 387e0fe2d75SToby 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); 388e0fe2d75SToby 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); 389906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 390906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 391e0fe2d75SToby 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); 392e0fe2d75SToby 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); 393906d46d4SStefano Zampini } 3941dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 395b9b85e73SStefano Zampini PetscFunctionReturn(0); 396b9b85e73SStefano Zampini } 3972d505d7fSStefano Zampini 39830368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 39930368db7SStefano Zampini { 40030368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 40156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 40230368db7SStefano Zampini PetscErrorCode ierr; 40330368db7SStefano Zampini 40430368db7SStefano Zampini PetscFunctionBegin; 40556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 40656282151SStefano Zampini if (pcbddc->user_primal_vertices) { 40756282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 40856282151SStefano Zampini } 40930368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 41030368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 41130368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 41256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 41330368db7SStefano Zampini PetscFunctionReturn(0); 41430368db7SStefano Zampini } 41530368db7SStefano Zampini /*@ 41630368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 41730368db7SStefano Zampini 41830368db7SStefano Zampini Collective 41930368db7SStefano Zampini 42030368db7SStefano Zampini Input Parameters: 42130368db7SStefano Zampini + pc - the preconditioning context 42230368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 42330368db7SStefano Zampini 42430368db7SStefano Zampini Level: intermediate 42530368db7SStefano Zampini 42630368db7SStefano Zampini Notes: 42730368db7SStefano Zampini Any process can list any global node 42830368db7SStefano Zampini 42930368db7SStefano Zampini .seealso: PCBDDC 43030368db7SStefano Zampini @*/ 43130368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 43230368db7SStefano Zampini { 43330368db7SStefano Zampini PetscErrorCode ierr; 43430368db7SStefano Zampini 43530368db7SStefano Zampini PetscFunctionBegin; 43630368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 43730368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 43830368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 43930368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 44030368db7SStefano Zampini PetscFunctionReturn(0); 44130368db7SStefano Zampini } 4422d505d7fSStefano Zampini 443674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 444674ae819SStefano Zampini { 445674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 44656282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 447674ae819SStefano Zampini PetscErrorCode ierr; 4481e6b0712SBarry Smith 449674ae819SStefano Zampini PetscFunctionBegin; 45056282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 45156282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 45256282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 45356282151SStefano Zampini } 454674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 45530368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 45630368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 45756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 458674ae819SStefano Zampini PetscFunctionReturn(0); 459674ae819SStefano Zampini } 460674ae819SStefano Zampini /*@ 46128509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 462674ae819SStefano Zampini 46317eb1463SStefano Zampini Collective 464674ae819SStefano Zampini 465674ae819SStefano Zampini Input Parameters: 466674ae819SStefano Zampini + pc - the preconditioning context 46717eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 468674ae819SStefano Zampini 469674ae819SStefano Zampini Level: intermediate 470674ae819SStefano Zampini 471674ae819SStefano Zampini Notes: 472674ae819SStefano Zampini 473674ae819SStefano Zampini .seealso: PCBDDC 474674ae819SStefano Zampini @*/ 475674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 476674ae819SStefano Zampini { 477674ae819SStefano Zampini PetscErrorCode ierr; 478674ae819SStefano Zampini 479674ae819SStefano Zampini PetscFunctionBegin; 480674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 481674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 48217eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 483674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 484674ae819SStefano Zampini PetscFunctionReturn(0); 485674ae819SStefano Zampini } 4862d505d7fSStefano Zampini 4874fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 4884fad6a16SStefano Zampini { 4894fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4904fad6a16SStefano Zampini 4914fad6a16SStefano Zampini PetscFunctionBegin; 4924fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 4934fad6a16SStefano Zampini PetscFunctionReturn(0); 4944fad6a16SStefano Zampini } 4951e6b0712SBarry Smith 4964fad6a16SStefano Zampini /*@ 49728509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 4984fad6a16SStefano Zampini 4994fad6a16SStefano Zampini Logically collective on PC 5004fad6a16SStefano Zampini 5014fad6a16SStefano Zampini Input Parameters: 5024fad6a16SStefano Zampini + pc - the preconditioning context 50328509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 5044fad6a16SStefano Zampini 5050f202f7eSStefano Zampini Options Database Keys: 5060f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 5074fad6a16SStefano Zampini 5084fad6a16SStefano Zampini Level: intermediate 5094fad6a16SStefano Zampini 5104fad6a16SStefano Zampini Notes: 5110f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 5124fad6a16SStefano Zampini 5130f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 5144fad6a16SStefano Zampini @*/ 5154fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 5164fad6a16SStefano Zampini { 5174fad6a16SStefano Zampini PetscErrorCode ierr; 5184fad6a16SStefano Zampini 5194fad6a16SStefano Zampini PetscFunctionBegin; 5204fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5212b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 5224fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 5234fad6a16SStefano Zampini PetscFunctionReturn(0); 5244fad6a16SStefano Zampini } 5252b510759SStefano Zampini 526b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 527b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 528b8ffe317SStefano Zampini { 529b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 530b8ffe317SStefano Zampini 531b8ffe317SStefano Zampini PetscFunctionBegin; 53285c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 533b8ffe317SStefano Zampini PetscFunctionReturn(0); 534b8ffe317SStefano Zampini } 535b8ffe317SStefano Zampini 536b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 5372b510759SStefano Zampini { 5382b510759SStefano Zampini PetscErrorCode ierr; 5392b510759SStefano Zampini 5402b510759SStefano Zampini PetscFunctionBegin; 5412b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 542b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 543b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 5442b510759SStefano Zampini PetscFunctionReturn(0); 5452b510759SStefano Zampini } 5461e6b0712SBarry Smith 5472b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 5484fad6a16SStefano Zampini { 5494fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5504fad6a16SStefano Zampini 5514fad6a16SStefano Zampini PetscFunctionBegin; 5522b510759SStefano Zampini pcbddc->current_level = level; 5534fad6a16SStefano Zampini PetscFunctionReturn(0); 5544fad6a16SStefano Zampini } 5551e6b0712SBarry Smith 556b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 557b8ffe317SStefano Zampini { 558b8ffe317SStefano Zampini PetscErrorCode ierr; 559b8ffe317SStefano Zampini 560b8ffe317SStefano Zampini PetscFunctionBegin; 561b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 562b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 563b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 564b8ffe317SStefano Zampini PetscFunctionReturn(0); 565b8ffe317SStefano Zampini } 566b8ffe317SStefano Zampini 5672b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 5682b510759SStefano Zampini { 5692b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5702b510759SStefano Zampini 5712b510759SStefano Zampini PetscFunctionBegin; 5722b510759SStefano Zampini pcbddc->max_levels = levels; 5732b510759SStefano Zampini PetscFunctionReturn(0); 5742b510759SStefano Zampini } 5752b510759SStefano Zampini 5764fad6a16SStefano Zampini /*@ 57728509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 5784fad6a16SStefano Zampini 5794fad6a16SStefano Zampini Logically collective on PC 5804fad6a16SStefano Zampini 5814fad6a16SStefano Zampini Input Parameters: 5824fad6a16SStefano Zampini + pc - the preconditioning context 58328509bceSStefano Zampini - levels - the maximum number of levels (max 9) 5844fad6a16SStefano Zampini 5850f202f7eSStefano Zampini Options Database Keys: 5860f202f7eSStefano Zampini . -pc_bddc_levels 5874fad6a16SStefano Zampini 5884fad6a16SStefano Zampini Level: intermediate 5894fad6a16SStefano Zampini 5904fad6a16SStefano Zampini Notes: 5910f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 5924fad6a16SStefano Zampini 5930f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 5944fad6a16SStefano Zampini @*/ 5952b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 5964fad6a16SStefano Zampini { 5974fad6a16SStefano Zampini PetscErrorCode ierr; 5984fad6a16SStefano Zampini 5994fad6a16SStefano Zampini PetscFunctionBegin; 6004fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6012b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 602312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 6032b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 6044fad6a16SStefano Zampini PetscFunctionReturn(0); 6054fad6a16SStefano Zampini } 6061e6b0712SBarry Smith 6073b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 6083b03a366Sstefano_zampini { 6093b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 61056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6113b03a366Sstefano_zampini PetscErrorCode ierr; 6123b03a366Sstefano_zampini 6133b03a366Sstefano_zampini PetscFunctionBegin; 61456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 61556282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 61656282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 61756282151SStefano Zampini } 618785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 619785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6203b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 62136e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 62256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6233b03a366Sstefano_zampini PetscFunctionReturn(0); 6243b03a366Sstefano_zampini } 6251e6b0712SBarry Smith 6263b03a366Sstefano_zampini /*@ 62728509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 6283b03a366Sstefano_zampini 629785d1243SStefano Zampini Collective 6303b03a366Sstefano_zampini 6313b03a366Sstefano_zampini Input Parameters: 6323b03a366Sstefano_zampini + pc - the preconditioning context 633785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 6343b03a366Sstefano_zampini 6353b03a366Sstefano_zampini Level: intermediate 6363b03a366Sstefano_zampini 6370f202f7eSStefano Zampini Notes: 6380f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 6393b03a366Sstefano_zampini 6400f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 6413b03a366Sstefano_zampini @*/ 6423b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 6433b03a366Sstefano_zampini { 6443b03a366Sstefano_zampini PetscErrorCode ierr; 6453b03a366Sstefano_zampini 6463b03a366Sstefano_zampini PetscFunctionBegin; 6473b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 648674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 649785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 6503b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6513b03a366Sstefano_zampini PetscFunctionReturn(0); 6523b03a366Sstefano_zampini } 6531e6b0712SBarry Smith 65482ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 6553b03a366Sstefano_zampini { 6563b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 65756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6583b03a366Sstefano_zampini PetscErrorCode ierr; 6593b03a366Sstefano_zampini 6603b03a366Sstefano_zampini PetscFunctionBegin; 66156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 66256282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 66356282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 66456282151SStefano Zampini } 665785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 666785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6673b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 668785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 66956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6703b03a366Sstefano_zampini PetscFunctionReturn(0); 6713b03a366Sstefano_zampini } 6723b03a366Sstefano_zampini 6733b03a366Sstefano_zampini /*@ 67482ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 6753b03a366Sstefano_zampini 676785d1243SStefano Zampini Collective 6773b03a366Sstefano_zampini 6783b03a366Sstefano_zampini Input Parameters: 6793b03a366Sstefano_zampini + pc - the preconditioning context 68082ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 6813b03a366Sstefano_zampini 6823b03a366Sstefano_zampini Level: intermediate 6833b03a366Sstefano_zampini 6843b03a366Sstefano_zampini Notes: 6853b03a366Sstefano_zampini 6860f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 6873b03a366Sstefano_zampini @*/ 68882ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 6893b03a366Sstefano_zampini { 6903b03a366Sstefano_zampini PetscErrorCode ierr; 6913b03a366Sstefano_zampini 6923b03a366Sstefano_zampini PetscFunctionBegin; 6933b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6943b03a366Sstefano_zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 69582ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 69682ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6973b03a366Sstefano_zampini PetscFunctionReturn(0); 6983b03a366Sstefano_zampini } 6993b03a366Sstefano_zampini 70053cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 7010c7d97c5SJed Brown { 7020c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 70356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 70453cdbc3dSStefano Zampini PetscErrorCode ierr; 7050c7d97c5SJed Brown 7060c7d97c5SJed Brown PetscFunctionBegin; 70756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 70856282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 70956282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 71056282151SStefano Zampini } 711785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 712785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 71353cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 71436e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 71556282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7160c7d97c5SJed Brown PetscFunctionReturn(0); 7170c7d97c5SJed Brown } 7181e6b0712SBarry Smith 71957527edcSJed Brown /*@ 72028509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 72157527edcSJed Brown 722785d1243SStefano Zampini Collective 72357527edcSJed Brown 72457527edcSJed Brown Input Parameters: 72557527edcSJed Brown + pc - the preconditioning context 726785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 72757527edcSJed Brown 72857527edcSJed Brown Level: intermediate 72957527edcSJed Brown 7300f202f7eSStefano Zampini Notes: 7310f202f7eSStefano Zampini Any process can list any global node 73257527edcSJed Brown 7330f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 73457527edcSJed Brown @*/ 73553cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 7360c7d97c5SJed Brown { 7370c7d97c5SJed Brown PetscErrorCode ierr; 7380c7d97c5SJed Brown 7390c7d97c5SJed Brown PetscFunctionBegin; 7400c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 741674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 742785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 74353cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 74453cdbc3dSStefano Zampini PetscFunctionReturn(0); 74553cdbc3dSStefano Zampini } 7461e6b0712SBarry Smith 74782ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 7480c7d97c5SJed Brown { 7490c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 75056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7510c7d97c5SJed Brown PetscErrorCode ierr; 7520c7d97c5SJed Brown 7530c7d97c5SJed Brown PetscFunctionBegin; 75456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 75556282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 75656282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 75756282151SStefano Zampini } 758785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 759785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 7600c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 761785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 76256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7630c7d97c5SJed Brown PetscFunctionReturn(0); 7640c7d97c5SJed Brown } 7650c7d97c5SJed Brown 7660c7d97c5SJed Brown /*@ 76782ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 7680c7d97c5SJed Brown 769785d1243SStefano Zampini Collective 7700c7d97c5SJed Brown 7710c7d97c5SJed Brown Input Parameters: 7720c7d97c5SJed Brown + pc - the preconditioning context 77382ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 7740c7d97c5SJed Brown 7750c7d97c5SJed Brown Level: intermediate 7760c7d97c5SJed Brown 7770c7d97c5SJed Brown Notes: 7780c7d97c5SJed Brown 7790f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 7800c7d97c5SJed Brown @*/ 78182ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 7820c7d97c5SJed Brown { 7830c7d97c5SJed Brown PetscErrorCode ierr; 7840c7d97c5SJed Brown 7850c7d97c5SJed Brown PetscFunctionBegin; 7860c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7870c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 78882ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 78982ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 79053cdbc3dSStefano Zampini PetscFunctionReturn(0); 79153cdbc3dSStefano Zampini } 79253cdbc3dSStefano Zampini 793da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 794da1bb401SStefano Zampini { 795da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 796da1bb401SStefano Zampini 797da1bb401SStefano Zampini PetscFunctionBegin; 798da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 799da1bb401SStefano Zampini PetscFunctionReturn(0); 800da1bb401SStefano Zampini } 8011e6b0712SBarry Smith 802da1bb401SStefano Zampini /*@ 803785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 804da1bb401SStefano Zampini 805785d1243SStefano Zampini Collective 806785d1243SStefano Zampini 807785d1243SStefano Zampini Input Parameters: 808785d1243SStefano Zampini . pc - the preconditioning context 809785d1243SStefano Zampini 810785d1243SStefano Zampini Output Parameters: 811785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 812785d1243SStefano Zampini 813785d1243SStefano Zampini Level: intermediate 814785d1243SStefano Zampini 8150f202f7eSStefano Zampini Notes: 8160f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 817785d1243SStefano Zampini 818785d1243SStefano Zampini .seealso: PCBDDC 819785d1243SStefano Zampini @*/ 820785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 821785d1243SStefano Zampini { 822785d1243SStefano Zampini PetscErrorCode ierr; 823785d1243SStefano Zampini 824785d1243SStefano Zampini PetscFunctionBegin; 825785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 826785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 827785d1243SStefano Zampini PetscFunctionReturn(0); 828785d1243SStefano Zampini } 829785d1243SStefano Zampini 830785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 831785d1243SStefano Zampini { 832785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 833785d1243SStefano Zampini 834785d1243SStefano Zampini PetscFunctionBegin; 835785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 836785d1243SStefano Zampini PetscFunctionReturn(0); 837785d1243SStefano Zampini } 838785d1243SStefano Zampini 839da1bb401SStefano Zampini /*@ 84082ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 841da1bb401SStefano Zampini 842785d1243SStefano Zampini Collective 843da1bb401SStefano Zampini 844da1bb401SStefano Zampini Input Parameters: 84528509bceSStefano Zampini . pc - the preconditioning context 846da1bb401SStefano Zampini 847da1bb401SStefano Zampini Output Parameters: 84828509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 849da1bb401SStefano Zampini 850da1bb401SStefano Zampini Level: intermediate 851da1bb401SStefano Zampini 852da1bb401SStefano Zampini Notes: 8530f202f7eSStefano 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). 8540f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 855da1bb401SStefano Zampini 856da1bb401SStefano Zampini .seealso: PCBDDC 857da1bb401SStefano Zampini @*/ 85882ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 859da1bb401SStefano Zampini { 860da1bb401SStefano Zampini PetscErrorCode ierr; 861da1bb401SStefano Zampini 862da1bb401SStefano Zampini PetscFunctionBegin; 863da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 86482ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 865da1bb401SStefano Zampini PetscFunctionReturn(0); 866da1bb401SStefano Zampini } 8671e6b0712SBarry Smith 86853cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 86953cdbc3dSStefano Zampini { 87053cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 87153cdbc3dSStefano Zampini 87253cdbc3dSStefano Zampini PetscFunctionBegin; 87353cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 87453cdbc3dSStefano Zampini PetscFunctionReturn(0); 87553cdbc3dSStefano Zampini } 8761e6b0712SBarry Smith 87753cdbc3dSStefano Zampini /*@ 878785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 87953cdbc3dSStefano Zampini 880785d1243SStefano Zampini Collective 881785d1243SStefano Zampini 882785d1243SStefano Zampini Input Parameters: 883785d1243SStefano Zampini . pc - the preconditioning context 884785d1243SStefano Zampini 885785d1243SStefano Zampini Output Parameters: 886785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 887785d1243SStefano Zampini 888785d1243SStefano Zampini Level: intermediate 889785d1243SStefano Zampini 8900f202f7eSStefano Zampini Notes: 8910f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 892785d1243SStefano Zampini 893785d1243SStefano Zampini .seealso: PCBDDC 894785d1243SStefano Zampini @*/ 895785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 896785d1243SStefano Zampini { 897785d1243SStefano Zampini PetscErrorCode ierr; 898785d1243SStefano Zampini 899785d1243SStefano Zampini PetscFunctionBegin; 900785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 901785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 902785d1243SStefano Zampini PetscFunctionReturn(0); 903785d1243SStefano Zampini } 904785d1243SStefano Zampini 905785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 906785d1243SStefano Zampini { 907785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 908785d1243SStefano Zampini 909785d1243SStefano Zampini PetscFunctionBegin; 910785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 911785d1243SStefano Zampini PetscFunctionReturn(0); 912785d1243SStefano Zampini } 913785d1243SStefano Zampini 91453cdbc3dSStefano Zampini /*@ 91582ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 91653cdbc3dSStefano Zampini 917785d1243SStefano Zampini Collective 91853cdbc3dSStefano Zampini 91953cdbc3dSStefano Zampini Input Parameters: 92028509bceSStefano Zampini . pc - the preconditioning context 92153cdbc3dSStefano Zampini 92253cdbc3dSStefano Zampini Output Parameters: 92328509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 92453cdbc3dSStefano Zampini 92553cdbc3dSStefano Zampini Level: intermediate 92653cdbc3dSStefano Zampini 92753cdbc3dSStefano Zampini Notes: 9280f202f7eSStefano 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). 9290f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 93053cdbc3dSStefano Zampini 93153cdbc3dSStefano Zampini .seealso: PCBDDC 93253cdbc3dSStefano Zampini @*/ 93382ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 93453cdbc3dSStefano Zampini { 93553cdbc3dSStefano Zampini PetscErrorCode ierr; 93653cdbc3dSStefano Zampini 93753cdbc3dSStefano Zampini PetscFunctionBegin; 93853cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 93982ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 9400c7d97c5SJed Brown PetscFunctionReturn(0); 9410c7d97c5SJed Brown } 9421e6b0712SBarry Smith 9431a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 94436e030ebSStefano Zampini { 94536e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 946da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 94756282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 948da1bb401SStefano Zampini PetscErrorCode ierr; 94936e030ebSStefano Zampini 95036e030ebSStefano Zampini PetscFunctionBegin; 9518687889aSStefano Zampini if (!nvtxs) { 95204194a47SStefano Zampini if (copymode == PETSC_OWN_POINTER) { 95304194a47SStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 95404194a47SStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 95504194a47SStefano Zampini } 9568687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 9578687889aSStefano Zampini PetscFunctionReturn(0); 9588687889aSStefano Zampini } 95966da6bd7Sstefano_zampini if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */ 96056282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 96156282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 9622d505d7fSStefano Zampini ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9632d505d7fSStefano Zampini if (same_data) { 9642d505d7fSStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9652d505d7fSStefano Zampini } 96656282151SStefano Zampini } 96756282151SStefano Zampini } 96856282151SStefano Zampini if (!same_data) { 969674ae819SStefano Zampini /* free old CSR */ 970674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 971674ae819SStefano Zampini /* get CSR into graph structure */ 972da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 973854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 974785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 975674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 976674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 977a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 978da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 9791a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 9801a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 981a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 982a1dbd327SStefano Zampini } else if (copymode == PETSC_USE_POINTER) { 983a1dbd327SStefano Zampini mat_graph->xadj = (PetscInt*)xadj; 984a1dbd327SStefano Zampini mat_graph->adjncy = (PetscInt*)adjncy; 985a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_FALSE; 986e0fe2d75SToby Isaac } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %D",copymode); 987575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 98856282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 98956282151SStefano Zampini } 99036e030ebSStefano Zampini PetscFunctionReturn(0); 99136e030ebSStefano Zampini } 9921e6b0712SBarry Smith 99336e030ebSStefano Zampini /*@ 99454fffbccSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom. 99536e030ebSStefano Zampini 99636e030ebSStefano Zampini Not collective 99736e030ebSStefano Zampini 99836e030ebSStefano Zampini Input Parameters: 99954fffbccSStefano Zampini + pc - the preconditioning context. 100054fffbccSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the number of local dofs). 100154fffbccSStefano Zampini . xadj, adjncy - the connectivity of the dofs in CSR format. 100254fffbccSStefano Zampini - copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER. 100336e030ebSStefano Zampini 100436e030ebSStefano Zampini Level: intermediate 100536e030ebSStefano Zampini 100654fffbccSStefano Zampini Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative. 100736e030ebSStefano Zampini 100828509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 100936e030ebSStefano Zampini @*/ 10101a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 101136e030ebSStefano Zampini { 1012575ad6abSStefano Zampini void (*f)(void) = 0; 101336e030ebSStefano Zampini PetscErrorCode ierr; 101436e030ebSStefano Zampini 101536e030ebSStefano Zampini PetscFunctionBegin; 101636e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 10178687889aSStefano Zampini if (nvtxs) { 1018674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 10191633d1f0SStefano Zampini if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4); 10208687889aSStefano Zampini } 10211a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 1022575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 1023575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 1024575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 1025575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 1026575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 1027da1bb401SStefano Zampini } 102836e030ebSStefano Zampini PetscFunctionReturn(0); 102936e030ebSStefano Zampini } 10301e6b0712SBarry Smith 103163602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 103263602bcaSStefano Zampini { 103363602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 103463602bcaSStefano Zampini PetscInt i; 103556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 103663602bcaSStefano Zampini PetscErrorCode ierr; 103763602bcaSStefano Zampini 103863602bcaSStefano Zampini PetscFunctionBegin; 103956282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 104056282151SStefano Zampini for (i=0;i<n_is;i++) { 104156282151SStefano Zampini PetscBool isequalt; 104256282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 104356282151SStefano Zampini if (!isequalt) break; 104456282151SStefano Zampini } 104556282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 104656282151SStefano Zampini } 104756282151SStefano Zampini for (i=0;i<n_is;i++) { 104856282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 104956282151SStefano Zampini } 105063602bcaSStefano Zampini /* Destroy ISes if they were already set */ 105163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 105263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 105363602bcaSStefano Zampini } 105463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 105563602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 105663602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 105763602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 105863602bcaSStefano Zampini } 105963602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 106063602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 106163602bcaSStefano Zampini /* allocate space then set */ 1062d02579f5SStefano Zampini if (n_is) { 1063d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 1064d02579f5SStefano Zampini } 106563602bcaSStefano Zampini for (i=0;i<n_is;i++) { 106663602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 106763602bcaSStefano Zampini } 106863602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 106963602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 107056282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 107163602bcaSStefano Zampini PetscFunctionReturn(0); 107263602bcaSStefano Zampini } 107363602bcaSStefano Zampini 107463602bcaSStefano Zampini /*@ 107563602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 107663602bcaSStefano Zampini 107763602bcaSStefano Zampini Collective 107863602bcaSStefano Zampini 107963602bcaSStefano Zampini Input Parameters: 108063602bcaSStefano Zampini + pc - the preconditioning context 10810f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10820f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 108363602bcaSStefano Zampini 108463602bcaSStefano Zampini Level: intermediate 108563602bcaSStefano Zampini 10860f202f7eSStefano Zampini Notes: 10870f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 108863602bcaSStefano Zampini 108963602bcaSStefano Zampini .seealso: PCBDDC 109063602bcaSStefano Zampini @*/ 109163602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 109263602bcaSStefano Zampini { 109363602bcaSStefano Zampini PetscInt i; 109463602bcaSStefano Zampini PetscErrorCode ierr; 109563602bcaSStefano Zampini 109663602bcaSStefano Zampini PetscFunctionBegin; 109763602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 109863602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 109963602bcaSStefano Zampini for (i=0;i<n_is;i++) { 110063602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 110163602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 110263602bcaSStefano Zampini } 1103e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 110463602bcaSStefano Zampini PetscFunctionReturn(0); 110563602bcaSStefano Zampini } 110663602bcaSStefano Zampini 11079c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 11089c0446d6SStefano Zampini { 11099c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 11109c0446d6SStefano Zampini PetscInt i; 111156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 11129c0446d6SStefano Zampini PetscErrorCode ierr; 11139c0446d6SStefano Zampini 11149c0446d6SStefano Zampini PetscFunctionBegin; 111556282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 111656282151SStefano Zampini for (i=0;i<n_is;i++) { 111756282151SStefano Zampini PetscBool isequalt; 111856282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 111956282151SStefano Zampini if (!isequalt) break; 112056282151SStefano Zampini } 112156282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 112256282151SStefano Zampini } 112356282151SStefano Zampini for (i=0;i<n_is;i++) { 112456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 112556282151SStefano Zampini } 1126da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 11279c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 11289c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 11299c0446d6SStefano Zampini } 1130d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 113163602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 113263602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 113363602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 113463602bcaSStefano Zampini } 113563602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 113663602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1137da1bb401SStefano Zampini /* allocate space then set */ 1138d02579f5SStefano Zampini if (n_is) { 1139785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1140d02579f5SStefano Zampini } 11419c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1142da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 11439c0446d6SStefano Zampini } 11449c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 114563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 114656282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 11479c0446d6SStefano Zampini PetscFunctionReturn(0); 11489c0446d6SStefano Zampini } 11491e6b0712SBarry Smith 11509c0446d6SStefano Zampini /*@ 115163602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 11529c0446d6SStefano Zampini 115363602bcaSStefano Zampini Collective 11549c0446d6SStefano Zampini 11559c0446d6SStefano Zampini Input Parameters: 11569c0446d6SStefano Zampini + pc - the preconditioning context 11570f202f7eSStefano Zampini . n_is - number of index sets defining the fields 11580f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 11599c0446d6SStefano Zampini 11609c0446d6SStefano Zampini Level: intermediate 11619c0446d6SStefano Zampini 11620f202f7eSStefano Zampini Notes: 11630f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 11649c0446d6SStefano Zampini 11659c0446d6SStefano Zampini .seealso: PCBDDC 11669c0446d6SStefano Zampini @*/ 11679c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 11689c0446d6SStefano Zampini { 11692b510759SStefano Zampini PetscInt i; 11709c0446d6SStefano Zampini PetscErrorCode ierr; 11719c0446d6SStefano Zampini 11729c0446d6SStefano Zampini PetscFunctionBegin; 11739c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 117463602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 11752b510759SStefano Zampini for (i=0;i<n_is;i++) { 117663602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 1177a011d5a7Sstefano_zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 11782b510759SStefano Zampini } 11799c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 11809c0446d6SStefano Zampini PetscFunctionReturn(0); 11819c0446d6SStefano Zampini } 1182906d46d4SStefano Zampini 1183534831adSStefano Zampini /* 1184534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1185534831adSStefano Zampini guess if a transformation of basis approach has been selected. 11869c0446d6SStefano Zampini 1187534831adSStefano Zampini Input Parameter: 1188534831adSStefano Zampini + pc - the preconditioner contex 1189534831adSStefano Zampini 1190534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1191534831adSStefano Zampini 1192534831adSStefano Zampini Notes: 1193534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1194534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1195534831adSStefano Zampini */ 1196534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1197534831adSStefano Zampini { 1198534831adSStefano Zampini PetscErrorCode ierr; 1199534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1200534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 12013972b0daSStefano Zampini Vec used_vec; 120227b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1203534831adSStefano Zampini 1204534831adSStefano Zampini PetscFunctionBegin; 12051f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 120685c4d303SStefano Zampini if (ksp) { 12071f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 120885c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 120927b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 121027b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1211f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 12121f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 121385c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 121485c4d303SStefano Zampini } 121585c4d303SStefano Zampini } 1216fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1217fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1218fc17d649SStefano Zampini } 12191f4df5f7SStefano Zampini 122085c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 122162a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 122262a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 122362a6ff1dSStefano Zampini } 122462a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 122562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 122662a6ff1dSStefano Zampini } 12278d00608fSStefano Zampini 122827b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 12293972b0daSStefano Zampini if (x) { 12303972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 12313972b0daSStefano Zampini used_vec = x; 12328d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 12333972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 12343972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 12353972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 123627b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1237266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1238266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1239266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 12403972b0daSStefano Zampini } 12418efcfb23SStefano Zampini 12428efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 12433972b0daSStefano Zampini if (ksp) { 1244a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 12458efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12468efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 12473972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 12483972b0daSStefano Zampini } 12493972b0daSStefano Zampini } 12503308cffdSStefano Zampini 12518d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12523972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 125370c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 12543975b054SStefano Zampini IS dirIS = NULL; 12553975b054SStefano Zampini 1256a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 12573975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 12583975b054SStefano Zampini if (dirIS) { 1259906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1260785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 12612b095fd8SStefano Zampini PetscScalar *array_x; 12622b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1263785d1243SStefano Zampini 12643972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 12653972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1266e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1267e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1268e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1269e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 127082ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 12713972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 12722b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12733972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12742fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 12753972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12762b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12773972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1278e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1279e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12808d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12811b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 12828efcfb23SStefano Zampini } 1283a07ea27aSStefano Zampini } 1284b76ba322SStefano Zampini 12858efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 12868d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 128727b6a85dSStefano Zampini /* save the original rhs */ 128827b6a85dSStefano Zampini if (save_rhs) { 128927b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 129027b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 12918d00608fSStefano Zampini } 12928d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12933972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 129427b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 12953972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 12968efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 129727b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 12987acc28cbSStefano Zampini if (ksp) { 12997acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 13007acc28cbSStefano Zampini } 13013308cffdSStefano Zampini } 13028efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1303b76ba322SStefano Zampini 1304fc17d649SStefano Zampini /* compute initial vector in benign space if needed 130527b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 130627b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 130727b6a85dSStefano Zampini if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) { 13081f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 13091f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 13100369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 13110369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 13120369aaf7SStefano Zampini } 13134fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 131427b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 13151dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 13163bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 13171f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 13181f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13191f4df5f7SStefano Zampini } 13201f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 132127b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 132227b6a85dSStefano Zampini if (save_rhs) { 132327b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 132492e3dcfbSStefano Zampini } 132527b6a85dSStefano Zampini if (pcbddc->rhs_change) { 13260369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 132727b6a85dSStefano Zampini } else { 132827b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 132927b6a85dSStefano Zampini } 13300369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 133127b6a85dSStefano Zampini } 133227b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 13330369aaf7SStefano Zampini } 13342d4c4fecSStefano Zampini 13352d4c4fecSStefano Zampini /* dbg output */ 1336a198735bSStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 13371f4df5f7SStefano Zampini Vec v; 13381f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 13391f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 13401f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1341e0fe2d75SToby Isaac ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %D: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr); 13422d4c4fecSStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr); 13431f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 13441f4df5f7SStefano Zampini } 13450369aaf7SStefano Zampini 13460369aaf7SStefano Zampini /* set initial guess if using PCG */ 13478ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 13480369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 13490369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 13501dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 135127b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 13521dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13531dd7afcfSStefano Zampini } else { 13541dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 13551dd7afcfSStefano Zampini } 13561dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13571dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13581dd7afcfSStefano Zampini } else { 13590369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13600369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13611dd7afcfSStefano Zampini } 13620369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13631dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 13641dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 13651dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13661dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13671dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 13681dd7afcfSStefano Zampini } else { 13690369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13700369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13711dd7afcfSStefano Zampini } 13720369aaf7SStefano Zampini if (ksp) { 13730369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 13740369aaf7SStefano Zampini } 13758ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_TRUE; 1376266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1377266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13780369aaf7SStefano Zampini } 1379534831adSStefano Zampini PetscFunctionReturn(0); 1380534831adSStefano Zampini } 1381906d46d4SStefano Zampini 1382534831adSStefano Zampini /* 1383534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1384534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1385534831adSStefano Zampini 1386534831adSStefano Zampini Input Parameter: 1387534831adSStefano Zampini + pc - the preconditioner contex 1388534831adSStefano Zampini 1389534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1390534831adSStefano Zampini 1391534831adSStefano Zampini Notes: 1392534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1393534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1394534831adSStefano Zampini */ 1395534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1396534831adSStefano Zampini { 1397534831adSStefano Zampini PetscErrorCode ierr; 1398534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1399534831adSStefano Zampini 1400534831adSStefano Zampini PetscFunctionBegin; 14013972b0daSStefano Zampini /* add solution removed in presolve */ 14026bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 140327b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 14043425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1405af140850Sstefano_zampini } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) { 140627b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14073425bc38SStefano Zampini } 1408af140850Sstefano_zampini /* restore to original state (not for FETI-DP) */ 1409af140850Sstefano_zampini if (ksp) pcbddc->temp_solution_used = PETSC_FALSE; 141027b6a85dSStefano Zampini } 141127b6a85dSStefano Zampini 1412266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 14138d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 141427b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 14158d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 1416af140850Sstefano_zampini } 14178efcfb23SStefano Zampini /* restore ksp guess state */ 14188efcfb23SStefano Zampini if (ksp) { 14198efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 14208ae0ca82SStefano Zampini /* reset flag for exact dirichlet trick */ 14218ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 1422af140850Sstefano_zampini } 1423534831adSStefano Zampini PetscFunctionReturn(0); 1424534831adSStefano Zampini } 1425af140850Sstefano_zampini 14260c7d97c5SJed Brown /* 14270c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 14280c7d97c5SJed Brown by setting data structures and options. 14290c7d97c5SJed Brown 14300c7d97c5SJed Brown Input Parameter: 143153cdbc3dSStefano Zampini + pc - the preconditioner context 14320c7d97c5SJed Brown 14330c7d97c5SJed Brown Application Interface Routine: PCSetUp() 14340c7d97c5SJed Brown 14350c7d97c5SJed Brown Notes: 14360c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 14370c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 14380c7d97c5SJed Brown */ 143953cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 14400c7d97c5SJed Brown { 14410c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1442c703fcc7SStefano Zampini PCBDDCSubSchurs sub_schurs; 14435e8657edSStefano Zampini Mat_IS* matis; 144408122e43SStefano Zampini MatNullSpace nearnullspace; 144535509ce9Sstefano_zampini Mat lA; 144635509ce9Sstefano_zampini IS lP,zerodiag = NULL; 144791e8d312SStefano Zampini PetscInt nrows,ncols; 1448c703fcc7SStefano Zampini PetscBool computesubschurs; 14498de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 14505e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 1451c703fcc7SStefano Zampini PetscErrorCode ierr; 14520c7d97c5SJed Brown 14530c7d97c5SJed Brown PetscFunctionBegin; 14545e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 14556c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 145691e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 14576c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 14585e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1459f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 14603b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 146171582508SStefano Zampini Also, BDDC builds its own KSP for the Dirichlet problem */ 1462a485753aSstefano_zampini if (!pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE; 1463c83e1ba7SStefano Zampini if (pcbddc->recompute_topography) { 1464c83e1ba7SStefano Zampini pcbddc->graphanalyzed = PETSC_FALSE; 1465c83e1ba7SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1466c83e1ba7SStefano Zampini } else { 14678de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1468c83e1ba7SStefano Zampini } 1469b087196eSStefano Zampini 1470b087196eSStefano Zampini /* check parameters' compatibility */ 1471b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 14729d54b7f4SStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0); 1473bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1474862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1475862806e4SStefano Zampini 14765a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 147716909a7fSStefano Zampini if (pcbddc->switch_static) { 147816909a7fSStefano Zampini PetscBool ismatis; 147916909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 148016909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 148116909a7fSStefano Zampini } 148216909a7fSStefano Zampini 148371582508SStefano Zampini /* activate all connected components if the netflux has been requested */ 1484bb05f991SStefano Zampini if (pcbddc->compute_nonetflux) { 1485bb05f991SStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 1486bb05f991SStefano Zampini pcbddc->use_edges = PETSC_TRUE; 1487bb05f991SStefano Zampini pcbddc->use_faces = PETSC_TRUE; 1488bb05f991SStefano Zampini } 1489bb05f991SStefano Zampini 1490f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 149170cf5478SStefano Zampini if (pcbddc->dbg_flag) { 149270cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 149358a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 14941575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1495f4ddd8eeSStefano Zampini } 149658a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1497f4ddd8eeSStefano Zampini } 1498f4ddd8eeSStefano Zampini 1499c703fcc7SStefano Zampini /* process topology information */ 150071582508SStefano Zampini if (pcbddc->recompute_topography) { 150171582508SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 1502c703fcc7SStefano Zampini if (pcbddc->discretegradient) { 1503a13144ffSStefano Zampini ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr); 1504a13144ffSStefano Zampini } 1505c703fcc7SStefano Zampini } 1506a13144ffSStefano Zampini 1507c703fcc7SStefano Zampini /* change basis if requested by the user */ 15085e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 15095e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 15105e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 15115e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 15125e8657edSStefano Zampini } else { 1513b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15145e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15155e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1516d16cbb6bSStefano Zampini } 1517d16cbb6bSStefano Zampini 15184f1b2e48SStefano Zampini /* 1519c703fcc7SStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick 15204f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 15214f1b2e48SStefano Zampini */ 15221dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1523d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 15249f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 15259f47a83aSStefano Zampini 152605b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1527669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 1528339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1529a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1530c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 15311dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 15321dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 15331dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 15341dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 15351dd7afcfSStefano Zampini } else { 1536a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 15371dd7afcfSStefano Zampini } 1538a3df083aSStefano Zampini } else { 15399f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1540674ae819SStefano Zampini } 1541a3df083aSStefano Zampini } 154227b6a85dSStefano Zampini 1543c703fcc7SStefano Zampini /* propagate relevant information -> TODO remove*/ 15444f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 15453301b35fSStefano Zampini if (matis->A->symmetric_set) { 15463301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 15473301b35fSStefano Zampini } 1548e496cd5dSStefano Zampini #endif 154906a4e24aSStefano Zampini if (matis->A->symmetric_set) { 155006a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 155106a4e24aSStefano Zampini } 155206a4e24aSStefano Zampini if (matis->A->spd_set) { 155306a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 155406a4e24aSStefano Zampini } 1555e496cd5dSStefano Zampini 15565e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 15575e8657edSStefano Zampini { 15585e8657edSStefano Zampini Mat temp_mat; 15595e8657edSStefano Zampini 15605e8657edSStefano Zampini temp_mat = matis->A; 15615e8657edSStefano Zampini matis->A = pcbddc->local_mat; 15625e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 15635e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15645e8657edSStefano Zampini matis->A = temp_mat; 15655e8657edSStefano Zampini } 1566684f6988SStefano Zampini 156781d14e9dSStefano Zampini /* Analyze interface */ 156864ac59b8SStefano Zampini if (!pcbddc->graphanalyzed) { 1569674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15708de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1571345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 15724247aa23Sstefano_zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1573345ecf6cSStefano Zampini } 1574a198735bSStefano Zampini if (pcbddc->compute_nonetflux) { 1575669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1576669cc0f4SStefano Zampini 157721ef3d20SStefano Zampini if (!pcbddc->divudotp) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Missing divudotp operator"); 15788ae0ca82SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 157971582508SStefano Zampini /* TODO what if a nearnullspace is already attached? */ 1580669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1581669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1582669cc0f4SStefano Zampini } 1583674ae819SStefano Zampini } 1584fb8d54d4SStefano Zampini 15855408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 15865408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 15875408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 1588ff1f7e73Sstefano_zampini if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) { 15895408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 159009f581a4SStefano Zampini } 15914f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 159206f24817SStefano Zampini 1593b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1594c703fcc7SStefano Zampini if (computesubschurs && pcbddc->recompute_topography) { 159508122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1596b1b3d7a2SStefano Zampini } 15979d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 15989d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 15999d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16009d54b7f4SStefano Zampini } 1601c703fcc7SStefano Zampini 1602c703fcc7SStefano Zampini /* finish setup solvers and do adaptive selection of constraints */ 1603b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1604b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 16052070dbb6SStefano Zampini if (computesubschurs) { 160608122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 16072070dbb6SStefano Zampini } 1608d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1609d5574798SStefano Zampini } else { 1610d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 16112070dbb6SStefano Zampini if (computesubschurs) { 1612d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1613d5574798SStefano Zampini } 16142070dbb6SStefano Zampini } 161508122e43SStefano Zampini if (pcbddc->adaptive_selection) { 161608122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 16178de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1618b7eb3628SStefano Zampini } 1619684f6988SStefano Zampini 1620f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1621fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1622f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1623f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1624f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1625f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1626f4ddd8eeSStefano Zampini } else { 1627f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1628f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1629f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1630165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1631f4ddd8eeSStefano Zampini PetscInt i; 1632165b64e2SStefano Zampini const Vec *nearnullvecs; 1633165b64e2SStefano Zampini PetscObjectState state; 1634165b64e2SStefano Zampini PetscInt nnsp_size; 1635165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1636f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1637f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1638165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1639f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1640f4ddd8eeSStefano Zampini break; 1641f4ddd8eeSStefano Zampini } 1642f4ddd8eeSStefano Zampini } 1643f4ddd8eeSStefano Zampini } 1644f4ddd8eeSStefano Zampini } 1645f4ddd8eeSStefano Zampini } else { 1646f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1647f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1648f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1649f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1650f4ddd8eeSStefano Zampini } 1651f4ddd8eeSStefano Zampini } 1652f4ddd8eeSStefano Zampini 1653f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1654727cdba6SStefano Zampini /* reset primal space flags */ 1655f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1656727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 16578de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1658727cdba6SStefano Zampini /* It also sets the primal space flags */ 1659674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 16609543d0ffSStefano Zampini } 1661e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1662f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 16635e8657edSStefano Zampini 16645e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16655e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 16665e8657edSStefano Zampini 16675e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16684f1b2e48SStefano Zampini if (pcbddc->benign_change) { 16691dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1670c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1671c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1672c263805aSStefano Zampini } 16735e8657edSStefano Zampini /* get submatrices */ 16745e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16755e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16765e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16777dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16787dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16797dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16803975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16813975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16829c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1683b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16845e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16855e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16865e8657edSStefano Zampini } 168735509ce9Sstefano_zampini 168835509ce9Sstefano_zampini /* interface pressure block row for B_C */ 168935509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr); 169035509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr); 169135509ce9Sstefano_zampini if (lA && lP) { 169235509ce9Sstefano_zampini PC_IS* pcis = (PC_IS*)pc->data; 169335509ce9Sstefano_zampini Mat B_BI,B_BB,Bt_BI,Bt_BB; 169435509ce9Sstefano_zampini PetscBool issym; 169535509ce9Sstefano_zampini ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr); 16966cc1294bSstefano_zampini if (issym) { 16977dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 16987dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 169935509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr); 170035509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr); 170135509ce9Sstefano_zampini } else { 17027dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17037dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 17047dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr); 17057dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr); 170635509ce9Sstefano_zampini } 170735509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr); 170835509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr); 170935509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr); 171035509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr); 171135509ce9Sstefano_zampini ierr = MatDestroy(&B_BI);CHKERRQ(ierr); 171235509ce9Sstefano_zampini ierr = MatDestroy(&B_BB);CHKERRQ(ierr); 171335509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr); 171435509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr); 171535509ce9Sstefano_zampini } 171635509ce9Sstefano_zampini 1717b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 171899cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1719b96c3477SStefano Zampini /* SetUp Scaling operator */ 17209d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1721674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 17220c7d97c5SJed Brown } 1723c703fcc7SStefano Zampini 17241dd7afcfSStefano Zampini /* mark topography as done */ 172556282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 17260369aaf7SStefano Zampini 17271dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 17281dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 17291dd7afcfSStefano Zampini 173058a03d70SStefano Zampini if (pcbddc->dbg_flag) { 173158a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 17322b510759SStefano Zampini } 17330c7d97c5SJed Brown PetscFunctionReturn(0); 17340c7d97c5SJed Brown } 17350c7d97c5SJed Brown 17360c7d97c5SJed Brown /* 173750efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 17380c7d97c5SJed Brown 17390c7d97c5SJed Brown Input Parameters: 17400f202f7eSStefano Zampini + pc - the preconditioner context 17410f202f7eSStefano Zampini - r - input vector (global) 17420c7d97c5SJed Brown 17430c7d97c5SJed Brown Output Parameter: 17440c7d97c5SJed Brown . z - output vector (global) 17450c7d97c5SJed Brown 17460c7d97c5SJed Brown Application Interface Routine: PCApply() 17470c7d97c5SJed Brown */ 174853cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 17490c7d97c5SJed Brown { 17500c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 17510c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1752b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 17530c7d97c5SJed Brown PetscErrorCode ierr; 17543b03a366Sstefano_zampini const PetscScalar one = 1.0; 17553b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 17562617d88aSStefano Zampini const PetscScalar zero = 0.0; 17570c7d97c5SJed Brown 17580c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 17590c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1760b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 17610c7d97c5SJed Brown 17620c7d97c5SJed Brown PetscFunctionBegin; 1763f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 17641dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 17651dd7afcfSStefano Zampini Vec swap; 176627b6a85dSStefano Zampini 176727b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 17681dd7afcfSStefano Zampini swap = pcbddc->work_change; 17691dd7afcfSStefano Zampini pcbddc->work_change = r; 17701dd7afcfSStefano Zampini r = swap; 17711dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 17729cc2a9b1Sstefano_zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 17731dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 17741dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 17751dd7afcfSStefano Zampini } 17761dd7afcfSStefano Zampini } 177727b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1778015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1779efc2fbd9SStefano Zampini } 17808ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1781b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 17820c7d97c5SJed Brown /* First Dirichlet solve */ 17830c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17840c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17850c7d97c5SJed Brown /* 17860c7d97c5SJed Brown Assembling right hand side for BDDC operator 1787b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1788674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 17890c7d97c5SJed Brown */ 1790b097fa66SStefano Zampini if (n_D) { 1791b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 17920c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 179316909a7fSStefano Zampini if (pcbddc->switch_static) { 179416909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 179516909a7fSStefano Zampini 179616909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 179716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 179816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 179916909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 180016909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 180116909a7fSStefano Zampini } else { 180216909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 180316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 180416909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 180516909a7fSStefano Zampini } 180616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 180716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 180816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 180916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 181016909a7fSStefano Zampini } else { 1811b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 181216909a7fSStefano Zampini } 1813b097fa66SStefano Zampini } else { 1814b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1815b097fa66SStefano Zampini } 18160c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18170c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1818674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1819b76ba322SStefano Zampini } else { 18204fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1821674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1822b76ba322SStefano Zampini } 18234fee134fSStefano Zampini } 1824b76ba322SStefano Zampini 18252617d88aSStefano Zampini /* Apply interface preconditioner 18262617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1827dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 18282617d88aSStefano Zampini 1829674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1830674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 18310c7d97c5SJed Brown 18323b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 18330c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18340c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1835b097fa66SStefano Zampini if (n_B) { 183616909a7fSStefano Zampini if (pcbddc->switch_static) { 183716909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 183816909a7fSStefano Zampini 183916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 184016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 184116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 184216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 184316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 184416909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 184516909a7fSStefano Zampini } else { 184616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 184716909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 184816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 184916909a7fSStefano Zampini } 185016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 185116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 185216909a7fSStefano Zampini } else { 18530c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 185416909a7fSStefano Zampini } 185516909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 185616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 185716909a7fSStefano Zampini 185816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 185916909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 186016909a7fSStefano Zampini } else { 186116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 186216909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 186416909a7fSStefano Zampini } 1865b097fa66SStefano Zampini } 1866df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1867efc2fbd9SStefano Zampini 18688ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1869b097fa66SStefano Zampini if (pcbddc->switch_static) { 1870b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1871b097fa66SStefano Zampini } else { 1872b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1873b097fa66SStefano Zampini } 18740c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18750c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1876b097fa66SStefano Zampini } else { 1877b097fa66SStefano Zampini if (pcbddc->switch_static) { 1878b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1879b097fa66SStefano Zampini } else { 1880b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1881b097fa66SStefano Zampini } 1882b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1883b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1884b097fa66SStefano Zampini } 188527b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 18861dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 18871dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 18881dd7afcfSStefano Zampini } 1889015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1890efc2fbd9SStefano Zampini } 18911f4df5f7SStefano Zampini 18921dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1893f913dca9SStefano Zampini pcbddc->work_change = r; 18941dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 18951dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 18961dd7afcfSStefano Zampini } 18970c7d97c5SJed Brown PetscFunctionReturn(0); 18980c7d97c5SJed Brown } 189950efa1b5SStefano Zampini 190050efa1b5SStefano Zampini /* 190150efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 190250efa1b5SStefano Zampini 190350efa1b5SStefano Zampini Input Parameters: 19040f202f7eSStefano Zampini + pc - the preconditioner context 19050f202f7eSStefano Zampini - r - input vector (global) 190650efa1b5SStefano Zampini 190750efa1b5SStefano Zampini Output Parameter: 190850efa1b5SStefano Zampini . z - output vector (global) 190950efa1b5SStefano Zampini 191050efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 191150efa1b5SStefano Zampini */ 191250efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 191350efa1b5SStefano Zampini { 191450efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 191550efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1916b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 191750efa1b5SStefano Zampini PetscErrorCode ierr; 191850efa1b5SStefano Zampini const PetscScalar one = 1.0; 191950efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 192050efa1b5SStefano Zampini const PetscScalar zero = 0.0; 192150efa1b5SStefano Zampini 192250efa1b5SStefano Zampini PetscFunctionBegin; 1923f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 19241dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19251dd7afcfSStefano Zampini Vec swap; 192627b6a85dSStefano Zampini 192727b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 19281dd7afcfSStefano Zampini swap = pcbddc->work_change; 19291dd7afcfSStefano Zampini pcbddc->work_change = r; 19301dd7afcfSStefano Zampini r = swap; 193127b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 19328ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 193327b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 193427b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 19351dd7afcfSStefano Zampini } 193627b6a85dSStefano Zampini } 193727b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1938537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1939537c1cdfSStefano Zampini } 19408ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1941b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 194250efa1b5SStefano Zampini /* First Dirichlet solve */ 194350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 194450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 194550efa1b5SStefano Zampini /* 194650efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1947b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 194850efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 194950efa1b5SStefano Zampini */ 1950b097fa66SStefano Zampini if (n_D) { 1951b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 195250efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 195316909a7fSStefano Zampini if (pcbddc->switch_static) { 195416909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 195516909a7fSStefano Zampini 195616909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 195716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 195816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 195916909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 196016909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 196116909a7fSStefano Zampini } else { 196216909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 196316909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 196416909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 196516909a7fSStefano Zampini } 196616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 197016909a7fSStefano Zampini } else { 1971b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 197216909a7fSStefano Zampini } 1973b097fa66SStefano Zampini } else { 1974b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1975b097fa66SStefano Zampini } 197650efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 197750efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 197850efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 197950efa1b5SStefano Zampini } else { 198050efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 198150efa1b5SStefano Zampini } 198250efa1b5SStefano Zampini 198350efa1b5SStefano Zampini /* Apply interface preconditioner 198450efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1985dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 198650efa1b5SStefano Zampini 198750efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 198850efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 198950efa1b5SStefano Zampini 199050efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 199150efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199250efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1993b097fa66SStefano Zampini if (n_B) { 199416909a7fSStefano Zampini if (pcbddc->switch_static) { 199516909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 199616909a7fSStefano Zampini 199716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 199816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 199916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 200016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 200116909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 200216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 200316909a7fSStefano Zampini } else { 200416909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 200516909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 200616909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 200716909a7fSStefano Zampini } 200816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 200916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 201016909a7fSStefano Zampini } else { 201150efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 201216909a7fSStefano Zampini } 201316909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 201416909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 201516909a7fSStefano Zampini 201616909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 201716909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 201816909a7fSStefano Zampini } else { 201916909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 202016909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202116909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 202216909a7fSStefano Zampini } 2023b097fa66SStefano Zampini } 2024b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 20258ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2026b097fa66SStefano Zampini if (pcbddc->switch_static) { 2027b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 2028b097fa66SStefano Zampini } else { 2029b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 2030b097fa66SStefano Zampini } 203150efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 203250efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2033b097fa66SStefano Zampini } else { 2034b097fa66SStefano Zampini if (pcbddc->switch_static) { 2035b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 2036b097fa66SStefano Zampini } else { 2037b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 2038b097fa66SStefano Zampini } 2039b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2040b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2041b097fa66SStefano Zampini } 204227b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 2043537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 2044537c1cdfSStefano Zampini } 20451dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2046f913dca9SStefano Zampini pcbddc->work_change = r; 20471dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 20481dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 20491dd7afcfSStefano Zampini } 205050efa1b5SStefano Zampini PetscFunctionReturn(0); 205150efa1b5SStefano Zampini } 2052674ae819SStefano Zampini 20539326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc) 2054da1bb401SStefano Zampini { 2055da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 20569326c5c6Sstefano_zampini PC_IS *pcis = (PC_IS*)pc->data; 20579326c5c6Sstefano_zampini KSP kspD,kspR,kspC; 2058da1bb401SStefano Zampini PetscErrorCode ierr; 2059da1bb401SStefano Zampini 2060da1bb401SStefano Zampini PetscFunctionBegin; 2061674ae819SStefano Zampini /* free BDDC custom data */ 2062674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2063674ae819SStefano Zampini /* destroy objects related to topography */ 2064674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 206534a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2066674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 2067674ae819SStefano Zampini /* free solvers stuff */ 2068674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 206962a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 207062a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 207162a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 20721dd7afcfSStefano Zampini /* free data created by PCIS */ 20731dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 20749326c5c6Sstefano_zampini 20759326c5c6Sstefano_zampini /* restore defaults */ 20769326c5c6Sstefano_zampini kspD = pcbddc->ksp_D; 20779326c5c6Sstefano_zampini kspR = pcbddc->ksp_R; 20789326c5c6Sstefano_zampini kspC = pcbddc->coarse_ksp; 20799326c5c6Sstefano_zampini ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr); 20809326c5c6Sstefano_zampini pcis->n_neigh = -1; 20819326c5c6Sstefano_zampini pcis->scaling_factor = 1.0; 20829326c5c6Sstefano_zampini pcis->reusesubmatrices = PETSC_TRUE; 20839326c5c6Sstefano_zampini pcbddc->use_local_adj = PETSC_TRUE; 20849326c5c6Sstefano_zampini pcbddc->use_vertices = PETSC_TRUE; 20859326c5c6Sstefano_zampini pcbddc->use_edges = PETSC_TRUE; 20869326c5c6Sstefano_zampini pcbddc->symmetric_primal = PETSC_TRUE; 20879326c5c6Sstefano_zampini pcbddc->vertex_size = 1; 20889326c5c6Sstefano_zampini pcbddc->recompute_topography = PETSC_TRUE; 20899326c5c6Sstefano_zampini pcbddc->coarse_size = -1; 20909326c5c6Sstefano_zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 20919326c5c6Sstefano_zampini pcbddc->coarsening_ratio = 8; 20929326c5c6Sstefano_zampini pcbddc->coarse_eqs_per_proc = 1; 20939326c5c6Sstefano_zampini pcbddc->benign_compute_correction = PETSC_TRUE; 20949326c5c6Sstefano_zampini pcbddc->nedfield = -1; 20959326c5c6Sstefano_zampini pcbddc->nedglobal = PETSC_TRUE; 20969326c5c6Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 20979326c5c6Sstefano_zampini pcbddc->sub_schurs_layers = -1; 20989326c5c6Sstefano_zampini pcbddc->ksp_D = kspD; 20999326c5c6Sstefano_zampini pcbddc->ksp_R = kspR; 21009326c5c6Sstefano_zampini pcbddc->coarse_ksp = kspC; 21019326c5c6Sstefano_zampini PetscFunctionReturn(0); 21029326c5c6Sstefano_zampini } 21039326c5c6Sstefano_zampini 21049326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc) 21059326c5c6Sstefano_zampini { 21069326c5c6Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 21079326c5c6Sstefano_zampini PetscErrorCode ierr; 21089326c5c6Sstefano_zampini 21099326c5c6Sstefano_zampini PetscFunctionBegin; 21109326c5c6Sstefano_zampini ierr = PCReset_BDDC(pc);CHKERRQ(ierr); 21119326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr); 21129326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr); 21139326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr); 2114a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr); 2115a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr); 2116906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2117674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 211830368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2119bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 21202b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2121b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 21222b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2123bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 212482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2125bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 212682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2127bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 212882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2129bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2130785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2131bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 213263602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2133bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2134bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2135bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2136bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2137a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2138674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2139da1bb401SStefano Zampini PetscFunctionReturn(0); 2140da1bb401SStefano Zampini } 21411e6b0712SBarry Smith 2142a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2143a06fd7f2SStefano Zampini { 2144a06fd7f2SStefano Zampini PetscFunctionBegin; 2145a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2146a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2147a06fd7f2SStefano Zampini } 2148a06fd7f2SStefano Zampini 21493425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21503425bc38SStefano Zampini { 2151674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2152266e20e9SStefano Zampini Vec work; 21533425bc38SStefano Zampini PC_IS* pcis; 21543425bc38SStefano Zampini PC_BDDC* pcbddc; 21553425bc38SStefano Zampini PetscErrorCode ierr; 21560c7d97c5SJed Brown 21573425bc38SStefano Zampini PetscFunctionBegin; 21583425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21593425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21603425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21613425bc38SStefano Zampini 2162229984c5Sstefano_zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 2163229984c5Sstefano_zampini /* copy rhs since we may change it during PCPreSolve_BDDC */ 2164229984c5Sstefano_zampini if (!pcbddc->original_rhs) { 2165229984c5Sstefano_zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 2166229984c5Sstefano_zampini } 21676cc1294bSstefano_zampini if (mat_ctx->rhs_flip) { 21686cc1294bSstefano_zampini ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr); 21696cc1294bSstefano_zampini } else { 2170229984c5Sstefano_zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 21716cc1294bSstefano_zampini } 2172af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2173229984c5Sstefano_zampini /* interface pressure rhs */ 2174022d8d2bSstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2175022d8d2bSstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2176229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2177229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21786cc1294bSstefano_zampini if (!mat_ctx->rhs_flip) { 2179229984c5Sstefano_zampini ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr); 2180229984c5Sstefano_zampini } 21816cc1294bSstefano_zampini } 2182c08af4c6SStefano Zampini /* 2183c08af4c6SStefano Zampini change of basis for physical rhs if needed 2184c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2185c08af4c6SStefano Zampini */ 21863738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2187fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 21883738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 21893738a8e6SStefano Zampini work = pcbddc->work_change; 2190fc17d649SStefano Zampini } else { 21913738a8e6SStefano Zampini work = pcbddc->original_rhs; 2192fc17d649SStefano Zampini } 21933425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2194266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2195266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2196fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2197fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2198266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2199266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2200674ae819SStefano Zampini /* Apply partition of unity */ 22013425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2202266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 22038eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 22043425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 22053425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22063425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 22073425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2208266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2209266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2210266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2211266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2212266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2213266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22143425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 22153425bc38SStefano Zampini } 22163425bc38SStefano Zampini /* BDDC rhs */ 22173425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 22188eeda7d8SStefano Zampini if (pcbddc->switch_static) { 22193425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22203425bc38SStefano Zampini } 22213425bc38SStefano Zampini /* apply BDDC */ 2222229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2223dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2224266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2225229984c5Sstefano_zampini 22263425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 22273425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 22283425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22293425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2230229984c5Sstefano_zampini /* Add contribution to interface pressures */ 2231229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2232229984c5Sstefano_zampini ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr); 2233229984c5Sstefano_zampini if (pcbddc->switch_static) { 2234229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr); 2235229984c5Sstefano_zampini } 2236229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2237229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2238229984c5Sstefano_zampini } 22393425bc38SStefano Zampini PetscFunctionReturn(0); 22403425bc38SStefano Zampini } 22411e6b0712SBarry Smith 22423425bc38SStefano Zampini /*@ 22430f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 22443425bc38SStefano Zampini 22453425bc38SStefano Zampini Collective 22463425bc38SStefano Zampini 22473425bc38SStefano Zampini Input Parameters: 22480f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 22490f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 22503425bc38SStefano Zampini 22513425bc38SStefano Zampini Output Parameters: 22520f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 22533425bc38SStefano Zampini 22543425bc38SStefano Zampini Level: developer 22553425bc38SStefano Zampini 22563425bc38SStefano Zampini Notes: 22573425bc38SStefano Zampini 22580f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 22593425bc38SStefano Zampini @*/ 22603425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 22613425bc38SStefano Zampini { 2262674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22633425bc38SStefano Zampini PetscErrorCode ierr; 22643425bc38SStefano Zampini 22653425bc38SStefano Zampini PetscFunctionBegin; 2266266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2267266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2268266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 22693425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2270163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 22713425bc38SStefano Zampini PetscFunctionReturn(0); 22723425bc38SStefano Zampini } 22731e6b0712SBarry Smith 22743425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22753425bc38SStefano Zampini { 2276674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22773425bc38SStefano Zampini PC_IS* pcis; 22783425bc38SStefano Zampini PC_BDDC* pcbddc; 22793425bc38SStefano Zampini PetscErrorCode ierr; 2280229984c5Sstefano_zampini Vec work; 22813425bc38SStefano Zampini 22823425bc38SStefano Zampini PetscFunctionBegin; 22833425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 22843425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 22853425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 22863425bc38SStefano Zampini 22873425bc38SStefano Zampini /* apply B_delta^T */ 2288af140850Sstefano_zampini ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr); 22893425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22903425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22913425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 2292229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2293229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2294229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2295229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 2296229984c5Sstefano_zampini } 2297229984c5Sstefano_zampini 22983425bc38SStefano Zampini /* compute rhs for BDDC application */ 22993425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 23008eeda7d8SStefano Zampini if (pcbddc->switch_static) { 23013425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 2302229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2303229984c5Sstefano_zampini ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr); 2304229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23053425bc38SStefano Zampini } 2306229984c5Sstefano_zampini } 2307229984c5Sstefano_zampini 23083425bc38SStefano Zampini /* apply BDDC */ 2309229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2310dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2311229984c5Sstefano_zampini 2312229984c5Sstefano_zampini /* put values into global vector */ 2313af140850Sstefano_zampini if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change; 2314af140850Sstefano_zampini else work = standard_sol; 2315229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2316229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23178eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 23183425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 23193425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 232000f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 232100f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23223425bc38SStefano Zampini } 2323229984c5Sstefano_zampini 2324229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2325229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2326266e20e9SStefano Zampini /* add p0 solution to final solution */ 2327229984c5Sstefano_zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr); 2328fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2329af140850Sstefano_zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr); 2330fc17d649SStefano Zampini } 2331af140850Sstefano_zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 2332af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2333229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2334229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2335229984c5Sstefano_zampini } 23363425bc38SStefano Zampini PetscFunctionReturn(0); 23373425bc38SStefano Zampini } 23381e6b0712SBarry Smith 23393425bc38SStefano Zampini /*@ 23400f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 23413425bc38SStefano Zampini 23423425bc38SStefano Zampini Collective 23433425bc38SStefano Zampini 23443425bc38SStefano Zampini Input Parameters: 23450f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 23460f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 23473425bc38SStefano Zampini 23483425bc38SStefano Zampini Output Parameters: 23490f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 23503425bc38SStefano Zampini 23513425bc38SStefano Zampini Level: developer 23523425bc38SStefano Zampini 23533425bc38SStefano Zampini Notes: 23543425bc38SStefano Zampini 23550f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 23563425bc38SStefano Zampini @*/ 23573425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23583425bc38SStefano Zampini { 2359674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23603425bc38SStefano Zampini PetscErrorCode ierr; 23613425bc38SStefano Zampini 23623425bc38SStefano Zampini PetscFunctionBegin; 2363266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2364266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2365266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 23663425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2367163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 23683425bc38SStefano Zampini PetscFunctionReturn(0); 23693425bc38SStefano Zampini } 23701e6b0712SBarry Smith 2371547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc) 23723425bc38SStefano Zampini { 2373674ae819SStefano Zampini 2374674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 23753425bc38SStefano Zampini Mat newmat; 2376674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 23773425bc38SStefano Zampini PC newpc; 2378ce94432eSBarry Smith MPI_Comm comm; 23793425bc38SStefano Zampini PetscErrorCode ierr; 23803425bc38SStefano Zampini 23813425bc38SStefano Zampini PetscFunctionBegin; 2382ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 23833425bc38SStefano Zampini /* FETIDP linear matrix */ 23843425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 23851720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 23863425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 2387a5bb87b3Sstefano_zampini ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 23883425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2389edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 23903425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 2391547c9a8eSstefano_zampini ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr); 2392547c9a8eSstefano_zampini ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr); 23933425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 23943425bc38SStefano Zampini /* FETIDP preconditioner */ 23953425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 23963425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 23973425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 2398e1214c54Sstefano_zampini ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 2399e1214c54Sstefano_zampini ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr); 2400e1214c54Sstefano_zampini ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr); 2401e1214c54Sstefano_zampini if (!fetidpmat_ctx->l2g_lambda_only) { 24023425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 24033425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 24043425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2405edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2406c45b8d2dSstefano_zampini ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr); 24073425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2408e1214c54Sstefano_zampini } else { 2409e1214c54Sstefano_zampini KSP *ksps; 2410e1214c54Sstefano_zampini PC lagpc; 2411e1214c54Sstefano_zampini Mat M,AM,PM; 2412e1214c54Sstefano_zampini PetscInt nn; 2413e1214c54Sstefano_zampini 2414e1214c54Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_PPmat",(PetscObject*)&M);CHKERRQ(ierr); 2415e1214c54Sstefano_zampini ierr = PCSetType(newpc,PCFIELDSPLIT);CHKERRQ(ierr); 2416e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"lag",fetidpmat_ctx->lagrange);CHKERRQ(ierr); 2417e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"p",fetidpmat_ctx->pressure);CHKERRQ(ierr); 2418e1214c54Sstefano_zampini ierr = PCFieldSplitSetType(newpc,PC_COMPOSITE_SCHUR);CHKERRQ(ierr); 241940c75d76SStefano Zampini ierr = PCFieldSplitSetSchurFactType(newpc,PC_FIELDSPLIT_SCHUR_FACT_DIAG);CHKERRQ(ierr); 2420e1214c54Sstefano_zampini ierr = PCFieldSplitSetSchurPre(newpc,PC_FIELDSPLIT_SCHUR_PRE_USER,M);CHKERRQ(ierr); 2421c096484dSStefano Zampini ierr = PCFieldSplitSetSchurScale(newpc,1.0);CHKERRQ(ierr); 2422e1214c54Sstefano_zampini ierr = PCSetFromOptions(newpc);CHKERRQ(ierr); 2423e1214c54Sstefano_zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 2424e1214c54Sstefano_zampini 2425e1214c54Sstefano_zampini /* set the solver for the (0,0) block */ 242640c75d76SStefano Zampini ierr = PCFieldSplitGetSubKSP(newpc,&nn,&ksps);CHKERRQ(ierr); 2427e1214c54Sstefano_zampini ierr = PCCreate(comm,&lagpc);CHKERRQ(ierr); 2428e1214c54Sstefano_zampini ierr = PCSetType(lagpc,PCSHELL);CHKERRQ(ierr); 2429e1214c54Sstefano_zampini ierr = KSPGetOperators(ksps[0],&AM,&PM);CHKERRQ(ierr); 2430e1214c54Sstefano_zampini ierr = PCSetOperators(lagpc,AM,PM);CHKERRQ(ierr); 2431e1214c54Sstefano_zampini ierr = PCShellSetContext(lagpc,fetidppc_ctx);CHKERRQ(ierr); 2432e1214c54Sstefano_zampini ierr = PCShellSetApply(lagpc,FETIDPPCApply);CHKERRQ(ierr); 2433e1214c54Sstefano_zampini ierr = PCShellSetApplyTranspose(lagpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2434e1214c54Sstefano_zampini ierr = PCShellSetView(lagpc,FETIDPPCView);CHKERRQ(ierr); 2435e1214c54Sstefano_zampini ierr = PCShellSetDestroy(lagpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2436e1214c54Sstefano_zampini ierr = KSPSetPC(ksps[0],lagpc);CHKERRQ(ierr); 2437e1214c54Sstefano_zampini ierr = PCDestroy(&lagpc);CHKERRQ(ierr); 2438e1214c54Sstefano_zampini ierr = PetscFree(ksps);CHKERRQ(ierr); 2439e1214c54Sstefano_zampini } 24403425bc38SStefano Zampini /* return pointers for objects created */ 24413425bc38SStefano Zampini *fetidp_mat = newmat; 24423425bc38SStefano Zampini *fetidp_pc = newpc; 24433425bc38SStefano Zampini PetscFunctionReturn(0); 24443425bc38SStefano Zampini } 24451e6b0712SBarry Smith 244694ef8ddeSSatish Balay /*@C 24470f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 24483425bc38SStefano Zampini 24493425bc38SStefano Zampini Collective 24503425bc38SStefano Zampini 24513425bc38SStefano Zampini Input Parameters: 24521720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 2453547c9a8eSstefano_zampini . fully_redundant - true for a fully redundant set of Lagrange multipliers 2454547c9a8eSstefano_zampini - prefix - optional options database prefix for the objects to be created (can be NULL) 245528509bceSStefano Zampini 245628509bceSStefano Zampini Output Parameters: 24570f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 24580f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 245928509bceSStefano Zampini 24603425bc38SStefano Zampini Level: developer 24613425bc38SStefano Zampini 24623425bc38SStefano Zampini Notes: 24630f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 24643425bc38SStefano Zampini 24650f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 24663425bc38SStefano Zampini @*/ 2467547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc) 24683425bc38SStefano Zampini { 24693425bc38SStefano Zampini PetscErrorCode ierr; 24703425bc38SStefano Zampini 24713425bc38SStefano Zampini PetscFunctionBegin; 24723425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 24733425bc38SStefano Zampini if (pc->setupcalled) { 2474547c9a8eSstefano_zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 24754247aa23Sstefano_zampini } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 24763425bc38SStefano Zampini PetscFunctionReturn(0); 24773425bc38SStefano Zampini } 24780c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2479da1bb401SStefano Zampini /*MC 2480da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 24810c7d97c5SJed Brown 248228509bceSStefano Zampini An implementation of the BDDC preconditioner based on 248328509bceSStefano Zampini 248428509bceSStefano Zampini .vb 248528509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 248628509bceSStefano 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 248728509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 24880f202f7eSStefano 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 248928509bceSStefano Zampini .ve 249028509bceSStefano Zampini 249128509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 249228509bceSStefano Zampini 24930f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 249428509bceSStefano Zampini 249528509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 249628509bceSStefano Zampini 2497b6fdb6dfSStefano 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. 2498b6fdb6dfSStefano Zampini 2499c7017625SStefano 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). 250028509bceSStefano Zampini 25010f202f7eSStefano 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() 250230368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 250328509bceSStefano Zampini 25040f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 250528509bceSStefano Zampini 25060f202f7eSStefano 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. 25070f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 250828509bceSStefano Zampini 25090f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 251028509bceSStefano Zampini 2511df4d28bfSStefano 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. 251228509bceSStefano Zampini 25130f202f7eSStefano 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. 25140f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 25150f202f7eSStefano Zampini 25160f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 25170f202f7eSStefano Zampini 25180f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 25190f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 25200f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 25210f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 25220f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 25230f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 25240f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 252528509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 25260f202f7eSStefano 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) 25275459c157SBarry Smith . -pc_bddc_coarse_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) 25280f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 25290f202f7eSStefano 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) 2530df4d28bfSStefano 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) 253128509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 253228509bceSStefano Zampini 253328509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 253428509bceSStefano Zampini .vb 253528509bceSStefano Zampini -pc_bddc_dirichlet_ 253628509bceSStefano Zampini -pc_bddc_neumann_ 253728509bceSStefano Zampini -pc_bddc_coarse_ 253828509bceSStefano Zampini .ve 25390f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 254028509bceSStefano Zampini 25410f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 254228509bceSStefano Zampini .vb 2543312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2544312be037SStefano Zampini -pc_bddc_neumann_lN_ 2545312be037SStefano Zampini -pc_bddc_coarse_lN_ 254628509bceSStefano Zampini .ve 25470f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 25480f202f7eSStefano 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. 25490f202f7eSStefano Zampini .vb 25500f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 25510f202f7eSStefano Zampini .ve 25520f202f7eSStefano 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 2553da1bb401SStefano Zampini 2554da1bb401SStefano Zampini Level: intermediate 2555da1bb401SStefano Zampini 2556b6fdb6dfSStefano Zampini Developer notes: 2557da1bb401SStefano Zampini 2558da1bb401SStefano Zampini Contributed by Stefano Zampini 2559da1bb401SStefano Zampini 2560da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2561da1bb401SStefano Zampini M*/ 2562b2573a8aSBarry Smith 25638cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2564da1bb401SStefano Zampini { 2565da1bb401SStefano Zampini PetscErrorCode ierr; 2566da1bb401SStefano Zampini PC_BDDC *pcbddc; 2567da1bb401SStefano Zampini 2568da1bb401SStefano Zampini PetscFunctionBegin; 2569b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2570da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2571da1bb401SStefano Zampini 2572da1bb401SStefano Zampini /* create PCIS data structure */ 2573da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2574da1bb401SStefano Zampini 25759326c5c6Sstefano_zampini /* create local graph structure */ 25769326c5c6Sstefano_zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 25779326c5c6Sstefano_zampini 25789326c5c6Sstefano_zampini /* BDDC nonzero defaults */ 257908a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 258047d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 258147d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 25823301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 258314f95afaSStefano Zampini pcbddc->vertex_size = 1; 2584c703fcc7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 258568457ee5SStefano Zampini pcbddc->coarse_size = -1; 258685c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 258747d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 258857de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 258927b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 25901e0482f5SStefano Zampini pcbddc->nedfield = -1; 25911e0482f5SStefano Zampini pcbddc->nedglobal = PETSC_TRUE; 2592be12c134Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 2593b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2594b7eb3628SStefano Zampini 2595da1bb401SStefano Zampini /* function pointers */ 2596da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 259793bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2598da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2599da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2600da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 26016b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2602da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2603da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2604da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2605534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2606534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 26079326c5c6Sstefano_zampini pc->ops->reset = PCReset_BDDC; 2608da1bb401SStefano Zampini 2609da1bb401SStefano Zampini /* composing function */ 2610a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr); 2611a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr); 2612906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2613674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 261430368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2615bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 26162b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2617b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 26182b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2619bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 262082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2621bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 262282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2623bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 262482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2625bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 262682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2627bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 262863602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2629bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2630bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2631bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2632bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2633a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2634da1bb401SStefano Zampini PetscFunctionReturn(0); 2635da1bb401SStefano Zampini } 26363425bc38SStefano Zampini 2637