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 2643371fb9SStefano Zampini static PetscBool PCBDDCPackageInitialized = PETSC_FALSE; 2743371fb9SStefano Zampini 28f3d41395Sstefano_zampini static PetscBool cited = PETSC_FALSE; 29f3d41395Sstefano_zampini static const char citation[] = 30f3d41395Sstefano_zampini "@article{ZampiniPCBDDC,\n" 31f3d41395Sstefano_zampini "author = {Stefano Zampini},\n" 32f3d41395Sstefano_zampini "title = {{PCBDDC}: A Class of Robust Dual-Primal Methods in {PETS}c},\n" 33f3d41395Sstefano_zampini "journal = {SIAM Journal on Scientific Computing},\n" 34f3d41395Sstefano_zampini "volume = {38},\n" 35f3d41395Sstefano_zampini "number = {5},\n" 36f3d41395Sstefano_zampini "pages = {S282-S306},\n" 37f3d41395Sstefano_zampini "year = {2016},\n" 38f3d41395Sstefano_zampini "doi = {10.1137/15M1025785},\n" 39f3d41395Sstefano_zampini "URL = {http://dx.doi.org/10.1137/15M1025785},\n" 40f3d41395Sstefano_zampini "eprint = {http://dx.doi.org/10.1137/15M1025785}\n" 41f3d41395Sstefano_zampini "}\n"; 42f3d41395Sstefano_zampini 4343371fb9SStefano Zampini PetscLogEvent PC_BDDC_Topology[PETSC_PCBDDC_MAXLEVELS]; 4443371fb9SStefano Zampini PetscLogEvent PC_BDDC_LocalSolvers[PETSC_PCBDDC_MAXLEVELS]; 4543371fb9SStefano Zampini PetscLogEvent PC_BDDC_LocalWork[PETSC_PCBDDC_MAXLEVELS]; 4643371fb9SStefano Zampini PetscLogEvent PC_BDDC_CorrectionSetUp[PETSC_PCBDDC_MAXLEVELS]; 4743371fb9SStefano Zampini PetscLogEvent PC_BDDC_CoarseSetUp[PETSC_PCBDDC_MAXLEVELS]; 4843371fb9SStefano Zampini PetscLogEvent PC_BDDC_CoarseSolver[PETSC_PCBDDC_MAXLEVELS]; 4943371fb9SStefano Zampini PetscLogEvent PC_BDDC_AdaptiveSetUp[PETSC_PCBDDC_MAXLEVELS]; 5043371fb9SStefano Zampini PetscLogEvent PC_BDDC_Scaling[PETSC_PCBDDC_MAXLEVELS]; 5143371fb9SStefano Zampini PetscLogEvent PC_BDDC_Schurs[PETSC_PCBDDC_MAXLEVELS]; 5243371fb9SStefano Zampini 530369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 540369aaf7SStefano Zampini 554416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 560c7d97c5SJed Brown { 570c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 58*e569e4e1SStefano Zampini PetscInt nt,i; 590c7d97c5SJed Brown PetscErrorCode ierr; 600c7d97c5SJed Brown 610c7d97c5SJed Brown PetscFunctionBegin; 62e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 638eeda7d8SStefano Zampini /* Verbose debugging */ 64a13144ffSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 65a13144ffSStefano Zampini /* Approximate solvers */ 66c7017625SStefano 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); 67c7017625SStefano 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); 68c7017625SStefano 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); 69c7017625SStefano 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); 706b78500eSPatrick Sanan /* Primal space customization */ 7108a5cf49SStefano 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); 72be12c134Sstefano_zampini ierr = PetscOptionsInt("-pc_bddc_graph_maxcount","Maximum number of shared subdomains for a connected component","none",pcbddc->graphmaxcount,&pcbddc->graphmaxcount,NULL);CHKERRQ(ierr); 731c7a958bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_corner_selection","Activates face-based corner selection","none",pcbddc->corner_selection,&pcbddc->corner_selection,NULL);CHKERRQ(ierr); 748eeda7d8SStefano 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); 758eeda7d8SStefano 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); 768eeda7d8SStefano 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); 7714f95afaSStefano 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); 786661aa1dSStefano 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); 7914f95afaSStefano 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); 808eeda7d8SStefano Zampini /* Change of basis */ 81b9b85e73SStefano 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); 82b9b85e73SStefano 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); 83674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 84674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 85674ae819SStefano Zampini } 868eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 878eeda7d8SStefano 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); 88*e569e4e1SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_eqs_per_proc","Target 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); 89*e569e4e1SStefano Zampini i = pcbddc->coarsening_ratio; 90*e569e4e1SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","PCBDDCSetCoarseningRatio",i,&i,NULL);CHKERRQ(ierr); 91*e569e4e1SStefano Zampini ierr = PCBDDCSetCoarseningRatio(pc,i);CHKERRQ(ierr); 92*e569e4e1SStefano Zampini i = pcbddc->max_levels; 93*e569e4e1SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","PCBDDCSetLevels",i,&i,NULL);CHKERRQ(ierr); 94*e569e4e1SStefano Zampini ierr = PCBDDCSetLevels(pc,i);CHKERRQ(ierr); 95*e569e4e1SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_eqs_limit","Set maximum number of equations on coarsest grid to aim for","none",pcbddc->coarse_eqs_limit,&pcbddc->coarse_eqs_limit,NULL);CHKERRQ(ierr); 96323d395dSStefano 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); 97674ae819SStefano 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); 98b96c3477SStefano 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); 99b96c3477SStefano 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); 100b96c3477SStefano 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); 101683d3df6SStefano 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); 102bf3a8328SStefano 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); 103839e9adbSstefano_zampini ierr = PetscOptionsBool("-pc_bddc_deluxe_singlemat","Collapse deluxe operators","none",pcbddc->deluxe_singlemat,&pcbddc->deluxe_singlemat,NULL);CHKERRQ(ierr); 104bf3a8328SStefano 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); 105bd2a564bSStefano Zampini nt = 2; 106bd2a564bSStefano Zampini ierr = PetscOptionsRealArray("-pc_bddc_adaptive_threshold","Thresholds to be used for adaptive selection of constraints","none",pcbddc->adaptive_threshold,&nt,NULL);CHKERRQ(ierr); 107bd2a564bSStefano Zampini if (nt == 1) pcbddc->adaptive_threshold[1] = pcbddc->adaptive_threshold[0]; 10808122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 10908122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 1103301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 111b0c7d250SStefano 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); 112b3afcdbeSStefano 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); 113e9627c49SStefano 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); 11427b6a85dSStefano 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); 115a198735bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->compute_nonetflux,&pcbddc->compute_nonetflux,NULL);CHKERRQ(ierr); 1164f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 1178361f951SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected_filter","Filters out small entries in the local matrix when detecting disconnected subdomains","none",pcbddc->detect_disconnected_filter,&pcbddc->detect_disconnected_filter,NULL);CHKERRQ(ierr); 11870c64980SStefano 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); 1190c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 1200c7d97c5SJed Brown PetscFunctionReturn(0); 1210c7d97c5SJed Brown } 1226b78500eSPatrick Sanan 1236b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 1246b78500eSPatrick Sanan { 1256b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 126e9627c49SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 1276b78500eSPatrick Sanan PetscErrorCode ierr; 12871783a16SStefano Zampini PetscBool isascii; 129e9627c49SStefano Zampini PetscSubcomm subcomm; 130e9627c49SStefano Zampini PetscViewer subviewer; 1316b78500eSPatrick Sanan 1326b78500eSPatrick Sanan PetscFunctionBegin; 1336b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 1346b78500eSPatrick Sanan /* ASCII viewer */ 1356b78500eSPatrick Sanan if (isascii) { 1364b2aedd3SStefano Zampini PetscMPIInt color,rank,size; 137fbad9177SStefano Zampini PetscInt64 loc[7],gsum[6],gmax[6],gmin[6],totbenign; 138e9627c49SStefano Zampini PetscScalar interface_size; 139e9627c49SStefano Zampini PetscReal ratio1=0.,ratio2=0.; 140e9627c49SStefano Zampini Vec counter; 1416b78500eSPatrick Sanan 142b74ba07aSstefano_zampini if (!pc->setupcalled) { 143b74ba07aSstefano_zampini ierr = PetscViewerASCIIPrintf(viewer," Partial information available: preconditioner has not been setup yet\n");CHKERRQ(ierr); 144b74ba07aSstefano_zampini } 145efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use verbose output: %D\n",pcbddc->dbg_flag);CHKERRQ(ierr); 1466f0c0a6aSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use user-defined CSR: %d\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 1476f0c0a6aSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use local mat graph: %d\n",pcbddc->use_local_adj && !pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 148e9627c49SStefano Zampini if (pcbddc->mat_graph->twodim) { 149efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Connectivity graph topological dimension: 2\n");CHKERRQ(ierr); 150e9627c49SStefano Zampini } else { 151efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Connectivity graph topological dimension: 3\n");CHKERRQ(ierr); 152e9627c49SStefano Zampini } 153aefa1729SStefano Zampini if (pcbddc->graphmaxcount != PETSC_MAX_INT) { 154efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Graph max count: %D\n",pcbddc->graphmaxcount);CHKERRQ(ierr); 155aefa1729SStefano Zampini } 15650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Use vertices: %d (vertex size %D)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 157efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 158efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 159efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 160efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 161efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 162efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 163efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 164efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 165efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 166efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 167efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 16850e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Multilevel max levels: %D\n",pcbddc->max_levels);CHKERRQ(ierr); 16950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Multilevel coarsening ratio: %D\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 170efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 171efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 172efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 173efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use deluxe singlemat: %d\n",pcbddc->deluxe_singlemat);CHKERRQ(ierr); 174efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 17550e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Number of dofs' layers for the computation of principal minors: %D\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 176efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 177bd2a564bSStefano Zampini if (pcbddc->adaptive_threshold[1] != pcbddc->adaptive_threshold[0]) { 178bd2a564bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Adaptive constraint selection thresholds (active %d, userdefined %d): %g,%g\n",pcbddc->adaptive_selection,pcbddc->adaptive_userdefined,pcbddc->adaptive_threshold[0],pcbddc->adaptive_threshold[1]);CHKERRQ(ierr); 179bd2a564bSStefano Zampini } else { 180bd2a564bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_selection,pcbddc->adaptive_userdefined,pcbddc->adaptive_threshold[0]);CHKERRQ(ierr); 181bd2a564bSStefano Zampini } 18250e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Min constraints / connected component: %D\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 18350e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Max constraints / connected component: %D\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 184efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 185efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 18650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Num. Procs. to map coarse adjacency list: %D\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 18750e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Coarse eqs per proc (significant at the coarsest level): %D\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 1888361f951SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Detect disconnected: %d (filter %d)\n",pcbddc->detect_disconnected,pcbddc->detect_disconnected_filter);CHKERRQ(ierr); 189efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 190efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr); 19115579a77SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Algebraic computation of no-net-flux: %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr); 192b74ba07aSstefano_zampini if (!pc->setupcalled) PetscFunctionReturn(0); 1936b78500eSPatrick Sanan 194fbad9177SStefano Zampini /* compute interface size */ 195e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 196e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 197e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 198e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 199e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 200e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 201e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 202fbad9177SStefano Zampini 203fbad9177SStefano Zampini /* compute some statistics on the domain decomposition */ 204e9627c49SStefano Zampini gsum[0] = 1; 205fbad9177SStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = gsum[5] = 0; 206e9627c49SStefano Zampini loc[0] = !!pcis->n; 207e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 208e9627c49SStefano Zampini loc[2] = pcis->n_B; 209e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 210345ecf6cSStefano Zampini loc[4] = pcis->n; 211fbad9177SStefano Zampini loc[5] = pcbddc->n_local_subs > 0 ? pcbddc->n_local_subs : (pcis->n ? 1 : 0); 212fbad9177SStefano Zampini loc[6] = pcbddc->benign_n; 213fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gsum,6,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 214fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = -1; 215fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmax,6,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 216fbad9177SStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = loc[5] = PETSC_MAX_INT; 217fbad9177SStefano Zampini ierr = MPI_Reduce(loc,gmin,6,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 218fbad9177SStefano Zampini ierr = MPI_Reduce(&loc[6],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 219e9627c49SStefano Zampini if (pcbddc->coarse_size) { 220e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 221e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 222e9627c49SStefano Zampini } 223efd4aadfSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 22450e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Global dofs sizes: all %D interface %D coarse %D\n",pc->pmat->rmap->N,(PetscInt)PetscRealPart(interface_size),pcbddc->coarse_size);CHKERRQ(ierr); 22550e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Coarsening ratios: all/coarse %D interface/coarse %D\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 22650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Active processes : %D\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 22750e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Total subdomains : %D\n",(PetscInt)gsum[5]);CHKERRQ(ierr); 228345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 22950e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Benign subs : %D\n",(PetscInt)totbenign);CHKERRQ(ierr); 230345ecf6cSStefano Zampini } 23150e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Dofs type :\tMIN\tMAX\tMEAN\n");CHKERRQ(ierr); 23250e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Interior dofs :\t%D\t%D\t%D\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 23350e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Interface dofs :\t%D\t%D\t%D\n",(PetscInt)gmin[2],(PetscInt)gmax[2],(PetscInt)(gsum[2]/gsum[0]));CHKERRQ(ierr); 23450e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Primal dofs :\t%D\t%D\t%D\n",(PetscInt)gmin[3],(PetscInt)gmax[3],(PetscInt)(gsum[3]/gsum[0]));CHKERRQ(ierr); 23550e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Local dofs :\t%D\t%D\t%D\n",(PetscInt)gmin[4],(PetscInt)gmax[4],(PetscInt)(gsum[4]/gsum[0]));CHKERRQ(ierr); 23650e0721cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," Local subs :\t%D\t%D\n" ,(PetscInt)gmin[5],(PetscInt)gmax[5]);CHKERRQ(ierr); 23715579a77SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 23815579a77SStefano Zampini 23915579a77SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 24015579a77SStefano Zampini 24115579a77SStefano Zampini /* local solvers */ 24215579a77SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscObjectComm((PetscObject)pcbddc->ksp_D),&subviewer);CHKERRQ(ierr); 24315579a77SStefano Zampini if (!rank) { 24415579a77SStefano Zampini ierr = PetscViewerASCIIPrintf(subviewer,"--- Interior solver (rank 0)\n");CHKERRQ(ierr); 24515579a77SStefano Zampini ierr = PetscViewerASCIIPushTab(subviewer);CHKERRQ(ierr); 24615579a77SStefano Zampini ierr = KSPView(pcbddc->ksp_D,subviewer);CHKERRQ(ierr); 24715579a77SStefano Zampini ierr = PetscViewerASCIIPopTab(subviewer);CHKERRQ(ierr); 24815579a77SStefano Zampini ierr = PetscViewerASCIIPrintf(subviewer,"--- Correction solver (rank 0)\n");CHKERRQ(ierr); 24915579a77SStefano Zampini ierr = PetscViewerASCIIPushTab(subviewer);CHKERRQ(ierr); 25015579a77SStefano Zampini ierr = KSPView(pcbddc->ksp_R,subviewer);CHKERRQ(ierr); 25115579a77SStefano Zampini ierr = PetscViewerASCIIPopTab(subviewer);CHKERRQ(ierr); 25215579a77SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 25315579a77SStefano Zampini } 25415579a77SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscObjectComm((PetscObject)pcbddc->ksp_D),&subviewer);CHKERRQ(ierr); 25527b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 256e9627c49SStefano Zampini 257fbad9177SStefano Zampini /* the coarse problem can be handled by a different communicator */ 258e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 259e9627c49SStefano Zampini else color = 0; 2604b2aedd3SStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 261e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 2624b2aedd3SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,2));CHKERRQ(ierr); 263e9627c49SStefano Zampini ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr); 264e9627c49SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 265e9627c49SStefano Zampini if (color == 1) { 26615579a77SStefano Zampini ierr = PetscViewerASCIIPrintf(subviewer,"--- Coarse solver\n");CHKERRQ(ierr); 26715579a77SStefano Zampini ierr = PetscViewerASCIIPushTab(subviewer);CHKERRQ(ierr); 268e9627c49SStefano Zampini ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr); 26915579a77SStefano Zampini ierr = PetscViewerASCIIPopTab(subviewer);CHKERRQ(ierr); 270e9627c49SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 271e9627c49SStefano Zampini } 272e9627c49SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 273e9627c49SStefano Zampini ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr); 274e9627c49SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 275e9627c49SStefano Zampini } 2766b78500eSPatrick Sanan PetscFunctionReturn(0); 2776b78500eSPatrick Sanan } 278a13144ffSStefano Zampini 2791e0482f5SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 280a13144ffSStefano Zampini { 281a13144ffSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 282a13144ffSStefano Zampini PetscErrorCode ierr; 283a13144ffSStefano Zampini 284a13144ffSStefano Zampini PetscFunctionBegin; 285a13144ffSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 286a13144ffSStefano Zampini ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); 287a13144ffSStefano Zampini pcbddc->discretegradient = G; 288a13144ffSStefano Zampini pcbddc->nedorder = order > 0 ? order : -order; 289495a2a07SStefano Zampini pcbddc->nedfield = field; 2901e0482f5SStefano Zampini pcbddc->nedglobal = global; 2911e0482f5SStefano Zampini pcbddc->conforming = conforming; 292a13144ffSStefano Zampini PetscFunctionReturn(0); 293a13144ffSStefano Zampini } 294a13144ffSStefano Zampini 295a13144ffSStefano Zampini /*@ 296a13144ffSStefano Zampini PCBDDCSetDiscreteGradient - Sets the discrete gradient 297a13144ffSStefano Zampini 298a13144ffSStefano Zampini Collective on PC 299a13144ffSStefano Zampini 300a13144ffSStefano Zampini Input Parameters: 301a13144ffSStefano Zampini + pc - the preconditioning context 302a13144ffSStefano Zampini . G - the discrete gradient matrix (should be in AIJ format) 303a13144ffSStefano Zampini . order - the order of the Nedelec space (1 for the lowest order) 304495a2a07SStefano Zampini . field - the field id of the Nedelec dofs (not used if the fields have not been specified) 3051e0482f5SStefano Zampini . global - the type of global ordering for the rows of G 306a13144ffSStefano Zampini - conforming - whether the mesh is conforming or not 307a13144ffSStefano Zampini 308a13144ffSStefano Zampini Level: advanced 309a13144ffSStefano Zampini 31095452b02SPatrick Sanan Notes: 31195452b02SPatrick Sanan The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry. 312495a2a07SStefano Zampini For variable order spaces, the order should be set to zero. 3131e0482f5SStefano Zampini If global is true, the rows of G should be given in global ordering for the whole dofs; 3141e0482f5SStefano Zampini if false, the ordering should be global for the Nedelec field. 3151e0482f5SStefano 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 3161e0482f5SStefano Zampini and geid the one for the Nedelec field. 317a13144ffSStefano Zampini 318495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal() 319a13144ffSStefano Zampini @*/ 3201e0482f5SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool global, PetscBool conforming) 321a13144ffSStefano Zampini { 322a13144ffSStefano Zampini PetscErrorCode ierr; 323a13144ffSStefano Zampini 324a13144ffSStefano Zampini PetscFunctionBegin; 325a13144ffSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 326a13144ffSStefano Zampini PetscValidHeaderSpecific(G,MAT_CLASSID,2); 327a13144ffSStefano Zampini PetscValidLogicalCollectiveInt(pc,order,3); 3281e0482f5SStefano Zampini PetscValidLogicalCollectiveInt(pc,field,4); 3291e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,global,5); 3301e0482f5SStefano Zampini PetscValidLogicalCollectiveBool(pc,conforming,6); 3311e0482f5SStefano Zampini PetscCheckSameComm(pc,1,G,2); 3321e0482f5SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool,PetscBool),(pc,G,order,field,global,conforming));CHKERRQ(ierr); 333a13144ffSStefano Zampini PetscFunctionReturn(0); 334a13144ffSStefano Zampini } 335a13144ffSStefano Zampini 3368ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 337a198735bSStefano Zampini { 338a198735bSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 339a198735bSStefano Zampini PetscErrorCode ierr; 3406b78500eSPatrick Sanan 341a198735bSStefano Zampini PetscFunctionBegin; 342a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr); 343a198735bSStefano Zampini ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); 344a198735bSStefano Zampini pcbddc->divudotp = divudotp; 3458ae0ca82SStefano Zampini pcbddc->divudotp_trans = trans; 346a198735bSStefano Zampini pcbddc->compute_nonetflux = PETSC_TRUE; 347a198735bSStefano Zampini if (vl2l) { 348a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr); 349fa23a32eSStefano Zampini ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); 350a198735bSStefano Zampini pcbddc->divudotp_vl2l = vl2l; 351a198735bSStefano Zampini } 352a198735bSStefano Zampini PetscFunctionReturn(0); 353a198735bSStefano Zampini } 3543d996552SStefano Zampini 355a198735bSStefano Zampini /*@ 356a198735bSStefano Zampini PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx 357a198735bSStefano Zampini 358a198735bSStefano Zampini Collective on PC 359a198735bSStefano Zampini 360a198735bSStefano Zampini Input Parameters: 361a198735bSStefano Zampini + pc - the preconditioning context 362a198735bSStefano Zampini . divudotp - the matrix (must be of type MATIS) 3638ae0ca82SStefano Zampini . trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space. 36405a3bf82SStefano Zampini - vl2l - optional index set describing the local (wrt the local matrix in divudotp) to local (wrt the local matrix in the preconditioning matrix) map for the velocities 365a198735bSStefano Zampini 366a198735bSStefano Zampini Level: advanced 367a198735bSStefano Zampini 36895452b02SPatrick Sanan Notes: 36995452b02SPatrick Sanan This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries 37005a3bf82SStefano Zampini If vl2l is NULL, the local ordering for velocities in divudotp should match that of the preconditioning matrix 371a198735bSStefano Zampini 372a198735bSStefano Zampini .seealso: PCBDDC 373a198735bSStefano Zampini @*/ 3748ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 375a198735bSStefano Zampini { 376a198735bSStefano Zampini PetscBool ismatis; 377a198735bSStefano Zampini PetscErrorCode ierr; 378a198735bSStefano Zampini 379a198735bSStefano Zampini PetscFunctionBegin; 380a198735bSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 381a198735bSStefano Zampini PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2); 382a198735bSStefano Zampini PetscCheckSameComm(pc,1,divudotp,2); 3838ae0ca82SStefano Zampini PetscValidLogicalCollectiveBool(pc,trans,3); 3848ae0ca82SStefano Zampini if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4); 385a198735bSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr); 386a198735bSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS"); 3878ae0ca82SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr); 388a198735bSStefano Zampini PetscFunctionReturn(0); 389a198735bSStefano Zampini } 3902d505d7fSStefano Zampini 3911dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 392b9b85e73SStefano Zampini { 393b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 394b9b85e73SStefano Zampini PetscErrorCode ierr; 395b9b85e73SStefano Zampini 396b9b85e73SStefano Zampini PetscFunctionBegin; 397b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 39856282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 399b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 4001dd7afcfSStefano Zampini pcbddc->change_interior = interior; 401b9b85e73SStefano Zampini PetscFunctionReturn(0); 402b9b85e73SStefano Zampini } 403b9b85e73SStefano Zampini /*@ 404906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 405b9b85e73SStefano Zampini 406b9b85e73SStefano Zampini Collective on PC 407b9b85e73SStefano Zampini 408b9b85e73SStefano Zampini Input Parameters: 409b9b85e73SStefano Zampini + pc - the preconditioning context 4101dd7afcfSStefano Zampini . change - the change of basis matrix 4111dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 412b9b85e73SStefano Zampini 413b9b85e73SStefano Zampini Level: intermediate 414b9b85e73SStefano Zampini 415b9b85e73SStefano Zampini Notes: 416b9b85e73SStefano Zampini 417b9b85e73SStefano Zampini .seealso: PCBDDC 418b9b85e73SStefano Zampini @*/ 4191dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 420b9b85e73SStefano Zampini { 421b9b85e73SStefano Zampini PetscErrorCode ierr; 422b9b85e73SStefano Zampini 423b9b85e73SStefano Zampini PetscFunctionBegin; 424b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 425b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 426906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 427906d46d4SStefano Zampini if (pc->mat) { 428906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 429906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 430906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 431e0fe2d75SToby Isaac if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %D != %D",rows_c,rows); 432e0fe2d75SToby Isaac if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %D != %D",cols_c,cols); 433906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 434906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 435e0fe2d75SToby Isaac if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %D != %D",rows_c,rows); 436e0fe2d75SToby Isaac if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %D != %D",cols_c,cols); 437906d46d4SStefano Zampini } 4381dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 439b9b85e73SStefano Zampini PetscFunctionReturn(0); 440b9b85e73SStefano Zampini } 4412d505d7fSStefano Zampini 44230368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 44330368db7SStefano Zampini { 44430368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 44556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 44630368db7SStefano Zampini PetscErrorCode ierr; 44730368db7SStefano Zampini 44830368db7SStefano Zampini PetscFunctionBegin; 44956282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 45056282151SStefano Zampini if (pcbddc->user_primal_vertices) { 45156282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 45256282151SStefano Zampini } 45330368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 45430368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 45530368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 45656282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 45730368db7SStefano Zampini PetscFunctionReturn(0); 45830368db7SStefano Zampini } 459ab8c8b98SStefano Zampini 46030368db7SStefano Zampini /*@ 46130368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 46230368db7SStefano Zampini 46330368db7SStefano Zampini Collective 46430368db7SStefano Zampini 46530368db7SStefano Zampini Input Parameters: 46630368db7SStefano Zampini + pc - the preconditioning context 46730368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 46830368db7SStefano Zampini 46930368db7SStefano Zampini Level: intermediate 47030368db7SStefano Zampini 47130368db7SStefano Zampini Notes: 47230368db7SStefano Zampini Any process can list any global node 47330368db7SStefano Zampini 47430368db7SStefano Zampini .seealso: PCBDDC 47530368db7SStefano Zampini @*/ 47630368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 47730368db7SStefano Zampini { 47830368db7SStefano Zampini PetscErrorCode ierr; 47930368db7SStefano Zampini 48030368db7SStefano Zampini PetscFunctionBegin; 48130368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 48230368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 48330368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 48430368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 48530368db7SStefano Zampini PetscFunctionReturn(0); 48630368db7SStefano Zampini } 4872d505d7fSStefano Zampini 488674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 489674ae819SStefano Zampini { 490674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 49156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 492674ae819SStefano Zampini PetscErrorCode ierr; 4931e6b0712SBarry Smith 494674ae819SStefano Zampini PetscFunctionBegin; 49556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 49656282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 49756282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 49856282151SStefano Zampini } 499674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 50030368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 50130368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 50256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 503674ae819SStefano Zampini PetscFunctionReturn(0); 504674ae819SStefano Zampini } 505674ae819SStefano Zampini /*@ 50628509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 507674ae819SStefano Zampini 50817eb1463SStefano Zampini Collective 509674ae819SStefano Zampini 510674ae819SStefano Zampini Input Parameters: 511674ae819SStefano Zampini + pc - the preconditioning context 51217eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 513674ae819SStefano Zampini 514674ae819SStefano Zampini Level: intermediate 515674ae819SStefano Zampini 516674ae819SStefano Zampini Notes: 517674ae819SStefano Zampini 518674ae819SStefano Zampini .seealso: PCBDDC 519674ae819SStefano Zampini @*/ 520674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 521674ae819SStefano Zampini { 522674ae819SStefano Zampini PetscErrorCode ierr; 523674ae819SStefano Zampini 524674ae819SStefano Zampini PetscFunctionBegin; 525674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 526674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 52717eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 528674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 529674ae819SStefano Zampini PetscFunctionReturn(0); 530674ae819SStefano Zampini } 5312d505d7fSStefano Zampini 5324fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 5334fad6a16SStefano Zampini { 5344fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5354fad6a16SStefano Zampini 5364fad6a16SStefano Zampini PetscFunctionBegin; 5374fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 5384fad6a16SStefano Zampini PetscFunctionReturn(0); 5394fad6a16SStefano Zampini } 5401e6b0712SBarry Smith 5414fad6a16SStefano Zampini /*@ 54228509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 5434fad6a16SStefano Zampini 5444fad6a16SStefano Zampini Logically collective on PC 5454fad6a16SStefano Zampini 5464fad6a16SStefano Zampini Input Parameters: 5474fad6a16SStefano Zampini + pc - the preconditioning context 54828509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 5494fad6a16SStefano Zampini 5500f202f7eSStefano Zampini Options Database Keys: 5510f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 5524fad6a16SStefano Zampini 5534fad6a16SStefano Zampini Level: intermediate 5544fad6a16SStefano Zampini 5554fad6a16SStefano Zampini Notes: 5560f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 5574fad6a16SStefano Zampini 5580f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 5594fad6a16SStefano Zampini @*/ 5604fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 5614fad6a16SStefano Zampini { 5624fad6a16SStefano Zampini PetscErrorCode ierr; 5634fad6a16SStefano Zampini 5644fad6a16SStefano Zampini PetscFunctionBegin; 5654fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5662b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 5674fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 5684fad6a16SStefano Zampini PetscFunctionReturn(0); 5694fad6a16SStefano Zampini } 5702b510759SStefano Zampini 571b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 572b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 573b8ffe317SStefano Zampini { 574b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 575b8ffe317SStefano Zampini 576b8ffe317SStefano Zampini PetscFunctionBegin; 57785c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 578b8ffe317SStefano Zampini PetscFunctionReturn(0); 579b8ffe317SStefano Zampini } 580b8ffe317SStefano Zampini 581b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 5822b510759SStefano Zampini { 5832b510759SStefano Zampini PetscErrorCode ierr; 5842b510759SStefano Zampini 5852b510759SStefano Zampini PetscFunctionBegin; 5862b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 587b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 588b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 5892b510759SStefano Zampini PetscFunctionReturn(0); 5902b510759SStefano Zampini } 5911e6b0712SBarry Smith 5922b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 5934fad6a16SStefano Zampini { 5944fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5954fad6a16SStefano Zampini 5964fad6a16SStefano Zampini PetscFunctionBegin; 5972b510759SStefano Zampini pcbddc->current_level = level; 5984fad6a16SStefano Zampini PetscFunctionReturn(0); 5994fad6a16SStefano Zampini } 6001e6b0712SBarry Smith 601b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 602b8ffe317SStefano Zampini { 603b8ffe317SStefano Zampini PetscErrorCode ierr; 604b8ffe317SStefano Zampini 605b8ffe317SStefano Zampini PetscFunctionBegin; 606b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 607b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 608b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 609b8ffe317SStefano Zampini PetscFunctionReturn(0); 610b8ffe317SStefano Zampini } 611b8ffe317SStefano Zampini 6122b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 6132b510759SStefano Zampini { 6142b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 6152b510759SStefano Zampini 6162b510759SStefano Zampini PetscFunctionBegin; 617*e569e4e1SStefano Zampini if (levels > PETSC_PCBDDC_MAXLEVELS-1) SETERRQ1(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of additional levels for BDDC is %d\n",PETSC_PCBDDC_MAXLEVELS-1); 6182b510759SStefano Zampini pcbddc->max_levels = levels; 6192b510759SStefano Zampini PetscFunctionReturn(0); 6202b510759SStefano Zampini } 6212b510759SStefano Zampini 6224fad6a16SStefano Zampini /*@ 62337ebbdf7SStefano Zampini PCBDDCSetLevels - Sets the maximum number of additional levels allowed for multilevel BDDC 6244fad6a16SStefano Zampini 6254fad6a16SStefano Zampini Logically collective on PC 6264fad6a16SStefano Zampini 6274fad6a16SStefano Zampini Input Parameters: 6284fad6a16SStefano Zampini + pc - the preconditioning context 62937ebbdf7SStefano Zampini - levels - the maximum number of levels 6304fad6a16SStefano Zampini 6310f202f7eSStefano Zampini Options Database Keys: 6320f202f7eSStefano Zampini . -pc_bddc_levels 6334fad6a16SStefano Zampini 6344fad6a16SStefano Zampini Level: intermediate 6354fad6a16SStefano Zampini 6364fad6a16SStefano Zampini Notes: 63737ebbdf7SStefano Zampini The default value is 0, that gives the classical two-levels BDDC 6384fad6a16SStefano Zampini 6390f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 6404fad6a16SStefano Zampini @*/ 6412b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 6424fad6a16SStefano Zampini { 6434fad6a16SStefano Zampini PetscErrorCode ierr; 6444fad6a16SStefano Zampini 6454fad6a16SStefano Zampini PetscFunctionBegin; 6464fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6472b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 6482b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 6494fad6a16SStefano Zampini PetscFunctionReturn(0); 6504fad6a16SStefano Zampini } 6511e6b0712SBarry Smith 6523b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 6533b03a366Sstefano_zampini { 6543b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 65556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6563b03a366Sstefano_zampini PetscErrorCode ierr; 6573b03a366Sstefano_zampini 6583b03a366Sstefano_zampini PetscFunctionBegin; 65956282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 66056282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 66156282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 66256282151SStefano Zampini } 663785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 664785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6653b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 66636e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 66756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6683b03a366Sstefano_zampini PetscFunctionReturn(0); 6693b03a366Sstefano_zampini } 6701e6b0712SBarry Smith 6713b03a366Sstefano_zampini /*@ 67228509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 6733b03a366Sstefano_zampini 674785d1243SStefano Zampini Collective 6753b03a366Sstefano_zampini 6763b03a366Sstefano_zampini Input Parameters: 6773b03a366Sstefano_zampini + pc - the preconditioning context 678785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 6793b03a366Sstefano_zampini 6803b03a366Sstefano_zampini Level: intermediate 6813b03a366Sstefano_zampini 6820f202f7eSStefano Zampini Notes: 6830f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 6843b03a366Sstefano_zampini 6850f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 6863b03a366Sstefano_zampini @*/ 6873b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 6883b03a366Sstefano_zampini { 6893b03a366Sstefano_zampini PetscErrorCode ierr; 6903b03a366Sstefano_zampini 6913b03a366Sstefano_zampini PetscFunctionBegin; 6923b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 693674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 694785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 6953b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6963b03a366Sstefano_zampini PetscFunctionReturn(0); 6973b03a366Sstefano_zampini } 6981e6b0712SBarry Smith 69982ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 7003b03a366Sstefano_zampini { 7013b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 70256282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7033b03a366Sstefano_zampini PetscErrorCode ierr; 7043b03a366Sstefano_zampini 7053b03a366Sstefano_zampini PetscFunctionBegin; 70656282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 70756282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 70856282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 70956282151SStefano Zampini } 710785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 711785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 7123b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 713785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 71456282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7153b03a366Sstefano_zampini PetscFunctionReturn(0); 7163b03a366Sstefano_zampini } 7173b03a366Sstefano_zampini 7183b03a366Sstefano_zampini /*@ 71982ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 7203b03a366Sstefano_zampini 721785d1243SStefano Zampini Collective 7223b03a366Sstefano_zampini 7233b03a366Sstefano_zampini Input Parameters: 7243b03a366Sstefano_zampini + pc - the preconditioning context 72582ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 7263b03a366Sstefano_zampini 7273b03a366Sstefano_zampini Level: intermediate 7283b03a366Sstefano_zampini 7293b03a366Sstefano_zampini Notes: 7303b03a366Sstefano_zampini 7310f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 7323b03a366Sstefano_zampini @*/ 73382ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 7343b03a366Sstefano_zampini { 7353b03a366Sstefano_zampini PetscErrorCode ierr; 7363b03a366Sstefano_zampini 7373b03a366Sstefano_zampini PetscFunctionBegin; 7383b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7393b03a366Sstefano_zampini 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); 7423b03a366Sstefano_zampini PetscFunctionReturn(0); 7433b03a366Sstefano_zampini } 7443b03a366Sstefano_zampini 74553cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 7460c7d97c5SJed Brown { 7470c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 74856282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 74953cdbc3dSStefano Zampini PetscErrorCode ierr; 7500c7d97c5SJed Brown 7510c7d97c5SJed Brown PetscFunctionBegin; 75256282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 75356282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 75456282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 75556282151SStefano Zampini } 756785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 757785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 75853cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 75936e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 76056282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7610c7d97c5SJed Brown PetscFunctionReturn(0); 7620c7d97c5SJed Brown } 7631e6b0712SBarry Smith 76457527edcSJed Brown /*@ 76528509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 76657527edcSJed Brown 767785d1243SStefano Zampini Collective 76857527edcSJed Brown 76957527edcSJed Brown Input Parameters: 77057527edcSJed Brown + pc - the preconditioning context 771785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 77257527edcSJed Brown 77357527edcSJed Brown Level: intermediate 77457527edcSJed Brown 7750f202f7eSStefano Zampini Notes: 7760f202f7eSStefano Zampini Any process can list any global node 77757527edcSJed Brown 7780f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 77957527edcSJed Brown @*/ 78053cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 7810c7d97c5SJed Brown { 7820c7d97c5SJed Brown PetscErrorCode ierr; 7830c7d97c5SJed Brown 7840c7d97c5SJed Brown PetscFunctionBegin; 7850c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 786674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 787785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 78853cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 78953cdbc3dSStefano Zampini PetscFunctionReturn(0); 79053cdbc3dSStefano Zampini } 7911e6b0712SBarry Smith 79282ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 7930c7d97c5SJed Brown { 7940c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79556282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7960c7d97c5SJed Brown PetscErrorCode ierr; 7970c7d97c5SJed Brown 7980c7d97c5SJed Brown PetscFunctionBegin; 79956282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 80056282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 80156282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 80256282151SStefano Zampini } 803785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 804785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 8050c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 806785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 80756282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 8080c7d97c5SJed Brown PetscFunctionReturn(0); 8090c7d97c5SJed Brown } 8100c7d97c5SJed Brown 8110c7d97c5SJed Brown /*@ 81282ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 8130c7d97c5SJed Brown 814785d1243SStefano Zampini Collective 8150c7d97c5SJed Brown 8160c7d97c5SJed Brown Input Parameters: 8170c7d97c5SJed Brown + pc - the preconditioning context 81882ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 8190c7d97c5SJed Brown 8200c7d97c5SJed Brown Level: intermediate 8210c7d97c5SJed Brown 8220c7d97c5SJed Brown Notes: 8230c7d97c5SJed Brown 8240f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 8250c7d97c5SJed Brown @*/ 82682ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 8270c7d97c5SJed Brown { 8280c7d97c5SJed Brown PetscErrorCode ierr; 8290c7d97c5SJed Brown 8300c7d97c5SJed Brown PetscFunctionBegin; 8310c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 8320c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 83382ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 83482ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 83553cdbc3dSStefano Zampini PetscFunctionReturn(0); 83653cdbc3dSStefano Zampini } 83753cdbc3dSStefano Zampini 838da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 839da1bb401SStefano Zampini { 840da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 841da1bb401SStefano Zampini 842da1bb401SStefano Zampini PetscFunctionBegin; 843da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 844da1bb401SStefano Zampini PetscFunctionReturn(0); 845da1bb401SStefano Zampini } 8461e6b0712SBarry Smith 847da1bb401SStefano Zampini /*@ 848785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 849da1bb401SStefano Zampini 850785d1243SStefano Zampini Collective 851785d1243SStefano Zampini 852785d1243SStefano Zampini Input Parameters: 853785d1243SStefano Zampini . pc - the preconditioning context 854785d1243SStefano Zampini 855785d1243SStefano Zampini Output Parameters: 856785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 857785d1243SStefano Zampini 858785d1243SStefano Zampini Level: intermediate 859785d1243SStefano Zampini 8600f202f7eSStefano Zampini Notes: 8610f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 862785d1243SStefano Zampini 863785d1243SStefano Zampini .seealso: PCBDDC 864785d1243SStefano Zampini @*/ 865785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 866785d1243SStefano Zampini { 867785d1243SStefano Zampini PetscErrorCode ierr; 868785d1243SStefano Zampini 869785d1243SStefano Zampini PetscFunctionBegin; 870785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 871785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 872785d1243SStefano Zampini PetscFunctionReturn(0); 873785d1243SStefano Zampini } 874785d1243SStefano Zampini 875785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 876785d1243SStefano Zampini { 877785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 878785d1243SStefano Zampini 879785d1243SStefano Zampini PetscFunctionBegin; 880785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 881785d1243SStefano Zampini PetscFunctionReturn(0); 882785d1243SStefano Zampini } 883785d1243SStefano Zampini 884da1bb401SStefano Zampini /*@ 88582ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 886da1bb401SStefano Zampini 887785d1243SStefano Zampini Collective 888da1bb401SStefano Zampini 889da1bb401SStefano Zampini Input Parameters: 89028509bceSStefano Zampini . pc - the preconditioning context 891da1bb401SStefano Zampini 892da1bb401SStefano Zampini Output Parameters: 89328509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 894da1bb401SStefano Zampini 895da1bb401SStefano Zampini Level: intermediate 896da1bb401SStefano Zampini 897da1bb401SStefano Zampini Notes: 8980f202f7eSStefano 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). 8990f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 900da1bb401SStefano Zampini 901da1bb401SStefano Zampini .seealso: PCBDDC 902da1bb401SStefano Zampini @*/ 90382ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 904da1bb401SStefano Zampini { 905da1bb401SStefano Zampini PetscErrorCode ierr; 906da1bb401SStefano Zampini 907da1bb401SStefano Zampini PetscFunctionBegin; 908da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 90982ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 910da1bb401SStefano Zampini PetscFunctionReturn(0); 911da1bb401SStefano Zampini } 9121e6b0712SBarry Smith 91353cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 91453cdbc3dSStefano Zampini { 91553cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 91653cdbc3dSStefano Zampini 91753cdbc3dSStefano Zampini PetscFunctionBegin; 91853cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 91953cdbc3dSStefano Zampini PetscFunctionReturn(0); 92053cdbc3dSStefano Zampini } 9211e6b0712SBarry Smith 92253cdbc3dSStefano Zampini /*@ 923785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 92453cdbc3dSStefano Zampini 925785d1243SStefano Zampini Collective 926785d1243SStefano Zampini 927785d1243SStefano Zampini Input Parameters: 928785d1243SStefano Zampini . pc - the preconditioning context 929785d1243SStefano Zampini 930785d1243SStefano Zampini Output Parameters: 931785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 932785d1243SStefano Zampini 933785d1243SStefano Zampini Level: intermediate 934785d1243SStefano Zampini 9350f202f7eSStefano Zampini Notes: 9360f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 937785d1243SStefano Zampini 938785d1243SStefano Zampini .seealso: PCBDDC 939785d1243SStefano Zampini @*/ 940785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 941785d1243SStefano Zampini { 942785d1243SStefano Zampini PetscErrorCode ierr; 943785d1243SStefano Zampini 944785d1243SStefano Zampini PetscFunctionBegin; 945785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 946785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 947785d1243SStefano Zampini PetscFunctionReturn(0); 948785d1243SStefano Zampini } 949785d1243SStefano Zampini 950785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 951785d1243SStefano Zampini { 952785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 953785d1243SStefano Zampini 954785d1243SStefano Zampini PetscFunctionBegin; 955785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 956785d1243SStefano Zampini PetscFunctionReturn(0); 957785d1243SStefano Zampini } 958785d1243SStefano Zampini 95953cdbc3dSStefano Zampini /*@ 96082ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 96153cdbc3dSStefano Zampini 962785d1243SStefano Zampini Collective 96353cdbc3dSStefano Zampini 96453cdbc3dSStefano Zampini Input Parameters: 96528509bceSStefano Zampini . pc - the preconditioning context 96653cdbc3dSStefano Zampini 96753cdbc3dSStefano Zampini Output Parameters: 96828509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 96953cdbc3dSStefano Zampini 97053cdbc3dSStefano Zampini Level: intermediate 97153cdbc3dSStefano Zampini 97253cdbc3dSStefano Zampini Notes: 9730f202f7eSStefano 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). 9740f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 97553cdbc3dSStefano Zampini 97653cdbc3dSStefano Zampini .seealso: PCBDDC 97753cdbc3dSStefano Zampini @*/ 97882ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 97953cdbc3dSStefano Zampini { 98053cdbc3dSStefano Zampini PetscErrorCode ierr; 98153cdbc3dSStefano Zampini 98253cdbc3dSStefano Zampini PetscFunctionBegin; 98353cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 98482ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 9850c7d97c5SJed Brown PetscFunctionReturn(0); 9860c7d97c5SJed Brown } 9871e6b0712SBarry Smith 9881a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 98936e030ebSStefano Zampini { 99036e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 991da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 99256282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 993da1bb401SStefano Zampini PetscErrorCode ierr; 99436e030ebSStefano Zampini 99536e030ebSStefano Zampini PetscFunctionBegin; 9968687889aSStefano Zampini if (!nvtxs) { 99704194a47SStefano Zampini if (copymode == PETSC_OWN_POINTER) { 99804194a47SStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 99904194a47SStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 100004194a47SStefano Zampini } 10018687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 10028687889aSStefano Zampini PetscFunctionReturn(0); 10038687889aSStefano Zampini } 100466da6bd7Sstefano_zampini if (mat_graph->nvtxs == nvtxs && mat_graph->freecsr) { /* we own the data */ 100556282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 100656282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 10072d505d7fSStefano Zampini ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 10082d505d7fSStefano Zampini if (same_data) { 10092d505d7fSStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 10102d505d7fSStefano Zampini } 101156282151SStefano Zampini } 101256282151SStefano Zampini } 101356282151SStefano Zampini if (!same_data) { 1014674ae819SStefano Zampini /* free old CSR */ 1015674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 1016674ae819SStefano Zampini /* get CSR into graph structure */ 1017da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 1018854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 1019785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 1020674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 1021674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 1022a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 1023da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 10241a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 10251a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 1026a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 1027a1dbd327SStefano Zampini } else if (copymode == PETSC_USE_POINTER) { 1028a1dbd327SStefano Zampini mat_graph->xadj = (PetscInt*)xadj; 1029a1dbd327SStefano Zampini mat_graph->adjncy = (PetscInt*)adjncy; 1030a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_FALSE; 1031e0fe2d75SToby Isaac } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %D",copymode); 1032575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 103356282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 103456282151SStefano Zampini } 103536e030ebSStefano Zampini PetscFunctionReturn(0); 103636e030ebSStefano Zampini } 10371e6b0712SBarry Smith 103836e030ebSStefano Zampini /*@ 103954fffbccSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom. 104036e030ebSStefano Zampini 104136e030ebSStefano Zampini Not collective 104236e030ebSStefano Zampini 104336e030ebSStefano Zampini Input Parameters: 104454fffbccSStefano Zampini + pc - the preconditioning context. 104554fffbccSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the number of local dofs). 104654fffbccSStefano Zampini . xadj, adjncy - the connectivity of the dofs in CSR format. 104754fffbccSStefano Zampini - copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER. 104836e030ebSStefano Zampini 104936e030ebSStefano Zampini Level: intermediate 105036e030ebSStefano Zampini 105195452b02SPatrick Sanan Notes: 105295452b02SPatrick Sanan A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative. 105336e030ebSStefano Zampini 105428509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 105536e030ebSStefano Zampini @*/ 10561a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 105736e030ebSStefano Zampini { 1058575ad6abSStefano Zampini void (*f)(void) = 0; 105936e030ebSStefano Zampini PetscErrorCode ierr; 106036e030ebSStefano Zampini 106136e030ebSStefano Zampini PetscFunctionBegin; 106236e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 10638687889aSStefano Zampini if (nvtxs) { 1064674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 10651633d1f0SStefano Zampini if (xadj[nvtxs]) PetscValidIntPointer(adjncy,4); 10668687889aSStefano Zampini } 10671a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 1068575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 1069575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 1070575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 1071575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 1072575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 1073da1bb401SStefano Zampini } 107436e030ebSStefano Zampini PetscFunctionReturn(0); 107536e030ebSStefano Zampini } 10761e6b0712SBarry Smith 107763602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 107863602bcaSStefano Zampini { 107963602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 108063602bcaSStefano Zampini PetscInt i; 108156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 108263602bcaSStefano Zampini PetscErrorCode ierr; 108363602bcaSStefano Zampini 108463602bcaSStefano Zampini PetscFunctionBegin; 108556282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 108656282151SStefano Zampini for (i=0;i<n_is;i++) { 108756282151SStefano Zampini PetscBool isequalt; 108856282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 108956282151SStefano Zampini if (!isequalt) break; 109056282151SStefano Zampini } 109156282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 109256282151SStefano Zampini } 109356282151SStefano Zampini for (i=0;i<n_is;i++) { 109456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 109556282151SStefano Zampini } 109663602bcaSStefano Zampini /* Destroy ISes if they were already set */ 109763602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 109863602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 109963602bcaSStefano Zampini } 110063602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 110163602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 110263602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 110363602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 110463602bcaSStefano Zampini } 110563602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 110663602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 110763602bcaSStefano Zampini /* allocate space then set */ 1108d02579f5SStefano Zampini if (n_is) { 1109d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 1110d02579f5SStefano Zampini } 111163602bcaSStefano Zampini for (i=0;i<n_is;i++) { 111263602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 111363602bcaSStefano Zampini } 111463602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 111563602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 111656282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 111763602bcaSStefano Zampini PetscFunctionReturn(0); 111863602bcaSStefano Zampini } 111963602bcaSStefano Zampini 112063602bcaSStefano Zampini /*@ 112163602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 112263602bcaSStefano Zampini 112363602bcaSStefano Zampini Collective 112463602bcaSStefano Zampini 112563602bcaSStefano Zampini Input Parameters: 112663602bcaSStefano Zampini + pc - the preconditioning context 11270f202f7eSStefano Zampini . n_is - number of index sets defining the fields 11280f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 112963602bcaSStefano Zampini 113063602bcaSStefano Zampini Level: intermediate 113163602bcaSStefano Zampini 11320f202f7eSStefano Zampini Notes: 11330f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 113463602bcaSStefano Zampini 113563602bcaSStefano Zampini .seealso: PCBDDC 113663602bcaSStefano Zampini @*/ 113763602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 113863602bcaSStefano Zampini { 113963602bcaSStefano Zampini PetscInt i; 114063602bcaSStefano Zampini PetscErrorCode ierr; 114163602bcaSStefano Zampini 114263602bcaSStefano Zampini PetscFunctionBegin; 114363602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 114463602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 114563602bcaSStefano Zampini for (i=0;i<n_is;i++) { 114663602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 114763602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 114863602bcaSStefano Zampini } 1149e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 115063602bcaSStefano Zampini PetscFunctionReturn(0); 115163602bcaSStefano Zampini } 115263602bcaSStefano Zampini 11539c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 11549c0446d6SStefano Zampini { 11559c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 11569c0446d6SStefano Zampini PetscInt i; 115756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 11589c0446d6SStefano Zampini PetscErrorCode ierr; 11599c0446d6SStefano Zampini 11609c0446d6SStefano Zampini PetscFunctionBegin; 116156282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 116256282151SStefano Zampini for (i=0;i<n_is;i++) { 116356282151SStefano Zampini PetscBool isequalt; 116456282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 116556282151SStefano Zampini if (!isequalt) break; 116656282151SStefano Zampini } 116756282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 116856282151SStefano Zampini } 116956282151SStefano Zampini for (i=0;i<n_is;i++) { 117056282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 117156282151SStefano Zampini } 1172da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 11739c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 11749c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 11759c0446d6SStefano Zampini } 1176d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 117763602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 117863602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 117963602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 118063602bcaSStefano Zampini } 118163602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 118263602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1183da1bb401SStefano Zampini /* allocate space then set */ 1184d02579f5SStefano Zampini if (n_is) { 1185785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1186d02579f5SStefano Zampini } 11879c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1188da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 11899c0446d6SStefano Zampini } 11909c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 119163602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 119256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 11939c0446d6SStefano Zampini PetscFunctionReturn(0); 11949c0446d6SStefano Zampini } 11951e6b0712SBarry Smith 11969c0446d6SStefano Zampini /*@ 119763602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 11989c0446d6SStefano Zampini 119963602bcaSStefano Zampini Collective 12009c0446d6SStefano Zampini 12019c0446d6SStefano Zampini Input Parameters: 12029c0446d6SStefano Zampini + pc - the preconditioning context 12030f202f7eSStefano Zampini . n_is - number of index sets defining the fields 12040f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 12059c0446d6SStefano Zampini 12069c0446d6SStefano Zampini Level: intermediate 12079c0446d6SStefano Zampini 12080f202f7eSStefano Zampini Notes: 12090f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 12109c0446d6SStefano Zampini 12119c0446d6SStefano Zampini .seealso: PCBDDC 12129c0446d6SStefano Zampini @*/ 12139c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 12149c0446d6SStefano Zampini { 12152b510759SStefano Zampini PetscInt i; 12169c0446d6SStefano Zampini PetscErrorCode ierr; 12179c0446d6SStefano Zampini 12189c0446d6SStefano Zampini PetscFunctionBegin; 12199c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 122063602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 12212b510759SStefano Zampini for (i=0;i<n_is;i++) { 122263602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 1223a011d5a7Sstefano_zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 12242b510759SStefano Zampini } 12259c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 12269c0446d6SStefano Zampini PetscFunctionReturn(0); 12279c0446d6SStefano Zampini } 1228906d46d4SStefano Zampini 1229534831adSStefano Zampini /* 1230534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1231534831adSStefano Zampini guess if a transformation of basis approach has been selected. 12329c0446d6SStefano Zampini 1233534831adSStefano Zampini Input Parameter: 1234534831adSStefano Zampini + pc - the preconditioner contex 1235534831adSStefano Zampini 1236534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1237534831adSStefano Zampini 1238534831adSStefano Zampini Notes: 1239534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1240534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1241534831adSStefano Zampini */ 1242534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1243534831adSStefano Zampini { 1244534831adSStefano Zampini PetscErrorCode ierr; 1245534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1246534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 12473972b0daSStefano Zampini Vec used_vec; 124827b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1249534831adSStefano Zampini 1250534831adSStefano Zampini PetscFunctionBegin; 12511f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 125285c4d303SStefano Zampini if (ksp) { 12531f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 125485c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 125527b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 125627b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1257f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 12581f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 125985c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 126085c4d303SStefano Zampini } 126185c4d303SStefano Zampini } 1262fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1263fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1264fc17d649SStefano Zampini } 12651f4df5f7SStefano Zampini 126685c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 126762a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 126862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 126962a6ff1dSStefano Zampini } 127062a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 127162a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 127262a6ff1dSStefano Zampini } 12738d00608fSStefano Zampini 127427b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 12753972b0daSStefano Zampini if (x) { 12763972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 12773972b0daSStefano Zampini used_vec = x; 12788d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 12793972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 12803972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 12813972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 128227b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1283266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1284266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1285266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 12863972b0daSStefano Zampini } 12878efcfb23SStefano Zampini 12888efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 12893972b0daSStefano Zampini if (ksp) { 1290a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 12918efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12928efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 12933972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 12943972b0daSStefano Zampini } 12953972b0daSStefano Zampini } 12963308cffdSStefano Zampini 12978d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12983972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 129970c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 13003975b054SStefano Zampini IS dirIS = NULL; 13013975b054SStefano Zampini 1302a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 13033975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 13043975b054SStefano Zampini if (dirIS) { 1305906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1306785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 13072b095fd8SStefano Zampini PetscScalar *array_x; 13082b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1309785d1243SStefano Zampini 13103972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 13113972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1312e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1313e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1314e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1315e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 131682ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 13173972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 13182b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 13193972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 13202fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 13213972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 13222b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 13233972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1324e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1325e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13268d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13271b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 13288efcfb23SStefano Zampini } 1329a07ea27aSStefano Zampini } 1330b76ba322SStefano Zampini 13318efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 13328d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 133327b6a85dSStefano Zampini /* save the original rhs */ 133427b6a85dSStefano Zampini if (save_rhs) { 133527b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 133627b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 13378d00608fSStefano Zampini } 13388d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13393972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 134027b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 13413972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 13428efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 134327b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 13447acc28cbSStefano Zampini if (ksp) { 13457acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 13467acc28cbSStefano Zampini } 13473308cffdSStefano Zampini } 13488efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1349b76ba322SStefano Zampini 1350fc17d649SStefano Zampini /* compute initial vector in benign space if needed 135127b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 135227b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 1353c69e9cc1SStefano Zampini if (rhs && pcbddc->benign_compute_correction && (pcbddc->benign_have_null || pcbddc->benign_apply_coarse_only)) { 13541f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 13551f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 13560369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 13570369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 13580369aaf7SStefano Zampini } 1359c69e9cc1SStefano Zampini /* keep applying coarse solver unless we no longer have benign subdomains */ 1360c69e9cc1SStefano Zampini pcbddc->benign_apply_coarse_only = pcbddc->benign_have_null ? PETSC_TRUE : PETSC_FALSE; 136127b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 13621dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 13633bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 13641f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 13651f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13661f4df5f7SStefano Zampini } 13671f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 136827b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 136927b6a85dSStefano Zampini if (save_rhs) { 137027b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 137192e3dcfbSStefano Zampini } 137227b6a85dSStefano Zampini if (pcbddc->rhs_change) { 13730369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 137427b6a85dSStefano Zampini } else { 137527b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 137627b6a85dSStefano Zampini } 13770369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 137827b6a85dSStefano Zampini } 137927b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 13800369aaf7SStefano Zampini } 13812d4c4fecSStefano Zampini 13822d4c4fecSStefano Zampini /* dbg output */ 1383a198735bSStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 13841f4df5f7SStefano Zampini Vec v; 1385c69e9cc1SStefano Zampini 13861f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 1387c69e9cc1SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 13881f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 1389c69e9cc1SStefano Zampini } else { 1390c69e9cc1SStefano Zampini ierr = VecCopy(rhs,v);CHKERRQ(ierr); 1391c69e9cc1SStefano Zampini } 13921f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1393e0fe2d75SToby Isaac ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %D: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr); 1394c69e9cc1SStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,pcbddc->dbg_viewer);CHKERRQ(ierr); 1395c69e9cc1SStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 13961f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 13971f4df5f7SStefano Zampini } 13980369aaf7SStefano Zampini 13990369aaf7SStefano Zampini /* set initial guess if using PCG */ 14008ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 14010369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 14020369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 14031dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 140427b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 14051dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 14061dd7afcfSStefano Zampini } else { 14071dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 14081dd7afcfSStefano Zampini } 14091dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14101dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14111dd7afcfSStefano Zampini } else { 14120369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14130369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14141dd7afcfSStefano Zampini } 14150369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 14161dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 14171dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 14181dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14191dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14201dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 14211dd7afcfSStefano Zampini } else { 14220369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14230369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14241dd7afcfSStefano Zampini } 14250369aaf7SStefano Zampini if (ksp) { 14260369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 14270369aaf7SStefano Zampini } 14288ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_TRUE; 1429266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1430266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 14310369aaf7SStefano Zampini } 1432534831adSStefano Zampini PetscFunctionReturn(0); 1433534831adSStefano Zampini } 1434906d46d4SStefano Zampini 1435534831adSStefano Zampini /* 1436534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1437534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1438534831adSStefano Zampini 1439534831adSStefano Zampini Input Parameter: 1440534831adSStefano Zampini + pc - the preconditioner contex 1441534831adSStefano Zampini 1442534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1443534831adSStefano Zampini 1444534831adSStefano Zampini Notes: 1445534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1446534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1447534831adSStefano Zampini */ 1448534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1449534831adSStefano Zampini { 1450534831adSStefano Zampini PetscErrorCode ierr; 1451534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1452534831adSStefano Zampini 1453534831adSStefano Zampini PetscFunctionBegin; 14543972b0daSStefano Zampini /* add solution removed in presolve */ 14556bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 145627b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 14573425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1458af140850Sstefano_zampini } else if (pcbddc->benign_compute_correction && pcbddc->benign_vec) { 145927b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14603425bc38SStefano Zampini } 1461af140850Sstefano_zampini /* restore to original state (not for FETI-DP) */ 1462af140850Sstefano_zampini if (ksp) pcbddc->temp_solution_used = PETSC_FALSE; 146327b6a85dSStefano Zampini } 146427b6a85dSStefano Zampini 1465266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 14668d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 146727b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 14688d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 1469af140850Sstefano_zampini } 14708efcfb23SStefano Zampini /* restore ksp guess state */ 14718efcfb23SStefano Zampini if (ksp) { 14728efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 14738ae0ca82SStefano Zampini /* reset flag for exact dirichlet trick */ 14748ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 1475af140850Sstefano_zampini } 1476534831adSStefano Zampini PetscFunctionReturn(0); 1477534831adSStefano Zampini } 1478af140850Sstefano_zampini 14790c7d97c5SJed Brown /* 14800c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 14810c7d97c5SJed Brown by setting data structures and options. 14820c7d97c5SJed Brown 14830c7d97c5SJed Brown Input Parameter: 148453cdbc3dSStefano Zampini + pc - the preconditioner context 14850c7d97c5SJed Brown 14860c7d97c5SJed Brown Application Interface Routine: PCSetUp() 14870c7d97c5SJed Brown 14880c7d97c5SJed Brown Notes: 14890c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 14900c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 14910c7d97c5SJed Brown */ 149253cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 14930c7d97c5SJed Brown { 14940c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1495c703fcc7SStefano Zampini PCBDDCSubSchurs sub_schurs; 14965e8657edSStefano Zampini Mat_IS* matis; 149708122e43SStefano Zampini MatNullSpace nearnullspace; 149835509ce9Sstefano_zampini Mat lA; 149935509ce9Sstefano_zampini IS lP,zerodiag = NULL; 150091e8d312SStefano Zampini PetscInt nrows,ncols; 150186bfa4cfSStefano Zampini PetscMPIInt size; 1502c703fcc7SStefano Zampini PetscBool computesubschurs; 15038de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 15045e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 1505c703fcc7SStefano Zampini PetscErrorCode ierr; 15060c7d97c5SJed Brown 15070c7d97c5SJed Brown PetscFunctionBegin; 15085e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 15096c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 151091e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 15116c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 151286bfa4cfSStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 151386bfa4cfSStefano Zampini 15145e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1515f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 15163b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 151771582508SStefano Zampini Also, BDDC builds its own KSP for the Dirichlet problem */ 1518a485753aSstefano_zampini if (!pc->setupcalled || pc->flag == DIFFERENT_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE; 1519c83e1ba7SStefano Zampini if (pcbddc->recompute_topography) { 1520c83e1ba7SStefano Zampini pcbddc->graphanalyzed = PETSC_FALSE; 1521c83e1ba7SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1522c83e1ba7SStefano Zampini } else { 15238de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1524c83e1ba7SStefano Zampini } 1525b087196eSStefano Zampini 1526b087196eSStefano Zampini /* check parameters' compatibility */ 1527b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 1528bd2a564bSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold[0] != 0.0 || pcbddc->adaptive_threshold[1] != 0.0); 152986bfa4cfSStefano Zampini pcbddc->use_deluxe_scaling = (PetscBool)(pcbddc->use_deluxe_scaling && size > 1); 153086bfa4cfSStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_selection && size > 1); 1531bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1532862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1533862806e4SStefano Zampini 15345a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 153516909a7fSStefano Zampini if (pcbddc->switch_static) { 153616909a7fSStefano Zampini PetscBool ismatis; 1537d9869140SStefano Zampini 153816909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 153916909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 154016909a7fSStefano Zampini } 154116909a7fSStefano Zampini 154271582508SStefano Zampini /* activate all connected components if the netflux has been requested */ 1543bb05f991SStefano Zampini if (pcbddc->compute_nonetflux) { 1544bb05f991SStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 1545bb05f991SStefano Zampini pcbddc->use_edges = PETSC_TRUE; 1546bb05f991SStefano Zampini pcbddc->use_faces = PETSC_TRUE; 1547bb05f991SStefano Zampini } 1548bb05f991SStefano Zampini 1549f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 155070cf5478SStefano Zampini if (pcbddc->dbg_flag) { 155170cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 155258a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 1553f4ddd8eeSStefano Zampini } 1554d9869140SStefano Zampini ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 155558a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1556f4ddd8eeSStefano Zampini } 1557f4ddd8eeSStefano Zampini 1558c703fcc7SStefano Zampini /* process topology information */ 155943371fb9SStefano Zampini ierr = PetscLogEventBegin(PC_BDDC_Topology[pcbddc->current_level],pc,0,0,0);CHKERRQ(ierr); 156071582508SStefano Zampini if (pcbddc->recompute_topography) { 156171582508SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 1562c703fcc7SStefano Zampini if (pcbddc->discretegradient) { 1563a13144ffSStefano Zampini ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr); 1564a13144ffSStefano Zampini } 1565c703fcc7SStefano Zampini } 1566a13144ffSStefano Zampini 1567c703fcc7SStefano Zampini /* change basis if requested by the user */ 15685e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 15695e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 15705e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 15715e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 15725e8657edSStefano Zampini } else { 1573b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15745e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15755e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1576d16cbb6bSStefano Zampini } 1577d16cbb6bSStefano Zampini 15784f1b2e48SStefano Zampini /* 1579c703fcc7SStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick 15804f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 15814f1b2e48SStefano Zampini */ 15821dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1583d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 15849f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 15859f47a83aSStefano Zampini 158605b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1587669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 1588339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1589a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1590c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 15911dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 15921dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 15931dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 15941dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 15951dd7afcfSStefano Zampini } else { 1596a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 15971dd7afcfSStefano Zampini } 1598a3df083aSStefano Zampini } else { 15999f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1600674ae819SStefano Zampini } 1601a3df083aSStefano Zampini } 160227b6a85dSStefano Zampini 16038037d520SStefano Zampini /* propagate relevant information */ 160406a4e24aSStefano Zampini if (matis->A->symmetric_set) { 160506a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 160606a4e24aSStefano Zampini } 160706a4e24aSStefano Zampini if (matis->A->spd_set) { 160806a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 160906a4e24aSStefano Zampini } 1610e496cd5dSStefano Zampini 16115e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 16125e8657edSStefano Zampini { 16135e8657edSStefano Zampini Mat temp_mat; 16145e8657edSStefano Zampini 16155e8657edSStefano Zampini temp_mat = matis->A; 16165e8657edSStefano Zampini matis->A = pcbddc->local_mat; 1617d9869140SStefano Zampini ierr = PCISSetUp(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 16185e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16195e8657edSStefano Zampini matis->A = temp_mat; 16205e8657edSStefano Zampini } 1621684f6988SStefano Zampini 162281d14e9dSStefano Zampini /* Analyze interface */ 162364ac59b8SStefano Zampini if (!pcbddc->graphanalyzed) { 1624674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 16258de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1626345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 16274247aa23Sstefano_zampini SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1628345ecf6cSStefano Zampini } 1629a198735bSStefano Zampini if (pcbddc->compute_nonetflux) { 1630669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1631669cc0f4SStefano Zampini 163221ef3d20SStefano Zampini if (!pcbddc->divudotp) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Missing divudotp operator"); 16338ae0ca82SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 163471582508SStefano Zampini /* TODO what if a nearnullspace is already attached? */ 16358037d520SStefano Zampini if (nnfnnsp) { 1636669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1637669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1638669cc0f4SStefano Zampini } 1639674ae819SStefano Zampini } 16408037d520SStefano Zampini } 164143371fb9SStefano Zampini ierr = PetscLogEventEnd(PC_BDDC_Topology[pcbddc->current_level],pc,0,0,0);CHKERRQ(ierr); 1642fb8d54d4SStefano Zampini 16435408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 16445408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 16455408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 1646ff1f7e73Sstefano_zampini if (pcbddc->benign_saddle_point && pcbddc->dbg_flag > 1) { 16475408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 164809f581a4SStefano Zampini } 16494f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 165006f24817SStefano Zampini 1651b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1652c703fcc7SStefano Zampini if (computesubschurs && pcbddc->recompute_topography) { 165308122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1654b1b3d7a2SStefano Zampini } 16559d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 16569d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 16579d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16589d54b7f4SStefano Zampini } 1659c703fcc7SStefano Zampini 1660c703fcc7SStefano Zampini /* finish setup solvers and do adaptive selection of constraints */ 1661b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1662b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 16632070dbb6SStefano Zampini if (computesubschurs) { 166408122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 16652070dbb6SStefano Zampini } 1666d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1667d5574798SStefano Zampini } else { 1668d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 16692070dbb6SStefano Zampini if (computesubschurs) { 1670d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1671d5574798SStefano Zampini } 16722070dbb6SStefano Zampini } 167308122e43SStefano Zampini if (pcbddc->adaptive_selection) { 167408122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 16758de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1676b7eb3628SStefano Zampini } 1677684f6988SStefano Zampini 1678f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1679fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1680f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1681f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1682f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1683f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1684f4ddd8eeSStefano Zampini } else { 1685f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1686f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1687f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1688165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1689f4ddd8eeSStefano Zampini PetscInt i; 1690165b64e2SStefano Zampini const Vec *nearnullvecs; 1691165b64e2SStefano Zampini PetscObjectState state; 1692165b64e2SStefano Zampini PetscInt nnsp_size; 1693165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1694f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1695f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1696165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1697f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1698f4ddd8eeSStefano Zampini break; 1699f4ddd8eeSStefano Zampini } 1700f4ddd8eeSStefano Zampini } 1701f4ddd8eeSStefano Zampini } 1702f4ddd8eeSStefano Zampini } 1703f4ddd8eeSStefano Zampini } else { 1704f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1705f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1706f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1707f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1708f4ddd8eeSStefano Zampini } 1709f4ddd8eeSStefano Zampini } 1710f4ddd8eeSStefano Zampini 1711f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1712727cdba6SStefano Zampini /* reset primal space flags */ 171343371fb9SStefano Zampini ierr = PetscLogEventBegin(PC_BDDC_LocalWork[pcbddc->current_level],pc,0,0,0);CHKERRQ(ierr); 1714f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1715727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 17168de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1717727cdba6SStefano Zampini /* It also sets the primal space flags */ 1718674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 17199543d0ffSStefano Zampini } 1720e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1721f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 17225e8657edSStefano Zampini 17235e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 17245e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 17255e8657edSStefano Zampini 17265e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 17274f1b2e48SStefano Zampini if (pcbddc->benign_change) { 17281dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1729c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1730c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1731c263805aSStefano Zampini } 17325e8657edSStefano Zampini /* get submatrices */ 17335e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 17345e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 17355e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 17367dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 17377dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 17387dae84e0SHong Zhang ierr = MatCreateSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 17393975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 17403975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 17419c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1742b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 17435e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 17445e8657edSStefano Zampini pcbddc->local_mat = matis->A; 17455e8657edSStefano Zampini } 174635509ce9Sstefano_zampini 174735509ce9Sstefano_zampini /* interface pressure block row for B_C */ 174835509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lP" ,(PetscObject*)&lP);CHKERRQ(ierr); 174935509ce9Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_lA" ,(PetscObject*)&lA);CHKERRQ(ierr); 175035509ce9Sstefano_zampini if (lA && lP) { 175135509ce9Sstefano_zampini PC_IS* pcis = (PC_IS*)pc->data; 175235509ce9Sstefano_zampini Mat B_BI,B_BB,Bt_BI,Bt_BB; 175335509ce9Sstefano_zampini PetscBool issym; 175435509ce9Sstefano_zampini ierr = MatIsSymmetric(lA,PETSC_SMALL,&issym);CHKERRQ(ierr); 17556cc1294bSstefano_zampini if (issym) { 17567dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17577dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 175835509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BI,&Bt_BI);CHKERRQ(ierr); 175935509ce9Sstefano_zampini ierr = MatCreateTranspose(B_BB,&Bt_BB);CHKERRQ(ierr); 176035509ce9Sstefano_zampini } else { 17617dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_I_local,MAT_INITIAL_MATRIX,&B_BI);CHKERRQ(ierr); 17627dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,lP,pcis->is_B_local,MAT_INITIAL_MATRIX,&B_BB);CHKERRQ(ierr); 17637dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_I_local,lP,MAT_INITIAL_MATRIX,&Bt_BI);CHKERRQ(ierr); 17647dae84e0SHong Zhang ierr = MatCreateSubMatrix(lA,pcis->is_B_local,lP,MAT_INITIAL_MATRIX,&Bt_BB);CHKERRQ(ierr); 176535509ce9Sstefano_zampini } 176635509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BI",(PetscObject)B_BI);CHKERRQ(ierr); 176735509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_B_BB",(PetscObject)B_BB);CHKERRQ(ierr); 176835509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BI",(PetscObject)Bt_BI);CHKERRQ(ierr); 176935509ce9Sstefano_zampini ierr = PetscObjectCompose((PetscObject)pc,"__KSPFETIDP_Bt_BB",(PetscObject)Bt_BB);CHKERRQ(ierr); 177035509ce9Sstefano_zampini ierr = MatDestroy(&B_BI);CHKERRQ(ierr); 177135509ce9Sstefano_zampini ierr = MatDestroy(&B_BB);CHKERRQ(ierr); 177235509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BI);CHKERRQ(ierr); 177335509ce9Sstefano_zampini ierr = MatDestroy(&Bt_BB);CHKERRQ(ierr); 177435509ce9Sstefano_zampini } 177543371fb9SStefano Zampini ierr = PetscLogEventEnd(PC_BDDC_LocalWork[pcbddc->current_level],pc,0,0,0);CHKERRQ(ierr); 177635509ce9Sstefano_zampini 1777b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 177899cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1779b96c3477SStefano Zampini /* SetUp Scaling operator */ 17809d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1781674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 17820c7d97c5SJed Brown } 1783c703fcc7SStefano Zampini 17841dd7afcfSStefano Zampini /* mark topography as done */ 178556282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 17860369aaf7SStefano Zampini 17871dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 17881dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 17891dd7afcfSStefano Zampini 179058a03d70SStefano Zampini if (pcbddc->dbg_flag) { 179158a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1792d9869140SStefano Zampini ierr = PetscViewerASCIIPopSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 17932b510759SStefano Zampini } 17940c7d97c5SJed Brown PetscFunctionReturn(0); 17950c7d97c5SJed Brown } 17960c7d97c5SJed Brown 17970c7d97c5SJed Brown /* 179850efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 17990c7d97c5SJed Brown 18000c7d97c5SJed Brown Input Parameters: 18010f202f7eSStefano Zampini + pc - the preconditioner context 18020f202f7eSStefano Zampini - r - input vector (global) 18030c7d97c5SJed Brown 18040c7d97c5SJed Brown Output Parameter: 18050c7d97c5SJed Brown . z - output vector (global) 18060c7d97c5SJed Brown 18070c7d97c5SJed Brown Application Interface Routine: PCApply() 18080c7d97c5SJed Brown */ 180953cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 18100c7d97c5SJed Brown { 18110c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 18120c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1813b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 18140c7d97c5SJed Brown PetscErrorCode ierr; 18153b03a366Sstefano_zampini const PetscScalar one = 1.0; 18163b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 18172617d88aSStefano Zampini const PetscScalar zero = 0.0; 18180c7d97c5SJed Brown 18190c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 18200c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1821b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 18220c7d97c5SJed Brown 18230c7d97c5SJed Brown PetscFunctionBegin; 1824f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 18251dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18261dd7afcfSStefano Zampini Vec swap; 182727b6a85dSStefano Zampini 182827b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 18291dd7afcfSStefano Zampini swap = pcbddc->work_change; 18301dd7afcfSStefano Zampini pcbddc->work_change = r; 18311dd7afcfSStefano Zampini r = swap; 18321dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 18339cc2a9b1Sstefano_zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 18341dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 18351dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 18361dd7afcfSStefano Zampini } 18371dd7afcfSStefano Zampini } 183827b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1839015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1840efc2fbd9SStefano Zampini } 18418ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1842b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 18430c7d97c5SJed Brown /* First Dirichlet solve */ 18440c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18450c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18460c7d97c5SJed Brown /* 18470c7d97c5SJed Brown Assembling right hand side for BDDC operator 1848b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1849674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 18500c7d97c5SJed Brown */ 1851b097fa66SStefano Zampini if (n_D) { 1852b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 18530c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 185416909a7fSStefano Zampini if (pcbddc->switch_static) { 185516909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 185616909a7fSStefano Zampini 185716909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 185816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 185916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186016909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 186116909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186216909a7fSStefano Zampini } else { 186316909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186416909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 186516909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186616909a7fSStefano Zampini } 186716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187116909a7fSStefano Zampini } else { 1872b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 187316909a7fSStefano Zampini } 1874b097fa66SStefano Zampini } else { 1875b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1876b097fa66SStefano Zampini } 18770c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18780c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1879674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1880b76ba322SStefano Zampini } else { 18814fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1882674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1883b76ba322SStefano Zampini } 18844fee134fSStefano Zampini } 1885b76ba322SStefano Zampini 18862617d88aSStefano Zampini /* Apply interface preconditioner 18872617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1888dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 18892617d88aSStefano Zampini 1890674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1891674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 18920c7d97c5SJed Brown 18933b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 18940c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18950c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1896b097fa66SStefano Zampini if (n_B) { 189716909a7fSStefano Zampini if (pcbddc->switch_static) { 189816909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 189916909a7fSStefano Zampini 190016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,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->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191316909a7fSStefano Zampini } else { 19140c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 191516909a7fSStefano Zampini } 191616909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 191716909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 191816909a7fSStefano Zampini 191916909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 192016909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 192116909a7fSStefano Zampini } else { 192216909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 192316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 192416909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 192516909a7fSStefano Zampini } 1926b097fa66SStefano Zampini } 1927df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1928efc2fbd9SStefano Zampini 19298ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1930b097fa66SStefano Zampini if (pcbddc->switch_static) { 1931b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1932b097fa66SStefano Zampini } else { 1933b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1934b097fa66SStefano Zampini } 19350c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 19360c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1937b097fa66SStefano Zampini } else { 1938b097fa66SStefano Zampini if (pcbddc->switch_static) { 1939b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1940b097fa66SStefano Zampini } else { 1941b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1942b097fa66SStefano Zampini } 1943b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1944b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1945b097fa66SStefano Zampini } 194627b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 19471dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 19481dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 19491dd7afcfSStefano Zampini } 1950015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1951efc2fbd9SStefano Zampini } 19521f4df5f7SStefano Zampini 19531dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1954f913dca9SStefano Zampini pcbddc->work_change = r; 19551dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19561dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19571dd7afcfSStefano Zampini } 19580c7d97c5SJed Brown PetscFunctionReturn(0); 19590c7d97c5SJed Brown } 196050efa1b5SStefano Zampini 196150efa1b5SStefano Zampini /* 196250efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 196350efa1b5SStefano Zampini 196450efa1b5SStefano Zampini Input Parameters: 19650f202f7eSStefano Zampini + pc - the preconditioner context 19660f202f7eSStefano Zampini - r - input vector (global) 196750efa1b5SStefano Zampini 196850efa1b5SStefano Zampini Output Parameter: 196950efa1b5SStefano Zampini . z - output vector (global) 197050efa1b5SStefano Zampini 197150efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 197250efa1b5SStefano Zampini */ 197350efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 197450efa1b5SStefano Zampini { 197550efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 197650efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1977b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 197850efa1b5SStefano Zampini PetscErrorCode ierr; 197950efa1b5SStefano Zampini const PetscScalar one = 1.0; 198050efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 198150efa1b5SStefano Zampini const PetscScalar zero = 0.0; 198250efa1b5SStefano Zampini 198350efa1b5SStefano Zampini PetscFunctionBegin; 1984f3d41395Sstefano_zampini ierr = PetscCitationsRegister(citation,&cited);CHKERRQ(ierr); 19851dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19861dd7afcfSStefano Zampini Vec swap; 198727b6a85dSStefano Zampini 198827b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 19891dd7afcfSStefano Zampini swap = pcbddc->work_change; 19901dd7afcfSStefano Zampini pcbddc->work_change = r; 19911dd7afcfSStefano Zampini r = swap; 199227b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 19938ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 199427b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 199527b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 19961dd7afcfSStefano Zampini } 199727b6a85dSStefano Zampini } 199827b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1999537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 2000537c1cdfSStefano Zampini } 20018ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2002b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 200350efa1b5SStefano Zampini /* First Dirichlet solve */ 200450efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 200550efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 200650efa1b5SStefano Zampini /* 200750efa1b5SStefano Zampini Assembling right hand side for BDDC operator 2008b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 200950efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 201050efa1b5SStefano Zampini */ 2011b097fa66SStefano Zampini if (n_D) { 2012b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 201350efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 201416909a7fSStefano Zampini if (pcbddc->switch_static) { 201516909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 201616909a7fSStefano Zampini 201716909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 201816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 201916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202016909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 202116909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202216909a7fSStefano Zampini } else { 202316909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202416909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 202516909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202616909a7fSStefano Zampini } 202716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 202816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 202916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 203016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 203116909a7fSStefano Zampini } else { 2032b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 203316909a7fSStefano Zampini } 2034b097fa66SStefano Zampini } else { 2035b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 2036b097fa66SStefano Zampini } 203750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 203850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 203950efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 204050efa1b5SStefano Zampini } else { 204150efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 204250efa1b5SStefano Zampini } 204350efa1b5SStefano Zampini 204450efa1b5SStefano Zampini /* Apply interface preconditioner 204550efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 2046dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 204750efa1b5SStefano Zampini 204850efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 204950efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 205050efa1b5SStefano Zampini 205150efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 205250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 205350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2054b097fa66SStefano Zampini if (n_B) { 205516909a7fSStefano Zampini if (pcbddc->switch_static) { 205616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 205716909a7fSStefano Zampini 205816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 205916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 206016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 206116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 206216909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 206316909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 206416909a7fSStefano Zampini } else { 206516909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 206616909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 206716909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 206816909a7fSStefano Zampini } 206916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 207016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 207116909a7fSStefano Zampini } else { 207250efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 207316909a7fSStefano Zampini } 207416909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 207516909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 207616909a7fSStefano Zampini 207716909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 207816909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 207916909a7fSStefano Zampini } else { 208016909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 208116909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 208216909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 208316909a7fSStefano Zampini } 2084b097fa66SStefano Zampini } 2085b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 20868ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2087b097fa66SStefano Zampini if (pcbddc->switch_static) { 2088b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 2089b097fa66SStefano Zampini } else { 2090b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 2091b097fa66SStefano Zampini } 209250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 209350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2094b097fa66SStefano Zampini } else { 2095b097fa66SStefano Zampini if (pcbddc->switch_static) { 2096b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 2097b097fa66SStefano Zampini } else { 2098b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 2099b097fa66SStefano Zampini } 2100b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2101b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2102b097fa66SStefano Zampini } 210327b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 2104537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 2105537c1cdfSStefano Zampini } 21061dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2107f913dca9SStefano Zampini pcbddc->work_change = r; 21081dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 21091dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 21101dd7afcfSStefano Zampini } 211150efa1b5SStefano Zampini PetscFunctionReturn(0); 211250efa1b5SStefano Zampini } 2113674ae819SStefano Zampini 21149326c5c6Sstefano_zampini PetscErrorCode PCReset_BDDC(PC pc) 2115da1bb401SStefano Zampini { 2116da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 21179326c5c6Sstefano_zampini PC_IS *pcis = (PC_IS*)pc->data; 21189326c5c6Sstefano_zampini KSP kspD,kspR,kspC; 2119da1bb401SStefano Zampini PetscErrorCode ierr; 2120da1bb401SStefano Zampini 2121da1bb401SStefano Zampini PetscFunctionBegin; 2122674ae819SStefano Zampini /* free BDDC custom data */ 2123674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2124674ae819SStefano Zampini /* destroy objects related to topography */ 2125674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 212634a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2127674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 2128674ae819SStefano Zampini /* free solvers stuff */ 2129674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 213062a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 213162a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 213262a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 21331dd7afcfSStefano Zampini /* free data created by PCIS */ 21341dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 21359326c5c6Sstefano_zampini 21369326c5c6Sstefano_zampini /* restore defaults */ 21379326c5c6Sstefano_zampini kspD = pcbddc->ksp_D; 21389326c5c6Sstefano_zampini kspR = pcbddc->ksp_R; 21399326c5c6Sstefano_zampini kspC = pcbddc->coarse_ksp; 21409326c5c6Sstefano_zampini ierr = PetscMemzero(pc->data,sizeof(*pcbddc));CHKERRQ(ierr); 21419326c5c6Sstefano_zampini pcis->n_neigh = -1; 21429326c5c6Sstefano_zampini pcis->scaling_factor = 1.0; 21439326c5c6Sstefano_zampini pcis->reusesubmatrices = PETSC_TRUE; 21449326c5c6Sstefano_zampini pcbddc->use_local_adj = PETSC_TRUE; 21459326c5c6Sstefano_zampini pcbddc->use_vertices = PETSC_TRUE; 21469326c5c6Sstefano_zampini pcbddc->use_edges = PETSC_TRUE; 21479326c5c6Sstefano_zampini pcbddc->symmetric_primal = PETSC_TRUE; 21489326c5c6Sstefano_zampini pcbddc->vertex_size = 1; 21499326c5c6Sstefano_zampini pcbddc->recompute_topography = PETSC_TRUE; 21509326c5c6Sstefano_zampini pcbddc->coarse_size = -1; 21519326c5c6Sstefano_zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 21529326c5c6Sstefano_zampini pcbddc->coarsening_ratio = 8; 21539326c5c6Sstefano_zampini pcbddc->coarse_eqs_per_proc = 1; 21549326c5c6Sstefano_zampini pcbddc->benign_compute_correction = PETSC_TRUE; 21559326c5c6Sstefano_zampini pcbddc->nedfield = -1; 21569326c5c6Sstefano_zampini pcbddc->nedglobal = PETSC_TRUE; 21579326c5c6Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 21589326c5c6Sstefano_zampini pcbddc->sub_schurs_layers = -1; 21599326c5c6Sstefano_zampini pcbddc->ksp_D = kspD; 21609326c5c6Sstefano_zampini pcbddc->ksp_R = kspR; 21619326c5c6Sstefano_zampini pcbddc->coarse_ksp = kspC; 21629326c5c6Sstefano_zampini PetscFunctionReturn(0); 21639326c5c6Sstefano_zampini } 21649326c5c6Sstefano_zampini 21659326c5c6Sstefano_zampini PetscErrorCode PCDestroy_BDDC(PC pc) 21669326c5c6Sstefano_zampini { 21679326c5c6Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 21689326c5c6Sstefano_zampini PetscErrorCode ierr; 21699326c5c6Sstefano_zampini 21709326c5c6Sstefano_zampini PetscFunctionBegin; 21719326c5c6Sstefano_zampini ierr = PCReset_BDDC(pc);CHKERRQ(ierr); 21729326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_D);CHKERRQ(ierr); 21739326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->ksp_R);CHKERRQ(ierr); 21749326c5c6Sstefano_zampini ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr); 2175a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr); 2176a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr); 2177906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2178674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 217930368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2180bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 21812b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2182b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 21832b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2184bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 218582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2186bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 218782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2188bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 218982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2190bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2191785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2192bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 219363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2194bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2195bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2196bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2197bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2198a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2199ab8c8b98SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCSetCoordinates_C",NULL);CHKERRQ(ierr); 2200674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2201da1bb401SStefano Zampini PetscFunctionReturn(0); 2202da1bb401SStefano Zampini } 22031e6b0712SBarry Smith 2204ab8c8b98SStefano Zampini static PetscErrorCode PCSetCoordinates_BDDC(PC pc, PetscInt dim, PetscInt nloc, PetscReal *coords) 2205ab8c8b98SStefano Zampini { 2206ab8c8b98SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2207ab8c8b98SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 2208ab8c8b98SStefano Zampini PetscErrorCode ierr; 2209ab8c8b98SStefano Zampini 2210ab8c8b98SStefano Zampini PetscFunctionBegin; 2211ab8c8b98SStefano Zampini ierr = PetscFree(mat_graph->coords);CHKERRQ(ierr); 2212ab8c8b98SStefano Zampini ierr = PetscMalloc1(nloc*dim,&mat_graph->coords);CHKERRQ(ierr); 2213ab8c8b98SStefano Zampini ierr = PetscMemcpy(mat_graph->coords,coords,nloc*dim*sizeof(PetscReal));CHKERRQ(ierr); 2214ab8c8b98SStefano Zampini mat_graph->cnloc = nloc; 2215ab8c8b98SStefano Zampini mat_graph->cdim = dim; 2216ab8c8b98SStefano Zampini mat_graph->cloc = PETSC_FALSE; 2217ab8c8b98SStefano Zampini PetscFunctionReturn(0); 2218ab8c8b98SStefano Zampini } 2219ab8c8b98SStefano Zampini 2220a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2221a06fd7f2SStefano Zampini { 2222a06fd7f2SStefano Zampini PetscFunctionBegin; 2223a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2224a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2225a06fd7f2SStefano Zampini } 2226a06fd7f2SStefano Zampini 22273425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 22283425bc38SStefano Zampini { 2229674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2230266e20e9SStefano Zampini Vec work; 22313425bc38SStefano Zampini PC_IS* pcis; 22323425bc38SStefano Zampini PC_BDDC* pcbddc; 22333425bc38SStefano Zampini PetscErrorCode ierr; 22340c7d97c5SJed Brown 22353425bc38SStefano Zampini PetscFunctionBegin; 22363425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 22373425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 22383425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 22393425bc38SStefano Zampini 2240229984c5Sstefano_zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 2241229984c5Sstefano_zampini /* copy rhs since we may change it during PCPreSolve_BDDC */ 2242229984c5Sstefano_zampini if (!pcbddc->original_rhs) { 2243229984c5Sstefano_zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 2244229984c5Sstefano_zampini } 22456cc1294bSstefano_zampini if (mat_ctx->rhs_flip) { 22466cc1294bSstefano_zampini ierr = VecPointwiseMult(pcbddc->original_rhs,standard_rhs,mat_ctx->rhs_flip);CHKERRQ(ierr); 22476cc1294bSstefano_zampini } else { 2248229984c5Sstefano_zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 22496cc1294bSstefano_zampini } 2250af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2251229984c5Sstefano_zampini /* interface pressure rhs */ 2252022d8d2bSstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2253022d8d2bSstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_rhs,pcbddc->original_rhs,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2254229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2255229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,standard_rhs,fetidp_flux_rhs,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22566cc1294bSstefano_zampini if (!mat_ctx->rhs_flip) { 2257229984c5Sstefano_zampini ierr = VecScale(fetidp_flux_rhs,-1.);CHKERRQ(ierr); 2258229984c5Sstefano_zampini } 22596cc1294bSstefano_zampini } 2260c08af4c6SStefano Zampini /* 2261c08af4c6SStefano Zampini change of basis for physical rhs if needed 2262c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2263c08af4c6SStefano Zampini */ 22643738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2265fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 22663738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 22673738a8e6SStefano Zampini work = pcbddc->work_change; 2268fc17d649SStefano Zampini } else { 22693738a8e6SStefano Zampini work = pcbddc->original_rhs; 2270fc17d649SStefano Zampini } 22713425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2272266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2273266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2274fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2275fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2276266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2277266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2278674ae819SStefano Zampini /* Apply partition of unity */ 22793425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2280266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 22818eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 22823425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 22833425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22843425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 22853425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2286266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2287266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2288266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2289266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2290266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2291266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22923425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 22933425bc38SStefano Zampini } 22943425bc38SStefano Zampini /* BDDC rhs */ 22953425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 22968eeda7d8SStefano Zampini if (pcbddc->switch_static) { 22973425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22983425bc38SStefano Zampini } 22993425bc38SStefano Zampini /* apply BDDC */ 2300229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2301dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2302266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2303229984c5Sstefano_zampini 23043425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 23053425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 23063425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 23073425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2308229984c5Sstefano_zampini /* Add contribution to interface pressures */ 2309229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2310229984c5Sstefano_zampini ierr = MatMult(mat_ctx->B_BB,pcis->vec1_B,mat_ctx->vP);CHKERRQ(ierr); 2311229984c5Sstefano_zampini if (pcbddc->switch_static) { 2312229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->B_BI,pcis->vec1_D,mat_ctx->vP,mat_ctx->vP);CHKERRQ(ierr); 2313229984c5Sstefano_zampini } 2314229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2315229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,mat_ctx->vP,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2316229984c5Sstefano_zampini } 23173425bc38SStefano Zampini PetscFunctionReturn(0); 23183425bc38SStefano Zampini } 23191e6b0712SBarry Smith 23203425bc38SStefano Zampini /*@ 23210f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 23223425bc38SStefano Zampini 23233425bc38SStefano Zampini Collective 23243425bc38SStefano Zampini 23253425bc38SStefano Zampini Input Parameters: 23260f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 23270f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 23283425bc38SStefano Zampini 23293425bc38SStefano Zampini Output Parameters: 23300f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 23313425bc38SStefano Zampini 23323425bc38SStefano Zampini Level: developer 23333425bc38SStefano Zampini 23343425bc38SStefano Zampini Notes: 23353425bc38SStefano Zampini 23360f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 23373425bc38SStefano Zampini @*/ 23383425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 23393425bc38SStefano Zampini { 2340674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23413425bc38SStefano Zampini PetscErrorCode ierr; 23423425bc38SStefano Zampini 23433425bc38SStefano Zampini PetscFunctionBegin; 2344266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2345266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2346266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 23473425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2348163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 23493425bc38SStefano Zampini PetscFunctionReturn(0); 23503425bc38SStefano Zampini } 23511e6b0712SBarry Smith 23523425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23533425bc38SStefano Zampini { 2354674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23553425bc38SStefano Zampini PC_IS* pcis; 23563425bc38SStefano Zampini PC_BDDC* pcbddc; 23573425bc38SStefano Zampini PetscErrorCode ierr; 2358229984c5Sstefano_zampini Vec work; 23593425bc38SStefano Zampini 23603425bc38SStefano Zampini PetscFunctionBegin; 23613425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 23623425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 23633425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 23643425bc38SStefano Zampini 23653425bc38SStefano Zampini /* apply B_delta^T */ 2366af140850Sstefano_zampini ierr = VecSet(pcis->vec1_B,0.);CHKERRQ(ierr); 23673425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23683425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23693425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 2370229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2371229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2372229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->l2g_p,fetidp_flux_sol,mat_ctx->vP,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2373229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BB,mat_ctx->vP,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 2374229984c5Sstefano_zampini } 2375229984c5Sstefano_zampini 23763425bc38SStefano Zampini /* compute rhs for BDDC application */ 23773425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 23788eeda7d8SStefano Zampini if (pcbddc->switch_static) { 23793425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 2380229984c5Sstefano_zampini if (mat_ctx->l2g_p) { 2381229984c5Sstefano_zampini ierr = VecScale(mat_ctx->vP,-1.);CHKERRQ(ierr); 2382229984c5Sstefano_zampini ierr = MatMultAdd(mat_ctx->Bt_BI,mat_ctx->vP,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 23833425bc38SStefano Zampini } 2384229984c5Sstefano_zampini } 2385229984c5Sstefano_zampini 23863425bc38SStefano Zampini /* apply BDDC */ 2387229984c5Sstefano_zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 2388dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2389229984c5Sstefano_zampini 2390229984c5Sstefano_zampini /* put values into global vector */ 2391af140850Sstefano_zampini if (pcbddc->ChangeOfBasisMatrix) work = pcbddc->work_change; 2392af140850Sstefano_zampini else work = standard_sol; 2393229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2394229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 23958eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 23963425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 23973425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 239800f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 239900f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 24003425bc38SStefano Zampini } 2401229984c5Sstefano_zampini 2402229984c5Sstefano_zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2403229984c5Sstefano_zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,work,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2404266e20e9SStefano Zampini /* add p0 solution to final solution */ 2405229984c5Sstefano_zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,work,PETSC_FALSE);CHKERRQ(ierr); 2406fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2407af140850Sstefano_zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,work,standard_sol);CHKERRQ(ierr); 2408fc17d649SStefano Zampini } 2409af140850Sstefano_zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 2410af140850Sstefano_zampini if (mat_ctx->g2g_p) { 2411229984c5Sstefano_zampini ierr = VecScatterBegin(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2412229984c5Sstefano_zampini ierr = VecScatterEnd(mat_ctx->g2g_p,fetidp_flux_sol,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2413229984c5Sstefano_zampini } 24143425bc38SStefano Zampini PetscFunctionReturn(0); 24153425bc38SStefano Zampini } 24161e6b0712SBarry Smith 24173425bc38SStefano Zampini /*@ 24180f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 24193425bc38SStefano Zampini 24203425bc38SStefano Zampini Collective 24213425bc38SStefano Zampini 24223425bc38SStefano Zampini Input Parameters: 24230f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 24240f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 24253425bc38SStefano Zampini 24263425bc38SStefano Zampini Output Parameters: 24270f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 24283425bc38SStefano Zampini 24293425bc38SStefano Zampini Level: developer 24303425bc38SStefano Zampini 24313425bc38SStefano Zampini Notes: 24323425bc38SStefano Zampini 24330f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 24343425bc38SStefano Zampini @*/ 24353425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 24363425bc38SStefano Zampini { 2437674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 24383425bc38SStefano Zampini PetscErrorCode ierr; 24393425bc38SStefano Zampini 24403425bc38SStefano Zampini PetscFunctionBegin; 2441266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2442266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2443266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 24443425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2445163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 24463425bc38SStefano Zampini PetscFunctionReturn(0); 24473425bc38SStefano Zampini } 24481e6b0712SBarry Smith 2449547c9a8eSstefano_zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, const char* prefix, Mat *fetidp_mat, PC *fetidp_pc) 24503425bc38SStefano Zampini { 2451674ae819SStefano Zampini 2452674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 24533425bc38SStefano Zampini Mat newmat; 2454674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 24553425bc38SStefano Zampini PC newpc; 2456ce94432eSBarry Smith MPI_Comm comm; 24573425bc38SStefano Zampini PetscErrorCode ierr; 24583425bc38SStefano Zampini 24593425bc38SStefano Zampini PetscFunctionBegin; 2460ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 246115579a77SStefano Zampini /* FETI-DP matrix */ 24623425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 24631720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 24643425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 2465a5bb87b3Sstefano_zampini ierr = MatCreateShell(comm,fetidpmat_ctx->n,fetidpmat_ctx->n,fetidpmat_ctx->N,fetidpmat_ctx->N,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 246615579a77SStefano Zampini ierr = PetscObjectSetName((PetscObject)newmat,!fetidpmat_ctx->l2g_lambda_only ? "F" : "G");CHKERRQ(ierr); 24673425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2468edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 24693425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 247015579a77SStefano Zampini /* propagate MatOptions */ 247115579a77SStefano Zampini { 247215579a77SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)fetidpmat_ctx->pc->data; 247315579a77SStefano Zampini PetscBool issym; 247415579a77SStefano Zampini 247515579a77SStefano Zampini ierr = MatGetOption(pc->mat,MAT_SYMMETRIC,&issym);CHKERRQ(ierr); 247615579a77SStefano Zampini if (issym || pcbddc->symmetric_primal) { 247715579a77SStefano Zampini ierr = MatSetOption(newmat,MAT_SYMMETRIC,PETSC_TRUE);CHKERRQ(ierr); 247815579a77SStefano Zampini } 247915579a77SStefano Zampini } 2480547c9a8eSstefano_zampini ierr = MatSetOptionsPrefix(newmat,prefix);CHKERRQ(ierr); 2481547c9a8eSstefano_zampini ierr = MatAppendOptionsPrefix(newmat,"fetidp_");CHKERRQ(ierr); 24823425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 248315579a77SStefano Zampini /* FETI-DP preconditioner */ 24843425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 24853425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 24863425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 2487e1214c54Sstefano_zampini ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 2488e1214c54Sstefano_zampini ierr = PCSetOptionsPrefix(newpc,prefix);CHKERRQ(ierr); 2489e1214c54Sstefano_zampini ierr = PCAppendOptionsPrefix(newpc,"fetidp_");CHKERRQ(ierr); 2490399ffe99SStefano Zampini ierr = PCSetErrorIfFailure(newpc,pc->erroriffailure);CHKERRQ(ierr); 249115579a77SStefano Zampini if (!fetidpmat_ctx->l2g_lambda_only) { /* standard FETI-DP */ 24923425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 249315579a77SStefano Zampini ierr = PCShellSetName(newpc,"FETI-DP multipliers");CHKERRQ(ierr); 24943425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 24953425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2496edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2497c45b8d2dSstefano_zampini ierr = PCShellSetView(newpc,FETIDPPCView);CHKERRQ(ierr); 24983425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2499e1214c54Sstefano_zampini } else { 2500e1214c54Sstefano_zampini KSP *ksps; 2501e1214c54Sstefano_zampini PC lagpc; 2502e1214c54Sstefano_zampini Mat M,AM,PM; 2503e1214c54Sstefano_zampini PetscInt nn; 2504e1214c54Sstefano_zampini 250515579a77SStefano Zampini ierr = ISViewFromOptions(fetidpmat_ctx->lagrange,NULL,"-lag_view");CHKERRQ(ierr); 250615579a77SStefano Zampini ierr = ISViewFromOptions(fetidpmat_ctx->pressure,NULL,"-press_view");CHKERRQ(ierr); 2507e1214c54Sstefano_zampini ierr = PetscObjectQuery((PetscObject)pc,"__KSPFETIDP_PPmat",(PetscObject*)&M);CHKERRQ(ierr); 2508e1214c54Sstefano_zampini ierr = PCSetType(newpc,PCFIELDSPLIT);CHKERRQ(ierr); 2509e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"lag",fetidpmat_ctx->lagrange);CHKERRQ(ierr); 2510e1214c54Sstefano_zampini ierr = PCFieldSplitSetIS(newpc,"p",fetidpmat_ctx->pressure);CHKERRQ(ierr); 2511e1214c54Sstefano_zampini ierr = PCFieldSplitSetType(newpc,PC_COMPOSITE_SCHUR);CHKERRQ(ierr); 251240c75d76SStefano Zampini ierr = PCFieldSplitSetSchurFactType(newpc,PC_FIELDSPLIT_SCHUR_FACT_DIAG);CHKERRQ(ierr); 2513e1214c54Sstefano_zampini ierr = PCFieldSplitSetSchurPre(newpc,PC_FIELDSPLIT_SCHUR_PRE_USER,M);CHKERRQ(ierr); 2514c096484dSStefano Zampini ierr = PCFieldSplitSetSchurScale(newpc,1.0);CHKERRQ(ierr); 251515579a77SStefano Zampini 251615579a77SStefano Zampini /* we need to setfromoptions and setup here to access the blocks */ 2517e1214c54Sstefano_zampini ierr = PCSetFromOptions(newpc);CHKERRQ(ierr); 2518e1214c54Sstefano_zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 2519e1214c54Sstefano_zampini 2520e1214c54Sstefano_zampini /* set the solver for the (0,0) block */ 252140c75d76SStefano Zampini ierr = PCFieldSplitGetSubKSP(newpc,&nn,&ksps);CHKERRQ(ierr); 2522e1214c54Sstefano_zampini ierr = PCCreate(comm,&lagpc);CHKERRQ(ierr); 2523e1214c54Sstefano_zampini ierr = PCSetType(lagpc,PCSHELL);CHKERRQ(ierr); 2524e1214c54Sstefano_zampini ierr = KSPGetOperators(ksps[0],&AM,&PM);CHKERRQ(ierr); 2525e1214c54Sstefano_zampini ierr = PCSetOperators(lagpc,AM,PM);CHKERRQ(ierr); 2526e1214c54Sstefano_zampini ierr = PCShellSetContext(lagpc,fetidppc_ctx);CHKERRQ(ierr); 2527e1214c54Sstefano_zampini ierr = PCShellSetApply(lagpc,FETIDPPCApply);CHKERRQ(ierr); 2528e1214c54Sstefano_zampini ierr = PCShellSetApplyTranspose(lagpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 2529e1214c54Sstefano_zampini ierr = PCShellSetView(lagpc,FETIDPPCView);CHKERRQ(ierr); 2530e1214c54Sstefano_zampini ierr = PCShellSetDestroy(lagpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 2531e1214c54Sstefano_zampini ierr = KSPSetPC(ksps[0],lagpc);CHKERRQ(ierr); 253215579a77SStefano Zampini ierr = PetscObjectIncrementTabLevel((PetscObject)lagpc,(PetscObject)ksps[0],0);CHKERRQ(ierr); 2533e1214c54Sstefano_zampini ierr = PCDestroy(&lagpc);CHKERRQ(ierr); 2534e1214c54Sstefano_zampini ierr = PetscFree(ksps);CHKERRQ(ierr); 2535e1214c54Sstefano_zampini } 25363425bc38SStefano Zampini /* return pointers for objects created */ 25373425bc38SStefano Zampini *fetidp_mat = newmat; 25383425bc38SStefano Zampini *fetidp_pc = newpc; 25393425bc38SStefano Zampini PetscFunctionReturn(0); 25403425bc38SStefano Zampini } 25411e6b0712SBarry Smith 254294ef8ddeSSatish Balay /*@C 25430f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 25443425bc38SStefano Zampini 25453425bc38SStefano Zampini Collective 25463425bc38SStefano Zampini 25473425bc38SStefano Zampini Input Parameters: 25481720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 2549547c9a8eSstefano_zampini . fully_redundant - true for a fully redundant set of Lagrange multipliers 2550547c9a8eSstefano_zampini - prefix - optional options database prefix for the objects to be created (can be NULL) 255128509bceSStefano Zampini 255228509bceSStefano Zampini Output Parameters: 25530f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 25540f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 255528509bceSStefano Zampini 25563425bc38SStefano Zampini Level: developer 25573425bc38SStefano Zampini 25583425bc38SStefano Zampini Notes: 25590f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 25603425bc38SStefano Zampini 25610f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 25623425bc38SStefano Zampini @*/ 2563547c9a8eSstefano_zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, const char *prefix, Mat *fetidp_mat, PC *fetidp_pc) 25643425bc38SStefano Zampini { 25653425bc38SStefano Zampini PetscErrorCode ierr; 25663425bc38SStefano Zampini 25673425bc38SStefano Zampini PetscFunctionBegin; 25683425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 25693425bc38SStefano Zampini if (pc->setupcalled) { 2570547c9a8eSstefano_zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,const char*,Mat*,PC*),(pc,fully_redundant,prefix,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 25714247aa23Sstefano_zampini } else SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 25723425bc38SStefano Zampini PetscFunctionReturn(0); 25733425bc38SStefano Zampini } 25740c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2575da1bb401SStefano Zampini /*MC 2576da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 25770c7d97c5SJed Brown 257828509bceSStefano Zampini An implementation of the BDDC preconditioner based on 257928509bceSStefano Zampini 258028509bceSStefano Zampini .vb 258128509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 258228509bceSStefano 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 258328509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 25840f202f7eSStefano 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 258528509bceSStefano Zampini .ve 258628509bceSStefano Zampini 258728509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 258828509bceSStefano Zampini 25890f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 259028509bceSStefano Zampini 259128509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 259228509bceSStefano Zampini 2593b6fdb6dfSStefano 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. 2594b6fdb6dfSStefano Zampini 2595c7017625SStefano 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). 259628509bceSStefano Zampini 25970f202f7eSStefano 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() 259830368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 259928509bceSStefano Zampini 26000f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 260128509bceSStefano Zampini 26020f202f7eSStefano 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. 26030f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 260428509bceSStefano Zampini 26050f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 260628509bceSStefano Zampini 2607df4d28bfSStefano 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. 260828509bceSStefano Zampini 26090f202f7eSStefano 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. 26100f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 26110f202f7eSStefano Zampini 26120f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 26130f202f7eSStefano Zampini 26140f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 26150f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 26160f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 26170f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 26180f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 26190f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 26200f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 262128509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 26220f202f7eSStefano 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) 26235459c157SBarry Smith . -pc_bddc_coarse_redistribute <0> - size of a subset of processors where the coarse problem will be remapped (the value is ignored if not at the coarsest level) 26240f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 26250f202f7eSStefano 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) 2626bd2a564bSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value different than zero is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 262728509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 262828509bceSStefano Zampini 262928509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 263028509bceSStefano Zampini .vb 263128509bceSStefano Zampini -pc_bddc_dirichlet_ 263228509bceSStefano Zampini -pc_bddc_neumann_ 263328509bceSStefano Zampini -pc_bddc_coarse_ 263428509bceSStefano Zampini .ve 26350f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 263628509bceSStefano Zampini 26370f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 263828509bceSStefano Zampini .vb 2639312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2640312be037SStefano Zampini -pc_bddc_neumann_lN_ 2641312be037SStefano Zampini -pc_bddc_coarse_lN_ 264228509bceSStefano Zampini .ve 26430f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 26440f202f7eSStefano 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. 26450f202f7eSStefano Zampini .vb 26460f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 26470f202f7eSStefano Zampini .ve 26480f202f7eSStefano 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 2649da1bb401SStefano Zampini 2650da1bb401SStefano Zampini Level: intermediate 2651da1bb401SStefano Zampini 2652e94cfbe0SPatrick Sanan Developer Notes: 2653da1bb401SStefano Zampini 2654da1bb401SStefano Zampini Contributed by Stefano Zampini 2655da1bb401SStefano Zampini 2656da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2657da1bb401SStefano Zampini M*/ 2658b2573a8aSBarry Smith 26598cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2660da1bb401SStefano Zampini { 2661da1bb401SStefano Zampini PetscErrorCode ierr; 2662da1bb401SStefano Zampini PC_BDDC *pcbddc; 2663da1bb401SStefano Zampini 2664da1bb401SStefano Zampini PetscFunctionBegin; 2665b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2666da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2667da1bb401SStefano Zampini 2668da1bb401SStefano Zampini /* create PCIS data structure */ 2669da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2670da1bb401SStefano Zampini 26719326c5c6Sstefano_zampini /* create local graph structure */ 26729326c5c6Sstefano_zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 26739326c5c6Sstefano_zampini 26749326c5c6Sstefano_zampini /* BDDC nonzero defaults */ 267508a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 267647d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 267747d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 26783301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 267914f95afaSStefano Zampini pcbddc->vertex_size = 1; 2680c703fcc7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 268168457ee5SStefano Zampini pcbddc->coarse_size = -1; 268285c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 268347d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 268457de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 268527b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 26861e0482f5SStefano Zampini pcbddc->nedfield = -1; 26871e0482f5SStefano Zampini pcbddc->nedglobal = PETSC_TRUE; 2688be12c134Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 2689b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2690bd2a564bSStefano Zampini pcbddc->adaptive_threshold[0] = 0.0; 2691bd2a564bSStefano Zampini pcbddc->adaptive_threshold[1] = 0.0; 2692b7eb3628SStefano Zampini 2693da1bb401SStefano Zampini /* function pointers */ 2694da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 269593bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2696da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2697da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2698da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 26996b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2700da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2701da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2702da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2703534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2704534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 27059326c5c6Sstefano_zampini pc->ops->reset = PCReset_BDDC; 2706da1bb401SStefano Zampini 2707da1bb401SStefano Zampini /* composing function */ 2708a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr); 2709a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr); 2710906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2711674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 271230368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2713bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 27142b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2715b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 27162b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2717bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 271882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2719bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 272082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2721bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 272282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2723bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 272482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2725bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 272663602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2727bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2728bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2729bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2730bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2731a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2732ab8c8b98SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCSetCoordinates_C",PCSetCoordinates_BDDC);CHKERRQ(ierr); 2733da1bb401SStefano Zampini PetscFunctionReturn(0); 2734da1bb401SStefano Zampini } 273543371fb9SStefano Zampini 273643371fb9SStefano Zampini /*@C 273743371fb9SStefano Zampini PCBDDCInitializePackage - This function initializes everything in the PCBDDC package. It is called 273843371fb9SStefano Zampini from PetscDLLibraryRegister() when using dynamic libraries, and on the first call to PCCreate_BDDC() 273943371fb9SStefano Zampini when using static libraries. 274043371fb9SStefano Zampini 274143371fb9SStefano Zampini Level: developer 274243371fb9SStefano Zampini 274343371fb9SStefano Zampini .keywords: PC, PCBDDC, initialize, package 274443371fb9SStefano Zampini .seealso: PetscInitialize() 274543371fb9SStefano Zampini @*/ 274643371fb9SStefano Zampini PetscErrorCode PCBDDCInitializePackage(void) 274743371fb9SStefano Zampini { 274843371fb9SStefano Zampini PetscErrorCode ierr; 274943371fb9SStefano Zampini int i; 275043371fb9SStefano Zampini 275143371fb9SStefano Zampini PetscFunctionBegin; 275243371fb9SStefano Zampini if (PCBDDCPackageInitialized) PetscFunctionReturn(0); 275343371fb9SStefano Zampini PCBDDCPackageInitialized = PETSC_TRUE; 275443371fb9SStefano Zampini ierr = PetscRegisterFinalize(PCBDDCFinalizePackage);CHKERRQ(ierr); 275543371fb9SStefano Zampini 275643371fb9SStefano Zampini /* general events */ 275743371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCTopo",PC_CLASSID,&PC_BDDC_Topology[0]);CHKERRQ(ierr); 275843371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCLKSP",PC_CLASSID,&PC_BDDC_LocalSolvers[0]);CHKERRQ(ierr); 275943371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCLWor",PC_CLASSID,&PC_BDDC_LocalWork[0]);CHKERRQ(ierr); 276043371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCCorr",PC_CLASSID,&PC_BDDC_CorrectionSetUp[0]);CHKERRQ(ierr); 276143371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCCSet",PC_CLASSID,&PC_BDDC_CoarseSetUp[0]);CHKERRQ(ierr); 276243371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCCKSP",PC_CLASSID,&PC_BDDC_CoarseSolver[0]);CHKERRQ(ierr); 276343371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCAdap",PC_CLASSID,&PC_BDDC_AdaptiveSetUp[0]);CHKERRQ(ierr); 276443371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCScal",PC_CLASSID,&PC_BDDC_Scaling[0]);CHKERRQ(ierr); 276543371fb9SStefano Zampini ierr = PetscLogEventRegister("PCBDDCSchr",PC_CLASSID,&PC_BDDC_Schurs[0]);CHKERRQ(ierr); 276643371fb9SStefano Zampini for (i=1;i<PETSC_PCBDDC_MAXLEVELS;i++) { 276743371fb9SStefano Zampini char ename[32]; 276843371fb9SStefano Zampini 276943371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCTopo l%02d",i);CHKERRQ(ierr); 277043371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_Topology[i]);CHKERRQ(ierr); 277143371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCLKSP l%02d",i);CHKERRQ(ierr); 277243371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_LocalSolvers[i]);CHKERRQ(ierr); 277343371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCLWor l%02d",i);CHKERRQ(ierr); 277443371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_LocalWork[i]);CHKERRQ(ierr); 277543371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCCorr l%02d",i);CHKERRQ(ierr); 277643371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_CorrectionSetUp[i]);CHKERRQ(ierr); 277743371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCCSet l%02d",i);CHKERRQ(ierr); 277843371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_CoarseSetUp[i]);CHKERRQ(ierr); 277943371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCCKSP l%02d",i);CHKERRQ(ierr); 278043371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_CoarseSolver[i]);CHKERRQ(ierr); 278143371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCAdap l%02d",i);CHKERRQ(ierr); 278243371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_AdaptiveSetUp[i]);CHKERRQ(ierr); 278343371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCScal l%02d",i);CHKERRQ(ierr); 278443371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_Scaling[i]);CHKERRQ(ierr); 278543371fb9SStefano Zampini ierr = PetscSNPrintf(ename,sizeof(ename),"PCBDDCSchr l%02d",i);CHKERRQ(ierr); 278643371fb9SStefano Zampini ierr = PetscLogEventRegister(ename,PC_CLASSID,&PC_BDDC_Schurs[i]);CHKERRQ(ierr); 278743371fb9SStefano Zampini } 278843371fb9SStefano Zampini PetscFunctionReturn(0); 278943371fb9SStefano Zampini } 279043371fb9SStefano Zampini 279143371fb9SStefano Zampini /*@C 279243371fb9SStefano Zampini PCBDDCFinalizePackage - This function frees everything from the PCBDDC package. It is 279343371fb9SStefano Zampini called from PetscFinalize() automatically. 279443371fb9SStefano Zampini 279543371fb9SStefano Zampini Level: developer 279643371fb9SStefano Zampini 279743371fb9SStefano Zampini .keywords: Petsc, destroy, package 279843371fb9SStefano Zampini .seealso: PetscFinalize() 279943371fb9SStefano Zampini @*/ 280043371fb9SStefano Zampini PetscErrorCode PCBDDCFinalizePackage(void) 280143371fb9SStefano Zampini { 280243371fb9SStefano Zampini PetscFunctionBegin; 280343371fb9SStefano Zampini PCBDDCPackageInitialized = PETSC_FALSE; 280443371fb9SStefano Zampini PetscFunctionReturn(0); 280543371fb9SStefano Zampini } 2806