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 } 1277*2d4c4fecSStefano Zampini 1278*2d4c4fecSStefano Zampini /* dbg output */ 1279*2d4c4fecSStefano Zampini if (pcbddc->dbg_flag && pcbddc->benign_have_null) { 12801f4df5f7SStefano Zampini Vec v; 12811f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 12821f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 12831f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1284*2d4c4fecSStefano Zampini ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %d: IS CORRECTION BENIGN?\n",pcbddc->current_level);CHKERRQ(ierr); 1285*2d4c4fecSStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr); 12861f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 12871f4df5f7SStefano Zampini } 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; 1475669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 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 } 15225ca1260dSStefano Zampini if (pcbddc->benign_compute_nonetflux) { 1523669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1524669cc0f4SStefano Zampini 1525669cc0f4SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 1526669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1527669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1528669cc0f4SStefano Zampini } 1529674ae819SStefano Zampini } 1530fb8d54d4SStefano Zampini 15315408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 15325408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 15335408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 15345408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 153509f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 15365408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 153709f581a4SStefano Zampini } 15385408967cSStefano Zampini #endif 15394f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 154006f24817SStefano Zampini 1541b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1542684f6988SStefano Zampini if (computesolvers) { 1543b334f244SStefano Zampini PCBDDCSubSchurs sub_schurs; 1544d5574798SStefano Zampini 1545d5574798SStefano Zampini if (computesubschurs && computetopography) { 154608122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1547b1b3d7a2SStefano Zampini } 15489d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 15499d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 15509d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 15519d54b7f4SStefano Zampini } 1552b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1553b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 15542070dbb6SStefano Zampini if (computesubschurs) { 155508122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 15562070dbb6SStefano Zampini } 1557d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1558d5574798SStefano Zampini } else { 1559d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 15602070dbb6SStefano Zampini if (computesubschurs) { 1561d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1562d5574798SStefano Zampini } 15632070dbb6SStefano Zampini } 156408122e43SStefano Zampini if (pcbddc->adaptive_selection) { 156508122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 15668de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1567b7eb3628SStefano Zampini } 1568b1b3d7a2SStefano Zampini } 1569684f6988SStefano Zampini 1570f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1571fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1572f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1573f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1574f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1575f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1576f4ddd8eeSStefano Zampini } else { 1577f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1578f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1579f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1580165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1581f4ddd8eeSStefano Zampini PetscInt i; 1582165b64e2SStefano Zampini const Vec *nearnullvecs; 1583165b64e2SStefano Zampini PetscObjectState state; 1584165b64e2SStefano Zampini PetscInt nnsp_size; 1585165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1586f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1587f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1588165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1589f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1590f4ddd8eeSStefano Zampini break; 1591f4ddd8eeSStefano Zampini } 1592f4ddd8eeSStefano Zampini } 1593f4ddd8eeSStefano Zampini } 1594f4ddd8eeSStefano Zampini } 1595f4ddd8eeSStefano Zampini } else { 1596f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1597f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1598f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1599f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1600f4ddd8eeSStefano Zampini } 1601f4ddd8eeSStefano Zampini } 1602f4ddd8eeSStefano Zampini 1603f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1604727cdba6SStefano Zampini /* reset primal space flags */ 1605f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1606727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 16078de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1608727cdba6SStefano Zampini /* It also sets the primal space flags */ 1609674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1610e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1611f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 16123975b054SStefano Zampini } 16135e8657edSStefano Zampini 16143975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 16155e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16165e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 16175e8657edSStefano Zampini 16185e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16194f1b2e48SStefano Zampini if (pcbddc->benign_change) { 16201dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1621c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1622c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1623c263805aSStefano Zampini } 16245e8657edSStefano Zampini /* get submatrices */ 16255e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16265e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16275e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16285e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16295e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16305e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16313975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16323975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16339c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1634b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16355e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16365e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16375e8657edSStefano Zampini } 1638b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 163999cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1640b96c3477SStefano Zampini /* SetUp Scaling operator */ 16419d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1642674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16430c7d97c5SJed Brown } 16449d54b7f4SStefano Zampini } 16451dd7afcfSStefano Zampini /* mark topography as done */ 164656282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 16470369aaf7SStefano Zampini 16481dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 16491dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 16501dd7afcfSStefano Zampini 165158a03d70SStefano Zampini if (pcbddc->dbg_flag) { 165258a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 16532b510759SStefano Zampini } 16540c7d97c5SJed Brown PetscFunctionReturn(0); 16550c7d97c5SJed Brown } 16560c7d97c5SJed Brown 16570c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 16580c7d97c5SJed Brown /* 165950efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 16600c7d97c5SJed Brown 16610c7d97c5SJed Brown Input Parameters: 16620f202f7eSStefano Zampini + pc - the preconditioner context 16630f202f7eSStefano Zampini - r - input vector (global) 16640c7d97c5SJed Brown 16650c7d97c5SJed Brown Output Parameter: 16660c7d97c5SJed Brown . z - output vector (global) 16670c7d97c5SJed Brown 16680c7d97c5SJed Brown Application Interface Routine: PCApply() 16690c7d97c5SJed Brown */ 16700c7d97c5SJed Brown #undef __FUNCT__ 16710c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 167253cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 16730c7d97c5SJed Brown { 16740c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 16750c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1676b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 16770c7d97c5SJed Brown PetscErrorCode ierr; 16783b03a366Sstefano_zampini const PetscScalar one = 1.0; 16793b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 16802617d88aSStefano Zampini const PetscScalar zero = 0.0; 16810c7d97c5SJed Brown 16820c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 16830c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1684b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 16850c7d97c5SJed Brown 16860c7d97c5SJed Brown PetscFunctionBegin; 16871dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 16881dd7afcfSStefano Zampini Vec swap; 168927b6a85dSStefano Zampini 169027b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 16911dd7afcfSStefano Zampini swap = pcbddc->work_change; 16921dd7afcfSStefano Zampini pcbddc->work_change = r; 16931dd7afcfSStefano Zampini r = swap; 16941dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 16951dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 16961dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 16971dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 16981dd7afcfSStefano Zampini } 16991dd7afcfSStefano Zampini } 170027b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1701015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1702efc2fbd9SStefano Zampini } 170327b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1704b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 17050c7d97c5SJed Brown /* First Dirichlet solve */ 17060c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17070c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17080c7d97c5SJed Brown /* 17090c7d97c5SJed Brown Assembling right hand side for BDDC operator 1710b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1711674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 17120c7d97c5SJed Brown */ 1713b097fa66SStefano Zampini if (n_D) { 1714b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 17150c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 171616909a7fSStefano Zampini if (pcbddc->switch_static) { 171716909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 171816909a7fSStefano Zampini 171916909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 172016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 172116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 172216909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 172316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 172416909a7fSStefano Zampini } else { 172516909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 172616909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 172716909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 172816909a7fSStefano Zampini } 172916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 173016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 173116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 173216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 173316909a7fSStefano Zampini } else { 1734b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 173516909a7fSStefano Zampini } 1736b097fa66SStefano Zampini } else { 1737b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1738b097fa66SStefano Zampini } 17390c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17400c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1741674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1742b76ba322SStefano Zampini } else { 17434fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1744674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1745b76ba322SStefano Zampini } 17464fee134fSStefano Zampini } 1747b76ba322SStefano Zampini 17482617d88aSStefano Zampini /* Apply interface preconditioner 17492617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1750dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 17512617d88aSStefano Zampini 1752674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1753674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 17540c7d97c5SJed Brown 17553b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 17560c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17570c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1758b097fa66SStefano Zampini if (n_B) { 175916909a7fSStefano Zampini if (pcbddc->switch_static) { 176016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 176116909a7fSStefano Zampini 176216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 176316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 176416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 176516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 176616909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 176716909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 176816909a7fSStefano Zampini } else { 176916909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 177016909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 177116909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 177216909a7fSStefano Zampini } 177316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 177416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 177516909a7fSStefano Zampini } else { 17760c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 177716909a7fSStefano Zampini } 177816909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 177916909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 178016909a7fSStefano Zampini 178116909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 178216909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 178316909a7fSStefano Zampini } else { 178416909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 178516909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 178616909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 178716909a7fSStefano Zampini } 1788b097fa66SStefano Zampini } 1789df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1790efc2fbd9SStefano Zampini 179127b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1792b097fa66SStefano Zampini if (pcbddc->switch_static) { 1793b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1794b097fa66SStefano Zampini } else { 1795b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1796b097fa66SStefano Zampini } 17970c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17980c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1799b097fa66SStefano Zampini } else { 1800b097fa66SStefano Zampini if (pcbddc->switch_static) { 1801b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1802b097fa66SStefano Zampini } else { 1803b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1804b097fa66SStefano Zampini } 1805b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1806b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1807b097fa66SStefano Zampini } 180827b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 18091dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 18101dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 18111dd7afcfSStefano Zampini } 1812015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1813efc2fbd9SStefano Zampini } 18141f4df5f7SStefano Zampini 18151dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18161dd7afcfSStefano Zampini Vec swap; 18171dd7afcfSStefano Zampini 18181dd7afcfSStefano Zampini swap = r; 18191dd7afcfSStefano Zampini r = pcbddc->work_change; 18201dd7afcfSStefano Zampini pcbddc->work_change = swap; 18211dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 18221dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 18231dd7afcfSStefano Zampini } 18240c7d97c5SJed Brown PetscFunctionReturn(0); 18250c7d97c5SJed Brown } 182650efa1b5SStefano Zampini 182750efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 182850efa1b5SStefano Zampini /* 182950efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 183050efa1b5SStefano Zampini 183150efa1b5SStefano Zampini Input Parameters: 18320f202f7eSStefano Zampini + pc - the preconditioner context 18330f202f7eSStefano Zampini - r - input vector (global) 183450efa1b5SStefano Zampini 183550efa1b5SStefano Zampini Output Parameter: 183650efa1b5SStefano Zampini . z - output vector (global) 183750efa1b5SStefano Zampini 183850efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 183950efa1b5SStefano Zampini */ 184050efa1b5SStefano Zampini #undef __FUNCT__ 184150efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 184250efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 184350efa1b5SStefano Zampini { 184450efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 184550efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1846b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 184750efa1b5SStefano Zampini PetscErrorCode ierr; 184850efa1b5SStefano Zampini const PetscScalar one = 1.0; 184950efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 185050efa1b5SStefano Zampini const PetscScalar zero = 0.0; 185150efa1b5SStefano Zampini 185250efa1b5SStefano Zampini PetscFunctionBegin; 18531dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18541dd7afcfSStefano Zampini Vec swap; 185527b6a85dSStefano Zampini 185627b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 18571dd7afcfSStefano Zampini swap = pcbddc->work_change; 18581dd7afcfSStefano Zampini pcbddc->work_change = r; 18591dd7afcfSStefano Zampini r = swap; 186027b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 186127b6a85dSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 186227b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 186327b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 18641dd7afcfSStefano Zampini } 186527b6a85dSStefano Zampini } 186627b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1867537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1868537c1cdfSStefano Zampini } 186927b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1870b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 187150efa1b5SStefano Zampini /* First Dirichlet solve */ 187250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187450efa1b5SStefano Zampini /* 187550efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1876b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 187750efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 187850efa1b5SStefano Zampini */ 1879b097fa66SStefano Zampini if (n_D) { 1880b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 188150efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 188216909a7fSStefano Zampini if (pcbddc->switch_static) { 188316909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 188416909a7fSStefano Zampini 188516909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 188616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 188716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 188816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 188916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 189016909a7fSStefano Zampini } else { 189116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 189216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 189316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 189416909a7fSStefano Zampini } 189516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 189916909a7fSStefano Zampini } else { 1900b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 190116909a7fSStefano Zampini } 1902b097fa66SStefano Zampini } else { 1903b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1904b097fa66SStefano Zampini } 190550efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190650efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 190750efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 190850efa1b5SStefano Zampini } else { 190950efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 191050efa1b5SStefano Zampini } 191150efa1b5SStefano Zampini 191250efa1b5SStefano Zampini /* Apply interface preconditioner 191350efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1914dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 191550efa1b5SStefano Zampini 191650efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 191750efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 191850efa1b5SStefano Zampini 191950efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 192050efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 192150efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1922b097fa66SStefano Zampini if (n_B) { 192316909a7fSStefano Zampini if (pcbddc->switch_static) { 192416909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 192516909a7fSStefano Zampini 192616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 192916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 193016909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 193116909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 193216909a7fSStefano Zampini } else { 193316909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 193416909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 193516909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 193616909a7fSStefano Zampini } 193716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 193816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 193916909a7fSStefano Zampini } else { 194050efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 194116909a7fSStefano Zampini } 194216909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 194316909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 194416909a7fSStefano Zampini 194516909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 194616909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 194716909a7fSStefano Zampini } else { 194816909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 194916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 195016909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 195116909a7fSStefano Zampini } 1952b097fa66SStefano Zampini } 1953b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 195427b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1955b097fa66SStefano Zampini if (pcbddc->switch_static) { 1956b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1957b097fa66SStefano Zampini } else { 1958b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1959b097fa66SStefano Zampini } 196050efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 196150efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1962b097fa66SStefano Zampini } else { 1963b097fa66SStefano Zampini if (pcbddc->switch_static) { 1964b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1965b097fa66SStefano Zampini } else { 1966b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1967b097fa66SStefano Zampini } 1968b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1969b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1970b097fa66SStefano Zampini } 197127b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 1972537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1973537c1cdfSStefano Zampini } 19741dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19751dd7afcfSStefano Zampini Vec swap; 19761dd7afcfSStefano Zampini 19771dd7afcfSStefano Zampini swap = r; 19781dd7afcfSStefano Zampini r = pcbddc->work_change; 19791dd7afcfSStefano Zampini pcbddc->work_change = swap; 19801dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19811dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19821dd7afcfSStefano Zampini } 198350efa1b5SStefano Zampini PetscFunctionReturn(0); 198450efa1b5SStefano Zampini } 1985da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1986674ae819SStefano Zampini 1987da1bb401SStefano Zampini #undef __FUNCT__ 1988da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1989da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1990da1bb401SStefano Zampini { 1991da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1992da1bb401SStefano Zampini PetscErrorCode ierr; 1993da1bb401SStefano Zampini 1994da1bb401SStefano Zampini PetscFunctionBegin; 1995674ae819SStefano Zampini /* free BDDC custom data */ 1996674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1997674ae819SStefano Zampini /* destroy objects related to topography */ 1998674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1999674ae819SStefano Zampini /* free allocated graph structure */ 2000da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 2001b96c3477SStefano Zampini /* free allocated sub schurs structure */ 2002b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 200334a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2004674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 200534a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 2006674ae819SStefano Zampini /* free solvers stuff */ 2007674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 200862a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 200962a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 201062a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 20111dd7afcfSStefano Zampini /* free data created by PCIS */ 20121dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 20133425bc38SStefano Zampini /* remove functions */ 2014906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2015674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 201630368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2017bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 20182b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2019b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 20202b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2021bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 202282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2023bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 202482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2025bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 202682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2027bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2028785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2029bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 203063602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2031bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2032bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2033bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2034bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2035a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2036674ae819SStefano Zampini /* Free the private data structure */ 2037674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2038da1bb401SStefano Zampini PetscFunctionReturn(0); 2039da1bb401SStefano Zampini } 20403425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20411e6b0712SBarry Smith 20423425bc38SStefano Zampini #undef __FUNCT__ 2043a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC" 2044a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2045a06fd7f2SStefano Zampini { 2046a06fd7f2SStefano Zampini PetscFunctionBegin; 2047a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2048a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2049a06fd7f2SStefano Zampini } 2050a06fd7f2SStefano Zampini 2051a06fd7f2SStefano Zampini #undef __FUNCT__ 20523425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 20533425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 20543425bc38SStefano Zampini { 2055674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2056266e20e9SStefano Zampini Vec work; 20573425bc38SStefano Zampini PC_IS* pcis; 20583425bc38SStefano Zampini PC_BDDC* pcbddc; 20593425bc38SStefano Zampini PetscErrorCode ierr; 20600c7d97c5SJed Brown 20613425bc38SStefano Zampini PetscFunctionBegin; 20623425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20633425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20643425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20653425bc38SStefano Zampini 2066c08af4c6SStefano Zampini /* 2067c08af4c6SStefano Zampini change of basis for physical rhs if needed 2068c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2069c08af4c6SStefano Zampini */ 20703738a8e6SStefano Zampini if (!pcbddc->original_rhs) { 20713738a8e6SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 20723738a8e6SStefano Zampini } 20733738a8e6SStefano Zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 20743738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2075fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 20763738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 20773738a8e6SStefano Zampini work = pcbddc->work_change; 2078fc17d649SStefano Zampini } else { 20793738a8e6SStefano Zampini work = pcbddc->original_rhs; 2080fc17d649SStefano Zampini } 20813425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2082266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2083266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2084fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2085fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2086266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2087266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2088674ae819SStefano Zampini /* Apply partition of unity */ 20893425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2090266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 20918eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 20923425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 20933425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20943425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 20953425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2096266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2097266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2098266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2099266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2100266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2101266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21023425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21033425bc38SStefano Zampini } 21043425bc38SStefano Zampini /* BDDC rhs */ 21053425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 21068eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21073425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21083425bc38SStefano Zampini } 2109fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21103425bc38SStefano Zampini /* apply BDDC */ 2111dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2112266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21133425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 21143425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 21153425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 21163425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21173425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21183425bc38SStefano Zampini PetscFunctionReturn(0); 21193425bc38SStefano Zampini } 21201e6b0712SBarry Smith 21213425bc38SStefano Zampini #undef __FUNCT__ 21223425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 21233425bc38SStefano Zampini /*@ 21240f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 21253425bc38SStefano Zampini 21263425bc38SStefano Zampini Collective 21273425bc38SStefano Zampini 21283425bc38SStefano Zampini Input Parameters: 21290f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 21300f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 21313425bc38SStefano Zampini 21323425bc38SStefano Zampini Output Parameters: 21330f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 21343425bc38SStefano Zampini 21353425bc38SStefano Zampini Level: developer 21363425bc38SStefano Zampini 21373425bc38SStefano Zampini Notes: 21383425bc38SStefano Zampini 21390f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 21403425bc38SStefano Zampini @*/ 21413425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21423425bc38SStefano Zampini { 2143674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21443425bc38SStefano Zampini PetscErrorCode ierr; 21453425bc38SStefano Zampini 21463425bc38SStefano Zampini PetscFunctionBegin; 2147266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2148266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2149266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 21503425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2151163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 21523425bc38SStefano Zampini PetscFunctionReturn(0); 21533425bc38SStefano Zampini } 21543425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 21551e6b0712SBarry Smith 21563425bc38SStefano Zampini #undef __FUNCT__ 21573425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 21583425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 21593425bc38SStefano Zampini { 2160674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21613425bc38SStefano Zampini PC_IS* pcis; 21623425bc38SStefano Zampini PC_BDDC* pcbddc; 21633425bc38SStefano Zampini PetscErrorCode ierr; 21643425bc38SStefano Zampini 21653425bc38SStefano Zampini PetscFunctionBegin; 21663425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21673425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21683425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21693425bc38SStefano Zampini 21703425bc38SStefano Zampini /* apply B_delta^T */ 21713425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21723425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21733425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 21743425bc38SStefano Zampini /* compute rhs for BDDC application */ 21753425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21768eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21773425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21783425bc38SStefano Zampini } 2179fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21803425bc38SStefano Zampini /* apply BDDC */ 2181dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 21823425bc38SStefano Zampini /* put values into standard global vector */ 21833425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21843425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21858eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 21863425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 21873425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 21883425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 21893425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21903425bc38SStefano Zampini } 21913425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21923425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2193266e20e9SStefano Zampini /* add p0 solution to final solution */ 2194266e20e9SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,standard_sol,PETSC_FALSE);CHKERRQ(ierr); 2195fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2196fc17d649SStefano Zampini Vec v2; 2197fc17d649SStefano Zampini ierr = VecDuplicate(standard_sol,&v2);CHKERRQ(ierr); 2198fc17d649SStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,standard_sol,v2);CHKERRQ(ierr); 2199fc17d649SStefano Zampini ierr = VecCopy(v2,standard_sol);CHKERRQ(ierr); 2200fc17d649SStefano Zampini ierr = VecDestroy(&v2);CHKERRQ(ierr); 2201fc17d649SStefano Zampini } 22023308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 22033425bc38SStefano Zampini PetscFunctionReturn(0); 22043425bc38SStefano Zampini } 22051e6b0712SBarry Smith 22063425bc38SStefano Zampini #undef __FUNCT__ 22073425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 22083425bc38SStefano Zampini /*@ 22090f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 22103425bc38SStefano Zampini 22113425bc38SStefano Zampini Collective 22123425bc38SStefano Zampini 22133425bc38SStefano Zampini Input Parameters: 22140f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 22150f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 22163425bc38SStefano Zampini 22173425bc38SStefano Zampini Output Parameters: 22180f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 22193425bc38SStefano Zampini 22203425bc38SStefano Zampini Level: developer 22213425bc38SStefano Zampini 22223425bc38SStefano Zampini Notes: 22233425bc38SStefano Zampini 22240f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 22253425bc38SStefano Zampini @*/ 22263425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22273425bc38SStefano Zampini { 2228674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22293425bc38SStefano Zampini PetscErrorCode ierr; 22303425bc38SStefano Zampini 22313425bc38SStefano Zampini PetscFunctionBegin; 2232266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2233266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2234266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 22353425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2236163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 22373425bc38SStefano Zampini PetscFunctionReturn(0); 22383425bc38SStefano Zampini } 22393425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 22401e6b0712SBarry Smith 2241f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2242edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2243f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2244f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2245edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2246f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2247674ae819SStefano Zampini 22483425bc38SStefano Zampini #undef __FUNCT__ 22493425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 22503425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 22513425bc38SStefano Zampini { 2252674ae819SStefano Zampini 2253674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 22543425bc38SStefano Zampini Mat newmat; 2255674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 22563425bc38SStefano Zampini PC newpc; 2257ce94432eSBarry Smith MPI_Comm comm; 22583425bc38SStefano Zampini PetscErrorCode ierr; 22593425bc38SStefano Zampini 22603425bc38SStefano Zampini PetscFunctionBegin; 2261ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 22623425bc38SStefano Zampini /* FETIDP linear matrix */ 22633425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 22643425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 22653425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 22663425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2267edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 22683425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 22693425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 22703425bc38SStefano Zampini /* FETIDP preconditioner */ 22713425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 22723425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 22733425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 22743425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 22753425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 22763425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2277edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 22783425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 227923ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 22803425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 22813425bc38SStefano Zampini /* return pointers for objects created */ 22823425bc38SStefano Zampini *fetidp_mat=newmat; 22833425bc38SStefano Zampini *fetidp_pc=newpc; 22843425bc38SStefano Zampini PetscFunctionReturn(0); 22853425bc38SStefano Zampini } 22861e6b0712SBarry Smith 22873425bc38SStefano Zampini #undef __FUNCT__ 22883425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 22893425bc38SStefano Zampini /*@ 22900f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 22913425bc38SStefano Zampini 22923425bc38SStefano Zampini Collective 22933425bc38SStefano Zampini 22943425bc38SStefano Zampini Input Parameters: 22950f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 229628509bceSStefano Zampini 229728509bceSStefano Zampini Output Parameters: 22980f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 22990f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 230028509bceSStefano Zampini 230128509bceSStefano Zampini Options Database Keys: 23020f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 23033425bc38SStefano Zampini 23043425bc38SStefano Zampini Level: developer 23053425bc38SStefano Zampini 23063425bc38SStefano Zampini Notes: 23070f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 23083425bc38SStefano Zampini 23090f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 23103425bc38SStefano Zampini @*/ 23113425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 23123425bc38SStefano Zampini { 23133425bc38SStefano Zampini PetscErrorCode ierr; 23143425bc38SStefano Zampini 23153425bc38SStefano Zampini PetscFunctionBegin; 23163425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 23173425bc38SStefano Zampini if (pc->setupcalled) { 2318516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2319f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 23203425bc38SStefano Zampini PetscFunctionReturn(0); 23213425bc38SStefano Zampini } 23220c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2323da1bb401SStefano Zampini /*MC 2324da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 23250c7d97c5SJed Brown 232628509bceSStefano Zampini An implementation of the BDDC preconditioner based on 232728509bceSStefano Zampini 232828509bceSStefano Zampini .vb 232928509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 233028509bceSStefano 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 233128509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 23320f202f7eSStefano 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 233328509bceSStefano Zampini .ve 233428509bceSStefano Zampini 233528509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 233628509bceSStefano Zampini 23370f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 233828509bceSStefano Zampini 233928509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 234028509bceSStefano Zampini 2341b6fdb6dfSStefano 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. 2342b6fdb6dfSStefano Zampini 2343c7017625SStefano 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). 234428509bceSStefano Zampini 23450f202f7eSStefano 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() 234630368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 234728509bceSStefano Zampini 23480f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 234928509bceSStefano Zampini 23500f202f7eSStefano 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. 23510f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 235228509bceSStefano Zampini 23530f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 235428509bceSStefano Zampini 2355df4d28bfSStefano 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. 235628509bceSStefano Zampini 23570f202f7eSStefano 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. 23580f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 23590f202f7eSStefano Zampini 23600f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 23610f202f7eSStefano Zampini 23620f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 23630f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 23640f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 23650f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 23660f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 23670f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 23680f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 236928509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 23700f202f7eSStefano 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) 23710f202f7eSStefano 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) 23720f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 23730f202f7eSStefano 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) 2374df4d28bfSStefano 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) 237528509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 237628509bceSStefano Zampini 237728509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 237828509bceSStefano Zampini .vb 237928509bceSStefano Zampini -pc_bddc_dirichlet_ 238028509bceSStefano Zampini -pc_bddc_neumann_ 238128509bceSStefano Zampini -pc_bddc_coarse_ 238228509bceSStefano Zampini .ve 23830f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 238428509bceSStefano Zampini 23850f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 238628509bceSStefano Zampini .vb 2387312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2388312be037SStefano Zampini -pc_bddc_neumann_lN_ 2389312be037SStefano Zampini -pc_bddc_coarse_lN_ 239028509bceSStefano Zampini .ve 23910f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 23920f202f7eSStefano 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. 23930f202f7eSStefano Zampini .vb 23940f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 23950f202f7eSStefano Zampini .ve 23960f202f7eSStefano 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 2397da1bb401SStefano Zampini 2398da1bb401SStefano Zampini Level: intermediate 2399da1bb401SStefano Zampini 2400b6fdb6dfSStefano Zampini Developer notes: 2401da1bb401SStefano Zampini 2402da1bb401SStefano Zampini Contributed by Stefano Zampini 2403da1bb401SStefano Zampini 2404da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2405da1bb401SStefano Zampini M*/ 2406b2573a8aSBarry Smith 2407da1bb401SStefano Zampini #undef __FUNCT__ 2408da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 24098cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2410da1bb401SStefano Zampini { 2411da1bb401SStefano Zampini PetscErrorCode ierr; 2412da1bb401SStefano Zampini PC_BDDC *pcbddc; 2413da1bb401SStefano Zampini 2414da1bb401SStefano Zampini PetscFunctionBegin; 2415da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2416b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2417da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2418da1bb401SStefano Zampini 2419da1bb401SStefano Zampini /* create PCIS data structure */ 2420da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2421da1bb401SStefano Zampini 242247d04d0dSStefano Zampini /* BDDC customization */ 242308a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 242447d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 242547d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 242647d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 242747d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 242847d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 242947d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2430fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2431fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 24323301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 243306a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 243427b6a85dSStefano Zampini pcbddc->benign_have_null = PETSC_FALSE; 243514f95afaSStefano Zampini pcbddc->vertex_size = 1; 243647d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2437b9d89cd5SStefano Zampini /* private */ 2438727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 24390e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 24400e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 24410e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2442e9189074SStefano Zampini pcbddc->n_vertices = 0; 2443727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2444fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 244568457ee5SStefano Zampini pcbddc->coarse_size = -1; 2446f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2447727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2448f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2449f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2450f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2451674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 245230368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 24533972b0daSStefano Zampini pcbddc->temp_solution = 0; 2454534831adSStefano Zampini pcbddc->original_rhs = 0; 2455534831adSStefano Zampini pcbddc->local_mat = 0; 2456534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2457b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2458da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2459da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2460da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2461da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 246215aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 246315aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2464da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2465da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2466da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2467da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2468da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2469da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2470da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2471da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2472da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2473da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2474785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2475785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2476785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 247760d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 247860d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 247963602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2480da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 248163602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2482da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 248385c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 248447d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 248547d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2486b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 24874fad6a16SStefano Zampini pcbddc->current_level = 0; 24882b510759SStefano Zampini pcbddc->max_levels = 0; 2489323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 249057de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 2491f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 24924f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 24934f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 24944f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 249581d14e9dSStefano Zampini 249681d14e9dSStefano Zampini /* benign subspace trick */ 249781d14e9dSStefano Zampini pcbddc->benign_change = 0; 249827b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 24990369aaf7SStefano Zampini pcbddc->benign_vec = 0; 25000369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 25010369aaf7SStefano Zampini pcbddc->benign_sf = 0; 25024f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 25034f1b2e48SStefano Zampini pcbddc->benign_n = 0; 25044f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 25054f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 25064f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2507b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 250881d14e9dSStefano Zampini 2509674ae819SStefano Zampini /* create local graph structure */ 2510674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2511674ae819SStefano Zampini 2512674ae819SStefano Zampini /* scaling */ 2513674ae819SStefano Zampini pcbddc->work_scaling = 0; 251434a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2515b96c3477SStefano Zampini 2516b334f244SStefano Zampini /* sub schurs options */ 2517b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2518b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2519b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2520b96c3477SStefano Zampini 2521b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2522da1bb401SStefano Zampini 2523b7eb3628SStefano Zampini /* adaptivity */ 2524f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 252508122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2526f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2527b7eb3628SStefano Zampini 2528da1bb401SStefano Zampini /* function pointers */ 2529da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 253093bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2531da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2532da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2533da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 25346b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2535da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2536da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2537da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2538534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2539534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2540da1bb401SStefano Zampini 2541da1bb401SStefano Zampini /* composing function */ 2542906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2543674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 254430368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2545bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 25462b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2547b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 25482b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2549bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 255082ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2551bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 255282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2553bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 255482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2555bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 255682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2557bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 255863602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2559bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2560bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2561bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2562bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2563a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2564da1bb401SStefano Zampini PetscFunctionReturn(0); 2565da1bb401SStefano Zampini } 25663425bc38SStefano Zampini 2567