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 440c7d97c5SJed Brown #undef __FUNCT__ 450c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 464416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 470c7d97c5SJed Brown { 480c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 490c7d97c5SJed Brown PetscErrorCode ierr; 500c7d97c5SJed Brown 510c7d97c5SJed Brown PetscFunctionBegin; 52e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 538eeda7d8SStefano Zampini /* Verbose debugging */ 54a13144ffSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 55a13144ffSStefano Zampini /* Approximate solvers */ 56c7017625SStefano 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); 57c7017625SStefano 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); 58c7017625SStefano 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); 59c7017625SStefano 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); 606b78500eSPatrick Sanan /* Primal space customization */ 6108a5cf49SStefano 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); 62be12c134Sstefano_zampini ierr = PetscOptionsInt("-pc_bddc_graph_maxcount","Maximum number of shared subdomains for a connected component","none",pcbddc->graphmaxcount,&pcbddc->graphmaxcount,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); 87*839e9adbSstefano_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); 894c6709b3SStefano Zampini ierr = PetscOptionsReal("-pc_bddc_adaptive_threshold","Threshold to be used for adaptive selection of constraints","none",pcbddc->adaptive_threshold,&pcbddc->adaptive_threshold,NULL);CHKERRQ(ierr); 9008122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 9108122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 923301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 93b0c7d250SStefano 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); 94b3afcdbeSStefano 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); 95e9627c49SStefano 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); 9627b6a85dSStefano 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); 97a198735bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->compute_nonetflux,&pcbddc->compute_nonetflux,NULL);CHKERRQ(ierr); 984f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 9970c64980SStefano 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); 1000c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 1010c7d97c5SJed Brown PetscFunctionReturn(0); 1020c7d97c5SJed Brown } 1036b78500eSPatrick Sanan 1046b78500eSPatrick Sanan #undef __FUNCT__ 1056b78500eSPatrick Sanan #define __FUNCT__ "PCView_BDDC" 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; 1116b78500eSPatrick Sanan PetscBool isascii,isstring; 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 ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 1186b78500eSPatrick Sanan /* Nothing printed for the String viewer */ 1196b78500eSPatrick Sanan /* ASCII viewer */ 1206b78500eSPatrick Sanan if (isascii) { 1214b2aedd3SStefano Zampini PetscMPIInt color,rank,size; 122fbad9177SStefano Zampini PetscInt64 loc[7],gsum[6],gmax[6],gmin[6],totbenign; 123e9627c49SStefano Zampini PetscScalar interface_size; 124e9627c49SStefano Zampini PetscReal ratio1=0.,ratio2=0.; 125e9627c49SStefano Zampini Vec counter; 1266b78500eSPatrick Sanan 127b74ba07aSstefano_zampini if (!pc->setupcalled) { 128b74ba07aSstefano_zampini ierr = PetscViewerASCIIPrintf(viewer," Partial information available: preconditioner has not been setup yet\n");CHKERRQ(ierr); 129b74ba07aSstefano_zampini } 130e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use verbose output: %d\n",pcbddc->dbg_flag);CHKERRQ(ierr); 131e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use user-defined CSR: %d\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 13266da6bd7Sstefano_zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use local mat graph: %d\n",pcbddc->use_local_adj && !pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 133e9627c49SStefano Zampini if (pcbddc->mat_graph->twodim) { 134e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Connectivity graph topological dimension: 2\n");CHKERRQ(ierr); 135e9627c49SStefano Zampini } else { 136e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Connectivity graph topological dimension: 3\n");CHKERRQ(ierr); 137e9627c49SStefano Zampini } 138a70d66d9SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Graph max count: %d\n",pcbddc->graphmaxcount);CHKERRQ(ierr); 139a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use vertices: %d (vertex size %d)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 1406b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 1416b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 1426b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 1436b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 1446b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 1456b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 14627b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 14727b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 14827b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 1496b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 15027b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 1516b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr); 152e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 1536b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 1546b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 155e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 156*839e9adbSstefano_zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr); 1576b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 1586b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 1596b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 160e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_threshold,pcbddc->adaptive_selection,pcbddc->adaptive_userdefined);CHKERRQ(ierr); 1616b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 1626b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 163e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 1646b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 1656b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 166e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Coarse eqs per proc (significant at the coarsest level): %d\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 167e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr); 168fc17d649SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 169a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr); 170a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr); 171b74ba07aSstefano_zampini if (!pc->setupcalled) PetscFunctionReturn(0); 1726b78500eSPatrick Sanan 173fbad9177SStefano Zampini /* compute interface size */ 174e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 175e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 176e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 177e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 178e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 179e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 180e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 181fbad9177SStefano Zampini 182fbad9177SStefano Zampini /* compute some statistics on the domain decomposition */ 183e9627c49SStefano Zampini gsum[0] = 1; 184fbad9177SStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0; 185e9627c49SStefano Zampini loc[0] = !!pcis->n; 186e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 187e9627c49SStefano Zampini loc[2] = pcis->n_B; 188e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 189345ecf6cSStefano Zampini loc[4] = pcis->n; 190fbad9177SStefano Zampini loc[5] = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0); 191fbad9177SStefano Zampini loc[6] = pcbddc->benign_n; 192fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 193fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1; 194fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 195fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT; 196fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 197fbad9177SStefano Zampini ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 198e9627c49SStefano Zampini if (pcbddc->coarse_size) { 199e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 200e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 201e9627c49SStefano Zampini } 202e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 203e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Global dofs sizes: all %d interface %d coarse %d\n",pc->pmat->rmap->N,(PetscInt)PetscRealPart(interface_size),pcbddc->coarse_size);CHKERRQ(ierr); 204e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Coarsening ratios: all/coarse %d interface/coarse %d\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 205e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Active processes : %d\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 206fbad9177SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Total subdomains : %d\n",(PetscInt)gsum[5]);CHKERRQ(ierr); 207345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 208345ecf6cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subs : %d\n",(PetscInt)totbenign);CHKERRQ(ierr); 209345ecf6cSStefano Zampini } 210e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Dofs type :\tMIN\tMAX\tMEAN\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 211e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Interior dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 212e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Interface dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[2],(PetscInt)gmax[2],(PetscInt)(gsum[2]/gsum[0]));CHKERRQ(ierr); 213e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Primal dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[3],(PetscInt)gmax[3],(PetscInt)(gsum[3]/gsum[0]));CHKERRQ(ierr); 214345ecf6cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Local dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[4],(PetscInt)gmax[4],(PetscInt)(gsum[4]/gsum[0]));CHKERRQ(ierr); 215fbad9177SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Local subs :\t%d\t%d\n",(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr); 21627b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: ********************************** COARSE PROBLEM DETAILS *********************************\n",pcbddc->current_level);CHKERRQ(ierr); 21727b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 218e9627c49SStefano Zampini 219fbad9177SStefano Zampini /* the coarse problem can be handled by a different communicator */ 220e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 221e9627c49SStefano Zampini else color = 0; 222e9627c49SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 2234b2aedd3SStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 224e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 2254b2aedd3SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr); 226e9627c49SStefano Zampini ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr); 227e9627c49SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 228e9627c49SStefano Zampini if (color == 1) { 229e9627c49SStefano Zampini ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr); 230e9627c49SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 231e9627c49SStefano Zampini } 232e9627c49SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 233e9627c49SStefano Zampini ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr); 234e9627c49SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 235e9627c49SStefano Zampini } 2366b78500eSPatrick Sanan PetscFunctionReturn(0); 2376b78500eSPatrick Sanan } 238a13144ffSStefano Zampini 239a13144ffSStefano Zampini #undef __FUNCT__ 240a13144ffSStefano Zampini #define __FUNCT__ "PCBDDCSetDiscreteGradient_BDDC" 2411e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 242a13144ffSStefano Zampini { 243a13144ffSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 244a13144ffSStefano Zampini PetscErrorCode ierr; 245a13144ffSStefano Zampini 246a13144ffSStefano Zampini PetscFunctionBegin; 247a13144ffSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 248a13144ffSStefano Zampini ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); 249a13144ffSStefano Zampini pcbddc->discretegradient = G; 250a13144ffSStefano Zampini pcbddc->nedorder = order > 0 ? order : -order; 251495a2a07SStefano Zampini pcbddc->nedfield = field; 2521e0482f5SStefano Zampini pcbddc->nedglobal = global; 2531e0482f5SStefano Zampini pcbddc->conforming = conforming; 254a13144ffSStefano Zampini PetscFunctionReturn(0); 255a13144ffSStefano Zampini } 256a13144ffSStefano Zampini 257a13144ffSStefano Zampini #undef __FUNCT__ 258a13144ffSStefano Zampini #define __FUNCT__ "PCBDDCSetDiscreteGradient" 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 299a198735bSStefano Zampini #undef __FUNCT__ 300a198735bSStefano Zampini #define __FUNCT__ "PCBDDCSetDivergenceMat_BDDC" 3018ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 302a198735bSStefano Zampini { 303a198735bSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 304a198735bSStefano Zampini PetscErrorCode ierr; 3056b78500eSPatrick Sanan 306a198735bSStefano Zampini PetscFunctionBegin; 307a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr); 308a198735bSStefano Zampini ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); 309a198735bSStefano Zampini pcbddc->divudotp = divudotp; 3108ae0ca82SStefano Zampini pcbddc->divudotp_trans = trans; 311a198735bSStefano Zampini pcbddc->compute_nonetflux = PETSC_TRUE; 312a198735bSStefano Zampini if (vl2l) { 313a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr); 314fa23a32eSStefano Zampini ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); 315a198735bSStefano Zampini pcbddc->divudotp_vl2l = vl2l; 316a198735bSStefano Zampini } 317a198735bSStefano Zampini PetscFunctionReturn(0); 318a198735bSStefano Zampini } 3193d996552SStefano Zampini 320a198735bSStefano Zampini #undef __FUNCT__ 321a198735bSStefano Zampini #define __FUNCT__ "PCBDDCSetDivergenceMat" 322a198735bSStefano Zampini /*@ 323a198735bSStefano Zampini PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx 324a198735bSStefano Zampini 325a198735bSStefano Zampini Collective on PC 326a198735bSStefano Zampini 327a198735bSStefano Zampini Input Parameters: 328a198735bSStefano Zampini + pc - the preconditioning context 329a198735bSStefano Zampini . divudotp - the matrix (must be of type MATIS) 3308ae0ca82SStefano Zampini . trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space. 331bb05f991SStefano Zampini - vl2l - optional IS describing the local (wrt the local mat in divudotp) to local (wrt the local mat in pc->pmat) map for the velocities 332a198735bSStefano Zampini 333a198735bSStefano Zampini Level: advanced 334a198735bSStefano Zampini 3358ae0ca82SStefano Zampini Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries 336a198735bSStefano Zampini 337a198735bSStefano Zampini .seealso: PCBDDC 338a198735bSStefano Zampini @*/ 3398ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 340a198735bSStefano Zampini { 341a198735bSStefano Zampini PetscBool ismatis; 342a198735bSStefano Zampini PetscErrorCode ierr; 343a198735bSStefano Zampini 344a198735bSStefano Zampini PetscFunctionBegin; 345a198735bSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 346a198735bSStefano Zampini PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2); 347a198735bSStefano Zampini PetscCheckSameComm(pc,1,divudotp,2); 3488ae0ca82SStefano Zampini PetscValidLogicalCollectiveBool(pc,trans,3); 3498ae0ca82SStefano Zampini if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4); 350a198735bSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr); 351a198735bSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS"); 3528ae0ca82SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr); 353a198735bSStefano Zampini PetscFunctionReturn(0); 354a198735bSStefano Zampini } 3552d505d7fSStefano Zampini 356674ae819SStefano Zampini #undef __FUNCT__ 357906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 3581dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 359b9b85e73SStefano Zampini { 360b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 361b9b85e73SStefano Zampini PetscErrorCode ierr; 362b9b85e73SStefano Zampini 363b9b85e73SStefano Zampini PetscFunctionBegin; 364b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 36556282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 366b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 3671dd7afcfSStefano Zampini pcbddc->change_interior = interior; 368b9b85e73SStefano Zampini PetscFunctionReturn(0); 369b9b85e73SStefano Zampini } 370b9b85e73SStefano Zampini #undef __FUNCT__ 371906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 372b9b85e73SStefano Zampini /*@ 373906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 374b9b85e73SStefano Zampini 375b9b85e73SStefano Zampini Collective on PC 376b9b85e73SStefano Zampini 377b9b85e73SStefano Zampini Input Parameters: 378b9b85e73SStefano Zampini + pc - the preconditioning context 3791dd7afcfSStefano Zampini . change - the change of basis matrix 3801dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 381b9b85e73SStefano Zampini 382b9b85e73SStefano Zampini Level: intermediate 383b9b85e73SStefano Zampini 384b9b85e73SStefano Zampini Notes: 385b9b85e73SStefano Zampini 386b9b85e73SStefano Zampini .seealso: PCBDDC 387b9b85e73SStefano Zampini @*/ 3881dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 389b9b85e73SStefano Zampini { 390b9b85e73SStefano Zampini PetscErrorCode ierr; 391b9b85e73SStefano Zampini 392b9b85e73SStefano Zampini PetscFunctionBegin; 393b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 394b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 395906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 396906d46d4SStefano Zampini if (pc->mat) { 397906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 398906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 399906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 4006c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 4016c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 402906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 403906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 4046c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 4056c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 406906d46d4SStefano Zampini } 4071dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 408b9b85e73SStefano Zampini PetscFunctionReturn(0); 409b9b85e73SStefano Zampini } 4102d505d7fSStefano Zampini 411b9b85e73SStefano Zampini #undef __FUNCT__ 41230368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS_BDDC" 41330368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 41430368db7SStefano Zampini { 41530368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 41656282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 41730368db7SStefano Zampini PetscErrorCode ierr; 41830368db7SStefano Zampini 41930368db7SStefano Zampini PetscFunctionBegin; 42056282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 42156282151SStefano Zampini if (pcbddc->user_primal_vertices) { 42256282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 42356282151SStefano Zampini } 42430368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 42530368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 42630368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 42756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 42830368db7SStefano Zampini PetscFunctionReturn(0); 42930368db7SStefano Zampini } 43030368db7SStefano Zampini #undef __FUNCT__ 43130368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS" 43230368db7SStefano Zampini /*@ 43330368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 43430368db7SStefano Zampini 43530368db7SStefano Zampini Collective 43630368db7SStefano Zampini 43730368db7SStefano Zampini Input Parameters: 43830368db7SStefano Zampini + pc - the preconditioning context 43930368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 44030368db7SStefano Zampini 44130368db7SStefano Zampini Level: intermediate 44230368db7SStefano Zampini 44330368db7SStefano Zampini Notes: 44430368db7SStefano Zampini Any process can list any global node 44530368db7SStefano Zampini 44630368db7SStefano Zampini .seealso: PCBDDC 44730368db7SStefano Zampini @*/ 44830368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 44930368db7SStefano Zampini { 45030368db7SStefano Zampini PetscErrorCode ierr; 45130368db7SStefano Zampini 45230368db7SStefano Zampini PetscFunctionBegin; 45330368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 45430368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 45530368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 45630368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 45730368db7SStefano Zampini PetscFunctionReturn(0); 45830368db7SStefano Zampini } 4592d505d7fSStefano Zampini 46030368db7SStefano Zampini #undef __FUNCT__ 461674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 462674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 463674ae819SStefano Zampini { 464674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 46556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 466674ae819SStefano Zampini PetscErrorCode ierr; 4671e6b0712SBarry Smith 468674ae819SStefano Zampini PetscFunctionBegin; 46956282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 47056282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 47156282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 47256282151SStefano Zampini } 473674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 47430368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 47530368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 47656282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 477674ae819SStefano Zampini PetscFunctionReturn(0); 478674ae819SStefano Zampini } 479674ae819SStefano Zampini #undef __FUNCT__ 480674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 481674ae819SStefano Zampini /*@ 48228509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 483674ae819SStefano Zampini 48417eb1463SStefano Zampini Collective 485674ae819SStefano Zampini 486674ae819SStefano Zampini Input Parameters: 487674ae819SStefano Zampini + pc - the preconditioning context 48817eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 489674ae819SStefano Zampini 490674ae819SStefano Zampini Level: intermediate 491674ae819SStefano Zampini 492674ae819SStefano Zampini Notes: 493674ae819SStefano Zampini 494674ae819SStefano Zampini .seealso: PCBDDC 495674ae819SStefano Zampini @*/ 496674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 497674ae819SStefano Zampini { 498674ae819SStefano Zampini PetscErrorCode ierr; 499674ae819SStefano Zampini 500674ae819SStefano Zampini PetscFunctionBegin; 501674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 502674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 50317eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 504674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 505674ae819SStefano Zampini PetscFunctionReturn(0); 506674ae819SStefano Zampini } 5072d505d7fSStefano Zampini 5080c7d97c5SJed Brown #undef __FUNCT__ 5094fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 5104fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 5114fad6a16SStefano Zampini { 5124fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5134fad6a16SStefano Zampini 5144fad6a16SStefano Zampini PetscFunctionBegin; 5154fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 5164fad6a16SStefano Zampini PetscFunctionReturn(0); 5174fad6a16SStefano Zampini } 5181e6b0712SBarry Smith 5194fad6a16SStefano Zampini #undef __FUNCT__ 5204fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 5214fad6a16SStefano Zampini /*@ 52228509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 5234fad6a16SStefano Zampini 5244fad6a16SStefano Zampini Logically collective on PC 5254fad6a16SStefano Zampini 5264fad6a16SStefano Zampini Input Parameters: 5274fad6a16SStefano Zampini + pc - the preconditioning context 52828509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 5294fad6a16SStefano Zampini 5300f202f7eSStefano Zampini Options Database Keys: 5310f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 5324fad6a16SStefano Zampini 5334fad6a16SStefano Zampini Level: intermediate 5344fad6a16SStefano Zampini 5354fad6a16SStefano Zampini Notes: 5360f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 5374fad6a16SStefano Zampini 5380f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 5394fad6a16SStefano Zampini @*/ 5404fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 5414fad6a16SStefano Zampini { 5424fad6a16SStefano Zampini PetscErrorCode ierr; 5434fad6a16SStefano Zampini 5444fad6a16SStefano Zampini PetscFunctionBegin; 5454fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5462b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 5474fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 5484fad6a16SStefano Zampini PetscFunctionReturn(0); 5494fad6a16SStefano Zampini } 5502b510759SStefano Zampini 551b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 5522b510759SStefano Zampini #undef __FUNCT__ 553b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 554b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 555b8ffe317SStefano Zampini { 556b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 557b8ffe317SStefano Zampini 558b8ffe317SStefano Zampini PetscFunctionBegin; 55985c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 560b8ffe317SStefano Zampini PetscFunctionReturn(0); 561b8ffe317SStefano Zampini } 562b8ffe317SStefano Zampini 563b8ffe317SStefano Zampini #undef __FUNCT__ 564b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 565b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 5662b510759SStefano Zampini { 5672b510759SStefano Zampini PetscErrorCode ierr; 5682b510759SStefano Zampini 5692b510759SStefano Zampini PetscFunctionBegin; 5702b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 571b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 572b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 5732b510759SStefano Zampini PetscFunctionReturn(0); 5742b510759SStefano Zampini } 5751e6b0712SBarry Smith 5764fad6a16SStefano Zampini #undef __FUNCT__ 5772b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 5782b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 5794fad6a16SStefano Zampini { 5804fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5814fad6a16SStefano Zampini 5824fad6a16SStefano Zampini PetscFunctionBegin; 5832b510759SStefano Zampini pcbddc->current_level = level; 5844fad6a16SStefano Zampini PetscFunctionReturn(0); 5854fad6a16SStefano Zampini } 5861e6b0712SBarry Smith 5874fad6a16SStefano Zampini #undef __FUNCT__ 588b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 589b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 590b8ffe317SStefano Zampini { 591b8ffe317SStefano Zampini PetscErrorCode ierr; 592b8ffe317SStefano Zampini 593b8ffe317SStefano Zampini PetscFunctionBegin; 594b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 595b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 596b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 597b8ffe317SStefano Zampini PetscFunctionReturn(0); 598b8ffe317SStefano Zampini } 599b8ffe317SStefano Zampini 600b8ffe317SStefano Zampini #undef __FUNCT__ 6012b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 6022b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 6032b510759SStefano Zampini { 6042b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 6052b510759SStefano Zampini 6062b510759SStefano Zampini PetscFunctionBegin; 6072b510759SStefano Zampini pcbddc->max_levels = levels; 6082b510759SStefano Zampini PetscFunctionReturn(0); 6092b510759SStefano Zampini } 6102b510759SStefano Zampini 6112b510759SStefano Zampini #undef __FUNCT__ 6122b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 6134fad6a16SStefano Zampini /*@ 61428509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 6154fad6a16SStefano Zampini 6164fad6a16SStefano Zampini Logically collective on PC 6174fad6a16SStefano Zampini 6184fad6a16SStefano Zampini Input Parameters: 6194fad6a16SStefano Zampini + pc - the preconditioning context 62028509bceSStefano Zampini - levels - the maximum number of levels (max 9) 6214fad6a16SStefano Zampini 6220f202f7eSStefano Zampini Options Database Keys: 6230f202f7eSStefano Zampini . -pc_bddc_levels 6244fad6a16SStefano Zampini 6254fad6a16SStefano Zampini Level: intermediate 6264fad6a16SStefano Zampini 6274fad6a16SStefano Zampini Notes: 6280f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 6294fad6a16SStefano Zampini 6300f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 6314fad6a16SStefano Zampini @*/ 6322b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 6334fad6a16SStefano Zampini { 6344fad6a16SStefano Zampini PetscErrorCode ierr; 6354fad6a16SStefano Zampini 6364fad6a16SStefano Zampini PetscFunctionBegin; 6374fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6382b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 639312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 6402b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 6414fad6a16SStefano Zampini PetscFunctionReturn(0); 6424fad6a16SStefano Zampini } 6431e6b0712SBarry Smith 6444fad6a16SStefano Zampini #undef __FUNCT__ 6453b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 6463b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 6473b03a366Sstefano_zampini { 6483b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 64956282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6503b03a366Sstefano_zampini PetscErrorCode ierr; 6513b03a366Sstefano_zampini 6523b03a366Sstefano_zampini PetscFunctionBegin; 65356282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 65456282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 65556282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 65656282151SStefano Zampini } 657785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 658785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6593b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 66036e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 66156282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6623b03a366Sstefano_zampini PetscFunctionReturn(0); 6633b03a366Sstefano_zampini } 6641e6b0712SBarry Smith 6653b03a366Sstefano_zampini #undef __FUNCT__ 6663b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 6673b03a366Sstefano_zampini /*@ 66828509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 6693b03a366Sstefano_zampini 670785d1243SStefano Zampini Collective 6713b03a366Sstefano_zampini 6723b03a366Sstefano_zampini Input Parameters: 6733b03a366Sstefano_zampini + pc - the preconditioning context 674785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 6753b03a366Sstefano_zampini 6763b03a366Sstefano_zampini Level: intermediate 6773b03a366Sstefano_zampini 6780f202f7eSStefano Zampini Notes: 6790f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 6803b03a366Sstefano_zampini 6810f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 6823b03a366Sstefano_zampini @*/ 6833b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 6843b03a366Sstefano_zampini { 6853b03a366Sstefano_zampini PetscErrorCode ierr; 6863b03a366Sstefano_zampini 6873b03a366Sstefano_zampini PetscFunctionBegin; 6883b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 689674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 690785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 6913b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6923b03a366Sstefano_zampini PetscFunctionReturn(0); 6933b03a366Sstefano_zampini } 6941e6b0712SBarry Smith 6953b03a366Sstefano_zampini #undef __FUNCT__ 69682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 69782ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 6980c7d97c5SJed Brown { 6990c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 70056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7010c7d97c5SJed Brown PetscErrorCode ierr; 7020c7d97c5SJed Brown 7030c7d97c5SJed Brown PetscFunctionBegin; 70456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 70556282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 70656282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 70756282151SStefano Zampini } 708785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 709785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 7100c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 711785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 71256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7130c7d97c5SJed Brown PetscFunctionReturn(0); 7140c7d97c5SJed Brown } 7150c7d97c5SJed Brown 7160c7d97c5SJed Brown #undef __FUNCT__ 71782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 7189c0446d6SStefano Zampini /*@ 71982ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 7209c0446d6SStefano Zampini 721785d1243SStefano Zampini Collective 72253cdbc3dSStefano Zampini 72353cdbc3dSStefano Zampini Input Parameters: 72453cdbc3dSStefano Zampini + pc - the preconditioning context 72582ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 72653cdbc3dSStefano Zampini 72753cdbc3dSStefano Zampini Level: intermediate 72853cdbc3dSStefano Zampini 7299c0446d6SStefano Zampini Notes: 73053cdbc3dSStefano Zampini 7310f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 73253cdbc3dSStefano Zampini @*/ 73382ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 7340c7d97c5SJed Brown { 7350c7d97c5SJed Brown PetscErrorCode ierr; 7360c7d97c5SJed Brown 7370c7d97c5SJed Brown PetscFunctionBegin; 7380c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7390c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 74082ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 74182ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 7420c7d97c5SJed Brown PetscFunctionReturn(0); 7430c7d97c5SJed Brown } 7440c7d97c5SJed Brown 7450c7d97c5SJed Brown #undef __FUNCT__ 7460c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 74753cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 7480c7d97c5SJed Brown { 7490c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 75056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 75153cdbc3dSStefano Zampini PetscErrorCode ierr; 7520c7d97c5SJed Brown 7530c7d97c5SJed Brown PetscFunctionBegin; 75456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 75556282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 75656282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 75756282151SStefano Zampini } 758785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 759785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 76053cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 76136e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 76256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7630c7d97c5SJed Brown PetscFunctionReturn(0); 7640c7d97c5SJed Brown } 7651e6b0712SBarry Smith 7660c7d97c5SJed Brown #undef __FUNCT__ 7670c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 76857527edcSJed Brown /*@ 76928509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 77057527edcSJed Brown 771785d1243SStefano Zampini Collective 77257527edcSJed Brown 77357527edcSJed Brown Input Parameters: 77457527edcSJed Brown + pc - the preconditioning context 775785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 77657527edcSJed Brown 77757527edcSJed Brown Level: intermediate 77857527edcSJed Brown 7790f202f7eSStefano Zampini Notes: 7800f202f7eSStefano Zampini Any process can list any global node 78157527edcSJed Brown 7820f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 78357527edcSJed Brown @*/ 78453cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 7850c7d97c5SJed Brown { 7860c7d97c5SJed Brown PetscErrorCode ierr; 7870c7d97c5SJed Brown 7880c7d97c5SJed Brown PetscFunctionBegin; 7890c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 790674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 791785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 79253cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 79353cdbc3dSStefano Zampini PetscFunctionReturn(0); 79453cdbc3dSStefano Zampini } 7951e6b0712SBarry Smith 79653cdbc3dSStefano Zampini #undef __FUNCT__ 79782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 79882ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 7990c7d97c5SJed Brown { 8000c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 80156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 8020c7d97c5SJed Brown PetscErrorCode ierr; 8030c7d97c5SJed Brown 8040c7d97c5SJed Brown PetscFunctionBegin; 80556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 80656282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 80756282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 80856282151SStefano Zampini } 809785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 810785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 8110c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 812785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 81356282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 8140c7d97c5SJed Brown PetscFunctionReturn(0); 8150c7d97c5SJed Brown } 8160c7d97c5SJed Brown 8170c7d97c5SJed Brown #undef __FUNCT__ 81882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 8190c7d97c5SJed Brown /*@ 82082ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 8210c7d97c5SJed Brown 822785d1243SStefano Zampini Collective 8230c7d97c5SJed Brown 8240c7d97c5SJed Brown Input Parameters: 8250c7d97c5SJed Brown + pc - the preconditioning context 82682ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 8270c7d97c5SJed Brown 8280c7d97c5SJed Brown Level: intermediate 8290c7d97c5SJed Brown 8300c7d97c5SJed Brown Notes: 8310c7d97c5SJed Brown 8320f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 8330c7d97c5SJed Brown @*/ 83482ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 8350c7d97c5SJed Brown { 8360c7d97c5SJed Brown PetscErrorCode ierr; 8370c7d97c5SJed Brown 8380c7d97c5SJed Brown PetscFunctionBegin; 8390c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 8400c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 84182ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 84282ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 84353cdbc3dSStefano Zampini PetscFunctionReturn(0); 84453cdbc3dSStefano Zampini } 84553cdbc3dSStefano Zampini 84653cdbc3dSStefano Zampini #undef __FUNCT__ 847da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 848da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 849da1bb401SStefano Zampini { 850da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 851da1bb401SStefano Zampini 852da1bb401SStefano Zampini PetscFunctionBegin; 853da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 854da1bb401SStefano Zampini PetscFunctionReturn(0); 855da1bb401SStefano Zampini } 8561e6b0712SBarry Smith 857da1bb401SStefano Zampini #undef __FUNCT__ 858da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 859da1bb401SStefano Zampini /*@ 860785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 861da1bb401SStefano Zampini 862785d1243SStefano Zampini Collective 863785d1243SStefano Zampini 864785d1243SStefano Zampini Input Parameters: 865785d1243SStefano Zampini . pc - the preconditioning context 866785d1243SStefano Zampini 867785d1243SStefano Zampini Output Parameters: 868785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 869785d1243SStefano Zampini 870785d1243SStefano Zampini Level: intermediate 871785d1243SStefano Zampini 8720f202f7eSStefano Zampini Notes: 8730f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 874785d1243SStefano Zampini 875785d1243SStefano Zampini .seealso: PCBDDC 876785d1243SStefano Zampini @*/ 877785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 878785d1243SStefano Zampini { 879785d1243SStefano Zampini PetscErrorCode ierr; 880785d1243SStefano Zampini 881785d1243SStefano Zampini PetscFunctionBegin; 882785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 883785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 884785d1243SStefano Zampini PetscFunctionReturn(0); 885785d1243SStefano Zampini } 886785d1243SStefano Zampini 887785d1243SStefano Zampini #undef __FUNCT__ 888785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 889785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 890785d1243SStefano Zampini { 891785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 892785d1243SStefano Zampini 893785d1243SStefano Zampini PetscFunctionBegin; 894785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 895785d1243SStefano Zampini PetscFunctionReturn(0); 896785d1243SStefano Zampini } 897785d1243SStefano Zampini 898785d1243SStefano Zampini #undef __FUNCT__ 89982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 900da1bb401SStefano Zampini /*@ 90182ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 902da1bb401SStefano Zampini 903785d1243SStefano Zampini Collective 904da1bb401SStefano Zampini 905da1bb401SStefano Zampini Input Parameters: 90628509bceSStefano Zampini . pc - the preconditioning context 907da1bb401SStefano Zampini 908da1bb401SStefano Zampini Output Parameters: 90928509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 910da1bb401SStefano Zampini 911da1bb401SStefano Zampini Level: intermediate 912da1bb401SStefano Zampini 913da1bb401SStefano Zampini Notes: 9140f202f7eSStefano 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). 9150f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 916da1bb401SStefano Zampini 917da1bb401SStefano Zampini .seealso: PCBDDC 918da1bb401SStefano Zampini @*/ 91982ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 920da1bb401SStefano Zampini { 921da1bb401SStefano Zampini PetscErrorCode ierr; 922da1bb401SStefano Zampini 923da1bb401SStefano Zampini PetscFunctionBegin; 924da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 92582ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 926da1bb401SStefano Zampini PetscFunctionReturn(0); 927da1bb401SStefano Zampini } 9281e6b0712SBarry Smith 929da1bb401SStefano Zampini #undef __FUNCT__ 93053cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 93153cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 93253cdbc3dSStefano Zampini { 93353cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 93453cdbc3dSStefano Zampini 93553cdbc3dSStefano Zampini PetscFunctionBegin; 93653cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 93753cdbc3dSStefano Zampini PetscFunctionReturn(0); 93853cdbc3dSStefano Zampini } 9391e6b0712SBarry Smith 94053cdbc3dSStefano Zampini #undef __FUNCT__ 94153cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 94253cdbc3dSStefano Zampini /*@ 943785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 94453cdbc3dSStefano Zampini 945785d1243SStefano Zampini Collective 946785d1243SStefano Zampini 947785d1243SStefano Zampini Input Parameters: 948785d1243SStefano Zampini . pc - the preconditioning context 949785d1243SStefano Zampini 950785d1243SStefano Zampini Output Parameters: 951785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 952785d1243SStefano Zampini 953785d1243SStefano Zampini Level: intermediate 954785d1243SStefano Zampini 9550f202f7eSStefano Zampini Notes: 9560f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 957785d1243SStefano Zampini 958785d1243SStefano Zampini .seealso: PCBDDC 959785d1243SStefano Zampini @*/ 960785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 961785d1243SStefano Zampini { 962785d1243SStefano Zampini PetscErrorCode ierr; 963785d1243SStefano Zampini 964785d1243SStefano Zampini PetscFunctionBegin; 965785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 966785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 967785d1243SStefano Zampini PetscFunctionReturn(0); 968785d1243SStefano Zampini } 969785d1243SStefano Zampini 970785d1243SStefano Zampini #undef __FUNCT__ 971785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 972785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 973785d1243SStefano Zampini { 974785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 975785d1243SStefano Zampini 976785d1243SStefano Zampini PetscFunctionBegin; 977785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 978785d1243SStefano Zampini PetscFunctionReturn(0); 979785d1243SStefano Zampini } 980785d1243SStefano Zampini 981785d1243SStefano Zampini #undef __FUNCT__ 98282ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 98353cdbc3dSStefano Zampini /*@ 98482ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 98553cdbc3dSStefano Zampini 986785d1243SStefano Zampini Collective 98753cdbc3dSStefano Zampini 98853cdbc3dSStefano Zampini Input Parameters: 98928509bceSStefano Zampini . pc - the preconditioning context 99053cdbc3dSStefano Zampini 99153cdbc3dSStefano Zampini Output Parameters: 99228509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 99353cdbc3dSStefano Zampini 99453cdbc3dSStefano Zampini Level: intermediate 99553cdbc3dSStefano Zampini 99653cdbc3dSStefano Zampini Notes: 9970f202f7eSStefano 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). 9980f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 99953cdbc3dSStefano Zampini 100053cdbc3dSStefano Zampini .seealso: PCBDDC 100153cdbc3dSStefano Zampini @*/ 100282ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 100353cdbc3dSStefano Zampini { 100453cdbc3dSStefano Zampini PetscErrorCode ierr; 100553cdbc3dSStefano Zampini 100653cdbc3dSStefano Zampini PetscFunctionBegin; 100753cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 100882ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 10090c7d97c5SJed Brown PetscFunctionReturn(0); 10100c7d97c5SJed Brown } 10111e6b0712SBarry Smith 101236e030ebSStefano Zampini #undef __FUNCT__ 1013da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 10141a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 101536e030ebSStefano Zampini { 101636e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1017da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 101856282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 1019da1bb401SStefano Zampini PetscErrorCode ierr; 102036e030ebSStefano Zampini 102136e030ebSStefano Zampini PetscFunctionBegin; 10228687889aSStefano Zampini if (!nvtxs) { 102304194a47SStefano Zampini if (copymode == PETSC_OWN_POINTER) { 102404194a47SStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 102504194a47SStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 102604194a47SStefano Zampini } 10278687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 10288687889aSStefano Zampini PetscFunctionReturn(0); 10298687889aSStefano Zampini } 103066da6bd7Sstefano_zampini if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */ 103156282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 103256282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 10332d505d7fSStefano Zampini ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 10342d505d7fSStefano Zampini if (same_data) { 10352d505d7fSStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 10362d505d7fSStefano Zampini } 103756282151SStefano Zampini } 103856282151SStefano Zampini } 103956282151SStefano Zampini if (!same_data) { 1040674ae819SStefano Zampini /* free old CSR */ 1041674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 1042674ae819SStefano Zampini /* get CSR into graph structure */ 1043da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 1044854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 1045785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 1046674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 1047674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 1048a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 1049da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 10501a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 10511a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 1052a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 1053a1dbd327SStefano Zampini } else if (copymode == PETSC_USE_POINTER) { 1054a1dbd327SStefano Zampini mat_graph->xadj = (PetscInt*)xadj; 1055a1dbd327SStefano Zampini mat_graph->adjncy = (PetscInt*)adjncy; 1056a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_FALSE; 1057a1dbd327SStefano Zampini } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode); 1058575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 105956282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 106056282151SStefano Zampini } 106136e030ebSStefano Zampini PetscFunctionReturn(0); 106236e030ebSStefano Zampini } 10631e6b0712SBarry Smith 106436e030ebSStefano Zampini #undef __FUNCT__ 1065da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 106636e030ebSStefano Zampini /*@ 106754fffbccSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom. 106836e030ebSStefano Zampini 106936e030ebSStefano Zampini Not collective 107036e030ebSStefano Zampini 107136e030ebSStefano Zampini Input Parameters: 107254fffbccSStefano Zampini + pc - the preconditioning context. 107354fffbccSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the number of local dofs). 107454fffbccSStefano Zampini . xadj, adjncy - the connectivity of the dofs in CSR format. 107554fffbccSStefano Zampini - copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER. 107636e030ebSStefano Zampini 107736e030ebSStefano Zampini Level: intermediate 107836e030ebSStefano Zampini 107954fffbccSStefano Zampini Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative. 108036e030ebSStefano Zampini 108128509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 108236e030ebSStefano Zampini @*/ 10831a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 108436e030ebSStefano Zampini { 1085575ad6abSStefano Zampini void (*f)(void) = 0; 108636e030ebSStefano Zampini PetscErrorCode ierr; 108736e030ebSStefano Zampini 108836e030ebSStefano Zampini PetscFunctionBegin; 108936e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 10908687889aSStefano Zampini if (nvtxs) { 1091674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 10921633d1f0SStefano Zampini if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4); 10938687889aSStefano Zampini } 10941a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 1095575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 1096575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 1097575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 1098575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 1099575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 1100da1bb401SStefano Zampini } 110136e030ebSStefano Zampini PetscFunctionReturn(0); 110236e030ebSStefano Zampini } 11031e6b0712SBarry Smith 11049c0446d6SStefano Zampini #undef __FUNCT__ 110563602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 110663602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 110763602bcaSStefano Zampini { 110863602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 110963602bcaSStefano Zampini PetscInt i; 111056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 111163602bcaSStefano Zampini PetscErrorCode ierr; 111263602bcaSStefano Zampini 111363602bcaSStefano Zampini PetscFunctionBegin; 111456282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 111556282151SStefano Zampini for (i=0;i<n_is;i++) { 111656282151SStefano Zampini PetscBool isequalt; 111756282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 111856282151SStefano Zampini if (!isequalt) break; 111956282151SStefano Zampini } 112056282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 112156282151SStefano Zampini } 112256282151SStefano Zampini for (i=0;i<n_is;i++) { 112356282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 112456282151SStefano Zampini } 112563602bcaSStefano Zampini /* Destroy ISes if they were already set */ 112663602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 112763602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 112863602bcaSStefano Zampini } 112963602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 113063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 113163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 113263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 113363602bcaSStefano Zampini } 113463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 113563602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 113663602bcaSStefano Zampini /* allocate space then set */ 1137d02579f5SStefano Zampini if (n_is) { 1138d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 1139d02579f5SStefano Zampini } 114063602bcaSStefano Zampini for (i=0;i<n_is;i++) { 114163602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 114263602bcaSStefano Zampini } 114363602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 114463602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 114556282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 114663602bcaSStefano Zampini PetscFunctionReturn(0); 114763602bcaSStefano Zampini } 114863602bcaSStefano Zampini 114963602bcaSStefano Zampini #undef __FUNCT__ 115063602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 115163602bcaSStefano Zampini /*@ 115263602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 115363602bcaSStefano Zampini 115463602bcaSStefano Zampini Collective 115563602bcaSStefano Zampini 115663602bcaSStefano Zampini Input Parameters: 115763602bcaSStefano Zampini + pc - the preconditioning context 11580f202f7eSStefano Zampini . n_is - number of index sets defining the fields 11590f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 116063602bcaSStefano Zampini 116163602bcaSStefano Zampini Level: intermediate 116263602bcaSStefano Zampini 11630f202f7eSStefano Zampini Notes: 11640f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 116563602bcaSStefano Zampini 116663602bcaSStefano Zampini .seealso: PCBDDC 116763602bcaSStefano Zampini @*/ 116863602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 116963602bcaSStefano Zampini { 117063602bcaSStefano Zampini PetscInt i; 117163602bcaSStefano Zampini PetscErrorCode ierr; 117263602bcaSStefano Zampini 117363602bcaSStefano Zampini PetscFunctionBegin; 117463602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 117563602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 117663602bcaSStefano Zampini for (i=0;i<n_is;i++) { 117763602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 117863602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 117963602bcaSStefano Zampini } 1180e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 118163602bcaSStefano Zampini PetscFunctionReturn(0); 118263602bcaSStefano Zampini } 118363602bcaSStefano Zampini 118463602bcaSStefano Zampini #undef __FUNCT__ 11859c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 11869c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 11879c0446d6SStefano Zampini { 11889c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 11899c0446d6SStefano Zampini PetscInt i; 119056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 11919c0446d6SStefano Zampini PetscErrorCode ierr; 11929c0446d6SStefano Zampini 11939c0446d6SStefano Zampini PetscFunctionBegin; 119456282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 119556282151SStefano Zampini for (i=0;i<n_is;i++) { 119656282151SStefano Zampini PetscBool isequalt; 119756282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 119856282151SStefano Zampini if (!isequalt) break; 119956282151SStefano Zampini } 120056282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 120156282151SStefano Zampini } 120256282151SStefano Zampini for (i=0;i<n_is;i++) { 120356282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 120456282151SStefano Zampini } 1205da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 12069c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 12079c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 12089c0446d6SStefano Zampini } 1209d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 121063602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 121163602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 121263602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 121363602bcaSStefano Zampini } 121463602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 121563602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1216da1bb401SStefano Zampini /* allocate space then set */ 1217d02579f5SStefano Zampini if (n_is) { 1218785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1219d02579f5SStefano Zampini } 12209c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1221da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 12229c0446d6SStefano Zampini } 12239c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 122463602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 122556282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 12269c0446d6SStefano Zampini PetscFunctionReturn(0); 12279c0446d6SStefano Zampini } 12281e6b0712SBarry Smith 12299c0446d6SStefano Zampini #undef __FUNCT__ 12309c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 12319c0446d6SStefano Zampini /*@ 123263602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 12339c0446d6SStefano Zampini 123463602bcaSStefano Zampini Collective 12359c0446d6SStefano Zampini 12369c0446d6SStefano Zampini Input Parameters: 12379c0446d6SStefano Zampini + pc - the preconditioning context 12380f202f7eSStefano Zampini . n_is - number of index sets defining the fields 12390f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 12409c0446d6SStefano Zampini 12419c0446d6SStefano Zampini Level: intermediate 12429c0446d6SStefano Zampini 12430f202f7eSStefano Zampini Notes: 12440f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 12459c0446d6SStefano Zampini 12469c0446d6SStefano Zampini .seealso: PCBDDC 12479c0446d6SStefano Zampini @*/ 12489c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 12499c0446d6SStefano Zampini { 12502b510759SStefano Zampini PetscInt i; 12519c0446d6SStefano Zampini PetscErrorCode ierr; 12529c0446d6SStefano Zampini 12539c0446d6SStefano Zampini PetscFunctionBegin; 12549c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 125563602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 12562b510759SStefano Zampini for (i=0;i<n_is;i++) { 125763602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 1258a011d5a7Sstefano_zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 12592b510759SStefano Zampini } 12609c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 12619c0446d6SStefano Zampini PetscFunctionReturn(0); 12629c0446d6SStefano Zampini } 1263906d46d4SStefano Zampini 1264534831adSStefano Zampini #undef __FUNCT__ 1265534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 1266534831adSStefano Zampini /* 1267534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1268534831adSStefano Zampini guess if a transformation of basis approach has been selected. 12699c0446d6SStefano Zampini 1270534831adSStefano Zampini Input Parameter: 1271534831adSStefano Zampini + pc - the preconditioner contex 1272534831adSStefano Zampini 1273534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1274534831adSStefano Zampini 1275534831adSStefano Zampini Notes: 1276534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1277534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1278534831adSStefano Zampini */ 1279534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1280534831adSStefano Zampini { 1281534831adSStefano Zampini PetscErrorCode ierr; 1282534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1283534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 12843972b0daSStefano Zampini Vec used_vec; 128527b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1286534831adSStefano Zampini 1287534831adSStefano Zampini PetscFunctionBegin; 12881f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 128985c4d303SStefano Zampini if (ksp) { 12901f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 129185c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 129227b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 129327b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1294f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 12951f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 129685c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 129785c4d303SStefano Zampini } 129885c4d303SStefano Zampini } 1299fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1300fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1301fc17d649SStefano Zampini } 13021f4df5f7SStefano Zampini 130385c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 130462a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 130562a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 130662a6ff1dSStefano Zampini } 130762a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 130862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 130962a6ff1dSStefano Zampini } 13108d00608fSStefano Zampini 131127b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 13123972b0daSStefano Zampini if (x) { 13133972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 13143972b0daSStefano Zampini used_vec = x; 13158d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 13163972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 13173972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 13183972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 131927b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1320266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1321266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1322266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 13233972b0daSStefano Zampini } 13248efcfb23SStefano Zampini 13258efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 13263972b0daSStefano Zampini if (ksp) { 1327a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 13288efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 13298efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 13303972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 13313972b0daSStefano Zampini } 13323972b0daSStefano Zampini } 13333308cffdSStefano Zampini 13348d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 13353972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 133670c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 13373975b054SStefano Zampini IS dirIS = NULL; 13383975b054SStefano Zampini 1339a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 13403975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 13413975b054SStefano Zampini if (dirIS) { 1342906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1343785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 13442b095fd8SStefano Zampini PetscScalar *array_x; 13452b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1346785d1243SStefano Zampini 13473972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 13483972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1349e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1350e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1351e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1352e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 135382ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 13543972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 13552b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 13563972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 13572fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 13583972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 13592b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 13603972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1361e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1362e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13638d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13641b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 13658efcfb23SStefano Zampini } 1366a07ea27aSStefano Zampini } 1367b76ba322SStefano Zampini 13688efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 13698d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 137027b6a85dSStefano Zampini /* save the original rhs */ 137127b6a85dSStefano Zampini if (save_rhs) { 137227b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 137327b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 13748d00608fSStefano Zampini } 13758d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13763972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 137727b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 13783972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 13798efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 138027b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 13817acc28cbSStefano Zampini if (ksp) { 13827acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 13837acc28cbSStefano Zampini } 13843308cffdSStefano Zampini } 13858efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1386b76ba322SStefano Zampini 1387fc17d649SStefano Zampini /* compute initial vector in benign space if needed 138827b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 138927b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 139027b6a85dSStefano Zampini if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) { 13911f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 13921f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 13930369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 13940369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 13950369aaf7SStefano Zampini } 13964fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 139727b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 13981dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 13993bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 14001f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 14011f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14021f4df5f7SStefano Zampini } 14031f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 140427b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 140527b6a85dSStefano Zampini if (save_rhs) { 140627b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 140792e3dcfbSStefano Zampini } 140827b6a85dSStefano Zampini if (pcbddc->rhs_change) { 14090369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 141027b6a85dSStefano Zampini } else { 141127b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 141227b6a85dSStefano Zampini } 14130369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 141427b6a85dSStefano Zampini } 141527b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 14160369aaf7SStefano Zampini } 14172d4c4fecSStefano Zampini 14182d4c4fecSStefano Zampini /* dbg output */ 1419a198735bSStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 14201f4df5f7SStefano Zampini Vec v; 14211f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 14221f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 14231f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1424a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %d: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr); 14252d4c4fecSStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr); 14261f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 14271f4df5f7SStefano Zampini } 14280369aaf7SStefano Zampini 14290369aaf7SStefano Zampini /* set initial guess if using PCG */ 14308ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 14310369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 14320369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 14331dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 143427b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 14351dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 14361dd7afcfSStefano Zampini } else { 14371dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 14381dd7afcfSStefano Zampini } 14391dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14401dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14411dd7afcfSStefano Zampini } else { 14420369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14430369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14441dd7afcfSStefano Zampini } 14450369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 14461dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 14471dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 14481dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14491dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14501dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 14511dd7afcfSStefano Zampini } else { 14520369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14530369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14541dd7afcfSStefano Zampini } 14550369aaf7SStefano Zampini if (ksp) { 14560369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 14570369aaf7SStefano Zampini } 14588ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_TRUE; 1459266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1460266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 14610369aaf7SStefano Zampini } 1462534831adSStefano Zampini PetscFunctionReturn(0); 1463534831adSStefano Zampini } 1464906d46d4SStefano Zampini 1465534831adSStefano Zampini #undef __FUNCT__ 1466534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1467534831adSStefano Zampini /* 1468534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1469534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1470534831adSStefano Zampini 1471534831adSStefano Zampini Input Parameter: 1472534831adSStefano Zampini + pc - the preconditioner contex 1473534831adSStefano Zampini 1474534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1475534831adSStefano Zampini 1476534831adSStefano Zampini Notes: 1477534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1478534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1479534831adSStefano Zampini */ 1480534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1481534831adSStefano Zampini { 1482534831adSStefano Zampini PetscErrorCode ierr; 1483534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1484534831adSStefano Zampini 1485534831adSStefano Zampini PetscFunctionBegin; 14863972b0daSStefano Zampini /* add solution removed in presolve */ 14876bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 148827b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 14893425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1490af140850Sstefano_zampini } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) { 149127b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14923425bc38SStefano Zampini } 1493af140850Sstefano_zampini /* restore to original state (not for FETI-DP) */ 1494af140850Sstefano_zampini if (ksp) pcbddc->temp_solution_used = PETSC_FALSE; 149527b6a85dSStefano Zampini } 149627b6a85dSStefano Zampini 1497266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 14988d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 149927b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 15008d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 1501af140850Sstefano_zampini } 15028efcfb23SStefano Zampini /* restore ksp guess state */ 15038efcfb23SStefano Zampini if (ksp) { 15048efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 15058ae0ca82SStefano Zampini /* reset flag for exact dirichlet trick */ 15068ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 1507af140850Sstefano_zampini } 1508534831adSStefano Zampini PetscFunctionReturn(0); 1509534831adSStefano Zampini } 1510af140850Sstefano_zampini 151153cdbc3dSStefano Zampini #undef __FUNCT__ 151253cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 15130c7d97c5SJed Brown /* 15140c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 15150c7d97c5SJed Brown by setting data structures and options. 15160c7d97c5SJed Brown 15170c7d97c5SJed Brown Input Parameter: 151853cdbc3dSStefano Zampini + pc - the preconditioner context 15190c7d97c5SJed Brown 15200c7d97c5SJed Brown Application Interface Routine: PCSetUp() 15210c7d97c5SJed Brown 15220c7d97c5SJed Brown Notes: 15230c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 15240c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 15250c7d97c5SJed Brown */ 152653cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 15270c7d97c5SJed Brown { 15280c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1529c703fcc7SStefano Zampini PCBDDCSubSchurs sub_schurs; 15305e8657edSStefano Zampini Mat_IS* matis; 153108122e43SStefano Zampini MatNullSpace nearnullspace; 153235509ce9Sstefano_zampini Mat lA; 153335509ce9Sstefano_zampini IS lP,zerodiag = NULL; 153491e8d312SStefano Zampini PetscInt nrows,ncols; 1535c703fcc7SStefano Zampini PetscBool computesubschurs; 15368de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 15375e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 1538c703fcc7SStefano Zampini PetscErrorCode ierr; 15390c7d97c5SJed Brown 15400c7d97c5SJed Brown PetscFunctionBegin; 15415e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 15426c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 154391e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 15446c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 15455e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1546f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 15473b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 154871582508SStefano Zampini Also, BDDC builds its own KSP for the Dirichlet problem */ 1549c703fcc7SStefano Zampini if (pc->setupcalled && pc->flag != SAME_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE; 1550c83e1ba7SStefano Zampini if (pcbddc->recompute_topography) { 1551c83e1ba7SStefano Zampini pcbddc->graphanalyzed = PETSC_FALSE; 1552c83e1ba7SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1553c83e1ba7SStefano Zampini } else { 15548de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1555c83e1ba7SStefano Zampini } 1556b087196eSStefano Zampini 1557b087196eSStefano Zampini /* check parameters' compatibility */ 1558b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 15599d54b7f4SStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0); 1560bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1561862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1562862806e4SStefano Zampini 15635a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 156416909a7fSStefano Zampini if (pcbddc->switch_static) { 156516909a7fSStefano Zampini PetscBool ismatis; 156616909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 156716909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 156816909a7fSStefano Zampini } 156916909a7fSStefano Zampini 157071582508SStefano Zampini /* activate all connected components if the netflux has been requested */ 1571bb05f991SStefano Zampini if (pcbddc->compute_nonetflux) { 1572bb05f991SStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 1573bb05f991SStefano Zampini pcbddc->use_edges = PETSC_TRUE; 1574bb05f991SStefano Zampini pcbddc->use_faces = PETSC_TRUE; 1575bb05f991SStefano Zampini } 1576bb05f991SStefano Zampini 1577f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 157870cf5478SStefano Zampini if (pcbddc->dbg_flag) { 157970cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 158058a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 15811575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1582f4ddd8eeSStefano Zampini } 158358a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1584f4ddd8eeSStefano Zampini } 1585f4ddd8eeSStefano Zampini 1586c703fcc7SStefano Zampini /* process topology information */ 158771582508SStefano Zampini if (pcbddc->recompute_topography) { 15885b003df0Sstefano_zampini PetscContainer c; 15895b003df0Sstefano_zampini 15905b003df0Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc->pmat,"_convert_nest_lfields",(PetscObject*)&c);CHKERRQ(ierr); 15915b003df0Sstefano_zampini if (c) { 15925b003df0Sstefano_zampini MatISLocalFields lf; 15935b003df0Sstefano_zampini ierr = PetscContainerGetPointer(c,(void**)&lf);CHKERRQ(ierr); 15945b003df0Sstefano_zampini ierr = PCBDDCSetDofsSplittingLocal(pc,lf->nr,lf->rf);CHKERRQ(ierr); 15955b003df0Sstefano_zampini } 159671582508SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 1597c83e1ba7SStefano Zampini /* detect local disconnected subdomains if requested (use matis->A) */ 1598c83e1ba7SStefano Zampini if (pcbddc->detect_disconnected) { 1599c83e1ba7SStefano Zampini PetscInt i; 1600c83e1ba7SStefano Zampini for (i=0;i<pcbddc->n_local_subs;i++) { 1601c83e1ba7SStefano Zampini ierr = ISDestroy(&pcbddc->local_subs[i]);CHKERRQ(ierr); 1602c83e1ba7SStefano Zampini } 1603c83e1ba7SStefano Zampini ierr = PetscFree(pcbddc->local_subs);CHKERRQ(ierr); 1604c83e1ba7SStefano Zampini ierr = MatDetectDisconnectedComponents(matis->A,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 1605c83e1ba7SStefano Zampini } 1606c703fcc7SStefano Zampini if (pcbddc->discretegradient) { 1607a13144ffSStefano Zampini ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr); 1608a13144ffSStefano Zampini } 1609c703fcc7SStefano Zampini } 1610a13144ffSStefano Zampini 1611c703fcc7SStefano Zampini /* change basis if requested by the user */ 16125e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 16135e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 16145e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 16155e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 16165e8657edSStefano Zampini } else { 1617b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16185e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16195e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1620d16cbb6bSStefano Zampini } 1621d16cbb6bSStefano Zampini 16224f1b2e48SStefano Zampini /* 1623c703fcc7SStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick 16244f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 16254f1b2e48SStefano Zampini */ 16261dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1627d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 16289f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 16299f47a83aSStefano Zampini 163005b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1631669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 1632339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1633a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1634c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 16351dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 16361dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 16371dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 16381dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16391dd7afcfSStefano Zampini } else { 1640a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 16411dd7afcfSStefano Zampini } 1642a3df083aSStefano Zampini } else { 16439f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1644674ae819SStefano Zampini } 1645a3df083aSStefano Zampini } 164627b6a85dSStefano Zampini 1647c703fcc7SStefano Zampini /* propagate relevant information -> TODO remove*/ 16484f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 16493301b35fSStefano Zampini if (matis->A->symmetric_set) { 16503301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 16513301b35fSStefano Zampini } 1652e496cd5dSStefano Zampini #endif 165306a4e24aSStefano Zampini if (matis->A->symmetric_set) { 165406a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 165506a4e24aSStefano Zampini } 165606a4e24aSStefano Zampini if (matis->A->spd_set) { 165706a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 165806a4e24aSStefano Zampini } 1659e496cd5dSStefano Zampini 16605e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 16615e8657edSStefano Zampini { 16625e8657edSStefano Zampini Mat temp_mat; 16635e8657edSStefano Zampini 16645e8657edSStefano Zampini temp_mat = matis->A; 16655e8657edSStefano Zampini matis->A = pcbddc->local_mat; 16665e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 16675e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16685e8657edSStefano Zampini matis->A = temp_mat; 16695e8657edSStefano Zampini } 1670684f6988SStefano Zampini 167181d14e9dSStefano Zampini /* Analyze interface */ 167264ac59b8SStefano Zampini if (!pcbddc->graphanalyzed) { 1673674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 16748de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1675345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 1676345ecf6cSStefano Zampini SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1677345ecf6cSStefano Zampini } 1678a198735bSStefano Zampini if (pcbddc->compute_nonetflux) { 1679669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1680669cc0f4SStefano Zampini 16818ae0ca82SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 168271582508SStefano Zampini /* TODO what if a nearnullspace is already attached? */ 1683669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1684669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1685669cc0f4SStefano Zampini } 1686674ae819SStefano Zampini } 1687fb8d54d4SStefano Zampini 16885408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 16895408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 16905408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 1691ff1f7e73Sstefano_zampini if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) { 16925408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 169309f581a4SStefano Zampini } 16944f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 169506f24817SStefano Zampini 1696b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1697c703fcc7SStefano Zampini if (computesubschurs && pcbddc->recompute_topography) { 169808122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1699b1b3d7a2SStefano Zampini } 17009d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 17019d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 17029d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 17039d54b7f4SStefano Zampini } 1704c703fcc7SStefano Zampini 1705c703fcc7SStefano Zampini /* finish setup solvers and do adaptive selection of constraints */ 1706b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1707b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 17082070dbb6SStefano Zampini if (computesubschurs) { 170908122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 17102070dbb6SStefano Zampini } 1711d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1712d5574798SStefano Zampini } else { 1713d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 17142070dbb6SStefano Zampini if (computesubschurs) { 1715d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1716d5574798SStefano Zampini } 17172070dbb6SStefano Zampini } 171808122e43SStefano Zampini if (pcbddc->adaptive_selection) { 171908122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 17208de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1721b7eb3628SStefano Zampini } 1722684f6988SStefano Zampini 1723f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1724fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1725f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1726f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1727f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1728f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1729f4ddd8eeSStefano Zampini } else { 1730f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1731f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1732f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1733165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1734f4ddd8eeSStefano Zampini PetscInt i; 1735165b64e2SStefano Zampini const Vec *nearnullvecs; 1736165b64e2SStefano Zampini PetscObjectState state; 1737165b64e2SStefano Zampini PetscInt nnsp_size; 1738165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1739f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1740f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1741165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1742f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1743f4ddd8eeSStefano Zampini break; 1744f4ddd8eeSStefano Zampini } 1745f4ddd8eeSStefano Zampini } 1746f4ddd8eeSStefano Zampini } 1747f4ddd8eeSStefano Zampini } 1748f4ddd8eeSStefano Zampini } else { 1749f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1750f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1751f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1752f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1753f4ddd8eeSStefano Zampini } 1754f4ddd8eeSStefano Zampini } 1755f4ddd8eeSStefano Zampini 1756f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1757727cdba6SStefano Zampini /* reset primal space flags */ 1758f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1759727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 17608de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1761727cdba6SStefano Zampini /* It also sets the primal space flags */ 1762674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 17639543d0ffSStefano Zampini } 1764e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1765f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 17665e8657edSStefano Zampini 17675e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 17685e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 17695e8657edSStefano Zampini 17705e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 17714f1b2e48SStefano Zampini if (pcbddc->benign_change) { 17721dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1773c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1774c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1775c263805aSStefano Zampini } 17765e8657edSStefano Zampini /* get submatrices */ 17775e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 17785e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 17795e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 17805e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 17815e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 17825e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 17833975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 17843975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 17859c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1786b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 17875e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 17885e8657edSStefano Zampini pcbddc->local_mat = matis->A; 17895e8657edSStefano Zampini } 179035509ce9Sstefano_zampini 179135509ce9Sstefano_zampini /* interface pressure block row for B_C */ 179235509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr); 179335509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr); 179435509ce9Sstefano_zampini if (lA && lP) { 179535509ce9Sstefano_zampini PC_IS* pcis = (PC_IS*)pc->data; 179635509ce9Sstefano_zampini Mat B_BI,B_BB,Bt_BI,Bt_BB; 179735509ce9Sstefano_zampini PetscBool issym; 179835509ce9Sstefano_zampini ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr); 17996cc1294bSstefano_zampini if (issym) { 180035509ce9Sstefano_zampini ierr = MatGetSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 180135509ce9Sstefano_zampini ierr = MatGetSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 180235509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr); 180335509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr); 180435509ce9Sstefano_zampini } else { 18056cc1294bSstefano_zampini ierr = MatGetSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 18066cc1294bSstefano_zampini ierr = MatGetSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 180735509ce9Sstefano_zampini ierr = MatGetSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr); 180835509ce9Sstefano_zampini ierr = MatGetSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr); 180935509ce9Sstefano_zampini } 181035509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr); 181135509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr); 181235509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr); 181335509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr); 181435509ce9Sstefano_zampini ierr = MatDestroy(&B_BI);CHKERRQ(ierr); 181535509ce9Sstefano_zampini ierr = MatDestroy(&B_BB);CHKERRQ(ierr); 181635509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr); 181735509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr); 181835509ce9Sstefano_zampini } 181935509ce9Sstefano_zampini 1820b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 182199cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1822b96c3477SStefano Zampini /* SetUp Scaling operator */ 18239d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1824674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 18250c7d97c5SJed Brown } 1826c703fcc7SStefano Zampini 18271dd7afcfSStefano Zampini /* mark topography as done */ 182856282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 18290369aaf7SStefano Zampini 18301dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 18311dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 18321dd7afcfSStefano Zampini 183358a03d70SStefano Zampini if (pcbddc->dbg_flag) { 183458a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 18352b510759SStefano Zampini } 18360c7d97c5SJed Brown PetscFunctionReturn(0); 18370c7d97c5SJed Brown } 18380c7d97c5SJed Brown 18390c7d97c5SJed Brown /* 184050efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 18410c7d97c5SJed Brown 18420c7d97c5SJed Brown Input Parameters: 18430f202f7eSStefano Zampini + pc - the preconditioner context 18440f202f7eSStefano Zampini - r - input vector (global) 18450c7d97c5SJed Brown 18460c7d97c5SJed Brown Output Parameter: 18470c7d97c5SJed Brown . z - output vector (global) 18480c7d97c5SJed Brown 18490c7d97c5SJed Brown Application Interface Routine: PCApply() 18500c7d97c5SJed Brown */ 18510c7d97c5SJed Brown #undef __FUNCT__ 18520c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 185353cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 18540c7d97c5SJed Brown { 18550c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 18560c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1857b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 18580c7d97c5SJed Brown PetscErrorCode ierr; 18593b03a366Sstefano_zampini const PetscScalar one = 1.0; 18603b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 18612617d88aSStefano Zampini const PetscScalar zero = 0.0; 18620c7d97c5SJed Brown 18630c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 18640c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1865b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 18660c7d97c5SJed Brown 18670c7d97c5SJed Brown PetscFunctionBegin; 1868f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 18691dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18701dd7afcfSStefano Zampini Vec swap; 187127b6a85dSStefano Zampini 187227b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 18731dd7afcfSStefano Zampini swap = pcbddc->work_change; 18741dd7afcfSStefano Zampini pcbddc->work_change = r; 18751dd7afcfSStefano Zampini r = swap; 18761dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 18779cc2a9b1Sstefano_zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 18781dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 18791dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 18801dd7afcfSStefano Zampini } 18811dd7afcfSStefano Zampini } 188227b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1883015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1884efc2fbd9SStefano Zampini } 18858ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1886b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 18870c7d97c5SJed Brown /* First Dirichlet solve */ 18880c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18890c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18900c7d97c5SJed Brown /* 18910c7d97c5SJed Brown Assembling right hand side for BDDC operator 1892b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1893674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 18940c7d97c5SJed Brown */ 1895b097fa66SStefano Zampini if (n_D) { 1896b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 18970c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 189816909a7fSStefano Zampini if (pcbddc->switch_static) { 189916909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 190016909a7fSStefano Zampini 190116909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 190216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190416909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 190516909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 190616909a7fSStefano Zampini } else { 190716909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 190816909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 190916909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 191016909a7fSStefano Zampini } 191116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191516909a7fSStefano Zampini } else { 1916b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 191716909a7fSStefano Zampini } 1918b097fa66SStefano Zampini } else { 1919b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1920b097fa66SStefano Zampini } 19210c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 19220c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1923674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1924b76ba322SStefano Zampini } else { 19254fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1926674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1927b76ba322SStefano Zampini } 19284fee134fSStefano Zampini } 1929b76ba322SStefano Zampini 19302617d88aSStefano Zampini /* Apply interface preconditioner 19312617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1932dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 19332617d88aSStefano Zampini 1934674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1935674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 19360c7d97c5SJed Brown 19373b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 19380c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 19390c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1940b097fa66SStefano Zampini if (n_B) { 194116909a7fSStefano Zampini if (pcbddc->switch_static) { 194216909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 194316909a7fSStefano Zampini 194416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 194516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 194616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 194716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 194816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 194916909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 195016909a7fSStefano Zampini } else { 195116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 195216909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 195316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 195416909a7fSStefano Zampini } 195516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 195616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 195716909a7fSStefano Zampini } else { 19580c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 195916909a7fSStefano Zampini } 196016909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 196116909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 196216909a7fSStefano Zampini 196316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 196416909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 196516909a7fSStefano Zampini } else { 196616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 196716909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 196816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 196916909a7fSStefano Zampini } 1970b097fa66SStefano Zampini } 1971df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1972efc2fbd9SStefano Zampini 19738ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1974b097fa66SStefano Zampini if (pcbddc->switch_static) { 1975b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1976b097fa66SStefano Zampini } else { 1977b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1978b097fa66SStefano Zampini } 19790c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 19800c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1981b097fa66SStefano Zampini } else { 1982b097fa66SStefano Zampini if (pcbddc->switch_static) { 1983b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1984b097fa66SStefano Zampini } else { 1985b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1986b097fa66SStefano Zampini } 1987b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1988b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1989b097fa66SStefano Zampini } 199027b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 19911dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 19921dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 19931dd7afcfSStefano Zampini } 1994015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1995efc2fbd9SStefano Zampini } 19961f4df5f7SStefano Zampini 19971dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1998f913dca9SStefano Zampini pcbddc->work_change = r; 19991dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 20001dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 20011dd7afcfSStefano Zampini } 20020c7d97c5SJed Brown PetscFunctionReturn(0); 20030c7d97c5SJed Brown } 200450efa1b5SStefano Zampini 200550efa1b5SStefano Zampini /* 200650efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 200750efa1b5SStefano Zampini 200850efa1b5SStefano Zampini Input Parameters: 20090f202f7eSStefano Zampini + pc - the preconditioner context 20100f202f7eSStefano Zampini - r - input vector (global) 201150efa1b5SStefano Zampini 201250efa1b5SStefano Zampini Output Parameter: 201350efa1b5SStefano Zampini . z - output vector (global) 201450efa1b5SStefano Zampini 201550efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 201650efa1b5SStefano Zampini */ 201750efa1b5SStefano Zampini #undef __FUNCT__ 201850efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 201950efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 202050efa1b5SStefano Zampini { 202150efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 202250efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 2023b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 202450efa1b5SStefano Zampini PetscErrorCode ierr; 202550efa1b5SStefano Zampini const PetscScalar one = 1.0; 202650efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 202750efa1b5SStefano Zampini const PetscScalar zero = 0.0; 202850efa1b5SStefano Zampini 202950efa1b5SStefano Zampini PetscFunctionBegin; 2030f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 20311dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 20321dd7afcfSStefano Zampini Vec swap; 203327b6a85dSStefano Zampini 203427b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 20351dd7afcfSStefano Zampini swap = pcbddc->work_change; 20361dd7afcfSStefano Zampini pcbddc->work_change = r; 20371dd7afcfSStefano Zampini r = swap; 203827b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 20398ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 204027b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 204127b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 20421dd7afcfSStefano Zampini } 204327b6a85dSStefano Zampini } 204427b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 2045537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 2046537c1cdfSStefano Zampini } 20478ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2048b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 204950efa1b5SStefano Zampini /* First Dirichlet solve */ 205050efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 205150efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 205250efa1b5SStefano Zampini /* 205350efa1b5SStefano Zampini Assembling right hand side for BDDC operator 2054b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 205550efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 205650efa1b5SStefano Zampini */ 2057b097fa66SStefano Zampini if (n_D) { 2058b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 205950efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 206016909a7fSStefano Zampini if (pcbddc->switch_static) { 206116909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 206216909a7fSStefano Zampini 206316909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 206416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 206516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 206616909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 206716909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 206816909a7fSStefano Zampini } else { 206916909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 207016909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 207116909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 207216909a7fSStefano Zampini } 207316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 207416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 207516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 207616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 207716909a7fSStefano Zampini } else { 2078b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 207916909a7fSStefano Zampini } 2080b097fa66SStefano Zampini } else { 2081b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 2082b097fa66SStefano Zampini } 208350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 208450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 208550efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 208650efa1b5SStefano Zampini } else { 208750efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 208850efa1b5SStefano Zampini } 208950efa1b5SStefano Zampini 209050efa1b5SStefano Zampini /* Apply interface preconditioner 209150efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 2092dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 209350efa1b5SStefano Zampini 209450efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 209550efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 209650efa1b5SStefano Zampini 209750efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 209850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 209950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2100b097fa66SStefano Zampini if (n_B) { 210116909a7fSStefano Zampini if (pcbddc->switch_static) { 210216909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 210316909a7fSStefano Zampini 210416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 210516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 210616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 210716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 210816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 210916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 211016909a7fSStefano Zampini } else { 211116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 211216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 211316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 211416909a7fSStefano Zampini } 211516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 211616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 211716909a7fSStefano Zampini } else { 211850efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 211916909a7fSStefano Zampini } 212016909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 212116909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 212216909a7fSStefano Zampini 212316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 212416909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 212516909a7fSStefano Zampini } else { 212616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 212716909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 212816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 212916909a7fSStefano Zampini } 2130b097fa66SStefano Zampini } 2131b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 21328ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2133b097fa66SStefano Zampini if (pcbddc->switch_static) { 2134b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 2135b097fa66SStefano Zampini } else { 2136b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 2137b097fa66SStefano Zampini } 213850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 213950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2140b097fa66SStefano Zampini } else { 2141b097fa66SStefano Zampini if (pcbddc->switch_static) { 2142b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 2143b097fa66SStefano Zampini } else { 2144b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 2145b097fa66SStefano Zampini } 2146b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2147b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2148b097fa66SStefano Zampini } 214927b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 2150537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 2151537c1cdfSStefano Zampini } 21521dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2153f913dca9SStefano Zampini pcbddc->work_change = r; 21541dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 21551dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 21561dd7afcfSStefano Zampini } 215750efa1b5SStefano Zampini PetscFunctionReturn(0); 215850efa1b5SStefano Zampini } 2159674ae819SStefano Zampini 2160da1bb401SStefano Zampini #undef __FUNCT__ 21619326c5c6Sstefano_zampini #define __FUNCT__ "PCReset_BDDC" 21629326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc) 2163da1bb401SStefano Zampini { 2164da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 21659326c5c6Sstefano_zampini PC_IS *pcis = (PC_IS*)pc->data; 21669326c5c6Sstefano_zampini KSP kspD,kspR,kspC; 2167da1bb401SStefano Zampini PetscErrorCode ierr; 2168da1bb401SStefano Zampini 2169da1bb401SStefano Zampini PetscFunctionBegin; 2170674ae819SStefano Zampini /* free BDDC custom data */ 2171674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2172674ae819SStefano Zampini /* destroy objects related to topography */ 2173674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 217434a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2175674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 2176674ae819SStefano Zampini /* free solvers stuff */ 2177674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 217862a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 217962a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 218062a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 21811dd7afcfSStefano Zampini /* free data created by PCIS */ 21821dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 21839326c5c6Sstefano_zampini 21849326c5c6Sstefano_zampini /* restore defaults */ 21859326c5c6Sstefano_zampini kspD = pcbddc->ksp_D; 21869326c5c6Sstefano_zampini kspR = pcbddc->ksp_R; 21879326c5c6Sstefano_zampini kspC = pcbddc->coarse_ksp; 21889326c5c6Sstefano_zampini ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr); 21899326c5c6Sstefano_zampini pcis->n_neigh = -1; 21909326c5c6Sstefano_zampini pcis->scaling_factor = 1.0; 21919326c5c6Sstefano_zampini pcis->reusesubmatrices = PETSC_TRUE; 21929326c5c6Sstefano_zampini pcbddc->use_local_adj = PETSC_TRUE; 21939326c5c6Sstefano_zampini pcbddc->use_vertices = PETSC_TRUE; 21949326c5c6Sstefano_zampini pcbddc->use_edges = PETSC_TRUE; 21959326c5c6Sstefano_zampini pcbddc->symmetric_primal = PETSC_TRUE; 21969326c5c6Sstefano_zampini pcbddc->vertex_size = 1; 21979326c5c6Sstefano_zampini pcbddc->recompute_topography = PETSC_TRUE; 21989326c5c6Sstefano_zampini pcbddc->coarse_size = -1; 21999326c5c6Sstefano_zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 22009326c5c6Sstefano_zampini pcbddc->coarsening_ratio = 8; 22019326c5c6Sstefano_zampini pcbddc->coarse_eqs_per_proc = 1; 22029326c5c6Sstefano_zampini pcbddc->benign_compute_correction = PETSC_TRUE; 22039326c5c6Sstefano_zampini pcbddc->nedfield = -1; 22049326c5c6Sstefano_zampini pcbddc->nedglobal = PETSC_TRUE; 22059326c5c6Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 22069326c5c6Sstefano_zampini pcbddc->sub_schurs_layers = -1; 22079326c5c6Sstefano_zampini pcbddc->ksp_D = kspD; 22089326c5c6Sstefano_zampini pcbddc->ksp_R = kspR; 22099326c5c6Sstefano_zampini pcbddc->coarse_ksp = kspC; 22109326c5c6Sstefano_zampini PetscFunctionReturn(0); 22119326c5c6Sstefano_zampini } 22129326c5c6Sstefano_zampini 22139326c5c6Sstefano_zampini #undef __FUNCT__ 22149326c5c6Sstefano_zampini #define __FUNCT__ "PCDestroy_BDDC" 22159326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc) 22169326c5c6Sstefano_zampini { 22179326c5c6Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 22189326c5c6Sstefano_zampini PetscErrorCode ierr; 22199326c5c6Sstefano_zampini 22209326c5c6Sstefano_zampini PetscFunctionBegin; 22219326c5c6Sstefano_zampini ierr = PCReset_BDDC(pc);CHKERRQ(ierr); 22229326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr); 22239326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr); 22249326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr); 2225a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr); 2226a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr); 2227906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2228674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 222930368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2230bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 22312b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2232b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 22332b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2234bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 223582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2236bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 223782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2238bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 223982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2240bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2241785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2242bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 224363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2244bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2245bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2246bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2247bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2248a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2249674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2250da1bb401SStefano Zampini PetscFunctionReturn(0); 2251da1bb401SStefano Zampini } 22521e6b0712SBarry Smith 22533425bc38SStefano Zampini #undef __FUNCT__ 2254a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC" 2255a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2256a06fd7f2SStefano Zampini { 2257a06fd7f2SStefano Zampini PetscFunctionBegin; 2258a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2259a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2260a06fd7f2SStefano Zampini } 2261a06fd7f2SStefano Zampini 2262a06fd7f2SStefano Zampini #undef __FUNCT__ 22633425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 22643425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 22653425bc38SStefano Zampini { 2266674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2267266e20e9SStefano Zampini Vec work; 22683425bc38SStefano Zampini PC_IS* pcis; 22693425bc38SStefano Zampini PC_BDDC* pcbddc; 22703425bc38SStefano Zampini PetscErrorCode ierr; 22710c7d97c5SJed Brown 22723425bc38SStefano Zampini PetscFunctionBegin; 22733425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 22743425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 22753425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 22763425bc38SStefano Zampini 2277229984c5Sstefano_zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 2278229984c5Sstefano_zampini /* copy rhs since we may change it during PCPreSolve_BDDC */ 2279229984c5Sstefano_zampini if (!pcbddc->original_rhs) { 2280229984c5Sstefano_zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 2281229984c5Sstefano_zampini } 22826cc1294bSstefano_zampini if (mat_ctx->rhs_flip) { 22836cc1294bSstefano_zampini ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr); 22846cc1294bSstefano_zampini } else { 2285229984c5Sstefano_zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 22866cc1294bSstefano_zampini } 2287af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2288229984c5Sstefano_zampini /* interface pressure rhs */ 2289022d8d2bSstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2290022d8d2bSstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2291229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2292229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22936cc1294bSstefano_zampini if (!mat_ctx->rhs_flip) { 2294229984c5Sstefano_zampini ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr); 2295229984c5Sstefano_zampini } 22966cc1294bSstefano_zampini } 2297c08af4c6SStefano Zampini /* 2298c08af4c6SStefano Zampini change of basis for physical rhs if needed 2299c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2300c08af4c6SStefano Zampini */ 23013738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2302fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 23033738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 23043738a8e6SStefano Zampini work = pcbddc->work_change; 2305fc17d649SStefano Zampini } else { 23063738a8e6SStefano Zampini work = pcbddc->original_rhs; 2307fc17d649SStefano Zampini } 23083425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2309266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2310266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2311fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2312fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2313266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2314266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2315674ae819SStefano Zampini /* Apply partition of unity */ 23163425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2317266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 23188eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 23193425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 23203425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 23213425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 23223425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2323266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2324266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2325266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2326266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2327266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2328266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 23293425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 23303425bc38SStefano Zampini } 23313425bc38SStefano Zampini /* BDDC rhs */ 23323425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 23338eeda7d8SStefano Zampini if (pcbddc->switch_static) { 23343425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 23353425bc38SStefano Zampini } 23363425bc38SStefano Zampini /* apply BDDC */ 2337229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2338dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2339266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2340229984c5Sstefano_zampini 23413425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 23423425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 23433425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 23443425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2345229984c5Sstefano_zampini /* Add contribution to interface pressures */ 2346229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2347229984c5Sstefano_zampini ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr); 2348229984c5Sstefano_zampini if (pcbddc->switch_static) { 2349229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr); 2350229984c5Sstefano_zampini } 2351229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2352229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2353229984c5Sstefano_zampini } 23543425bc38SStefano Zampini PetscFunctionReturn(0); 23553425bc38SStefano Zampini } 23561e6b0712SBarry Smith 23573425bc38SStefano Zampini #undef __FUNCT__ 23583425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 23593425bc38SStefano Zampini /*@ 23600f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 23613425bc38SStefano Zampini 23623425bc38SStefano Zampini Collective 23633425bc38SStefano Zampini 23643425bc38SStefano Zampini Input Parameters: 23650f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 23660f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 23673425bc38SStefano Zampini 23683425bc38SStefano Zampini Output Parameters: 23690f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 23703425bc38SStefano Zampini 23713425bc38SStefano Zampini Level: developer 23723425bc38SStefano Zampini 23733425bc38SStefano Zampini Notes: 23743425bc38SStefano Zampini 23750f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 23763425bc38SStefano Zampini @*/ 23773425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 23783425bc38SStefano Zampini { 2379674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23803425bc38SStefano Zampini PetscErrorCode ierr; 23813425bc38SStefano Zampini 23823425bc38SStefano Zampini PetscFunctionBegin; 2383266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2384266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2385266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 23863425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2387163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 23883425bc38SStefano Zampini PetscFunctionReturn(0); 23893425bc38SStefano Zampini } 23901e6b0712SBarry Smith 23913425bc38SStefano Zampini #undef __FUNCT__ 23923425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 23933425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23943425bc38SStefano Zampini { 2395674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23963425bc38SStefano Zampini PC_IS* pcis; 23973425bc38SStefano Zampini PC_BDDC* pcbddc; 23983425bc38SStefano Zampini PetscErrorCode ierr; 2399229984c5Sstefano_zampini Vec work; 24003425bc38SStefano Zampini 24013425bc38SStefano Zampini PetscFunctionBegin; 24023425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 24033425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 24043425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 24053425bc38SStefano Zampini 24063425bc38SStefano Zampini /* apply B_delta^T */ 2407af140850Sstefano_zampini ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr); 24083425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 24093425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 24103425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 2411229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2412229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2413229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2414229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 2415229984c5Sstefano_zampini } 2416229984c5Sstefano_zampini 24173425bc38SStefano Zampini /* compute rhs for BDDC application */ 24183425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 24198eeda7d8SStefano Zampini if (pcbddc->switch_static) { 24203425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 2421229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2422229984c5Sstefano_zampini ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr); 2423229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 24243425bc38SStefano Zampini } 2425229984c5Sstefano_zampini } 2426229984c5Sstefano_zampini 24273425bc38SStefano Zampini /* apply BDDC */ 2428229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2429dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2430229984c5Sstefano_zampini 2431229984c5Sstefano_zampini /* put values into global vector */ 2432af140850Sstefano_zampini if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change; 2433af140850Sstefano_zampini else work = standard_sol; 2434229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2435229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 24368eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 24373425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 24383425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 243900f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 244000f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 24413425bc38SStefano Zampini } 2442229984c5Sstefano_zampini 2443229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2444229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2445266e20e9SStefano Zampini /* add p0 solution to final solution */ 2446229984c5Sstefano_zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr); 2447fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2448af140850Sstefano_zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr); 2449fc17d649SStefano Zampini } 2450af140850Sstefano_zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 2451af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2452229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2453229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2454229984c5Sstefano_zampini } 24553425bc38SStefano Zampini PetscFunctionReturn(0); 24563425bc38SStefano Zampini } 24571e6b0712SBarry Smith 24583425bc38SStefano Zampini #undef __FUNCT__ 24593425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 24603425bc38SStefano Zampini /*@ 24610f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 24623425bc38SStefano Zampini 24633425bc38SStefano Zampini Collective 24643425bc38SStefano Zampini 24653425bc38SStefano Zampini Input Parameters: 24660f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 24670f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 24683425bc38SStefano Zampini 24693425bc38SStefano Zampini Output Parameters: 24700f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 24713425bc38SStefano Zampini 24723425bc38SStefano Zampini Level: developer 24733425bc38SStefano Zampini 24743425bc38SStefano Zampini Notes: 24753425bc38SStefano Zampini 24760f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 24773425bc38SStefano Zampini @*/ 24783425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 24793425bc38SStefano Zampini { 2480674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 24813425bc38SStefano Zampini PetscErrorCode ierr; 24823425bc38SStefano Zampini 24833425bc38SStefano Zampini PetscFunctionBegin; 2484266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2485266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2486266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 24873425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2488163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 24893425bc38SStefano Zampini PetscFunctionReturn(0); 24903425bc38SStefano Zampini } 24911e6b0712SBarry Smith 24923425bc38SStefano Zampini #undef __FUNCT__ 24933425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 2494547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc) 24953425bc38SStefano Zampini { 2496674ae819SStefano Zampini 2497674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 24983425bc38SStefano Zampini Mat newmat; 2499674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 25003425bc38SStefano Zampini PC newpc; 2501ce94432eSBarry Smith MPI_Comm comm; 25023425bc38SStefano Zampini PetscErrorCode ierr; 25033425bc38SStefano Zampini 25043425bc38SStefano Zampini PetscFunctionBegin; 2505ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 25063425bc38SStefano Zampini /* FETIDP linear matrix */ 25073425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 25081720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 25093425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 2510a5bb87b3Sstefano_zampini ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 25113425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2512edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 25133425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 2514547c9a8eSstefano_zampini ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr); 2515547c9a8eSstefano_zampini ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr); 25163425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 25173425bc38SStefano Zampini /* FETIDP preconditioner */ 25183425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 25193425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 25203425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 25213425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 25223425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 25233425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2524edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2525c45b8d2dSstefano_zampini ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr); 25263425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 252723ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 2528547c9a8eSstefano_zampini ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr); 2529547c9a8eSstefano_zampini ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr); 25303425bc38SStefano Zampini /* return pointers for objects created */ 25313425bc38SStefano Zampini *fetidp_mat=newmat; 25323425bc38SStefano Zampini *fetidp_pc=newpc; 25333425bc38SStefano Zampini PetscFunctionReturn(0); 25343425bc38SStefano Zampini } 25351e6b0712SBarry Smith 25363425bc38SStefano Zampini #undef __FUNCT__ 25373425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 25383425bc38SStefano Zampini /*@ 25390f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 25403425bc38SStefano Zampini 25413425bc38SStefano Zampini Collective 25423425bc38SStefano Zampini 25433425bc38SStefano Zampini Input Parameters: 25441720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 2545547c9a8eSstefano_zampini . fully_redundant - true for a fully redundant set of Lagrange multipliers 2546547c9a8eSstefano_zampini - prefix - optional options database prefix for the objects to be created (can be NULL) 254728509bceSStefano Zampini 254828509bceSStefano Zampini Output Parameters: 25490f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 25500f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 255128509bceSStefano Zampini 25523425bc38SStefano Zampini Level: developer 25533425bc38SStefano Zampini 25543425bc38SStefano Zampini Notes: 25550f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 25563425bc38SStefano Zampini 25570f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 25583425bc38SStefano Zampini @*/ 2559547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc) 25603425bc38SStefano Zampini { 25613425bc38SStefano Zampini PetscErrorCode ierr; 25623425bc38SStefano Zampini 25633425bc38SStefano Zampini PetscFunctionBegin; 25643425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 25653425bc38SStefano Zampini if (pc->setupcalled) { 2566547c9a8eSstefano_zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2567f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 25683425bc38SStefano Zampini PetscFunctionReturn(0); 25693425bc38SStefano Zampini } 25700c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2571da1bb401SStefano Zampini /*MC 2572da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 25730c7d97c5SJed Brown 257428509bceSStefano Zampini An implementation of the BDDC preconditioner based on 257528509bceSStefano Zampini 257628509bceSStefano Zampini .vb 257728509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 257828509bceSStefano 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 257928509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 25800f202f7eSStefano 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 258128509bceSStefano Zampini .ve 258228509bceSStefano Zampini 258328509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 258428509bceSStefano Zampini 25850f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 258628509bceSStefano Zampini 258728509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 258828509bceSStefano Zampini 2589b6fdb6dfSStefano 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. 2590b6fdb6dfSStefano Zampini 2591c7017625SStefano 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). 259228509bceSStefano Zampini 25930f202f7eSStefano 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() 259430368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 259528509bceSStefano Zampini 25960f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 259728509bceSStefano Zampini 25980f202f7eSStefano 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. 25990f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 260028509bceSStefano Zampini 26010f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 260228509bceSStefano Zampini 2603df4d28bfSStefano 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. 260428509bceSStefano Zampini 26050f202f7eSStefano 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. 26060f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 26070f202f7eSStefano Zampini 26080f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 26090f202f7eSStefano Zampini 26100f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 26110f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 26120f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 26130f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 26140f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 26150f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 26160f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 261728509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 26180f202f7eSStefano 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) 26190f202f7eSStefano Zampini . -pc_bddc_redistribute <0> - size of a subset of processors where the coarse problem will be remapped (the value is ignored if not at the coarsest level) 26200f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 26210f202f7eSStefano 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) 2622df4d28bfSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 262328509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 262428509bceSStefano Zampini 262528509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 262628509bceSStefano Zampini .vb 262728509bceSStefano Zampini -pc_bddc_dirichlet_ 262828509bceSStefano Zampini -pc_bddc_neumann_ 262928509bceSStefano Zampini -pc_bddc_coarse_ 263028509bceSStefano Zampini .ve 26310f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 263228509bceSStefano Zampini 26330f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 263428509bceSStefano Zampini .vb 2635312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2636312be037SStefano Zampini -pc_bddc_neumann_lN_ 2637312be037SStefano Zampini -pc_bddc_coarse_lN_ 263828509bceSStefano Zampini .ve 26390f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 26400f202f7eSStefano 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. 26410f202f7eSStefano Zampini .vb 26420f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 26430f202f7eSStefano Zampini .ve 26440f202f7eSStefano 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 2645da1bb401SStefano Zampini 2646da1bb401SStefano Zampini Level: intermediate 2647da1bb401SStefano Zampini 2648b6fdb6dfSStefano Zampini Developer notes: 2649da1bb401SStefano Zampini 2650da1bb401SStefano Zampini Contributed by Stefano Zampini 2651da1bb401SStefano Zampini 2652da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2653da1bb401SStefano Zampini M*/ 2654b2573a8aSBarry Smith 2655da1bb401SStefano Zampini #undef __FUNCT__ 2656da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 26578cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2658da1bb401SStefano Zampini { 2659da1bb401SStefano Zampini PetscErrorCode ierr; 2660da1bb401SStefano Zampini PC_BDDC *pcbddc; 2661da1bb401SStefano Zampini 2662da1bb401SStefano Zampini PetscFunctionBegin; 2663b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2664da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2665da1bb401SStefano Zampini 2666da1bb401SStefano Zampini /* create PCIS data structure */ 2667da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2668da1bb401SStefano Zampini 26699326c5c6Sstefano_zampini /* create local graph structure */ 26709326c5c6Sstefano_zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 26719326c5c6Sstefano_zampini 26729326c5c6Sstefano_zampini /* BDDC nonzero defaults */ 267308a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 267447d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 267547d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 26763301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 267714f95afaSStefano Zampini pcbddc->vertex_size = 1; 2678c703fcc7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 267968457ee5SStefano Zampini pcbddc->coarse_size = -1; 268085c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 268147d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 268257de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 268327b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 26841e0482f5SStefano Zampini pcbddc->nedfield = -1; 26851e0482f5SStefano Zampini pcbddc->nedglobal = PETSC_TRUE; 2686be12c134Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 2687b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2688b7eb3628SStefano Zampini 2689da1bb401SStefano Zampini /* function pointers */ 2690da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 269193bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2692da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2693da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2694da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 26956b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2696da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2697da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2698da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2699534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2700534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 27019326c5c6Sstefano_zampini pc->ops->reset = PCReset_BDDC; 2702da1bb401SStefano Zampini 2703da1bb401SStefano Zampini /* composing function */ 2704a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr); 2705a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr); 2706906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2707674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 270830368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2709bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 27102b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2711b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 27122b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2713bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 271482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2715bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 271682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2717bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 271882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2719bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 272082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2721bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 272263602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2723bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2724bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2725bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2726bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2727a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2728da1bb401SStefano Zampini PetscFunctionReturn(0); 2729da1bb401SStefano Zampini } 27303425bc38SStefano Zampini 2731