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; 47bd2a564bSStefano Zampini PetscInt nt; 480c7d97c5SJed Brown PetscErrorCode ierr; 490c7d97c5SJed Brown 500c7d97c5SJed Brown PetscFunctionBegin; 51e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 528eeda7d8SStefano Zampini /* Verbose debugging */ 53a13144ffSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 54a13144ffSStefano Zampini /* Approximate solvers */ 55c7017625SStefano 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); 56c7017625SStefano 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); 57c7017625SStefano 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); 58c7017625SStefano 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); 596b78500eSPatrick Sanan /* Primal space customization */ 6008a5cf49SStefano 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); 61be12c134Sstefano_zampini ierr = PetscOptionsInt("-pc_bddc_graph_maxcount","Maximum number of shared subdomains for a connected component","none",pcbddc->graphmaxcount,&pcbddc->graphmaxcount,NULL);CHKERRQ(ierr); 621c7a958bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_corner_selection","Activates face-based corner selection","none",pcbddc->corner_selection,&pcbddc->corner_selection,NULL);CHKERRQ(ierr); 638eeda7d8SStefano 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); 648eeda7d8SStefano 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); 658eeda7d8SStefano 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); 6614f95afaSStefano 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); 676661aa1dSStefano 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); 6814f95afaSStefano 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); 698eeda7d8SStefano Zampini /* Change of basis */ 70b9b85e73SStefano 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); 71b9b85e73SStefano 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); 72674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 73674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 74674ae819SStefano Zampini } 758eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 768eeda7d8SStefano 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); 7757de7509SStefano 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); 780298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 792b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 80323d395dSStefano 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); 81674ae819SStefano 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); 82b96c3477SStefano 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); 83b96c3477SStefano 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); 84b96c3477SStefano 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); 85683d3df6SStefano 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); 86bf3a8328SStefano 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); 87839e9adbSstefano_zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_singlemat","Collapse deluxe operators","none",pcbddc->deluxe_singlemat,&pcbddc->deluxe_singlemat,NULL);CHKERRQ(ierr); 88bf3a8328SStefano 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); 89bd2a564bSStefano Zampini nt = 2; 90bd2a564bSStefano Zampini ierr = PetscOptionsRealArray("-pc_bddc_adaptive_threshold","Thresholds to be used for adaptive selection of constraints","none",pcbddc->adaptive_threshold,&nt,NULL);CHKERRQ(ierr); 91bd2a564bSStefano Zampini if (nt == 1) pcbddc->adaptive_threshold[1] = pcbddc->adaptive_threshold[0]; 9208122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 9308122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 943301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 95b0c7d250SStefano 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); 96b3afcdbeSStefano 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); 97e9627c49SStefano 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); 9827b6a85dSStefano 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); 99a198735bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->compute_nonetflux,&pcbddc->compute_nonetflux,NULL);CHKERRQ(ierr); 1004f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 10170c64980SStefano 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); 1020c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 1030c7d97c5SJed Brown PetscFunctionReturn(0); 1040c7d97c5SJed Brown } 1056b78500eSPatrick Sanan 1066b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 1076b78500eSPatrick Sanan { 1086b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 109e9627c49SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 1106b78500eSPatrick Sanan PetscErrorCode ierr; 11171783a16SStefano Zampini PetscBool isascii; 112e9627c49SStefano Zampini PetscSubcomm subcomm; 113e9627c49SStefano Zampini PetscViewer subviewer; 1146b78500eSPatrick Sanan 1156b78500eSPatrick Sanan PetscFunctionBegin; 1166b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 1176b78500eSPatrick Sanan /* ASCII viewer */ 1186b78500eSPatrick Sanan if (isascii) { 1194b2aedd3SStefano Zampini PetscMPIInt color,rank,size; 120fbad9177SStefano Zampini PetscInt64 loc[7],gsum[6],gmax[6],gmin[6],totbenign; 121e9627c49SStefano Zampini PetscScalar interface_size; 122e9627c49SStefano Zampini PetscReal ratio1=0.,ratio2=0.; 123e9627c49SStefano Zampini Vec counter; 1246b78500eSPatrick Sanan 125b74ba07aSstefano_zampini if (!pc->setupcalled) { 126b74ba07aSstefano_zampini ierr = PetscViewerASCIIPrintf(viewer," Partial information available: preconditioner has not been setup yet\n");CHKERRQ(ierr); 127b74ba07aSstefano_zampini } 128efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use verbose output: %D\n",pcbddc->dbg_flag);CHKERRQ(ierr); 1296f0c0a6aSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use user-defined CSR: %d\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 1306f0c0a6aSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use local mat graph: %d\n",pcbddc->use_local_adj && !pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 131e9627c49SStefano Zampini if (pcbddc->mat_graph->twodim) { 132efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Connectivity graph topological dimension: 2\n");CHKERRQ(ierr); 133e9627c49SStefano Zampini } else { 134efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Connectivity graph topological dimension: 3\n");CHKERRQ(ierr); 135e9627c49SStefano Zampini } 136aefa1729SStefano Zampini if (pcbddc->graphmaxcount != PETSC_MAX_INT) { 137efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Graph max count: %D\n",pcbddc->graphmaxcount);CHKERRQ(ierr); 138aefa1729SStefano Zampini } 13950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use vertices: %d (vertex size %D)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 140efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 141efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 142efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 143efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 144efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 145efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 146efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 147efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 148efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 149efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 150efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 15150e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Multilevel max levels: %D\n",pcbddc->max_levels);CHKERRQ(ierr); 15250e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Multilevel coarsening ratio: %D\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 153efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 154efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 155efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 156efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr); 157efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 15850e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Number of dofs' layers for the computation of principal minors: %D\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 159efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 160bd2a564bSStefano Zampini if (pcbddc->adaptive_threshold[1] != pcbddc->adaptive_threshold[0]) { 161bd2a564bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Adaptive constraint selection thresholds (active %d, userdefined %d): %g,%g\n",pcbddc->adaptive_selection,pcbddc->adaptive_userdefined,pcbddc->adaptive_threshold[0],pcbddc->adaptive_threshold[1]);CHKERRQ(ierr); 162bd2a564bSStefano Zampini } else { 163bd2a564bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_selection,pcbddc->adaptive_userdefined,pcbddc->adaptive_threshold[0]);CHKERRQ(ierr); 164bd2a564bSStefano Zampini } 16550e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Min constraints / connected component: %D\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 16650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Max constraints / connected component: %D\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 167efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 168efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 16950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Num. Procs. to map coarse adjacency list: %D\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 17050e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Coarse eqs per proc (significant at the coarsest level): %D\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 171efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr); 172efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 173efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr); 174efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr); 175b74ba07aSstefano_zampini if (!pc->setupcalled) PetscFunctionReturn(0); 1766b78500eSPatrick Sanan 177fbad9177SStefano Zampini /* compute interface size */ 178e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 179e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 180e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 181e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 182e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 183e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 184e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 185fbad9177SStefano Zampini 186fbad9177SStefano Zampini /* compute some statistics on the domain decomposition */ 187e9627c49SStefano Zampini gsum[0] = 1; 188fbad9177SStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0; 189e9627c49SStefano Zampini loc[0] = !!pcis->n; 190e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 191e9627c49SStefano Zampini loc[2] = pcis->n_B; 192e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 193345ecf6cSStefano Zampini loc[4] = pcis->n; 194fbad9177SStefano Zampini loc[5] = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0); 195fbad9177SStefano Zampini loc[6] = pcbddc->benign_n; 196fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 197fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1; 198fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 199fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT; 200fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 201fbad9177SStefano Zampini ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 202e9627c49SStefano Zampini if (pcbddc->coarse_size) { 203e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 204e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 205e9627c49SStefano Zampini } 206efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 20750e0721cSStefano 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); 20850e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Coarsening ratios: all/coarse %D interface/coarse %D\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 20950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Active processes : %D\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 21050e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Total subdomains : %D\n",(PetscInt)gsum[5]);CHKERRQ(ierr); 211345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 21250e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Benign subs : %D\n",(PetscInt)totbenign);CHKERRQ(ierr); 213345ecf6cSStefano Zampini } 21450e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Dofs type :\tMIN\tMAX\tMEAN\n");CHKERRQ(ierr); 21550e0721cSStefano 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); 21650e0721cSStefano 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); 21750e0721cSStefano 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); 21850e0721cSStefano 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); 21950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Local subs :\t%D\t%D\n" ,(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr); 22050e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," ********************************** COARSE PROBLEM DETAILS *********************************\n");CHKERRQ(ierr); 22127b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 222e9627c49SStefano Zampini 223fbad9177SStefano Zampini /* the coarse problem can be handled by a different communicator */ 224e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 225e9627c49SStefano Zampini else color = 0; 226e9627c49SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 2274b2aedd3SStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 228e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 2294b2aedd3SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr); 230e9627c49SStefano Zampini ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr); 231e9627c49SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 232e9627c49SStefano Zampini if (color == 1) { 233e9627c49SStefano Zampini ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr); 234e9627c49SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 235e9627c49SStefano Zampini } 236e9627c49SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 237e9627c49SStefano Zampini ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr); 238e9627c49SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 239e9627c49SStefano Zampini } 2406b78500eSPatrick Sanan PetscFunctionReturn(0); 2416b78500eSPatrick Sanan } 242a13144ffSStefano Zampini 2431e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 244a13144ffSStefano Zampini { 245a13144ffSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 246a13144ffSStefano Zampini PetscErrorCode ierr; 247a13144ffSStefano Zampini 248a13144ffSStefano Zampini PetscFunctionBegin; 249a13144ffSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 250a13144ffSStefano Zampini ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); 251a13144ffSStefano Zampini pcbddc->discretegradient = G; 252a13144ffSStefano Zampini pcbddc->nedorder = order > 0 ? order : -order; 253495a2a07SStefano Zampini pcbddc->nedfield = field; 2541e0482f5SStefano Zampini pcbddc->nedglobal = global; 2551e0482f5SStefano Zampini pcbddc->conforming = conforming; 256a13144ffSStefano Zampini PetscFunctionReturn(0); 257a13144ffSStefano Zampini } 258a13144ffSStefano Zampini 259a13144ffSStefano Zampini /*@ 260a13144ffSStefano Zampini PCBDDCSetDiscreteGradient - Sets the discrete gradient 261a13144ffSStefano Zampini 262a13144ffSStefano Zampini Collective on PC 263a13144ffSStefano Zampini 264a13144ffSStefano Zampini Input Parameters: 265a13144ffSStefano Zampini + pc - the preconditioning context 266a13144ffSStefano Zampini . G - the discrete gradient matrix (should be in AIJ format) 267a13144ffSStefano Zampini . order - the order of the Nedelec space (1 for the lowest order) 268495a2a07SStefano Zampini . field - the field id of the Nedelec dofs (not used if the fields have not been specified) 2691e0482f5SStefano Zampini . global - the type of global ordering for the rows of G 270a13144ffSStefano Zampini - conforming - whether the mesh is conforming or not 271a13144ffSStefano Zampini 272a13144ffSStefano Zampini Level: advanced 273a13144ffSStefano Zampini 274a13144ffSStefano Zampini Notes: The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry. 275495a2a07SStefano Zampini For variable order spaces, the order should be set to zero. 2761e0482f5SStefano Zampini If global is true, the rows of G should be given in global ordering for the whole dofs; 2771e0482f5SStefano Zampini if false, the ordering should be global for the Nedelec field. 2781e0482f5SStefano 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 2791e0482f5SStefano Zampini and geid the one for the Nedelec field. 280a13144ffSStefano Zampini 281495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal() 282a13144ffSStefano Zampini @*/ 2831e0482f5SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 284a13144ffSStefano Zampini { 285a13144ffSStefano Zampini PetscErrorCode ierr; 286a13144ffSStefano Zampini 287a13144ffSStefano Zampini PetscFunctionBegin; 288a13144ffSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 289a13144ffSStefano Zampini PetscValidHeaderSpecific(G,MAT_CLASSID,2); 290a13144ffSStefano Zampini PetscValidLogicalCollectiveInt(pc,order,3); 2911e0482f5SStefano Zampini PetscValidLogicalCollectiveInt(pc,field,4); 2921e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,global,5); 2931e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,conforming,6); 2941e0482f5SStefano Zampini PetscCheckSameComm(pc,1,G,2); 2951e0482f5SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool,PetscBool),(pc,G,order,field,global,conforming));CHKERRQ(ierr); 296a13144ffSStefano Zampini PetscFunctionReturn(0); 297a13144ffSStefano Zampini } 298a13144ffSStefano Zampini 2998ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 300a198735bSStefano Zampini { 301a198735bSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 302a198735bSStefano Zampini PetscErrorCode ierr; 3036b78500eSPatrick Sanan 304a198735bSStefano Zampini PetscFunctionBegin; 305a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr); 306a198735bSStefano Zampini ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); 307a198735bSStefano Zampini pcbddc->divudotp = divudotp; 3088ae0ca82SStefano Zampini pcbddc->divudotp_trans = trans; 309a198735bSStefano Zampini pcbddc->compute_nonetflux = PETSC_TRUE; 310a198735bSStefano Zampini if (vl2l) { 311a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr); 312fa23a32eSStefano Zampini ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); 313a198735bSStefano Zampini pcbddc->divudotp_vl2l = vl2l; 314a198735bSStefano Zampini } 315a198735bSStefano Zampini PetscFunctionReturn(0); 316a198735bSStefano Zampini } 3173d996552SStefano Zampini 318a198735bSStefano Zampini /*@ 319a198735bSStefano Zampini PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx 320a198735bSStefano Zampini 321a198735bSStefano Zampini Collective on PC 322a198735bSStefano Zampini 323a198735bSStefano Zampini Input Parameters: 324a198735bSStefano Zampini + pc - the preconditioning context 325a198735bSStefano Zampini . divudotp - the matrix (must be of type MATIS) 3268ae0ca82SStefano Zampini . trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space. 32705a3bf82SStefano 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 328a198735bSStefano Zampini 329a198735bSStefano Zampini Level: advanced 330a198735bSStefano Zampini 3318ae0ca82SStefano Zampini Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries 33205a3bf82SStefano Zampini If vl2l is NULL, the local ordering for velocities in divudotp should match that of the preconditioning matrix 333a198735bSStefano Zampini 334a198735bSStefano Zampini .seealso: PCBDDC 335a198735bSStefano Zampini @*/ 3368ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 337a198735bSStefano Zampini { 338a198735bSStefano Zampini PetscBool ismatis; 339a198735bSStefano Zampini PetscErrorCode ierr; 340a198735bSStefano Zampini 341a198735bSStefano Zampini PetscFunctionBegin; 342a198735bSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 343a198735bSStefano Zampini PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2); 344a198735bSStefano Zampini PetscCheckSameComm(pc,1,divudotp,2); 3458ae0ca82SStefano Zampini PetscValidLogicalCollectiveBool(pc,trans,3); 3468ae0ca82SStefano Zampini if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4); 347a198735bSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr); 348a198735bSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS"); 3498ae0ca82SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr); 350a198735bSStefano Zampini PetscFunctionReturn(0); 351a198735bSStefano Zampini } 3522d505d7fSStefano Zampini 3531dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 354b9b85e73SStefano Zampini { 355b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 356b9b85e73SStefano Zampini PetscErrorCode ierr; 357b9b85e73SStefano Zampini 358b9b85e73SStefano Zampini PetscFunctionBegin; 359b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 36056282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 361b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 3621dd7afcfSStefano Zampini pcbddc->change_interior = interior; 363b9b85e73SStefano Zampini PetscFunctionReturn(0); 364b9b85e73SStefano Zampini } 365b9b85e73SStefano Zampini /*@ 366906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 367b9b85e73SStefano Zampini 368b9b85e73SStefano Zampini Collective on PC 369b9b85e73SStefano Zampini 370b9b85e73SStefano Zampini Input Parameters: 371b9b85e73SStefano Zampini + pc - the preconditioning context 3721dd7afcfSStefano Zampini . change - the change of basis matrix 3731dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 374b9b85e73SStefano Zampini 375b9b85e73SStefano Zampini Level: intermediate 376b9b85e73SStefano Zampini 377b9b85e73SStefano Zampini Notes: 378b9b85e73SStefano Zampini 379b9b85e73SStefano Zampini .seealso: PCBDDC 380b9b85e73SStefano Zampini @*/ 3811dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 382b9b85e73SStefano Zampini { 383b9b85e73SStefano Zampini PetscErrorCode ierr; 384b9b85e73SStefano Zampini 385b9b85e73SStefano Zampini PetscFunctionBegin; 386b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 387b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 388906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 389906d46d4SStefano Zampini if (pc->mat) { 390906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 391906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 392906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 393e0fe2d75SToby 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); 394e0fe2d75SToby 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); 395906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 396906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 397e0fe2d75SToby 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); 398e0fe2d75SToby 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); 399906d46d4SStefano Zampini } 4001dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 401b9b85e73SStefano Zampini PetscFunctionReturn(0); 402b9b85e73SStefano Zampini } 4032d505d7fSStefano Zampini 40430368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 40530368db7SStefano Zampini { 40630368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 40756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 40830368db7SStefano Zampini PetscErrorCode ierr; 40930368db7SStefano Zampini 41030368db7SStefano Zampini PetscFunctionBegin; 41156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 41256282151SStefano Zampini if (pcbddc->user_primal_vertices) { 41356282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 41456282151SStefano Zampini } 41530368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 41630368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 41730368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 41856282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 41930368db7SStefano Zampini PetscFunctionReturn(0); 42030368db7SStefano Zampini } 421ab8c8b98SStefano Zampini 42230368db7SStefano Zampini /*@ 42330368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 42430368db7SStefano Zampini 42530368db7SStefano Zampini Collective 42630368db7SStefano Zampini 42730368db7SStefano Zampini Input Parameters: 42830368db7SStefano Zampini + pc - the preconditioning context 42930368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 43030368db7SStefano Zampini 43130368db7SStefano Zampini Level: intermediate 43230368db7SStefano Zampini 43330368db7SStefano Zampini Notes: 43430368db7SStefano Zampini Any process can list any global node 43530368db7SStefano Zampini 43630368db7SStefano Zampini .seealso: PCBDDC 43730368db7SStefano Zampini @*/ 43830368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 43930368db7SStefano Zampini { 44030368db7SStefano Zampini PetscErrorCode ierr; 44130368db7SStefano Zampini 44230368db7SStefano Zampini PetscFunctionBegin; 44330368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 44430368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 44530368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 44630368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 44730368db7SStefano Zampini PetscFunctionReturn(0); 44830368db7SStefano Zampini } 4492d505d7fSStefano Zampini 450674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 451674ae819SStefano Zampini { 452674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 45356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 454674ae819SStefano Zampini PetscErrorCode ierr; 4551e6b0712SBarry Smith 456674ae819SStefano Zampini PetscFunctionBegin; 45756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 45856282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 45956282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 46056282151SStefano Zampini } 461674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 46230368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 46330368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 46456282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 465674ae819SStefano Zampini PetscFunctionReturn(0); 466674ae819SStefano Zampini } 467674ae819SStefano Zampini /*@ 46828509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 469674ae819SStefano Zampini 47017eb1463SStefano Zampini Collective 471674ae819SStefano Zampini 472674ae819SStefano Zampini Input Parameters: 473674ae819SStefano Zampini + pc - the preconditioning context 47417eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 475674ae819SStefano Zampini 476674ae819SStefano Zampini Level: intermediate 477674ae819SStefano Zampini 478674ae819SStefano Zampini Notes: 479674ae819SStefano Zampini 480674ae819SStefano Zampini .seealso: PCBDDC 481674ae819SStefano Zampini @*/ 482674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 483674ae819SStefano Zampini { 484674ae819SStefano Zampini PetscErrorCode ierr; 485674ae819SStefano Zampini 486674ae819SStefano Zampini PetscFunctionBegin; 487674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 488674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 48917eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 490674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 491674ae819SStefano Zampini PetscFunctionReturn(0); 492674ae819SStefano Zampini } 4932d505d7fSStefano Zampini 4944fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 4954fad6a16SStefano Zampini { 4964fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4974fad6a16SStefano Zampini 4984fad6a16SStefano Zampini PetscFunctionBegin; 4994fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 5004fad6a16SStefano Zampini PetscFunctionReturn(0); 5014fad6a16SStefano Zampini } 5021e6b0712SBarry Smith 5034fad6a16SStefano Zampini /*@ 50428509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 5054fad6a16SStefano Zampini 5064fad6a16SStefano Zampini Logically collective on PC 5074fad6a16SStefano Zampini 5084fad6a16SStefano Zampini Input Parameters: 5094fad6a16SStefano Zampini + pc - the preconditioning context 51028509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 5114fad6a16SStefano Zampini 5120f202f7eSStefano Zampini Options Database Keys: 5130f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 5144fad6a16SStefano Zampini 5154fad6a16SStefano Zampini Level: intermediate 5164fad6a16SStefano Zampini 5174fad6a16SStefano Zampini Notes: 5180f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 5194fad6a16SStefano Zampini 5200f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 5214fad6a16SStefano Zampini @*/ 5224fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 5234fad6a16SStefano Zampini { 5244fad6a16SStefano Zampini PetscErrorCode ierr; 5254fad6a16SStefano Zampini 5264fad6a16SStefano Zampini PetscFunctionBegin; 5274fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5282b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 5294fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 5304fad6a16SStefano Zampini PetscFunctionReturn(0); 5314fad6a16SStefano Zampini } 5322b510759SStefano Zampini 533b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 534b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 535b8ffe317SStefano Zampini { 536b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 537b8ffe317SStefano Zampini 538b8ffe317SStefano Zampini PetscFunctionBegin; 53985c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 540b8ffe317SStefano Zampini PetscFunctionReturn(0); 541b8ffe317SStefano Zampini } 542b8ffe317SStefano Zampini 543b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 5442b510759SStefano Zampini { 5452b510759SStefano Zampini PetscErrorCode ierr; 5462b510759SStefano Zampini 5472b510759SStefano Zampini PetscFunctionBegin; 5482b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 549b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 550b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 5512b510759SStefano Zampini PetscFunctionReturn(0); 5522b510759SStefano Zampini } 5531e6b0712SBarry Smith 5542b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 5554fad6a16SStefano Zampini { 5564fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5574fad6a16SStefano Zampini 5584fad6a16SStefano Zampini PetscFunctionBegin; 5592b510759SStefano Zampini pcbddc->current_level = level; 5604fad6a16SStefano Zampini PetscFunctionReturn(0); 5614fad6a16SStefano Zampini } 5621e6b0712SBarry Smith 563b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 564b8ffe317SStefano Zampini { 565b8ffe317SStefano Zampini PetscErrorCode ierr; 566b8ffe317SStefano Zampini 567b8ffe317SStefano Zampini PetscFunctionBegin; 568b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 569b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 570b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 571b8ffe317SStefano Zampini PetscFunctionReturn(0); 572b8ffe317SStefano Zampini } 573b8ffe317SStefano Zampini 5742b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 5752b510759SStefano Zampini { 5762b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5772b510759SStefano Zampini 5782b510759SStefano Zampini PetscFunctionBegin; 5792b510759SStefano Zampini pcbddc->max_levels = levels; 5802b510759SStefano Zampini PetscFunctionReturn(0); 5812b510759SStefano Zampini } 5822b510759SStefano Zampini 5834fad6a16SStefano Zampini /*@ 58428509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 5854fad6a16SStefano Zampini 5864fad6a16SStefano Zampini Logically collective on PC 5874fad6a16SStefano Zampini 5884fad6a16SStefano Zampini Input Parameters: 5894fad6a16SStefano Zampini + pc - the preconditioning context 59028509bceSStefano Zampini - levels - the maximum number of levels (max 9) 5914fad6a16SStefano Zampini 5920f202f7eSStefano Zampini Options Database Keys: 5930f202f7eSStefano Zampini . -pc_bddc_levels 5944fad6a16SStefano Zampini 5954fad6a16SStefano Zampini Level: intermediate 5964fad6a16SStefano Zampini 5974fad6a16SStefano Zampini Notes: 5980f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 5994fad6a16SStefano Zampini 6000f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 6014fad6a16SStefano Zampini @*/ 6022b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 6034fad6a16SStefano Zampini { 6044fad6a16SStefano Zampini PetscErrorCode ierr; 6054fad6a16SStefano Zampini 6064fad6a16SStefano Zampini PetscFunctionBegin; 6074fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6082b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 609312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 6102b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 6114fad6a16SStefano Zampini PetscFunctionReturn(0); 6124fad6a16SStefano Zampini } 6131e6b0712SBarry Smith 6143b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 6153b03a366Sstefano_zampini { 6163b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 61756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6183b03a366Sstefano_zampini PetscErrorCode ierr; 6193b03a366Sstefano_zampini 6203b03a366Sstefano_zampini PetscFunctionBegin; 62156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 62256282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 62356282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 62456282151SStefano Zampini } 625785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 626785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6273b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 62836e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 62956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6303b03a366Sstefano_zampini PetscFunctionReturn(0); 6313b03a366Sstefano_zampini } 6321e6b0712SBarry Smith 6333b03a366Sstefano_zampini /*@ 63428509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 6353b03a366Sstefano_zampini 636785d1243SStefano Zampini Collective 6373b03a366Sstefano_zampini 6383b03a366Sstefano_zampini Input Parameters: 6393b03a366Sstefano_zampini + pc - the preconditioning context 640785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 6413b03a366Sstefano_zampini 6423b03a366Sstefano_zampini Level: intermediate 6433b03a366Sstefano_zampini 6440f202f7eSStefano Zampini Notes: 6450f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 6463b03a366Sstefano_zampini 6470f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 6483b03a366Sstefano_zampini @*/ 6493b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 6503b03a366Sstefano_zampini { 6513b03a366Sstefano_zampini PetscErrorCode ierr; 6523b03a366Sstefano_zampini 6533b03a366Sstefano_zampini PetscFunctionBegin; 6543b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 655674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 656785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 6573b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6583b03a366Sstefano_zampini PetscFunctionReturn(0); 6593b03a366Sstefano_zampini } 6601e6b0712SBarry Smith 66182ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 6623b03a366Sstefano_zampini { 6633b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 66456282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6653b03a366Sstefano_zampini PetscErrorCode ierr; 6663b03a366Sstefano_zampini 6673b03a366Sstefano_zampini PetscFunctionBegin; 66856282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 66956282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 67056282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 67156282151SStefano Zampini } 672785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 673785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6743b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 675785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 67656282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6773b03a366Sstefano_zampini PetscFunctionReturn(0); 6783b03a366Sstefano_zampini } 6793b03a366Sstefano_zampini 6803b03a366Sstefano_zampini /*@ 68182ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 6823b03a366Sstefano_zampini 683785d1243SStefano Zampini Collective 6843b03a366Sstefano_zampini 6853b03a366Sstefano_zampini Input Parameters: 6863b03a366Sstefano_zampini + pc - the preconditioning context 68782ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 6883b03a366Sstefano_zampini 6893b03a366Sstefano_zampini Level: intermediate 6903b03a366Sstefano_zampini 6913b03a366Sstefano_zampini Notes: 6923b03a366Sstefano_zampini 6930f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 6943b03a366Sstefano_zampini @*/ 69582ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 6963b03a366Sstefano_zampini { 6973b03a366Sstefano_zampini PetscErrorCode ierr; 6983b03a366Sstefano_zampini 6993b03a366Sstefano_zampini PetscFunctionBegin; 7003b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7013b03a366Sstefano_zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 70282ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 70382ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 7043b03a366Sstefano_zampini PetscFunctionReturn(0); 7053b03a366Sstefano_zampini } 7063b03a366Sstefano_zampini 70753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 7080c7d97c5SJed Brown { 7090c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 71056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 71153cdbc3dSStefano Zampini PetscErrorCode ierr; 7120c7d97c5SJed Brown 7130c7d97c5SJed Brown PetscFunctionBegin; 71456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 71556282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 71656282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 71756282151SStefano Zampini } 718785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 719785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 72053cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 72136e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 72256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7230c7d97c5SJed Brown PetscFunctionReturn(0); 7240c7d97c5SJed Brown } 7251e6b0712SBarry Smith 72657527edcSJed Brown /*@ 72728509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 72857527edcSJed Brown 729785d1243SStefano Zampini Collective 73057527edcSJed Brown 73157527edcSJed Brown Input Parameters: 73257527edcSJed Brown + pc - the preconditioning context 733785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 73457527edcSJed Brown 73557527edcSJed Brown Level: intermediate 73657527edcSJed Brown 7370f202f7eSStefano Zampini Notes: 7380f202f7eSStefano Zampini Any process can list any global node 73957527edcSJed Brown 7400f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 74157527edcSJed Brown @*/ 74253cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 7430c7d97c5SJed Brown { 7440c7d97c5SJed Brown PetscErrorCode ierr; 7450c7d97c5SJed Brown 7460c7d97c5SJed Brown PetscFunctionBegin; 7470c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 748674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 749785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 75053cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 75153cdbc3dSStefano Zampini PetscFunctionReturn(0); 75253cdbc3dSStefano Zampini } 7531e6b0712SBarry Smith 75482ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 7550c7d97c5SJed Brown { 7560c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 75756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7580c7d97c5SJed Brown PetscErrorCode ierr; 7590c7d97c5SJed Brown 7600c7d97c5SJed Brown PetscFunctionBegin; 76156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 76256282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 76356282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 76456282151SStefano Zampini } 765785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 766785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 7670c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 768785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 76956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7700c7d97c5SJed Brown PetscFunctionReturn(0); 7710c7d97c5SJed Brown } 7720c7d97c5SJed Brown 7730c7d97c5SJed Brown /*@ 77482ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 7750c7d97c5SJed Brown 776785d1243SStefano Zampini Collective 7770c7d97c5SJed Brown 7780c7d97c5SJed Brown Input Parameters: 7790c7d97c5SJed Brown + pc - the preconditioning context 78082ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 7810c7d97c5SJed Brown 7820c7d97c5SJed Brown Level: intermediate 7830c7d97c5SJed Brown 7840c7d97c5SJed Brown Notes: 7850c7d97c5SJed Brown 7860f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 7870c7d97c5SJed Brown @*/ 78882ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 7890c7d97c5SJed Brown { 7900c7d97c5SJed Brown PetscErrorCode ierr; 7910c7d97c5SJed Brown 7920c7d97c5SJed Brown PetscFunctionBegin; 7930c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7940c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 79582ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 79682ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 79753cdbc3dSStefano Zampini PetscFunctionReturn(0); 79853cdbc3dSStefano Zampini } 79953cdbc3dSStefano Zampini 800da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 801da1bb401SStefano Zampini { 802da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 803da1bb401SStefano Zampini 804da1bb401SStefano Zampini PetscFunctionBegin; 805da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 806da1bb401SStefano Zampini PetscFunctionReturn(0); 807da1bb401SStefano Zampini } 8081e6b0712SBarry Smith 809da1bb401SStefano Zampini /*@ 810785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 811da1bb401SStefano Zampini 812785d1243SStefano Zampini Collective 813785d1243SStefano Zampini 814785d1243SStefano Zampini Input Parameters: 815785d1243SStefano Zampini . pc - the preconditioning context 816785d1243SStefano Zampini 817785d1243SStefano Zampini Output Parameters: 818785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 819785d1243SStefano Zampini 820785d1243SStefano Zampini Level: intermediate 821785d1243SStefano Zampini 8220f202f7eSStefano Zampini Notes: 8230f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 824785d1243SStefano Zampini 825785d1243SStefano Zampini .seealso: PCBDDC 826785d1243SStefano Zampini @*/ 827785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 828785d1243SStefano Zampini { 829785d1243SStefano Zampini PetscErrorCode ierr; 830785d1243SStefano Zampini 831785d1243SStefano Zampini PetscFunctionBegin; 832785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 833785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 834785d1243SStefano Zampini PetscFunctionReturn(0); 835785d1243SStefano Zampini } 836785d1243SStefano Zampini 837785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 838785d1243SStefano Zampini { 839785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 840785d1243SStefano Zampini 841785d1243SStefano Zampini PetscFunctionBegin; 842785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 843785d1243SStefano Zampini PetscFunctionReturn(0); 844785d1243SStefano Zampini } 845785d1243SStefano Zampini 846da1bb401SStefano Zampini /*@ 84782ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 848da1bb401SStefano Zampini 849785d1243SStefano Zampini Collective 850da1bb401SStefano Zampini 851da1bb401SStefano Zampini Input Parameters: 85228509bceSStefano Zampini . pc - the preconditioning context 853da1bb401SStefano Zampini 854da1bb401SStefano Zampini Output Parameters: 85528509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 856da1bb401SStefano Zampini 857da1bb401SStefano Zampini Level: intermediate 858da1bb401SStefano Zampini 859da1bb401SStefano Zampini Notes: 8600f202f7eSStefano 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). 8610f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 862da1bb401SStefano Zampini 863da1bb401SStefano Zampini .seealso: PCBDDC 864da1bb401SStefano Zampini @*/ 86582ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 866da1bb401SStefano Zampini { 867da1bb401SStefano Zampini PetscErrorCode ierr; 868da1bb401SStefano Zampini 869da1bb401SStefano Zampini PetscFunctionBegin; 870da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 87182ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 872da1bb401SStefano Zampini PetscFunctionReturn(0); 873da1bb401SStefano Zampini } 8741e6b0712SBarry Smith 87553cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 87653cdbc3dSStefano Zampini { 87753cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 87853cdbc3dSStefano Zampini 87953cdbc3dSStefano Zampini PetscFunctionBegin; 88053cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 88153cdbc3dSStefano Zampini PetscFunctionReturn(0); 88253cdbc3dSStefano Zampini } 8831e6b0712SBarry Smith 88453cdbc3dSStefano Zampini /*@ 885785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 88653cdbc3dSStefano Zampini 887785d1243SStefano Zampini Collective 888785d1243SStefano Zampini 889785d1243SStefano Zampini Input Parameters: 890785d1243SStefano Zampini . pc - the preconditioning context 891785d1243SStefano Zampini 892785d1243SStefano Zampini Output Parameters: 893785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 894785d1243SStefano Zampini 895785d1243SStefano Zampini Level: intermediate 896785d1243SStefano Zampini 8970f202f7eSStefano Zampini Notes: 8980f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 899785d1243SStefano Zampini 900785d1243SStefano Zampini .seealso: PCBDDC 901785d1243SStefano Zampini @*/ 902785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 903785d1243SStefano Zampini { 904785d1243SStefano Zampini PetscErrorCode ierr; 905785d1243SStefano Zampini 906785d1243SStefano Zampini PetscFunctionBegin; 907785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 908785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 909785d1243SStefano Zampini PetscFunctionReturn(0); 910785d1243SStefano Zampini } 911785d1243SStefano Zampini 912785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 913785d1243SStefano Zampini { 914785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 915785d1243SStefano Zampini 916785d1243SStefano Zampini PetscFunctionBegin; 917785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 918785d1243SStefano Zampini PetscFunctionReturn(0); 919785d1243SStefano Zampini } 920785d1243SStefano Zampini 92153cdbc3dSStefano Zampini /*@ 92282ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 92353cdbc3dSStefano Zampini 924785d1243SStefano Zampini Collective 92553cdbc3dSStefano Zampini 92653cdbc3dSStefano Zampini Input Parameters: 92728509bceSStefano Zampini . pc - the preconditioning context 92853cdbc3dSStefano Zampini 92953cdbc3dSStefano Zampini Output Parameters: 93028509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 93153cdbc3dSStefano Zampini 93253cdbc3dSStefano Zampini Level: intermediate 93353cdbc3dSStefano Zampini 93453cdbc3dSStefano Zampini Notes: 9350f202f7eSStefano 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). 9360f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 93753cdbc3dSStefano Zampini 93853cdbc3dSStefano Zampini .seealso: PCBDDC 93953cdbc3dSStefano Zampini @*/ 94082ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 94153cdbc3dSStefano Zampini { 94253cdbc3dSStefano Zampini PetscErrorCode ierr; 94353cdbc3dSStefano Zampini 94453cdbc3dSStefano Zampini PetscFunctionBegin; 94553cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 94682ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 9470c7d97c5SJed Brown PetscFunctionReturn(0); 9480c7d97c5SJed Brown } 9491e6b0712SBarry Smith 9501a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 95136e030ebSStefano Zampini { 95236e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 953da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 95456282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 955da1bb401SStefano Zampini PetscErrorCode ierr; 95636e030ebSStefano Zampini 95736e030ebSStefano Zampini PetscFunctionBegin; 9588687889aSStefano Zampini if (!nvtxs) { 95904194a47SStefano Zampini if (copymode == PETSC_OWN_POINTER) { 96004194a47SStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 96104194a47SStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 96204194a47SStefano Zampini } 9638687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 9648687889aSStefano Zampini PetscFunctionReturn(0); 9658687889aSStefano Zampini } 96666da6bd7Sstefano_zampini if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */ 96756282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 96856282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 9692d505d7fSStefano Zampini ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9702d505d7fSStefano Zampini if (same_data) { 9712d505d7fSStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9722d505d7fSStefano Zampini } 97356282151SStefano Zampini } 97456282151SStefano Zampini } 97556282151SStefano Zampini if (!same_data) { 976674ae819SStefano Zampini /* free old CSR */ 977674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 978674ae819SStefano Zampini /* get CSR into graph structure */ 979da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 980854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 981785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 982674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 983674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 984a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 985da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 9861a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 9871a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 988a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 989a1dbd327SStefano Zampini } else if (copymode == PETSC_USE_POINTER) { 990a1dbd327SStefano Zampini mat_graph->xadj = (PetscInt*)xadj; 991a1dbd327SStefano Zampini mat_graph->adjncy = (PetscInt*)adjncy; 992a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_FALSE; 993e0fe2d75SToby Isaac } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %D",copymode); 994575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 99556282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 99656282151SStefano Zampini } 99736e030ebSStefano Zampini PetscFunctionReturn(0); 99836e030ebSStefano Zampini } 9991e6b0712SBarry Smith 100036e030ebSStefano Zampini /*@ 100154fffbccSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom. 100236e030ebSStefano Zampini 100336e030ebSStefano Zampini Not collective 100436e030ebSStefano Zampini 100536e030ebSStefano Zampini Input Parameters: 100654fffbccSStefano Zampini + pc - the preconditioning context. 100754fffbccSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the number of local dofs). 100854fffbccSStefano Zampini . xadj, adjncy - the connectivity of the dofs in CSR format. 100954fffbccSStefano Zampini - copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER. 101036e030ebSStefano Zampini 101136e030ebSStefano Zampini Level: intermediate 101236e030ebSStefano Zampini 101354fffbccSStefano Zampini Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative. 101436e030ebSStefano Zampini 101528509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 101636e030ebSStefano Zampini @*/ 10171a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 101836e030ebSStefano Zampini { 1019575ad6abSStefano Zampini void (*f)(void) = 0; 102036e030ebSStefano Zampini PetscErrorCode ierr; 102136e030ebSStefano Zampini 102236e030ebSStefano Zampini PetscFunctionBegin; 102336e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 10248687889aSStefano Zampini if (nvtxs) { 1025674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 10261633d1f0SStefano Zampini if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4); 10278687889aSStefano Zampini } 10281a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 1029575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 1030575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 1031575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 1032575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 1033575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 1034da1bb401SStefano Zampini } 103536e030ebSStefano Zampini PetscFunctionReturn(0); 103636e030ebSStefano Zampini } 10371e6b0712SBarry Smith 103863602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 103963602bcaSStefano Zampini { 104063602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 104163602bcaSStefano Zampini PetscInt i; 104256282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 104363602bcaSStefano Zampini PetscErrorCode ierr; 104463602bcaSStefano Zampini 104563602bcaSStefano Zampini PetscFunctionBegin; 104656282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 104756282151SStefano Zampini for (i=0;i<n_is;i++) { 104856282151SStefano Zampini PetscBool isequalt; 104956282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 105056282151SStefano Zampini if (!isequalt) break; 105156282151SStefano Zampini } 105256282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 105356282151SStefano Zampini } 105456282151SStefano Zampini for (i=0;i<n_is;i++) { 105556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 105656282151SStefano Zampini } 105763602bcaSStefano Zampini /* Destroy ISes if they were already set */ 105863602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 105963602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 106063602bcaSStefano Zampini } 106163602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 106263602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 106363602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 106463602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 106563602bcaSStefano Zampini } 106663602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 106763602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 106863602bcaSStefano Zampini /* allocate space then set */ 1069d02579f5SStefano Zampini if (n_is) { 1070d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 1071d02579f5SStefano Zampini } 107263602bcaSStefano Zampini for (i=0;i<n_is;i++) { 107363602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 107463602bcaSStefano Zampini } 107563602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 107663602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 107756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 107863602bcaSStefano Zampini PetscFunctionReturn(0); 107963602bcaSStefano Zampini } 108063602bcaSStefano Zampini 108163602bcaSStefano Zampini /*@ 108263602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 108363602bcaSStefano Zampini 108463602bcaSStefano Zampini Collective 108563602bcaSStefano Zampini 108663602bcaSStefano Zampini Input Parameters: 108763602bcaSStefano Zampini + pc - the preconditioning context 10880f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10890f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 109063602bcaSStefano Zampini 109163602bcaSStefano Zampini Level: intermediate 109263602bcaSStefano Zampini 10930f202f7eSStefano Zampini Notes: 10940f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 109563602bcaSStefano Zampini 109663602bcaSStefano Zampini .seealso: PCBDDC 109763602bcaSStefano Zampini @*/ 109863602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 109963602bcaSStefano Zampini { 110063602bcaSStefano Zampini PetscInt i; 110163602bcaSStefano Zampini PetscErrorCode ierr; 110263602bcaSStefano Zampini 110363602bcaSStefano Zampini PetscFunctionBegin; 110463602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 110563602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 110663602bcaSStefano Zampini for (i=0;i<n_is;i++) { 110763602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 110863602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 110963602bcaSStefano Zampini } 1110e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 111163602bcaSStefano Zampini PetscFunctionReturn(0); 111263602bcaSStefano Zampini } 111363602bcaSStefano Zampini 11149c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 11159c0446d6SStefano Zampini { 11169c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 11179c0446d6SStefano Zampini PetscInt i; 111856282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 11199c0446d6SStefano Zampini PetscErrorCode ierr; 11209c0446d6SStefano Zampini 11219c0446d6SStefano Zampini PetscFunctionBegin; 112256282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 112356282151SStefano Zampini for (i=0;i<n_is;i++) { 112456282151SStefano Zampini PetscBool isequalt; 112556282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 112656282151SStefano Zampini if (!isequalt) break; 112756282151SStefano Zampini } 112856282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 112956282151SStefano Zampini } 113056282151SStefano Zampini for (i=0;i<n_is;i++) { 113156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 113256282151SStefano Zampini } 1133da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 11349c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 11359c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 11369c0446d6SStefano Zampini } 1137d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 113863602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 113963602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 114063602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 114163602bcaSStefano Zampini } 114263602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 114363602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1144da1bb401SStefano Zampini /* allocate space then set */ 1145d02579f5SStefano Zampini if (n_is) { 1146785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1147d02579f5SStefano Zampini } 11489c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1149da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 11509c0446d6SStefano Zampini } 11519c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 115263602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 115356282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 11549c0446d6SStefano Zampini PetscFunctionReturn(0); 11559c0446d6SStefano Zampini } 11561e6b0712SBarry Smith 11579c0446d6SStefano Zampini /*@ 115863602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 11599c0446d6SStefano Zampini 116063602bcaSStefano Zampini Collective 11619c0446d6SStefano Zampini 11629c0446d6SStefano Zampini Input Parameters: 11639c0446d6SStefano Zampini + pc - the preconditioning context 11640f202f7eSStefano Zampini . n_is - number of index sets defining the fields 11650f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 11669c0446d6SStefano Zampini 11679c0446d6SStefano Zampini Level: intermediate 11689c0446d6SStefano Zampini 11690f202f7eSStefano Zampini Notes: 11700f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 11719c0446d6SStefano Zampini 11729c0446d6SStefano Zampini .seealso: PCBDDC 11739c0446d6SStefano Zampini @*/ 11749c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 11759c0446d6SStefano Zampini { 11762b510759SStefano Zampini PetscInt i; 11779c0446d6SStefano Zampini PetscErrorCode ierr; 11789c0446d6SStefano Zampini 11799c0446d6SStefano Zampini PetscFunctionBegin; 11809c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 118163602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 11822b510759SStefano Zampini for (i=0;i<n_is;i++) { 118363602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 1184a011d5a7Sstefano_zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 11852b510759SStefano Zampini } 11869c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 11879c0446d6SStefano Zampini PetscFunctionReturn(0); 11889c0446d6SStefano Zampini } 1189906d46d4SStefano Zampini 1190534831adSStefano Zampini /* 1191534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1192534831adSStefano Zampini guess if a transformation of basis approach has been selected. 11939c0446d6SStefano Zampini 1194534831adSStefano Zampini Input Parameter: 1195534831adSStefano Zampini + pc - the preconditioner contex 1196534831adSStefano Zampini 1197534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1198534831adSStefano Zampini 1199534831adSStefano Zampini Notes: 1200534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1201534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1202534831adSStefano Zampini */ 1203534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1204534831adSStefano Zampini { 1205534831adSStefano Zampini PetscErrorCode ierr; 1206534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1207534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 12083972b0daSStefano Zampini Vec used_vec; 120927b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1210534831adSStefano Zampini 1211534831adSStefano Zampini PetscFunctionBegin; 12121f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 121385c4d303SStefano Zampini if (ksp) { 12141f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 121585c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 121627b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 121727b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1218f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 12191f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 122085c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 122185c4d303SStefano Zampini } 122285c4d303SStefano Zampini } 1223fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1224fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1225fc17d649SStefano Zampini } 12261f4df5f7SStefano Zampini 122785c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 122862a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 122962a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 123062a6ff1dSStefano Zampini } 123162a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 123262a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 123362a6ff1dSStefano Zampini } 12348d00608fSStefano Zampini 123527b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 12363972b0daSStefano Zampini if (x) { 12373972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 12383972b0daSStefano Zampini used_vec = x; 12398d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 12403972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 12413972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 12423972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 124327b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1244266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1245266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1246266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 12473972b0daSStefano Zampini } 12488efcfb23SStefano Zampini 12498efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 12503972b0daSStefano Zampini if (ksp) { 1251a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 12528efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12538efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 12543972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 12553972b0daSStefano Zampini } 12563972b0daSStefano Zampini } 12573308cffdSStefano Zampini 12588d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12593972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 126070c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 12613975b054SStefano Zampini IS dirIS = NULL; 12623975b054SStefano Zampini 1263a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 12643975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 12653975b054SStefano Zampini if (dirIS) { 1266906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1267785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 12682b095fd8SStefano Zampini PetscScalar *array_x; 12692b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1270785d1243SStefano Zampini 12713972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 12723972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1273e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1274e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1275e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1276e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 127782ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 12783972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 12792b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12803972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12812fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 12823972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12832b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12843972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1285e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1286e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12878d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12881b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 12898efcfb23SStefano Zampini } 1290a07ea27aSStefano Zampini } 1291b76ba322SStefano Zampini 12928efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 12938d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 129427b6a85dSStefano Zampini /* save the original rhs */ 129527b6a85dSStefano Zampini if (save_rhs) { 129627b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 129727b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 12988d00608fSStefano Zampini } 12998d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13003972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 130127b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 13023972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 13038efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 130427b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 13057acc28cbSStefano Zampini if (ksp) { 13067acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 13077acc28cbSStefano Zampini } 13083308cffdSStefano Zampini } 13098efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1310b76ba322SStefano Zampini 1311fc17d649SStefano Zampini /* compute initial vector in benign space if needed 131227b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 131327b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 1314c69e9cc1SStefano Zampini if (rhs && pcbddc->benign_compute_correction && (pcbddc->benign_have_null || pcbddc->benign_apply_coarse_only)) { 13151f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 13161f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 13170369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 13180369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 13190369aaf7SStefano Zampini } 1320c69e9cc1SStefano Zampini /* keep applying coarse solver unless we no longer have benign subdomains */ 1321c69e9cc1SStefano Zampini pcbddc->benign_apply_coarse_only = pcbddc->benign_have_null ? PETSC_TRUE : PETSC_FALSE; 132227b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 13231dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 13243bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 13251f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 13261f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13271f4df5f7SStefano Zampini } 13281f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 132927b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 133027b6a85dSStefano Zampini if (save_rhs) { 133127b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 133292e3dcfbSStefano Zampini } 133327b6a85dSStefano Zampini if (pcbddc->rhs_change) { 13340369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 133527b6a85dSStefano Zampini } else { 133627b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 133727b6a85dSStefano Zampini } 13380369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 133927b6a85dSStefano Zampini } 134027b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 13410369aaf7SStefano Zampini } 13422d4c4fecSStefano Zampini 13432d4c4fecSStefano Zampini /* dbg output */ 1344a198735bSStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 13451f4df5f7SStefano Zampini Vec v; 1346c69e9cc1SStefano Zampini 13471f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 1348c69e9cc1SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 13491f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 1350c69e9cc1SStefano Zampini } else { 1351c69e9cc1SStefano Zampini ierr = VecCopy(rhs,v);CHKERRQ(ierr); 1352c69e9cc1SStefano Zampini } 13531f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1354e0fe2d75SToby Isaac ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %D: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr); 1355c69e9cc1SStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,pcbddc->dbg_viewer);CHKERRQ(ierr); 1356c69e9cc1SStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 13571f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 13581f4df5f7SStefano Zampini } 13590369aaf7SStefano Zampini 13600369aaf7SStefano Zampini /* set initial guess if using PCG */ 13618ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 13620369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 13630369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 13641dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 136527b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 13661dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13671dd7afcfSStefano Zampini } else { 13681dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 13691dd7afcfSStefano Zampini } 13701dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13711dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13721dd7afcfSStefano Zampini } else { 13730369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13740369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13751dd7afcfSStefano Zampini } 13760369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13771dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 13781dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 13791dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13801dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13811dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 13821dd7afcfSStefano Zampini } else { 13830369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13840369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13851dd7afcfSStefano Zampini } 13860369aaf7SStefano Zampini if (ksp) { 13870369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 13880369aaf7SStefano Zampini } 13898ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_TRUE; 1390266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1391266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13920369aaf7SStefano Zampini } 1393534831adSStefano Zampini PetscFunctionReturn(0); 1394534831adSStefano Zampini } 1395906d46d4SStefano Zampini 1396534831adSStefano Zampini /* 1397534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1398534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1399534831adSStefano Zampini 1400534831adSStefano Zampini Input Parameter: 1401534831adSStefano Zampini + pc - the preconditioner contex 1402534831adSStefano Zampini 1403534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1404534831adSStefano Zampini 1405534831adSStefano Zampini Notes: 1406534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1407534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1408534831adSStefano Zampini */ 1409534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1410534831adSStefano Zampini { 1411534831adSStefano Zampini PetscErrorCode ierr; 1412534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1413534831adSStefano Zampini 1414534831adSStefano Zampini PetscFunctionBegin; 14153972b0daSStefano Zampini /* add solution removed in presolve */ 14166bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 141727b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 14183425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1419af140850Sstefano_zampini } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) { 142027b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14213425bc38SStefano Zampini } 1422af140850Sstefano_zampini /* restore to original state (not for FETI-DP) */ 1423af140850Sstefano_zampini if (ksp) pcbddc->temp_solution_used = PETSC_FALSE; 142427b6a85dSStefano Zampini } 142527b6a85dSStefano Zampini 1426266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 14278d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 142827b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 14298d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 1430af140850Sstefano_zampini } 14318efcfb23SStefano Zampini /* restore ksp guess state */ 14328efcfb23SStefano Zampini if (ksp) { 14338efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 14348ae0ca82SStefano Zampini /* reset flag for exact dirichlet trick */ 14358ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 1436af140850Sstefano_zampini } 1437534831adSStefano Zampini PetscFunctionReturn(0); 1438534831adSStefano Zampini } 1439af140850Sstefano_zampini 14400c7d97c5SJed Brown /* 14410c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 14420c7d97c5SJed Brown by setting data structures and options. 14430c7d97c5SJed Brown 14440c7d97c5SJed Brown Input Parameter: 144553cdbc3dSStefano Zampini + pc - the preconditioner context 14460c7d97c5SJed Brown 14470c7d97c5SJed Brown Application Interface Routine: PCSetUp() 14480c7d97c5SJed Brown 14490c7d97c5SJed Brown Notes: 14500c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 14510c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 14520c7d97c5SJed Brown */ 145353cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 14540c7d97c5SJed Brown { 14550c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1456c703fcc7SStefano Zampini PCBDDCSubSchurs sub_schurs; 14575e8657edSStefano Zampini Mat_IS* matis; 145808122e43SStefano Zampini MatNullSpace nearnullspace; 145935509ce9Sstefano_zampini Mat lA; 146035509ce9Sstefano_zampini IS lP,zerodiag = NULL; 146191e8d312SStefano Zampini PetscInt nrows,ncols; 1462*86bfa4cfSStefano Zampini PetscMPIInt size; 1463c703fcc7SStefano Zampini PetscBool computesubschurs; 14648de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 14655e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 1466c703fcc7SStefano Zampini PetscErrorCode ierr; 14670c7d97c5SJed Brown 14680c7d97c5SJed Brown PetscFunctionBegin; 14695e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 14706c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 147191e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 14726c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 1473*86bfa4cfSStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 1474*86bfa4cfSStefano Zampini 14755e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1476f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 14773b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 147871582508SStefano Zampini Also, BDDC builds its own KSP for the Dirichlet problem */ 1479a485753aSstefano_zampini if (!pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE; 1480c83e1ba7SStefano Zampini if (pcbddc->recompute_topography) { 1481c83e1ba7SStefano Zampini pcbddc->graphanalyzed = PETSC_FALSE; 1482c83e1ba7SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1483c83e1ba7SStefano Zampini } else { 14848de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1485c83e1ba7SStefano Zampini } 1486b087196eSStefano Zampini 1487b087196eSStefano Zampini /* check parameters' compatibility */ 1488b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 1489bd2a564bSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold[0] != 0.0 || pcbddc->adaptive_threshold[1] != 0.0); 1490*86bfa4cfSStefano Zampini pcbddc->use_deluxe_scaling = (PetscBool)(pcbddc->use_deluxe_scaling && size > 1); 1491*86bfa4cfSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_selection && size > 1); 1492bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1493862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1494862806e4SStefano Zampini 14955a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 149616909a7fSStefano Zampini if (pcbddc->switch_static) { 149716909a7fSStefano Zampini PetscBool ismatis; 149816909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 149916909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 150016909a7fSStefano Zampini } 150116909a7fSStefano Zampini 150271582508SStefano Zampini /* activate all connected components if the netflux has been requested */ 1503bb05f991SStefano Zampini if (pcbddc->compute_nonetflux) { 1504bb05f991SStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 1505bb05f991SStefano Zampini pcbddc->use_edges = PETSC_TRUE; 1506bb05f991SStefano Zampini pcbddc->use_faces = PETSC_TRUE; 1507bb05f991SStefano Zampini } 1508bb05f991SStefano Zampini 1509f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 151070cf5478SStefano Zampini if (pcbddc->dbg_flag) { 151170cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 151258a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 15131575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1514f4ddd8eeSStefano Zampini } 151558a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1516f4ddd8eeSStefano Zampini } 1517f4ddd8eeSStefano Zampini 1518c703fcc7SStefano Zampini /* process topology information */ 151971582508SStefano Zampini if (pcbddc->recompute_topography) { 152071582508SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 1521c703fcc7SStefano Zampini if (pcbddc->discretegradient) { 1522a13144ffSStefano Zampini ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr); 1523a13144ffSStefano Zampini } 1524c703fcc7SStefano Zampini } 1525a13144ffSStefano Zampini 1526c703fcc7SStefano Zampini /* change basis if requested by the user */ 15275e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 15285e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 15295e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 15305e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 15315e8657edSStefano Zampini } else { 1532b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15335e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15345e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1535d16cbb6bSStefano Zampini } 1536d16cbb6bSStefano Zampini 15374f1b2e48SStefano Zampini /* 1538c703fcc7SStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick 15394f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 15404f1b2e48SStefano Zampini */ 15411dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1542d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 15439f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 15449f47a83aSStefano Zampini 154505b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1546669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 1547339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1548a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1549c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 15501dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 15511dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 15521dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 15531dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 15541dd7afcfSStefano Zampini } else { 1555a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 15561dd7afcfSStefano Zampini } 1557a3df083aSStefano Zampini } else { 15589f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1559674ae819SStefano Zampini } 1560a3df083aSStefano Zampini } 156127b6a85dSStefano Zampini 15628037d520SStefano Zampini /* propagate relevant information */ 156306a4e24aSStefano Zampini if (matis->A->symmetric_set) { 156406a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 156506a4e24aSStefano Zampini } 156606a4e24aSStefano Zampini if (matis->A->spd_set) { 156706a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 156806a4e24aSStefano Zampini } 1569e496cd5dSStefano Zampini 15705e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 15715e8657edSStefano Zampini { 15725e8657edSStefano Zampini Mat temp_mat; 15735e8657edSStefano Zampini 15745e8657edSStefano Zampini temp_mat = matis->A; 15755e8657edSStefano Zampini matis->A = pcbddc->local_mat; 15765e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 15775e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15785e8657edSStefano Zampini matis->A = temp_mat; 15795e8657edSStefano Zampini } 1580684f6988SStefano Zampini 158181d14e9dSStefano Zampini /* Analyze interface */ 158264ac59b8SStefano Zampini if (!pcbddc->graphanalyzed) { 1583674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15848de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1585345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 15864247aa23Sstefano_zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1587345ecf6cSStefano Zampini } 1588a198735bSStefano Zampini if (pcbddc->compute_nonetflux) { 1589669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1590669cc0f4SStefano Zampini 159121ef3d20SStefano Zampini if (!pcbddc->divudotp) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Missing divudotp operator"); 15928ae0ca82SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 159371582508SStefano Zampini /* TODO what if a nearnullspace is already attached? */ 15948037d520SStefano Zampini if (nnfnnsp) { 1595669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1596669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1597669cc0f4SStefano Zampini } 1598674ae819SStefano Zampini } 15998037d520SStefano Zampini } 1600fb8d54d4SStefano Zampini 16015408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 16025408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 16035408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 1604ff1f7e73Sstefano_zampini if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) { 16055408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 160609f581a4SStefano Zampini } 16074f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 160806f24817SStefano Zampini 1609b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1610c703fcc7SStefano Zampini if (computesubschurs && pcbddc->recompute_topography) { 161108122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1612b1b3d7a2SStefano Zampini } 16139d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 16149d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 16159d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16169d54b7f4SStefano Zampini } 1617c703fcc7SStefano Zampini 1618c703fcc7SStefano Zampini /* finish setup solvers and do adaptive selection of constraints */ 1619b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1620b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 16212070dbb6SStefano Zampini if (computesubschurs) { 162208122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 16232070dbb6SStefano Zampini } 1624d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1625d5574798SStefano Zampini } else { 1626d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 16272070dbb6SStefano Zampini if (computesubschurs) { 1628d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1629d5574798SStefano Zampini } 16302070dbb6SStefano Zampini } 163108122e43SStefano Zampini if (pcbddc->adaptive_selection) { 163208122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 16338de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1634b7eb3628SStefano Zampini } 1635684f6988SStefano Zampini 1636f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1637fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1638f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1639f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1640f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1641f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1642f4ddd8eeSStefano Zampini } else { 1643f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1644f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1645f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1646165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1647f4ddd8eeSStefano Zampini PetscInt i; 1648165b64e2SStefano Zampini const Vec *nearnullvecs; 1649165b64e2SStefano Zampini PetscObjectState state; 1650165b64e2SStefano Zampini PetscInt nnsp_size; 1651165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1652f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1653f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1654165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1655f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1656f4ddd8eeSStefano Zampini break; 1657f4ddd8eeSStefano Zampini } 1658f4ddd8eeSStefano Zampini } 1659f4ddd8eeSStefano Zampini } 1660f4ddd8eeSStefano Zampini } 1661f4ddd8eeSStefano Zampini } else { 1662f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1663f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1664f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1665f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1666f4ddd8eeSStefano Zampini } 1667f4ddd8eeSStefano Zampini } 1668f4ddd8eeSStefano Zampini 1669f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1670727cdba6SStefano Zampini /* reset primal space flags */ 1671f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1672727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 16738de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1674727cdba6SStefano Zampini /* It also sets the primal space flags */ 1675674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 16769543d0ffSStefano Zampini } 1677e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1678f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 16795e8657edSStefano Zampini 16805e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16815e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 16825e8657edSStefano Zampini 16835e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16844f1b2e48SStefano Zampini if (pcbddc->benign_change) { 16851dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1686c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1687c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1688c263805aSStefano Zampini } 16895e8657edSStefano Zampini /* get submatrices */ 16905e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16915e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16925e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16937dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16947dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16957dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16963975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16973975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16989c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1699b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 17005e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 17015e8657edSStefano Zampini pcbddc->local_mat = matis->A; 17025e8657edSStefano Zampini } 170335509ce9Sstefano_zampini 170435509ce9Sstefano_zampini /* interface pressure block row for B_C */ 170535509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr); 170635509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr); 170735509ce9Sstefano_zampini if (lA && lP) { 170835509ce9Sstefano_zampini PC_IS* pcis = (PC_IS*)pc->data; 170935509ce9Sstefano_zampini Mat B_BI,B_BB,Bt_BI,Bt_BB; 171035509ce9Sstefano_zampini PetscBool issym; 171135509ce9Sstefano_zampini ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr); 17126cc1294bSstefano_zampini if (issym) { 17137dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17147dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 171535509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr); 171635509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr); 171735509ce9Sstefano_zampini } else { 17187dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17197dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 17207dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr); 17217dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr); 172235509ce9Sstefano_zampini } 172335509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr); 172435509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr); 172535509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr); 172635509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr); 172735509ce9Sstefano_zampini ierr = MatDestroy(&B_BI);CHKERRQ(ierr); 172835509ce9Sstefano_zampini ierr = MatDestroy(&B_BB);CHKERRQ(ierr); 172935509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr); 173035509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr); 173135509ce9Sstefano_zampini } 173235509ce9Sstefano_zampini 1733b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 173499cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1735b96c3477SStefano Zampini /* SetUp Scaling operator */ 17369d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1737674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 17380c7d97c5SJed Brown } 1739c703fcc7SStefano Zampini 17401dd7afcfSStefano Zampini /* mark topography as done */ 174156282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 17420369aaf7SStefano Zampini 17431dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 17441dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 17451dd7afcfSStefano Zampini 174658a03d70SStefano Zampini if (pcbddc->dbg_flag) { 174758a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 17482b510759SStefano Zampini } 17490c7d97c5SJed Brown PetscFunctionReturn(0); 17500c7d97c5SJed Brown } 17510c7d97c5SJed Brown 17520c7d97c5SJed Brown /* 175350efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 17540c7d97c5SJed Brown 17550c7d97c5SJed Brown Input Parameters: 17560f202f7eSStefano Zampini + pc - the preconditioner context 17570f202f7eSStefano Zampini - r - input vector (global) 17580c7d97c5SJed Brown 17590c7d97c5SJed Brown Output Parameter: 17600c7d97c5SJed Brown . z - output vector (global) 17610c7d97c5SJed Brown 17620c7d97c5SJed Brown Application Interface Routine: PCApply() 17630c7d97c5SJed Brown */ 176453cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 17650c7d97c5SJed Brown { 17660c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 17670c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1768b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 17690c7d97c5SJed Brown PetscErrorCode ierr; 17703b03a366Sstefano_zampini const PetscScalar one = 1.0; 17713b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 17722617d88aSStefano Zampini const PetscScalar zero = 0.0; 17730c7d97c5SJed Brown 17740c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 17750c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1776b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 17770c7d97c5SJed Brown 17780c7d97c5SJed Brown PetscFunctionBegin; 1779f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 17801dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 17811dd7afcfSStefano Zampini Vec swap; 178227b6a85dSStefano Zampini 178327b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 17841dd7afcfSStefano Zampini swap = pcbddc->work_change; 17851dd7afcfSStefano Zampini pcbddc->work_change = r; 17861dd7afcfSStefano Zampini r = swap; 17871dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 17889cc2a9b1Sstefano_zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 17891dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 17901dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 17911dd7afcfSStefano Zampini } 17921dd7afcfSStefano Zampini } 179327b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1794015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1795efc2fbd9SStefano Zampini } 17968ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1797b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 17980c7d97c5SJed Brown /* First Dirichlet solve */ 17990c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18000c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18010c7d97c5SJed Brown /* 18020c7d97c5SJed Brown Assembling right hand side for BDDC operator 1803b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1804674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 18050c7d97c5SJed Brown */ 1806b097fa66SStefano Zampini if (n_D) { 1807b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 18080c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 180916909a7fSStefano Zampini if (pcbddc->switch_static) { 181016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 181116909a7fSStefano Zampini 181216909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 181316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 181416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 181516909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 181616909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 181716909a7fSStefano Zampini } else { 181816909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 181916909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 182016909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182116909a7fSStefano Zampini } 182216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 182316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 182416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 182516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 182616909a7fSStefano Zampini } else { 1827b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 182816909a7fSStefano Zampini } 1829b097fa66SStefano Zampini } else { 1830b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1831b097fa66SStefano Zampini } 18320c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18330c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1834674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1835b76ba322SStefano Zampini } else { 18364fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1837674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1838b76ba322SStefano Zampini } 18394fee134fSStefano Zampini } 1840b76ba322SStefano Zampini 18412617d88aSStefano Zampini /* Apply interface preconditioner 18422617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1843dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 18442617d88aSStefano Zampini 1845674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1846674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 18470c7d97c5SJed Brown 18483b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 18490c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18500c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1851b097fa66SStefano Zampini if (n_B) { 185216909a7fSStefano Zampini if (pcbddc->switch_static) { 185316909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 185416909a7fSStefano Zampini 185516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 185616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 185716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 185816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 185916909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 186016909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186116909a7fSStefano Zampini } else { 186216909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 186416909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186516909a7fSStefano Zampini } 186616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186816909a7fSStefano Zampini } else { 18690c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 187016909a7fSStefano Zampini } 187116909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 187216909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 187316909a7fSStefano Zampini 187416909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 187516909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 187616909a7fSStefano Zampini } else { 187716909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 187816909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 187916909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 188016909a7fSStefano Zampini } 1881b097fa66SStefano Zampini } 1882df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1883efc2fbd9SStefano Zampini 18848ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1885b097fa66SStefano Zampini if (pcbddc->switch_static) { 1886b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1887b097fa66SStefano Zampini } else { 1888b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1889b097fa66SStefano Zampini } 18900c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18910c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1892b097fa66SStefano Zampini } else { 1893b097fa66SStefano Zampini if (pcbddc->switch_static) { 1894b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1895b097fa66SStefano Zampini } else { 1896b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1897b097fa66SStefano Zampini } 1898b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1899b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1900b097fa66SStefano Zampini } 190127b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 19021dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 19031dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 19041dd7afcfSStefano Zampini } 1905015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1906efc2fbd9SStefano Zampini } 19071f4df5f7SStefano Zampini 19081dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1909f913dca9SStefano Zampini pcbddc->work_change = r; 19101dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19111dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19121dd7afcfSStefano Zampini } 19130c7d97c5SJed Brown PetscFunctionReturn(0); 19140c7d97c5SJed Brown } 191550efa1b5SStefano Zampini 191650efa1b5SStefano Zampini /* 191750efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 191850efa1b5SStefano Zampini 191950efa1b5SStefano Zampini Input Parameters: 19200f202f7eSStefano Zampini + pc - the preconditioner context 19210f202f7eSStefano Zampini - r - input vector (global) 192250efa1b5SStefano Zampini 192350efa1b5SStefano Zampini Output Parameter: 192450efa1b5SStefano Zampini . z - output vector (global) 192550efa1b5SStefano Zampini 192650efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 192750efa1b5SStefano Zampini */ 192850efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 192950efa1b5SStefano Zampini { 193050efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 193150efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1932b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 193350efa1b5SStefano Zampini PetscErrorCode ierr; 193450efa1b5SStefano Zampini const PetscScalar one = 1.0; 193550efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 193650efa1b5SStefano Zampini const PetscScalar zero = 0.0; 193750efa1b5SStefano Zampini 193850efa1b5SStefano Zampini PetscFunctionBegin; 1939f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 19401dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19411dd7afcfSStefano Zampini Vec swap; 194227b6a85dSStefano Zampini 194327b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 19441dd7afcfSStefano Zampini swap = pcbddc->work_change; 19451dd7afcfSStefano Zampini pcbddc->work_change = r; 19461dd7afcfSStefano Zampini r = swap; 194727b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 19488ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 194927b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 195027b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 19511dd7afcfSStefano Zampini } 195227b6a85dSStefano Zampini } 195327b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1954537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1955537c1cdfSStefano Zampini } 19568ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1957b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 195850efa1b5SStefano Zampini /* First Dirichlet solve */ 195950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196150efa1b5SStefano Zampini /* 196250efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1963b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 196450efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 196550efa1b5SStefano Zampini */ 1966b097fa66SStefano Zampini if (n_D) { 1967b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 196850efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 196916909a7fSStefano Zampini if (pcbddc->switch_static) { 197016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 197116909a7fSStefano Zampini 197216909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 197316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 197416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 197516909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 197616909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 197716909a7fSStefano Zampini } else { 197816909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 197916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 198016909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 198116909a7fSStefano Zampini } 198216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 198316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 198416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 198516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 198616909a7fSStefano Zampini } else { 1987b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 198816909a7fSStefano Zampini } 1989b097fa66SStefano Zampini } else { 1990b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1991b097fa66SStefano Zampini } 199250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 199350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 199450efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 199550efa1b5SStefano Zampini } else { 199650efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 199750efa1b5SStefano Zampini } 199850efa1b5SStefano Zampini 199950efa1b5SStefano Zampini /* Apply interface preconditioner 200050efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 2001dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 200250efa1b5SStefano Zampini 200350efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 200450efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 200550efa1b5SStefano Zampini 200650efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 200750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 200850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2009b097fa66SStefano Zampini if (n_B) { 201016909a7fSStefano Zampini if (pcbddc->switch_static) { 201116909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 201216909a7fSStefano Zampini 201316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 201416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 201516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 201616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 201716909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 201816909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 201916909a7fSStefano Zampini } else { 202016909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202116909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 202216909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202316909a7fSStefano Zampini } 202416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 202516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 202616909a7fSStefano Zampini } else { 202750efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 202816909a7fSStefano Zampini } 202916909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 203016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 203116909a7fSStefano Zampini 203216909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 203316909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 203416909a7fSStefano Zampini } else { 203516909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 203616909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 203716909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 203816909a7fSStefano Zampini } 2039b097fa66SStefano Zampini } 2040b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 20418ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2042b097fa66SStefano Zampini if (pcbddc->switch_static) { 2043b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 2044b097fa66SStefano Zampini } else { 2045b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 2046b097fa66SStefano Zampini } 204750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 204850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2049b097fa66SStefano Zampini } else { 2050b097fa66SStefano Zampini if (pcbddc->switch_static) { 2051b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 2052b097fa66SStefano Zampini } else { 2053b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 2054b097fa66SStefano Zampini } 2055b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2056b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2057b097fa66SStefano Zampini } 205827b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 2059537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 2060537c1cdfSStefano Zampini } 20611dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2062f913dca9SStefano Zampini pcbddc->work_change = r; 20631dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 20641dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 20651dd7afcfSStefano Zampini } 206650efa1b5SStefano Zampini PetscFunctionReturn(0); 206750efa1b5SStefano Zampini } 2068674ae819SStefano Zampini 20699326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc) 2070da1bb401SStefano Zampini { 2071da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 20729326c5c6Sstefano_zampini PC_IS *pcis = (PC_IS*)pc->data; 20739326c5c6Sstefano_zampini KSP kspD,kspR,kspC; 2074da1bb401SStefano Zampini PetscErrorCode ierr; 2075da1bb401SStefano Zampini 2076da1bb401SStefano Zampini PetscFunctionBegin; 2077674ae819SStefano Zampini /* free BDDC custom data */ 2078674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2079674ae819SStefano Zampini /* destroy objects related to topography */ 2080674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 208134a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2082674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 2083674ae819SStefano Zampini /* free solvers stuff */ 2084674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 208562a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 208662a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 208762a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 20881dd7afcfSStefano Zampini /* free data created by PCIS */ 20891dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 20909326c5c6Sstefano_zampini 20919326c5c6Sstefano_zampini /* restore defaults */ 20929326c5c6Sstefano_zampini kspD = pcbddc->ksp_D; 20939326c5c6Sstefano_zampini kspR = pcbddc->ksp_R; 20949326c5c6Sstefano_zampini kspC = pcbddc->coarse_ksp; 20959326c5c6Sstefano_zampini ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr); 20969326c5c6Sstefano_zampini pcis->n_neigh = -1; 20979326c5c6Sstefano_zampini pcis->scaling_factor = 1.0; 20989326c5c6Sstefano_zampini pcis->reusesubmatrices = PETSC_TRUE; 20999326c5c6Sstefano_zampini pcbddc->use_local_adj = PETSC_TRUE; 21009326c5c6Sstefano_zampini pcbddc->use_vertices = PETSC_TRUE; 21019326c5c6Sstefano_zampini pcbddc->use_edges = PETSC_TRUE; 21029326c5c6Sstefano_zampini pcbddc->symmetric_primal = PETSC_TRUE; 21039326c5c6Sstefano_zampini pcbddc->vertex_size = 1; 21049326c5c6Sstefano_zampini pcbddc->recompute_topography = PETSC_TRUE; 21059326c5c6Sstefano_zampini pcbddc->coarse_size = -1; 21069326c5c6Sstefano_zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 21079326c5c6Sstefano_zampini pcbddc->coarsening_ratio = 8; 21089326c5c6Sstefano_zampini pcbddc->coarse_eqs_per_proc = 1; 21099326c5c6Sstefano_zampini pcbddc->benign_compute_correction = PETSC_TRUE; 21109326c5c6Sstefano_zampini pcbddc->nedfield = -1; 21119326c5c6Sstefano_zampini pcbddc->nedglobal = PETSC_TRUE; 21129326c5c6Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 21139326c5c6Sstefano_zampini pcbddc->sub_schurs_layers = -1; 21149326c5c6Sstefano_zampini pcbddc->ksp_D = kspD; 21159326c5c6Sstefano_zampini pcbddc->ksp_R = kspR; 21169326c5c6Sstefano_zampini pcbddc->coarse_ksp = kspC; 21179326c5c6Sstefano_zampini PetscFunctionReturn(0); 21189326c5c6Sstefano_zampini } 21199326c5c6Sstefano_zampini 21209326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc) 21219326c5c6Sstefano_zampini { 21229326c5c6Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 21239326c5c6Sstefano_zampini PetscErrorCode ierr; 21249326c5c6Sstefano_zampini 21259326c5c6Sstefano_zampini PetscFunctionBegin; 21269326c5c6Sstefano_zampini ierr = PCReset_BDDC(pc);CHKERRQ(ierr); 21279326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr); 21289326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr); 21299326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr); 2130a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr); 2131a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr); 2132906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2133674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 213430368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2135bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 21362b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2137b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 21382b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2139bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 214082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2141bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 214282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2143bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 214482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2145bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2146785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2147bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 214863602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2149bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2150bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2151bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2152bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2153a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2154ab8c8b98SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCSetCoordinates_C",NULL);CHKERRQ(ierr); 2155674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2156da1bb401SStefano Zampini PetscFunctionReturn(0); 2157da1bb401SStefano Zampini } 21581e6b0712SBarry Smith 2159ab8c8b98SStefano Zampini static PetscErrorCode PCSetCoordinates_BDDC(PC pc, PetscInt dim, PetscInt nloc, PetscReal *coords) 2160ab8c8b98SStefano Zampini { 2161ab8c8b98SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2162ab8c8b98SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 2163ab8c8b98SStefano Zampini PetscErrorCode ierr; 2164ab8c8b98SStefano Zampini 2165ab8c8b98SStefano Zampini PetscFunctionBegin; 2166ab8c8b98SStefano Zampini ierr = PetscFree(mat_graph->coords);CHKERRQ(ierr); 2167ab8c8b98SStefano Zampini ierr = PetscMalloc1(nloc*dim,&mat_graph->coords);CHKERRQ(ierr); 2168ab8c8b98SStefano Zampini ierr = PetscMemcpy(mat_graph->coords,coords,nloc*dim*sizeof(PetscReal));CHKERRQ(ierr); 2169ab8c8b98SStefano Zampini mat_graph->cnloc = nloc; 2170ab8c8b98SStefano Zampini mat_graph->cdim = dim; 2171ab8c8b98SStefano Zampini mat_graph->cloc = PETSC_FALSE; 2172ab8c8b98SStefano Zampini PetscFunctionReturn(0); 2173ab8c8b98SStefano Zampini } 2174ab8c8b98SStefano Zampini 2175a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2176a06fd7f2SStefano Zampini { 2177a06fd7f2SStefano Zampini PetscFunctionBegin; 2178a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2179a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2180a06fd7f2SStefano Zampini } 2181a06fd7f2SStefano Zampini 21823425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21833425bc38SStefano Zampini { 2184674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2185266e20e9SStefano Zampini Vec work; 21863425bc38SStefano Zampini PC_IS* pcis; 21873425bc38SStefano Zampini PC_BDDC* pcbddc; 21883425bc38SStefano Zampini PetscErrorCode ierr; 21890c7d97c5SJed Brown 21903425bc38SStefano Zampini PetscFunctionBegin; 21913425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21923425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21933425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21943425bc38SStefano Zampini 2195229984c5Sstefano_zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 2196229984c5Sstefano_zampini /* copy rhs since we may change it during PCPreSolve_BDDC */ 2197229984c5Sstefano_zampini if (!pcbddc->original_rhs) { 2198229984c5Sstefano_zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 2199229984c5Sstefano_zampini } 22006cc1294bSstefano_zampini if (mat_ctx->rhs_flip) { 22016cc1294bSstefano_zampini ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr); 22026cc1294bSstefano_zampini } else { 2203229984c5Sstefano_zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 22046cc1294bSstefano_zampini } 2205af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2206229984c5Sstefano_zampini /* interface pressure rhs */ 2207022d8d2bSstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2208022d8d2bSstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2209229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2210229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22116cc1294bSstefano_zampini if (!mat_ctx->rhs_flip) { 2212229984c5Sstefano_zampini ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr); 2213229984c5Sstefano_zampini } 22146cc1294bSstefano_zampini } 2215c08af4c6SStefano Zampini /* 2216c08af4c6SStefano Zampini change of basis for physical rhs if needed 2217c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2218c08af4c6SStefano Zampini */ 22193738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2220fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 22213738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 22223738a8e6SStefano Zampini work = pcbddc->work_change; 2223fc17d649SStefano Zampini } else { 22243738a8e6SStefano Zampini work = pcbddc->original_rhs; 2225fc17d649SStefano Zampini } 22263425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2227266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2228266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2229fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2230fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2231266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2232266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2233674ae819SStefano Zampini /* Apply partition of unity */ 22343425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2235266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 22368eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 22373425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 22383425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22393425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 22403425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2241266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2242266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2243266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2244266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2245266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2246266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22473425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 22483425bc38SStefano Zampini } 22493425bc38SStefano Zampini /* BDDC rhs */ 22503425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 22518eeda7d8SStefano Zampini if (pcbddc->switch_static) { 22523425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22533425bc38SStefano Zampini } 22543425bc38SStefano Zampini /* apply BDDC */ 2255229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2256dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2257266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2258229984c5Sstefano_zampini 22593425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 22603425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 22613425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22623425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2263229984c5Sstefano_zampini /* Add contribution to interface pressures */ 2264229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2265229984c5Sstefano_zampini ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr); 2266229984c5Sstefano_zampini if (pcbddc->switch_static) { 2267229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr); 2268229984c5Sstefano_zampini } 2269229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2270229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2271229984c5Sstefano_zampini } 22723425bc38SStefano Zampini PetscFunctionReturn(0); 22733425bc38SStefano Zampini } 22741e6b0712SBarry Smith 22753425bc38SStefano Zampini /*@ 22760f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 22773425bc38SStefano Zampini 22783425bc38SStefano Zampini Collective 22793425bc38SStefano Zampini 22803425bc38SStefano Zampini Input Parameters: 22810f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 22820f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 22833425bc38SStefano Zampini 22843425bc38SStefano Zampini Output Parameters: 22850f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 22863425bc38SStefano Zampini 22873425bc38SStefano Zampini Level: developer 22883425bc38SStefano Zampini 22893425bc38SStefano Zampini Notes: 22903425bc38SStefano Zampini 22910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 22923425bc38SStefano Zampini @*/ 22933425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 22943425bc38SStefano Zampini { 2295674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22963425bc38SStefano Zampini PetscErrorCode ierr; 22973425bc38SStefano Zampini 22983425bc38SStefano Zampini PetscFunctionBegin; 2299266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2300266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2301266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 23023425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2303163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 23043425bc38SStefano Zampini PetscFunctionReturn(0); 23053425bc38SStefano Zampini } 23061e6b0712SBarry Smith 23073425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23083425bc38SStefano Zampini { 2309674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23103425bc38SStefano Zampini PC_IS* pcis; 23113425bc38SStefano Zampini PC_BDDC* pcbddc; 23123425bc38SStefano Zampini PetscErrorCode ierr; 2313229984c5Sstefano_zampini Vec work; 23143425bc38SStefano Zampini 23153425bc38SStefano Zampini PetscFunctionBegin; 23163425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 23173425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 23183425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 23193425bc38SStefano Zampini 23203425bc38SStefano Zampini /* apply B_delta^T */ 2321af140850Sstefano_zampini ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr); 23223425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23233425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23243425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 2325229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2326229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2327229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2328229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 2329229984c5Sstefano_zampini } 2330229984c5Sstefano_zampini 23313425bc38SStefano Zampini /* compute rhs for BDDC application */ 23323425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 23338eeda7d8SStefano Zampini if (pcbddc->switch_static) { 23343425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 2335229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2336229984c5Sstefano_zampini ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr); 2337229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23383425bc38SStefano Zampini } 2339229984c5Sstefano_zampini } 2340229984c5Sstefano_zampini 23413425bc38SStefano Zampini /* apply BDDC */ 2342229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2343dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2344229984c5Sstefano_zampini 2345229984c5Sstefano_zampini /* put values into global vector */ 2346af140850Sstefano_zampini if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change; 2347af140850Sstefano_zampini else work = standard_sol; 2348229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2349229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23508eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 23513425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 23523425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 235300f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 235400f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23553425bc38SStefano Zampini } 2356229984c5Sstefano_zampini 2357229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2358229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2359266e20e9SStefano Zampini /* add p0 solution to final solution */ 2360229984c5Sstefano_zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr); 2361fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2362af140850Sstefano_zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr); 2363fc17d649SStefano Zampini } 2364af140850Sstefano_zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 2365af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2366229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2367229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2368229984c5Sstefano_zampini } 23693425bc38SStefano Zampini PetscFunctionReturn(0); 23703425bc38SStefano Zampini } 23711e6b0712SBarry Smith 23723425bc38SStefano Zampini /*@ 23730f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 23743425bc38SStefano Zampini 23753425bc38SStefano Zampini Collective 23763425bc38SStefano Zampini 23773425bc38SStefano Zampini Input Parameters: 23780f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 23790f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 23803425bc38SStefano Zampini 23813425bc38SStefano Zampini Output Parameters: 23820f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 23833425bc38SStefano Zampini 23843425bc38SStefano Zampini Level: developer 23853425bc38SStefano Zampini 23863425bc38SStefano Zampini Notes: 23873425bc38SStefano Zampini 23880f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 23893425bc38SStefano Zampini @*/ 23903425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23913425bc38SStefano Zampini { 2392674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23933425bc38SStefano Zampini PetscErrorCode ierr; 23943425bc38SStefano Zampini 23953425bc38SStefano Zampini PetscFunctionBegin; 2396266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2397266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2398266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 23993425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2400163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 24013425bc38SStefano Zampini PetscFunctionReturn(0); 24023425bc38SStefano Zampini } 24031e6b0712SBarry Smith 2404547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc) 24053425bc38SStefano Zampini { 2406674ae819SStefano Zampini 2407674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 24083425bc38SStefano Zampini Mat newmat; 2409674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 24103425bc38SStefano Zampini PC newpc; 2411ce94432eSBarry Smith MPI_Comm comm; 24123425bc38SStefano Zampini PetscErrorCode ierr; 24133425bc38SStefano Zampini 24143425bc38SStefano Zampini PetscFunctionBegin; 2415ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 24163425bc38SStefano Zampini /* FETIDP linear matrix */ 24173425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 24181720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 24193425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 2420a5bb87b3Sstefano_zampini ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 24213425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2422edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 24233425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 2424547c9a8eSstefano_zampini ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr); 2425547c9a8eSstefano_zampini ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr); 24263425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 24273425bc38SStefano Zampini /* FETIDP preconditioner */ 24283425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 24293425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 24303425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 2431e1214c54Sstefano_zampini ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 2432e1214c54Sstefano_zampini ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr); 2433e1214c54Sstefano_zampini ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr); 2434e1214c54Sstefano_zampini if (!fetidpmat_ctx->l2g_lambda_only) { 24353425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 24363425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 24373425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2438edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2439c45b8d2dSstefano_zampini ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr); 24403425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2441e1214c54Sstefano_zampini } else { 2442e1214c54Sstefano_zampini KSP *ksps; 2443e1214c54Sstefano_zampini PC lagpc; 2444e1214c54Sstefano_zampini Mat M,AM,PM; 2445e1214c54Sstefano_zampini PetscInt nn; 2446e1214c54Sstefano_zampini 2447e1214c54Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_PPmat",(PetscObject*)&M);CHKERRQ(ierr); 2448e1214c54Sstefano_zampini ierr = PCSetType(newpc,PCFIELDSPLIT);CHKERRQ(ierr); 2449e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"lag",fetidpmat_ctx->lagrange);CHKERRQ(ierr); 2450e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"p",fetidpmat_ctx->pressure);CHKERRQ(ierr); 2451e1214c54Sstefano_zampini ierr = PCFieldSplitSetType(newpc,PC_COMPOSITE_SCHUR);CHKERRQ(ierr); 245240c75d76SStefano Zampini ierr = PCFieldSplitSetSchurFactType(newpc,PC_FIELDSPLIT_SCHUR_FACT_DIAG);CHKERRQ(ierr); 2453e1214c54Sstefano_zampini ierr = PCFieldSplitSetSchurPre(newpc,PC_FIELDSPLIT_SCHUR_PRE_USER,M);CHKERRQ(ierr); 2454c096484dSStefano Zampini ierr = PCFieldSplitSetSchurScale(newpc,1.0);CHKERRQ(ierr); 2455e1214c54Sstefano_zampini ierr = PCSetFromOptions(newpc);CHKERRQ(ierr); 2456e1214c54Sstefano_zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 2457e1214c54Sstefano_zampini 2458e1214c54Sstefano_zampini /* set the solver for the (0,0) block */ 245940c75d76SStefano Zampini ierr = PCFieldSplitGetSubKSP(newpc,&nn,&ksps);CHKERRQ(ierr); 2460e1214c54Sstefano_zampini ierr = PCCreate(comm,&lagpc);CHKERRQ(ierr); 2461e1214c54Sstefano_zampini ierr = PCSetType(lagpc,PCSHELL);CHKERRQ(ierr); 2462e1214c54Sstefano_zampini ierr = KSPGetOperators(ksps[0],&AM,&PM);CHKERRQ(ierr); 2463e1214c54Sstefano_zampini ierr = PCSetOperators(lagpc,AM,PM);CHKERRQ(ierr); 2464e1214c54Sstefano_zampini ierr = PCShellSetContext(lagpc,fetidppc_ctx);CHKERRQ(ierr); 2465e1214c54Sstefano_zampini ierr = PCShellSetApply(lagpc,FETIDPPCApply);CHKERRQ(ierr); 2466e1214c54Sstefano_zampini ierr = PCShellSetApplyTranspose(lagpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2467e1214c54Sstefano_zampini ierr = PCShellSetView(lagpc,FETIDPPCView);CHKERRQ(ierr); 2468e1214c54Sstefano_zampini ierr = PCShellSetDestroy(lagpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2469e1214c54Sstefano_zampini ierr = KSPSetPC(ksps[0],lagpc);CHKERRQ(ierr); 2470e1214c54Sstefano_zampini ierr = PCDestroy(&lagpc);CHKERRQ(ierr); 2471e1214c54Sstefano_zampini ierr = PetscFree(ksps);CHKERRQ(ierr); 2472e1214c54Sstefano_zampini } 24733425bc38SStefano Zampini /* return pointers for objects created */ 24743425bc38SStefano Zampini *fetidp_mat = newmat; 24753425bc38SStefano Zampini *fetidp_pc = newpc; 24763425bc38SStefano Zampini PetscFunctionReturn(0); 24773425bc38SStefano Zampini } 24781e6b0712SBarry Smith 247994ef8ddeSSatish Balay /*@C 24800f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 24813425bc38SStefano Zampini 24823425bc38SStefano Zampini Collective 24833425bc38SStefano Zampini 24843425bc38SStefano Zampini Input Parameters: 24851720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 2486547c9a8eSstefano_zampini . fully_redundant - true for a fully redundant set of Lagrange multipliers 2487547c9a8eSstefano_zampini - prefix - optional options database prefix for the objects to be created (can be NULL) 248828509bceSStefano Zampini 248928509bceSStefano Zampini Output Parameters: 24900f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 24910f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 249228509bceSStefano Zampini 24933425bc38SStefano Zampini Level: developer 24943425bc38SStefano Zampini 24953425bc38SStefano Zampini Notes: 24960f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 24973425bc38SStefano Zampini 24980f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 24993425bc38SStefano Zampini @*/ 2500547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc) 25013425bc38SStefano Zampini { 25023425bc38SStefano Zampini PetscErrorCode ierr; 25033425bc38SStefano Zampini 25043425bc38SStefano Zampini PetscFunctionBegin; 25053425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 25063425bc38SStefano Zampini if (pc->setupcalled) { 2507547c9a8eSstefano_zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 25084247aa23Sstefano_zampini } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 25093425bc38SStefano Zampini PetscFunctionReturn(0); 25103425bc38SStefano Zampini } 25110c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2512da1bb401SStefano Zampini /*MC 2513da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 25140c7d97c5SJed Brown 251528509bceSStefano Zampini An implementation of the BDDC preconditioner based on 251628509bceSStefano Zampini 251728509bceSStefano Zampini .vb 251828509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 251928509bceSStefano 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 252028509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 25210f202f7eSStefano 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 252228509bceSStefano Zampini .ve 252328509bceSStefano Zampini 252428509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 252528509bceSStefano Zampini 25260f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 252728509bceSStefano Zampini 252828509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 252928509bceSStefano Zampini 2530b6fdb6dfSStefano 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. 2531b6fdb6dfSStefano Zampini 2532c7017625SStefano 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). 253328509bceSStefano Zampini 25340f202f7eSStefano 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() 253530368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 253628509bceSStefano Zampini 25370f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 253828509bceSStefano Zampini 25390f202f7eSStefano 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. 25400f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 254128509bceSStefano Zampini 25420f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 254328509bceSStefano Zampini 2544df4d28bfSStefano 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. 254528509bceSStefano Zampini 25460f202f7eSStefano 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. 25470f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 25480f202f7eSStefano Zampini 25490f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 25500f202f7eSStefano Zampini 25510f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 25520f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 25530f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 25540f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 25550f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 25560f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 25570f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 255828509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 25590f202f7eSStefano 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) 25605459c157SBarry 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) 25610f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 25620f202f7eSStefano 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) 2563bd2a564bSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value different than zero is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 256428509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 256528509bceSStefano Zampini 256628509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 256728509bceSStefano Zampini .vb 256828509bceSStefano Zampini -pc_bddc_dirichlet_ 256928509bceSStefano Zampini -pc_bddc_neumann_ 257028509bceSStefano Zampini -pc_bddc_coarse_ 257128509bceSStefano Zampini .ve 25720f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 257328509bceSStefano Zampini 25740f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 257528509bceSStefano Zampini .vb 2576312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2577312be037SStefano Zampini -pc_bddc_neumann_lN_ 2578312be037SStefano Zampini -pc_bddc_coarse_lN_ 257928509bceSStefano Zampini .ve 25800f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 25810f202f7eSStefano 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. 25820f202f7eSStefano Zampini .vb 25830f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 25840f202f7eSStefano Zampini .ve 25850f202f7eSStefano 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 2586da1bb401SStefano Zampini 2587da1bb401SStefano Zampini Level: intermediate 2588da1bb401SStefano Zampini 2589b6fdb6dfSStefano Zampini Developer notes: 2590da1bb401SStefano Zampini 2591da1bb401SStefano Zampini Contributed by Stefano Zampini 2592da1bb401SStefano Zampini 2593da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2594da1bb401SStefano Zampini M*/ 2595b2573a8aSBarry Smith 25968cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2597da1bb401SStefano Zampini { 2598da1bb401SStefano Zampini PetscErrorCode ierr; 2599da1bb401SStefano Zampini PC_BDDC *pcbddc; 2600da1bb401SStefano Zampini 2601da1bb401SStefano Zampini PetscFunctionBegin; 2602b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2603da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2604da1bb401SStefano Zampini 2605da1bb401SStefano Zampini /* create PCIS data structure */ 2606da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2607da1bb401SStefano Zampini 26089326c5c6Sstefano_zampini /* create local graph structure */ 26099326c5c6Sstefano_zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 26109326c5c6Sstefano_zampini 26119326c5c6Sstefano_zampini /* BDDC nonzero defaults */ 261208a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 261347d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 261447d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 26153301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 261614f95afaSStefano Zampini pcbddc->vertex_size = 1; 2617c703fcc7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 261868457ee5SStefano Zampini pcbddc->coarse_size = -1; 261985c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 262047d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 262157de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 262227b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 26231e0482f5SStefano Zampini pcbddc->nedfield = -1; 26241e0482f5SStefano Zampini pcbddc->nedglobal = PETSC_TRUE; 2625be12c134Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 2626b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2627bd2a564bSStefano Zampini pcbddc->adaptive_threshold[0] = 0.0; 2628bd2a564bSStefano Zampini pcbddc->adaptive_threshold[1] = 0.0; 2629b7eb3628SStefano Zampini 2630da1bb401SStefano Zampini /* function pointers */ 2631da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 263293bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2633da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2634da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2635da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 26366b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2637da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2638da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2639da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2640534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2641534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 26429326c5c6Sstefano_zampini pc->ops->reset = PCReset_BDDC; 2643da1bb401SStefano Zampini 2644da1bb401SStefano Zampini /* composing function */ 2645a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr); 2646a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr); 2647906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2648674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 264930368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2650bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 26512b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2652b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 26532b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2654bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 265582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2656bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 265782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2658bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 265982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2660bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 266182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2662bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 266363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2664bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2665bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2666bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2667bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2668a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2669ab8c8b98SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCSetCoordinates_C",PCSetCoordinates_BDDC);CHKERRQ(ierr); 2670da1bb401SStefano Zampini PetscFunctionReturn(0); 2671da1bb401SStefano Zampini } 2672