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 FETIDP 18eb97c9d2SStefano Zampini - Move FETIDP code to its own classes 19eb97c9d2SStefano Zampini 20eb97c9d2SStefano Zampini MATIS related operations contained in BDDC code 21eb97c9d2SStefano Zampini - Provide general case for subassembling 22eb97c9d2SStefano Zampini 2353cdbc3dSStefano Zampini */ 240c7d97c5SJed Brown 25ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 26ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 273b03a366Sstefano_zampini #include <petscblaslapack.h> 28674ae819SStefano Zampini 290369aaf7SStefano Zampini /* temporarily declare it */ 300369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 310369aaf7SStefano Zampini 320c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 330c7d97c5SJed Brown #undef __FUNCT__ 340c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 354416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 360c7d97c5SJed Brown { 370c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 380c7d97c5SJed Brown PetscErrorCode ierr; 390c7d97c5SJed Brown 400c7d97c5SJed Brown PetscFunctionBegin; 41e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 428eeda7d8SStefano Zampini /* Verbose debugging */ 43c7017625SStefano 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); 44c7017625SStefano 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); 45c7017625SStefano 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); 46c7017625SStefano 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); 478eeda7d8SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 486b78500eSPatrick Sanan /* Primal space customization */ 4908a5cf49SStefano 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); 508eeda7d8SStefano 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); 518eeda7d8SStefano 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); 528eeda7d8SStefano 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); 5314f95afaSStefano 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); 546661aa1dSStefano 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); 5514f95afaSStefano 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); 568eeda7d8SStefano Zampini /* Change of basis */ 57b9b85e73SStefano 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); 58b9b85e73SStefano 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); 59674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 60674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 61674ae819SStefano Zampini } 628eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 638eeda7d8SStefano 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); 6457de7509SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_coarse_eqs_per_proc","Number of equations per process for coarse problem redistribution (significant only at the coarsest level)","none",pcbddc->coarse_eqs_per_proc,&pcbddc->coarse_eqs_per_proc,NULL);CHKERRQ(ierr); 650298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 662b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 67323d395dSStefano 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); 68674ae819SStefano 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); 69b96c3477SStefano 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); 70b96c3477SStefano 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); 71b96c3477SStefano 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); 72683d3df6SStefano 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); 73bf3a8328SStefano 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); 74bf3a8328SStefano 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); 754c6709b3SStefano Zampini ierr = PetscOptionsReal("-pc_bddc_adaptive_threshold","Threshold to be used for adaptive selection of constraints","none",pcbddc->adaptive_threshold,&pcbddc->adaptive_threshold,NULL);CHKERRQ(ierr); 7608122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 7708122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 783301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 79b0c7d250SStefano 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); 80b3afcdbeSStefano 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); 81b3afcdbeSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_benign_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->benign_compute_nonetflux,&pcbddc->benign_compute_nonetflux,NULL);CHKERRQ(ierr); 82e9627c49SStefano 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); 8327b6a85dSStefano 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); 844f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 8570c64980SStefano 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); 860c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 870c7d97c5SJed Brown PetscFunctionReturn(0); 880c7d97c5SJed Brown } 896b78500eSPatrick Sanan 906b78500eSPatrick Sanan /* -------------------------------------------------------------------------- */ 916b78500eSPatrick Sanan #undef __FUNCT__ 926b78500eSPatrick Sanan #define __FUNCT__ "PCView_BDDC" 936b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 946b78500eSPatrick Sanan { 956b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 96e9627c49SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 976b78500eSPatrick Sanan PetscErrorCode ierr; 986b78500eSPatrick Sanan PetscBool isascii,isstring; 99e9627c49SStefano Zampini PetscSubcomm subcomm; 100e9627c49SStefano Zampini PetscViewer subviewer; 1016b78500eSPatrick Sanan 1026b78500eSPatrick Sanan PetscFunctionBegin; 1036b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 1046b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 1056b78500eSPatrick Sanan /* Nothing printed for the String viewer */ 1066b78500eSPatrick Sanan /* ASCII viewer */ 1076b78500eSPatrick Sanan if (isascii) { 108e9627c49SStefano Zampini PetscMPIInt color,rank; 109345ecf6cSStefano Zampini Petsc64bitInt loc[6],gsum[5],gmax[5],gmin[5],totbenign; 110e9627c49SStefano Zampini PetscScalar interface_size; 111e9627c49SStefano Zampini PetscReal ratio1=0.,ratio2=0.; 112e9627c49SStefano Zampini Vec counter; 1136b78500eSPatrick Sanan 114e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use verbose output: %d\n",pcbddc->dbg_flag);CHKERRQ(ierr); 115e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use user-defined CSR: %d\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 1166b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use local mat graph: %d\n",pcbddc->use_local_adj);CHKERRQ(ierr); 117e9627c49SStefano Zampini if (pcbddc->mat_graph->twodim) { 118e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Connectivity graph topological dimension: 2\n");CHKERRQ(ierr); 119e9627c49SStefano Zampini } else { 120e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Connectivity graph topological dimension: 3\n");CHKERRQ(ierr); 121e9627c49SStefano Zampini } 122e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use vertices: %d (vertext size %d)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 1236b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 1246b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 1256b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 1266b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 1276b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 1286b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 12927b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 13027b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 13127b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 1326b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 13327b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 1346b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr); 135e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 1366b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 1376b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 138e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 1396b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 1406b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 1416b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 142e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Adaptive constraint selection threshold (active %d, userdefined %d): %g\n",pcbddc->adaptive_threshold,pcbddc->adaptive_selection,pcbddc->adaptive_userdefined);CHKERRQ(ierr); 1436b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 1446b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 145e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 1466b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 1476b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 148e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Coarse eqs per proc (significant at the coarsest level): %d\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 149e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr); 150fc17d649SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 15127b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subspace trick is active: %d (algebraic computation of no-net-flux %d)\n",pcbddc->benign_have_null,pcbddc->benign_compute_nonetflux);CHKERRQ(ierr); 1526b78500eSPatrick Sanan 153345ecf6cSStefano Zampini /* compute some numbers on the domain decomposition */ 154e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 155e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 156e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 157e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 158e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 159e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 160e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 161e9627c49SStefano Zampini gsum[0] = 1; 162345ecf6cSStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = 0; 163e9627c49SStefano Zampini loc[0] = !!pcis->n; 164e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 165e9627c49SStefano Zampini loc[2] = pcis->n_B; 166e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 167345ecf6cSStefano Zampini loc[4] = pcis->n; 168345ecf6cSStefano Zampini loc[5] = pcbddc->benign_n; 169345ecf6cSStefano Zampini ierr = MPI_Reduce(loc,gsum,5,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 170345ecf6cSStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = -1; 171345ecf6cSStefano Zampini ierr = MPI_Reduce(loc,gmax,5,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 172345ecf6cSStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = PETSC_MAX_INT; 173345ecf6cSStefano Zampini ierr = MPI_Reduce(loc,gmin,5,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 174345ecf6cSStefano Zampini ierr = MPI_Reduce(&loc[5],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 175e9627c49SStefano Zampini if (pcbddc->coarse_size) { 176e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 177e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 178e9627c49SStefano Zampini } 179e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 180e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Global dofs sizes: all %d interface %d coarse %d\n",pc->pmat->rmap->N,(PetscInt)PetscRealPart(interface_size),pcbddc->coarse_size);CHKERRQ(ierr); 181e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Coarsening ratios: all/coarse %d interface/coarse %d\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 182e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Active processes : %d\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 183345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 184345ecf6cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subs : %d\n",(PetscInt)totbenign);CHKERRQ(ierr); 185345ecf6cSStefano Zampini } 186e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Dofs type :\tMIN\tMAX\tMEAN\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 187e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Interior dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 188e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Interface dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[2],(PetscInt)gmax[2],(PetscInt)(gsum[2]/gsum[0]));CHKERRQ(ierr); 189e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Primal dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[3],(PetscInt)gmax[3],(PetscInt)(gsum[3]/gsum[0]));CHKERRQ(ierr); 190345ecf6cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Local dofs :\t%d\t%d\t%d\n",(PetscInt)gmin[4],(PetscInt)gmax[4],(PetscInt)(gsum[4]/gsum[0]));CHKERRQ(ierr); 19127b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: ********************************** COARSE PROBLEM DETAILS *********************************\n",pcbddc->current_level);CHKERRQ(ierr); 19227b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 193e9627c49SStefano Zampini 194e9627c49SStefano Zampini /* the coarse problem can be handled with a different communicator */ 195e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 196e9627c49SStefano Zampini else color = 0; 197e9627c49SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 198e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 199e9627c49SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,2);CHKERRQ(ierr); 200e9627c49SStefano Zampini ierr = PetscSubcommSetTypeGeneral(subcomm,color,rank);CHKERRQ(ierr); 201e9627c49SStefano Zampini ierr = PetscViewerGetSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 202e9627c49SStefano Zampini if (color == 1) { 203e9627c49SStefano Zampini ierr = KSPView(pcbddc->coarse_ksp,subviewer);CHKERRQ(ierr); 204e9627c49SStefano Zampini ierr = PetscViewerFlush(subviewer);CHKERRQ(ierr); 205e9627c49SStefano Zampini } 206e9627c49SStefano Zampini ierr = PetscViewerRestoreSubViewer(viewer,PetscSubcommChild(subcomm),&subviewer);CHKERRQ(ierr); 207e9627c49SStefano Zampini ierr = PetscSubcommDestroy(&subcomm);CHKERRQ(ierr); 208e9627c49SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 209e9627c49SStefano Zampini } 2106b78500eSPatrick Sanan PetscFunctionReturn(0); 2116b78500eSPatrick Sanan } 2126b78500eSPatrick Sanan 2130c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 214674ae819SStefano Zampini #undef __FUNCT__ 215906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 2161dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 217b9b85e73SStefano Zampini { 218b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 219b9b85e73SStefano Zampini PetscErrorCode ierr; 220b9b85e73SStefano Zampini 221b9b85e73SStefano Zampini PetscFunctionBegin; 222b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 22356282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 224b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 2251dd7afcfSStefano Zampini pcbddc->change_interior = interior; 226b9b85e73SStefano Zampini PetscFunctionReturn(0); 227b9b85e73SStefano Zampini } 228b9b85e73SStefano Zampini #undef __FUNCT__ 229906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 230b9b85e73SStefano Zampini /*@ 231906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 232b9b85e73SStefano Zampini 233b9b85e73SStefano Zampini Collective on PC 234b9b85e73SStefano Zampini 235b9b85e73SStefano Zampini Input Parameters: 236b9b85e73SStefano Zampini + pc - the preconditioning context 2371dd7afcfSStefano Zampini . change - the change of basis matrix 2381dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 239b9b85e73SStefano Zampini 240b9b85e73SStefano Zampini Level: intermediate 241b9b85e73SStefano Zampini 242b9b85e73SStefano Zampini Notes: 243b9b85e73SStefano Zampini 244b9b85e73SStefano Zampini .seealso: PCBDDC 245b9b85e73SStefano Zampini @*/ 2461dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 247b9b85e73SStefano Zampini { 248b9b85e73SStefano Zampini PetscErrorCode ierr; 249b9b85e73SStefano Zampini 250b9b85e73SStefano Zampini PetscFunctionBegin; 251b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 252b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 253906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 254906d46d4SStefano Zampini if (pc->mat) { 255906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 256906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 257906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 2586c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 2596c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 260906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 261906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 2626c4ed002SBarry Smith if (rows_c != rows) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 2636c4ed002SBarry Smith if (cols_c != cols) SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 264906d46d4SStefano Zampini } 2651dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 266b9b85e73SStefano Zampini PetscFunctionReturn(0); 267b9b85e73SStefano Zampini } 268b9b85e73SStefano Zampini /* -------------------------------------------------------------------------- */ 269b9b85e73SStefano Zampini #undef __FUNCT__ 27030368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS_BDDC" 27130368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 27230368db7SStefano Zampini { 27330368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 27456282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 27530368db7SStefano Zampini PetscErrorCode ierr; 27630368db7SStefano Zampini 27730368db7SStefano Zampini PetscFunctionBegin; 27856282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 27956282151SStefano Zampini if (pcbddc->user_primal_vertices) { 28056282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 28156282151SStefano Zampini } 28230368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 28330368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 28430368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 28556282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 28630368db7SStefano Zampini PetscFunctionReturn(0); 28730368db7SStefano Zampini } 28830368db7SStefano Zampini #undef __FUNCT__ 28930368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS" 29030368db7SStefano Zampini /*@ 29130368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 29230368db7SStefano Zampini 29330368db7SStefano Zampini Collective 29430368db7SStefano Zampini 29530368db7SStefano Zampini Input Parameters: 29630368db7SStefano Zampini + pc - the preconditioning context 29730368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 29830368db7SStefano Zampini 29930368db7SStefano Zampini Level: intermediate 30030368db7SStefano Zampini 30130368db7SStefano Zampini Notes: 30230368db7SStefano Zampini Any process can list any global node 30330368db7SStefano Zampini 30430368db7SStefano Zampini .seealso: PCBDDC 30530368db7SStefano Zampini @*/ 30630368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 30730368db7SStefano Zampini { 30830368db7SStefano Zampini PetscErrorCode ierr; 30930368db7SStefano Zampini 31030368db7SStefano Zampini PetscFunctionBegin; 31130368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 31230368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 31330368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 31430368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 31530368db7SStefano Zampini PetscFunctionReturn(0); 31630368db7SStefano Zampini } 31730368db7SStefano Zampini /* -------------------------------------------------------------------------- */ 31830368db7SStefano Zampini #undef __FUNCT__ 319674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 320674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 321674ae819SStefano Zampini { 322674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 32356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 324674ae819SStefano Zampini PetscErrorCode ierr; 3251e6b0712SBarry Smith 326674ae819SStefano Zampini PetscFunctionBegin; 32756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 32856282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 32956282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 33056282151SStefano Zampini } 331674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 33230368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 33330368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 33456282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 335674ae819SStefano Zampini PetscFunctionReturn(0); 336674ae819SStefano Zampini } 337674ae819SStefano Zampini #undef __FUNCT__ 338674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 339674ae819SStefano Zampini /*@ 34028509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 341674ae819SStefano Zampini 34217eb1463SStefano Zampini Collective 343674ae819SStefano Zampini 344674ae819SStefano Zampini Input Parameters: 345674ae819SStefano Zampini + pc - the preconditioning context 34617eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 347674ae819SStefano Zampini 348674ae819SStefano Zampini Level: intermediate 349674ae819SStefano Zampini 350674ae819SStefano Zampini Notes: 351674ae819SStefano Zampini 352674ae819SStefano Zampini .seealso: PCBDDC 353674ae819SStefano Zampini @*/ 354674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 355674ae819SStefano Zampini { 356674ae819SStefano Zampini PetscErrorCode ierr; 357674ae819SStefano Zampini 358674ae819SStefano Zampini PetscFunctionBegin; 359674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 360674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 36117eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 362674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 363674ae819SStefano Zampini PetscFunctionReturn(0); 364674ae819SStefano Zampini } 365674ae819SStefano Zampini /* -------------------------------------------------------------------------- */ 3660c7d97c5SJed Brown #undef __FUNCT__ 3674fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 3684fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 3694fad6a16SStefano Zampini { 3704fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 3714fad6a16SStefano Zampini 3724fad6a16SStefano Zampini PetscFunctionBegin; 3734fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 3744fad6a16SStefano Zampini PetscFunctionReturn(0); 3754fad6a16SStefano Zampini } 3761e6b0712SBarry Smith 3774fad6a16SStefano Zampini #undef __FUNCT__ 3784fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 3794fad6a16SStefano Zampini /*@ 38028509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 3814fad6a16SStefano Zampini 3824fad6a16SStefano Zampini Logically collective on PC 3834fad6a16SStefano Zampini 3844fad6a16SStefano Zampini Input Parameters: 3854fad6a16SStefano Zampini + pc - the preconditioning context 38628509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 3874fad6a16SStefano Zampini 3880f202f7eSStefano Zampini Options Database Keys: 3890f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 3904fad6a16SStefano Zampini 3914fad6a16SStefano Zampini Level: intermediate 3924fad6a16SStefano Zampini 3934fad6a16SStefano Zampini Notes: 3940f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 3954fad6a16SStefano Zampini 3960f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 3974fad6a16SStefano Zampini @*/ 3984fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 3994fad6a16SStefano Zampini { 4004fad6a16SStefano Zampini PetscErrorCode ierr; 4014fad6a16SStefano Zampini 4024fad6a16SStefano Zampini PetscFunctionBegin; 4034fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4042b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 4054fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 4064fad6a16SStefano Zampini PetscFunctionReturn(0); 4074fad6a16SStefano Zampini } 4082b510759SStefano Zampini 409b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 4102b510759SStefano Zampini #undef __FUNCT__ 411b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 412b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 413b8ffe317SStefano Zampini { 414b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 415b8ffe317SStefano Zampini 416b8ffe317SStefano Zampini PetscFunctionBegin; 41785c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 418b8ffe317SStefano Zampini PetscFunctionReturn(0); 419b8ffe317SStefano Zampini } 420b8ffe317SStefano Zampini 421b8ffe317SStefano Zampini #undef __FUNCT__ 422b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 423b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 4242b510759SStefano Zampini { 4252b510759SStefano Zampini PetscErrorCode ierr; 4262b510759SStefano Zampini 4272b510759SStefano Zampini PetscFunctionBegin; 4282b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 429b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 430b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 4312b510759SStefano Zampini PetscFunctionReturn(0); 4322b510759SStefano Zampini } 4331e6b0712SBarry Smith 4344fad6a16SStefano Zampini #undef __FUNCT__ 4352b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 4362b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 4374fad6a16SStefano Zampini { 4384fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4394fad6a16SStefano Zampini 4404fad6a16SStefano Zampini PetscFunctionBegin; 4412b510759SStefano Zampini pcbddc->current_level = level; 4424fad6a16SStefano Zampini PetscFunctionReturn(0); 4434fad6a16SStefano Zampini } 4441e6b0712SBarry Smith 4454fad6a16SStefano Zampini #undef __FUNCT__ 446b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 447b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 448b8ffe317SStefano Zampini { 449b8ffe317SStefano Zampini PetscErrorCode ierr; 450b8ffe317SStefano Zampini 451b8ffe317SStefano Zampini PetscFunctionBegin; 452b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 453b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 454b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 455b8ffe317SStefano Zampini PetscFunctionReturn(0); 456b8ffe317SStefano Zampini } 457b8ffe317SStefano Zampini 458b8ffe317SStefano Zampini #undef __FUNCT__ 4592b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 4602b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 4612b510759SStefano Zampini { 4622b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4632b510759SStefano Zampini 4642b510759SStefano Zampini PetscFunctionBegin; 4652b510759SStefano Zampini pcbddc->max_levels = levels; 4662b510759SStefano Zampini PetscFunctionReturn(0); 4672b510759SStefano Zampini } 4682b510759SStefano Zampini 4692b510759SStefano Zampini #undef __FUNCT__ 4702b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 4714fad6a16SStefano Zampini /*@ 47228509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 4734fad6a16SStefano Zampini 4744fad6a16SStefano Zampini Logically collective on PC 4754fad6a16SStefano Zampini 4764fad6a16SStefano Zampini Input Parameters: 4774fad6a16SStefano Zampini + pc - the preconditioning context 47828509bceSStefano Zampini - levels - the maximum number of levels (max 9) 4794fad6a16SStefano Zampini 4800f202f7eSStefano Zampini Options Database Keys: 4810f202f7eSStefano Zampini . -pc_bddc_levels 4824fad6a16SStefano Zampini 4834fad6a16SStefano Zampini Level: intermediate 4844fad6a16SStefano Zampini 4854fad6a16SStefano Zampini Notes: 4860f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 4874fad6a16SStefano Zampini 4880f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 4894fad6a16SStefano Zampini @*/ 4902b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 4914fad6a16SStefano Zampini { 4924fad6a16SStefano Zampini PetscErrorCode ierr; 4934fad6a16SStefano Zampini 4944fad6a16SStefano Zampini PetscFunctionBegin; 4954fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 4962b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 497312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 4982b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 4994fad6a16SStefano Zampini PetscFunctionReturn(0); 5004fad6a16SStefano Zampini } 5014fad6a16SStefano Zampini /* -------------------------------------------------------------------------- */ 5021e6b0712SBarry Smith 5034fad6a16SStefano Zampini #undef __FUNCT__ 5043b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 5053b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 5063b03a366Sstefano_zampini { 5073b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 50856282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 5093b03a366Sstefano_zampini PetscErrorCode ierr; 5103b03a366Sstefano_zampini 5113b03a366Sstefano_zampini PetscFunctionBegin; 51256282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 51356282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 51456282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 51556282151SStefano Zampini } 516785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 517785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 5183b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 51936e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 52056282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 5213b03a366Sstefano_zampini PetscFunctionReturn(0); 5223b03a366Sstefano_zampini } 5231e6b0712SBarry Smith 5243b03a366Sstefano_zampini #undef __FUNCT__ 5253b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 5263b03a366Sstefano_zampini /*@ 52728509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 5283b03a366Sstefano_zampini 529785d1243SStefano Zampini Collective 5303b03a366Sstefano_zampini 5313b03a366Sstefano_zampini Input Parameters: 5323b03a366Sstefano_zampini + pc - the preconditioning context 533785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 5343b03a366Sstefano_zampini 5353b03a366Sstefano_zampini Level: intermediate 5363b03a366Sstefano_zampini 5370f202f7eSStefano Zampini Notes: 5380f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 5393b03a366Sstefano_zampini 5400f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 5413b03a366Sstefano_zampini @*/ 5423b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 5433b03a366Sstefano_zampini { 5443b03a366Sstefano_zampini PetscErrorCode ierr; 5453b03a366Sstefano_zampini 5463b03a366Sstefano_zampini PetscFunctionBegin; 5473b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 548674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 549785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 5503b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 5513b03a366Sstefano_zampini PetscFunctionReturn(0); 5523b03a366Sstefano_zampini } 5533b03a366Sstefano_zampini /* -------------------------------------------------------------------------- */ 5541e6b0712SBarry Smith 5553b03a366Sstefano_zampini #undef __FUNCT__ 55682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 55782ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 5580c7d97c5SJed Brown { 5590c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 56056282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 5610c7d97c5SJed Brown PetscErrorCode ierr; 5620c7d97c5SJed Brown 5630c7d97c5SJed Brown PetscFunctionBegin; 56456282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 56556282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 56656282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 56756282151SStefano Zampini } 568785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 569785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 5700c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 571785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 57256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 5730c7d97c5SJed Brown PetscFunctionReturn(0); 5740c7d97c5SJed Brown } 5750c7d97c5SJed Brown 5760c7d97c5SJed Brown #undef __FUNCT__ 57782ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 5789c0446d6SStefano Zampini /*@ 57982ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 5809c0446d6SStefano Zampini 581785d1243SStefano Zampini Collective 58253cdbc3dSStefano Zampini 58353cdbc3dSStefano Zampini Input Parameters: 58453cdbc3dSStefano Zampini + pc - the preconditioning context 58582ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 58653cdbc3dSStefano Zampini 58753cdbc3dSStefano Zampini Level: intermediate 58853cdbc3dSStefano Zampini 5899c0446d6SStefano Zampini Notes: 59053cdbc3dSStefano Zampini 5910f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 59253cdbc3dSStefano Zampini @*/ 59382ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 5940c7d97c5SJed Brown { 5950c7d97c5SJed Brown PetscErrorCode ierr; 5960c7d97c5SJed Brown 5970c7d97c5SJed Brown PetscFunctionBegin; 5980c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5990c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 60082ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 60182ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6020c7d97c5SJed Brown PetscFunctionReturn(0); 6030c7d97c5SJed Brown } 6040c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 6050c7d97c5SJed Brown 6060c7d97c5SJed Brown #undef __FUNCT__ 6070c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 60853cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 6090c7d97c5SJed Brown { 6100c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 61156282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 61253cdbc3dSStefano Zampini PetscErrorCode ierr; 6130c7d97c5SJed Brown 6140c7d97c5SJed Brown PetscFunctionBegin; 61556282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 61656282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 61756282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 61856282151SStefano Zampini } 619785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 620785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 62153cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 62236e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 62356282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6240c7d97c5SJed Brown PetscFunctionReturn(0); 6250c7d97c5SJed Brown } 6261e6b0712SBarry Smith 6270c7d97c5SJed Brown #undef __FUNCT__ 6280c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 62957527edcSJed Brown /*@ 63028509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 63157527edcSJed Brown 632785d1243SStefano Zampini Collective 63357527edcSJed Brown 63457527edcSJed Brown Input Parameters: 63557527edcSJed Brown + pc - the preconditioning context 636785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 63757527edcSJed Brown 63857527edcSJed Brown Level: intermediate 63957527edcSJed Brown 6400f202f7eSStefano Zampini Notes: 6410f202f7eSStefano Zampini Any process can list any global node 64257527edcSJed Brown 6430f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 64457527edcSJed Brown @*/ 64553cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 6460c7d97c5SJed Brown { 6470c7d97c5SJed Brown PetscErrorCode ierr; 6480c7d97c5SJed Brown 6490c7d97c5SJed Brown PetscFunctionBegin; 6500c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 651674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 652785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 65353cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 65453cdbc3dSStefano Zampini PetscFunctionReturn(0); 65553cdbc3dSStefano Zampini } 65653cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 6571e6b0712SBarry Smith 65853cdbc3dSStefano Zampini #undef __FUNCT__ 65982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 66082ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 6610c7d97c5SJed Brown { 6620c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 66356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6640c7d97c5SJed Brown PetscErrorCode ierr; 6650c7d97c5SJed Brown 6660c7d97c5SJed Brown PetscFunctionBegin; 66756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 66856282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 66956282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 67056282151SStefano Zampini } 671785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 672785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 6730c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 674785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 67556282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6760c7d97c5SJed Brown PetscFunctionReturn(0); 6770c7d97c5SJed Brown } 6780c7d97c5SJed Brown 6790c7d97c5SJed Brown #undef __FUNCT__ 68082ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 6810c7d97c5SJed Brown /*@ 68282ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 6830c7d97c5SJed Brown 684785d1243SStefano Zampini Collective 6850c7d97c5SJed Brown 6860c7d97c5SJed Brown Input Parameters: 6870c7d97c5SJed Brown + pc - the preconditioning context 68882ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 6890c7d97c5SJed Brown 6900c7d97c5SJed Brown Level: intermediate 6910c7d97c5SJed Brown 6920c7d97c5SJed Brown Notes: 6930c7d97c5SJed Brown 6940f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 6950c7d97c5SJed Brown @*/ 69682ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 6970c7d97c5SJed Brown { 6980c7d97c5SJed Brown PetscErrorCode ierr; 6990c7d97c5SJed Brown 7000c7d97c5SJed Brown PetscFunctionBegin; 7010c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7020c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 70382ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 70482ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 70553cdbc3dSStefano Zampini PetscFunctionReturn(0); 70653cdbc3dSStefano Zampini } 70753cdbc3dSStefano Zampini /* -------------------------------------------------------------------------- */ 70853cdbc3dSStefano Zampini 70953cdbc3dSStefano Zampini #undef __FUNCT__ 710da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 711da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 712da1bb401SStefano Zampini { 713da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 714da1bb401SStefano Zampini 715da1bb401SStefano Zampini PetscFunctionBegin; 716da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 717da1bb401SStefano Zampini PetscFunctionReturn(0); 718da1bb401SStefano Zampini } 7191e6b0712SBarry Smith 720da1bb401SStefano Zampini #undef __FUNCT__ 721da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 722da1bb401SStefano Zampini /*@ 723785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 724da1bb401SStefano Zampini 725785d1243SStefano Zampini Collective 726785d1243SStefano Zampini 727785d1243SStefano Zampini Input Parameters: 728785d1243SStefano Zampini . pc - the preconditioning context 729785d1243SStefano Zampini 730785d1243SStefano Zampini Output Parameters: 731785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 732785d1243SStefano Zampini 733785d1243SStefano Zampini Level: intermediate 734785d1243SStefano Zampini 7350f202f7eSStefano Zampini Notes: 7360f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 737785d1243SStefano Zampini 738785d1243SStefano Zampini .seealso: PCBDDC 739785d1243SStefano Zampini @*/ 740785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 741785d1243SStefano Zampini { 742785d1243SStefano Zampini PetscErrorCode ierr; 743785d1243SStefano Zampini 744785d1243SStefano Zampini PetscFunctionBegin; 745785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 746785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 747785d1243SStefano Zampini PetscFunctionReturn(0); 748785d1243SStefano Zampini } 749785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 750785d1243SStefano Zampini 751785d1243SStefano Zampini #undef __FUNCT__ 752785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 753785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 754785d1243SStefano Zampini { 755785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 756785d1243SStefano Zampini 757785d1243SStefano Zampini PetscFunctionBegin; 758785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 759785d1243SStefano Zampini PetscFunctionReturn(0); 760785d1243SStefano Zampini } 761785d1243SStefano Zampini 762785d1243SStefano Zampini #undef __FUNCT__ 76382ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 764da1bb401SStefano Zampini /*@ 76582ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 766da1bb401SStefano Zampini 767785d1243SStefano Zampini Collective 768da1bb401SStefano Zampini 769da1bb401SStefano Zampini Input Parameters: 77028509bceSStefano Zampini . pc - the preconditioning context 771da1bb401SStefano Zampini 772da1bb401SStefano Zampini Output Parameters: 77328509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 774da1bb401SStefano Zampini 775da1bb401SStefano Zampini Level: intermediate 776da1bb401SStefano Zampini 777da1bb401SStefano Zampini Notes: 7780f202f7eSStefano 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). 7790f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 780da1bb401SStefano Zampini 781da1bb401SStefano Zampini .seealso: PCBDDC 782da1bb401SStefano Zampini @*/ 78382ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 784da1bb401SStefano Zampini { 785da1bb401SStefano Zampini PetscErrorCode ierr; 786da1bb401SStefano Zampini 787da1bb401SStefano Zampini PetscFunctionBegin; 788da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 78982ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 790da1bb401SStefano Zampini PetscFunctionReturn(0); 791da1bb401SStefano Zampini } 792da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 7931e6b0712SBarry Smith 794da1bb401SStefano Zampini #undef __FUNCT__ 79553cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 79653cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 79753cdbc3dSStefano Zampini { 79853cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 79953cdbc3dSStefano Zampini 80053cdbc3dSStefano Zampini PetscFunctionBegin; 80153cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 80253cdbc3dSStefano Zampini PetscFunctionReturn(0); 80353cdbc3dSStefano Zampini } 8041e6b0712SBarry Smith 80553cdbc3dSStefano Zampini #undef __FUNCT__ 80653cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 80753cdbc3dSStefano Zampini /*@ 808785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 80953cdbc3dSStefano Zampini 810785d1243SStefano Zampini Collective 811785d1243SStefano Zampini 812785d1243SStefano Zampini Input Parameters: 813785d1243SStefano Zampini . pc - the preconditioning context 814785d1243SStefano Zampini 815785d1243SStefano Zampini Output Parameters: 816785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 817785d1243SStefano Zampini 818785d1243SStefano Zampini Level: intermediate 819785d1243SStefano Zampini 8200f202f7eSStefano Zampini Notes: 8210f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 822785d1243SStefano Zampini 823785d1243SStefano Zampini .seealso: PCBDDC 824785d1243SStefano Zampini @*/ 825785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 826785d1243SStefano Zampini { 827785d1243SStefano Zampini PetscErrorCode ierr; 828785d1243SStefano Zampini 829785d1243SStefano Zampini PetscFunctionBegin; 830785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 831785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 832785d1243SStefano Zampini PetscFunctionReturn(0); 833785d1243SStefano Zampini } 834785d1243SStefano Zampini /* -------------------------------------------------------------------------- */ 835785d1243SStefano Zampini 836785d1243SStefano Zampini #undef __FUNCT__ 837785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 838785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 839785d1243SStefano Zampini { 840785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 841785d1243SStefano Zampini 842785d1243SStefano Zampini PetscFunctionBegin; 843785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 844785d1243SStefano Zampini PetscFunctionReturn(0); 845785d1243SStefano Zampini } 846785d1243SStefano Zampini 847785d1243SStefano Zampini #undef __FUNCT__ 84882ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 84953cdbc3dSStefano Zampini /*@ 85082ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 85153cdbc3dSStefano Zampini 852785d1243SStefano Zampini Collective 85353cdbc3dSStefano Zampini 85453cdbc3dSStefano Zampini Input Parameters: 85528509bceSStefano Zampini . pc - the preconditioning context 85653cdbc3dSStefano Zampini 85753cdbc3dSStefano Zampini Output Parameters: 85828509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 85953cdbc3dSStefano Zampini 86053cdbc3dSStefano Zampini Level: intermediate 86153cdbc3dSStefano Zampini 86253cdbc3dSStefano Zampini Notes: 8630f202f7eSStefano 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). 8640f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 86553cdbc3dSStefano Zampini 86653cdbc3dSStefano Zampini .seealso: PCBDDC 86753cdbc3dSStefano Zampini @*/ 86882ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 86953cdbc3dSStefano Zampini { 87053cdbc3dSStefano Zampini PetscErrorCode ierr; 87153cdbc3dSStefano Zampini 87253cdbc3dSStefano Zampini PetscFunctionBegin; 87353cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 87482ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 8750c7d97c5SJed Brown PetscFunctionReturn(0); 8760c7d97c5SJed Brown } 87736e030ebSStefano Zampini /* -------------------------------------------------------------------------- */ 8781e6b0712SBarry Smith 87936e030ebSStefano Zampini #undef __FUNCT__ 880da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 8811a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 88236e030ebSStefano Zampini { 88336e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 884da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 88556282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 886da1bb401SStefano Zampini PetscErrorCode ierr; 88736e030ebSStefano Zampini 88836e030ebSStefano Zampini PetscFunctionBegin; 8898687889aSStefano Zampini if (!nvtxs) { 8908687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 8918687889aSStefano Zampini PetscFunctionReturn(0); 8928687889aSStefano Zampini } 89356282151SStefano Zampini if (mat_graph->nvtxs_csr == nvtxs) { 89456282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 89556282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 89656282151SStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,nvtxs*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 89756282151SStefano Zampini } 89856282151SStefano Zampini } 89956282151SStefano Zampini if (!same_data) { 900674ae819SStefano Zampini /* free old CSR */ 901674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 902674ae819SStefano Zampini /* get CSR into graph structure */ 903da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 904854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 905785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 906674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 907674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 908da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 9091a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 9101a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 911674ae819SStefano Zampini } 912575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 91356282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 91456282151SStefano Zampini } 91536e030ebSStefano Zampini PetscFunctionReturn(0); 91636e030ebSStefano Zampini } 9171e6b0712SBarry Smith 91836e030ebSStefano Zampini #undef __FUNCT__ 919da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 92036e030ebSStefano Zampini /*@ 9210f202f7eSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 92236e030ebSStefano Zampini 92336e030ebSStefano Zampini Not collective 92436e030ebSStefano Zampini 92536e030ebSStefano Zampini Input Parameters: 92636e030ebSStefano Zampini + pc - the preconditioning context 9270f202f7eSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 92828509bceSStefano Zampini . xadj, adjncy - the CSR graph 92928509bceSStefano Zampini - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 93036e030ebSStefano Zampini 93136e030ebSStefano Zampini Level: intermediate 93236e030ebSStefano Zampini 93336e030ebSStefano Zampini Notes: 93436e030ebSStefano Zampini 93528509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 93636e030ebSStefano Zampini @*/ 9371a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 93836e030ebSStefano Zampini { 939575ad6abSStefano Zampini void (*f)(void) = 0; 94036e030ebSStefano Zampini PetscErrorCode ierr; 94136e030ebSStefano Zampini 94236e030ebSStefano Zampini PetscFunctionBegin; 94336e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 9448687889aSStefano Zampini if (nvtxs) { 945674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 946d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 9478687889aSStefano Zampini } 9486c4ed002SBarry Smith if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode); 9491a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 950575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 951575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 952575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 953575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 954575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 955da1bb401SStefano Zampini } 95636e030ebSStefano Zampini PetscFunctionReturn(0); 95736e030ebSStefano Zampini } 9589c0446d6SStefano Zampini /* -------------------------------------------------------------------------- */ 9591e6b0712SBarry Smith 9609c0446d6SStefano Zampini #undef __FUNCT__ 96163602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 96263602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 96363602bcaSStefano Zampini { 96463602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 96563602bcaSStefano Zampini PetscInt i; 96656282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 96763602bcaSStefano Zampini PetscErrorCode ierr; 96863602bcaSStefano Zampini 96963602bcaSStefano Zampini PetscFunctionBegin; 97056282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 97156282151SStefano Zampini for (i=0;i<n_is;i++) { 97256282151SStefano Zampini PetscBool isequalt; 97356282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 97456282151SStefano Zampini if (!isequalt) break; 97556282151SStefano Zampini } 97656282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 97756282151SStefano Zampini } 97856282151SStefano Zampini for (i=0;i<n_is;i++) { 97956282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 98056282151SStefano Zampini } 98163602bcaSStefano Zampini /* Destroy ISes if they were already set */ 98263602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 98363602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 98463602bcaSStefano Zampini } 98563602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 98663602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 98763602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 98863602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 98963602bcaSStefano Zampini } 99063602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 99163602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 99263602bcaSStefano Zampini /* allocate space then set */ 993d02579f5SStefano Zampini if (n_is) { 994d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 995d02579f5SStefano Zampini } 99663602bcaSStefano Zampini for (i=0;i<n_is;i++) { 99763602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 99863602bcaSStefano Zampini } 99963602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 100063602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 100156282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 100263602bcaSStefano Zampini PetscFunctionReturn(0); 100363602bcaSStefano Zampini } 100463602bcaSStefano Zampini 100563602bcaSStefano Zampini #undef __FUNCT__ 100663602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 100763602bcaSStefano Zampini /*@ 100863602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 100963602bcaSStefano Zampini 101063602bcaSStefano Zampini Collective 101163602bcaSStefano Zampini 101263602bcaSStefano Zampini Input Parameters: 101363602bcaSStefano Zampini + pc - the preconditioning context 10140f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10150f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 101663602bcaSStefano Zampini 101763602bcaSStefano Zampini Level: intermediate 101863602bcaSStefano Zampini 10190f202f7eSStefano Zampini Notes: 10200f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 102163602bcaSStefano Zampini 102263602bcaSStefano Zampini .seealso: PCBDDC 102363602bcaSStefano Zampini @*/ 102463602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 102563602bcaSStefano Zampini { 102663602bcaSStefano Zampini PetscInt i; 102763602bcaSStefano Zampini PetscErrorCode ierr; 102863602bcaSStefano Zampini 102963602bcaSStefano Zampini PetscFunctionBegin; 103063602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 103163602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 103263602bcaSStefano Zampini for (i=0;i<n_is;i++) { 103363602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 103463602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 103563602bcaSStefano Zampini } 1036e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 103763602bcaSStefano Zampini PetscFunctionReturn(0); 103863602bcaSStefano Zampini } 103963602bcaSStefano Zampini /* -------------------------------------------------------------------------- */ 104063602bcaSStefano Zampini 104163602bcaSStefano Zampini #undef __FUNCT__ 10429c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 10439c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 10449c0446d6SStefano Zampini { 10459c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 10469c0446d6SStefano Zampini PetscInt i; 104756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 10489c0446d6SStefano Zampini PetscErrorCode ierr; 10499c0446d6SStefano Zampini 10509c0446d6SStefano Zampini PetscFunctionBegin; 105156282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 105256282151SStefano Zampini for (i=0;i<n_is;i++) { 105356282151SStefano Zampini PetscBool isequalt; 105456282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 105556282151SStefano Zampini if (!isequalt) break; 105656282151SStefano Zampini } 105756282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 105856282151SStefano Zampini } 105956282151SStefano Zampini for (i=0;i<n_is;i++) { 106056282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 106156282151SStefano Zampini } 1062da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 10639c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 10649c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 10659c0446d6SStefano Zampini } 1066d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 106763602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 106863602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 106963602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 107063602bcaSStefano Zampini } 107163602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 107263602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1073da1bb401SStefano Zampini /* allocate space then set */ 1074d02579f5SStefano Zampini if (n_is) { 1075785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1076d02579f5SStefano Zampini } 10779c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1078da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 10799c0446d6SStefano Zampini } 10809c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 108163602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 108256282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 10839c0446d6SStefano Zampini PetscFunctionReturn(0); 10849c0446d6SStefano Zampini } 10851e6b0712SBarry Smith 10869c0446d6SStefano Zampini #undef __FUNCT__ 10879c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 10889c0446d6SStefano Zampini /*@ 108963602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 10909c0446d6SStefano Zampini 109163602bcaSStefano Zampini Collective 10929c0446d6SStefano Zampini 10939c0446d6SStefano Zampini Input Parameters: 10949c0446d6SStefano Zampini + pc - the preconditioning context 10950f202f7eSStefano Zampini . n_is - number of index sets defining the fields 10960f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 10979c0446d6SStefano Zampini 10989c0446d6SStefano Zampini Level: intermediate 10999c0446d6SStefano Zampini 11000f202f7eSStefano Zampini Notes: 11010f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 11029c0446d6SStefano Zampini 11039c0446d6SStefano Zampini .seealso: PCBDDC 11049c0446d6SStefano Zampini @*/ 11059c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 11069c0446d6SStefano Zampini { 11072b510759SStefano Zampini PetscInt i; 11089c0446d6SStefano Zampini PetscErrorCode ierr; 11099c0446d6SStefano Zampini 11109c0446d6SStefano Zampini PetscFunctionBegin; 11119c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 111263602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 11132b510759SStefano Zampini for (i=0;i<n_is;i++) { 111463602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 111563602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 11162b510759SStefano Zampini } 11179c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 11189c0446d6SStefano Zampini PetscFunctionReturn(0); 11199c0446d6SStefano Zampini } 1120906d46d4SStefano Zampini 1121da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1122534831adSStefano Zampini #undef __FUNCT__ 1123534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 1124534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1125534831adSStefano Zampini /* 1126534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1127534831adSStefano Zampini guess if a transformation of basis approach has been selected. 11289c0446d6SStefano Zampini 1129534831adSStefano Zampini Input Parameter: 1130534831adSStefano Zampini + pc - the preconditioner contex 1131534831adSStefano Zampini 1132534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1133534831adSStefano Zampini 1134534831adSStefano Zampini Notes: 1135534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1136534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1137534831adSStefano Zampini */ 1138534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1139534831adSStefano Zampini { 1140534831adSStefano Zampini PetscErrorCode ierr; 1141534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1142534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 11433972b0daSStefano Zampini Vec used_vec; 114427b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1145534831adSStefano Zampini 1146534831adSStefano Zampini PetscFunctionBegin; 11471f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 114885c4d303SStefano Zampini if (ksp) { 11491f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 115085c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 115127b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 115227b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1153f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 11541f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 115585c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 115685c4d303SStefano Zampini } 115785c4d303SStefano Zampini } 1158fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1159fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1160fc17d649SStefano Zampini } 11611f4df5f7SStefano Zampini 116285c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 116362a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 116462a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 116562a6ff1dSStefano Zampini } 116662a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 116762a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 116862a6ff1dSStefano Zampini } 11698d00608fSStefano Zampini 117027b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 11713972b0daSStefano Zampini if (x) { 11723972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 11733972b0daSStefano Zampini used_vec = x; 11748d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 11753972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 11763972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 11773972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 117827b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1179266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1180266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1181266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 11823972b0daSStefano Zampini } 11838efcfb23SStefano Zampini 11848efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 11853972b0daSStefano Zampini if (ksp) { 1186a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 11878efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 11888efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 11893972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 11903972b0daSStefano Zampini } 11913972b0daSStefano Zampini } 11923308cffdSStefano Zampini 11938d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11943972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 119570c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 11963975b054SStefano Zampini IS dirIS = NULL; 11973975b054SStefano Zampini 1198a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 11993975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 12003975b054SStefano Zampini if (dirIS) { 1201906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1202785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 12032b095fd8SStefano Zampini PetscScalar *array_x; 12042b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1205785d1243SStefano Zampini 12063972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 12073972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1208e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1209e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1210e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1211e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 121282ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 12133972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 12142b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12153972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12162fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 12173972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12182b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12193972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1220e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1221e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12228d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12231b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 12248efcfb23SStefano Zampini } 1225a07ea27aSStefano Zampini } 1226b76ba322SStefano Zampini 12278efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 12288d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 122927b6a85dSStefano Zampini /* save the original rhs */ 123027b6a85dSStefano Zampini if (save_rhs) { 123127b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 123227b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 12338d00608fSStefano Zampini } 12348d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12353972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 123627b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 12373972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 12388efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 123927b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 12407acc28cbSStefano Zampini if (ksp) { 12417acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 12427acc28cbSStefano Zampini } 12433308cffdSStefano Zampini } 12448efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1245b76ba322SStefano Zampini 1246fc17d649SStefano Zampini /* compute initial vector in benign space if needed 124727b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 124827b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 124927b6a85dSStefano Zampini if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) { 12501f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 12511f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 12520369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 12530369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 12540369aaf7SStefano Zampini } 12554fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 125627b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 12571dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 12583bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 12591f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 12601f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 12611f4df5f7SStefano Zampini } 12621f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 126327b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 126427b6a85dSStefano Zampini if (save_rhs) { 126527b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 126627b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 126792e3dcfbSStefano Zampini } 126827b6a85dSStefano Zampini if (pcbddc->rhs_change) { 12690369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 127027b6a85dSStefano Zampini } else { 127127b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 127227b6a85dSStefano Zampini } 12730369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 127427b6a85dSStefano Zampini } 127527b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 12760369aaf7SStefano Zampini } 12771f4df5f7SStefano Zampini #if 0 12781f4df5f7SStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 12791f4df5f7SStefano Zampini Vec v; 12801f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 12811f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 12821f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 12831f4df5f7SStefano Zampini PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)),"LEVEL %d: IS CORRECTION BENIGN?\n",pcbddc->current_level); 12841f4df5f7SStefano Zampini PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc))); 12851f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 12861f4df5f7SStefano Zampini } 12871f4df5f7SStefano Zampini #endif 12880369aaf7SStefano Zampini 12890369aaf7SStefano Zampini /* set initial guess if using PCG */ 12900369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 12910369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 12921dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 129327b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 12941dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 12951dd7afcfSStefano Zampini } else { 12961dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 12971dd7afcfSStefano Zampini } 12981dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12991dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13001dd7afcfSStefano Zampini } else { 13010369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13020369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13031dd7afcfSStefano Zampini } 13040369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13051dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 13061dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 13071dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13081dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13091dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 13101dd7afcfSStefano Zampini } else { 13110369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13120369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13131dd7afcfSStefano Zampini } 13140369aaf7SStefano Zampini if (ksp) { 13150369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 13160369aaf7SStefano Zampini } 1317266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1318266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13190369aaf7SStefano Zampini } 1320534831adSStefano Zampini PetscFunctionReturn(0); 1321534831adSStefano Zampini } 1322906d46d4SStefano Zampini 1323534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1324534831adSStefano Zampini #undef __FUNCT__ 1325534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1326534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1327534831adSStefano Zampini /* 1328534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1329534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1330534831adSStefano Zampini 1331534831adSStefano Zampini Input Parameter: 1332534831adSStefano Zampini + pc - the preconditioner contex 1333534831adSStefano Zampini 1334534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1335534831adSStefano Zampini 1336534831adSStefano Zampini Notes: 1337534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1338534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1339534831adSStefano Zampini */ 1340534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1341534831adSStefano Zampini { 1342534831adSStefano Zampini PetscErrorCode ierr; 1343534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1344534831adSStefano Zampini 1345534831adSStefano Zampini PetscFunctionBegin; 13463972b0daSStefano Zampini /* add solution removed in presolve */ 13476bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 134827b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 13493425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 135027b6a85dSStefano Zampini } else if (pcbddc->benign_compute_correction) { 135127b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13523425bc38SStefano Zampini } 135327b6a85dSStefano Zampini } 135427b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 135527b6a85dSStefano Zampini 1356266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 13578d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 135827b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1359fb223d50SStefano Zampini } 13608d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 13618efcfb23SStefano Zampini /* restore ksp guess state */ 13628efcfb23SStefano Zampini if (ksp) { 13638efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 13648efcfb23SStefano Zampini } 1365534831adSStefano Zampini PetscFunctionReturn(0); 1366534831adSStefano Zampini } 1367534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 136853cdbc3dSStefano Zampini #undef __FUNCT__ 136953cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 13700c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13710c7d97c5SJed Brown /* 13720c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 13730c7d97c5SJed Brown by setting data structures and options. 13740c7d97c5SJed Brown 13750c7d97c5SJed Brown Input Parameter: 137653cdbc3dSStefano Zampini + pc - the preconditioner context 13770c7d97c5SJed Brown 13780c7d97c5SJed Brown Application Interface Routine: PCSetUp() 13790c7d97c5SJed Brown 13800c7d97c5SJed Brown Notes: 13810c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 13820c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 13830c7d97c5SJed Brown */ 138453cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 13850c7d97c5SJed Brown { 13860c7d97c5SJed Brown PetscErrorCode ierr; 13870c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 13885e8657edSStefano Zampini Mat_IS* matis; 138908122e43SStefano Zampini MatNullSpace nearnullspace; 1390c9ed8603SStefano Zampini IS zerodiag = NULL; 139191e8d312SStefano Zampini PetscInt nrows,ncols; 139208122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 13938de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 13945e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 13950c7d97c5SJed Brown 13960c7d97c5SJed Brown PetscFunctionBegin; 13975e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 13986c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 139991e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 14006c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 14015e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1402f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 14033b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1404fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1405f4ddd8eeSStefano Zampini /* split work */ 1406674ae819SStefano Zampini if (pc->setupcalled) { 1407d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1408674ae819SStefano Zampini computetopography = PETSC_FALSE; 1409674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1410674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1411674ae819SStefano Zampini computetopography = PETSC_TRUE; 1412674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1413674ae819SStefano Zampini } 1414674ae819SStefano Zampini } else { 1415674ae819SStefano Zampini computetopography = PETSC_TRUE; 1416674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1417674ae819SStefano Zampini } 1418fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1419fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1420fb180af4SStefano Zampini } 142156282151SStefano Zampini pcbddc->recompute_topography = computetopography; 14228de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1423b087196eSStefano Zampini 1424b087196eSStefano Zampini /* check parameters' compatibility */ 1425b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 14269d54b7f4SStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0); 1427bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1428862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1429862806e4SStefano Zampini 14305a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 143116909a7fSStefano Zampini if (pcbddc->switch_static) { 143216909a7fSStefano Zampini PetscBool ismatis; 143316909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 143416909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 143516909a7fSStefano Zampini } 143616909a7fSStefano Zampini 1437f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 143870cf5478SStefano Zampini if (pcbddc->dbg_flag) { 143970cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 144058a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 14411575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1442f4ddd8eeSStefano Zampini } 144358a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1444f4ddd8eeSStefano Zampini } 1445f4ddd8eeSStefano Zampini 14465e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 14475e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 14485e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 14495e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 14505e8657edSStefano Zampini } else { 1451b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 14525e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 14535e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1454d16cbb6bSStefano Zampini } 1455d16cbb6bSStefano Zampini 14564f1b2e48SStefano Zampini /* detect local disconnected subdomains if requested and not done before */ 14574f1b2e48SStefano Zampini if (pcbddc->detect_disconnected && !pcbddc->n_local_subs) { 14584f1b2e48SStefano Zampini ierr = MatDetectDisconnectedComponents(pcbddc->local_mat,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 14594f1b2e48SStefano Zampini } 14604f1b2e48SStefano Zampini 14611f4df5f7SStefano Zampini /* compute topology info in local ordering */ 14621f4df5f7SStefano Zampini if (pcbddc->recompute_topography) { 14631f4df5f7SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 14641f4df5f7SStefano Zampini } 14651f4df5f7SStefano Zampini 14664f1b2e48SStefano Zampini /* 14671dd7afcfSStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) 14684f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 14694f1b2e48SStefano Zampini */ 14701dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1471d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 14729f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 14739f47a83aSStefano Zampini 147405b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1475339f8db1SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat */ 1476339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1477a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1478c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 14791dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 14801dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 14811dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 14821dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 14831dd7afcfSStefano Zampini } else { 1484a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 14851dd7afcfSStefano Zampini } 1486a3df083aSStefano Zampini } else { 14879f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1488674ae819SStefano Zampini } 1489a3df083aSStefano Zampini } 149027b6a85dSStefano Zampini 1491f1a72664SStefano Zampini /* propagate relevant information */ 14924f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 14933301b35fSStefano Zampini if (matis->A->symmetric_set) { 14943301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 14953301b35fSStefano Zampini } 1496e496cd5dSStefano Zampini #endif 149706a4e24aSStefano Zampini if (matis->A->symmetric_set) { 149806a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 149906a4e24aSStefano Zampini } 150006a4e24aSStefano Zampini if (matis->A->spd_set) { 150106a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 150206a4e24aSStefano Zampini } 1503e496cd5dSStefano Zampini 15045e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 15055e8657edSStefano Zampini { 15065e8657edSStefano Zampini Mat temp_mat; 15075e8657edSStefano Zampini 15085e8657edSStefano Zampini temp_mat = matis->A; 15095e8657edSStefano Zampini matis->A = pcbddc->local_mat; 15105e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 15115e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15125e8657edSStefano Zampini matis->A = temp_mat; 15135e8657edSStefano Zampini } 1514684f6988SStefano Zampini 151581d14e9dSStefano Zampini /* Analyze interface */ 1516674ae819SStefano Zampini if (computetopography) { 1517674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15188de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1519345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 1520345ecf6cSStefano Zampini SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1521345ecf6cSStefano Zampini } 1522674ae819SStefano Zampini } 1523fb8d54d4SStefano Zampini 15245408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 15255408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 15265408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 15275408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 152809f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 15295408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 153009f581a4SStefano Zampini } 15315408967cSStefano Zampini #endif 15324f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 153306f24817SStefano Zampini 1534b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1535684f6988SStefano Zampini if (computesolvers) { 1536*b334f244SStefano Zampini PCBDDCSubSchurs sub_schurs; 1537d5574798SStefano Zampini 1538d5574798SStefano Zampini if (computesubschurs && computetopography) { 153908122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1540b1b3d7a2SStefano Zampini } 15419d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 15429d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 15439d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 15449d54b7f4SStefano Zampini } 1545*b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1546*b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 15472070dbb6SStefano Zampini if (computesubschurs) { 154808122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 15492070dbb6SStefano Zampini } 1550d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1551d5574798SStefano Zampini } else { 1552d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 15532070dbb6SStefano Zampini if (computesubschurs) { 1554d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1555d5574798SStefano Zampini } 15562070dbb6SStefano Zampini } 155708122e43SStefano Zampini if (pcbddc->adaptive_selection) { 155808122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 15598de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1560b7eb3628SStefano Zampini } 1561b1b3d7a2SStefano Zampini } 1562684f6988SStefano Zampini 1563f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1564fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1565f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1566f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1567f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1568f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1569f4ddd8eeSStefano Zampini } else { 1570f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1571f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1572f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1573165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1574f4ddd8eeSStefano Zampini PetscInt i; 1575165b64e2SStefano Zampini const Vec *nearnullvecs; 1576165b64e2SStefano Zampini PetscObjectState state; 1577165b64e2SStefano Zampini PetscInt nnsp_size; 1578165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1579f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1580f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1581165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1582f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1583f4ddd8eeSStefano Zampini break; 1584f4ddd8eeSStefano Zampini } 1585f4ddd8eeSStefano Zampini } 1586f4ddd8eeSStefano Zampini } 1587f4ddd8eeSStefano Zampini } 1588f4ddd8eeSStefano Zampini } else { 1589f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1590f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1591f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1592f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1593f4ddd8eeSStefano Zampini } 1594f4ddd8eeSStefano Zampini } 1595f4ddd8eeSStefano Zampini 1596f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1597727cdba6SStefano Zampini /* reset primal space flags */ 1598f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1599727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 16008de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1601727cdba6SStefano Zampini /* It also sets the primal space flags */ 1602674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1603e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1604f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 16053975b054SStefano Zampini } 16065e8657edSStefano Zampini 16073975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 16085e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16095e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 16105e8657edSStefano Zampini 16115e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16124f1b2e48SStefano Zampini if (pcbddc->benign_change) { 16131dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1614c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1615c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1616c263805aSStefano Zampini } 16175e8657edSStefano Zampini /* get submatrices */ 16185e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16195e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16205e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16215e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16225e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16235e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16243975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16253975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16269c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1627b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16285e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16295e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16305e8657edSStefano Zampini } 1631b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 163299cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1633b96c3477SStefano Zampini /* SetUp Scaling operator */ 16349d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1635674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16360c7d97c5SJed Brown } 16379d54b7f4SStefano Zampini } 16381dd7afcfSStefano Zampini /* mark topography as done */ 163956282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 16400369aaf7SStefano Zampini 16411dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 16421dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 16431dd7afcfSStefano Zampini 164458a03d70SStefano Zampini if (pcbddc->dbg_flag) { 164558a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 16462b510759SStefano Zampini } 16470c7d97c5SJed Brown PetscFunctionReturn(0); 16480c7d97c5SJed Brown } 16490c7d97c5SJed Brown 16500c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 16510c7d97c5SJed Brown /* 165250efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 16530c7d97c5SJed Brown 16540c7d97c5SJed Brown Input Parameters: 16550f202f7eSStefano Zampini + pc - the preconditioner context 16560f202f7eSStefano Zampini - r - input vector (global) 16570c7d97c5SJed Brown 16580c7d97c5SJed Brown Output Parameter: 16590c7d97c5SJed Brown . z - output vector (global) 16600c7d97c5SJed Brown 16610c7d97c5SJed Brown Application Interface Routine: PCApply() 16620c7d97c5SJed Brown */ 16630c7d97c5SJed Brown #undef __FUNCT__ 16640c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 166553cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 16660c7d97c5SJed Brown { 16670c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 16680c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1669b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 16700c7d97c5SJed Brown PetscErrorCode ierr; 16713b03a366Sstefano_zampini const PetscScalar one = 1.0; 16723b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 16732617d88aSStefano Zampini const PetscScalar zero = 0.0; 16740c7d97c5SJed Brown 16750c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 16760c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1677b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 16780c7d97c5SJed Brown 16790c7d97c5SJed Brown PetscFunctionBegin; 16801dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 16811dd7afcfSStefano Zampini Vec swap; 168227b6a85dSStefano Zampini 168327b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 16841dd7afcfSStefano Zampini swap = pcbddc->work_change; 16851dd7afcfSStefano Zampini pcbddc->work_change = r; 16861dd7afcfSStefano Zampini r = swap; 16871dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 16881dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 16891dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 16901dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 16911dd7afcfSStefano Zampini } 16921dd7afcfSStefano Zampini } 169327b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1694015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1695efc2fbd9SStefano Zampini } 169627b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1697b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 16980c7d97c5SJed Brown /* First Dirichlet solve */ 16990c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17000c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17010c7d97c5SJed Brown /* 17020c7d97c5SJed Brown Assembling right hand side for BDDC operator 1703b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1704674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 17050c7d97c5SJed Brown */ 1706b097fa66SStefano Zampini if (n_D) { 1707b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 17080c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 170916909a7fSStefano Zampini if (pcbddc->switch_static) { 171016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 171116909a7fSStefano Zampini 171216909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 171316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 171416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 171516909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 171616909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 171716909a7fSStefano Zampini } else { 171816909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 171916909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 172016909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 172116909a7fSStefano Zampini } 172216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 172316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 172416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 172516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 172616909a7fSStefano Zampini } else { 1727b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 172816909a7fSStefano Zampini } 1729b097fa66SStefano Zampini } else { 1730b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1731b097fa66SStefano Zampini } 17320c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17330c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1734674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1735b76ba322SStefano Zampini } else { 17364fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1737674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1738b76ba322SStefano Zampini } 17394fee134fSStefano Zampini } 1740b76ba322SStefano Zampini 17412617d88aSStefano Zampini /* Apply interface preconditioner 17422617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1743dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 17442617d88aSStefano Zampini 1745674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1746674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 17470c7d97c5SJed Brown 17483b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 17490c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17500c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1751b097fa66SStefano Zampini if (n_B) { 175216909a7fSStefano Zampini if (pcbddc->switch_static) { 175316909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 175416909a7fSStefano Zampini 175516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175916909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 176016909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 176116909a7fSStefano Zampini } else { 176216909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 176316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 176416909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 176516909a7fSStefano Zampini } 176616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 176716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 176816909a7fSStefano Zampini } else { 17690c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 177016909a7fSStefano Zampini } 177116909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 177216909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 177316909a7fSStefano Zampini 177416909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 177516909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 177616909a7fSStefano Zampini } else { 177716909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 177816909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 177916909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 178016909a7fSStefano Zampini } 1781b097fa66SStefano Zampini } 1782df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1783efc2fbd9SStefano Zampini 178427b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1785b097fa66SStefano Zampini if (pcbddc->switch_static) { 1786b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1787b097fa66SStefano Zampini } else { 1788b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1789b097fa66SStefano Zampini } 17900c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17910c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1792b097fa66SStefano Zampini } else { 1793b097fa66SStefano Zampini if (pcbddc->switch_static) { 1794b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1795b097fa66SStefano Zampini } else { 1796b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1797b097fa66SStefano Zampini } 1798b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1799b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1800b097fa66SStefano Zampini } 180127b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 18021dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 18031dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 18041dd7afcfSStefano Zampini } 1805015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1806efc2fbd9SStefano Zampini } 18071f4df5f7SStefano Zampini 18081dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18091dd7afcfSStefano Zampini Vec swap; 18101dd7afcfSStefano Zampini 18111dd7afcfSStefano Zampini swap = r; 18121dd7afcfSStefano Zampini r = pcbddc->work_change; 18131dd7afcfSStefano Zampini pcbddc->work_change = swap; 18141dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 18151dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 18161dd7afcfSStefano Zampini } 18170c7d97c5SJed Brown PetscFunctionReturn(0); 18180c7d97c5SJed Brown } 181950efa1b5SStefano Zampini 182050efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 182150efa1b5SStefano Zampini /* 182250efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 182350efa1b5SStefano Zampini 182450efa1b5SStefano Zampini Input Parameters: 18250f202f7eSStefano Zampini + pc - the preconditioner context 18260f202f7eSStefano Zampini - r - input vector (global) 182750efa1b5SStefano Zampini 182850efa1b5SStefano Zampini Output Parameter: 182950efa1b5SStefano Zampini . z - output vector (global) 183050efa1b5SStefano Zampini 183150efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 183250efa1b5SStefano Zampini */ 183350efa1b5SStefano Zampini #undef __FUNCT__ 183450efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 183550efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 183650efa1b5SStefano Zampini { 183750efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 183850efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1839b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 184050efa1b5SStefano Zampini PetscErrorCode ierr; 184150efa1b5SStefano Zampini const PetscScalar one = 1.0; 184250efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 184350efa1b5SStefano Zampini const PetscScalar zero = 0.0; 184450efa1b5SStefano Zampini 184550efa1b5SStefano Zampini PetscFunctionBegin; 18461dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18471dd7afcfSStefano Zampini Vec swap; 184827b6a85dSStefano Zampini 184927b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 18501dd7afcfSStefano Zampini swap = pcbddc->work_change; 18511dd7afcfSStefano Zampini pcbddc->work_change = r; 18521dd7afcfSStefano Zampini r = swap; 185327b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 185427b6a85dSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 185527b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 185627b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 18571dd7afcfSStefano Zampini } 185827b6a85dSStefano Zampini } 185927b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1860537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1861537c1cdfSStefano Zampini } 186227b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1863b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 186450efa1b5SStefano Zampini /* First Dirichlet solve */ 186550efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186650efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186750efa1b5SStefano Zampini /* 186850efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1869b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 187050efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 187150efa1b5SStefano Zampini */ 1872b097fa66SStefano Zampini if (n_D) { 1873b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 187450efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 187516909a7fSStefano Zampini if (pcbddc->switch_static) { 187616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 187716909a7fSStefano Zampini 187816909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 187916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 188016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 188116909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 188216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 188316909a7fSStefano Zampini } else { 188416909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 188516909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 188616909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 188716909a7fSStefano Zampini } 188816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 188916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189216909a7fSStefano Zampini } else { 1893b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 189416909a7fSStefano Zampini } 1895b097fa66SStefano Zampini } else { 1896b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1897b097fa66SStefano Zampini } 189850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 189950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190050efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 190150efa1b5SStefano Zampini } else { 190250efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 190350efa1b5SStefano Zampini } 190450efa1b5SStefano Zampini 190550efa1b5SStefano Zampini /* Apply interface preconditioner 190650efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1907dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 190850efa1b5SStefano Zampini 190950efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 191050efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 191150efa1b5SStefano Zampini 191250efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 191350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 191450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1915b097fa66SStefano Zampini if (n_B) { 191616909a7fSStefano Zampini if (pcbddc->switch_static) { 191716909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 191816909a7fSStefano Zampini 191916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 192416909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 192516909a7fSStefano Zampini } else { 192616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 192716909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 192816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 192916909a7fSStefano Zampini } 193016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 193116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 193216909a7fSStefano Zampini } else { 193350efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 193416909a7fSStefano Zampini } 193516909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 193616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 193716909a7fSStefano Zampini 193816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 193916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 194016909a7fSStefano Zampini } else { 194116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 194216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 194316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 194416909a7fSStefano Zampini } 1945b097fa66SStefano Zampini } 1946b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 194727b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1948b097fa66SStefano Zampini if (pcbddc->switch_static) { 1949b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1950b097fa66SStefano Zampini } else { 1951b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1952b097fa66SStefano Zampini } 195350efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 195450efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1955b097fa66SStefano Zampini } else { 1956b097fa66SStefano Zampini if (pcbddc->switch_static) { 1957b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1958b097fa66SStefano Zampini } else { 1959b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1960b097fa66SStefano Zampini } 1961b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1962b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1963b097fa66SStefano Zampini } 196427b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 1965537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1966537c1cdfSStefano Zampini } 19671dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19681dd7afcfSStefano Zampini Vec swap; 19691dd7afcfSStefano Zampini 19701dd7afcfSStefano Zampini swap = r; 19711dd7afcfSStefano Zampini r = pcbddc->work_change; 19721dd7afcfSStefano Zampini pcbddc->work_change = swap; 19731dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19741dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19751dd7afcfSStefano Zampini } 197650efa1b5SStefano Zampini PetscFunctionReturn(0); 197750efa1b5SStefano Zampini } 1978da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1979674ae819SStefano Zampini 1980da1bb401SStefano Zampini #undef __FUNCT__ 1981da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1982da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1983da1bb401SStefano Zampini { 1984da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1985da1bb401SStefano Zampini PetscErrorCode ierr; 1986da1bb401SStefano Zampini 1987da1bb401SStefano Zampini PetscFunctionBegin; 1988674ae819SStefano Zampini /* free BDDC custom data */ 1989674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1990674ae819SStefano Zampini /* destroy objects related to topography */ 1991674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1992674ae819SStefano Zampini /* free allocated graph structure */ 1993da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1994b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1995b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 199634a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1997674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 199834a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1999674ae819SStefano Zampini /* free solvers stuff */ 2000674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 200162a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 200262a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 200362a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 20041dd7afcfSStefano Zampini /* free data created by PCIS */ 20051dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 20063425bc38SStefano Zampini /* remove functions */ 2007906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2008674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 200930368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2010bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 20112b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2012b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 20132b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2014bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 201582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2016bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 201782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2018bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 201982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2020bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2021785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2022bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 202363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2024bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2025bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2026bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2027bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2028a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2029674ae819SStefano Zampini /* Free the private data structure */ 2030674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2031da1bb401SStefano Zampini PetscFunctionReturn(0); 2032da1bb401SStefano Zampini } 20333425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20341e6b0712SBarry Smith 20353425bc38SStefano Zampini #undef __FUNCT__ 2036a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC" 2037a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2038a06fd7f2SStefano Zampini { 2039a06fd7f2SStefano Zampini PetscFunctionBegin; 2040a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2041a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2042a06fd7f2SStefano Zampini } 2043a06fd7f2SStefano Zampini 2044a06fd7f2SStefano Zampini #undef __FUNCT__ 20453425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 20463425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 20473425bc38SStefano Zampini { 2048674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2049266e20e9SStefano Zampini Vec work; 20503425bc38SStefano Zampini PC_IS* pcis; 20513425bc38SStefano Zampini PC_BDDC* pcbddc; 20523425bc38SStefano Zampini PetscErrorCode ierr; 20530c7d97c5SJed Brown 20543425bc38SStefano Zampini PetscFunctionBegin; 20553425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20563425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20573425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20583425bc38SStefano Zampini 2059c08af4c6SStefano Zampini /* 2060c08af4c6SStefano Zampini change of basis for physical rhs if needed 2061c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2062c08af4c6SStefano Zampini */ 2063fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2064266e20e9SStefano Zampini ierr = VecDuplicate(standard_rhs,&work);CHKERRQ(ierr); 2065266e20e9SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,standard_rhs,NULL);CHKERRQ(ierr); 2066266e20e9SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,standard_rhs,work);CHKERRQ(ierr); 2067fc17d649SStefano Zampini } else { 2068266e20e9SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,standard_rhs,NULL);CHKERRQ(ierr); 2069266e20e9SStefano Zampini ierr = PetscObjectReference((PetscObject)standard_rhs);CHKERRQ(ierr); 2070266e20e9SStefano Zampini work = standard_rhs; 2071fc17d649SStefano Zampini } 20723425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2073266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2074266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2075fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2076fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2077266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2078266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2079674ae819SStefano Zampini /* Apply partition of unity */ 20803425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2081266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 20828eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 20833425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 20843425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20853425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 20863425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2087266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2088266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2089266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2090266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2091266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2092266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 20933425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 20943425bc38SStefano Zampini } 2095266e20e9SStefano Zampini ierr = VecDestroy(&work);CHKERRQ(ierr); 2096266e20e9SStefano Zampini ierr = VecCopy(pcbddc->original_rhs,standard_rhs);CHKERRQ(ierr); 20973425bc38SStefano Zampini /* BDDC rhs */ 20983425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 20998eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21003425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21013425bc38SStefano Zampini } 2102fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21033425bc38SStefano Zampini /* apply BDDC */ 2104dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2105266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21063425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 21073425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 21083425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 21093425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21103425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21113425bc38SStefano Zampini PetscFunctionReturn(0); 21123425bc38SStefano Zampini } 21131e6b0712SBarry Smith 21143425bc38SStefano Zampini #undef __FUNCT__ 21153425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 21163425bc38SStefano Zampini /*@ 21170f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 21183425bc38SStefano Zampini 21193425bc38SStefano Zampini Collective 21203425bc38SStefano Zampini 21213425bc38SStefano Zampini Input Parameters: 21220f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 21230f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 21243425bc38SStefano Zampini 21253425bc38SStefano Zampini Output Parameters: 21260f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 21273425bc38SStefano Zampini 21283425bc38SStefano Zampini Level: developer 21293425bc38SStefano Zampini 21303425bc38SStefano Zampini Notes: 21313425bc38SStefano Zampini 21320f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 21333425bc38SStefano Zampini @*/ 21343425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21353425bc38SStefano Zampini { 2136674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21373425bc38SStefano Zampini PetscErrorCode ierr; 21383425bc38SStefano Zampini 21393425bc38SStefano Zampini PetscFunctionBegin; 2140266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2141266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2142266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 21433425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2144163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 21453425bc38SStefano Zampini PetscFunctionReturn(0); 21463425bc38SStefano Zampini } 21473425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 21481e6b0712SBarry Smith 21493425bc38SStefano Zampini #undef __FUNCT__ 21503425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 21513425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 21523425bc38SStefano Zampini { 2153674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21543425bc38SStefano Zampini PC_IS* pcis; 21553425bc38SStefano Zampini PC_BDDC* pcbddc; 21563425bc38SStefano Zampini PetscErrorCode ierr; 21573425bc38SStefano Zampini 21583425bc38SStefano Zampini PetscFunctionBegin; 21593425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21603425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21613425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21623425bc38SStefano Zampini 21633425bc38SStefano Zampini /* apply B_delta^T */ 21643425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21653425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21663425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 21673425bc38SStefano Zampini /* compute rhs for BDDC application */ 21683425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21698eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21703425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21713425bc38SStefano Zampini } 2172fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21733425bc38SStefano Zampini /* apply BDDC */ 2174dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 21753425bc38SStefano Zampini /* put values into standard global vector */ 21763425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21773425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21788eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 21793425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 21803425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 21813425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 21823425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21833425bc38SStefano Zampini } 21843425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21853425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2186266e20e9SStefano Zampini /* add p0 solution to final solution */ 2187266e20e9SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,standard_sol,PETSC_FALSE);CHKERRQ(ierr); 2188fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2189fc17d649SStefano Zampini Vec v2; 2190fc17d649SStefano Zampini ierr = VecDuplicate(standard_sol,&v2);CHKERRQ(ierr); 2191fc17d649SStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,standard_sol,v2);CHKERRQ(ierr); 2192fc17d649SStefano Zampini ierr = VecCopy(v2,standard_sol);CHKERRQ(ierr); 2193fc17d649SStefano Zampini ierr = VecDestroy(&v2);CHKERRQ(ierr); 2194fc17d649SStefano Zampini } 21953308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 21963425bc38SStefano Zampini PetscFunctionReturn(0); 21973425bc38SStefano Zampini } 21981e6b0712SBarry Smith 21993425bc38SStefano Zampini #undef __FUNCT__ 22003425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 22013425bc38SStefano Zampini /*@ 22020f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 22033425bc38SStefano Zampini 22043425bc38SStefano Zampini Collective 22053425bc38SStefano Zampini 22063425bc38SStefano Zampini Input Parameters: 22070f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 22080f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 22093425bc38SStefano Zampini 22103425bc38SStefano Zampini Output Parameters: 22110f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 22123425bc38SStefano Zampini 22133425bc38SStefano Zampini Level: developer 22143425bc38SStefano Zampini 22153425bc38SStefano Zampini Notes: 22163425bc38SStefano Zampini 22170f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 22183425bc38SStefano Zampini @*/ 22193425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22203425bc38SStefano Zampini { 2221674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22223425bc38SStefano Zampini PetscErrorCode ierr; 22233425bc38SStefano Zampini 22243425bc38SStefano Zampini PetscFunctionBegin; 2225266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2226266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2227266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 22283425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2229163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 22303425bc38SStefano Zampini PetscFunctionReturn(0); 22313425bc38SStefano Zampini } 22323425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 22331e6b0712SBarry Smith 2234f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2235edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2236f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2237f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2238edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2239f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2240674ae819SStefano Zampini 22413425bc38SStefano Zampini #undef __FUNCT__ 22423425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 22433425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 22443425bc38SStefano Zampini { 2245674ae819SStefano Zampini 2246674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 22473425bc38SStefano Zampini Mat newmat; 2248674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 22493425bc38SStefano Zampini PC newpc; 2250ce94432eSBarry Smith MPI_Comm comm; 22513425bc38SStefano Zampini PetscErrorCode ierr; 22523425bc38SStefano Zampini 22533425bc38SStefano Zampini PetscFunctionBegin; 2254ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 22553425bc38SStefano Zampini /* FETIDP linear matrix */ 22563425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 22573425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 22583425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 22593425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2260edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 22613425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 22623425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 22633425bc38SStefano Zampini /* FETIDP preconditioner */ 22643425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 22653425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 22663425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 22673425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 22683425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 22693425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2270edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 22713425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 227223ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 22733425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 22743425bc38SStefano Zampini /* return pointers for objects created */ 22753425bc38SStefano Zampini *fetidp_mat=newmat; 22763425bc38SStefano Zampini *fetidp_pc=newpc; 22773425bc38SStefano Zampini PetscFunctionReturn(0); 22783425bc38SStefano Zampini } 22791e6b0712SBarry Smith 22803425bc38SStefano Zampini #undef __FUNCT__ 22813425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 22823425bc38SStefano Zampini /*@ 22830f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 22843425bc38SStefano Zampini 22853425bc38SStefano Zampini Collective 22863425bc38SStefano Zampini 22873425bc38SStefano Zampini Input Parameters: 22880f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 228928509bceSStefano Zampini 229028509bceSStefano Zampini Output Parameters: 22910f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 22920f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 229328509bceSStefano Zampini 229428509bceSStefano Zampini Options Database Keys: 22950f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 22963425bc38SStefano Zampini 22973425bc38SStefano Zampini Level: developer 22983425bc38SStefano Zampini 22993425bc38SStefano Zampini Notes: 23000f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 23013425bc38SStefano Zampini 23020f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 23033425bc38SStefano Zampini @*/ 23043425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 23053425bc38SStefano Zampini { 23063425bc38SStefano Zampini PetscErrorCode ierr; 23073425bc38SStefano Zampini 23083425bc38SStefano Zampini PetscFunctionBegin; 23093425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 23103425bc38SStefano Zampini if (pc->setupcalled) { 2311516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2312f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 23133425bc38SStefano Zampini PetscFunctionReturn(0); 23143425bc38SStefano Zampini } 23150c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2316da1bb401SStefano Zampini /*MC 2317da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 23180c7d97c5SJed Brown 231928509bceSStefano Zampini An implementation of the BDDC preconditioner based on 232028509bceSStefano Zampini 232128509bceSStefano Zampini .vb 232228509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 232328509bceSStefano 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 232428509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 23250f202f7eSStefano 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 232628509bceSStefano Zampini .ve 232728509bceSStefano Zampini 232828509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 232928509bceSStefano Zampini 23300f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 233128509bceSStefano Zampini 233228509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 233328509bceSStefano Zampini 2334b6fdb6dfSStefano 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. 2335b6fdb6dfSStefano Zampini 2336c7017625SStefano 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). 233728509bceSStefano Zampini 23380f202f7eSStefano 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() 233930368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 234028509bceSStefano Zampini 23410f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 234228509bceSStefano Zampini 23430f202f7eSStefano 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. 23440f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 234528509bceSStefano Zampini 23460f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 234728509bceSStefano Zampini 2348df4d28bfSStefano 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. 234928509bceSStefano Zampini 23500f202f7eSStefano 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. 23510f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 23520f202f7eSStefano Zampini 23530f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 23540f202f7eSStefano Zampini 23550f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 23560f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 23570f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 23580f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 23590f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 23600f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 23610f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 236228509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 23630f202f7eSStefano 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) 23640f202f7eSStefano Zampini . -pc_bddc_redistribute <0> - size of a subset of processors where the coarse problem will be remapped (the value is ignored if not at the coarsest level) 23650f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 23660f202f7eSStefano 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) 2367df4d28bfSStefano Zampini . -pc_bddc_adaptive_threshold <0.0> - when a value greater than one is specified, adaptive selection of constraints is performed on edges and faces (requires deluxe scaling and MUMPS or MKL_PARDISO installed) 236828509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 236928509bceSStefano Zampini 237028509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 237128509bceSStefano Zampini .vb 237228509bceSStefano Zampini -pc_bddc_dirichlet_ 237328509bceSStefano Zampini -pc_bddc_neumann_ 237428509bceSStefano Zampini -pc_bddc_coarse_ 237528509bceSStefano Zampini .ve 23760f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 237728509bceSStefano Zampini 23780f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 237928509bceSStefano Zampini .vb 2380312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2381312be037SStefano Zampini -pc_bddc_neumann_lN_ 2382312be037SStefano Zampini -pc_bddc_coarse_lN_ 238328509bceSStefano Zampini .ve 23840f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 23850f202f7eSStefano 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. 23860f202f7eSStefano Zampini .vb 23870f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 23880f202f7eSStefano Zampini .ve 23890f202f7eSStefano 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 2390da1bb401SStefano Zampini 2391da1bb401SStefano Zampini Level: intermediate 2392da1bb401SStefano Zampini 2393b6fdb6dfSStefano Zampini Developer notes: 2394da1bb401SStefano Zampini 2395da1bb401SStefano Zampini Contributed by Stefano Zampini 2396da1bb401SStefano Zampini 2397da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2398da1bb401SStefano Zampini M*/ 2399b2573a8aSBarry Smith 2400da1bb401SStefano Zampini #undef __FUNCT__ 2401da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 24028cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2403da1bb401SStefano Zampini { 2404da1bb401SStefano Zampini PetscErrorCode ierr; 2405da1bb401SStefano Zampini PC_BDDC *pcbddc; 2406da1bb401SStefano Zampini 2407da1bb401SStefano Zampini PetscFunctionBegin; 2408da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2409b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2410da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2411da1bb401SStefano Zampini 2412da1bb401SStefano Zampini /* create PCIS data structure */ 2413da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2414da1bb401SStefano Zampini 241547d04d0dSStefano Zampini /* BDDC customization */ 241608a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 241747d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 241847d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 241947d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 242047d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 242147d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 242247d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2423fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2424fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 24253301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 242606a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 242727b6a85dSStefano Zampini pcbddc->benign_have_null = PETSC_FALSE; 242814f95afaSStefano Zampini pcbddc->vertex_size = 1; 242947d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2430b9d89cd5SStefano Zampini /* private */ 2431727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 24320e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 24330e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 24340e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2435e9189074SStefano Zampini pcbddc->n_vertices = 0; 2436727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2437fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 243868457ee5SStefano Zampini pcbddc->coarse_size = -1; 2439f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2440727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2441f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2442f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2443f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2444674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 244530368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 24463972b0daSStefano Zampini pcbddc->temp_solution = 0; 2447534831adSStefano Zampini pcbddc->original_rhs = 0; 2448534831adSStefano Zampini pcbddc->local_mat = 0; 2449534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2450b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2451da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2452da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2453da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2454da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 245515aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 245615aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2457da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2458da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2459da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2460da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2461da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2462da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2463da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2464da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2465da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2466da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2467785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2468785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2469785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 247060d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 247160d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 247263602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2473da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 247463602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2475da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 247685c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 247747d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 247847d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2479b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 24804fad6a16SStefano Zampini pcbddc->current_level = 0; 24812b510759SStefano Zampini pcbddc->max_levels = 0; 2482323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 248357de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 2484f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 24854f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 24864f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 24874f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 248881d14e9dSStefano Zampini 248981d14e9dSStefano Zampini /* benign subspace trick */ 249081d14e9dSStefano Zampini pcbddc->benign_change = 0; 249127b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 24920369aaf7SStefano Zampini pcbddc->benign_vec = 0; 24930369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 24940369aaf7SStefano Zampini pcbddc->benign_sf = 0; 24954f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 24964f1b2e48SStefano Zampini pcbddc->benign_n = 0; 24974f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 24984f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 24994f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2500b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 250181d14e9dSStefano Zampini 2502674ae819SStefano Zampini /* create local graph structure */ 2503674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2504674ae819SStefano Zampini 2505674ae819SStefano Zampini /* scaling */ 2506674ae819SStefano Zampini pcbddc->work_scaling = 0; 250734a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2508b96c3477SStefano Zampini 2509*b334f244SStefano Zampini /* sub schurs options */ 2510b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2511b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2512b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2513b96c3477SStefano Zampini 2514b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2515da1bb401SStefano Zampini 2516b7eb3628SStefano Zampini /* adaptivity */ 2517f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 251808122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2519f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2520b7eb3628SStefano Zampini 2521da1bb401SStefano Zampini /* function pointers */ 2522da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 252393bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2524da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2525da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2526da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 25276b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2528da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2529da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2530da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2531534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2532534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2533da1bb401SStefano Zampini 2534da1bb401SStefano Zampini /* composing function */ 2535906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2536674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 253730368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2538bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 25392b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2540b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 25412b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2542bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 254382ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2544bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 254582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2546bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 254782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2548bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 254982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2550bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 255163602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2552bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2553bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2554bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2555bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2556a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2557da1bb401SStefano Zampini PetscFunctionReturn(0); 2558da1bb401SStefano Zampini } 25593425bc38SStefano Zampini 2560