153cdbc3dSStefano Zampini /* TODOLIST 2eb97c9d2SStefano Zampini 3eb97c9d2SStefano Zampini Solvers 4a0d3c3abSStefano Zampini - Add support for cholesky for coarse solver (similar to local solvers) 5eb97c9d2SStefano Zampini - Propagate ksp prefixes for solvers to mat objects? 6eb97c9d2SStefano Zampini 7eb97c9d2SStefano Zampini User interface 80f202f7eSStefano Zampini - ** DM attached to pc? 9eb97c9d2SStefano Zampini 10eb97c9d2SStefano Zampini Debugging output 11b9b85e73SStefano Zampini - * Better management of verbosity levels of debugging output 12eb97c9d2SStefano Zampini 13eb97c9d2SStefano Zampini Extra 14b9b85e73SStefano Zampini - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)? 15eb97c9d2SStefano Zampini - BDDC with MG framework? 16eb97c9d2SStefano Zampini 17eb97c9d2SStefano Zampini MATIS related operations contained in BDDC code 18eb97c9d2SStefano Zampini - Provide general case for subassembling 19eb97c9d2SStefano Zampini 2053cdbc3dSStefano Zampini */ 210c7d97c5SJed Brown 22ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 23ab5c6b0cSJed Brown #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 243b03a366Sstefano_zampini #include <petscblaslapack.h> 25674ae819SStefano Zampini 260369aaf7SStefano Zampini /* temporarily declare it */ 270369aaf7SStefano Zampini PetscErrorCode PCApply_BDDC(PC,Vec,Vec); 280369aaf7SStefano Zampini 290c7d97c5SJed Brown #undef __FUNCT__ 300c7d97c5SJed Brown #define __FUNCT__ "PCSetFromOptions_BDDC" 314416b707SBarry Smith PetscErrorCode PCSetFromOptions_BDDC(PetscOptionItems *PetscOptionsObject,PC pc) 320c7d97c5SJed Brown { 330c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 340c7d97c5SJed Brown PetscErrorCode ierr; 350c7d97c5SJed Brown 360c7d97c5SJed Brown PetscFunctionBegin; 37e55864a3SBarry Smith ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 388eeda7d8SStefano Zampini /* Verbose debugging */ 39a13144ffSStefano Zampini ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 40a13144ffSStefano Zampini /* Approximate solvers */ 41c7017625SStefano 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); 42c7017625SStefano 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); 43c7017625SStefano 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); 44c7017625SStefano 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); 456b78500eSPatrick Sanan /* Primal space customization */ 4608a5cf49SStefano 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); 47be12c134Sstefano_zampini ierr = PetscOptionsInt("-pc_bddc_graph_maxcount","Maximum number of shared subdomains for a connected component","none",pcbddc->graphmaxcount,&pcbddc->graphmaxcount,NULL);CHKERRQ(ierr); 488eeda7d8SStefano 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); 498eeda7d8SStefano 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); 508eeda7d8SStefano 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); 5114f95afaSStefano 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); 526661aa1dSStefano 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); 5314f95afaSStefano 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); 548eeda7d8SStefano Zampini /* Change of basis */ 55b9b85e73SStefano 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); 56b9b85e73SStefano 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); 57674ae819SStefano Zampini if (!pcbddc->use_change_of_basis) { 58674ae819SStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 59674ae819SStefano Zampini } 608eeda7d8SStefano Zampini /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 618eeda7d8SStefano 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); 6257de7509SStefano 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); 630298fd71SBarry Smith ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 642b510759SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 65323d395dSStefano 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); 66674ae819SStefano 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); 67b96c3477SStefano 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); 68b96c3477SStefano 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); 69b96c3477SStefano 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); 70683d3df6SStefano 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); 71bf3a8328SStefano 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); 72bf3a8328SStefano 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); 734c6709b3SStefano 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); 7408122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 7508122e43SStefano Zampini ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 763301b35fSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 77b0c7d250SStefano 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); 78b3afcdbeSStefano 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); 79e9627c49SStefano 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); 8027b6a85dSStefano 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); 81a198735bSStefano Zampini ierr = PetscOptionsBool("-pc_bddc_nonetflux","Automatic computation of no-net-flux quadrature weights","none",pcbddc->compute_nonetflux,&pcbddc->compute_nonetflux,NULL);CHKERRQ(ierr); 824f1b2e48SStefano Zampini ierr = PetscOptionsBool("-pc_bddc_detect_disconnected","Detects disconnected subdomains","none",pcbddc->detect_disconnected,&pcbddc->detect_disconnected,NULL);CHKERRQ(ierr); 8370c64980SStefano 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); 840c7d97c5SJed Brown ierr = PetscOptionsTail();CHKERRQ(ierr); 850c7d97c5SJed Brown PetscFunctionReturn(0); 860c7d97c5SJed Brown } 876b78500eSPatrick Sanan 886b78500eSPatrick Sanan #undef __FUNCT__ 896b78500eSPatrick Sanan #define __FUNCT__ "PCView_BDDC" 906b78500eSPatrick Sanan static PetscErrorCode PCView_BDDC(PC pc,PetscViewer viewer) 916b78500eSPatrick Sanan { 926b78500eSPatrick Sanan PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 93e9627c49SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 946b78500eSPatrick Sanan PetscErrorCode ierr; 956b78500eSPatrick Sanan PetscBool isascii,isstring; 96e9627c49SStefano Zampini PetscSubcomm subcomm; 97e9627c49SStefano Zampini PetscViewer subviewer; 986b78500eSPatrick Sanan 996b78500eSPatrick Sanan PetscFunctionBegin; 1006b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr); 1016b78500eSPatrick Sanan ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERSTRING,&isstring);CHKERRQ(ierr); 1026b78500eSPatrick Sanan /* Nothing printed for the String viewer */ 1036b78500eSPatrick Sanan /* ASCII viewer */ 1046b78500eSPatrick Sanan if (isascii) { 1054b2aedd3SStefano Zampini PetscMPIInt color,rank,size; 106695141f6SStefano Zampini PetscInt64 loc[6],gsum[5],gmax[5],gmin[5],totbenign; 107e9627c49SStefano Zampini PetscScalar interface_size; 108e9627c49SStefano Zampini PetscReal ratio1=0.,ratio2=0.; 109e9627c49SStefano Zampini Vec counter; 1106b78500eSPatrick Sanan 111e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use verbose output: %d\n",pcbddc->dbg_flag);CHKERRQ(ierr); 112e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use user-defined CSR: %d\n",!!pcbddc->mat_graph->nvtxs_csr);CHKERRQ(ierr); 1136b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use local mat graph: %d\n",pcbddc->use_local_adj);CHKERRQ(ierr); 114e9627c49SStefano Zampini if (pcbddc->mat_graph->twodim) { 115e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Connectivity graph topological dimension: 2\n");CHKERRQ(ierr); 116e9627c49SStefano Zampini } else { 117e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Connectivity graph topological dimension: 3\n");CHKERRQ(ierr); 118e9627c49SStefano Zampini } 119*a70d66d9SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Graph max count: %d\n",pcbddc->graphmaxcount);CHKERRQ(ierr); 120a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use vertices: %d (vertex size %d)\n",pcbddc->use_vertices,pcbddc->vertex_size);CHKERRQ(ierr); 1216b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use edges: %d\n",pcbddc->use_edges);CHKERRQ(ierr); 1226b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use faces: %d\n",pcbddc->use_faces);CHKERRQ(ierr); 1236b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use true near null space: %d\n",pcbddc->use_nnsp_true);CHKERRQ(ierr); 1246b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use QR for single constraints on cc: %d\n",pcbddc->use_qr_single);CHKERRQ(ierr); 1256b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use change of basis on local edge nodes: %d\n",pcbddc->use_change_of_basis);CHKERRQ(ierr); 1266b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use change of basis on local face nodes: %d\n",pcbddc->use_change_on_faces);CHKERRQ(ierr); 12727b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: User defined change of basis matrix: %d\n",!!pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 12827b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Has change of basis matrix: %d\n",!!pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 12927b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Eliminate dirichlet boundary dofs: %d\n",pcbddc->eliminate_dirdofs);CHKERRQ(ierr); 1306b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Switch on static condensation ops around the interface preconditioner: %d\n",pcbddc->switch_static);CHKERRQ(ierr); 13127b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use exact dirichlet trick: %d\n",pcbddc->use_exact_dirichlet_trick);CHKERRQ(ierr); 1326b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Multilevel max levels: %d\n",pcbddc->max_levels);CHKERRQ(ierr); 133e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Multilevel coarsening ratio: %d\n",pcbddc->coarsening_ratio);CHKERRQ(ierr); 1346b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use estimated eigs for coarse problem: %d\n",pcbddc->use_coarse_estimates);CHKERRQ(ierr); 1356b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe scaling: %d\n",pcbddc->use_deluxe_scaling);CHKERRQ(ierr); 136e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use deluxe zerorows: %d\n",pcbddc->deluxe_zerorows);CHKERRQ(ierr); 1376b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Rebuild interface graph for Schur principal minors: %d\n",pcbddc->sub_schurs_rebuild);CHKERRQ(ierr); 1386b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Number of dofs' layers for the computation of principal minors: %d\n",pcbddc->sub_schurs_layers);CHKERRQ(ierr); 1396b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Use user CSR graph to compute successive layers: %d\n",pcbddc->sub_schurs_use_useradj);CHKERRQ(ierr); 140e9627c49SStefano 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); 1416b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Min constraints / connected component: %d\n",pcbddc->adaptive_nmin);CHKERRQ(ierr); 1426b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Max constraints / connected component: %d\n",pcbddc->adaptive_nmax);CHKERRQ(ierr); 143e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Invert exact Schur complement for adaptive selection: %d\n",pcbddc->sub_schurs_exact_schur);CHKERRQ(ierr); 1446b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Symmetric computation of primal basis functions: %d\n",pcbddc->symmetric_primal);CHKERRQ(ierr); 1456b78500eSPatrick Sanan ierr = PetscViewerASCIIPrintf(viewer," BDDC: Num. Procs. to map coarse adjacency list: %d\n",pcbddc->coarse_adj_red);CHKERRQ(ierr); 146e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Coarse eqs per proc (significant at the coarsest level): %d\n",pcbddc->coarse_eqs_per_proc);CHKERRQ(ierr); 147e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Detect disconnected: %d\n",pcbddc->detect_disconnected);CHKERRQ(ierr); 148fc17d649SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subspace trick: %d (change explicit %d)\n",pcbddc->benign_saddle_point,pcbddc->benign_change_explicit);CHKERRQ(ierr); 149a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subspace trick is active: %d\n",pcbddc->benign_have_null);CHKERRQ(ierr); 150a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Algebraic computation of no-net-flux %d\n",pcbddc->compute_nonetflux);CHKERRQ(ierr); 1516b78500eSPatrick Sanan 152345ecf6cSStefano Zampini /* compute some numbers on the domain decomposition */ 153e9627c49SStefano Zampini ierr = VecSet(pcis->vec1_B,1.0);CHKERRQ(ierr); 154e9627c49SStefano Zampini ierr = MatCreateVecs(pc->pmat,&counter,0);CHKERRQ(ierr); 155e9627c49SStefano Zampini ierr = VecSet(counter,0.0);CHKERRQ(ierr); 156e9627c49SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 157e9627c49SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,counter,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 158e9627c49SStefano Zampini ierr = VecSum(counter,&interface_size);CHKERRQ(ierr); 159e9627c49SStefano Zampini ierr = VecDestroy(&counter);CHKERRQ(ierr); 160e9627c49SStefano Zampini gsum[0] = 1; 161345ecf6cSStefano Zampini gsum[1] = gsum[2] = gsum[3] = gsum[4] = 0; 162e9627c49SStefano Zampini loc[0] = !!pcis->n; 163e9627c49SStefano Zampini loc[1] = pcis->n - pcis->n_B; 164e9627c49SStefano Zampini loc[2] = pcis->n_B; 165e9627c49SStefano Zampini loc[3] = pcbddc->local_primal_size; 166345ecf6cSStefano Zampini loc[4] = pcis->n; 167345ecf6cSStefano Zampini loc[5] = pcbddc->benign_n; 168345ecf6cSStefano Zampini ierr = MPI_Reduce(loc,gsum,5,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 169345ecf6cSStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = -1; 170345ecf6cSStefano Zampini ierr = MPI_Reduce(loc,gmax,5,MPIU_INT64,MPI_MAX,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 171345ecf6cSStefano Zampini if (!loc[0]) loc[1] = loc[2] = loc[3] = loc[4] = PETSC_MAX_INT; 172345ecf6cSStefano Zampini ierr = MPI_Reduce(loc,gmin,5,MPIU_INT64,MPI_MIN,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 173345ecf6cSStefano Zampini ierr = MPI_Reduce(&loc[5],&totbenign,1,MPIU_INT64,MPI_SUM,0,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 174e9627c49SStefano Zampini if (pcbddc->coarse_size) { 175e9627c49SStefano Zampini ratio1 = pc->pmat->rmap->N/(1.*pcbddc->coarse_size); 176e9627c49SStefano Zampini ratio2 = PetscRealPart(interface_size)/pcbddc->coarse_size; 177e9627c49SStefano Zampini } 178e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: ********************************** STATISTICS AT LEVEL %d **********************************\n",pcbddc->current_level);CHKERRQ(ierr); 179e9627c49SStefano 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); 180e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Coarsening ratios: all/coarse %d interface/coarse %d\n",(PetscInt)ratio1,(PetscInt)ratio2);CHKERRQ(ierr); 181e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Active processes : %d\n",(PetscInt)gsum[0]);CHKERRQ(ierr); 182345ecf6cSStefano Zampini if (pcbddc->benign_have_null) { 183345ecf6cSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Benign subs : %d\n",(PetscInt)totbenign);CHKERRQ(ierr); 184345ecf6cSStefano Zampini } 185e9627c49SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: Dofs type :\tMIN\tMAX\tMEAN\n",(PetscInt)gmin[1],(PetscInt)gmax[1],(PetscInt)(gsum[1]/gsum[0]));CHKERRQ(ierr); 186e9627c49SStefano 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); 187e9627c49SStefano 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); 188e9627c49SStefano 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); 189345ecf6cSStefano 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); 19027b6a85dSStefano Zampini ierr = PetscViewerASCIIPrintf(viewer," BDDC: ********************************** COARSE PROBLEM DETAILS *********************************\n",pcbddc->current_level);CHKERRQ(ierr); 19127b6a85dSStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 192e9627c49SStefano Zampini 193e9627c49SStefano Zampini /* the coarse problem can be handled with a different communicator */ 194e9627c49SStefano Zampini if (pcbddc->coarse_ksp) color = 1; 195e9627c49SStefano Zampini else color = 0; 196e9627c49SStefano Zampini ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)pc),&rank);CHKERRQ(ierr); 1974b2aedd3SStefano Zampini ierr = MPI_Comm_size(PetscObjectComm((PetscObject)pc),&size);CHKERRQ(ierr); 198e9627c49SStefano Zampini ierr = PetscSubcommCreate(PetscObjectComm((PetscObject)pc),&subcomm);CHKERRQ(ierr); 1994b2aedd3SStefano Zampini ierr = PetscSubcommSetNumber(subcomm,PetscMin(size,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 } 212a13144ffSStefano Zampini 213a13144ffSStefano Zampini #undef __FUNCT__ 214a13144ffSStefano Zampini #define __FUNCT__ "PCBDDCSetDiscreteGradient_BDDC" 215495a2a07SStefano Zampini static PetscErrorCode PCBDDCSetDiscreteGradient_BDDC(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool conforming) 216a13144ffSStefano Zampini { 217a13144ffSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 218a13144ffSStefano Zampini PetscErrorCode ierr; 219a13144ffSStefano Zampini 220a13144ffSStefano Zampini PetscFunctionBegin; 221a13144ffSStefano Zampini ierr = PetscObjectReference((PetscObject)G);CHKERRQ(ierr); 222a13144ffSStefano Zampini ierr = MatDestroy(&pcbddc->discretegradient);CHKERRQ(ierr); 223a13144ffSStefano Zampini pcbddc->discretegradient = G; 224a13144ffSStefano Zampini pcbddc->nedorder = order > 0 ? order : -order; 225a13144ffSStefano Zampini pcbddc->conforming = conforming; 226495a2a07SStefano Zampini pcbddc->nedfield = field; 227a13144ffSStefano Zampini PetscFunctionReturn(0); 228a13144ffSStefano Zampini } 229a13144ffSStefano Zampini 230a13144ffSStefano Zampini #undef __FUNCT__ 231a13144ffSStefano Zampini #define __FUNCT__ "PCBDDCSetDiscreteGradient" 232a13144ffSStefano Zampini /*@ 233a13144ffSStefano Zampini PCBDDCSetDiscreteGradient - Sets the discrete gradient 234a13144ffSStefano Zampini 235a13144ffSStefano Zampini Collective on PC 236a13144ffSStefano Zampini 237a13144ffSStefano Zampini Input Parameters: 238a13144ffSStefano Zampini + pc - the preconditioning context 239a13144ffSStefano Zampini . G - the discrete gradient matrix (should be in AIJ format) 240a13144ffSStefano Zampini . order - the order of the Nedelec space (1 for the lowest order) 241495a2a07SStefano Zampini . field - the field id of the Nedelec dofs (not used if the fields have not been specified) 242a13144ffSStefano Zampini - conforming - whether the mesh is conforming or not 243a13144ffSStefano Zampini 244a13144ffSStefano Zampini Level: advanced 245a13144ffSStefano Zampini 246a13144ffSStefano Zampini Notes: The discrete gradient matrix G is used to analyze the subdomain edges, and it should not contain any zero entry. 247495a2a07SStefano Zampini For variable order spaces, the order should be set to zero. 248a13144ffSStefano Zampini The discrete gradient matrix is modified by PCBDDC. 249a13144ffSStefano Zampini 250495a2a07SStefano Zampini .seealso: PCBDDC,PCBDDCSetDofsSplitting(),PCBDDCSetDofsSplittingLocal() 251a13144ffSStefano Zampini @*/ 252495a2a07SStefano Zampini PetscErrorCode PCBDDCSetDiscreteGradient(PC pc, Mat G, PetscInt order, PetscInt field, PetscBool conforming) 253a13144ffSStefano Zampini { 254a13144ffSStefano Zampini PetscErrorCode ierr; 255a13144ffSStefano Zampini 256a13144ffSStefano Zampini PetscFunctionBegin; 257a13144ffSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 258a13144ffSStefano Zampini PetscValidHeaderSpecific(G,MAT_CLASSID,2); 259a13144ffSStefano Zampini PetscCheckSameComm(pc,1,G,2); 260a13144ffSStefano Zampini PetscValidLogicalCollectiveInt(pc,order,3); 261a13144ffSStefano Zampini PetscValidLogicalCollectiveBool(pc,conforming,4); 262495a2a07SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDiscreteGradient_C",(PC,Mat,PetscInt,PetscInt,PetscBool),(pc,G,order,field,conforming));CHKERRQ(ierr); 263a13144ffSStefano Zampini PetscFunctionReturn(0); 264a13144ffSStefano Zampini } 265a13144ffSStefano Zampini 266a198735bSStefano Zampini #undef __FUNCT__ 267a198735bSStefano Zampini #define __FUNCT__ "PCBDDCSetDivergenceMat_BDDC" 2688ae0ca82SStefano Zampini static PetscErrorCode PCBDDCSetDivergenceMat_BDDC(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 269a198735bSStefano Zampini { 270a198735bSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 271a198735bSStefano Zampini PetscErrorCode ierr; 2726b78500eSPatrick Sanan 273a198735bSStefano Zampini PetscFunctionBegin; 274a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)divudotp);CHKERRQ(ierr); 275a198735bSStefano Zampini ierr = MatDestroy(&pcbddc->divudotp);CHKERRQ(ierr); 276a198735bSStefano Zampini pcbddc->divudotp = divudotp; 2778ae0ca82SStefano Zampini pcbddc->divudotp_trans = trans; 278a198735bSStefano Zampini pcbddc->compute_nonetflux = PETSC_TRUE; 279a198735bSStefano Zampini if (vl2l) { 280a198735bSStefano Zampini ierr = PetscObjectReference((PetscObject)vl2l);CHKERRQ(ierr); 281fa23a32eSStefano Zampini ierr = ISDestroy(&pcbddc->divudotp_vl2l);CHKERRQ(ierr); 282a198735bSStefano Zampini pcbddc->divudotp_vl2l = vl2l; 283a198735bSStefano Zampini } 284a198735bSStefano Zampini PetscFunctionReturn(0); 285a198735bSStefano Zampini } 2863d996552SStefano Zampini 287a198735bSStefano Zampini #undef __FUNCT__ 288a198735bSStefano Zampini #define __FUNCT__ "PCBDDCSetDivergenceMat" 289a198735bSStefano Zampini /*@ 290a198735bSStefano Zampini PCBDDCSetDivergenceMat - Sets the linear operator representing \int_\Omega \div {\bf u} \cdot p dx 291a198735bSStefano Zampini 292a198735bSStefano Zampini Collective on PC 293a198735bSStefano Zampini 294a198735bSStefano Zampini Input Parameters: 295a198735bSStefano Zampini + pc - the preconditioning context 296a198735bSStefano Zampini . divudotp - the matrix (must be of type MATIS) 2978ae0ca82SStefano Zampini . trans - if trans if false (resp. true), then pressures are in the test (trial) space and velocities are in the trial (test) space. 298bb05f991SStefano Zampini - vl2l - optional IS describing the local (wrt the local mat in divudotp) to local (wrt the local mat in pc->pmat) map for the velocities 299a198735bSStefano Zampini 300a198735bSStefano Zampini Level: advanced 301a198735bSStefano Zampini 3028ae0ca82SStefano Zampini Notes: This auxiliary matrix is used to compute quadrature weights representing the net-flux across subdomain boundaries 303a198735bSStefano Zampini 304a198735bSStefano Zampini .seealso: PCBDDC 305a198735bSStefano Zampini @*/ 3068ae0ca82SStefano Zampini PetscErrorCode PCBDDCSetDivergenceMat(PC pc, Mat divudotp, PetscBool trans, IS vl2l) 307a198735bSStefano Zampini { 308a198735bSStefano Zampini PetscBool ismatis; 309a198735bSStefano Zampini PetscErrorCode ierr; 310a198735bSStefano Zampini 311a198735bSStefano Zampini PetscFunctionBegin; 312a198735bSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 313a198735bSStefano Zampini PetscValidHeaderSpecific(divudotp,MAT_CLASSID,2); 314a198735bSStefano Zampini PetscCheckSameComm(pc,1,divudotp,2); 3158ae0ca82SStefano Zampini PetscValidLogicalCollectiveBool(pc,trans,3); 3168ae0ca82SStefano Zampini if (vl2l) PetscValidHeaderSpecific(divudotp,IS_CLASSID,4); 317a198735bSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)divudotp,MATIS,&ismatis);CHKERRQ(ierr); 318a198735bSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"Divergence matrix needs to be of type MATIS"); 3198ae0ca82SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDivergenceMat_C",(PC,Mat,PetscBool,IS),(pc,divudotp,trans,vl2l));CHKERRQ(ierr); 320a198735bSStefano Zampini PetscFunctionReturn(0); 321a198735bSStefano Zampini } 3222d505d7fSStefano Zampini 323674ae819SStefano Zampini #undef __FUNCT__ 324906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 3251dd7afcfSStefano Zampini static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change, PetscBool interior) 326b9b85e73SStefano Zampini { 327b9b85e73SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 328b9b85e73SStefano Zampini PetscErrorCode ierr; 329b9b85e73SStefano Zampini 330b9b85e73SStefano Zampini PetscFunctionBegin; 331b9b85e73SStefano Zampini ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 33256282151SStefano Zampini ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 333b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = change; 3341dd7afcfSStefano Zampini pcbddc->change_interior = interior; 335b9b85e73SStefano Zampini PetscFunctionReturn(0); 336b9b85e73SStefano Zampini } 337b9b85e73SStefano Zampini #undef __FUNCT__ 338906d46d4SStefano Zampini #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 339b9b85e73SStefano Zampini /*@ 340906d46d4SStefano Zampini PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 341b9b85e73SStefano Zampini 342b9b85e73SStefano Zampini Collective on PC 343b9b85e73SStefano Zampini 344b9b85e73SStefano Zampini Input Parameters: 345b9b85e73SStefano Zampini + pc - the preconditioning context 3461dd7afcfSStefano Zampini . change - the change of basis matrix 3471dd7afcfSStefano Zampini - interior - whether or not the change of basis modifies interior dofs 348b9b85e73SStefano Zampini 349b9b85e73SStefano Zampini Level: intermediate 350b9b85e73SStefano Zampini 351b9b85e73SStefano Zampini Notes: 352b9b85e73SStefano Zampini 353b9b85e73SStefano Zampini .seealso: PCBDDC 354b9b85e73SStefano Zampini @*/ 3551dd7afcfSStefano Zampini PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change, PetscBool interior) 356b9b85e73SStefano Zampini { 357b9b85e73SStefano Zampini PetscErrorCode ierr; 358b9b85e73SStefano Zampini 359b9b85e73SStefano Zampini PetscFunctionBegin; 360b9b85e73SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 361b9b85e73SStefano Zampini PetscValidHeaderSpecific(change,MAT_CLASSID,2); 362906d46d4SStefano Zampini PetscCheckSameComm(pc,1,change,2); 363906d46d4SStefano Zampini if (pc->mat) { 364906d46d4SStefano Zampini PetscInt rows_c,cols_c,rows,cols; 365906d46d4SStefano Zampini ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 366906d46d4SStefano Zampini ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 3676c4ed002SBarry 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); 3686c4ed002SBarry 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); 369906d46d4SStefano Zampini ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 370906d46d4SStefano Zampini ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 3716c4ed002SBarry 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); 3726c4ed002SBarry 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); 373906d46d4SStefano Zampini } 3741dd7afcfSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat,PetscBool),(pc,change,interior));CHKERRQ(ierr); 375b9b85e73SStefano Zampini PetscFunctionReturn(0); 376b9b85e73SStefano Zampini } 3772d505d7fSStefano Zampini 378b9b85e73SStefano Zampini #undef __FUNCT__ 37930368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS_BDDC" 38030368db7SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesIS_BDDC(PC pc, IS PrimalVertices) 38130368db7SStefano Zampini { 38230368db7SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 38356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 38430368db7SStefano Zampini PetscErrorCode ierr; 38530368db7SStefano Zampini 38630368db7SStefano Zampini PetscFunctionBegin; 38756282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 38856282151SStefano Zampini if (pcbddc->user_primal_vertices) { 38956282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices,&isequal);CHKERRQ(ierr); 39056282151SStefano Zampini } 39130368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 39230368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 39330368db7SStefano Zampini pcbddc->user_primal_vertices = PrimalVertices; 39456282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 39530368db7SStefano Zampini PetscFunctionReturn(0); 39630368db7SStefano Zampini } 39730368db7SStefano Zampini #undef __FUNCT__ 39830368db7SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesIS" 39930368db7SStefano Zampini /*@ 40030368db7SStefano Zampini PCBDDCSetPrimalVerticesIS - Set additional user defined primal vertices in PCBDDC 40130368db7SStefano Zampini 40230368db7SStefano Zampini Collective 40330368db7SStefano Zampini 40430368db7SStefano Zampini Input Parameters: 40530368db7SStefano Zampini + pc - the preconditioning context 40630368db7SStefano Zampini - PrimalVertices - index set of primal vertices in global numbering (can be empty) 40730368db7SStefano Zampini 40830368db7SStefano Zampini Level: intermediate 40930368db7SStefano Zampini 41030368db7SStefano Zampini Notes: 41130368db7SStefano Zampini Any process can list any global node 41230368db7SStefano Zampini 41330368db7SStefano Zampini .seealso: PCBDDC 41430368db7SStefano Zampini @*/ 41530368db7SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesIS(PC pc, IS PrimalVertices) 41630368db7SStefano Zampini { 41730368db7SStefano Zampini PetscErrorCode ierr; 41830368db7SStefano Zampini 41930368db7SStefano Zampini PetscFunctionBegin; 42030368db7SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 42130368db7SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 42230368db7SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 42330368db7SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 42430368db7SStefano Zampini PetscFunctionReturn(0); 42530368db7SStefano Zampini } 4262d505d7fSStefano Zampini 42730368db7SStefano Zampini #undef __FUNCT__ 428674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 429674ae819SStefano Zampini static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 430674ae819SStefano Zampini { 431674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 43256282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 433674ae819SStefano Zampini PetscErrorCode ierr; 4341e6b0712SBarry Smith 435674ae819SStefano Zampini PetscFunctionBegin; 43656282151SStefano Zampini ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 43756282151SStefano Zampini if (pcbddc->user_primal_vertices_local) { 43856282151SStefano Zampini ierr = ISEqual(PrimalVertices,pcbddc->user_primal_vertices_local,&isequal);CHKERRQ(ierr); 43956282151SStefano Zampini } 440674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 44130368db7SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices_local);CHKERRQ(ierr); 44230368db7SStefano Zampini pcbddc->user_primal_vertices_local = PrimalVertices; 44356282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 444674ae819SStefano Zampini PetscFunctionReturn(0); 445674ae819SStefano Zampini } 446674ae819SStefano Zampini #undef __FUNCT__ 447674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 448674ae819SStefano Zampini /*@ 44928509bceSStefano Zampini PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 450674ae819SStefano Zampini 45117eb1463SStefano Zampini Collective 452674ae819SStefano Zampini 453674ae819SStefano Zampini Input Parameters: 454674ae819SStefano Zampini + pc - the preconditioning context 45517eb1463SStefano Zampini - PrimalVertices - index set of primal vertices in local numbering (can be empty) 456674ae819SStefano Zampini 457674ae819SStefano Zampini Level: intermediate 458674ae819SStefano Zampini 459674ae819SStefano Zampini Notes: 460674ae819SStefano Zampini 461674ae819SStefano Zampini .seealso: PCBDDC 462674ae819SStefano Zampini @*/ 463674ae819SStefano Zampini PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 464674ae819SStefano Zampini { 465674ae819SStefano Zampini PetscErrorCode ierr; 466674ae819SStefano Zampini 467674ae819SStefano Zampini PetscFunctionBegin; 468674ae819SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 469674ae819SStefano Zampini PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 47017eb1463SStefano Zampini PetscCheckSameComm(pc,1,PrimalVertices,2); 471674ae819SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 472674ae819SStefano Zampini PetscFunctionReturn(0); 473674ae819SStefano Zampini } 4742d505d7fSStefano Zampini 4750c7d97c5SJed Brown #undef __FUNCT__ 4764fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 4774fad6a16SStefano Zampini static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 4784fad6a16SStefano Zampini { 4794fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 4804fad6a16SStefano Zampini 4814fad6a16SStefano Zampini PetscFunctionBegin; 4824fad6a16SStefano Zampini pcbddc->coarsening_ratio = k; 4834fad6a16SStefano Zampini PetscFunctionReturn(0); 4844fad6a16SStefano Zampini } 4851e6b0712SBarry Smith 4864fad6a16SStefano Zampini #undef __FUNCT__ 4874fad6a16SStefano Zampini #define __FUNCT__ "PCBDDCSetCoarseningRatio" 4884fad6a16SStefano Zampini /*@ 48928509bceSStefano Zampini PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 4904fad6a16SStefano Zampini 4914fad6a16SStefano Zampini Logically collective on PC 4924fad6a16SStefano Zampini 4934fad6a16SStefano Zampini Input Parameters: 4944fad6a16SStefano Zampini + pc - the preconditioning context 49528509bceSStefano Zampini - k - coarsening ratio (H/h at the coarser level) 4964fad6a16SStefano Zampini 4970f202f7eSStefano Zampini Options Database Keys: 4980f202f7eSStefano Zampini . -pc_bddc_coarsening_ratio 4994fad6a16SStefano Zampini 5004fad6a16SStefano Zampini Level: intermediate 5014fad6a16SStefano Zampini 5024fad6a16SStefano Zampini Notes: 5030f202f7eSStefano Zampini Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 5044fad6a16SStefano Zampini 5050f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetLevels() 5064fad6a16SStefano Zampini @*/ 5074fad6a16SStefano Zampini PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 5084fad6a16SStefano Zampini { 5094fad6a16SStefano Zampini PetscErrorCode ierr; 5104fad6a16SStefano Zampini 5114fad6a16SStefano Zampini PetscFunctionBegin; 5124fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 5132b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,k,2); 5144fad6a16SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 5154fad6a16SStefano Zampini PetscFunctionReturn(0); 5164fad6a16SStefano Zampini } 5172b510759SStefano Zampini 518b8ffe317SStefano Zampini /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 5192b510759SStefano Zampini #undef __FUNCT__ 520b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 521b8ffe317SStefano Zampini static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 522b8ffe317SStefano Zampini { 523b8ffe317SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 524b8ffe317SStefano Zampini 525b8ffe317SStefano Zampini PetscFunctionBegin; 52685c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = flg; 527b8ffe317SStefano Zampini PetscFunctionReturn(0); 528b8ffe317SStefano Zampini } 529b8ffe317SStefano Zampini 530b8ffe317SStefano Zampini #undef __FUNCT__ 531b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 532b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 5332b510759SStefano Zampini { 5342b510759SStefano Zampini PetscErrorCode ierr; 5352b510759SStefano Zampini 5362b510759SStefano Zampini PetscFunctionBegin; 5372b510759SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 538b8ffe317SStefano Zampini PetscValidLogicalCollectiveBool(pc,flg,2); 539b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 5402b510759SStefano Zampini PetscFunctionReturn(0); 5412b510759SStefano Zampini } 5421e6b0712SBarry Smith 5434fad6a16SStefano Zampini #undef __FUNCT__ 5442b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel_BDDC" 5452b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 5464fad6a16SStefano Zampini { 5474fad6a16SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5484fad6a16SStefano Zampini 5494fad6a16SStefano Zampini PetscFunctionBegin; 5502b510759SStefano Zampini pcbddc->current_level = level; 5514fad6a16SStefano Zampini PetscFunctionReturn(0); 5524fad6a16SStefano Zampini } 5531e6b0712SBarry Smith 5544fad6a16SStefano Zampini #undef __FUNCT__ 555b8ffe317SStefano Zampini #define __FUNCT__ "PCBDDCSetLevel" 556b8ffe317SStefano Zampini PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 557b8ffe317SStefano Zampini { 558b8ffe317SStefano Zampini PetscErrorCode ierr; 559b8ffe317SStefano Zampini 560b8ffe317SStefano Zampini PetscFunctionBegin; 561b8ffe317SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 562b8ffe317SStefano Zampini PetscValidLogicalCollectiveInt(pc,level,2); 563b8ffe317SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 564b8ffe317SStefano Zampini PetscFunctionReturn(0); 565b8ffe317SStefano Zampini } 566b8ffe317SStefano Zampini 567b8ffe317SStefano Zampini #undef __FUNCT__ 5682b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels_BDDC" 5692b510759SStefano Zampini static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 5702b510759SStefano Zampini { 5712b510759SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 5722b510759SStefano Zampini 5732b510759SStefano Zampini PetscFunctionBegin; 5742b510759SStefano Zampini pcbddc->max_levels = levels; 5752b510759SStefano Zampini PetscFunctionReturn(0); 5762b510759SStefano Zampini } 5772b510759SStefano Zampini 5782b510759SStefano Zampini #undef __FUNCT__ 5792b510759SStefano Zampini #define __FUNCT__ "PCBDDCSetLevels" 5804fad6a16SStefano Zampini /*@ 58128509bceSStefano Zampini PCBDDCSetLevels - Sets the maximum number of levels for multilevel 5824fad6a16SStefano Zampini 5834fad6a16SStefano Zampini Logically collective on PC 5844fad6a16SStefano Zampini 5854fad6a16SStefano Zampini Input Parameters: 5864fad6a16SStefano Zampini + pc - the preconditioning context 58728509bceSStefano Zampini - levels - the maximum number of levels (max 9) 5884fad6a16SStefano Zampini 5890f202f7eSStefano Zampini Options Database Keys: 5900f202f7eSStefano Zampini . -pc_bddc_levels 5914fad6a16SStefano Zampini 5924fad6a16SStefano Zampini Level: intermediate 5934fad6a16SStefano Zampini 5944fad6a16SStefano Zampini Notes: 5950f202f7eSStefano Zampini Default value is 0, i.e. traditional one-level BDDC 5964fad6a16SStefano Zampini 5970f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 5984fad6a16SStefano Zampini @*/ 5992b510759SStefano Zampini PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 6004fad6a16SStefano Zampini { 6014fad6a16SStefano Zampini PetscErrorCode ierr; 6024fad6a16SStefano Zampini 6034fad6a16SStefano Zampini PetscFunctionBegin; 6044fad6a16SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 6052b510759SStefano Zampini PetscValidLogicalCollectiveInt(pc,levels,2); 606312be037SStefano Zampini if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 6072b510759SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 6084fad6a16SStefano Zampini PetscFunctionReturn(0); 6094fad6a16SStefano Zampini } 6101e6b0712SBarry Smith 6114fad6a16SStefano Zampini #undef __FUNCT__ 6123b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 6133b03a366Sstefano_zampini static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 6143b03a366Sstefano_zampini { 6153b03a366Sstefano_zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 61656282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6173b03a366Sstefano_zampini PetscErrorCode ierr; 6183b03a366Sstefano_zampini 6193b03a366Sstefano_zampini PetscFunctionBegin; 62056282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 62156282151SStefano Zampini if (pcbddc->DirichletBoundaries) { 62256282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundaries,&isequal);CHKERRQ(ierr); 62356282151SStefano Zampini } 624785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 625785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6263b03a366Sstefano_zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 62736e030ebSStefano Zampini pcbddc->DirichletBoundaries = DirichletBoundaries; 62856282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6293b03a366Sstefano_zampini PetscFunctionReturn(0); 6303b03a366Sstefano_zampini } 6311e6b0712SBarry Smith 6323b03a366Sstefano_zampini #undef __FUNCT__ 6333b03a366Sstefano_zampini #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 6343b03a366Sstefano_zampini /*@ 63528509bceSStefano Zampini PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 6363b03a366Sstefano_zampini 637785d1243SStefano Zampini Collective 6383b03a366Sstefano_zampini 6393b03a366Sstefano_zampini Input Parameters: 6403b03a366Sstefano_zampini + pc - the preconditioning context 641785d1243SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 6423b03a366Sstefano_zampini 6433b03a366Sstefano_zampini Level: intermediate 6443b03a366Sstefano_zampini 6450f202f7eSStefano Zampini Notes: 6460f202f7eSStefano Zampini Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 6473b03a366Sstefano_zampini 6480f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 6493b03a366Sstefano_zampini @*/ 6503b03a366Sstefano_zampini PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 6513b03a366Sstefano_zampini { 6523b03a366Sstefano_zampini PetscErrorCode ierr; 6533b03a366Sstefano_zampini 6543b03a366Sstefano_zampini PetscFunctionBegin; 6553b03a366Sstefano_zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 656674ae819SStefano Zampini PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 657785d1243SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 6583b03a366Sstefano_zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 6593b03a366Sstefano_zampini PetscFunctionReturn(0); 6603b03a366Sstefano_zampini } 6611e6b0712SBarry Smith 6623b03a366Sstefano_zampini #undef __FUNCT__ 66382ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 66482ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 6650c7d97c5SJed Brown { 6660c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 66756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 6680c7d97c5SJed Brown PetscErrorCode ierr; 6690c7d97c5SJed Brown 6700c7d97c5SJed Brown PetscFunctionBegin; 67156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 67256282151SStefano Zampini if (pcbddc->DirichletBoundariesLocal) { 67356282151SStefano Zampini ierr = ISEqual(DirichletBoundaries,pcbddc->DirichletBoundariesLocal,&isequal);CHKERRQ(ierr); 67456282151SStefano Zampini } 675785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 676785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 6770c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 678785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 67956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 6800c7d97c5SJed Brown PetscFunctionReturn(0); 6810c7d97c5SJed Brown } 6820c7d97c5SJed Brown 6830c7d97c5SJed Brown #undef __FUNCT__ 68482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 6859c0446d6SStefano Zampini /*@ 68682ba6b80SStefano Zampini PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 6879c0446d6SStefano Zampini 688785d1243SStefano Zampini Collective 68953cdbc3dSStefano Zampini 69053cdbc3dSStefano Zampini Input Parameters: 69153cdbc3dSStefano Zampini + pc - the preconditioning context 69282ba6b80SStefano Zampini - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 69353cdbc3dSStefano Zampini 69453cdbc3dSStefano Zampini Level: intermediate 69553cdbc3dSStefano Zampini 6969c0446d6SStefano Zampini Notes: 69753cdbc3dSStefano Zampini 6980f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 69953cdbc3dSStefano Zampini @*/ 70082ba6b80SStefano Zampini PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 7010c7d97c5SJed Brown { 7020c7d97c5SJed Brown PetscErrorCode ierr; 7030c7d97c5SJed Brown 7040c7d97c5SJed Brown PetscFunctionBegin; 7050c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 7060c7d97c5SJed Brown PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 70782ba6b80SStefano Zampini PetscCheckSameComm(pc,1,DirichletBoundaries,2); 70882ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 7090c7d97c5SJed Brown PetscFunctionReturn(0); 7100c7d97c5SJed Brown } 7110c7d97c5SJed Brown 7120c7d97c5SJed Brown #undef __FUNCT__ 7130c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 71453cdbc3dSStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 7150c7d97c5SJed Brown { 7160c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 71756282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 71853cdbc3dSStefano Zampini PetscErrorCode ierr; 7190c7d97c5SJed Brown 7200c7d97c5SJed Brown PetscFunctionBegin; 72156282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 72256282151SStefano Zampini if (pcbddc->NeumannBoundaries) { 72356282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundaries,&isequal);CHKERRQ(ierr); 72456282151SStefano Zampini } 725785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 726785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 72753cdbc3dSStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 72836e030ebSStefano Zampini pcbddc->NeumannBoundaries = NeumannBoundaries; 72956282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7300c7d97c5SJed Brown PetscFunctionReturn(0); 7310c7d97c5SJed Brown } 7321e6b0712SBarry Smith 7330c7d97c5SJed Brown #undef __FUNCT__ 7340c7d97c5SJed Brown #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 73557527edcSJed Brown /*@ 73628509bceSStefano Zampini PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 73757527edcSJed Brown 738785d1243SStefano Zampini Collective 73957527edcSJed Brown 74057527edcSJed Brown Input Parameters: 74157527edcSJed Brown + pc - the preconditioning context 742785d1243SStefano Zampini - NeumannBoundaries - parallel IS defining the Neumann boundaries 74357527edcSJed Brown 74457527edcSJed Brown Level: intermediate 74557527edcSJed Brown 7460f202f7eSStefano Zampini Notes: 7470f202f7eSStefano Zampini Any process can list any global node 74857527edcSJed Brown 7490f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 75057527edcSJed Brown @*/ 75153cdbc3dSStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 7520c7d97c5SJed Brown { 7530c7d97c5SJed Brown PetscErrorCode ierr; 7540c7d97c5SJed Brown 7550c7d97c5SJed Brown PetscFunctionBegin; 7560c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 757674ae819SStefano Zampini PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 758785d1243SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 75953cdbc3dSStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 76053cdbc3dSStefano Zampini PetscFunctionReturn(0); 76153cdbc3dSStefano Zampini } 7621e6b0712SBarry Smith 76353cdbc3dSStefano Zampini #undef __FUNCT__ 76482ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 76582ba6b80SStefano Zampini static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 7660c7d97c5SJed Brown { 7670c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 76856282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 7690c7d97c5SJed Brown PetscErrorCode ierr; 7700c7d97c5SJed Brown 7710c7d97c5SJed Brown PetscFunctionBegin; 77256282151SStefano Zampini ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 77356282151SStefano Zampini if (pcbddc->NeumannBoundariesLocal) { 77456282151SStefano Zampini ierr = ISEqual(NeumannBoundaries,pcbddc->NeumannBoundariesLocal,&isequal);CHKERRQ(ierr); 77556282151SStefano Zampini } 776785d1243SStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 777785d1243SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 7780c7d97c5SJed Brown ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 779785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 78056282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 7810c7d97c5SJed Brown PetscFunctionReturn(0); 7820c7d97c5SJed Brown } 7830c7d97c5SJed Brown 7840c7d97c5SJed Brown #undef __FUNCT__ 78582ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 7860c7d97c5SJed Brown /*@ 78782ba6b80SStefano Zampini PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 7880c7d97c5SJed Brown 789785d1243SStefano Zampini Collective 7900c7d97c5SJed Brown 7910c7d97c5SJed Brown Input Parameters: 7920c7d97c5SJed Brown + pc - the preconditioning context 79382ba6b80SStefano Zampini - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 7940c7d97c5SJed Brown 7950c7d97c5SJed Brown Level: intermediate 7960c7d97c5SJed Brown 7970c7d97c5SJed Brown Notes: 7980c7d97c5SJed Brown 7990f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 8000c7d97c5SJed Brown @*/ 80182ba6b80SStefano Zampini PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 8020c7d97c5SJed Brown { 8030c7d97c5SJed Brown PetscErrorCode ierr; 8040c7d97c5SJed Brown 8050c7d97c5SJed Brown PetscFunctionBegin; 8060c7d97c5SJed Brown PetscValidHeaderSpecific(pc,PC_CLASSID,1); 8070c7d97c5SJed Brown PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 80882ba6b80SStefano Zampini PetscCheckSameComm(pc,1,NeumannBoundaries,2); 80982ba6b80SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 81053cdbc3dSStefano Zampini PetscFunctionReturn(0); 81153cdbc3dSStefano Zampini } 81253cdbc3dSStefano Zampini 81353cdbc3dSStefano Zampini #undef __FUNCT__ 814da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 815da1bb401SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 816da1bb401SStefano Zampini { 817da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 818da1bb401SStefano Zampini 819da1bb401SStefano Zampini PetscFunctionBegin; 820da1bb401SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundaries; 821da1bb401SStefano Zampini PetscFunctionReturn(0); 822da1bb401SStefano Zampini } 8231e6b0712SBarry Smith 824da1bb401SStefano Zampini #undef __FUNCT__ 825da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 826da1bb401SStefano Zampini /*@ 827785d1243SStefano Zampini PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 828da1bb401SStefano Zampini 829785d1243SStefano Zampini Collective 830785d1243SStefano Zampini 831785d1243SStefano Zampini Input Parameters: 832785d1243SStefano Zampini . pc - the preconditioning context 833785d1243SStefano Zampini 834785d1243SStefano Zampini Output Parameters: 835785d1243SStefano Zampini . DirichletBoundaries - index set defining the Dirichlet boundaries 836785d1243SStefano Zampini 837785d1243SStefano Zampini Level: intermediate 838785d1243SStefano Zampini 8390f202f7eSStefano Zampini Notes: 8400f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 841785d1243SStefano Zampini 842785d1243SStefano Zampini .seealso: PCBDDC 843785d1243SStefano Zampini @*/ 844785d1243SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 845785d1243SStefano Zampini { 846785d1243SStefano Zampini PetscErrorCode ierr; 847785d1243SStefano Zampini 848785d1243SStefano Zampini PetscFunctionBegin; 849785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 850785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 851785d1243SStefano Zampini PetscFunctionReturn(0); 852785d1243SStefano Zampini } 853785d1243SStefano Zampini 854785d1243SStefano Zampini #undef __FUNCT__ 855785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 856785d1243SStefano Zampini static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 857785d1243SStefano Zampini { 858785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 859785d1243SStefano Zampini 860785d1243SStefano Zampini PetscFunctionBegin; 861785d1243SStefano Zampini *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 862785d1243SStefano Zampini PetscFunctionReturn(0); 863785d1243SStefano Zampini } 864785d1243SStefano Zampini 865785d1243SStefano Zampini #undef __FUNCT__ 86682ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 867da1bb401SStefano Zampini /*@ 86882ba6b80SStefano Zampini PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 869da1bb401SStefano Zampini 870785d1243SStefano Zampini Collective 871da1bb401SStefano Zampini 872da1bb401SStefano Zampini Input Parameters: 87328509bceSStefano Zampini . pc - the preconditioning context 874da1bb401SStefano Zampini 875da1bb401SStefano Zampini Output Parameters: 87628509bceSStefano Zampini . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 877da1bb401SStefano Zampini 878da1bb401SStefano Zampini Level: intermediate 879da1bb401SStefano Zampini 880da1bb401SStefano Zampini Notes: 8810f202f7eSStefano 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). 8820f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 883da1bb401SStefano Zampini 884da1bb401SStefano Zampini .seealso: PCBDDC 885da1bb401SStefano Zampini @*/ 88682ba6b80SStefano Zampini PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 887da1bb401SStefano Zampini { 888da1bb401SStefano Zampini PetscErrorCode ierr; 889da1bb401SStefano Zampini 890da1bb401SStefano Zampini PetscFunctionBegin; 891da1bb401SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 89282ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 893da1bb401SStefano Zampini PetscFunctionReturn(0); 894da1bb401SStefano Zampini } 8951e6b0712SBarry Smith 896da1bb401SStefano Zampini #undef __FUNCT__ 89753cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 89853cdbc3dSStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 89953cdbc3dSStefano Zampini { 90053cdbc3dSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 90153cdbc3dSStefano Zampini 90253cdbc3dSStefano Zampini PetscFunctionBegin; 90353cdbc3dSStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundaries; 90453cdbc3dSStefano Zampini PetscFunctionReturn(0); 90553cdbc3dSStefano Zampini } 9061e6b0712SBarry Smith 90753cdbc3dSStefano Zampini #undef __FUNCT__ 90853cdbc3dSStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 90953cdbc3dSStefano Zampini /*@ 910785d1243SStefano Zampini PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 91153cdbc3dSStefano Zampini 912785d1243SStefano Zampini Collective 913785d1243SStefano Zampini 914785d1243SStefano Zampini Input Parameters: 915785d1243SStefano Zampini . pc - the preconditioning context 916785d1243SStefano Zampini 917785d1243SStefano Zampini Output Parameters: 918785d1243SStefano Zampini . NeumannBoundaries - index set defining the Neumann boundaries 919785d1243SStefano Zampini 920785d1243SStefano Zampini Level: intermediate 921785d1243SStefano Zampini 9220f202f7eSStefano Zampini Notes: 9230f202f7eSStefano Zampini The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 924785d1243SStefano Zampini 925785d1243SStefano Zampini .seealso: PCBDDC 926785d1243SStefano Zampini @*/ 927785d1243SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 928785d1243SStefano Zampini { 929785d1243SStefano Zampini PetscErrorCode ierr; 930785d1243SStefano Zampini 931785d1243SStefano Zampini PetscFunctionBegin; 932785d1243SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 933785d1243SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 934785d1243SStefano Zampini PetscFunctionReturn(0); 935785d1243SStefano Zampini } 936785d1243SStefano Zampini 937785d1243SStefano Zampini #undef __FUNCT__ 938785d1243SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 939785d1243SStefano Zampini static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 940785d1243SStefano Zampini { 941785d1243SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 942785d1243SStefano Zampini 943785d1243SStefano Zampini PetscFunctionBegin; 944785d1243SStefano Zampini *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 945785d1243SStefano Zampini PetscFunctionReturn(0); 946785d1243SStefano Zampini } 947785d1243SStefano Zampini 948785d1243SStefano Zampini #undef __FUNCT__ 94982ba6b80SStefano Zampini #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 95053cdbc3dSStefano Zampini /*@ 95182ba6b80SStefano Zampini PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 95253cdbc3dSStefano Zampini 953785d1243SStefano Zampini Collective 95453cdbc3dSStefano Zampini 95553cdbc3dSStefano Zampini Input Parameters: 95628509bceSStefano Zampini . pc - the preconditioning context 95753cdbc3dSStefano Zampini 95853cdbc3dSStefano Zampini Output Parameters: 95928509bceSStefano Zampini . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 96053cdbc3dSStefano Zampini 96153cdbc3dSStefano Zampini Level: intermediate 96253cdbc3dSStefano Zampini 96353cdbc3dSStefano Zampini Notes: 9640f202f7eSStefano 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). 9650f202f7eSStefano Zampini In the latter case, the IS will be available after PCSetUp. 96653cdbc3dSStefano Zampini 96753cdbc3dSStefano Zampini .seealso: PCBDDC 96853cdbc3dSStefano Zampini @*/ 96982ba6b80SStefano Zampini PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 97053cdbc3dSStefano Zampini { 97153cdbc3dSStefano Zampini PetscErrorCode ierr; 97253cdbc3dSStefano Zampini 97353cdbc3dSStefano Zampini PetscFunctionBegin; 97453cdbc3dSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 97582ba6b80SStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 9760c7d97c5SJed Brown PetscFunctionReturn(0); 9770c7d97c5SJed Brown } 9781e6b0712SBarry Smith 97936e030ebSStefano Zampini #undef __FUNCT__ 980da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 9811a83f524SJed Brown static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 98236e030ebSStefano Zampini { 98336e030ebSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 984da1bb401SStefano Zampini PCBDDCGraph mat_graph = pcbddc->mat_graph; 98556282151SStefano Zampini PetscBool same_data = PETSC_FALSE; 986da1bb401SStefano Zampini PetscErrorCode ierr; 98736e030ebSStefano Zampini 98836e030ebSStefano Zampini PetscFunctionBegin; 9898687889aSStefano Zampini if (!nvtxs) { 9908687889aSStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 9918687889aSStefano Zampini PetscFunctionReturn(0); 9928687889aSStefano Zampini } 9932d505d7fSStefano Zampini if (mat_graph->nvtxs_csr == nvtxs && mat_graph->freecsr) { /* we own the data */ 99456282151SStefano Zampini if (mat_graph->xadj == xadj && mat_graph->adjncy == adjncy) same_data = PETSC_TRUE; 99556282151SStefano Zampini if (!same_data && mat_graph->xadj[nvtxs] == xadj[nvtxs]) { 9962d505d7fSStefano Zampini ierr = PetscMemcmp(xadj,mat_graph->xadj,(nvtxs+1)*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9972d505d7fSStefano Zampini if (same_data) { 9982d505d7fSStefano Zampini ierr = PetscMemcmp(adjncy,mat_graph->adjncy,xadj[nvtxs]*sizeof(PetscInt),&same_data);CHKERRQ(ierr); 9992d505d7fSStefano Zampini } 100056282151SStefano Zampini } 100156282151SStefano Zampini } 100256282151SStefano Zampini if (!same_data) { 1003674ae819SStefano Zampini /* free old CSR */ 1004674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 1005674ae819SStefano Zampini /* get CSR into graph structure */ 1006da1bb401SStefano Zampini if (copymode == PETSC_COPY_VALUES) { 1007854ce69bSBarry Smith ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 1008785e854fSJed Brown ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 1009674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 1010674ae819SStefano Zampini ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 1011a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 1012da1bb401SStefano Zampini } else if (copymode == PETSC_OWN_POINTER) { 10131a83f524SJed Brown mat_graph->xadj = (PetscInt*)xadj; 10141a83f524SJed Brown mat_graph->adjncy = (PetscInt*)adjncy; 1015a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_TRUE; 1016a1dbd327SStefano Zampini } else if (copymode == PETSC_USE_POINTER) { 1017a1dbd327SStefano Zampini mat_graph->xadj = (PetscInt*)xadj; 1018a1dbd327SStefano Zampini mat_graph->adjncy = (PetscInt*)adjncy; 1019a1dbd327SStefano Zampini mat_graph->freecsr = PETSC_FALSE; 1020a1dbd327SStefano Zampini } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d",copymode); 1021575ad6abSStefano Zampini mat_graph->nvtxs_csr = nvtxs; 102256282151SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 102356282151SStefano Zampini } 102436e030ebSStefano Zampini PetscFunctionReturn(0); 102536e030ebSStefano Zampini } 10261e6b0712SBarry Smith 102736e030ebSStefano Zampini #undef __FUNCT__ 1028da1bb401SStefano Zampini #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 102936e030ebSStefano Zampini /*@ 103054fffbccSStefano Zampini PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local degrees of freedom. 103136e030ebSStefano Zampini 103236e030ebSStefano Zampini Not collective 103336e030ebSStefano Zampini 103436e030ebSStefano Zampini Input Parameters: 103554fffbccSStefano Zampini + pc - the preconditioning context. 103654fffbccSStefano Zampini . nvtxs - number of local vertices of the graph (i.e., the number of local dofs). 103754fffbccSStefano Zampini . xadj, adjncy - the connectivity of the dofs in CSR format. 103854fffbccSStefano Zampini - copymode - supported modes are PETSC_COPY_VALUES, PETSC_USE_POINTER or PETSC_OWN_POINTER. 103936e030ebSStefano Zampini 104036e030ebSStefano Zampini Level: intermediate 104136e030ebSStefano Zampini 104254fffbccSStefano Zampini Notes: A dof is considered connected with all local dofs if xadj[dof+1]-xadj[dof] == 1 and adjncy[xadj[dof]] is negative. 104336e030ebSStefano Zampini 104428509bceSStefano Zampini .seealso: PCBDDC,PetscCopyMode 104536e030ebSStefano Zampini @*/ 10461a83f524SJed Brown PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 104736e030ebSStefano Zampini { 1048575ad6abSStefano Zampini void (*f)(void) = 0; 104936e030ebSStefano Zampini PetscErrorCode ierr; 105036e030ebSStefano Zampini 105136e030ebSStefano Zampini PetscFunctionBegin; 105236e030ebSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 10538687889aSStefano Zampini if (nvtxs) { 1054674ae819SStefano Zampini PetscValidIntPointer(xadj,3); 1055d7de1dd9SStefano Zampini PetscValidIntPointer(adjncy,4); 10568687889aSStefano Zampini } 10571a83f524SJed Brown ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 1058575ad6abSStefano Zampini /* free arrays if PCBDDC is not the PC type */ 1059575ad6abSStefano Zampini ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 1060575ad6abSStefano Zampini if (!f && copymode == PETSC_OWN_POINTER) { 1061575ad6abSStefano Zampini ierr = PetscFree(xadj);CHKERRQ(ierr); 1062575ad6abSStefano Zampini ierr = PetscFree(adjncy);CHKERRQ(ierr); 1063da1bb401SStefano Zampini } 106436e030ebSStefano Zampini PetscFunctionReturn(0); 106536e030ebSStefano Zampini } 10661e6b0712SBarry Smith 10679c0446d6SStefano Zampini #undef __FUNCT__ 106863602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 106963602bcaSStefano Zampini static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 107063602bcaSStefano Zampini { 107163602bcaSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 107263602bcaSStefano Zampini PetscInt i; 107356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 107463602bcaSStefano Zampini PetscErrorCode ierr; 107563602bcaSStefano Zampini 107663602bcaSStefano Zampini PetscFunctionBegin; 107756282151SStefano Zampini if (pcbddc->n_ISForDofsLocal == n_is) { 107856282151SStefano Zampini for (i=0;i<n_is;i++) { 107956282151SStefano Zampini PetscBool isequalt; 108056282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofsLocal[i],&isequalt);CHKERRQ(ierr); 108156282151SStefano Zampini if (!isequalt) break; 108256282151SStefano Zampini } 108356282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 108456282151SStefano Zampini } 108556282151SStefano Zampini for (i=0;i<n_is;i++) { 108656282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 108756282151SStefano Zampini } 108863602bcaSStefano Zampini /* Destroy ISes if they were already set */ 108963602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 109063602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 109163602bcaSStefano Zampini } 109263602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 109363602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 109463602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 109563602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 109663602bcaSStefano Zampini } 109763602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 109863602bcaSStefano Zampini pcbddc->n_ISForDofs = 0; 109963602bcaSStefano Zampini /* allocate space then set */ 1100d02579f5SStefano Zampini if (n_is) { 1101d02579f5SStefano Zampini ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 1102d02579f5SStefano Zampini } 110363602bcaSStefano Zampini for (i=0;i<n_is;i++) { 110463602bcaSStefano Zampini pcbddc->ISForDofsLocal[i] = ISForDofs[i]; 110563602bcaSStefano Zampini } 110663602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = n_is; 110763602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 110856282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 110963602bcaSStefano Zampini PetscFunctionReturn(0); 111063602bcaSStefano Zampini } 111163602bcaSStefano Zampini 111263602bcaSStefano Zampini #undef __FUNCT__ 111363602bcaSStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 111463602bcaSStefano Zampini /*@ 111563602bcaSStefano Zampini PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 111663602bcaSStefano Zampini 111763602bcaSStefano Zampini Collective 111863602bcaSStefano Zampini 111963602bcaSStefano Zampini Input Parameters: 112063602bcaSStefano Zampini + pc - the preconditioning context 11210f202f7eSStefano Zampini . n_is - number of index sets defining the fields 11220f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in local ordering 112363602bcaSStefano Zampini 112463602bcaSStefano Zampini Level: intermediate 112563602bcaSStefano Zampini 11260f202f7eSStefano Zampini Notes: 11270f202f7eSStefano Zampini n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 112863602bcaSStefano Zampini 112963602bcaSStefano Zampini .seealso: PCBDDC 113063602bcaSStefano Zampini @*/ 113163602bcaSStefano Zampini PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 113263602bcaSStefano Zampini { 113363602bcaSStefano Zampini PetscInt i; 113463602bcaSStefano Zampini PetscErrorCode ierr; 113563602bcaSStefano Zampini 113663602bcaSStefano Zampini PetscFunctionBegin; 113763602bcaSStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 113863602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 113963602bcaSStefano Zampini for (i=0;i<n_is;i++) { 114063602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 114163602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 114263602bcaSStefano Zampini } 1143e71e7a71SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 114463602bcaSStefano Zampini PetscFunctionReturn(0); 114563602bcaSStefano Zampini } 114663602bcaSStefano Zampini 114763602bcaSStefano Zampini #undef __FUNCT__ 11489c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 11499c0446d6SStefano Zampini static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 11509c0446d6SStefano Zampini { 11519c0446d6SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 11529c0446d6SStefano Zampini PetscInt i; 115356282151SStefano Zampini PetscBool isequal = PETSC_FALSE; 11549c0446d6SStefano Zampini PetscErrorCode ierr; 11559c0446d6SStefano Zampini 11569c0446d6SStefano Zampini PetscFunctionBegin; 115756282151SStefano Zampini if (pcbddc->n_ISForDofs == n_is) { 115856282151SStefano Zampini for (i=0;i<n_is;i++) { 115956282151SStefano Zampini PetscBool isequalt; 116056282151SStefano Zampini ierr = ISEqual(ISForDofs[i],pcbddc->ISForDofs[i],&isequalt);CHKERRQ(ierr); 116156282151SStefano Zampini if (!isequalt) break; 116256282151SStefano Zampini } 116356282151SStefano Zampini if (i == n_is) isequal = PETSC_TRUE; 116456282151SStefano Zampini } 116556282151SStefano Zampini for (i=0;i<n_is;i++) { 116656282151SStefano Zampini ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 116756282151SStefano Zampini } 1168da1bb401SStefano Zampini /* Destroy ISes if they were already set */ 11699c0446d6SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 11709c0446d6SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 11719c0446d6SStefano Zampini } 1172d11ae9bbSstefano_zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 117363602bcaSStefano Zampini /* last user setting takes precendence -> destroy any other customization */ 117463602bcaSStefano Zampini for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 117563602bcaSStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 117663602bcaSStefano Zampini } 117763602bcaSStefano Zampini ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 117863602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 1179da1bb401SStefano Zampini /* allocate space then set */ 1180d02579f5SStefano Zampini if (n_is) { 1181785e854fSJed Brown ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 1182d02579f5SStefano Zampini } 11839c0446d6SStefano Zampini for (i=0;i<n_is;i++) { 1184da1bb401SStefano Zampini pcbddc->ISForDofs[i] = ISForDofs[i]; 11859c0446d6SStefano Zampini } 11869c0446d6SStefano Zampini pcbddc->n_ISForDofs = n_is; 118763602bcaSStefano Zampini if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 118856282151SStefano Zampini if (!isequal) pcbddc->recompute_topography = PETSC_TRUE; 11899c0446d6SStefano Zampini PetscFunctionReturn(0); 11909c0446d6SStefano Zampini } 11911e6b0712SBarry Smith 11929c0446d6SStefano Zampini #undef __FUNCT__ 11939c0446d6SStefano Zampini #define __FUNCT__ "PCBDDCSetDofsSplitting" 11949c0446d6SStefano Zampini /*@ 119563602bcaSStefano Zampini PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 11969c0446d6SStefano Zampini 119763602bcaSStefano Zampini Collective 11989c0446d6SStefano Zampini 11999c0446d6SStefano Zampini Input Parameters: 12009c0446d6SStefano Zampini + pc - the preconditioning context 12010f202f7eSStefano Zampini . n_is - number of index sets defining the fields 12020f202f7eSStefano Zampini - ISForDofs - array of IS describing the fields in global ordering 12039c0446d6SStefano Zampini 12049c0446d6SStefano Zampini Level: intermediate 12059c0446d6SStefano Zampini 12060f202f7eSStefano Zampini Notes: 12070f202f7eSStefano Zampini Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 12089c0446d6SStefano Zampini 12099c0446d6SStefano Zampini .seealso: PCBDDC 12109c0446d6SStefano Zampini @*/ 12119c0446d6SStefano Zampini PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 12129c0446d6SStefano Zampini { 12132b510759SStefano Zampini PetscInt i; 12149c0446d6SStefano Zampini PetscErrorCode ierr; 12159c0446d6SStefano Zampini 12169c0446d6SStefano Zampini PetscFunctionBegin; 12179c0446d6SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 121863602bcaSStefano Zampini PetscValidLogicalCollectiveInt(pc,n_is,2); 12192b510759SStefano Zampini for (i=0;i<n_is;i++) { 122063602bcaSStefano Zampini PetscCheckSameComm(pc,1,ISForDofs[i],3); 122163602bcaSStefano Zampini PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 12222b510759SStefano Zampini } 12239c0446d6SStefano Zampini ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 12249c0446d6SStefano Zampini PetscFunctionReturn(0); 12259c0446d6SStefano Zampini } 1226906d46d4SStefano Zampini 1227534831adSStefano Zampini #undef __FUNCT__ 1228534831adSStefano Zampini #define __FUNCT__ "PCPreSolve_BDDC" 1229534831adSStefano Zampini /* 1230534831adSStefano Zampini PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 1231534831adSStefano Zampini guess if a transformation of basis approach has been selected. 12329c0446d6SStefano Zampini 1233534831adSStefano Zampini Input Parameter: 1234534831adSStefano Zampini + pc - the preconditioner contex 1235534831adSStefano Zampini 1236534831adSStefano Zampini Application Interface Routine: PCPreSolve() 1237534831adSStefano Zampini 1238534831adSStefano Zampini Notes: 1239534831adSStefano Zampini The interface routine PCPreSolve() is not usually called directly by 1240534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1241534831adSStefano Zampini */ 1242534831adSStefano Zampini static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1243534831adSStefano Zampini { 1244534831adSStefano Zampini PetscErrorCode ierr; 1245534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1246534831adSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 12473972b0daSStefano Zampini Vec used_vec; 124827b6a85dSStefano Zampini PetscBool save_rhs = PETSC_TRUE, benign_correction_computed; 1249534831adSStefano Zampini 1250534831adSStefano Zampini PetscFunctionBegin; 12511f4df5f7SStefano Zampini /* if we are working with CG, one dirichlet solve can be avoided during Krylov iterations */ 125285c4d303SStefano Zampini if (ksp) { 12531f4df5f7SStefano Zampini PetscBool iscg, isgroppcg, ispipecg, ispipecgrr; 125485c4d303SStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 125527b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPGROPPCG,&isgroppcg);CHKERRQ(ierr); 125627b6a85dSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECG,&ispipecg);CHKERRQ(ierr); 1257f94e96cbSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPPIPECGRR,&ispipecgrr);CHKERRQ(ierr); 12581f4df5f7SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static || (!iscg && !isgroppcg && !ispipecg && !ispipecgrr)) { 125985c4d303SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 126085c4d303SStefano Zampini } 126185c4d303SStefano Zampini } 1262fc17d649SStefano Zampini if (pcbddc->benign_apply_coarse_only || pcbddc->switch_static) { 1263fc17d649SStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 1264fc17d649SStefano Zampini } 12651f4df5f7SStefano Zampini 126685c4d303SStefano Zampini /* Creates parallel work vectors used in presolve */ 126762a6ff1dSStefano Zampini if (!pcbddc->original_rhs) { 126862a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 126962a6ff1dSStefano Zampini } 127062a6ff1dSStefano Zampini if (!pcbddc->temp_solution) { 127162a6ff1dSStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 127262a6ff1dSStefano Zampini } 12738d00608fSStefano Zampini 127427b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 12753972b0daSStefano Zampini if (x) { 12763972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 12773972b0daSStefano Zampini used_vec = x; 12788d00608fSStefano Zampini } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 12793972b0daSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 12803972b0daSStefano Zampini used_vec = pcbddc->temp_solution; 12813972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 128227b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 1283266e20e9SStefano Zampini ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1284266e20e9SStefano Zampini save_rhs = PETSC_FALSE; 1285266e20e9SStefano Zampini pcbddc->eliminate_dirdofs = PETSC_TRUE; 12863972b0daSStefano Zampini } 12878efcfb23SStefano Zampini 12888efcfb23SStefano Zampini /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 12893972b0daSStefano Zampini if (ksp) { 1290a0cb1b98SStefano Zampini /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 12918efcfb23SStefano Zampini ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 12928efcfb23SStefano Zampini if (!pcbddc->ksp_guess_nonzero) { 12933972b0daSStefano Zampini ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 12943972b0daSStefano Zampini } 12953972b0daSStefano Zampini } 12963308cffdSStefano Zampini 12978d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 12983972b0daSStefano Zampini /* Take into account zeroed rows -> change rhs and store solution removed */ 129970c64980SStefano Zampini if (rhs && pcbddc->eliminate_dirdofs) { 13003975b054SStefano Zampini IS dirIS = NULL; 13013975b054SStefano Zampini 1302a07ea27aSStefano Zampini /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 13033975b054SStefano Zampini ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 13043975b054SStefano Zampini if (dirIS) { 1305906d46d4SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1306785d1243SStefano Zampini PetscInt dirsize,i,*is_indices; 13072b095fd8SStefano Zampini PetscScalar *array_x; 13082b095fd8SStefano Zampini const PetscScalar *array_diagonal; 1309785d1243SStefano Zampini 13103972b0daSStefano Zampini ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 13113972b0daSStefano Zampini ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1312e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1313e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1314e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1315e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 131682ba6b80SStefano Zampini ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 13173972b0daSStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 13182b095fd8SStefano Zampini ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 13193972b0daSStefano Zampini ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 13202fa5cd67SKarl Rupp for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 13213972b0daSStefano Zampini ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 13222b095fd8SStefano Zampini ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 13233972b0daSStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1324e176bc59SStefano Zampini ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1325e176bc59SStefano Zampini ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 13268d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13271b968477SStefano Zampini ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 13288efcfb23SStefano Zampini } 1329a07ea27aSStefano Zampini } 1330b76ba322SStefano Zampini 13318efcfb23SStefano Zampini /* remove the computed solution or the initial guess from the rhs */ 13328d00608fSStefano Zampini if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 133327b6a85dSStefano Zampini /* save the original rhs */ 133427b6a85dSStefano Zampini if (save_rhs) { 133527b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 133627b6a85dSStefano Zampini save_rhs = PETSC_FALSE; 13378d00608fSStefano Zampini } 13388d00608fSStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 13393972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 134027b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,used_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 13413972b0daSStefano Zampini ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 13428efcfb23SStefano Zampini ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 134327b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_TRUE; 13447acc28cbSStefano Zampini if (ksp) { 13457acc28cbSStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 13467acc28cbSStefano Zampini } 13473308cffdSStefano Zampini } 13488efcfb23SStefano Zampini ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1349b76ba322SStefano Zampini 1350fc17d649SStefano Zampini /* compute initial vector in benign space if needed 135127b6a85dSStefano Zampini and remove non-benign solution from the rhs */ 135227b6a85dSStefano Zampini benign_correction_computed = PETSC_FALSE; 135327b6a85dSStefano Zampini if (rhs && pcbddc->benign_compute_correction && pcbddc->benign_have_null) { 13541f4df5f7SStefano Zampini /* compute u^*_h using ideas similar to those in Xuemin Tu's PhD thesis (see Section 4.8.1) 13551f4df5f7SStefano Zampini Recursively apply BDDC in the multilevel case */ 13560369aaf7SStefano Zampini if (!pcbddc->benign_vec) { 13570369aaf7SStefano Zampini ierr = VecDuplicate(rhs,&pcbddc->benign_vec);CHKERRQ(ierr); 13580369aaf7SStefano Zampini } 13594fee134fSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_TRUE; 136027b6a85dSStefano Zampini if (!pcbddc->benign_skip_correction) { 13611dd7afcfSStefano Zampini ierr = PCApply_BDDC(pc,rhs,pcbddc->benign_vec);CHKERRQ(ierr); 13623bca92a6SStefano Zampini benign_correction_computed = PETSC_TRUE; 13631f4df5f7SStefano Zampini if (pcbddc->temp_solution_used) { 13641f4df5f7SStefano Zampini ierr = VecAXPY(pcbddc->temp_solution,1.0,pcbddc->benign_vec);CHKERRQ(ierr); 13651f4df5f7SStefano Zampini } 13661f4df5f7SStefano Zampini ierr = VecScale(pcbddc->benign_vec,-1.0);CHKERRQ(ierr); 136727b6a85dSStefano Zampini /* store the original rhs if not done earlier */ 136827b6a85dSStefano Zampini if (save_rhs) { 136927b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 137092e3dcfbSStefano Zampini } 137127b6a85dSStefano Zampini if (pcbddc->rhs_change) { 13720369aaf7SStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,rhs,rhs);CHKERRQ(ierr); 137327b6a85dSStefano Zampini } else { 137427b6a85dSStefano Zampini ierr = MatMultAdd(pc->mat,pcbddc->benign_vec,pcbddc->original_rhs,rhs);CHKERRQ(ierr); 137527b6a85dSStefano Zampini } 13760369aaf7SStefano Zampini pcbddc->rhs_change = PETSC_TRUE; 137727b6a85dSStefano Zampini } 137827b6a85dSStefano Zampini pcbddc->benign_apply_coarse_only = PETSC_FALSE; 13790369aaf7SStefano Zampini } 13802d4c4fecSStefano Zampini 13812d4c4fecSStefano Zampini /* dbg output */ 1382a198735bSStefano Zampini if (pcbddc->dbg_flag && benign_correction_computed) { 13831f4df5f7SStefano Zampini Vec v; 13841f4df5f7SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&v);CHKERRQ(ierr); 13851f4df5f7SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,v);CHKERRQ(ierr); 13861f4df5f7SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,v,PETSC_TRUE);CHKERRQ(ierr); 1387a198735bSStefano Zampini ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"LEVEL %d: is the correction benign?\n",pcbddc->current_level);CHKERRQ(ierr); 13882d4c4fecSStefano Zampini ierr = PetscScalarView(pcbddc->benign_n,pcbddc->benign_p0,PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)));CHKERRQ(ierr); 13891f4df5f7SStefano Zampini ierr = VecDestroy(&v);CHKERRQ(ierr); 13901f4df5f7SStefano Zampini } 13910369aaf7SStefano Zampini 13920369aaf7SStefano Zampini /* set initial guess if using PCG */ 13938ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 13940369aaf7SStefano Zampini if (x && pcbddc->use_exact_dirichlet_trick) { 13950369aaf7SStefano Zampini ierr = VecSet(x,0.0);CHKERRQ(ierr); 13961dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 139727b6a85dSStefano Zampini if (benign_correction_computed) { /* we have already saved the changed rhs */ 13981dd7afcfSStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 13991dd7afcfSStefano Zampini } else { 14001dd7afcfSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,rhs,pcis->vec1_global);CHKERRQ(ierr); 14011dd7afcfSStefano Zampini } 14021dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14031dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_global,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14041dd7afcfSStefano Zampini } else { 14050369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14060369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 14071dd7afcfSStefano Zampini } 14080369aaf7SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 14091dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior) { 14101dd7afcfSStefano Zampini ierr = VecSet(pcis->vec1_global,0.);CHKERRQ(ierr); 14111dd7afcfSStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14121dd7afcfSStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,pcis->vec1_global,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14131dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcis->vec1_global,x);CHKERRQ(ierr); 14141dd7afcfSStefano Zampini } else { 14150369aaf7SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14160369aaf7SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 14171dd7afcfSStefano Zampini } 14180369aaf7SStefano Zampini if (ksp) { 14190369aaf7SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 14200369aaf7SStefano Zampini } 14218ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_TRUE; 1422266e20e9SStefano Zampini } else if (pcbddc->ChangeOfBasisMatrix && pcbddc->change_interior && benign_correction_computed && pcbddc->use_exact_dirichlet_trick) { 1423266e20e9SStefano Zampini ierr = VecLockPop(pcis->vec1_global);CHKERRQ(ierr); 14240369aaf7SStefano Zampini } 1425534831adSStefano Zampini PetscFunctionReturn(0); 1426534831adSStefano Zampini } 1427906d46d4SStefano Zampini 1428534831adSStefano Zampini #undef __FUNCT__ 1429534831adSStefano Zampini #define __FUNCT__ "PCPostSolve_BDDC" 1430534831adSStefano Zampini /* 1431534831adSStefano Zampini PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1432534831adSStefano Zampini approach has been selected. Also, restores rhs to its original state. 1433534831adSStefano Zampini 1434534831adSStefano Zampini Input Parameter: 1435534831adSStefano Zampini + pc - the preconditioner contex 1436534831adSStefano Zampini 1437534831adSStefano Zampini Application Interface Routine: PCPostSolve() 1438534831adSStefano Zampini 1439534831adSStefano Zampini Notes: 1440534831adSStefano Zampini The interface routine PCPostSolve() is not usually called directly by 1441534831adSStefano Zampini the user, but instead is called by KSPSolve(). 1442534831adSStefano Zampini */ 1443534831adSStefano Zampini static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1444534831adSStefano Zampini { 1445534831adSStefano Zampini PetscErrorCode ierr; 1446534831adSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1447534831adSStefano Zampini 1448534831adSStefano Zampini PetscFunctionBegin; 14493972b0daSStefano Zampini /* add solution removed in presolve */ 14506bcfc461SStefano Zampini if (x && pcbddc->rhs_change) { 145127b6a85dSStefano Zampini if (pcbddc->temp_solution_used) { 14523425bc38SStefano Zampini ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 145327b6a85dSStefano Zampini } else if (pcbddc->benign_compute_correction) { 145427b6a85dSStefano Zampini ierr = VecAXPY(x,-1.0,pcbddc->benign_vec);CHKERRQ(ierr); 14553425bc38SStefano Zampini } 145627b6a85dSStefano Zampini } 145727b6a85dSStefano Zampini pcbddc->temp_solution_used = PETSC_FALSE; 145827b6a85dSStefano Zampini 1459266e20e9SStefano Zampini /* restore rhs to its original state (not needed for FETI-DP) */ 14608d00608fSStefano Zampini if (rhs && pcbddc->rhs_change) { 146127b6a85dSStefano Zampini ierr = VecSwap(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1462fb223d50SStefano Zampini } 14638d00608fSStefano Zampini pcbddc->rhs_change = PETSC_FALSE; 14648efcfb23SStefano Zampini /* restore ksp guess state */ 14658efcfb23SStefano Zampini if (ksp) { 14668efcfb23SStefano Zampini ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 14678efcfb23SStefano Zampini } 14688ae0ca82SStefano Zampini /* reset flag for exact dirichlet trick */ 14698ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 1470534831adSStefano Zampini PetscFunctionReturn(0); 1471534831adSStefano Zampini } 147253cdbc3dSStefano Zampini #undef __FUNCT__ 147353cdbc3dSStefano Zampini #define __FUNCT__ "PCSetUp_BDDC" 14740c7d97c5SJed Brown /* 14750c7d97c5SJed Brown PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 14760c7d97c5SJed Brown by setting data structures and options. 14770c7d97c5SJed Brown 14780c7d97c5SJed Brown Input Parameter: 147953cdbc3dSStefano Zampini + pc - the preconditioner context 14800c7d97c5SJed Brown 14810c7d97c5SJed Brown Application Interface Routine: PCSetUp() 14820c7d97c5SJed Brown 14830c7d97c5SJed Brown Notes: 14840c7d97c5SJed Brown The interface routine PCSetUp() is not usually called directly by 14850c7d97c5SJed Brown the user, but instead is called by PCApply() if necessary. 14860c7d97c5SJed Brown */ 148753cdbc3dSStefano Zampini PetscErrorCode PCSetUp_BDDC(PC pc) 14880c7d97c5SJed Brown { 14890c7d97c5SJed Brown PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1490c703fcc7SStefano Zampini PCBDDCSubSchurs sub_schurs; 14915e8657edSStefano Zampini Mat_IS* matis; 149208122e43SStefano Zampini MatNullSpace nearnullspace; 1493c9ed8603SStefano Zampini IS zerodiag = NULL; 149491e8d312SStefano Zampini PetscInt nrows,ncols; 1495c703fcc7SStefano Zampini PetscBool computesubschurs; 14968de1fae6SStefano Zampini PetscBool computeconstraintsmatrix; 14975e8657edSStefano Zampini PetscBool new_nearnullspace_provided,ismatis; 1498c703fcc7SStefano Zampini PetscErrorCode ierr; 14990c7d97c5SJed Brown 15000c7d97c5SJed Brown PetscFunctionBegin; 15015e8657edSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 15026c4ed002SBarry Smith if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 150391e8d312SStefano Zampini ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 15046c4ed002SBarry Smith if (nrows != ncols) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 15055e8657edSStefano Zampini matis = (Mat_IS*)pc->pmat->data; 1506f4ddd8eeSStefano Zampini /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 15073b03a366Sstefano_zampini /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 150871582508SStefano Zampini Also, BDDC builds its own KSP for the Dirichlet problem */ 1509c703fcc7SStefano Zampini if (pc->setupcalled && pc->flag != SAME_NONZERO_PATTERN) pcbddc->recompute_topography = PETSC_TRUE; 15108de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_FALSE; 1511b087196eSStefano Zampini 1512b087196eSStefano Zampini /* check parameters' compatibility */ 1513b7ab4a40SStefano Zampini if (!pcbddc->use_deluxe_scaling) pcbddc->deluxe_zerorows = PETSC_FALSE; 15149d54b7f4SStefano Zampini pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0); 1515bf3a8328SStefano Zampini pcbddc->adaptive_userdefined = (PetscBool)(pcbddc->adaptive_selection && pcbddc->adaptive_userdefined); 1516862806e4SStefano Zampini if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1517862806e4SStefano Zampini 15185a95e1ceSStefano Zampini computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 151916909a7fSStefano Zampini if (pcbddc->switch_static) { 152016909a7fSStefano Zampini PetscBool ismatis; 152116909a7fSStefano Zampini ierr = PetscObjectTypeCompare((PetscObject)pc->mat,MATIS,&ismatis);CHKERRQ(ierr); 152216909a7fSStefano Zampini if (!ismatis) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"When the static switch is one, the iteration matrix should be of type MATIS"); 152316909a7fSStefano Zampini } 152416909a7fSStefano Zampini 152571582508SStefano Zampini /* activate all connected components if the netflux has been requested */ 1526bb05f991SStefano Zampini if (pcbddc->compute_nonetflux) { 1527bb05f991SStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 1528bb05f991SStefano Zampini pcbddc->use_edges = PETSC_TRUE; 1529bb05f991SStefano Zampini pcbddc->use_faces = PETSC_TRUE; 1530bb05f991SStefano Zampini } 1531bb05f991SStefano Zampini 1532f4ddd8eeSStefano Zampini /* Get stdout for dbg */ 153370cf5478SStefano Zampini if (pcbddc->dbg_flag) { 153470cf5478SStefano Zampini if (!pcbddc->dbg_viewer) { 153558a03d70SStefano Zampini pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 15361575c14dSBarry Smith ierr = PetscViewerASCIIPushSynchronized(pcbddc->dbg_viewer);CHKERRQ(ierr); 1537f4ddd8eeSStefano Zampini } 153858a03d70SStefano Zampini ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1539f4ddd8eeSStefano Zampini } 1540f4ddd8eeSStefano Zampini 1541c703fcc7SStefano Zampini /* process topology information */ 154271582508SStefano Zampini if (pcbddc->recompute_topography) { 154371582508SStefano Zampini ierr = PCBDDCComputeLocalTopologyInfo(pc);CHKERRQ(ierr); 1544c703fcc7SStefano Zampini if (pcbddc->discretegradient) { 1545a13144ffSStefano Zampini ierr = PCBDDCNedelecSupport(pc);CHKERRQ(ierr); 1546a13144ffSStefano Zampini } 1547c703fcc7SStefano Zampini } 1548a13144ffSStefano Zampini 1549c703fcc7SStefano Zampini /* change basis if requested by the user */ 15505e8657edSStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix) { 15515e8657edSStefano Zampini /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 15525e8657edSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 15535e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 15545e8657edSStefano Zampini } else { 1555b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 15565e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 15575e8657edSStefano Zampini pcbddc->local_mat = matis->A; 1558d16cbb6bSStefano Zampini } 1559d16cbb6bSStefano Zampini 1560c703fcc7SStefano Zampini /* detect local disconnected subdomains if requested */ 1561c703fcc7SStefano Zampini if (pcbddc->detect_disconnected && pcbddc->recompute_topography) { 1562c703fcc7SStefano Zampini PetscInt i; 1563c703fcc7SStefano Zampini for (i=0;i<pcbddc->n_local_subs;i++) { 1564c703fcc7SStefano Zampini ierr = ISDestroy(&pcbddc->local_subs[i]);CHKERRQ(ierr); 1565c703fcc7SStefano Zampini } 1566c703fcc7SStefano Zampini ierr = PetscFree(pcbddc->local_subs);CHKERRQ(ierr); 15674f1b2e48SStefano Zampini ierr = MatDetectDisconnectedComponents(pcbddc->local_mat,PETSC_FALSE,&pcbddc->n_local_subs,&pcbddc->local_subs);CHKERRQ(ierr); 15684f1b2e48SStefano Zampini } 15694f1b2e48SStefano Zampini 15704f1b2e48SStefano Zampini /* 1571c703fcc7SStefano Zampini Compute change of basis on local pressures (aka zerodiag dofs) with the benign trick 15724f1b2e48SStefano Zampini This should come earlier then PCISSetUp for extracting the correct subdomain matrices 15734f1b2e48SStefano Zampini */ 15741dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_TRUE);CHKERRQ(ierr); 1575d16cbb6bSStefano Zampini if (pcbddc->benign_saddle_point) { 15769f47a83aSStefano Zampini PC_IS* pcis = (PC_IS*)pc->data; 15779f47a83aSStefano Zampini 157805b28244SStefano Zampini if (pcbddc->user_ChangeOfBasisMatrix || pcbddc->use_change_of_basis || !computesubschurs) pcbddc->benign_change_explicit = PETSC_TRUE; 1579669cc0f4SStefano Zampini /* detect local saddle point and change the basis in pcbddc->local_mat (TODO: reuse case) */ 1580339f8db1SStefano Zampini ierr = PCBDDCBenignDetectSaddlePoint(pc,&zerodiag);CHKERRQ(ierr); 1581a3df083aSStefano Zampini /* pop B0 mat from local mat */ 1582c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 15831dd7afcfSStefano Zampini /* give pcis a hint to not reuse submatrices during PCISCreate */ 15841dd7afcfSStefano Zampini if (pc->flag == SAME_NONZERO_PATTERN && pcis->reusesubmatrices == PETSC_TRUE) { 15851dd7afcfSStefano Zampini if (pcbddc->benign_n && (pcbddc->benign_change_explicit || pcbddc->dbg_flag)) { 15861dd7afcfSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 15871dd7afcfSStefano Zampini } else { 1588a3df083aSStefano Zampini pcis->reusesubmatrices = PETSC_TRUE; 15891dd7afcfSStefano Zampini } 1590a3df083aSStefano Zampini } else { 15919f47a83aSStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 1592674ae819SStefano Zampini } 1593a3df083aSStefano Zampini } 159427b6a85dSStefano Zampini 1595c703fcc7SStefano Zampini /* propagate relevant information -> TODO remove*/ 15964f1b2e48SStefano Zampini #if !defined(PETSC_USE_COMPLEX) /* workaround for reals */ 15973301b35fSStefano Zampini if (matis->A->symmetric_set) { 15983301b35fSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 15993301b35fSStefano Zampini } 1600e496cd5dSStefano Zampini #endif 160106a4e24aSStefano Zampini if (matis->A->symmetric_set) { 160206a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SYMMETRIC,matis->A->symmetric);CHKERRQ(ierr); 160306a4e24aSStefano Zampini } 160406a4e24aSStefano Zampini if (matis->A->spd_set) { 160506a4e24aSStefano Zampini ierr = MatSetOption(pcbddc->local_mat,MAT_SPD,matis->A->spd);CHKERRQ(ierr); 160606a4e24aSStefano Zampini } 1607e496cd5dSStefano Zampini 16085e8657edSStefano Zampini /* Set up all the "iterative substructuring" common block without computing solvers */ 16095e8657edSStefano Zampini { 16105e8657edSStefano Zampini Mat temp_mat; 16115e8657edSStefano Zampini 16125e8657edSStefano Zampini temp_mat = matis->A; 16135e8657edSStefano Zampini matis->A = pcbddc->local_mat; 16145e8657edSStefano Zampini ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 16155e8657edSStefano Zampini pcbddc->local_mat = matis->A; 16165e8657edSStefano Zampini matis->A = temp_mat; 16175e8657edSStefano Zampini } 1618684f6988SStefano Zampini 161981d14e9dSStefano Zampini /* Analyze interface */ 1620c703fcc7SStefano Zampini if (pcbddc->recompute_topography) { 1621674ae819SStefano Zampini ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 16228de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1623345ecf6cSStefano Zampini if (pcbddc->adaptive_selection && !pcbddc->use_deluxe_scaling && !pcbddc->mat_graph->twodim) { 1624345ecf6cSStefano Zampini SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot compute the adaptive primal space for a problem with 3D edges without deluxe scaling"); 1625345ecf6cSStefano Zampini } 1626a198735bSStefano Zampini if (pcbddc->compute_nonetflux) { 1627669cc0f4SStefano Zampini MatNullSpace nnfnnsp; 1628669cc0f4SStefano Zampini 16298ae0ca82SStefano Zampini ierr = PCBDDCComputeNoNetFlux(pc->pmat,pcbddc->divudotp,pcbddc->divudotp_trans,pcbddc->divudotp_vl2l,pcbddc->mat_graph,&nnfnnsp);CHKERRQ(ierr); 163071582508SStefano Zampini /* TODO what if a nearnullspace is already attached? */ 1631669cc0f4SStefano Zampini ierr = MatSetNearNullSpace(pc->pmat,nnfnnsp);CHKERRQ(ierr); 1632669cc0f4SStefano Zampini ierr = MatNullSpaceDestroy(&nnfnnsp);CHKERRQ(ierr); 1633669cc0f4SStefano Zampini } 1634674ae819SStefano Zampini } 1635fb8d54d4SStefano Zampini 16365408967cSStefano Zampini /* check existence of a divergence free extension, i.e. 16375408967cSStefano Zampini b(v_I,p_0) = 0 for all v_I (raise error if not). 16385408967cSStefano Zampini Also, check that PCBDDCBenignGetOrSetP0 works */ 16395408967cSStefano Zampini #if defined(PETSC_USE_DEBUG) 164009f581a4SStefano Zampini if (pcbddc->benign_saddle_point) { 16415408967cSStefano Zampini ierr = PCBDDCBenignCheck(pc,zerodiag);CHKERRQ(ierr); 164209f581a4SStefano Zampini } 16435408967cSStefano Zampini #endif 16444f1b2e48SStefano Zampini ierr = ISDestroy(&zerodiag);CHKERRQ(ierr); 164506f24817SStefano Zampini 1646b96c3477SStefano Zampini /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1647c703fcc7SStefano Zampini if (computesubschurs && pcbddc->recompute_topography) { 164808122e43SStefano Zampini ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1649b1b3d7a2SStefano Zampini } 16509d54b7f4SStefano Zampini /* SetUp Scaling operator (scaling matrices could be needed in SubSchursSetUp)*/ 16519d54b7f4SStefano Zampini if (!pcbddc->use_deluxe_scaling) { 16529d54b7f4SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 16539d54b7f4SStefano Zampini } 1654c703fcc7SStefano Zampini 1655c703fcc7SStefano Zampini /* finish setup solvers and do adaptive selection of constraints */ 1656b334f244SStefano Zampini sub_schurs = pcbddc->sub_schurs; 1657b334f244SStefano Zampini if (sub_schurs && sub_schurs->schur_explicit) { 16582070dbb6SStefano Zampini if (computesubschurs) { 165908122e43SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 16602070dbb6SStefano Zampini } 1661d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1662d5574798SStefano Zampini } else { 1663d5574798SStefano Zampini ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 16642070dbb6SStefano Zampini if (computesubschurs) { 1665d5574798SStefano Zampini ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1666d5574798SStefano Zampini } 16672070dbb6SStefano Zampini } 166808122e43SStefano Zampini if (pcbddc->adaptive_selection) { 166908122e43SStefano Zampini ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 16708de1fae6SStefano Zampini computeconstraintsmatrix = PETSC_TRUE; 1671b7eb3628SStefano Zampini } 1672684f6988SStefano Zampini 1673f4ddd8eeSStefano Zampini /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1674fb8d54d4SStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1675f4ddd8eeSStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1676f4ddd8eeSStefano Zampini if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1677f4ddd8eeSStefano Zampini if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1678f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1679f4ddd8eeSStefano Zampini } else { 1680f4ddd8eeSStefano Zampini /* determine if the two nullspaces are different (should be lightweight) */ 1681f4ddd8eeSStefano Zampini if (nearnullspace != pcbddc->onearnullspace) { 1682f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1683165b64e2SStefano Zampini } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1684f4ddd8eeSStefano Zampini PetscInt i; 1685165b64e2SStefano Zampini const Vec *nearnullvecs; 1686165b64e2SStefano Zampini PetscObjectState state; 1687165b64e2SStefano Zampini PetscInt nnsp_size; 1688165b64e2SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1689f4ddd8eeSStefano Zampini for (i=0;i<nnsp_size;i++) { 1690f4ddd8eeSStefano Zampini ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1691165b64e2SStefano Zampini if (pcbddc->onearnullvecs_state[i] != state) { 1692f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1693f4ddd8eeSStefano Zampini break; 1694f4ddd8eeSStefano Zampini } 1695f4ddd8eeSStefano Zampini } 1696f4ddd8eeSStefano Zampini } 1697f4ddd8eeSStefano Zampini } 1698f4ddd8eeSStefano Zampini } else { 1699f4ddd8eeSStefano Zampini if (!nearnullspace) { /* both nearnullspaces are null */ 1700f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_FALSE; 1701f4ddd8eeSStefano Zampini } else { /* nearnullspace attached later */ 1702f4ddd8eeSStefano Zampini new_nearnullspace_provided = PETSC_TRUE; 1703f4ddd8eeSStefano Zampini } 1704f4ddd8eeSStefano Zampini } 1705f4ddd8eeSStefano Zampini 1706f4ddd8eeSStefano Zampini /* Setup constraints and related work vectors */ 1707727cdba6SStefano Zampini /* reset primal space flags */ 1708f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 1709727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 17108de1fae6SStefano Zampini if (computeconstraintsmatrix || new_nearnullspace_provided) { 1711727cdba6SStefano Zampini /* It also sets the primal space flags */ 1712674ae819SStefano Zampini ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1713e7b262bdSStefano Zampini /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1714f4ddd8eeSStefano Zampini ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 17153975b054SStefano Zampini } 17165e8657edSStefano Zampini 17175e8657edSStefano Zampini if (pcbddc->use_change_of_basis) { 17185e8657edSStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 17195e8657edSStefano Zampini 17205e8657edSStefano Zampini ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 17214f1b2e48SStefano Zampini if (pcbddc->benign_change) { 17221dd7afcfSStefano Zampini ierr = MatDestroy(&pcbddc->benign_B0);CHKERRQ(ierr); 1723c263805aSStefano Zampini /* pop B0 from pcbddc->local_mat */ 1724c263805aSStefano Zampini ierr = PCBDDCBenignPopOrPushB0(pc,PETSC_TRUE);CHKERRQ(ierr); 1725c263805aSStefano Zampini } 17265e8657edSStefano Zampini /* get submatrices */ 17275e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 17285e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 17295e8657edSStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 17305e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 17315e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 17325e8657edSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 17333975b054SStefano Zampini /* set flag in pcis to not reuse submatrices during PCISCreate */ 17343975b054SStefano Zampini pcis->reusesubmatrices = PETSC_FALSE; 17359c6a02ceSStefano Zampini } else if (!pcbddc->user_ChangeOfBasisMatrix && !pcbddc->benign_change) { 1736b96c3477SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 17375e8657edSStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 17385e8657edSStefano Zampini pcbddc->local_mat = matis->A; 17395e8657edSStefano Zampini } 1740b96c3477SStefano Zampini /* SetUp coarse and local Neumann solvers */ 174199cc7994SStefano Zampini ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1742b96c3477SStefano Zampini /* SetUp Scaling operator */ 17439d54b7f4SStefano Zampini if (pcbddc->use_deluxe_scaling) { 1744674ae819SStefano Zampini ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 17450c7d97c5SJed Brown } 1746c703fcc7SStefano Zampini 17471dd7afcfSStefano Zampini /* mark topography as done */ 174856282151SStefano Zampini pcbddc->recompute_topography = PETSC_FALSE; 17490369aaf7SStefano Zampini 17501dd7afcfSStefano Zampini /* wrap pcis->A_IB and pcis->A_BI if we did not change explicitly the variables on the pressures */ 17511dd7afcfSStefano Zampini ierr = PCBDDCBenignShellMat(pc,PETSC_FALSE);CHKERRQ(ierr); 17521dd7afcfSStefano Zampini 175358a03d70SStefano Zampini if (pcbddc->dbg_flag) { 175458a03d70SStefano Zampini ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 17552b510759SStefano Zampini } 17560c7d97c5SJed Brown PetscFunctionReturn(0); 17570c7d97c5SJed Brown } 17580c7d97c5SJed Brown 17590c7d97c5SJed Brown /* 176050efa1b5SStefano Zampini PCApply_BDDC - Applies the BDDC operator to a vector. 17610c7d97c5SJed Brown 17620c7d97c5SJed Brown Input Parameters: 17630f202f7eSStefano Zampini + pc - the preconditioner context 17640f202f7eSStefano Zampini - r - input vector (global) 17650c7d97c5SJed Brown 17660c7d97c5SJed Brown Output Parameter: 17670c7d97c5SJed Brown . z - output vector (global) 17680c7d97c5SJed Brown 17690c7d97c5SJed Brown Application Interface Routine: PCApply() 17700c7d97c5SJed Brown */ 17710c7d97c5SJed Brown #undef __FUNCT__ 17720c7d97c5SJed Brown #define __FUNCT__ "PCApply_BDDC" 177353cdbc3dSStefano Zampini PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 17740c7d97c5SJed Brown { 17750c7d97c5SJed Brown PC_IS *pcis = (PC_IS*)(pc->data); 17760c7d97c5SJed Brown PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1777b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 17780c7d97c5SJed Brown PetscErrorCode ierr; 17793b03a366Sstefano_zampini const PetscScalar one = 1.0; 17803b03a366Sstefano_zampini const PetscScalar m_one = -1.0; 17812617d88aSStefano Zampini const PetscScalar zero = 0.0; 17820c7d97c5SJed Brown 17830c7d97c5SJed Brown /* This code is similar to that provided in nn.c for PCNN 17840c7d97c5SJed Brown NN interface preconditioner changed to BDDC 1785b097fa66SStefano Zampini Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 17860c7d97c5SJed Brown 17870c7d97c5SJed Brown PetscFunctionBegin; 17881dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 17891dd7afcfSStefano Zampini Vec swap; 179027b6a85dSStefano Zampini 179127b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 17921dd7afcfSStefano Zampini swap = pcbddc->work_change; 17931dd7afcfSStefano Zampini pcbddc->work_change = r; 17941dd7afcfSStefano Zampini r = swap; 17951dd7afcfSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 17968ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 17971dd7afcfSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 17981dd7afcfSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 17991dd7afcfSStefano Zampini } 18001dd7afcfSStefano Zampini } 180127b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1802015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1803efc2fbd9SStefano Zampini } 18048ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1805b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 18060c7d97c5SJed Brown /* First Dirichlet solve */ 18070c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18080c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18090c7d97c5SJed Brown /* 18100c7d97c5SJed Brown Assembling right hand side for BDDC operator 1811b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1812674ae819SStefano Zampini - pcis->vec1_B the interface part of the global vector z 18130c7d97c5SJed Brown */ 1814b097fa66SStefano Zampini if (n_D) { 1815b097fa66SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 18160c7d97c5SJed Brown ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 181716909a7fSStefano Zampini if (pcbddc->switch_static) { 181816909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 181916909a7fSStefano Zampini 182016909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 182116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 182216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 182316909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 182416909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182516909a7fSStefano Zampini } else { 182616909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182716909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 182816909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 182916909a7fSStefano Zampini } 183016909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183116909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 183416909a7fSStefano Zampini } else { 1835b097fa66SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 183616909a7fSStefano Zampini } 1837b097fa66SStefano Zampini } else { 1838b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1839b097fa66SStefano Zampini } 18400c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18410c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1842674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1843b76ba322SStefano Zampini } else { 18444fee134fSStefano Zampini if (!pcbddc->benign_apply_coarse_only) { 1845674ae819SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1846b76ba322SStefano Zampini } 18474fee134fSStefano Zampini } 1848b76ba322SStefano Zampini 18492617d88aSStefano Zampini /* Apply interface preconditioner 18502617d88aSStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1851dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 18522617d88aSStefano Zampini 1853674ae819SStefano Zampini /* Apply transpose of partition of unity operator */ 1854674ae819SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 18550c7d97c5SJed Brown 18563b03a366Sstefano_zampini /* Second Dirichlet solve and assembling of output */ 18570c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 18580c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1859b097fa66SStefano Zampini if (n_B) { 186016909a7fSStefano Zampini if (pcbddc->switch_static) { 186116909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 186216909a7fSStefano Zampini 186316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186516909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186616909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 186716909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 186816909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 186916909a7fSStefano Zampini } else { 187016909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 187116909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 187216909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 187316909a7fSStefano Zampini } 187416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 187616909a7fSStefano Zampini } else { 18770c7d97c5SJed Brown ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 187816909a7fSStefano Zampini } 187916909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 188016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 188116909a7fSStefano Zampini 188216909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 188316909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 188416909a7fSStefano Zampini } else { 188516909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 188616909a7fSStefano Zampini ierr = MatMult(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 188716909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 188816909a7fSStefano Zampini } 1889b097fa66SStefano Zampini } 1890df187020SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1891efc2fbd9SStefano Zampini 18928ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1893b097fa66SStefano Zampini if (pcbddc->switch_static) { 1894b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1895b097fa66SStefano Zampini } else { 1896b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1897b097fa66SStefano Zampini } 18980c7d97c5SJed Brown ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 18990c7d97c5SJed Brown ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1900b097fa66SStefano Zampini } else { 1901b097fa66SStefano Zampini if (pcbddc->switch_static) { 1902b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1903b097fa66SStefano Zampini } else { 1904b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1905b097fa66SStefano Zampini } 1906b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1907b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1908b097fa66SStefano Zampini } 190927b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 19101dd7afcfSStefano Zampini if (pcbddc->benign_apply_coarse_only) { 19111dd7afcfSStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 19121dd7afcfSStefano Zampini } 1913015636ebSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 1914efc2fbd9SStefano Zampini } 19151f4df5f7SStefano Zampini 19161dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 1917f913dca9SStefano Zampini pcbddc->work_change = r; 19181dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 19191dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 19201dd7afcfSStefano Zampini } 19210c7d97c5SJed Brown PetscFunctionReturn(0); 19220c7d97c5SJed Brown } 192350efa1b5SStefano Zampini 192450efa1b5SStefano Zampini /* 192550efa1b5SStefano Zampini PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 192650efa1b5SStefano Zampini 192750efa1b5SStefano Zampini Input Parameters: 19280f202f7eSStefano Zampini + pc - the preconditioner context 19290f202f7eSStefano Zampini - r - input vector (global) 193050efa1b5SStefano Zampini 193150efa1b5SStefano Zampini Output Parameter: 193250efa1b5SStefano Zampini . z - output vector (global) 193350efa1b5SStefano Zampini 193450efa1b5SStefano Zampini Application Interface Routine: PCApplyTranspose() 193550efa1b5SStefano Zampini */ 193650efa1b5SStefano Zampini #undef __FUNCT__ 193750efa1b5SStefano Zampini #define __FUNCT__ "PCApplyTranspose_BDDC" 193850efa1b5SStefano Zampini PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 193950efa1b5SStefano Zampini { 194050efa1b5SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 194150efa1b5SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1942b097fa66SStefano Zampini PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 194350efa1b5SStefano Zampini PetscErrorCode ierr; 194450efa1b5SStefano Zampini const PetscScalar one = 1.0; 194550efa1b5SStefano Zampini const PetscScalar m_one = -1.0; 194650efa1b5SStefano Zampini const PetscScalar zero = 0.0; 194750efa1b5SStefano Zampini 194850efa1b5SStefano Zampini PetscFunctionBegin; 19491dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 19501dd7afcfSStefano Zampini Vec swap; 195127b6a85dSStefano Zampini 195227b6a85dSStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,r,pcbddc->work_change);CHKERRQ(ierr); 19531dd7afcfSStefano Zampini swap = pcbddc->work_change; 19541dd7afcfSStefano Zampini pcbddc->work_change = r; 19551dd7afcfSStefano Zampini r = swap; 195627b6a85dSStefano Zampini /* save rhs so that we don't need to apply the change of basis for the exact dirichlet trick in PreSolve */ 19578ae0ca82SStefano Zampini if (pcbddc->benign_apply_coarse_only && pcbddc->exact_dirichlet_trick_app && pcbddc->change_interior) { 195827b6a85dSStefano Zampini ierr = VecCopy(r,pcis->vec1_global);CHKERRQ(ierr); 195927b6a85dSStefano Zampini ierr = VecLockPush(pcis->vec1_global);CHKERRQ(ierr); 19601dd7afcfSStefano Zampini } 196127b6a85dSStefano Zampini } 196227b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* get p0 from r */ 1963537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,r,PETSC_TRUE);CHKERRQ(ierr); 1964537c1cdfSStefano Zampini } 19658ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 1966b097fa66SStefano Zampini ierr = VecCopy(r,z);CHKERRQ(ierr); 196750efa1b5SStefano Zampini /* First Dirichlet solve */ 196850efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 196950efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 197050efa1b5SStefano Zampini /* 197150efa1b5SStefano Zampini Assembling right hand side for BDDC operator 1972b097fa66SStefano Zampini - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 197350efa1b5SStefano Zampini - pcis->vec1_B the interface part of the global vector z 197450efa1b5SStefano Zampini */ 1975b097fa66SStefano Zampini if (n_D) { 1976b097fa66SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 197750efa1b5SStefano Zampini ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 197816909a7fSStefano Zampini if (pcbddc->switch_static) { 197916909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 198016909a7fSStefano Zampini 198116909a7fSStefano Zampini ierr = VecSet(pcis->vec1_N,0.);CHKERRQ(ierr); 198216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 198316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 198416909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 198516909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 198616909a7fSStefano Zampini } else { 198716909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 198816909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 198916909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 199016909a7fSStefano Zampini } 199116909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199216909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec1_D,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec2_N,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 199516909a7fSStefano Zampini } else { 1996b097fa66SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 199716909a7fSStefano Zampini } 1998b097fa66SStefano Zampini } else { 1999b097fa66SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 2000b097fa66SStefano Zampini } 200150efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 200250efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 200350efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 200450efa1b5SStefano Zampini } else { 200550efa1b5SStefano Zampini ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 200650efa1b5SStefano Zampini } 200750efa1b5SStefano Zampini 200850efa1b5SStefano Zampini /* Apply interface preconditioner 200950efa1b5SStefano Zampini input/output vecs: pcis->vec1_B and pcis->vec1_D */ 2010dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 201150efa1b5SStefano Zampini 201250efa1b5SStefano Zampini /* Apply transpose of partition of unity operator */ 201350efa1b5SStefano Zampini ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 201450efa1b5SStefano Zampini 201550efa1b5SStefano Zampini /* Second Dirichlet solve and assembling of output */ 201650efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 201750efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2018b097fa66SStefano Zampini if (n_B) { 201916909a7fSStefano Zampini if (pcbddc->switch_static) { 202016909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 202116909a7fSStefano Zampini 202216909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202316909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec1_D,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202416909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202516909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_B,pcis->vec1_B,pcis->vec1_N,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 202616909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 202716909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 202816909a7fSStefano Zampini } else { 202916909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 203016909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec2_N,pcis->vec1_N);CHKERRQ(ierr); 203116909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 203216909a7fSStefano Zampini } 203316909a7fSStefano Zampini ierr = VecScatterBegin(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 203416909a7fSStefano Zampini ierr = VecScatterEnd(pcis->N_to_D,pcis->vec2_N,pcis->vec3_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 203516909a7fSStefano Zampini } else { 203650efa1b5SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 203716909a7fSStefano Zampini } 203816909a7fSStefano Zampini } else if (pcbddc->switch_static) { /* n_B is zero */ 203916909a7fSStefano Zampini Mat_IS *matis = (Mat_IS*)(pc->mat->data); 204016909a7fSStefano Zampini 204116909a7fSStefano Zampini if (!pcbddc->switch_static_change) { 204216909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 204316909a7fSStefano Zampini } else { 204416909a7fSStefano Zampini ierr = MatMult(pcbddc->switch_static_change,pcis->vec1_D,pcis->vec1_N);CHKERRQ(ierr); 204516909a7fSStefano Zampini ierr = MatMultTranspose(matis->A,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 204616909a7fSStefano Zampini ierr = MatMultTranspose(pcbddc->switch_static_change,pcis->vec2_N,pcis->vec3_D);CHKERRQ(ierr); 204716909a7fSStefano Zampini } 2048b097fa66SStefano Zampini } 2049b0147a47SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 20508ae0ca82SStefano Zampini if (!pcbddc->exact_dirichlet_trick_app && !pcbddc->benign_apply_coarse_only) { 2051b097fa66SStefano Zampini if (pcbddc->switch_static) { 2052b097fa66SStefano Zampini ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 2053b097fa66SStefano Zampini } else { 2054b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 2055b097fa66SStefano Zampini } 205650efa1b5SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 205750efa1b5SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2058b097fa66SStefano Zampini } else { 2059b097fa66SStefano Zampini if (pcbddc->switch_static) { 2060b097fa66SStefano Zampini ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 2061b097fa66SStefano Zampini } else { 2062b097fa66SStefano Zampini ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 2063b097fa66SStefano Zampini } 2064b097fa66SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2065b097fa66SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2066b097fa66SStefano Zampini } 206727b6a85dSStefano Zampini if (pcbddc->benign_have_null) { /* set p0 (computed in PCBDDCApplyInterface) */ 2068537c1cdfSStefano Zampini ierr = PCBDDCBenignGetOrSetP0(pc,z,PETSC_FALSE);CHKERRQ(ierr); 2069537c1cdfSStefano Zampini } 20701dd7afcfSStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2071f913dca9SStefano Zampini pcbddc->work_change = r; 20721dd7afcfSStefano Zampini ierr = VecCopy(z,pcbddc->work_change);CHKERRQ(ierr); 20731dd7afcfSStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,pcbddc->work_change,z);CHKERRQ(ierr); 20741dd7afcfSStefano Zampini } 207550efa1b5SStefano Zampini PetscFunctionReturn(0); 207650efa1b5SStefano Zampini } 2077674ae819SStefano Zampini 2078da1bb401SStefano Zampini #undef __FUNCT__ 2079da1bb401SStefano Zampini #define __FUNCT__ "PCDestroy_BDDC" 2080da1bb401SStefano Zampini PetscErrorCode PCDestroy_BDDC(PC pc) 2081da1bb401SStefano Zampini { 2082da1bb401SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 2083da1bb401SStefano Zampini PetscErrorCode ierr; 2084da1bb401SStefano Zampini 2085da1bb401SStefano Zampini PetscFunctionBegin; 2086674ae819SStefano Zampini /* free BDDC custom data */ 2087674ae819SStefano Zampini ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 2088674ae819SStefano Zampini /* destroy objects related to topography */ 2089674ae819SStefano Zampini ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 2090674ae819SStefano Zampini /* free allocated graph structure */ 2091da1bb401SStefano Zampini ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 2092b96c3477SStefano Zampini /* free allocated sub schurs structure */ 2093b96c3477SStefano Zampini ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 209434a97f8cSStefano Zampini /* destroy objects for scaling operator */ 2095674ae819SStefano Zampini ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 209634a97f8cSStefano Zampini ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 2097674ae819SStefano Zampini /* free solvers stuff */ 2098674ae819SStefano Zampini ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 209962a6ff1dSStefano Zampini /* free global vectors needed in presolve */ 210062a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 210162a6ff1dSStefano Zampini ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 21021dd7afcfSStefano Zampini /* free data created by PCIS */ 21031dd7afcfSStefano Zampini ierr = PCISDestroy(pc);CHKERRQ(ierr); 21043425bc38SStefano Zampini /* remove functions */ 2105a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",NULL);CHKERRQ(ierr); 2106a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",NULL);CHKERRQ(ierr); 2107906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 2108674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 210930368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",NULL);CHKERRQ(ierr); 2110bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 21112b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 2112b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 21132b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 2114bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 211582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2116bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 211782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2118bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 211982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 2120bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 2121785d1243SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 2122bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 212363602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 2124bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 2125bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 2126bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 2127bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 2128a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",NULL);CHKERRQ(ierr); 2129674ae819SStefano Zampini /* Free the private data structure */ 2130674ae819SStefano Zampini ierr = PetscFree(pc->data);CHKERRQ(ierr); 2131da1bb401SStefano Zampini PetscFunctionReturn(0); 2132da1bb401SStefano Zampini } 21331e6b0712SBarry Smith 21343425bc38SStefano Zampini #undef __FUNCT__ 2135a06fd7f2SStefano Zampini #define __FUNCT__ "PCPreSolveChangeRHS_BDDC" 2136a06fd7f2SStefano Zampini static PetscErrorCode PCPreSolveChangeRHS_BDDC(PC pc, PetscBool* change) 2137a06fd7f2SStefano Zampini { 2138a06fd7f2SStefano Zampini PetscFunctionBegin; 2139a06fd7f2SStefano Zampini *change = PETSC_TRUE; 2140a06fd7f2SStefano Zampini PetscFunctionReturn(0); 2141a06fd7f2SStefano Zampini } 2142a06fd7f2SStefano Zampini 2143a06fd7f2SStefano Zampini #undef __FUNCT__ 21443425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 21453425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 21463425bc38SStefano Zampini { 2147674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 2148266e20e9SStefano Zampini Vec work; 21493425bc38SStefano Zampini PC_IS* pcis; 21503425bc38SStefano Zampini PC_BDDC* pcbddc; 21513425bc38SStefano Zampini PetscErrorCode ierr; 21520c7d97c5SJed Brown 21533425bc38SStefano Zampini PetscFunctionBegin; 21543425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 21553425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 21563425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 21573425bc38SStefano Zampini 2158c08af4c6SStefano Zampini /* 2159c08af4c6SStefano Zampini change of basis for physical rhs if needed 2160c08af4c6SStefano Zampini It also changes the rhs in case of dirichlet boundaries 2161c08af4c6SStefano Zampini */ 21623738a8e6SStefano Zampini if (!pcbddc->original_rhs) { 21633738a8e6SStefano Zampini ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 21643738a8e6SStefano Zampini } 21653738a8e6SStefano Zampini ierr = VecCopy(standard_rhs,pcbddc->original_rhs);CHKERRQ(ierr); 21663738a8e6SStefano Zampini ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,pcbddc->original_rhs,NULL);CHKERRQ(ierr); 2167fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 21683738a8e6SStefano Zampini ierr = MatMultTranspose(pcbddc->ChangeOfBasisMatrix,pcbddc->original_rhs,pcbddc->work_change);CHKERRQ(ierr); 21693738a8e6SStefano Zampini work = pcbddc->work_change; 2170fc17d649SStefano Zampini } else { 21713738a8e6SStefano Zampini work = pcbddc->original_rhs; 2172fc17d649SStefano Zampini } 21733425bc38SStefano Zampini /* store vectors for computation of fetidp final solution */ 2174266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2175266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,work,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2176fb223d50SStefano Zampini /* scale rhs since it should be unassembled */ 2177fb223d50SStefano Zampini /* TODO use counter scaling? (also below) */ 2178266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2179266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2180674ae819SStefano Zampini /* Apply partition of unity */ 21813425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 2182266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 21838eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 21843425bc38SStefano Zampini /* compute partially subassembled Schur complement right-hand side */ 21853425bc38SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 21863425bc38SStefano Zampini ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 21873425bc38SStefano Zampini ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 2188266e20e9SStefano Zampini ierr = VecSet(work,0.0);CHKERRQ(ierr); 2189266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2190266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,work,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2191266e20e9SStefano Zampini /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,work,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 2192266e20e9SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2193266e20e9SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,work,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 21943425bc38SStefano Zampini ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 21953425bc38SStefano Zampini } 21963425bc38SStefano Zampini /* BDDC rhs */ 21973425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 21988eeda7d8SStefano Zampini if (pcbddc->switch_static) { 21993425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22003425bc38SStefano Zampini } 2201fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 22023425bc38SStefano Zampini /* apply BDDC */ 2203dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 2204266e20e9SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 22053425bc38SStefano Zampini /* Application of B_delta and assembling of rhs for fetidp fluxes */ 22063425bc38SStefano Zampini ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 22073425bc38SStefano Zampini ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 22083425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22093425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 22103425bc38SStefano Zampini PetscFunctionReturn(0); 22113425bc38SStefano Zampini } 22121e6b0712SBarry Smith 22133425bc38SStefano Zampini #undef __FUNCT__ 22143425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 22153425bc38SStefano Zampini /*@ 22160f202f7eSStefano Zampini PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 22173425bc38SStefano Zampini 22183425bc38SStefano Zampini Collective 22193425bc38SStefano Zampini 22203425bc38SStefano Zampini Input Parameters: 22210f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 22220f202f7eSStefano Zampini - standard_rhs - the right-hand side of the original linear system 22233425bc38SStefano Zampini 22243425bc38SStefano Zampini Output Parameters: 22250f202f7eSStefano Zampini . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 22263425bc38SStefano Zampini 22273425bc38SStefano Zampini Level: developer 22283425bc38SStefano Zampini 22293425bc38SStefano Zampini Notes: 22303425bc38SStefano Zampini 22310f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 22323425bc38SStefano Zampini @*/ 22333425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 22343425bc38SStefano Zampini { 2235674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22363425bc38SStefano Zampini PetscErrorCode ierr; 22373425bc38SStefano Zampini 22383425bc38SStefano Zampini PetscFunctionBegin; 2239266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2240266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_rhs,VEC_CLASSID,2); 2241266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_rhs,VEC_CLASSID,3); 22423425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2243163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 22443425bc38SStefano Zampini PetscFunctionReturn(0); 22453425bc38SStefano Zampini } 22461e6b0712SBarry Smith 22473425bc38SStefano Zampini #undef __FUNCT__ 22483425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 22493425bc38SStefano Zampini static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 22503425bc38SStefano Zampini { 2251674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 22523425bc38SStefano Zampini PC_IS* pcis; 22533425bc38SStefano Zampini PC_BDDC* pcbddc; 22543425bc38SStefano Zampini PetscErrorCode ierr; 22553425bc38SStefano Zampini 22563425bc38SStefano Zampini PetscFunctionBegin; 22573425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 22583425bc38SStefano Zampini pcis = (PC_IS*)mat_ctx->pc->data; 22593425bc38SStefano Zampini pcbddc = (PC_BDDC*)mat_ctx->pc->data; 22603425bc38SStefano Zampini 22613425bc38SStefano Zampini /* apply B_delta^T */ 22623425bc38SStefano Zampini ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22633425bc38SStefano Zampini ierr = VecScatterEnd(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22643425bc38SStefano Zampini ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 22653425bc38SStefano Zampini /* compute rhs for BDDC application */ 22663425bc38SStefano Zampini ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 22678eeda7d8SStefano Zampini if (pcbddc->switch_static) { 22683425bc38SStefano Zampini ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 22693425bc38SStefano Zampini } 2270fc17d649SStefano Zampini ierr = PetscMemzero(pcbddc->benign_p0,pcbddc->benign_n*sizeof(PetscScalar));CHKERRQ(ierr); 22713425bc38SStefano Zampini /* apply BDDC */ 2272dc359a40SStefano Zampini ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 22733425bc38SStefano Zampini /* put values into standard global vector */ 22743425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22753425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22768eeda7d8SStefano Zampini if (!pcbddc->switch_static) { 22773425bc38SStefano Zampini /* compute values into the interior if solved for the partially subassembled Schur complement */ 22783425bc38SStefano Zampini ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 227900f6b531SStefano Zampini ierr = VecAYPX(pcis->vec1_D,-1.0,mat_ctx->temp_solution_D);CHKERRQ(ierr); 228000f6b531SStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); 22813425bc38SStefano Zampini } 22823425bc38SStefano Zampini ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 22833425bc38SStefano Zampini ierr = VecScatterEnd(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2284266e20e9SStefano Zampini /* add p0 solution to final solution */ 2285266e20e9SStefano Zampini ierr = PCBDDCBenignGetOrSetP0(mat_ctx->pc,standard_sol,PETSC_FALSE);CHKERRQ(ierr); 2286fc17d649SStefano Zampini if (pcbddc->ChangeOfBasisMatrix) { 2287fc17d649SStefano Zampini Vec v2; 2288fc17d649SStefano Zampini ierr = VecDuplicate(standard_sol,&v2);CHKERRQ(ierr); 2289fc17d649SStefano Zampini ierr = MatMult(pcbddc->ChangeOfBasisMatrix,standard_sol,v2);CHKERRQ(ierr); 2290fc17d649SStefano Zampini ierr = VecCopy(v2,standard_sol);CHKERRQ(ierr); 2291fc17d649SStefano Zampini ierr = VecDestroy(&v2);CHKERRQ(ierr); 2292fc17d649SStefano Zampini } 22933308cffdSStefano Zampini ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 22943425bc38SStefano Zampini PetscFunctionReturn(0); 22953425bc38SStefano Zampini } 22961e6b0712SBarry Smith 22973425bc38SStefano Zampini #undef __FUNCT__ 22983425bc38SStefano Zampini #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 22993425bc38SStefano Zampini /*@ 23000f202f7eSStefano Zampini PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 23013425bc38SStefano Zampini 23023425bc38SStefano Zampini Collective 23033425bc38SStefano Zampini 23043425bc38SStefano Zampini Input Parameters: 23050f202f7eSStefano Zampini + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 23060f202f7eSStefano Zampini - fetidp_flux_sol - the solution of the FETI-DP linear system 23073425bc38SStefano Zampini 23083425bc38SStefano Zampini Output Parameters: 23090f202f7eSStefano Zampini . standard_sol - the solution defined on the physical domain 23103425bc38SStefano Zampini 23113425bc38SStefano Zampini Level: developer 23123425bc38SStefano Zampini 23133425bc38SStefano Zampini Notes: 23143425bc38SStefano Zampini 23150f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 23163425bc38SStefano Zampini @*/ 23173425bc38SStefano Zampini PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 23183425bc38SStefano Zampini { 2319674ae819SStefano Zampini FETIDPMat_ctx mat_ctx; 23203425bc38SStefano Zampini PetscErrorCode ierr; 23213425bc38SStefano Zampini 23223425bc38SStefano Zampini PetscFunctionBegin; 2323266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_mat,MAT_CLASSID,1); 2324266e20e9SStefano Zampini PetscValidHeaderSpecific(fetidp_flux_sol,VEC_CLASSID,2); 2325266e20e9SStefano Zampini PetscValidHeaderSpecific(standard_sol,VEC_CLASSID,3); 23263425bc38SStefano Zampini ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 2327163d334eSBarry Smith ierr = PetscUseMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 23283425bc38SStefano Zampini PetscFunctionReturn(0); 23293425bc38SStefano Zampini } 23301e6b0712SBarry Smith 2331f23aa3ddSBarry Smith extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 2332edf7251bSStefano Zampini extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 2333f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 2334f23aa3ddSBarry Smith extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 2335edf7251bSStefano Zampini extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 2336f23aa3ddSBarry Smith extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 2337674ae819SStefano Zampini 23383425bc38SStefano Zampini #undef __FUNCT__ 23393425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 23401720468bSStefano Zampini static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, PetscBool fully_redundant, Mat *fetidp_mat, PC *fetidp_pc) 23413425bc38SStefano Zampini { 2342674ae819SStefano Zampini 2343674ae819SStefano Zampini FETIDPMat_ctx fetidpmat_ctx; 23443425bc38SStefano Zampini Mat newmat; 2345674ae819SStefano Zampini FETIDPPC_ctx fetidppc_ctx; 23463425bc38SStefano Zampini PC newpc; 2347ce94432eSBarry Smith MPI_Comm comm; 23483425bc38SStefano Zampini PetscErrorCode ierr; 23493425bc38SStefano Zampini 23503425bc38SStefano Zampini PetscFunctionBegin; 2351ce94432eSBarry Smith ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 23523425bc38SStefano Zampini /* FETIDP linear matrix */ 23533425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 23541720468bSStefano Zampini fetidpmat_ctx->fully_redundant = fully_redundant; 23553425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 23563425bc38SStefano Zampini ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 23573425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 2358edf7251bSStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 23593425bc38SStefano Zampini ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 23603425bc38SStefano Zampini ierr = MatSetUp(newmat);CHKERRQ(ierr); 23613425bc38SStefano Zampini /* FETIDP preconditioner */ 23623425bc38SStefano Zampini ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 23633425bc38SStefano Zampini ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 23643425bc38SStefano Zampini ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 23653425bc38SStefano Zampini ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 23663425bc38SStefano Zampini ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 23673425bc38SStefano Zampini ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 2368edf7251bSStefano Zampini ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 23693425bc38SStefano Zampini ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 237023ee1639SBarry Smith ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 23713425bc38SStefano Zampini ierr = PCSetUp(newpc);CHKERRQ(ierr); 23723425bc38SStefano Zampini /* return pointers for objects created */ 23733425bc38SStefano Zampini *fetidp_mat=newmat; 23743425bc38SStefano Zampini *fetidp_pc=newpc; 23753425bc38SStefano Zampini PetscFunctionReturn(0); 23763425bc38SStefano Zampini } 23771e6b0712SBarry Smith 23783425bc38SStefano Zampini #undef __FUNCT__ 23793425bc38SStefano Zampini #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 23803425bc38SStefano Zampini /*@ 23810f202f7eSStefano Zampini PCBDDCCreateFETIDPOperators - Create FETI-DP operators 23823425bc38SStefano Zampini 23833425bc38SStefano Zampini Collective 23843425bc38SStefano Zampini 23853425bc38SStefano Zampini Input Parameters: 23861720468bSStefano Zampini + pc - the BDDC preconditioning context (setup should have been called before) 23871720468bSStefano Zampini - fully_redundant - true for a fully redundant set of Lagrange multipliers 238828509bceSStefano Zampini 238928509bceSStefano Zampini Output Parameters: 23900f202f7eSStefano Zampini + fetidp_mat - shell FETI-DP matrix object 23910f202f7eSStefano Zampini - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 239228509bceSStefano Zampini 23933425bc38SStefano Zampini Level: developer 23943425bc38SStefano Zampini 23953425bc38SStefano Zampini Notes: 23960f202f7eSStefano Zampini Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 23973425bc38SStefano Zampini 23980f202f7eSStefano Zampini .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 23993425bc38SStefano Zampini @*/ 24001720468bSStefano Zampini PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, PetscBool fully_redundant, Mat *fetidp_mat, PC *fetidp_pc) 24013425bc38SStefano Zampini { 24023425bc38SStefano Zampini PetscErrorCode ierr; 24033425bc38SStefano Zampini 24043425bc38SStefano Zampini PetscFunctionBegin; 24053425bc38SStefano Zampini PetscValidHeaderSpecific(pc,PC_CLASSID,1); 24063425bc38SStefano Zampini if (pc->setupcalled) { 24071720468bSStefano Zampini ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,PetscBool,Mat*,PC*),(pc,fully_redundant,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 2408f23aa3ddSBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 24093425bc38SStefano Zampini PetscFunctionReturn(0); 24103425bc38SStefano Zampini } 24110c7d97c5SJed Brown /* -------------------------------------------------------------------------- */ 2412da1bb401SStefano Zampini /*MC 2413da1bb401SStefano Zampini PCBDDC - Balancing Domain Decomposition by Constraints. 24140c7d97c5SJed Brown 241528509bceSStefano Zampini An implementation of the BDDC preconditioner based on 241628509bceSStefano Zampini 241728509bceSStefano Zampini .vb 241828509bceSStefano Zampini [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 241928509bceSStefano 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 242028509bceSStefano Zampini [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 24210f202f7eSStefano 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 242228509bceSStefano Zampini .ve 242328509bceSStefano Zampini 242428509bceSStefano Zampini The matrix to be preconditioned (Pmat) must be of type MATIS. 242528509bceSStefano Zampini 24260f202f7eSStefano Zampini Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 242728509bceSStefano Zampini 242828509bceSStefano Zampini It also works with unsymmetric and indefinite problems. 242928509bceSStefano Zampini 2430b6fdb6dfSStefano 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. 2431b6fdb6dfSStefano Zampini 2432c7017625SStefano 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). 243328509bceSStefano Zampini 24340f202f7eSStefano 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() 243530368db7SStefano Zampini Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesIS() and their local counterparts. 243628509bceSStefano Zampini 24370f202f7eSStefano Zampini Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 243828509bceSStefano Zampini 24390f202f7eSStefano 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. 24400f202f7eSStefano Zampini User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 244128509bceSStefano Zampini 24420f202f7eSStefano Zampini The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 244328509bceSStefano Zampini 2444df4d28bfSStefano 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. 244528509bceSStefano Zampini 24460f202f7eSStefano 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. 24470f202f7eSStefano Zampini Deluxe scaling is not supported yet for FETI-DP. 24480f202f7eSStefano Zampini 24490f202f7eSStefano Zampini Options Database Keys (some of them, run with -h for a complete list): 24500f202f7eSStefano Zampini 24510f202f7eSStefano Zampini . -pc_bddc_use_vertices <true> - use or not vertices in primal space 24520f202f7eSStefano Zampini . -pc_bddc_use_edges <true> - use or not edges in primal space 24530f202f7eSStefano Zampini . -pc_bddc_use_faces <false> - use or not faces in primal space 24540f202f7eSStefano Zampini . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 24550f202f7eSStefano Zampini . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 24560f202f7eSStefano Zampini . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 24570f202f7eSStefano Zampini . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 245828509bceSStefano Zampini . -pc_bddc_levels <0> - maximum number of levels for multilevel 24590f202f7eSStefano 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) 24600f202f7eSStefano 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) 24610f202f7eSStefano Zampini . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 24620f202f7eSStefano 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) 2463df4d28bfSStefano 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) 246428509bceSStefano Zampini - -pc_bddc_check_level <0> - set verbosity level of debugging output 246528509bceSStefano Zampini 246628509bceSStefano Zampini Options for Dirichlet, Neumann or coarse solver can be set with 246728509bceSStefano Zampini .vb 246828509bceSStefano Zampini -pc_bddc_dirichlet_ 246928509bceSStefano Zampini -pc_bddc_neumann_ 247028509bceSStefano Zampini -pc_bddc_coarse_ 247128509bceSStefano Zampini .ve 24720f202f7eSStefano Zampini e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 247328509bceSStefano Zampini 24740f202f7eSStefano Zampini When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 247528509bceSStefano Zampini .vb 2476312be037SStefano Zampini -pc_bddc_dirichlet_lN_ 2477312be037SStefano Zampini -pc_bddc_neumann_lN_ 2478312be037SStefano Zampini -pc_bddc_coarse_lN_ 247928509bceSStefano Zampini .ve 24800f202f7eSStefano Zampini Note that level number ranges from the finest (0) to the coarsest (N). 24810f202f7eSStefano 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. 24820f202f7eSStefano Zampini .vb 24830f202f7eSStefano Zampini -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 24840f202f7eSStefano Zampini .ve 24850f202f7eSStefano 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 2486da1bb401SStefano Zampini 2487da1bb401SStefano Zampini Level: intermediate 2488da1bb401SStefano Zampini 2489b6fdb6dfSStefano Zampini Developer notes: 2490da1bb401SStefano Zampini 2491da1bb401SStefano Zampini Contributed by Stefano Zampini 2492da1bb401SStefano Zampini 2493da1bb401SStefano Zampini .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2494da1bb401SStefano Zampini M*/ 2495b2573a8aSBarry Smith 2496da1bb401SStefano Zampini #undef __FUNCT__ 2497da1bb401SStefano Zampini #define __FUNCT__ "PCCreate_BDDC" 24988cc058d9SJed Brown PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2499da1bb401SStefano Zampini { 2500da1bb401SStefano Zampini PetscErrorCode ierr; 2501da1bb401SStefano Zampini PC_BDDC *pcbddc; 2502da1bb401SStefano Zampini 2503da1bb401SStefano Zampini PetscFunctionBegin; 2504da1bb401SStefano Zampini /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2505b00a9115SJed Brown ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2506da1bb401SStefano Zampini pc->data = (void*)pcbddc; 2507da1bb401SStefano Zampini 2508da1bb401SStefano Zampini /* create PCIS data structure */ 2509da1bb401SStefano Zampini ierr = PCISCreate(pc);CHKERRQ(ierr); 2510da1bb401SStefano Zampini 251147d04d0dSStefano Zampini /* BDDC customization */ 251208a5cf49SStefano Zampini pcbddc->use_local_adj = PETSC_TRUE; 251347d04d0dSStefano Zampini pcbddc->use_vertices = PETSC_TRUE; 251447d04d0dSStefano Zampini pcbddc->use_edges = PETSC_TRUE; 251547d04d0dSStefano Zampini pcbddc->use_faces = PETSC_FALSE; 251647d04d0dSStefano Zampini pcbddc->use_change_of_basis = PETSC_FALSE; 251747d04d0dSStefano Zampini pcbddc->use_change_on_faces = PETSC_FALSE; 251847d04d0dSStefano Zampini pcbddc->switch_static = PETSC_FALSE; 2519fa434743SStefano Zampini pcbddc->use_nnsp_true = PETSC_FALSE; 2520fa434743SStefano Zampini pcbddc->use_qr_single = PETSC_FALSE; 25213301b35fSStefano Zampini pcbddc->symmetric_primal = PETSC_TRUE; 252206a4e24aSStefano Zampini pcbddc->benign_saddle_point = PETSC_FALSE; 252327b6a85dSStefano Zampini pcbddc->benign_have_null = PETSC_FALSE; 252414f95afaSStefano Zampini pcbddc->vertex_size = 1; 252547d04d0dSStefano Zampini pcbddc->dbg_flag = 0; 2526b9d89cd5SStefano Zampini /* private */ 2527727cdba6SStefano Zampini pcbddc->local_primal_size = 0; 25280e6343abSStefano Zampini pcbddc->local_primal_size_cc = 0; 25290e6343abSStefano Zampini pcbddc->local_primal_ref_node = 0; 25300e6343abSStefano Zampini pcbddc->local_primal_ref_mult = 0; 2531e9189074SStefano Zampini pcbddc->n_vertices = 0; 2532727cdba6SStefano Zampini pcbddc->primal_indices_local_idxs = 0; 2533c703fcc7SStefano Zampini pcbddc->recompute_topography = PETSC_TRUE; 253468457ee5SStefano Zampini pcbddc->coarse_size = -1; 2535f4ddd8eeSStefano Zampini pcbddc->new_primal_space = PETSC_FALSE; 2536727cdba6SStefano Zampini pcbddc->new_primal_space_local = PETSC_FALSE; 2537f4ddd8eeSStefano Zampini pcbddc->global_primal_indices = 0; 2538f4ddd8eeSStefano Zampini pcbddc->onearnullspace = 0; 2539f4ddd8eeSStefano Zampini pcbddc->onearnullvecs_state = 0; 2540674ae819SStefano Zampini pcbddc->user_primal_vertices = 0; 254130368db7SStefano Zampini pcbddc->user_primal_vertices_local = 0; 25423972b0daSStefano Zampini pcbddc->temp_solution = 0; 2543534831adSStefano Zampini pcbddc->original_rhs = 0; 2544534831adSStefano Zampini pcbddc->local_mat = 0; 2545534831adSStefano Zampini pcbddc->ChangeOfBasisMatrix = 0; 2546b9b85e73SStefano Zampini pcbddc->user_ChangeOfBasisMatrix = 0; 2547da1bb401SStefano Zampini pcbddc->coarse_vec = 0; 2548da1bb401SStefano Zampini pcbddc->coarse_ksp = 0; 2549da1bb401SStefano Zampini pcbddc->coarse_phi_B = 0; 2550da1bb401SStefano Zampini pcbddc->coarse_phi_D = 0; 255115aaf578SStefano Zampini pcbddc->coarse_psi_B = 0; 255215aaf578SStefano Zampini pcbddc->coarse_psi_D = 0; 2553da1bb401SStefano Zampini pcbddc->vec1_P = 0; 2554da1bb401SStefano Zampini pcbddc->vec1_R = 0; 2555da1bb401SStefano Zampini pcbddc->vec2_R = 0; 2556da1bb401SStefano Zampini pcbddc->local_auxmat1 = 0; 2557da1bb401SStefano Zampini pcbddc->local_auxmat2 = 0; 2558da1bb401SStefano Zampini pcbddc->R_to_B = 0; 2559da1bb401SStefano Zampini pcbddc->R_to_D = 0; 2560da1bb401SStefano Zampini pcbddc->ksp_D = 0; 2561da1bb401SStefano Zampini pcbddc->ksp_R = 0; 2562da1bb401SStefano Zampini pcbddc->NeumannBoundaries = 0; 2563785d1243SStefano Zampini pcbddc->NeumannBoundariesLocal = 0; 2564785d1243SStefano Zampini pcbddc->DirichletBoundaries = 0; 2565785d1243SStefano Zampini pcbddc->DirichletBoundariesLocal = 0; 256660d44989SStefano Zampini pcbddc->user_provided_isfordofs = PETSC_FALSE; 256760d44989SStefano Zampini pcbddc->n_ISForDofs = 0; 256863602bcaSStefano Zampini pcbddc->n_ISForDofsLocal = 0; 2569da1bb401SStefano Zampini pcbddc->ISForDofs = 0; 257063602bcaSStefano Zampini pcbddc->ISForDofsLocal = 0; 2571da1bb401SStefano Zampini pcbddc->ConstraintMatrix = 0; 257285c4d303SStefano Zampini pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 25738ae0ca82SStefano Zampini pcbddc->exact_dirichlet_trick_app = PETSC_FALSE; 257447d04d0dSStefano Zampini pcbddc->coarse_loc_to_glob = 0; 257547d04d0dSStefano Zampini pcbddc->coarsening_ratio = 8; 2576b0c7d250SStefano Zampini pcbddc->coarse_adj_red = 0; 25774fad6a16SStefano Zampini pcbddc->current_level = 0; 25782b510759SStefano Zampini pcbddc->max_levels = 0; 2579323d395dSStefano Zampini pcbddc->use_coarse_estimates = PETSC_FALSE; 258057de7509SStefano Zampini pcbddc->coarse_eqs_per_proc = 1; 2581f3bde8b3SStefano Zampini pcbddc->coarse_subassembling = 0; 25824f1b2e48SStefano Zampini pcbddc->detect_disconnected = PETSC_FALSE; 25834f1b2e48SStefano Zampini pcbddc->n_local_subs = 0; 25844f1b2e48SStefano Zampini pcbddc->local_subs = NULL; 258581d14e9dSStefano Zampini 258681d14e9dSStefano Zampini /* benign subspace trick */ 258781d14e9dSStefano Zampini pcbddc->benign_change = 0; 258827b6a85dSStefano Zampini pcbddc->benign_compute_correction = PETSC_TRUE; 25890369aaf7SStefano Zampini pcbddc->benign_vec = 0; 25900369aaf7SStefano Zampini pcbddc->benign_original_mat = 0; 25910369aaf7SStefano Zampini pcbddc->benign_sf = 0; 25924f1b2e48SStefano Zampini pcbddc->benign_B0 = 0; 25934f1b2e48SStefano Zampini pcbddc->benign_n = 0; 25944f1b2e48SStefano Zampini pcbddc->benign_p0 = NULL; 25954f1b2e48SStefano Zampini pcbddc->benign_p0_lidx = NULL; 25964f1b2e48SStefano Zampini pcbddc->benign_p0_gidx = NULL; 2597b0f5fe93SStefano Zampini pcbddc->benign_null = PETSC_FALSE; 259881d14e9dSStefano Zampini 2599674ae819SStefano Zampini /* create local graph structure */ 2600674ae819SStefano Zampini ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2601be12c134Sstefano_zampini pcbddc->graphmaxcount = PETSC_MAX_INT; 2602674ae819SStefano Zampini 2603674ae819SStefano Zampini /* scaling */ 2604674ae819SStefano Zampini pcbddc->work_scaling = 0; 260534a97f8cSStefano Zampini pcbddc->use_deluxe_scaling = PETSC_FALSE; 2606b96c3477SStefano Zampini 2607b334f244SStefano Zampini /* sub schurs options */ 2608b96c3477SStefano Zampini pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2609b96c3477SStefano Zampini pcbddc->sub_schurs_layers = -1; 2610b96c3477SStefano Zampini pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2611b96c3477SStefano Zampini 2612b96c3477SStefano Zampini pcbddc->computed_rowadj = PETSC_FALSE; 2613da1bb401SStefano Zampini 2614b7eb3628SStefano Zampini /* adaptivity */ 2615f6f667cfSStefano Zampini pcbddc->adaptive_threshold = 0.0; 261608122e43SStefano Zampini pcbddc->adaptive_nmax = 0; 2617f6f667cfSStefano Zampini pcbddc->adaptive_nmin = 0; 2618b7eb3628SStefano Zampini 2619da1bb401SStefano Zampini /* function pointers */ 2620da1bb401SStefano Zampini pc->ops->apply = PCApply_BDDC; 262193bd9ae7SStefano Zampini pc->ops->applytranspose = PCApplyTranspose_BDDC; 2622da1bb401SStefano Zampini pc->ops->setup = PCSetUp_BDDC; 2623da1bb401SStefano Zampini pc->ops->destroy = PCDestroy_BDDC; 2624da1bb401SStefano Zampini pc->ops->setfromoptions = PCSetFromOptions_BDDC; 26256b78500eSPatrick Sanan pc->ops->view = PCView_BDDC; 2626da1bb401SStefano Zampini pc->ops->applyrichardson = 0; 2627da1bb401SStefano Zampini pc->ops->applysymmetricleft = 0; 2628da1bb401SStefano Zampini pc->ops->applysymmetricright = 0; 2629534831adSStefano Zampini pc->ops->presolve = PCPreSolve_BDDC; 2630534831adSStefano Zampini pc->ops->postsolve = PCPostSolve_BDDC; 2631da1bb401SStefano Zampini 2632da1bb401SStefano Zampini /* composing function */ 2633a13144ffSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDiscreteGradient_C",PCBDDCSetDiscreteGradient_BDDC);CHKERRQ(ierr); 2634a198735bSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDivergenceMat_C",PCBDDCSetDivergenceMat_BDDC);CHKERRQ(ierr); 2635906d46d4SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2636674ae819SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 263730368db7SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesIS_C",PCBDDCSetPrimalVerticesIS_BDDC);CHKERRQ(ierr); 2638bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 26392b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2640b8ffe317SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 26412b510759SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2642bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 264382ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2644bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 264582ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2646bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 264782ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2648bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 264982ba6b80SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2650bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 265163602bcaSStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2652bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2653bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2654bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2655bdf89e91SBarry Smith ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2656a06fd7f2SStefano Zampini ierr = PetscObjectComposeFunction((PetscObject)pc,"PCPreSolveChangeRHS_C",PCPreSolveChangeRHS_BDDC);CHKERRQ(ierr); 2657da1bb401SStefano Zampini PetscFunctionReturn(0); 2658da1bb401SStefano Zampini } 26593425bc38SStefano Zampini 2660