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); 150*fc17d649SStefano 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 } 1158*fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1159*fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1160*fc17d649SStefano 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; 11793972b0daSStefano Zampini } 11808efcfb23SStefano Zampini 11818efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 11823972b0daSStefano Zampini if (ksp) { 1183a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 11848efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 11858efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 11863972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 11873972b0daSStefano Zampini } 11883972b0daSStefano Zampini } 11893308cffdSStefano Zampini 11908d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 11913972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 119270c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 11933975b054SStefano Zampini IS dirIS = NULL; 11943975b054SStefano Zampini 1195a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 11963975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 11973975b054SStefano Zampini if (dirIS) { 1198906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1199785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 12002b095fd8SStefano Zampini PetscScalar *array_x; 12012b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1202785d1243SStefano Zampini 12033972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 12043972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1205e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1206e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1207e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1208e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 120982ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 12103972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 12112b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12123972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12132fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 12143972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 12152b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 12163972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1217e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1218e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 12198d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12201b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 12218efcfb23SStefano Zampini } 1222a07ea27aSStefano Zampini } 1223b76ba322SStefano Zampini 12248efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 12258d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 122627b6a85dSStefano Zampini /* save the original rhs */ 122727b6a85dSStefano Zampini if (save_rhs) { 122827b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 122927b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 12308d00608fSStefano Zampini } 12318d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 12323972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 123327b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 12343972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 12358efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 123627b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 12377acc28cbSStefano Zampini if (ksp) { 12387acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 12397acc28cbSStefano Zampini } 12403308cffdSStefano Zampini } 12418efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1242b76ba322SStefano Zampini 1243*fc17d649SStefano Zampini /* compute initial vector in benign space if needed 124427b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 124527b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 124627b6a85dSStefano Zampini if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) { 12471f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 12481f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 12490369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 12500369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 12510369aaf7SStefano Zampini } 12524fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 125327b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 12541dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 12553bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 12561f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 12571f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 12581f4df5f7SStefano Zampini } 12591f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 126027b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 126127b6a85dSStefano Zampini if (save_rhs) { 126227b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 126327b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 126492e3dcfbSStefano Zampini } 126527b6a85dSStefano Zampini if (pcbddc->rhs_change) { 12660369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 126727b6a85dSStefano Zampini } else { 126827b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 126927b6a85dSStefano Zampini } 12700369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 127127b6a85dSStefano Zampini } 127227b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 12730369aaf7SStefano Zampini } 12741f4df5f7SStefano Zampini #if 0 12751f4df5f7SStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 12761f4df5f7SStefano Zampini Vec v; 12771f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 12781f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 12791f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 12801f4df5f7SStefano Zampini PetscViewerASCIIPrintf(PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)),"LEVEL %d: IS CORRECTION BENIGN?\n",pcbddc->current_level); 12811f4df5f7SStefano Zampini PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc))); 12821f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 12831f4df5f7SStefano Zampini } 12841f4df5f7SStefano Zampini #endif 12850369aaf7SStefano Zampini 12860369aaf7SStefano Zampini /* set initial guess if using PCG */ 12870369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 12880369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 12891dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 129027b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 12911dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 12921dd7afcfSStefano Zampini } else { 12931dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 12941dd7afcfSStefano Zampini } 12951dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12961dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12971dd7afcfSStefano Zampini } else { 12980369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 12990369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 13001dd7afcfSStefano Zampini } 13010369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 13021dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 13031dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 13041dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13051dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13061dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 13071dd7afcfSStefano Zampini } else { 13080369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13090369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13101dd7afcfSStefano Zampini } 13110369aaf7SStefano Zampini if (ksp) { 13120369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 13130369aaf7SStefano Zampini } 13140369aaf7SStefano Zampini } 1315534831adSStefano Zampini PetscFunctionReturn(0); 1316534831adSStefano Zampini } 1317906d46d4SStefano Zampini 1318534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1319534831adSStefano Zampini #undef __FUNCT__ 1320534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1321534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 1322534831adSStefano Zampini /* 1323534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1324534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1325534831adSStefano Zampini 1326534831adSStefano Zampini Input Parameter: 1327534831adSStefano Zampini + pc - the preconditioner contex 1328534831adSStefano Zampini 1329534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1330534831adSStefano Zampini 1331534831adSStefano Zampini Notes: 1332534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1333534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1334534831adSStefano Zampini */ 1335534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1336534831adSStefano Zampini { 1337534831adSStefano Zampini PetscErrorCode ierr; 1338534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1339534831adSStefano Zampini 1340534831adSStefano Zampini PetscFunctionBegin; 13413972b0daSStefano Zampini /* add solution removed in presolve */ 13426bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 134327b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 13443425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 134527b6a85dSStefano Zampini } else if (pcbddc->benign_compute_correction) { 134627b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13473425bc38SStefano Zampini } 134827b6a85dSStefano Zampini } 134927b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 135027b6a85dSStefano Zampini 1351fb223d50SStefano Zampini /* restore rhs to its original state */ 13528d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 135327b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1354fb223d50SStefano Zampini } 13558d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 13568efcfb23SStefano Zampini /* restore ksp guess state */ 13578efcfb23SStefano Zampini if (ksp) { 13588efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 13598efcfb23SStefano Zampini } 1360534831adSStefano Zampini PetscFunctionReturn(0); 1361534831adSStefano Zampini } 1362534831adSStefano Zampini /* -------------------------------------------------------------------------- */ 136353cdbc3dSStefano Zampini #undef __FUNCT__ 136453cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 13650c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 13660c7d97c5SJed Brown /* 13670c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 13680c7d97c5SJed Brown by setting data structures and options. 13690c7d97c5SJed Brown 13700c7d97c5SJed Brown Input Parameter: 137153cdbc3dSStefano Zampini + pc - the preconditioner context 13720c7d97c5SJed Brown 13730c7d97c5SJed Brown Application Interface Routine: PCSetUp() 13740c7d97c5SJed Brown 13750c7d97c5SJed Brown Notes: 13760c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 13770c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 13780c7d97c5SJed Brown */ 137953cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 13800c7d97c5SJed Brown { 13810c7d97c5SJed Brown PetscErrorCode ierr; 13820c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 13835e8657edSStefano Zampini Mat_IS* matis; 138408122e43SStefano Zampini MatNullSpace nearnullspace; 1385c9ed8603SStefano Zampini IS zerodiag = NULL; 138691e8d312SStefano Zampini PetscInt nrows,ncols; 138708122e43SStefano Zampini PetscBool computetopography,computesolvers,computesubschurs; 13888de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 13895e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 13900c7d97c5SJed Brown 13910c7d97c5SJed Brown PetscFunctionBegin; 13925e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 13936c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 139491e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 13956c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 13965e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1397f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 13983b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1399fcd409f5SStefano Zampini Also, BDDC directly build the Dirichlet problem */ 1400f4ddd8eeSStefano Zampini /* split work */ 1401674ae819SStefano Zampini if (pc->setupcalled) { 1402d1e9a80fSBarry Smith if (pc->flag == SAME_NONZERO_PATTERN) { 1403674ae819SStefano Zampini computetopography = PETSC_FALSE; 1404674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1405674ae819SStefano Zampini } else { /* DIFFERENT_NONZERO_PATTERN */ 1406674ae819SStefano Zampini computetopography = PETSC_TRUE; 1407674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1408674ae819SStefano Zampini } 1409674ae819SStefano Zampini } else { 1410674ae819SStefano Zampini computetopography = PETSC_TRUE; 1411674ae819SStefano Zampini computesolvers = PETSC_TRUE; 1412674ae819SStefano Zampini } 1413fb180af4SStefano Zampini if (pcbddc->recompute_topography) { 1414fb180af4SStefano Zampini computetopography = PETSC_TRUE; 1415fb180af4SStefano Zampini } 141656282151SStefano Zampini pcbddc->recompute_topography = computetopography; 14178de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1418b087196eSStefano Zampini 1419b087196eSStefano Zampini /* check parameters' compatibility */ 1420b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 14219d54b7f4SStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0); 1422bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1423862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1424862806e4SStefano Zampini 14255a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 142616909a7fSStefano Zampini if (pcbddc->switch_static) { 142716909a7fSStefano Zampini PetscBool ismatis; 142816909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 142916909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 143016909a7fSStefano Zampini } 143116909a7fSStefano Zampini 1432f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 143370cf5478SStefano Zampini if (pcbddc->dbg_flag) { 143470cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 143558a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 14361575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1437f4ddd8eeSStefano Zampini } 143858a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1439f4ddd8eeSStefano Zampini } 1440f4ddd8eeSStefano Zampini 14415e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 14425e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 14435e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 14445e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 14455e8657edSStefano Zampini } else { 1446b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 14475e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 14485e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1449d16cbb6bSStefano Zampini } 1450d16cbb6bSStefano Zampini 14514f1b2e48SStefano Zampini /* detect local disconnected subdomains if requested and not done before */ 14524f1b2e48SStefano Zampini if (pcbddc->detect_disconnected && !pcbddc->n_local_subs) { 14534f1b2e48SStefano Zampini ierr = MatDetectDisconnectedComponents(pcbddc->local_mat,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 14544f1b2e48SStefano Zampini } 14554f1b2e48SStefano Zampini 14561f4df5f7SStefano Zampini /* compute topology info in local ordering */ 14571f4df5f7SStefano Zampini if (pcbddc->recompute_topography) { 14581f4df5f7SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 14591f4df5f7SStefano Zampini } 14601f4df5f7SStefano Zampini 14614f1b2e48SStefano Zampini /* 14621dd7afcfSStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) 14634f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 14644f1b2e48SStefano Zampini */ 14651dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1466d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 14679f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 14689f47a83aSStefano Zampini 146905b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1470339f8db1SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat */ 1471339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1472a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1473c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 14741dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 14751dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 14761dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 14771dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 14781dd7afcfSStefano Zampini } else { 1479a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 14801dd7afcfSStefano Zampini } 1481a3df083aSStefano Zampini } else { 14829f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1483674ae819SStefano Zampini } 1484a3df083aSStefano Zampini } 148527b6a85dSStefano Zampini 1486f1a72664SStefano Zampini /* propagate relevant information */ 14874f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 14883301b35fSStefano Zampini if (matis->A->symmetric_set) { 14893301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 14903301b35fSStefano Zampini } 1491e496cd5dSStefano Zampini #endif 149206a4e24aSStefano Zampini if (matis->A->symmetric_set) { 149306a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 149406a4e24aSStefano Zampini } 149506a4e24aSStefano Zampini if (matis->A->spd_set) { 149606a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 149706a4e24aSStefano Zampini } 1498e496cd5dSStefano Zampini 14995e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 15005e8657edSStefano Zampini { 15015e8657edSStefano Zampini Mat temp_mat; 15025e8657edSStefano Zampini 15035e8657edSStefano Zampini temp_mat = matis->A; 15045e8657edSStefano Zampini matis->A = pcbddc->local_mat; 15055e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 15065e8657edSStefano Zampini pcbddc->local_mat = matis->A; 15075e8657edSStefano Zampini matis->A = temp_mat; 15085e8657edSStefano Zampini } 1509684f6988SStefano Zampini 151081d14e9dSStefano Zampini /* Analyze interface */ 1511674ae819SStefano Zampini if (computetopography) { 1512674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 15138de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1514345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 1515345ecf6cSStefano Zampini SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1516345ecf6cSStefano Zampini } 1517674ae819SStefano Zampini } 1518fb8d54d4SStefano Zampini 15195408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 15205408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 15215408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 15225408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 152309f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 15245408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 152509f581a4SStefano Zampini } 15265408967cSStefano Zampini #endif 15274f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 152806f24817SStefano Zampini 1529b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1530684f6988SStefano Zampini if (computesolvers) { 1531d5574798SStefano Zampini PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1532d5574798SStefano Zampini 1533d5574798SStefano Zampini if (computesubschurs && computetopography) { 153408122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1535b1b3d7a2SStefano Zampini } 15369d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 15379d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 15389d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 15399d54b7f4SStefano Zampini } 1540df4d28bfSStefano Zampini if (sub_schurs->schur_explicit) { 15412070dbb6SStefano Zampini if (computesubschurs) { 154208122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 15432070dbb6SStefano Zampini } 1544d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1545d5574798SStefano Zampini } else { 1546d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 15472070dbb6SStefano Zampini if (computesubschurs) { 1548d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1549d5574798SStefano Zampini } 15502070dbb6SStefano Zampini } 155108122e43SStefano Zampini if (pcbddc->adaptive_selection) { 155208122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 15538de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1554b7eb3628SStefano Zampini } 1555b1b3d7a2SStefano Zampini } 1556684f6988SStefano Zampini 1557f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1558fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1559f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1560f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1561f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1562f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1563f4ddd8eeSStefano Zampini } else { 1564f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1565f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1566f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1567165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1568f4ddd8eeSStefano Zampini PetscInt i; 1569165b64e2SStefano Zampini const Vec *nearnullvecs; 1570165b64e2SStefano Zampini PetscObjectState state; 1571165b64e2SStefano Zampini PetscInt nnsp_size; 1572165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1573f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1574f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1575165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1576f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1577f4ddd8eeSStefano Zampini break; 1578f4ddd8eeSStefano Zampini } 1579f4ddd8eeSStefano Zampini } 1580f4ddd8eeSStefano Zampini } 1581f4ddd8eeSStefano Zampini } 1582f4ddd8eeSStefano Zampini } else { 1583f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1584f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1585f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1586f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1587f4ddd8eeSStefano Zampini } 1588f4ddd8eeSStefano Zampini } 1589f4ddd8eeSStefano Zampini 1590f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1591727cdba6SStefano Zampini /* reset primal space flags */ 1592f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1593727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 15948de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1595727cdba6SStefano Zampini /* It also sets the primal space flags */ 1596674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1597e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1598f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 15993975b054SStefano Zampini } 16005e8657edSStefano Zampini 16013975b054SStefano Zampini if (computesolvers || pcbddc->new_primal_space) { 16025e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 16035e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 16045e8657edSStefano Zampini 16055e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 16064f1b2e48SStefano Zampini if (pcbddc->benign_change) { 16071dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1608c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1609c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1610c263805aSStefano Zampini } 16115e8657edSStefano Zampini /* get submatrices */ 16125e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 16135e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 16145e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 16155e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 16165e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 16175e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 16183975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 16193975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 16209c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1621b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 16225e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 16235e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16245e8657edSStefano Zampini } 1625b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 162699cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1627b96c3477SStefano Zampini /* SetUp Scaling operator */ 16289d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1629674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16300c7d97c5SJed Brown } 16319d54b7f4SStefano Zampini } 16321dd7afcfSStefano Zampini /* mark topography as done */ 163356282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 16340369aaf7SStefano Zampini 16351dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 16361dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 16371dd7afcfSStefano Zampini 163858a03d70SStefano Zampini if (pcbddc->dbg_flag) { 163958a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 16402b510759SStefano Zampini } 16410c7d97c5SJed Brown PetscFunctionReturn(0); 16420c7d97c5SJed Brown } 16430c7d97c5SJed Brown 16440c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 16450c7d97c5SJed Brown /* 164650efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 16470c7d97c5SJed Brown 16480c7d97c5SJed Brown Input Parameters: 16490f202f7eSStefano Zampini + pc - the preconditioner context 16500f202f7eSStefano Zampini - r - input vector (global) 16510c7d97c5SJed Brown 16520c7d97c5SJed Brown Output Parameter: 16530c7d97c5SJed Brown . z - output vector (global) 16540c7d97c5SJed Brown 16550c7d97c5SJed Brown Application Interface Routine: PCApply() 16560c7d97c5SJed Brown */ 16570c7d97c5SJed Brown #undef __FUNCT__ 16580c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 165953cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 16600c7d97c5SJed Brown { 16610c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 16620c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1663b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 16640c7d97c5SJed Brown PetscErrorCode ierr; 16653b03a366Sstefano_zampini const PetscScalar one = 1.0; 16663b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 16672617d88aSStefano Zampini const PetscScalar zero = 0.0; 16680c7d97c5SJed Brown 16690c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 16700c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1671b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 16720c7d97c5SJed Brown 16730c7d97c5SJed Brown PetscFunctionBegin; 16741dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 16751dd7afcfSStefano Zampini Vec swap; 167627b6a85dSStefano Zampini 167727b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 16781dd7afcfSStefano Zampini swap = pcbddc->work_change; 16791dd7afcfSStefano Zampini pcbddc->work_change = r; 16801dd7afcfSStefano Zampini r = swap; 16811dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 16821dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 16831dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 16841dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 16851dd7afcfSStefano Zampini } 16861dd7afcfSStefano Zampini } 168727b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1688015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1689efc2fbd9SStefano Zampini } 169027b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1691b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 16920c7d97c5SJed Brown /* First Dirichlet solve */ 16930c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16940c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 16950c7d97c5SJed Brown /* 16960c7d97c5SJed Brown Assembling right hand side for BDDC operator 1697b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1698674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 16990c7d97c5SJed Brown */ 1700b097fa66SStefano Zampini if (n_D) { 1701b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 17020c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 170316909a7fSStefano Zampini if (pcbddc->switch_static) { 170416909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 170516909a7fSStefano Zampini 170616909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 170716909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 170816909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 170916909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 171016909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 171116909a7fSStefano Zampini } else { 171216909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 171316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 171416909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 171516909a7fSStefano Zampini } 171616909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 171716909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 171816909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 171916909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 172016909a7fSStefano Zampini } else { 1721b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 172216909a7fSStefano Zampini } 1723b097fa66SStefano Zampini } else { 1724b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1725b097fa66SStefano Zampini } 17260c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17270c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1728674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1729b76ba322SStefano Zampini } else { 17304fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1731674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1732b76ba322SStefano Zampini } 17334fee134fSStefano Zampini } 1734b76ba322SStefano Zampini 17352617d88aSStefano Zampini /* Apply interface preconditioner 17362617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1737dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 17382617d88aSStefano Zampini 1739674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1740674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 17410c7d97c5SJed Brown 17423b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 17430c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 17440c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1745b097fa66SStefano Zampini if (n_B) { 174616909a7fSStefano Zampini if (pcbddc->switch_static) { 174716909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 174816909a7fSStefano Zampini 174916909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175016909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 175316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 175416909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 175516909a7fSStefano Zampini } else { 175616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 175716909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 175816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 175916909a7fSStefano Zampini } 176016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 176116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 176216909a7fSStefano Zampini } else { 17630c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 176416909a7fSStefano Zampini } 176516909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 176616909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 176716909a7fSStefano Zampini 176816909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 176916909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 177016909a7fSStefano Zampini } else { 177116909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 177216909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 177316909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 177416909a7fSStefano Zampini } 1775b097fa66SStefano Zampini } 1776df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1777efc2fbd9SStefano Zampini 177827b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1779b097fa66SStefano Zampini if (pcbddc->switch_static) { 1780b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1781b097fa66SStefano Zampini } else { 1782b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1783b097fa66SStefano Zampini } 17840c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 17850c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1786b097fa66SStefano Zampini } else { 1787b097fa66SStefano Zampini if (pcbddc->switch_static) { 1788b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1789b097fa66SStefano Zampini } else { 1790b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1791b097fa66SStefano Zampini } 1792b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1793b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1794b097fa66SStefano Zampini } 179527b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 17961dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 17971dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 17981dd7afcfSStefano Zampini } 1799015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1800efc2fbd9SStefano Zampini } 18011f4df5f7SStefano Zampini 18021dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18031dd7afcfSStefano Zampini Vec swap; 18041dd7afcfSStefano Zampini 18051dd7afcfSStefano Zampini swap = r; 18061dd7afcfSStefano Zampini r = pcbddc->work_change; 18071dd7afcfSStefano Zampini pcbddc->work_change = swap; 18081dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 18091dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 18101dd7afcfSStefano Zampini } 18110c7d97c5SJed Brown PetscFunctionReturn(0); 18120c7d97c5SJed Brown } 181350efa1b5SStefano Zampini 181450efa1b5SStefano Zampini /* -------------------------------------------------------------------------- */ 181550efa1b5SStefano Zampini /* 181650efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 181750efa1b5SStefano Zampini 181850efa1b5SStefano Zampini Input Parameters: 18190f202f7eSStefano Zampini + pc - the preconditioner context 18200f202f7eSStefano Zampini - r - input vector (global) 182150efa1b5SStefano Zampini 182250efa1b5SStefano Zampini Output Parameter: 182350efa1b5SStefano Zampini . z - output vector (global) 182450efa1b5SStefano Zampini 182550efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 182650efa1b5SStefano Zampini */ 182750efa1b5SStefano Zampini #undef __FUNCT__ 182850efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 182950efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 183050efa1b5SStefano Zampini { 183150efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 183250efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1833b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 183450efa1b5SStefano Zampini PetscErrorCode ierr; 183550efa1b5SStefano Zampini const PetscScalar one = 1.0; 183650efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 183750efa1b5SStefano Zampini const PetscScalar zero = 0.0; 183850efa1b5SStefano Zampini 183950efa1b5SStefano Zampini PetscFunctionBegin; 18401dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 18411dd7afcfSStefano Zampini Vec swap; 184227b6a85dSStefano Zampini 184327b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 18441dd7afcfSStefano Zampini swap = pcbddc->work_change; 18451dd7afcfSStefano Zampini pcbddc->work_change = r; 18461dd7afcfSStefano Zampini r = swap; 184727b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 184827b6a85dSStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->use_exact_dirichlet_trick && pcbddc->change_interior) { 184927b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 185027b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 18511dd7afcfSStefano Zampini } 185227b6a85dSStefano Zampini } 185327b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1854537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1855537c1cdfSStefano Zampini } 185627b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1857b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 185850efa1b5SStefano Zampini /* First Dirichlet solve */ 185950efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186050efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 186150efa1b5SStefano Zampini /* 186250efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1863b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 186450efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 186550efa1b5SStefano Zampini */ 1866b097fa66SStefano Zampini if (n_D) { 1867b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 186850efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 186916909a7fSStefano Zampini if (pcbddc->switch_static) { 187016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 187116909a7fSStefano Zampini 187216909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 187316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 187416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 187516909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 187616909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 187716909a7fSStefano Zampini } else { 187816909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 187916909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 188016909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 188116909a7fSStefano Zampini } 188216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 188316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 188416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 188516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 188616909a7fSStefano Zampini } else { 1887b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 188816909a7fSStefano Zampini } 1889b097fa66SStefano Zampini } else { 1890b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1891b097fa66SStefano Zampini } 189250efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 189350efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 189450efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 189550efa1b5SStefano Zampini } else { 189650efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 189750efa1b5SStefano Zampini } 189850efa1b5SStefano Zampini 189950efa1b5SStefano Zampini /* Apply interface preconditioner 190050efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1901dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 190250efa1b5SStefano Zampini 190350efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 190450efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 190550efa1b5SStefano Zampini 190650efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 190750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 190850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1909b097fa66SStefano Zampini if (n_B) { 191016909a7fSStefano Zampini if (pcbddc->switch_static) { 191116909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 191216909a7fSStefano Zampini 191316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 191416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 191516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 191616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 191716909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 191816909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 191916909a7fSStefano Zampini } else { 192016909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 192116909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 192216909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 192316909a7fSStefano Zampini } 192416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 192516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 192616909a7fSStefano Zampini } else { 192750efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 192816909a7fSStefano Zampini } 192916909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 193016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 193116909a7fSStefano Zampini 193216909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 193316909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 193416909a7fSStefano Zampini } else { 193516909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 193616909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 193716909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 193816909a7fSStefano Zampini } 1939b097fa66SStefano Zampini } 1940b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 194127b6a85dSStefano Zampini if (!pcbddc->use_exact_dirichlet_trick && !pcbddc->benign_apply_coarse_only) { 1942b097fa66SStefano Zampini if (pcbddc->switch_static) { 1943b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1944b097fa66SStefano Zampini } else { 1945b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1946b097fa66SStefano Zampini } 194750efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 194850efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1949b097fa66SStefano Zampini } else { 1950b097fa66SStefano Zampini if (pcbddc->switch_static) { 1951b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1952b097fa66SStefano Zampini } else { 1953b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1954b097fa66SStefano Zampini } 1955b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1956b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1957b097fa66SStefano Zampini } 195827b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 1959537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1960537c1cdfSStefano Zampini } 19611dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19621dd7afcfSStefano Zampini Vec swap; 19631dd7afcfSStefano Zampini 19641dd7afcfSStefano Zampini swap = r; 19651dd7afcfSStefano Zampini r = pcbddc->work_change; 19661dd7afcfSStefano Zampini pcbddc->work_change = swap; 19671dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19681dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19691dd7afcfSStefano Zampini } 197050efa1b5SStefano Zampini PetscFunctionReturn(0); 197150efa1b5SStefano Zampini } 1972da1bb401SStefano Zampini /* -------------------------------------------------------------------------- */ 1973674ae819SStefano Zampini 1974da1bb401SStefano Zampini #undef __FUNCT__ 1975da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 1976da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 1977da1bb401SStefano Zampini { 1978da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1979da1bb401SStefano Zampini PetscErrorCode ierr; 1980da1bb401SStefano Zampini 1981da1bb401SStefano Zampini PetscFunctionBegin; 1982674ae819SStefano Zampini /* free BDDC custom data */ 1983674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1984674ae819SStefano Zampini /* destroy objects related to topography */ 1985674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1986674ae819SStefano Zampini /* free allocated graph structure */ 1987da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1988b96c3477SStefano Zampini /* free allocated sub schurs structure */ 1989b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 199034a97f8cSStefano Zampini /* destroy objects for scaling operator */ 1991674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 199234a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1993674ae819SStefano Zampini /* free solvers stuff */ 1994674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 199562a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 199662a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 199762a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 19981dd7afcfSStefano Zampini /* free data created by PCIS */ 19991dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 20003425bc38SStefano Zampini /* remove functions */ 2001906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2002674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 200330368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2004bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 20052b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2006b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 20072b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2008bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 200982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2010bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 201182ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2012bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 201382ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2014bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2015785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2016bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 201763602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2018bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2019bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2020bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2021bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2022a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2023674ae819SStefano Zampini /* Free the private data structure */ 2024674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2025da1bb401SStefano Zampini PetscFunctionReturn(0); 2026da1bb401SStefano Zampini } 20273425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 20281e6b0712SBarry Smith 20293425bc38SStefano Zampini #undef __FUNCT__ 2030a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC" 2031a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2032a06fd7f2SStefano Zampini { 2033a06fd7f2SStefano Zampini PetscFunctionBegin; 2034a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2035a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2036a06fd7f2SStefano Zampini } 2037a06fd7f2SStefano Zampini 2038a06fd7f2SStefano Zampini #undef __FUNCT__ 20393425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 20403425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 20413425bc38SStefano Zampini { 2042674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2043c08af4c6SStefano Zampini Vec copy_standard_rhs; 20443425bc38SStefano Zampini PC_IS* pcis; 20453425bc38SStefano Zampini PC_BDDC* pcbddc; 20463425bc38SStefano Zampini PetscErrorCode ierr; 20470c7d97c5SJed Brown 20483425bc38SStefano Zampini PetscFunctionBegin; 20493425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 20503425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 20513425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 20523425bc38SStefano Zampini 2053c08af4c6SStefano Zampini /* 2054c08af4c6SStefano Zampini change of basis for physical rhs if needed 2055c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2056c08af4c6SStefano Zampini TODO: better management when FETIDP will have its own class 2057c08af4c6SStefano Zampini */ 2058c08af4c6SStefano Zampini ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 2059*fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2060*fc17d649SStefano Zampini Vec v2; 2061*fc17d649SStefano Zampini ierr = VecDuplicate(standard_rhs,&v2);CHKERRQ(ierr); 2062*fc17d649SStefano Zampini ierr = VecCopy(standard_rhs,v2);CHKERRQ(ierr); 2063*fc17d649SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,v2,NULL);CHKERRQ(ierr); 2064*fc17d649SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,v2,copy_standard_rhs);CHKERRQ(ierr); 2065*fc17d649SStefano Zampini ierr = VecDestroy(&v2);CHKERRQ(ierr); 2066*fc17d649SStefano Zampini } else { 2067c08af4c6SStefano Zampini ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 2068c08af4c6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 2069*fc17d649SStefano Zampini } 20703425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2071c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2072c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2073fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2074fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2075c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2076c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2077674ae819SStefano Zampini /* Apply partition of unity */ 20783425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2079c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 20808eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 20813425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 20823425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20833425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 20843425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2085c08af4c6SStefano Zampini ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 2086c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2087c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2088c08af4c6SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2089c08af4c6SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2090c08af4c6SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 20913425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 20923425bc38SStefano Zampini } 2093c08af4c6SStefano Zampini ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 20943425bc38SStefano Zampini /* BDDC rhs */ 20953425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 20968eeda7d8SStefano Zampini if (pcbddc->switch_static) { 20973425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 20983425bc38SStefano Zampini } 2099*fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21003425bc38SStefano Zampini /* apply BDDC */ 2101dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 21023425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 21033425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 21043425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 21053425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21063425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21073425bc38SStefano Zampini PetscFunctionReturn(0); 21083425bc38SStefano Zampini } 21091e6b0712SBarry Smith 21103425bc38SStefano Zampini #undef __FUNCT__ 21113425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 21123425bc38SStefano Zampini /*@ 21130f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 21143425bc38SStefano Zampini 21153425bc38SStefano Zampini Collective 21163425bc38SStefano Zampini 21173425bc38SStefano Zampini Input Parameters: 21180f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 21190f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 21203425bc38SStefano Zampini 21213425bc38SStefano Zampini Output Parameters: 21220f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 21233425bc38SStefano Zampini 21243425bc38SStefano Zampini Level: developer 21253425bc38SStefano Zampini 21263425bc38SStefano Zampini Notes: 21273425bc38SStefano Zampini 21280f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 21293425bc38SStefano Zampini @*/ 21303425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21313425bc38SStefano Zampini { 2132674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21333425bc38SStefano Zampini PetscErrorCode ierr; 21343425bc38SStefano Zampini 21353425bc38SStefano Zampini PetscFunctionBegin; 21363425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2137163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 21383425bc38SStefano Zampini PetscFunctionReturn(0); 21393425bc38SStefano Zampini } 21403425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 21411e6b0712SBarry Smith 21423425bc38SStefano Zampini #undef __FUNCT__ 21433425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 21443425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 21453425bc38SStefano Zampini { 2146674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 21473425bc38SStefano Zampini PC_IS* pcis; 21483425bc38SStefano Zampini PC_BDDC* pcbddc; 21493425bc38SStefano Zampini PetscErrorCode ierr; 21503425bc38SStefano Zampini 21513425bc38SStefano Zampini PetscFunctionBegin; 21523425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21533425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21543425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21553425bc38SStefano Zampini 21563425bc38SStefano Zampini /* apply B_delta^T */ 21573425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21583425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21593425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 21603425bc38SStefano Zampini /* compute rhs for BDDC application */ 21613425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21628eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21633425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21643425bc38SStefano Zampini } 2165*fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 21663425bc38SStefano Zampini /* apply BDDC */ 2167dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 21683425bc38SStefano Zampini /* put values into standard global vector */ 21693425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21703425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21718eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 21723425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 21733425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 21743425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 21753425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21763425bc38SStefano Zampini } 21773425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 21783425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2179*fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2180*fc17d649SStefano Zampini Vec v2; 2181*fc17d649SStefano Zampini ierr = VecDuplicate(standard_sol,&v2);CHKERRQ(ierr); 2182*fc17d649SStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,standard_sol,v2);CHKERRQ(ierr); 2183*fc17d649SStefano Zampini ierr = VecCopy(v2,standard_sol);CHKERRQ(ierr); 2184*fc17d649SStefano Zampini ierr = VecDestroy(&v2);CHKERRQ(ierr); 2185*fc17d649SStefano Zampini } 21863308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 21873425bc38SStefano Zampini PetscFunctionReturn(0); 21883425bc38SStefano Zampini } 21891e6b0712SBarry Smith 21903425bc38SStefano Zampini #undef __FUNCT__ 21913425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 21923425bc38SStefano Zampini /*@ 21930f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 21943425bc38SStefano Zampini 21953425bc38SStefano Zampini Collective 21963425bc38SStefano Zampini 21973425bc38SStefano Zampini Input Parameters: 21980f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 21990f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 22003425bc38SStefano Zampini 22013425bc38SStefano Zampini Output Parameters: 22020f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 22033425bc38SStefano Zampini 22043425bc38SStefano Zampini Level: developer 22053425bc38SStefano Zampini 22063425bc38SStefano Zampini Notes: 22073425bc38SStefano Zampini 22080f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 22093425bc38SStefano Zampini @*/ 22103425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22113425bc38SStefano Zampini { 2212674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22133425bc38SStefano Zampini PetscErrorCode ierr; 22143425bc38SStefano Zampini 22153425bc38SStefano Zampini PetscFunctionBegin; 22163425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2217163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 22183425bc38SStefano Zampini PetscFunctionReturn(0); 22193425bc38SStefano Zampini } 22203425bc38SStefano Zampini /* -------------------------------------------------------------------------- */ 22211e6b0712SBarry Smith 2222f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2223edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2224f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2225f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2226edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2227f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2228674ae819SStefano Zampini 22293425bc38SStefano Zampini #undef __FUNCT__ 22303425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 22313425bc38SStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 22323425bc38SStefano Zampini { 2233674ae819SStefano Zampini 2234674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 22353425bc38SStefano Zampini Mat newmat; 2236674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 22373425bc38SStefano Zampini PC newpc; 2238ce94432eSBarry Smith MPI_Comm comm; 22393425bc38SStefano Zampini PetscErrorCode ierr; 22403425bc38SStefano Zampini 22413425bc38SStefano Zampini PetscFunctionBegin; 2242ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 22433425bc38SStefano Zampini /* FETIDP linear matrix */ 22443425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 22453425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 22463425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 22473425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2248edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 22493425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 22503425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 22513425bc38SStefano Zampini /* FETIDP preconditioner */ 22523425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 22533425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 22543425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 22553425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 22563425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 22573425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2258edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 22593425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 226023ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 22613425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 22623425bc38SStefano Zampini /* return pointers for objects created */ 22633425bc38SStefano Zampini *fetidp_mat=newmat; 22643425bc38SStefano Zampini *fetidp_pc=newpc; 22653425bc38SStefano Zampini PetscFunctionReturn(0); 22663425bc38SStefano Zampini } 22671e6b0712SBarry Smith 22683425bc38SStefano Zampini #undef __FUNCT__ 22693425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 22703425bc38SStefano Zampini /*@ 22710f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 22723425bc38SStefano Zampini 22733425bc38SStefano Zampini Collective 22743425bc38SStefano Zampini 22753425bc38SStefano Zampini Input Parameters: 22760f202f7eSStefano Zampini . pc - the BDDC preconditioning context (setup should have been called before) 227728509bceSStefano Zampini 227828509bceSStefano Zampini Output Parameters: 22790f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 22800f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 228128509bceSStefano Zampini 228228509bceSStefano Zampini Options Database Keys: 22830f202f7eSStefano Zampini . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 22843425bc38SStefano Zampini 22853425bc38SStefano Zampini Level: developer 22863425bc38SStefano Zampini 22873425bc38SStefano Zampini Notes: 22880f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 22893425bc38SStefano Zampini 22900f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 22913425bc38SStefano Zampini @*/ 22923425bc38SStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 22933425bc38SStefano Zampini { 22943425bc38SStefano Zampini PetscErrorCode ierr; 22953425bc38SStefano Zampini 22963425bc38SStefano Zampini PetscFunctionBegin; 22973425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 22983425bc38SStefano Zampini if (pc->setupcalled) { 2299516d51a7SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2300f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 23013425bc38SStefano Zampini PetscFunctionReturn(0); 23023425bc38SStefano Zampini } 23030c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2304da1bb401SStefano Zampini /*MC 2305da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 23060c7d97c5SJed Brown 230728509bceSStefano Zampini An implementation of the BDDC preconditioner based on 230828509bceSStefano Zampini 230928509bceSStefano Zampini .vb 231028509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 231128509bceSStefano 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 231228509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 23130f202f7eSStefano 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 231428509bceSStefano Zampini .ve 231528509bceSStefano Zampini 231628509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 231728509bceSStefano Zampini 23180f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 231928509bceSStefano Zampini 232028509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 232128509bceSStefano Zampini 2322b6fdb6dfSStefano 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. 2323b6fdb6dfSStefano Zampini 2324c7017625SStefano 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). 232528509bceSStefano Zampini 23260f202f7eSStefano 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() 232730368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 232828509bceSStefano Zampini 23290f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 233028509bceSStefano Zampini 23310f202f7eSStefano 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. 23320f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 233328509bceSStefano Zampini 23340f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 233528509bceSStefano Zampini 2336df4d28bfSStefano 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. 233728509bceSStefano Zampini 23380f202f7eSStefano 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. 23390f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 23400f202f7eSStefano Zampini 23410f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 23420f202f7eSStefano Zampini 23430f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 23440f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 23450f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 23460f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 23470f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 23480f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 23490f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 235028509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 23510f202f7eSStefano 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) 23520f202f7eSStefano 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) 23530f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 23540f202f7eSStefano 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) 2355df4d28bfSStefano 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) 235628509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 235728509bceSStefano Zampini 235828509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 235928509bceSStefano Zampini .vb 236028509bceSStefano Zampini -pc_bddc_dirichlet_ 236128509bceSStefano Zampini -pc_bddc_neumann_ 236228509bceSStefano Zampini -pc_bddc_coarse_ 236328509bceSStefano Zampini .ve 23640f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 236528509bceSStefano Zampini 23660f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 236728509bceSStefano Zampini .vb 2368312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2369312be037SStefano Zampini -pc_bddc_neumann_lN_ 2370312be037SStefano Zampini -pc_bddc_coarse_lN_ 237128509bceSStefano Zampini .ve 23720f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 23730f202f7eSStefano 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. 23740f202f7eSStefano Zampini .vb 23750f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 23760f202f7eSStefano Zampini .ve 23770f202f7eSStefano 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 2378da1bb401SStefano Zampini 2379da1bb401SStefano Zampini Level: intermediate 2380da1bb401SStefano Zampini 2381b6fdb6dfSStefano Zampini Developer notes: 2382da1bb401SStefano Zampini 2383da1bb401SStefano Zampini Contributed by Stefano Zampini 2384da1bb401SStefano Zampini 2385da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2386da1bb401SStefano Zampini M*/ 2387b2573a8aSBarry Smith 2388da1bb401SStefano Zampini #undef __FUNCT__ 2389da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 23908cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2391da1bb401SStefano Zampini { 2392da1bb401SStefano Zampini PetscErrorCode ierr; 2393da1bb401SStefano Zampini PC_BDDC *pcbddc; 2394da1bb401SStefano Zampini 2395da1bb401SStefano Zampini PetscFunctionBegin; 2396da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2397b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2398da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2399da1bb401SStefano Zampini 2400da1bb401SStefano Zampini /* create PCIS data structure */ 2401da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2402da1bb401SStefano Zampini 240347d04d0dSStefano Zampini /* BDDC customization */ 240408a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 240547d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 240647d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 240747d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 240847d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 240947d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 241047d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2411fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2412fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 24133301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 241406a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 241527b6a85dSStefano Zampini pcbddc->benign_have_null = PETSC_FALSE; 241614f95afaSStefano Zampini pcbddc->vertex_size = 1; 241747d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2418b9d89cd5SStefano Zampini /* private */ 2419727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 24200e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 24210e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 24220e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2423e9189074SStefano Zampini pcbddc->n_vertices = 0; 2424727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2425fb180af4SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 242668457ee5SStefano Zampini pcbddc->coarse_size = -1; 2427f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2428727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2429f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2430f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2431f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2432674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 243330368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 24343972b0daSStefano Zampini pcbddc->temp_solution = 0; 2435534831adSStefano Zampini pcbddc->original_rhs = 0; 2436534831adSStefano Zampini pcbddc->local_mat = 0; 2437534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2438b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2439da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2440da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2441da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2442da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 244315aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 244415aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2445da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2446da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2447da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2448da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2449da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2450da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2451da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2452da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2453da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2454da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2455785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2456785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2457785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 245860d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 245960d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 246063602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2461da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 246263602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2463da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 246485c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 246547d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 246647d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2467b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 24684fad6a16SStefano Zampini pcbddc->current_level = 0; 24692b510759SStefano Zampini pcbddc->max_levels = 0; 2470323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 247157de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 2472f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 24734f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 24744f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 24754f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 247681d14e9dSStefano Zampini 247781d14e9dSStefano Zampini /* benign subspace trick */ 247881d14e9dSStefano Zampini pcbddc->benign_change = 0; 247927b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 24800369aaf7SStefano Zampini pcbddc->benign_vec = 0; 24810369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 24820369aaf7SStefano Zampini pcbddc->benign_sf = 0; 24834f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 24844f1b2e48SStefano Zampini pcbddc->benign_n = 0; 24854f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 24864f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 24874f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2488b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 248981d14e9dSStefano Zampini 2490674ae819SStefano Zampini /* create local graph structure */ 2491674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2492674ae819SStefano Zampini 2493674ae819SStefano Zampini /* scaling */ 2494674ae819SStefano Zampini pcbddc->work_scaling = 0; 249534a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2496b96c3477SStefano Zampini 2497b96c3477SStefano Zampini /* create sub schurs structure */ 2498b96c3477SStefano Zampini ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2499b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2500b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2501b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2502b96c3477SStefano Zampini 2503b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2504da1bb401SStefano Zampini 2505b7eb3628SStefano Zampini /* adaptivity */ 2506f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 250708122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2508f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2509b7eb3628SStefano Zampini 2510da1bb401SStefano Zampini /* function pointers */ 2511da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 251293bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2513da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2514da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2515da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 25166b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2517da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2518da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2519da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2520534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2521534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2522da1bb401SStefano Zampini 2523da1bb401SStefano Zampini /* composing function */ 2524906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2525674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 252630368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2527bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 25282b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2529b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 25302b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2531bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 253282ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2533bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 253482ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2535bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 253682ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2537bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 253882ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2539bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 254063602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2541bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2542bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2543bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2544bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2545a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2546da1bb401SStefano Zampini PetscFunctionReturn(0); 2547da1bb401SStefano Zampini } 25483425bc38SStefano Zampini 2549