1 /* TODOLIST 2 3 Solvers 4 - Add support for cholesky for coarse solver (similar to local solvers) 5 - Propagate ksp prefixes for solvers to mat objects? 6 7 User interface 8 - ** DM attached to pc? 9 10 Debugging output 11 - * Better management of verbosity levels of debugging output 12 13 Extra 14 - *** Is it possible to work with PCBDDCGraph on boundary indices only (less memory consumed)? 15 - BDDC with MG framework? 16 17 FETIDP 18 - Move FETIDP code to its own classes 19 20 MATIS related operations contained in BDDC code 21 - Provide general case for subassembling 22 23 */ 24 25 #include <../src/ksp/pc/impls/bddc/bddc.h> /*I "petscpc.h" I*/ /* includes for fortran wrappers */ 26 #include <../src/ksp/pc/impls/bddc/bddcprivate.h> 27 #include <petscblaslapack.h> 28 29 /* -------------------------------------------------------------------------- */ 30 #undef __FUNCT__ 31 #define __FUNCT__ "PCSetFromOptions_BDDC" 32 PetscErrorCode PCSetFromOptions_BDDC(PetscOptions *PetscOptionsObject,PC pc) 33 { 34 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 35 PetscErrorCode ierr; 36 37 PetscFunctionBegin; 38 ierr = PetscOptionsHead(PetscOptionsObject,"BDDC options");CHKERRQ(ierr); 39 /* Verbose debugging */ 40 ierr = PetscOptionsInt("-pc_bddc_check_level","Verbose output for PCBDDC (intended for debug)","none",pcbddc->dbg_flag,&pcbddc->dbg_flag,NULL);CHKERRQ(ierr); 41 /* Primal space cumstomization */ 42 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); 43 ierr = PetscOptionsBool("-pc_bddc_use_vertices","Use or not corner dofs in coarse space","none",pcbddc->use_vertices,&pcbddc->use_vertices,NULL);CHKERRQ(ierr); 44 ierr = PetscOptionsBool("-pc_bddc_use_edges","Use or not edge constraints in coarse space","none",pcbddc->use_edges,&pcbddc->use_edges,NULL);CHKERRQ(ierr); 45 ierr = PetscOptionsBool("-pc_bddc_use_faces","Use or not face constraints in coarse space","none",pcbddc->use_faces,&pcbddc->use_faces,NULL);CHKERRQ(ierr); 46 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); 47 ierr = PetscOptionsBool("-pc_bddc_use_qr_single","Use QR factorization for single constraints on cc (QR is used when multiple constraints are present)","none",pcbddc->use_qr_single,&pcbddc->use_qr_single,NULL);CHKERRQ(ierr); 48 /* Change of basis */ 49 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); 50 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); 51 if (!pcbddc->use_change_of_basis) { 52 pcbddc->use_change_on_faces = PETSC_FALSE; 53 } 54 /* Switch between M_2 (default) and M_3 preconditioners (as defined by C. Dohrmann in the ref. article) */ 55 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); 56 ierr = PetscOptionsInt("-pc_bddc_coarse_redistribute","Number of procs where to redistribute coarse problem","none",pcbddc->redistribute_coarse,&pcbddc->redistribute_coarse,NULL);CHKERRQ(ierr); 57 ierr = PetscOptionsInt("-pc_bddc_coarsening_ratio","Set coarsening ratio used in multilevel coarsening","none",pcbddc->coarsening_ratio,&pcbddc->coarsening_ratio,NULL);CHKERRQ(ierr); 58 ierr = PetscOptionsInt("-pc_bddc_levels","Set maximum number of levels for multilevel","none",pcbddc->max_levels,&pcbddc->max_levels,NULL);CHKERRQ(ierr); 59 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); 60 ierr = PetscOptionsBool("-pc_bddc_use_deluxe_scaling","Use deluxe scaling for BDDC","none",pcbddc->use_deluxe_scaling,&pcbddc->use_deluxe_scaling,NULL);CHKERRQ(ierr); 61 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); 62 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); 63 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); 64 ierr = PetscOptionsBool("-pc_bddc_deluxe_faster","Faster application of deluxe scaling (requires extra work during setup)","none",pcbddc->faster_deluxe,&pcbddc->faster_deluxe,NULL);CHKERRQ(ierr); 65 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); 66 ierr = PetscOptionsInt("-pc_bddc_adaptive_nmin","Minimum number of constraints per connected components","none",pcbddc->adaptive_nmin,&pcbddc->adaptive_nmin,NULL);CHKERRQ(ierr); 67 ierr = PetscOptionsInt("-pc_bddc_adaptive_nmax","Maximum number of constraints per connected components","none",pcbddc->adaptive_nmax,&pcbddc->adaptive_nmax,NULL);CHKERRQ(ierr); 68 ierr = PetscOptionsBool("-pc_bddc_symmetric","Symmetric computation of primal basis functions","none",pcbddc->symmetric_primal,&pcbddc->symmetric_primal,NULL);CHKERRQ(ierr); 69 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); 70 ierr = PetscOptionsBool("-pc_bddc_saddle_point","Apply the benign subspace trick","none",pcbddc->saddle_point,&pcbddc->saddle_point,NULL);CHKERRQ(ierr); 71 ierr = PetscOptionsTail();CHKERRQ(ierr); 72 PetscFunctionReturn(0); 73 } 74 /* -------------------------------------------------------------------------- */ 75 #undef __FUNCT__ 76 #define __FUNCT__ "PCBDDCSetChangeOfBasisMat_BDDC" 77 static PetscErrorCode PCBDDCSetChangeOfBasisMat_BDDC(PC pc, Mat change) 78 { 79 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 80 PetscErrorCode ierr; 81 82 PetscFunctionBegin; 83 ierr = MatDestroy(&pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 84 ierr = PetscObjectReference((PetscObject)change);CHKERRQ(ierr); 85 pcbddc->user_ChangeOfBasisMatrix = change; 86 PetscFunctionReturn(0); 87 } 88 #undef __FUNCT__ 89 #define __FUNCT__ "PCBDDCSetChangeOfBasisMat" 90 /*@ 91 PCBDDCSetChangeOfBasisMat - Set user defined change of basis for dofs 92 93 Collective on PC 94 95 Input Parameters: 96 + pc - the preconditioning context 97 - change - the change of basis matrix 98 99 Level: intermediate 100 101 Notes: 102 103 .seealso: PCBDDC 104 @*/ 105 PetscErrorCode PCBDDCSetChangeOfBasisMat(PC pc, Mat change) 106 { 107 PetscErrorCode ierr; 108 109 PetscFunctionBegin; 110 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 111 PetscValidHeaderSpecific(change,MAT_CLASSID,2); 112 PetscCheckSameComm(pc,1,change,2); 113 if (pc->mat) { 114 PetscInt rows_c,cols_c,rows,cols; 115 ierr = MatGetSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 116 ierr = MatGetSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 117 if (rows_c != rows) { 118 SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of rows for change of basis matrix! %d != %d",rows_c,rows); 119 } 120 if (cols_c != cols) { 121 SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of columns for change of basis matrix! %d != %d",cols_c,cols); 122 } 123 ierr = MatGetLocalSize(pc->mat,&rows,&cols);CHKERRQ(ierr); 124 ierr = MatGetLocalSize(change,&rows_c,&cols_c);CHKERRQ(ierr); 125 if (rows_c != rows) { 126 SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local rows for change of basis matrix! %d != %d",rows_c,rows); 127 } 128 if (cols_c != cols) { 129 SETERRQ2(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Invalid number of local columns for change of basis matrix! %d != %d",cols_c,cols); 130 } 131 } 132 ierr = PetscTryMethod(pc,"PCBDDCSetChangeOfBasisMat_C",(PC,Mat),(pc,change));CHKERRQ(ierr); 133 PetscFunctionReturn(0); 134 } 135 /* -------------------------------------------------------------------------- */ 136 #undef __FUNCT__ 137 #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS_BDDC" 138 static PetscErrorCode PCBDDCSetPrimalVerticesLocalIS_BDDC(PC pc, IS PrimalVertices) 139 { 140 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 141 PetscErrorCode ierr; 142 143 PetscFunctionBegin; 144 ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 145 ierr = PetscObjectReference((PetscObject)PrimalVertices);CHKERRQ(ierr); 146 pcbddc->user_primal_vertices = PrimalVertices; 147 PetscFunctionReturn(0); 148 } 149 #undef __FUNCT__ 150 #define __FUNCT__ "PCBDDCSetPrimalVerticesLocalIS" 151 /*@ 152 PCBDDCSetPrimalVerticesLocalIS - Set additional user defined primal vertices in PCBDDC 153 154 Collective 155 156 Input Parameters: 157 + pc - the preconditioning context 158 - PrimalVertices - index set of primal vertices in local numbering (can be empty) 159 160 Level: intermediate 161 162 Notes: 163 164 .seealso: PCBDDC 165 @*/ 166 PetscErrorCode PCBDDCSetPrimalVerticesLocalIS(PC pc, IS PrimalVertices) 167 { 168 PetscErrorCode ierr; 169 170 PetscFunctionBegin; 171 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 172 PetscValidHeaderSpecific(PrimalVertices,IS_CLASSID,2); 173 PetscCheckSameComm(pc,1,PrimalVertices,2); 174 ierr = PetscTryMethod(pc,"PCBDDCSetPrimalVerticesLocalIS_C",(PC,IS),(pc,PrimalVertices));CHKERRQ(ierr); 175 PetscFunctionReturn(0); 176 } 177 /* -------------------------------------------------------------------------- */ 178 #undef __FUNCT__ 179 #define __FUNCT__ "PCBDDCSetCoarseningRatio_BDDC" 180 static PetscErrorCode PCBDDCSetCoarseningRatio_BDDC(PC pc,PetscInt k) 181 { 182 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 183 184 PetscFunctionBegin; 185 pcbddc->coarsening_ratio = k; 186 PetscFunctionReturn(0); 187 } 188 189 #undef __FUNCT__ 190 #define __FUNCT__ "PCBDDCSetCoarseningRatio" 191 /*@ 192 PCBDDCSetCoarseningRatio - Set coarsening ratio used in multilevel 193 194 Logically collective on PC 195 196 Input Parameters: 197 + pc - the preconditioning context 198 - k - coarsening ratio (H/h at the coarser level) 199 200 Options Database Keys: 201 . -pc_bddc_coarsening_ratio 202 203 Level: intermediate 204 205 Notes: 206 Approximatively k subdomains at the finer level will be aggregated into a single subdomain at the coarser level 207 208 .seealso: PCBDDC, PCBDDCSetLevels() 209 @*/ 210 PetscErrorCode PCBDDCSetCoarseningRatio(PC pc,PetscInt k) 211 { 212 PetscErrorCode ierr; 213 214 PetscFunctionBegin; 215 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 216 PetscValidLogicalCollectiveInt(pc,k,2); 217 ierr = PetscTryMethod(pc,"PCBDDCSetCoarseningRatio_C",(PC,PetscInt),(pc,k));CHKERRQ(ierr); 218 PetscFunctionReturn(0); 219 } 220 221 /* The following functions (PCBDDCSetUseExactDirichlet PCBDDCSetLevel) are not public */ 222 #undef __FUNCT__ 223 #define __FUNCT__ "PCBDDCSetUseExactDirichlet_BDDC" 224 static PetscErrorCode PCBDDCSetUseExactDirichlet_BDDC(PC pc,PetscBool flg) 225 { 226 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 227 228 PetscFunctionBegin; 229 pcbddc->use_exact_dirichlet_trick = flg; 230 PetscFunctionReturn(0); 231 } 232 233 #undef __FUNCT__ 234 #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 235 PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool flg) 236 { 237 PetscErrorCode ierr; 238 239 PetscFunctionBegin; 240 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 241 PetscValidLogicalCollectiveBool(pc,flg,2); 242 ierr = PetscTryMethod(pc,"PCBDDCSetUseExactDirichlet_C",(PC,PetscBool),(pc,flg));CHKERRQ(ierr); 243 PetscFunctionReturn(0); 244 } 245 246 #undef __FUNCT__ 247 #define __FUNCT__ "PCBDDCSetLevel_BDDC" 248 static PetscErrorCode PCBDDCSetLevel_BDDC(PC pc,PetscInt level) 249 { 250 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 251 252 PetscFunctionBegin; 253 pcbddc->current_level = level; 254 PetscFunctionReturn(0); 255 } 256 257 #undef __FUNCT__ 258 #define __FUNCT__ "PCBDDCSetLevel" 259 PetscErrorCode PCBDDCSetLevel(PC pc,PetscInt level) 260 { 261 PetscErrorCode ierr; 262 263 PetscFunctionBegin; 264 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 265 PetscValidLogicalCollectiveInt(pc,level,2); 266 ierr = PetscTryMethod(pc,"PCBDDCSetLevel_C",(PC,PetscInt),(pc,level));CHKERRQ(ierr); 267 PetscFunctionReturn(0); 268 } 269 270 #undef __FUNCT__ 271 #define __FUNCT__ "PCBDDCSetLevels_BDDC" 272 static PetscErrorCode PCBDDCSetLevels_BDDC(PC pc,PetscInt levels) 273 { 274 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 275 276 PetscFunctionBegin; 277 pcbddc->max_levels = levels; 278 PetscFunctionReturn(0); 279 } 280 281 #undef __FUNCT__ 282 #define __FUNCT__ "PCBDDCSetLevels" 283 /*@ 284 PCBDDCSetLevels - Sets the maximum number of levels for multilevel 285 286 Logically collective on PC 287 288 Input Parameters: 289 + pc - the preconditioning context 290 - levels - the maximum number of levels (max 9) 291 292 Options Database Keys: 293 . -pc_bddc_levels 294 295 Level: intermediate 296 297 Notes: 298 Default value is 0, i.e. traditional one-level BDDC 299 300 .seealso: PCBDDC, PCBDDCSetCoarseningRatio() 301 @*/ 302 PetscErrorCode PCBDDCSetLevels(PC pc,PetscInt levels) 303 { 304 PetscErrorCode ierr; 305 306 PetscFunctionBegin; 307 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 308 PetscValidLogicalCollectiveInt(pc,levels,2); 309 if (levels > 99) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Maximum number of levels for bddc is 99\n"); 310 ierr = PetscTryMethod(pc,"PCBDDCSetLevels_C",(PC,PetscInt),(pc,levels));CHKERRQ(ierr); 311 PetscFunctionReturn(0); 312 } 313 /* -------------------------------------------------------------------------- */ 314 315 #undef __FUNCT__ 316 #define __FUNCT__ "PCBDDCSetNullSpace_BDDC" 317 static PetscErrorCode PCBDDCSetNullSpace_BDDC(PC pc,MatNullSpace NullSpace) 318 { 319 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 320 PetscErrorCode ierr; 321 322 PetscFunctionBegin; 323 ierr = PetscObjectReference((PetscObject)NullSpace);CHKERRQ(ierr); 324 ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 325 pcbddc->NullSpace = NullSpace; 326 PetscFunctionReturn(0); 327 } 328 329 #undef __FUNCT__ 330 #define __FUNCT__ "PCBDDCSetNullSpace" 331 /*@ 332 PCBDDCSetNullSpace - Set nullspace for BDDC operator 333 334 Logically collective on PC and MatNullSpace 335 336 Input Parameters: 337 + pc - the preconditioning context 338 - NullSpace - Null space of the linear operator to be preconditioned (Pmat) 339 340 Level: intermediate 341 342 Notes: 343 344 .seealso: PCBDDC 345 @*/ 346 PetscErrorCode PCBDDCSetNullSpace(PC pc,MatNullSpace NullSpace) 347 { 348 PetscErrorCode ierr; 349 350 PetscFunctionBegin; 351 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 352 PetscValidHeaderSpecific(NullSpace,MAT_NULLSPACE_CLASSID,2); 353 PetscCheckSameComm(pc,1,NullSpace,2); 354 ierr = PetscTryMethod(pc,"PCBDDCSetNullSpace_C",(PC,MatNullSpace),(pc,NullSpace));CHKERRQ(ierr); 355 PetscFunctionReturn(0); 356 } 357 /* -------------------------------------------------------------------------- */ 358 359 #undef __FUNCT__ 360 #define __FUNCT__ "PCBDDCSetDirichletBoundaries_BDDC" 361 static PetscErrorCode PCBDDCSetDirichletBoundaries_BDDC(PC pc,IS DirichletBoundaries) 362 { 363 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 364 PetscErrorCode ierr; 365 366 PetscFunctionBegin; 367 /* last user setting takes precendence -> destroy any other customization */ 368 ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 369 ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 370 ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 371 pcbddc->DirichletBoundaries = DirichletBoundaries; 372 pcbddc->recompute_topography = PETSC_TRUE; 373 PetscFunctionReturn(0); 374 } 375 376 #undef __FUNCT__ 377 #define __FUNCT__ "PCBDDCSetDirichletBoundaries" 378 /*@ 379 PCBDDCSetDirichletBoundaries - Set IS defining Dirichlet boundaries for the global problem. 380 381 Collective 382 383 Input Parameters: 384 + pc - the preconditioning context 385 - DirichletBoundaries - parallel IS defining the Dirichlet boundaries 386 387 Level: intermediate 388 389 Notes: 390 Provide the information if you used MatZeroRows/Columns routines. Any process can list any global node 391 392 .seealso: PCBDDC, PCBDDCSetDirichletBoundariesLocal(), MatZeroRows(), MatZeroRowsColumns() 393 @*/ 394 PetscErrorCode PCBDDCSetDirichletBoundaries(PC pc,IS DirichletBoundaries) 395 { 396 PetscErrorCode ierr; 397 398 PetscFunctionBegin; 399 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 400 PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 401 PetscCheckSameComm(pc,1,DirichletBoundaries,2); 402 ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundaries_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 403 PetscFunctionReturn(0); 404 } 405 /* -------------------------------------------------------------------------- */ 406 407 #undef __FUNCT__ 408 #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal_BDDC" 409 static PetscErrorCode PCBDDCSetDirichletBoundariesLocal_BDDC(PC pc,IS DirichletBoundaries) 410 { 411 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 412 PetscErrorCode ierr; 413 414 PetscFunctionBegin; 415 /* last user setting takes precendence -> destroy any other customization */ 416 ierr = ISDestroy(&pcbddc->DirichletBoundariesLocal);CHKERRQ(ierr); 417 ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 418 ierr = PetscObjectReference((PetscObject)DirichletBoundaries);CHKERRQ(ierr); 419 pcbddc->DirichletBoundariesLocal = DirichletBoundaries; 420 pcbddc->recompute_topography = PETSC_TRUE; 421 PetscFunctionReturn(0); 422 } 423 424 #undef __FUNCT__ 425 #define __FUNCT__ "PCBDDCSetDirichletBoundariesLocal" 426 /*@ 427 PCBDDCSetDirichletBoundariesLocal - Set IS defining Dirichlet boundaries for the global problem in local ordering. 428 429 Collective 430 431 Input Parameters: 432 + pc - the preconditioning context 433 - DirichletBoundaries - parallel IS defining the Dirichlet boundaries (in local ordering) 434 435 Level: intermediate 436 437 Notes: 438 439 .seealso: PCBDDC, PCBDDCSetDirichletBoundaries(), MatZeroRows(), MatZeroRowsColumns() 440 @*/ 441 PetscErrorCode PCBDDCSetDirichletBoundariesLocal(PC pc,IS DirichletBoundaries) 442 { 443 PetscErrorCode ierr; 444 445 PetscFunctionBegin; 446 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 447 PetscValidHeaderSpecific(DirichletBoundaries,IS_CLASSID,2); 448 PetscCheckSameComm(pc,1,DirichletBoundaries,2); 449 ierr = PetscTryMethod(pc,"PCBDDCSetDirichletBoundariesLocal_C",(PC,IS),(pc,DirichletBoundaries));CHKERRQ(ierr); 450 PetscFunctionReturn(0); 451 } 452 /* -------------------------------------------------------------------------- */ 453 454 #undef __FUNCT__ 455 #define __FUNCT__ "PCBDDCSetNeumannBoundaries_BDDC" 456 static PetscErrorCode PCBDDCSetNeumannBoundaries_BDDC(PC pc,IS NeumannBoundaries) 457 { 458 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 459 PetscErrorCode ierr; 460 461 PetscFunctionBegin; 462 /* last user setting takes precendence -> destroy any other customization */ 463 ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 464 ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 465 ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 466 pcbddc->NeumannBoundaries = NeumannBoundaries; 467 pcbddc->recompute_topography = PETSC_TRUE; 468 PetscFunctionReturn(0); 469 } 470 471 #undef __FUNCT__ 472 #define __FUNCT__ "PCBDDCSetNeumannBoundaries" 473 /*@ 474 PCBDDCSetNeumannBoundaries - Set IS defining Neumann boundaries for the global problem. 475 476 Collective 477 478 Input Parameters: 479 + pc - the preconditioning context 480 - NeumannBoundaries - parallel IS defining the Neumann boundaries 481 482 Level: intermediate 483 484 Notes: 485 Any process can list any global node 486 487 .seealso: PCBDDC, PCBDDCSetNeumannBoundariesLocal() 488 @*/ 489 PetscErrorCode PCBDDCSetNeumannBoundaries(PC pc,IS NeumannBoundaries) 490 { 491 PetscErrorCode ierr; 492 493 PetscFunctionBegin; 494 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 495 PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 496 PetscCheckSameComm(pc,1,NeumannBoundaries,2); 497 ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundaries_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 498 PetscFunctionReturn(0); 499 } 500 /* -------------------------------------------------------------------------- */ 501 502 #undef __FUNCT__ 503 #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal_BDDC" 504 static PetscErrorCode PCBDDCSetNeumannBoundariesLocal_BDDC(PC pc,IS NeumannBoundaries) 505 { 506 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 507 PetscErrorCode ierr; 508 509 PetscFunctionBegin; 510 /* last user setting takes precendence -> destroy any other customization */ 511 ierr = ISDestroy(&pcbddc->NeumannBoundariesLocal);CHKERRQ(ierr); 512 ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 513 ierr = PetscObjectReference((PetscObject)NeumannBoundaries);CHKERRQ(ierr); 514 pcbddc->NeumannBoundariesLocal = NeumannBoundaries; 515 pcbddc->recompute_topography = PETSC_TRUE; 516 PetscFunctionReturn(0); 517 } 518 519 #undef __FUNCT__ 520 #define __FUNCT__ "PCBDDCSetNeumannBoundariesLocal" 521 /*@ 522 PCBDDCSetNeumannBoundariesLocal - Set IS defining Neumann boundaries for the global problem in local ordering. 523 524 Collective 525 526 Input Parameters: 527 + pc - the preconditioning context 528 - NeumannBoundaries - parallel IS defining the subdomain part of Neumann boundaries (in local ordering) 529 530 Level: intermediate 531 532 Notes: 533 534 .seealso: PCBDDC, PCBDDCSetNeumannBoundaries() 535 @*/ 536 PetscErrorCode PCBDDCSetNeumannBoundariesLocal(PC pc,IS NeumannBoundaries) 537 { 538 PetscErrorCode ierr; 539 540 PetscFunctionBegin; 541 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 542 PetscValidHeaderSpecific(NeumannBoundaries,IS_CLASSID,2); 543 PetscCheckSameComm(pc,1,NeumannBoundaries,2); 544 ierr = PetscTryMethod(pc,"PCBDDCSetNeumannBoundariesLocal_C",(PC,IS),(pc,NeumannBoundaries));CHKERRQ(ierr); 545 PetscFunctionReturn(0); 546 } 547 /* -------------------------------------------------------------------------- */ 548 549 #undef __FUNCT__ 550 #define __FUNCT__ "PCBDDCGetDirichletBoundaries_BDDC" 551 static PetscErrorCode PCBDDCGetDirichletBoundaries_BDDC(PC pc,IS *DirichletBoundaries) 552 { 553 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 554 555 PetscFunctionBegin; 556 *DirichletBoundaries = pcbddc->DirichletBoundaries; 557 PetscFunctionReturn(0); 558 } 559 560 #undef __FUNCT__ 561 #define __FUNCT__ "PCBDDCGetDirichletBoundaries" 562 /*@ 563 PCBDDCGetDirichletBoundaries - Get parallel IS for Dirichlet boundaries 564 565 Collective 566 567 Input Parameters: 568 . pc - the preconditioning context 569 570 Output Parameters: 571 . DirichletBoundaries - index set defining the Dirichlet boundaries 572 573 Level: intermediate 574 575 Notes: 576 The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetDirichletBoundaries 577 578 .seealso: PCBDDC 579 @*/ 580 PetscErrorCode PCBDDCGetDirichletBoundaries(PC pc,IS *DirichletBoundaries) 581 { 582 PetscErrorCode ierr; 583 584 PetscFunctionBegin; 585 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 586 ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundaries_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 587 PetscFunctionReturn(0); 588 } 589 /* -------------------------------------------------------------------------- */ 590 591 #undef __FUNCT__ 592 #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal_BDDC" 593 static PetscErrorCode PCBDDCGetDirichletBoundariesLocal_BDDC(PC pc,IS *DirichletBoundaries) 594 { 595 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 596 597 PetscFunctionBegin; 598 *DirichletBoundaries = pcbddc->DirichletBoundariesLocal; 599 PetscFunctionReturn(0); 600 } 601 602 #undef __FUNCT__ 603 #define __FUNCT__ "PCBDDCGetDirichletBoundariesLocal" 604 /*@ 605 PCBDDCGetDirichletBoundariesLocal - Get parallel IS for Dirichlet boundaries (in local ordering) 606 607 Collective 608 609 Input Parameters: 610 . pc - the preconditioning context 611 612 Output Parameters: 613 . DirichletBoundaries - index set defining the subdomain part of Dirichlet boundaries 614 615 Level: intermediate 616 617 Notes: 618 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). 619 In the latter case, the IS will be available after PCSetUp. 620 621 .seealso: PCBDDC 622 @*/ 623 PetscErrorCode PCBDDCGetDirichletBoundariesLocal(PC pc,IS *DirichletBoundaries) 624 { 625 PetscErrorCode ierr; 626 627 PetscFunctionBegin; 628 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 629 ierr = PetscUseMethod(pc,"PCBDDCGetDirichletBoundariesLocal_C",(PC,IS*),(pc,DirichletBoundaries));CHKERRQ(ierr); 630 PetscFunctionReturn(0); 631 } 632 /* -------------------------------------------------------------------------- */ 633 634 #undef __FUNCT__ 635 #define __FUNCT__ "PCBDDCGetNeumannBoundaries_BDDC" 636 static PetscErrorCode PCBDDCGetNeumannBoundaries_BDDC(PC pc,IS *NeumannBoundaries) 637 { 638 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 639 640 PetscFunctionBegin; 641 *NeumannBoundaries = pcbddc->NeumannBoundaries; 642 PetscFunctionReturn(0); 643 } 644 645 #undef __FUNCT__ 646 #define __FUNCT__ "PCBDDCGetNeumannBoundaries" 647 /*@ 648 PCBDDCGetNeumannBoundaries - Get parallel IS for Neumann boundaries 649 650 Collective 651 652 Input Parameters: 653 . pc - the preconditioning context 654 655 Output Parameters: 656 . NeumannBoundaries - index set defining the Neumann boundaries 657 658 Level: intermediate 659 660 Notes: 661 The IS returned (if any) is the same passed in earlier by the user with PCBDDCSetNeumannBoundaries 662 663 .seealso: PCBDDC 664 @*/ 665 PetscErrorCode PCBDDCGetNeumannBoundaries(PC pc,IS *NeumannBoundaries) 666 { 667 PetscErrorCode ierr; 668 669 PetscFunctionBegin; 670 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 671 ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundaries_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 672 PetscFunctionReturn(0); 673 } 674 /* -------------------------------------------------------------------------- */ 675 676 #undef __FUNCT__ 677 #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal_BDDC" 678 static PetscErrorCode PCBDDCGetNeumannBoundariesLocal_BDDC(PC pc,IS *NeumannBoundaries) 679 { 680 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 681 682 PetscFunctionBegin; 683 *NeumannBoundaries = pcbddc->NeumannBoundariesLocal; 684 PetscFunctionReturn(0); 685 } 686 687 #undef __FUNCT__ 688 #define __FUNCT__ "PCBDDCGetNeumannBoundariesLocal" 689 /*@ 690 PCBDDCGetNeumannBoundariesLocal - Get parallel IS for Neumann boundaries (in local ordering) 691 692 Collective 693 694 Input Parameters: 695 . pc - the preconditioning context 696 697 Output Parameters: 698 . NeumannBoundaries - index set defining the subdomain part of Neumann boundaries 699 700 Level: intermediate 701 702 Notes: 703 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). 704 In the latter case, the IS will be available after PCSetUp. 705 706 .seealso: PCBDDC 707 @*/ 708 PetscErrorCode PCBDDCGetNeumannBoundariesLocal(PC pc,IS *NeumannBoundaries) 709 { 710 PetscErrorCode ierr; 711 712 PetscFunctionBegin; 713 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 714 ierr = PetscUseMethod(pc,"PCBDDCGetNeumannBoundariesLocal_C",(PC,IS*),(pc,NeumannBoundaries));CHKERRQ(ierr); 715 PetscFunctionReturn(0); 716 } 717 /* -------------------------------------------------------------------------- */ 718 719 #undef __FUNCT__ 720 #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph_BDDC" 721 static PetscErrorCode PCBDDCSetLocalAdjacencyGraph_BDDC(PC pc, PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 722 { 723 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 724 PCBDDCGraph mat_graph = pcbddc->mat_graph; 725 PetscErrorCode ierr; 726 727 PetscFunctionBegin; 728 /* free old CSR */ 729 ierr = PCBDDCGraphResetCSR(mat_graph);CHKERRQ(ierr); 730 /* TODO: PCBDDCGraphSetAdjacency */ 731 /* get CSR into graph structure */ 732 if (copymode == PETSC_COPY_VALUES) { 733 ierr = PetscMalloc1(nvtxs+1,&mat_graph->xadj);CHKERRQ(ierr); 734 ierr = PetscMalloc1(xadj[nvtxs],&mat_graph->adjncy);CHKERRQ(ierr); 735 ierr = PetscMemcpy(mat_graph->xadj,xadj,(nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 736 ierr = PetscMemcpy(mat_graph->adjncy,adjncy,xadj[nvtxs]*sizeof(PetscInt));CHKERRQ(ierr); 737 } else if (copymode == PETSC_OWN_POINTER) { 738 mat_graph->xadj = (PetscInt*)xadj; 739 mat_graph->adjncy = (PetscInt*)adjncy; 740 } 741 mat_graph->nvtxs_csr = nvtxs; 742 PetscFunctionReturn(0); 743 } 744 745 #undef __FUNCT__ 746 #define __FUNCT__ "PCBDDCSetLocalAdjacencyGraph" 747 /*@ 748 PCBDDCSetLocalAdjacencyGraph - Set adjacency structure (CSR graph) of the local matrix 749 750 Not collective 751 752 Input Parameters: 753 + pc - the preconditioning context 754 . nvtxs - number of local vertices of the graph (i.e., the size of the local problem) 755 . xadj, adjncy - the CSR graph 756 - copymode - either PETSC_COPY_VALUES or PETSC_OWN_POINTER. 757 758 Level: intermediate 759 760 Notes: 761 762 .seealso: PCBDDC,PetscCopyMode 763 @*/ 764 PetscErrorCode PCBDDCSetLocalAdjacencyGraph(PC pc,PetscInt nvtxs,const PetscInt xadj[],const PetscInt adjncy[], PetscCopyMode copymode) 765 { 766 void (*f)(void) = 0; 767 PetscErrorCode ierr; 768 769 PetscFunctionBegin; 770 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 771 PetscValidIntPointer(xadj,3); 772 PetscValidIntPointer(adjncy,4); 773 if (copymode != PETSC_COPY_VALUES && copymode != PETSC_OWN_POINTER) { 774 SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported copy mode %d in %s\n",copymode,__FUNCT__); 775 } 776 ierr = PetscTryMethod(pc,"PCBDDCSetLocalAdjacencyGraph_C",(PC,PetscInt,const PetscInt[],const PetscInt[],PetscCopyMode),(pc,nvtxs,xadj,adjncy,copymode));CHKERRQ(ierr); 777 /* free arrays if PCBDDC is not the PC type */ 778 ierr = PetscObjectQueryFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",&f);CHKERRQ(ierr); 779 if (!f && copymode == PETSC_OWN_POINTER) { 780 ierr = PetscFree(xadj);CHKERRQ(ierr); 781 ierr = PetscFree(adjncy);CHKERRQ(ierr); 782 } 783 PetscFunctionReturn(0); 784 } 785 /* -------------------------------------------------------------------------- */ 786 787 #undef __FUNCT__ 788 #define __FUNCT__ "PCBDDCSetDofsSplittingLocal_BDDC" 789 static PetscErrorCode PCBDDCSetDofsSplittingLocal_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 790 { 791 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 792 PetscInt i; 793 PetscErrorCode ierr; 794 795 PetscFunctionBegin; 796 /* Destroy ISes if they were already set */ 797 for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 798 ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 799 } 800 ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 801 /* last user setting takes precendence -> destroy any other customization */ 802 for (i=0;i<pcbddc->n_ISForDofs;i++) { 803 ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 804 } 805 ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 806 pcbddc->n_ISForDofs = 0; 807 /* allocate space then set */ 808 if (n_is) { 809 ierr = PetscMalloc1(n_is,&pcbddc->ISForDofsLocal);CHKERRQ(ierr); 810 } 811 for (i=0;i<n_is;i++) { 812 ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 813 pcbddc->ISForDofsLocal[i]=ISForDofs[i]; 814 } 815 pcbddc->n_ISForDofsLocal=n_is; 816 if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 817 pcbddc->recompute_topography = PETSC_TRUE; 818 PetscFunctionReturn(0); 819 } 820 821 #undef __FUNCT__ 822 #define __FUNCT__ "PCBDDCSetDofsSplittingLocal" 823 /*@ 824 PCBDDCSetDofsSplittingLocal - Set index sets defining fields of the local subdomain matrix 825 826 Collective 827 828 Input Parameters: 829 + pc - the preconditioning context 830 . n_is - number of index sets defining the fields 831 - ISForDofs - array of IS describing the fields in local ordering 832 833 Level: intermediate 834 835 Notes: 836 n_is should be the same among processes. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 837 838 .seealso: PCBDDC 839 @*/ 840 PetscErrorCode PCBDDCSetDofsSplittingLocal(PC pc,PetscInt n_is, IS ISForDofs[]) 841 { 842 PetscInt i; 843 PetscErrorCode ierr; 844 845 PetscFunctionBegin; 846 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 847 PetscValidLogicalCollectiveInt(pc,n_is,2); 848 for (i=0;i<n_is;i++) { 849 PetscCheckSameComm(pc,1,ISForDofs[i],3); 850 PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 851 } 852 ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplittingLocal_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 853 PetscFunctionReturn(0); 854 } 855 /* -------------------------------------------------------------------------- */ 856 857 #undef __FUNCT__ 858 #define __FUNCT__ "PCBDDCSetDofsSplitting_BDDC" 859 static PetscErrorCode PCBDDCSetDofsSplitting_BDDC(PC pc,PetscInt n_is, IS ISForDofs[]) 860 { 861 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 862 PetscInt i; 863 PetscErrorCode ierr; 864 865 PetscFunctionBegin; 866 /* Destroy ISes if they were already set */ 867 for (i=0;i<pcbddc->n_ISForDofs;i++) { 868 ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 869 } 870 ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 871 /* last user setting takes precendence -> destroy any other customization */ 872 for (i=0;i<pcbddc->n_ISForDofsLocal;i++) { 873 ierr = ISDestroy(&pcbddc->ISForDofsLocal[i]);CHKERRQ(ierr); 874 } 875 ierr = PetscFree(pcbddc->ISForDofsLocal);CHKERRQ(ierr); 876 pcbddc->n_ISForDofsLocal = 0; 877 /* allocate space then set */ 878 if (n_is) { 879 ierr = PetscMalloc1(n_is,&pcbddc->ISForDofs);CHKERRQ(ierr); 880 } 881 for (i=0;i<n_is;i++) { 882 ierr = PetscObjectReference((PetscObject)ISForDofs[i]);CHKERRQ(ierr); 883 pcbddc->ISForDofs[i]=ISForDofs[i]; 884 } 885 pcbddc->n_ISForDofs=n_is; 886 if (n_is) pcbddc->user_provided_isfordofs = PETSC_TRUE; 887 pcbddc->recompute_topography = PETSC_TRUE; 888 PetscFunctionReturn(0); 889 } 890 891 #undef __FUNCT__ 892 #define __FUNCT__ "PCBDDCSetDofsSplitting" 893 /*@ 894 PCBDDCSetDofsSplitting - Set index sets defining fields of the global matrix 895 896 Collective 897 898 Input Parameters: 899 + pc - the preconditioning context 900 . n_is - number of index sets defining the fields 901 - ISForDofs - array of IS describing the fields in global ordering 902 903 Level: intermediate 904 905 Notes: 906 Any process can list any global node. Not all nodes need to be listed: unlisted nodes will belong to the complement field. 907 908 .seealso: PCBDDC 909 @*/ 910 PetscErrorCode PCBDDCSetDofsSplitting(PC pc,PetscInt n_is, IS ISForDofs[]) 911 { 912 PetscInt i; 913 PetscErrorCode ierr; 914 915 PetscFunctionBegin; 916 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 917 PetscValidLogicalCollectiveInt(pc,n_is,2); 918 for (i=0;i<n_is;i++) { 919 PetscCheckSameComm(pc,1,ISForDofs[i],3); 920 PetscValidHeaderSpecific(ISForDofs[i],IS_CLASSID,3); 921 } 922 ierr = PetscTryMethod(pc,"PCBDDCSetDofsSplitting_C",(PC,PetscInt,IS[]),(pc,n_is,ISForDofs));CHKERRQ(ierr); 923 PetscFunctionReturn(0); 924 } 925 926 /* -------------------------------------------------------------------------- */ 927 #undef __FUNCT__ 928 #define __FUNCT__ "PCPreSolve_BDDC" 929 /* -------------------------------------------------------------------------- */ 930 /* 931 PCPreSolve_BDDC - Changes the right hand side and (if necessary) the initial 932 guess if a transformation of basis approach has been selected. 933 934 Input Parameter: 935 + pc - the preconditioner contex 936 937 Application Interface Routine: PCPreSolve() 938 939 Notes: 940 The interface routine PCPreSolve() is not usually called directly by 941 the user, but instead is called by KSPSolve(). 942 */ 943 static PetscErrorCode PCPreSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 944 { 945 PetscErrorCode ierr; 946 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 947 PC_IS *pcis = (PC_IS*)(pc->data); 948 Vec used_vec; 949 PetscBool copy_rhs = PETSC_TRUE; 950 951 PetscFunctionBegin; 952 /* if we are working with cg, one dirichlet solve can be avoided during Krylov iterations */ 953 if (ksp) { 954 PetscBool iscg; 955 ierr = PetscObjectTypeCompare((PetscObject)ksp,KSPCG,&iscg);CHKERRQ(ierr); 956 if (!iscg) { 957 ierr = PCBDDCSetUseExactDirichlet(pc,PETSC_FALSE);CHKERRQ(ierr); 958 } 959 } 960 /* Creates parallel work vectors used in presolve */ 961 if (!pcbddc->original_rhs) { 962 ierr = VecDuplicate(pcis->vec1_global,&pcbddc->original_rhs);CHKERRQ(ierr); 963 } 964 if (!pcbddc->temp_solution) { 965 ierr = VecDuplicate(pcis->vec1_global,&pcbddc->temp_solution);CHKERRQ(ierr); 966 } 967 968 if (x) { 969 ierr = PetscObjectReference((PetscObject)x);CHKERRQ(ierr); 970 used_vec = x; 971 } else { /* it can only happen when calling PCBDDCMatFETIDPGetRHS */ 972 ierr = PetscObjectReference((PetscObject)pcbddc->temp_solution);CHKERRQ(ierr); 973 used_vec = pcbddc->temp_solution; 974 ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 975 } 976 977 /* hack into ksp data structure since PCPreSolve comes earlier than setting to zero the guess in src/ksp/ksp/interface/itfunc.c */ 978 if (ksp) { 979 /* store the flag for the initial guess since it will be restored back during PCPostSolve_BDDC */ 980 ierr = KSPGetInitialGuessNonzero(ksp,&pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 981 if (!pcbddc->ksp_guess_nonzero) { 982 ierr = VecSet(used_vec,0.0);CHKERRQ(ierr); 983 } 984 } 985 986 pcbddc->rhs_change = PETSC_FALSE; 987 988 /* Take into account zeroed rows -> change rhs and store solution removed */ 989 if (rhs) { 990 IS dirIS = NULL; 991 992 /* DirichletBoundariesLocal may not be consistent among neighbours; gets a dirichlet dofs IS from graph (may be cached) */ 993 ierr = PCBDDCGraphGetDirichletDofs(pcbddc->mat_graph,&dirIS);CHKERRQ(ierr); 994 if (dirIS) { 995 Mat_IS *matis = (Mat_IS*)pc->pmat->data; 996 PetscInt dirsize,i,*is_indices; 997 PetscScalar *array_x; 998 const PetscScalar *array_diagonal; 999 1000 ierr = MatGetDiagonal(pc->pmat,pcis->vec1_global);CHKERRQ(ierr); 1001 ierr = VecPointwiseDivide(pcis->vec1_global,rhs,pcis->vec1_global);CHKERRQ(ierr); 1002 ierr = VecScatterBegin(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1003 ierr = VecScatterEnd(matis->rctx,pcis->vec1_global,pcis->vec2_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1004 ierr = VecScatterBegin(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1005 ierr = VecScatterEnd(matis->rctx,used_vec,pcis->vec1_N,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1006 ierr = ISGetLocalSize(dirIS,&dirsize);CHKERRQ(ierr); 1007 ierr = VecGetArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1008 ierr = VecGetArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 1009 ierr = ISGetIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1010 for (i=0; i<dirsize; i++) array_x[is_indices[i]] = array_diagonal[is_indices[i]]; 1011 ierr = ISRestoreIndices(dirIS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1012 ierr = VecRestoreArrayRead(pcis->vec2_N,&array_diagonal);CHKERRQ(ierr); 1013 ierr = VecRestoreArray(pcis->vec1_N,&array_x);CHKERRQ(ierr); 1014 ierr = VecScatterBegin(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1015 ierr = VecScatterEnd(matis->rctx,pcis->vec1_N,used_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1016 pcbddc->rhs_change = PETSC_TRUE; 1017 ierr = ISDestroy(&dirIS);CHKERRQ(ierr); 1018 } 1019 } 1020 1021 /* remove the computed solution or the initial guess from the rhs */ 1022 if (pcbddc->rhs_change || (ksp && pcbddc->ksp_guess_nonzero) ) { 1023 /* store the original rhs */ 1024 if (copy_rhs) { 1025 ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1026 copy_rhs = PETSC_FALSE; 1027 } 1028 pcbddc->rhs_change = PETSC_TRUE; 1029 ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 1030 ierr = MatMultAdd(pc->pmat,used_vec,rhs,rhs);CHKERRQ(ierr); 1031 ierr = VecScale(used_vec,-1.0);CHKERRQ(ierr); 1032 ierr = VecCopy(used_vec,pcbddc->temp_solution);CHKERRQ(ierr); 1033 if (ksp) { 1034 ierr = KSPSetInitialGuessNonzero(ksp,PETSC_FALSE);CHKERRQ(ierr); 1035 } 1036 } 1037 ierr = VecDestroy(&used_vec);CHKERRQ(ierr); 1038 1039 /* store partially computed solution and set initial guess */ 1040 if (x && pcbddc->use_exact_dirichlet_trick) { 1041 ierr = VecSet(x,0.0);CHKERRQ(ierr); 1042 ierr = VecScatterBegin(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1043 ierr = VecScatterEnd(pcis->global_to_D,rhs,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1044 ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 1045 ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1046 ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,x,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1047 if (ksp) { 1048 ierr = KSPSetInitialGuessNonzero(ksp,PETSC_TRUE);CHKERRQ(ierr); 1049 } 1050 } 1051 1052 if (pcbddc->ChangeOfBasisMatrix) { 1053 PCBDDCChange_ctx change_ctx; 1054 1055 /* get change ctx */ 1056 ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1057 1058 /* set current iteration matrix inside change context (change of basis has been already set into the ctx during PCSetUp) */ 1059 ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1060 ierr = PetscObjectReference((PetscObject)pc->mat);CHKERRQ(ierr); 1061 change_ctx->original_mat = pc->mat; 1062 1063 /* change iteration matrix */ 1064 ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1065 ierr = PetscObjectReference((PetscObject)pcbddc->new_global_mat);CHKERRQ(ierr); 1066 pc->mat = pcbddc->new_global_mat; 1067 1068 /* store the original rhs */ 1069 if (copy_rhs) { 1070 ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1071 copy_rhs = PETSC_FALSE; 1072 } 1073 1074 /* change rhs */ 1075 ierr = MatMultTranspose(change_ctx->global_change,rhs,pcis->vec1_global);CHKERRQ(ierr); 1076 ierr = VecCopy(pcis->vec1_global,rhs);CHKERRQ(ierr); 1077 pcbddc->rhs_change = PETSC_TRUE; 1078 } 1079 1080 /* remove nullspace if present */ 1081 if (ksp && x && pcbddc->NullSpace) { 1082 ierr = MatNullSpaceRemove(pcbddc->NullSpace,x);CHKERRQ(ierr); 1083 /* store the original rhs */ 1084 if (copy_rhs) { 1085 ierr = VecCopy(rhs,pcbddc->original_rhs);CHKERRQ(ierr); 1086 copy_rhs = PETSC_FALSE; 1087 } 1088 pcbddc->rhs_change = PETSC_TRUE; 1089 ierr = MatNullSpaceRemove(pcbddc->NullSpace,rhs);CHKERRQ(ierr); 1090 } 1091 PetscFunctionReturn(0); 1092 } 1093 1094 /* -------------------------------------------------------------------------- */ 1095 #undef __FUNCT__ 1096 #define __FUNCT__ "PCPostSolve_BDDC" 1097 /* -------------------------------------------------------------------------- */ 1098 /* 1099 PCPostSolve_BDDC - Changes the computed solution if a transformation of basis 1100 approach has been selected. Also, restores rhs to its original state. 1101 1102 Input Parameter: 1103 + pc - the preconditioner contex 1104 1105 Application Interface Routine: PCPostSolve() 1106 1107 Notes: 1108 The interface routine PCPostSolve() is not usually called directly by 1109 the user, but instead is called by KSPSolve(). 1110 */ 1111 static PetscErrorCode PCPostSolve_BDDC(PC pc, KSP ksp, Vec rhs, Vec x) 1112 { 1113 PetscErrorCode ierr; 1114 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1115 1116 PetscFunctionBegin; 1117 if (pcbddc->ChangeOfBasisMatrix) { 1118 PCBDDCChange_ctx change_ctx; 1119 1120 /* get change ctx */ 1121 ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1122 1123 /* restore iteration matrix */ 1124 ierr = MatDestroy(&pc->mat);CHKERRQ(ierr); 1125 ierr = PetscObjectReference((PetscObject)change_ctx->original_mat);CHKERRQ(ierr); 1126 pc->mat = change_ctx->original_mat; 1127 1128 /* get solution in original basis */ 1129 if (x) { 1130 PC_IS *pcis = (PC_IS*)(pc->data); 1131 ierr = MatMult(change_ctx->global_change,x,pcis->vec1_global);CHKERRQ(ierr); 1132 ierr = VecCopy(pcis->vec1_global,x);CHKERRQ(ierr); 1133 } 1134 } 1135 1136 /* add solution removed in presolve */ 1137 if (x && pcbddc->rhs_change) { 1138 ierr = VecAXPY(x,1.0,pcbddc->temp_solution);CHKERRQ(ierr); 1139 } 1140 1141 /* restore rhs to its original state */ 1142 if (rhs && pcbddc->rhs_change) { 1143 ierr = VecCopy(pcbddc->original_rhs,rhs);CHKERRQ(ierr); 1144 } 1145 pcbddc->rhs_change = PETSC_FALSE; 1146 1147 /* restore ksp guess state */ 1148 if (ksp) { 1149 ierr = KSPSetInitialGuessNonzero(ksp,pcbddc->ksp_guess_nonzero);CHKERRQ(ierr); 1150 } 1151 PetscFunctionReturn(0); 1152 } 1153 /* -------------------------------------------------------------------------- */ 1154 #undef __FUNCT__ 1155 #define __FUNCT__ "PCSetUp_BDDC" 1156 /* -------------------------------------------------------------------------- */ 1157 /* 1158 PCSetUp_BDDC - Prepares for the use of the BDDC preconditioner 1159 by setting data structures and options. 1160 1161 Input Parameter: 1162 + pc - the preconditioner context 1163 1164 Application Interface Routine: PCSetUp() 1165 1166 Notes: 1167 The interface routine PCSetUp() is not usually called directly by 1168 the user, but instead is called by PCApply() if necessary. 1169 */ 1170 PetscErrorCode PCSetUp_BDDC(PC pc) 1171 { 1172 PetscErrorCode ierr; 1173 PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1174 Mat_IS* matis; 1175 MatNullSpace nearnullspace; 1176 PetscInt nrows,ncols; 1177 PetscBool computetopography,computesolvers,computesubschurs; 1178 PetscBool computeconstraintsmatrix; 1179 PetscBool new_nearnullspace_provided,ismatis; 1180 1181 PetscFunctionBegin; 1182 ierr = PetscObjectTypeCompare((PetscObject)pc->pmat,MATIS,&ismatis);CHKERRQ(ierr); 1183 if (!ismatis) { 1184 SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_ARG_WRONG,"PCBDDC preconditioner requires matrix of type MATIS"); 1185 } 1186 ierr = MatGetSize(pc->pmat,&nrows,&ncols);CHKERRQ(ierr); 1187 if (nrows != ncols) { 1188 SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"PCBDDC preconditioner requires a square preconditioning matrix"); 1189 } 1190 matis = (Mat_IS*)pc->pmat->data; 1191 /* the following lines of code should be replaced by a better logic between PCIS, PCNN, PCBDDC and other future nonoverlapping preconditioners */ 1192 /* For BDDC we need to define a local "Neumann" problem different to that defined in PCISSetup 1193 Also, BDDC directly build the Dirichlet problem */ 1194 /* split work */ 1195 if (pc->setupcalled) { 1196 if (pc->flag == SAME_NONZERO_PATTERN) { 1197 computetopography = PETSC_FALSE; 1198 computesolvers = PETSC_TRUE; 1199 } else { /* DIFFERENT_NONZERO_PATTERN */ 1200 computetopography = PETSC_TRUE; 1201 computesolvers = PETSC_TRUE; 1202 } 1203 } else { 1204 computetopography = PETSC_TRUE; 1205 computesolvers = PETSC_TRUE; 1206 } 1207 if (pcbddc->recompute_topography) { 1208 computetopography = PETSC_TRUE; 1209 } 1210 computeconstraintsmatrix = PETSC_FALSE; 1211 if (pcbddc->adaptive_threshold > 0.0 && !pcbddc->use_deluxe_scaling) { 1212 SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute adaptive constraints without deluxe scaling. Rerun with -pc_bddc_use_deluxe_scaling"); 1213 } 1214 pcbddc->adaptive_selection = (PetscBool)(pcbddc->adaptive_threshold > 0.0 && pcbddc->use_deluxe_scaling); 1215 if (pcbddc->adaptive_selection) pcbddc->use_faces = PETSC_TRUE; 1216 1217 computesubschurs = (PetscBool)(pcbddc->adaptive_selection || pcbddc->use_deluxe_scaling); 1218 if (pcbddc->faster_deluxe && pcbddc->adaptive_selection && pcbddc->use_change_of_basis) { 1219 SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_SUP,"Cannot compute faster deluxe if adaptivity and change of basis are both requested. Rerun with -pc_bddc_deluxe_faster false"); 1220 } 1221 /* Get stdout for dbg */ 1222 if (pcbddc->dbg_flag) { 1223 if (!pcbddc->dbg_viewer) { 1224 pcbddc->dbg_viewer = PETSC_VIEWER_STDOUT_(PetscObjectComm((PetscObject)pc)); 1225 ierr = PetscViewerASCIISynchronizedAllow(pcbddc->dbg_viewer,PETSC_TRUE);CHKERRQ(ierr); 1226 } 1227 ierr = PetscViewerASCIIAddTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1228 } 1229 1230 if (pcbddc->user_ChangeOfBasisMatrix) { 1231 /* use_change_of_basis flag is used to automatically compute a change of basis from constraints */ 1232 pcbddc->use_change_of_basis = PETSC_FALSE; 1233 ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->user_ChangeOfBasisMatrix);CHKERRQ(ierr); 1234 } else { 1235 ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 1236 if (!pcbddc->saddle_point) { 1237 ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 1238 pcbddc->local_mat = matis->A; 1239 } else { 1240 PetscInt nz; 1241 PetscBool sorted; 1242 1243 ierr = ISDestroy(&pcbddc->zerodiag);CHKERRQ(ierr); 1244 /* should I also include nonzero pressures? */ 1245 ierr = MatFindZeroDiagonals(matis->A,&pcbddc->zerodiag);CHKERRQ(ierr); 1246 ierr = ISSorted(pcbddc->zerodiag,&sorted);CHKERRQ(ierr); 1247 if (!sorted) { 1248 ierr = ISSort(pcbddc->zerodiag);CHKERRQ(ierr); 1249 } 1250 if (!PetscGlobalRank) printf("ZERODIAG\n"); 1251 if (!PetscGlobalRank) ISView(pcbddc->zerodiag,NULL); 1252 ierr = ISGetLocalSize(pcbddc->zerodiag,&nz);CHKERRQ(ierr); 1253 if (nz) { 1254 IS zerodiagc; 1255 const PetscScalar *cB0_vals; 1256 PetscScalar *array; 1257 const PetscInt *idxs,*idxsc,*cB0_cols; 1258 PetscInt i,n,*nnz,cB0_ncol; 1259 1260 /* TODO: add check for shared dofs */ 1261 pcbddc->use_change_of_basis = PETSC_TRUE; 1262 ierr = MatGetLocalSize(matis->A,&n,NULL);CHKERRQ(ierr); 1263 ierr = ISComplement(pcbddc->zerodiag,0,n,&zerodiagc);CHKERRQ(ierr); 1264 ierr = ISGetIndices(pcbddc->zerodiag,&idxs);CHKERRQ(ierr); 1265 ierr = ISGetIndices(zerodiagc,&idxsc);CHKERRQ(ierr); 1266 /* local change of basis for pressures */ 1267 ierr = MatDestroy(&pcbddc->benign_change);CHKERRQ(ierr); 1268 ierr = MatCreate(PetscObjectComm((PetscObject)matis->A),&pcbddc->benign_change);CHKERRQ(ierr); 1269 ierr = MatSetType(pcbddc->benign_change,MATAIJ);CHKERRQ(ierr); 1270 ierr = MatSetSizes(pcbddc->benign_change,n,n,PETSC_DECIDE,PETSC_DECIDE);CHKERRQ(ierr); 1271 ierr = PetscMalloc1(n,&nnz);CHKERRQ(ierr); 1272 for (i=0;i<n-nz;i++) nnz[idxsc[i]] = 1; /* identity on velocities */ 1273 for (i=0;i<nz-1;i++) nnz[idxs[i]] = 2; /* change on pressures */ 1274 nnz[idxs[nz-1]] = nz; /* last local pressure dof: _0 set */ 1275 ierr = MatSeqAIJSetPreallocation(pcbddc->benign_change,0,nnz);CHKERRQ(ierr); 1276 ierr = PetscFree(nnz);CHKERRQ(ierr); 1277 /* set identity on velocities */ 1278 for (i=0;i<n-nz;i++) { 1279 ierr = MatSetValue(pcbddc->benign_change,idxsc[i],idxsc[i],1.,INSERT_VALUES);CHKERRQ(ierr); 1280 } 1281 /* set change on pressures */ 1282 for (i=0;i<nz-1;i++) { 1283 PetscScalar vals[2]; 1284 PetscInt cols[2]; 1285 1286 /* TODO: add quadrature */ 1287 cols[0] = idxs[i]; 1288 cols[1] = idxs[nz-1]; 1289 vals[0] = 1.; 1290 vals[1] = 1./nz; 1291 ierr = MatSetValues(pcbddc->benign_change,1,idxs+i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 1292 } 1293 ierr = PetscMalloc1(nz,&array);CHKERRQ(ierr); 1294 for (i=0;i<nz-1;i++) array[i] = -1.; 1295 array[nz-1] = 1./nz; 1296 ierr = MatSetValues(pcbddc->benign_change,1,idxs+nz-1,nz,idxs,array,INSERT_VALUES);CHKERRQ(ierr); 1297 ierr = PetscFree(array);CHKERRQ(ierr); 1298 ierr = MatAssemblyBegin(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1299 ierr = MatAssemblyEnd(pcbddc->benign_change,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1300 /* TODO: need optimization? */ 1301 ierr = MatPtAP(matis->A,pcbddc->benign_change,MAT_INITIAL_MATRIX,2.0,&pcbddc->local_mat);CHKERRQ(ierr); 1302 if (!PetscGlobalRank) printf("Local MAT\n"); 1303 if (!PetscGlobalRank) MatView(matis->A,NULL); 1304 if (!PetscGlobalRank) printf("Local pressure change\n"); 1305 if (!PetscGlobalRank) MatView(pcbddc->benign_change,NULL); 1306 if (!PetscGlobalRank) printf("Local AFTER CHANGE\n"); 1307 if (!PetscGlobalRank) MatView(pcbddc->local_mat,NULL); 1308 /* extract B_0 */ 1309 ierr = MatGetRow(pcbddc->local_mat,idxs[nz-1],&cB0_ncol,&cB0_cols,&cB0_vals);CHKERRQ(ierr); 1310 pcbddc->B0_ncol = cB0_ncol; 1311 ierr = PetscFree2(pcbddc->B0_cols,pcbddc->B0_vals);CHKERRQ(ierr); 1312 ierr = PetscMalloc2(cB0_ncol,&pcbddc->B0_cols,cB0_ncol,&pcbddc->B0_vals);CHKERRQ(ierr); 1313 ierr = PetscMemcpy(pcbddc->B0_cols,cB0_cols,cB0_ncol*sizeof(PetscInt));CHKERRQ(ierr); 1314 ierr = PetscMemcpy(pcbddc->B0_vals,cB0_vals,cB0_ncol*sizeof(PetscScalar));CHKERRQ(ierr); 1315 ierr = MatRestoreRow(pcbddc->local_mat,idxs[nz-1],&cB0_ncol,&cB0_cols,&cB0_vals);CHKERRQ(ierr); 1316 /* temporary remove rows and cols from local problem (added back at the end of setup) */ 1317 ierr = MatSetOption(pcbddc->local_mat,MAT_KEEP_NONZERO_PATTERN,PETSC_TRUE);CHKERRQ(ierr); 1318 ierr = MatZeroRowsColumns(pcbddc->local_mat,1,idxs+nz-1,1.,NULL,NULL);CHKERRQ(ierr); 1319 if (!PetscGlobalRank) printf("REMOVED AND B0\n"); 1320 if (!PetscGlobalRank) MatView(pcbddc->local_mat,NULL); 1321 if (!PetscGlobalRank) { 1322 for (i=0;i<pcbddc->B0_ncol;i++) printf("%d %f\n",pcbddc->B0_cols[i],pcbddc->B0_vals[i]); 1323 } 1324 ierr = MatAssemblyBegin(pcbddc->local_mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1325 ierr = MatAssemblyEnd(pcbddc->local_mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1326 ierr = ISRestoreIndices(pcbddc->zerodiag,&idxs);CHKERRQ(ierr); 1327 ierr = ISRestoreIndices(zerodiagc,&idxsc);CHKERRQ(ierr); 1328 ierr = ISDestroy(&zerodiagc);CHKERRQ(ierr); 1329 } else { /* this is unlikely to happen */ 1330 ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 1331 pcbddc->local_mat = matis->A; 1332 } 1333 } 1334 } 1335 1336 /* workaround for reals */ 1337 #if !defined(PETSC_USE_COMPLEX) 1338 if (matis->A->symmetric_set) { 1339 ierr = MatSetOption(pcbddc->local_mat,MAT_HERMITIAN,matis->A->symmetric);CHKERRQ(ierr); 1340 } 1341 #endif 1342 1343 /* Set up all the "iterative substructuring" common block without computing solvers */ 1344 { 1345 Mat temp_mat; 1346 1347 temp_mat = matis->A; 1348 matis->A = pcbddc->local_mat; 1349 ierr = PCISSetUp(pc,PETSC_FALSE);CHKERRQ(ierr); 1350 pcbddc->local_mat = matis->A; 1351 matis->A = temp_mat; 1352 if (!PetscGlobalRank) printf("IS_I_LOCAL\n"); 1353 if (!PetscGlobalRank) { PC_IS* pcis=(PC_IS*)(pc->data); ISView(pcis->is_I_local,NULL); } 1354 } 1355 1356 /* Analyze interface */ 1357 if (computetopography) { 1358 ierr = PCBDDCAnalyzeInterface(pc);CHKERRQ(ierr); 1359 computeconstraintsmatrix = PETSC_TRUE; 1360 } 1361 1362 /* Setup local dirichlet solver ksp_D and sub_schurs solvers */ 1363 if (computesolvers) { 1364 PCBDDCSubSchurs sub_schurs=pcbddc->sub_schurs; 1365 1366 if (computesubschurs && computetopography) { 1367 ierr = PCBDDCInitSubSchurs(pc);CHKERRQ(ierr); 1368 } 1369 if (sub_schurs->use_mumps) { 1370 if (computesubschurs) { 1371 ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1372 } 1373 ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1374 } else { 1375 ierr = PCBDDCSetUpLocalSolvers(pc,PETSC_TRUE,PETSC_FALSE);CHKERRQ(ierr); 1376 if (computesubschurs) { 1377 ierr = PCBDDCSetUpSubSchurs(pc);CHKERRQ(ierr); 1378 } 1379 } 1380 if (pcbddc->adaptive_selection) { 1381 ierr = PCBDDCAdaptiveSelection(pc);CHKERRQ(ierr); 1382 computeconstraintsmatrix = PETSC_TRUE; 1383 } 1384 } 1385 1386 /* infer if NullSpace object attached to Mat via MatSetNearNullSpace has changed */ 1387 new_nearnullspace_provided = PETSC_FALSE; 1388 ierr = MatGetNearNullSpace(pc->pmat,&nearnullspace);CHKERRQ(ierr); 1389 if (pcbddc->onearnullspace) { /* already used nearnullspace */ 1390 if (!nearnullspace) { /* near null space attached to mat has been destroyed */ 1391 new_nearnullspace_provided = PETSC_TRUE; 1392 } else { 1393 /* determine if the two nullspaces are different (should be lightweight) */ 1394 if (nearnullspace != pcbddc->onearnullspace) { 1395 new_nearnullspace_provided = PETSC_TRUE; 1396 } else { /* maybe the user has changed the content of the nearnullspace so check vectors ObjectStateId */ 1397 PetscInt i; 1398 const Vec *nearnullvecs; 1399 PetscObjectState state; 1400 PetscInt nnsp_size; 1401 ierr = MatNullSpaceGetVecs(nearnullspace,NULL,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1402 for (i=0;i<nnsp_size;i++) { 1403 ierr = PetscObjectStateGet((PetscObject)nearnullvecs[i],&state);CHKERRQ(ierr); 1404 if (pcbddc->onearnullvecs_state[i] != state) { 1405 new_nearnullspace_provided = PETSC_TRUE; 1406 break; 1407 } 1408 } 1409 } 1410 } 1411 } else { 1412 if (!nearnullspace) { /* both nearnullspaces are null */ 1413 new_nearnullspace_provided = PETSC_FALSE; 1414 } else { /* nearnullspace attached later */ 1415 new_nearnullspace_provided = PETSC_TRUE; 1416 } 1417 } 1418 1419 /* Setup constraints and related work vectors */ 1420 /* reset primal space flags */ 1421 pcbddc->new_primal_space = PETSC_FALSE; 1422 pcbddc->new_primal_space_local = PETSC_FALSE; 1423 if (computeconstraintsmatrix || new_nearnullspace_provided) { 1424 /* It also sets the primal space flags */ 1425 ierr = PCBDDCConstraintsSetUp(pc);CHKERRQ(ierr); 1426 /* Allocate needed local vectors (which depends on quantities defined during ConstraintsSetUp) */ 1427 ierr = PCBDDCSetUpLocalWorkVectors(pc);CHKERRQ(ierr); 1428 } 1429 1430 if (computesolvers || pcbddc->new_primal_space) { 1431 if (pcbddc->use_change_of_basis) { 1432 PC_IS *pcis = (PC_IS*)(pc->data); 1433 1434 ierr = PCBDDCComputeLocalMatrix(pc,pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 1435 /* get submatrices */ 1436 ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 1437 ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 1438 ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 1439 ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_BB);CHKERRQ(ierr); 1440 ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,MAT_INITIAL_MATRIX,&pcis->A_IB);CHKERRQ(ierr); 1441 ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,MAT_INITIAL_MATRIX,&pcis->A_BI);CHKERRQ(ierr); 1442 /* set flag in pcis to not reuse submatrices during PCISCreate */ 1443 pcis->reusesubmatrices = PETSC_FALSE; 1444 } else if (!pcbddc->user_ChangeOfBasisMatrix) { 1445 ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 1446 ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 1447 pcbddc->local_mat = matis->A; 1448 } 1449 /* SetUp coarse and local Neumann solvers */ 1450 ierr = PCBDDCSetUpSolvers(pc);CHKERRQ(ierr); 1451 /* SetUp Scaling operator */ 1452 ierr = PCBDDCScalingSetUp(pc);CHKERRQ(ierr); 1453 } 1454 1455 if (pcbddc->dbg_flag) { 1456 ierr = PetscViewerASCIISubtractTab(pcbddc->dbg_viewer,2*pcbddc->current_level);CHKERRQ(ierr); 1457 } 1458 PetscFunctionReturn(0); 1459 } 1460 1461 /* -------------------------------------------------------------------------- */ 1462 /* 1463 PCApply_BDDC - Applies the BDDC operator to a vector. 1464 1465 Input Parameters: 1466 + pc - the preconditioner context 1467 - r - input vector (global) 1468 1469 Output Parameter: 1470 . z - output vector (global) 1471 1472 Application Interface Routine: PCApply() 1473 */ 1474 #undef __FUNCT__ 1475 #define __FUNCT__ "PCApply_BDDC" 1476 PetscErrorCode PCApply_BDDC(PC pc,Vec r,Vec z) 1477 { 1478 PC_IS *pcis = (PC_IS*)(pc->data); 1479 PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1480 PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 1481 PetscErrorCode ierr; 1482 const PetscScalar one = 1.0; 1483 const PetscScalar m_one = -1.0; 1484 const PetscScalar zero = 0.0; 1485 1486 /* This code is similar to that provided in nn.c for PCNN 1487 NN interface preconditioner changed to BDDC 1488 Added support for M_3 preconditioner in the reference article (code is active if pcbddc->switch_static == PETSC_TRUE) */ 1489 1490 PetscFunctionBegin; 1491 if (!pcbddc->use_exact_dirichlet_trick) { 1492 ierr = VecCopy(r,z);CHKERRQ(ierr); 1493 /* First Dirichlet solve */ 1494 ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1495 ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1496 /* 1497 Assembling right hand side for BDDC operator 1498 - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1499 - pcis->vec1_B the interface part of the global vector z 1500 */ 1501 if (n_D) { 1502 ierr = KSPSolve(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 1503 ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 1504 if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1505 ierr = MatMult(pcis->A_BI,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1506 } else { 1507 ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1508 } 1509 ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1510 ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1511 ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1512 } else { 1513 if (pcbddc->switch_static) { 1514 ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1515 } 1516 ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1517 } 1518 1519 /* Apply interface preconditioner 1520 input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1521 ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_FALSE);CHKERRQ(ierr); 1522 1523 /* Apply transpose of partition of unity operator */ 1524 ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 1525 1526 /* Second Dirichlet solve and assembling of output */ 1527 ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1528 ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1529 if (n_B) { 1530 ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 1531 if (pcbddc->switch_static) { ierr = MatMultAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1532 } else if (pcbddc->switch_static) { 1533 ierr = MatMult(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1534 } 1535 ierr = KSPSolve(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1536 if (!pcbddc->use_exact_dirichlet_trick) { 1537 if (pcbddc->switch_static) { 1538 ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1539 } else { 1540 ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1541 } 1542 ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1543 ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1544 } else { 1545 if (pcbddc->switch_static) { 1546 ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1547 } else { 1548 ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1549 } 1550 ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1551 ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1552 } 1553 PetscFunctionReturn(0); 1554 } 1555 1556 /* -------------------------------------------------------------------------- */ 1557 /* 1558 PCApplyTranspose_BDDC - Applies the transpose of the BDDC operator to a vector. 1559 1560 Input Parameters: 1561 + pc - the preconditioner context 1562 - r - input vector (global) 1563 1564 Output Parameter: 1565 . z - output vector (global) 1566 1567 Application Interface Routine: PCApplyTranspose() 1568 */ 1569 #undef __FUNCT__ 1570 #define __FUNCT__ "PCApplyTranspose_BDDC" 1571 PetscErrorCode PCApplyTranspose_BDDC(PC pc,Vec r,Vec z) 1572 { 1573 PC_IS *pcis = (PC_IS*)(pc->data); 1574 PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1575 PetscInt n_B = pcis->n_B, n_D = pcis->n - n_B; 1576 PetscErrorCode ierr; 1577 const PetscScalar one = 1.0; 1578 const PetscScalar m_one = -1.0; 1579 const PetscScalar zero = 0.0; 1580 1581 PetscFunctionBegin; 1582 if (!pcbddc->use_exact_dirichlet_trick) { 1583 ierr = VecCopy(r,z);CHKERRQ(ierr); 1584 /* First Dirichlet solve */ 1585 ierr = VecScatterBegin(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1586 ierr = VecScatterEnd(pcis->global_to_D,r,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1587 /* 1588 Assembling right hand side for BDDC operator 1589 - pcis->vec1_D for the Dirichlet part (if needed, i.e. pcbddc->switch_static == PETSC_TRUE) 1590 - pcis->vec1_B the interface part of the global vector z 1591 */ 1592 if (n_D) { 1593 ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec1_D,pcis->vec2_D);CHKERRQ(ierr); 1594 ierr = VecScale(pcis->vec2_D,m_one);CHKERRQ(ierr); 1595 if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec2_D,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1596 ierr = MatMultTranspose(pcis->A_IB,pcis->vec2_D,pcis->vec1_B);CHKERRQ(ierr); 1597 } else { 1598 ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1599 } 1600 ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1601 ierr = VecScatterEnd(pcis->global_to_B,pcis->vec1_B,z,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1602 ierr = PCBDDCScalingRestriction(pc,z,pcis->vec1_B);CHKERRQ(ierr); 1603 } else { 1604 if (pcbddc->switch_static) { 1605 ierr = VecSet(pcis->vec1_D,zero);CHKERRQ(ierr); 1606 } 1607 ierr = PCBDDCScalingRestriction(pc,r,pcis->vec1_B);CHKERRQ(ierr); 1608 } 1609 1610 /* Apply interface preconditioner 1611 input/output vecs: pcis->vec1_B and pcis->vec1_D */ 1612 ierr = PCBDDCApplyInterfacePreconditioner(pc,PETSC_TRUE);CHKERRQ(ierr); 1613 1614 /* Apply transpose of partition of unity operator */ 1615 ierr = PCBDDCScalingExtension(pc,pcis->vec1_B,z);CHKERRQ(ierr); 1616 1617 /* Second Dirichlet solve and assembling of output */ 1618 ierr = VecScatterBegin(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1619 ierr = VecScatterEnd(pcis->global_to_B,z,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1620 if (n_B) { 1621 ierr = MatMultTranspose(pcis->A_BI,pcis->vec1_B,pcis->vec3_D);CHKERRQ(ierr); 1622 if (pcbddc->switch_static) { ierr = MatMultTransposeAdd(pcis->A_II,pcis->vec1_D,pcis->vec3_D,pcis->vec3_D);CHKERRQ(ierr); } 1623 } else if (pcbddc->switch_static) { 1624 ierr = MatMultTranspose(pcis->A_II,pcis->vec1_D,pcis->vec3_D);CHKERRQ(ierr); 1625 } 1626 ierr = KSPSolveTranspose(pcbddc->ksp_D,pcis->vec3_D,pcis->vec4_D);CHKERRQ(ierr); 1627 if (!pcbddc->use_exact_dirichlet_trick) { 1628 if (pcbddc->switch_static) { 1629 ierr = VecAXPBYPCZ(pcis->vec2_D,m_one,one,m_one,pcis->vec4_D,pcis->vec1_D);CHKERRQ(ierr); 1630 } else { 1631 ierr = VecAXPBY(pcis->vec2_D,m_one,m_one,pcis->vec4_D);CHKERRQ(ierr); 1632 } 1633 ierr = VecScatterBegin(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1634 ierr = VecScatterEnd(pcis->global_to_D,pcis->vec2_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1635 } else { 1636 if (pcbddc->switch_static) { 1637 ierr = VecAXPBY(pcis->vec4_D,one,m_one,pcis->vec1_D);CHKERRQ(ierr); 1638 } else { 1639 ierr = VecScale(pcis->vec4_D,m_one);CHKERRQ(ierr); 1640 } 1641 ierr = VecScatterBegin(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1642 ierr = VecScatterEnd(pcis->global_to_D,pcis->vec4_D,z,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1643 } 1644 PetscFunctionReturn(0); 1645 } 1646 /* -------------------------------------------------------------------------- */ 1647 1648 #undef __FUNCT__ 1649 #define __FUNCT__ "PCDestroy_BDDC" 1650 PetscErrorCode PCDestroy_BDDC(PC pc) 1651 { 1652 PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1653 PetscErrorCode ierr; 1654 1655 PetscFunctionBegin; 1656 /* free data created by PCIS */ 1657 ierr = PCISDestroy(pc);CHKERRQ(ierr); 1658 /* free BDDC custom data */ 1659 ierr = PCBDDCResetCustomization(pc);CHKERRQ(ierr); 1660 /* destroy objects related to topography */ 1661 ierr = PCBDDCResetTopography(pc);CHKERRQ(ierr); 1662 /* free allocated graph structure */ 1663 ierr = PetscFree(pcbddc->mat_graph);CHKERRQ(ierr); 1664 /* free allocated sub schurs structure */ 1665 ierr = PetscFree(pcbddc->sub_schurs);CHKERRQ(ierr); 1666 /* destroy objects for scaling operator */ 1667 ierr = PCBDDCScalingDestroy(pc);CHKERRQ(ierr); 1668 ierr = PetscFree(pcbddc->deluxe_ctx);CHKERRQ(ierr); 1669 /* free solvers stuff */ 1670 ierr = PCBDDCResetSolvers(pc);CHKERRQ(ierr); 1671 /* free global vectors needed in presolve */ 1672 ierr = VecDestroy(&pcbddc->temp_solution);CHKERRQ(ierr); 1673 ierr = VecDestroy(&pcbddc->original_rhs);CHKERRQ(ierr); 1674 /* free stuff for change of basis hooks */ 1675 if (pcbddc->new_global_mat) { 1676 PCBDDCChange_ctx change_ctx; 1677 ierr = MatShellGetContext(pcbddc->new_global_mat,&change_ctx);CHKERRQ(ierr); 1678 ierr = MatDestroy(&change_ctx->original_mat);CHKERRQ(ierr); 1679 ierr = MatDestroy(&change_ctx->global_change);CHKERRQ(ierr); 1680 ierr = VecDestroyVecs(2,&change_ctx->work);CHKERRQ(ierr); 1681 ierr = PetscFree(change_ctx);CHKERRQ(ierr); 1682 } 1683 ierr = MatDestroy(&pcbddc->new_global_mat);CHKERRQ(ierr); 1684 /* remove functions */ 1685 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",NULL);CHKERRQ(ierr); 1686 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",NULL);CHKERRQ(ierr); 1687 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",NULL);CHKERRQ(ierr); 1688 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",NULL);CHKERRQ(ierr); 1689 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",NULL);CHKERRQ(ierr); 1690 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",NULL);CHKERRQ(ierr); 1691 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",NULL);CHKERRQ(ierr); 1692 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 1693 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1694 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1695 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1696 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",NULL);CHKERRQ(ierr); 1697 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",NULL);CHKERRQ(ierr); 1698 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",NULL);CHKERRQ(ierr); 1699 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",NULL);CHKERRQ(ierr); 1700 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",NULL);CHKERRQ(ierr); 1701 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",NULL);CHKERRQ(ierr); 1702 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",NULL);CHKERRQ(ierr); 1703 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",NULL);CHKERRQ(ierr); 1704 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",NULL);CHKERRQ(ierr); 1705 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",NULL);CHKERRQ(ierr); 1706 /* Free the private data structure */ 1707 ierr = PetscFree(pc->data);CHKERRQ(ierr); 1708 PetscFunctionReturn(0); 1709 } 1710 /* -------------------------------------------------------------------------- */ 1711 1712 #undef __FUNCT__ 1713 #define __FUNCT__ "PCBDDCMatFETIDPGetRHS_BDDC" 1714 static PetscErrorCode PCBDDCMatFETIDPGetRHS_BDDC(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 1715 { 1716 FETIDPMat_ctx mat_ctx; 1717 Vec copy_standard_rhs; 1718 PC_IS* pcis; 1719 PC_BDDC* pcbddc; 1720 PetscErrorCode ierr; 1721 1722 PetscFunctionBegin; 1723 ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1724 pcis = (PC_IS*)mat_ctx->pc->data; 1725 pcbddc = (PC_BDDC*)mat_ctx->pc->data; 1726 1727 /* 1728 change of basis for physical rhs if needed 1729 It also changes the rhs in case of dirichlet boundaries 1730 TODO: better management when FETIDP will have its own class 1731 */ 1732 ierr = VecDuplicate(standard_rhs,©_standard_rhs);CHKERRQ(ierr); 1733 ierr = VecCopy(standard_rhs,copy_standard_rhs);CHKERRQ(ierr); 1734 ierr = PCPreSolve_BDDC(mat_ctx->pc,NULL,copy_standard_rhs,NULL);CHKERRQ(ierr); 1735 /* store vectors for computation of fetidp final solution */ 1736 ierr = VecScatterBegin(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1737 ierr = VecScatterEnd(pcis->global_to_D,copy_standard_rhs,mat_ctx->temp_solution_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1738 /* scale rhs since it should be unassembled */ 1739 /* TODO use counter scaling? (also below) */ 1740 ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1741 ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1742 /* Apply partition of unity */ 1743 ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1744 /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1745 if (!pcbddc->switch_static) { 1746 /* compute partially subassembled Schur complement right-hand side */ 1747 ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 1748 ierr = MatMult(pcis->A_BI,pcis->vec1_D,pcis->vec1_B);CHKERRQ(ierr); 1749 ierr = VecAXPY(mat_ctx->temp_solution_B,-1.0,pcis->vec1_B);CHKERRQ(ierr); 1750 ierr = VecSet(copy_standard_rhs,0.0);CHKERRQ(ierr); 1751 ierr = VecScatterBegin(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1752 ierr = VecScatterEnd(pcis->global_to_B,mat_ctx->temp_solution_B,copy_standard_rhs,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1753 /* ierr = PCBDDCScalingRestriction(mat_ctx->pc,copy_standard_rhs,mat_ctx->temp_solution_B);CHKERRQ(ierr); */ 1754 ierr = VecScatterBegin(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1755 ierr = VecScatterEnd(pcis->global_to_B,copy_standard_rhs,mat_ctx->temp_solution_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1756 ierr = VecPointwiseMult(mat_ctx->temp_solution_B,pcis->D,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1757 } 1758 ierr = VecDestroy(©_standard_rhs);CHKERRQ(ierr); 1759 /* BDDC rhs */ 1760 ierr = VecCopy(mat_ctx->temp_solution_B,pcis->vec1_B);CHKERRQ(ierr); 1761 if (pcbddc->switch_static) { 1762 ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 1763 } 1764 /* apply BDDC */ 1765 ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 1766 /* Application of B_delta and assembling of rhs for fetidp fluxes */ 1767 ierr = VecSet(fetidp_flux_rhs,0.0);CHKERRQ(ierr); 1768 ierr = MatMult(mat_ctx->B_delta,pcis->vec1_B,mat_ctx->lambda_local);CHKERRQ(ierr); 1769 ierr = VecScatterBegin(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1770 ierr = VecScatterEnd(mat_ctx->l2g_lambda,mat_ctx->lambda_local,fetidp_flux_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1771 PetscFunctionReturn(0); 1772 } 1773 1774 #undef __FUNCT__ 1775 #define __FUNCT__ "PCBDDCMatFETIDPGetRHS" 1776 /*@ 1777 PCBDDCMatFETIDPGetRHS - Compute the right-hand side for FETI-DP linear system using the physical right-hand side 1778 1779 Collective 1780 1781 Input Parameters: 1782 + fetidp_mat - the FETI-DP matrix object obtained by a call to PCBDDCCreateFETIDPOperators 1783 - standard_rhs - the right-hand side of the original linear system 1784 1785 Output Parameters: 1786 . fetidp_flux_rhs - the right-hand side for the FETI-DP linear system 1787 1788 Level: developer 1789 1790 Notes: 1791 1792 .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetSolution 1793 @*/ 1794 PetscErrorCode PCBDDCMatFETIDPGetRHS(Mat fetidp_mat, Vec standard_rhs, Vec fetidp_flux_rhs) 1795 { 1796 FETIDPMat_ctx mat_ctx; 1797 PetscErrorCode ierr; 1798 1799 PetscFunctionBegin; 1800 ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1801 ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetRHS_C",(Mat,Vec,Vec),(fetidp_mat,standard_rhs,fetidp_flux_rhs));CHKERRQ(ierr); 1802 PetscFunctionReturn(0); 1803 } 1804 /* -------------------------------------------------------------------------- */ 1805 1806 #undef __FUNCT__ 1807 #define __FUNCT__ "PCBDDCMatFETIDPGetSolution_BDDC" 1808 static PetscErrorCode PCBDDCMatFETIDPGetSolution_BDDC(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 1809 { 1810 FETIDPMat_ctx mat_ctx; 1811 PC_IS* pcis; 1812 PC_BDDC* pcbddc; 1813 PetscErrorCode ierr; 1814 1815 PetscFunctionBegin; 1816 ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1817 pcis = (PC_IS*)mat_ctx->pc->data; 1818 pcbddc = (PC_BDDC*)mat_ctx->pc->data; 1819 1820 /* apply B_delta^T */ 1821 ierr = VecScatterBegin(mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1822 ierr = VecScatterEnd (mat_ctx->l2g_lambda,fetidp_flux_sol,mat_ctx->lambda_local,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1823 ierr = MatMultTranspose(mat_ctx->B_delta,mat_ctx->lambda_local,pcis->vec1_B);CHKERRQ(ierr); 1824 /* compute rhs for BDDC application */ 1825 ierr = VecAYPX(pcis->vec1_B,-1.0,mat_ctx->temp_solution_B);CHKERRQ(ierr); 1826 if (pcbddc->switch_static) { 1827 ierr = VecCopy(mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 1828 } 1829 /* apply BDDC */ 1830 ierr = PCBDDCApplyInterfacePreconditioner(mat_ctx->pc,PETSC_FALSE);CHKERRQ(ierr); 1831 /* put values into standard global vector */ 1832 ierr = VecScatterBegin(pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1833 ierr = VecScatterEnd (pcis->global_to_B,pcis->vec1_B,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1834 if (!pcbddc->switch_static) { 1835 /* compute values into the interior if solved for the partially subassembled Schur complement */ 1836 ierr = MatMult(pcis->A_IB,pcis->vec1_B,pcis->vec1_D);CHKERRQ(ierr); 1837 ierr = VecAXPY(mat_ctx->temp_solution_D,-1.0,pcis->vec1_D);CHKERRQ(ierr); 1838 ierr = KSPSolve(pcbddc->ksp_D,mat_ctx->temp_solution_D,pcis->vec1_D);CHKERRQ(ierr); 1839 } 1840 ierr = VecScatterBegin(pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1841 ierr = VecScatterEnd (pcis->global_to_D,pcis->vec1_D,standard_sol,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1842 /* final change of basis if needed 1843 Is also sums the dirichlet part removed during RHS assembling */ 1844 ierr = PCPostSolve_BDDC(mat_ctx->pc,NULL,NULL,standard_sol);CHKERRQ(ierr); 1845 PetscFunctionReturn(0); 1846 } 1847 1848 #undef __FUNCT__ 1849 #define __FUNCT__ "PCBDDCMatFETIDPGetSolution" 1850 /*@ 1851 PCBDDCMatFETIDPGetSolution - Compute the physical solution using the solution of the FETI-DP linear system 1852 1853 Collective 1854 1855 Input Parameters: 1856 + fetidp_mat - the FETI-DP matrix obtained by a call to PCBDDCCreateFETIDPOperators 1857 - fetidp_flux_sol - the solution of the FETI-DP linear system 1858 1859 Output Parameters: 1860 . standard_sol - the solution defined on the physical domain 1861 1862 Level: developer 1863 1864 Notes: 1865 1866 .seealso: PCBDDC, PCBDDCCreateFETIDPOperators, PCBDDCMatFETIDPGetRHS 1867 @*/ 1868 PetscErrorCode PCBDDCMatFETIDPGetSolution(Mat fetidp_mat, Vec fetidp_flux_sol, Vec standard_sol) 1869 { 1870 FETIDPMat_ctx mat_ctx; 1871 PetscErrorCode ierr; 1872 1873 PetscFunctionBegin; 1874 ierr = MatShellGetContext(fetidp_mat,&mat_ctx);CHKERRQ(ierr); 1875 ierr = PetscTryMethod(mat_ctx->pc,"PCBDDCMatFETIDPGetSolution_C",(Mat,Vec,Vec),(fetidp_mat,fetidp_flux_sol,standard_sol));CHKERRQ(ierr); 1876 PetscFunctionReturn(0); 1877 } 1878 /* -------------------------------------------------------------------------- */ 1879 1880 extern PetscErrorCode FETIDPMatMult(Mat,Vec,Vec); 1881 extern PetscErrorCode FETIDPMatMultTranspose(Mat,Vec,Vec); 1882 extern PetscErrorCode PCBDDCDestroyFETIDPMat(Mat); 1883 extern PetscErrorCode FETIDPPCApply(PC,Vec,Vec); 1884 extern PetscErrorCode FETIDPPCApplyTranspose(PC,Vec,Vec); 1885 extern PetscErrorCode PCBDDCDestroyFETIDPPC(PC); 1886 1887 #undef __FUNCT__ 1888 #define __FUNCT__ "PCBDDCCreateFETIDPOperators_BDDC" 1889 static PetscErrorCode PCBDDCCreateFETIDPOperators_BDDC(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 1890 { 1891 1892 FETIDPMat_ctx fetidpmat_ctx; 1893 Mat newmat; 1894 FETIDPPC_ctx fetidppc_ctx; 1895 PC newpc; 1896 MPI_Comm comm; 1897 PetscErrorCode ierr; 1898 1899 PetscFunctionBegin; 1900 ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 1901 /* FETIDP linear matrix */ 1902 ierr = PCBDDCCreateFETIDPMatContext(pc,&fetidpmat_ctx);CHKERRQ(ierr); 1903 ierr = PCBDDCSetupFETIDPMatContext(fetidpmat_ctx);CHKERRQ(ierr); 1904 ierr = MatCreateShell(comm,PETSC_DECIDE,PETSC_DECIDE,fetidpmat_ctx->n_lambda,fetidpmat_ctx->n_lambda,fetidpmat_ctx,&newmat);CHKERRQ(ierr); 1905 ierr = MatShellSetOperation(newmat,MATOP_MULT,(void (*)(void))FETIDPMatMult);CHKERRQ(ierr); 1906 ierr = MatShellSetOperation(newmat,MATOP_MULT_TRANSPOSE,(void (*)(void))FETIDPMatMultTranspose);CHKERRQ(ierr); 1907 ierr = MatShellSetOperation(newmat,MATOP_DESTROY,(void (*)(void))PCBDDCDestroyFETIDPMat);CHKERRQ(ierr); 1908 ierr = MatSetUp(newmat);CHKERRQ(ierr); 1909 /* FETIDP preconditioner */ 1910 ierr = PCBDDCCreateFETIDPPCContext(pc,&fetidppc_ctx);CHKERRQ(ierr); 1911 ierr = PCBDDCSetupFETIDPPCContext(newmat,fetidppc_ctx);CHKERRQ(ierr); 1912 ierr = PCCreate(comm,&newpc);CHKERRQ(ierr); 1913 ierr = PCSetType(newpc,PCSHELL);CHKERRQ(ierr); 1914 ierr = PCShellSetContext(newpc,fetidppc_ctx);CHKERRQ(ierr); 1915 ierr = PCShellSetApply(newpc,FETIDPPCApply);CHKERRQ(ierr); 1916 ierr = PCShellSetApplyTranspose(newpc,FETIDPPCApplyTranspose);CHKERRQ(ierr); 1917 ierr = PCShellSetDestroy(newpc,PCBDDCDestroyFETIDPPC);CHKERRQ(ierr); 1918 ierr = PCSetOperators(newpc,newmat,newmat);CHKERRQ(ierr); 1919 ierr = PCSetUp(newpc);CHKERRQ(ierr); 1920 /* return pointers for objects created */ 1921 *fetidp_mat=newmat; 1922 *fetidp_pc=newpc; 1923 PetscFunctionReturn(0); 1924 } 1925 1926 #undef __FUNCT__ 1927 #define __FUNCT__ "PCBDDCCreateFETIDPOperators" 1928 /*@ 1929 PCBDDCCreateFETIDPOperators - Create FETI-DP operators 1930 1931 Collective 1932 1933 Input Parameters: 1934 . pc - the BDDC preconditioning context (setup should have been called before) 1935 1936 Output Parameters: 1937 + fetidp_mat - shell FETI-DP matrix object 1938 - fetidp_pc - shell Dirichlet preconditioner for FETI-DP matrix 1939 1940 Options Database Keys: 1941 . -fetidp_fullyredundant <false> - use or not a fully redundant set of Lagrange multipliers 1942 1943 Level: developer 1944 1945 Notes: 1946 Currently the only operations provided for FETI-DP matrix are MatMult and MatMultTranspose 1947 1948 .seealso: PCBDDC, PCBDDCMatFETIDPGetRHS, PCBDDCMatFETIDPGetSolution 1949 @*/ 1950 PetscErrorCode PCBDDCCreateFETIDPOperators(PC pc, Mat *fetidp_mat, PC *fetidp_pc) 1951 { 1952 PetscErrorCode ierr; 1953 1954 PetscFunctionBegin; 1955 PetscValidHeaderSpecific(pc,PC_CLASSID,1); 1956 if (pc->setupcalled) { 1957 ierr = PetscUseMethod(pc,"PCBDDCCreateFETIDPOperators_C",(PC,Mat*,PC*),(pc,fetidp_mat,fetidp_pc));CHKERRQ(ierr); 1958 } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"You must call PCSetup_BDDC() first \n"); 1959 PetscFunctionReturn(0); 1960 } 1961 /* -------------------------------------------------------------------------- */ 1962 /*MC 1963 PCBDDC - Balancing Domain Decomposition by Constraints. 1964 1965 An implementation of the BDDC preconditioner based on 1966 1967 .vb 1968 [1] C. R. Dohrmann. "An approximate BDDC preconditioner", Numerical Linear Algebra with Applications Volume 14, Issue 2, pages 149-168, March 2007 1969 [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 1970 [3] J. Mandel, B. Sousedik, C. R. Dohrmann. "Multispace and Multilevel BDDC", http://arxiv.org/abs/0712.3977 1971 [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 1972 .ve 1973 1974 The matrix to be preconditioned (Pmat) must be of type MATIS. 1975 1976 Currently works with MATIS matrices with local matrices of type MATSEQAIJ, MATSEQBAIJ or MATSEQSBAIJ, either with real or complex numbers. 1977 1978 It also works with unsymmetric and indefinite problems. 1979 1980 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. 1981 1982 Approximate local solvers are automatically adapted for singular linear problems (see [1]) if the user has provided the nullspace using PCBDDCSetNullSpace() 1983 1984 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() 1985 Additional information on dofs can be provided by using PCBDDCSetDofsSplitting(), PCBDDCSetDirichletBoundaries(), PCBDDCSetNeumannBoundaries(), and PCBDDCSetPrimalVerticesLocalIS() 1986 1987 Constraints can be customized by attaching a MatNullSpace object to the MATIS matrix via MatSetNearNullSpace(). Non-singular modes are retained via SVD. 1988 1989 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. 1990 User defined change of basis can be passed to PCBDDC by using PCBDDCSetChangeOfBasisMat() 1991 1992 The PETSc implementation also supports multilevel BDDC [3]. Coarse grids are partitioned using a MatPartitioning object. 1993 1994 Adaptive selection of primal constraints [4] is supported for SPD systems with high-contrast in the coefficients if MUMPS is present. Future versions of the code will also consider using MKL_PARDISO or PASTIX. 1995 1996 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. 1997 Deluxe scaling is not supported yet for FETI-DP. 1998 1999 Options Database Keys (some of them, run with -h for a complete list): 2000 2001 . -pc_bddc_use_vertices <true> - use or not vertices in primal space 2002 . -pc_bddc_use_edges <true> - use or not edges in primal space 2003 . -pc_bddc_use_faces <false> - use or not faces in primal space 2004 . -pc_bddc_symmetric <true> - symmetric computation of primal basis functions. Specify false for unsymmetric problems 2005 . -pc_bddc_use_change_of_basis <false> - use change of basis approach (on edges only) 2006 . -pc_bddc_use_change_on_faces <false> - use change of basis approach on faces if change of basis has been requested 2007 . -pc_bddc_switch_static <false> - switches from M_2 (default) to M_3 operator (see reference article [1]) 2008 . -pc_bddc_levels <0> - maximum number of levels for multilevel 2009 . -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) 2010 . -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) 2011 . -pc_bddc_use_deluxe_scaling <false> - use deluxe scaling 2012 . -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) 2013 . -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 installed) 2014 - -pc_bddc_check_level <0> - set verbosity level of debugging output 2015 2016 Options for Dirichlet, Neumann or coarse solver can be set with 2017 .vb 2018 -pc_bddc_dirichlet_ 2019 -pc_bddc_neumann_ 2020 -pc_bddc_coarse_ 2021 .ve 2022 e.g -pc_bddc_dirichlet_ksp_type richardson -pc_bddc_dirichlet_pc_type gamg. PCBDDC uses by default KPSPREONLY and PCLU. 2023 2024 When using a multilevel approach, solvers' options at the N-th level (N > 1) can be specified as 2025 .vb 2026 -pc_bddc_dirichlet_lN_ 2027 -pc_bddc_neumann_lN_ 2028 -pc_bddc_coarse_lN_ 2029 .ve 2030 Note that level number ranges from the finest (0) to the coarsest (N). 2031 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. 2032 .vb 2033 -pc_bddc_coarse_pc_bddc_adaptive_threshold 5 -pc_bddc_coarse_l1_pc_bddc_redistribute 3 2034 .ve 2035 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 2036 2037 Level: intermediate 2038 2039 Developer notes: 2040 2041 Contributed by Stefano Zampini 2042 2043 .seealso: PCCreate(), PCSetType(), PCType (for list of available types), PC, MATIS 2044 M*/ 2045 2046 #undef __FUNCT__ 2047 #define __FUNCT__ "PCCreate_BDDC" 2048 PETSC_EXTERN PetscErrorCode PCCreate_BDDC(PC pc) 2049 { 2050 PetscErrorCode ierr; 2051 PC_BDDC *pcbddc; 2052 2053 PetscFunctionBegin; 2054 /* Creates the private data structure for this preconditioner and attach it to the PC object. */ 2055 ierr = PetscNewLog(pc,&pcbddc);CHKERRQ(ierr); 2056 pc->data = (void*)pcbddc; 2057 2058 /* create PCIS data structure */ 2059 ierr = PCISCreate(pc);CHKERRQ(ierr); 2060 2061 /* BDDC customization */ 2062 pcbddc->use_local_adj = PETSC_TRUE; 2063 pcbddc->use_vertices = PETSC_TRUE; 2064 pcbddc->use_edges = PETSC_TRUE; 2065 pcbddc->use_faces = PETSC_FALSE; 2066 pcbddc->use_change_of_basis = PETSC_FALSE; 2067 pcbddc->use_change_on_faces = PETSC_FALSE; 2068 pcbddc->switch_static = PETSC_FALSE; 2069 pcbddc->use_nnsp_true = PETSC_FALSE; 2070 pcbddc->use_qr_single = PETSC_FALSE; 2071 pcbddc->symmetric_primal = PETSC_TRUE; 2072 pcbddc->saddle_point = PETSC_FALSE; 2073 pcbddc->dbg_flag = 0; 2074 /* private */ 2075 pcbddc->local_primal_size = 0; 2076 pcbddc->local_primal_size_cc = 0; 2077 pcbddc->local_primal_ref_node = 0; 2078 pcbddc->local_primal_ref_mult = 0; 2079 pcbddc->n_vertices = 0; 2080 pcbddc->primal_indices_local_idxs = 0; 2081 pcbddc->recompute_topography = PETSC_FALSE; 2082 pcbddc->coarse_size = -1; 2083 pcbddc->new_primal_space = PETSC_FALSE; 2084 pcbddc->new_primal_space_local = PETSC_FALSE; 2085 pcbddc->global_primal_indices = 0; 2086 pcbddc->onearnullspace = 0; 2087 pcbddc->onearnullvecs_state = 0; 2088 pcbddc->user_primal_vertices = 0; 2089 pcbddc->NullSpace = 0; 2090 pcbddc->temp_solution = 0; 2091 pcbddc->original_rhs = 0; 2092 pcbddc->local_mat = 0; 2093 pcbddc->ChangeOfBasisMatrix = 0; 2094 pcbddc->user_ChangeOfBasisMatrix = 0; 2095 pcbddc->new_global_mat = 0; 2096 pcbddc->coarse_vec = 0; 2097 pcbddc->coarse_ksp = 0; 2098 pcbddc->coarse_phi_B = 0; 2099 pcbddc->coarse_phi_D = 0; 2100 pcbddc->coarse_psi_B = 0; 2101 pcbddc->coarse_psi_D = 0; 2102 pcbddc->vec1_P = 0; 2103 pcbddc->vec1_R = 0; 2104 pcbddc->vec2_R = 0; 2105 pcbddc->local_auxmat1 = 0; 2106 pcbddc->local_auxmat2 = 0; 2107 pcbddc->R_to_B = 0; 2108 pcbddc->R_to_D = 0; 2109 pcbddc->ksp_D = 0; 2110 pcbddc->ksp_R = 0; 2111 pcbddc->NeumannBoundaries = 0; 2112 pcbddc->NeumannBoundariesLocal = 0; 2113 pcbddc->DirichletBoundaries = 0; 2114 pcbddc->DirichletBoundariesLocal = 0; 2115 pcbddc->user_provided_isfordofs = PETSC_FALSE; 2116 pcbddc->n_ISForDofs = 0; 2117 pcbddc->n_ISForDofsLocal = 0; 2118 pcbddc->ISForDofs = 0; 2119 pcbddc->ISForDofsLocal = 0; 2120 pcbddc->ConstraintMatrix = 0; 2121 pcbddc->use_exact_dirichlet_trick = PETSC_TRUE; 2122 pcbddc->coarse_loc_to_glob = 0; 2123 pcbddc->coarsening_ratio = 8; 2124 pcbddc->coarse_adj_red = 0; 2125 pcbddc->current_level = 0; 2126 pcbddc->max_levels = 0; 2127 pcbddc->use_coarse_estimates = PETSC_FALSE; 2128 pcbddc->redistribute_coarse = 0; 2129 pcbddc->coarse_subassembling = 0; 2130 pcbddc->coarse_subassembling_init = 0; 2131 2132 /* benign subspace trick */ 2133 pcbddc->zerodiag = 0; 2134 pcbddc->B0_ncol = 0; 2135 pcbddc->B0_cols = NULL; 2136 pcbddc->B0_vals = NULL; 2137 pcbddc->benign_change = 0; 2138 2139 /* create local graph structure */ 2140 ierr = PCBDDCGraphCreate(&pcbddc->mat_graph);CHKERRQ(ierr); 2141 2142 /* scaling */ 2143 pcbddc->work_scaling = 0; 2144 pcbddc->use_deluxe_scaling = PETSC_FALSE; 2145 pcbddc->faster_deluxe = PETSC_FALSE; 2146 2147 /* create sub schurs structure */ 2148 ierr = PCBDDCSubSchursCreate(&pcbddc->sub_schurs);CHKERRQ(ierr); 2149 pcbddc->sub_schurs_rebuild = PETSC_FALSE; 2150 pcbddc->sub_schurs_layers = -1; 2151 pcbddc->sub_schurs_use_useradj = PETSC_FALSE; 2152 2153 pcbddc->computed_rowadj = PETSC_FALSE; 2154 2155 /* adaptivity */ 2156 pcbddc->adaptive_threshold = 0.0; 2157 pcbddc->adaptive_nmax = 0; 2158 pcbddc->adaptive_nmin = 0; 2159 2160 /* function pointers */ 2161 pc->ops->apply = PCApply_BDDC; 2162 pc->ops->applytranspose = PCApplyTranspose_BDDC; 2163 pc->ops->setup = PCSetUp_BDDC; 2164 pc->ops->destroy = PCDestroy_BDDC; 2165 pc->ops->setfromoptions = PCSetFromOptions_BDDC; 2166 pc->ops->view = 0; 2167 pc->ops->applyrichardson = 0; 2168 pc->ops->applysymmetricleft = 0; 2169 pc->ops->applysymmetricright = 0; 2170 pc->ops->presolve = PCPreSolve_BDDC; 2171 pc->ops->postsolve = PCPostSolve_BDDC; 2172 2173 /* composing function */ 2174 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetChangeOfBasisMat_C",PCBDDCSetChangeOfBasisMat_BDDC);CHKERRQ(ierr); 2175 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetPrimalVerticesLocalIS_C",PCBDDCSetPrimalVerticesLocalIS_BDDC);CHKERRQ(ierr); 2176 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetCoarseningRatio_C",PCBDDCSetCoarseningRatio_BDDC);CHKERRQ(ierr); 2177 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevel_C",PCBDDCSetLevel_BDDC);CHKERRQ(ierr); 2178 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetUseExactDirichlet_C",PCBDDCSetUseExactDirichlet_BDDC);CHKERRQ(ierr); 2179 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLevels_C",PCBDDCSetLevels_BDDC);CHKERRQ(ierr); 2180 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNullSpace_C",PCBDDCSetNullSpace_BDDC);CHKERRQ(ierr); 2181 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundaries_C",PCBDDCSetDirichletBoundaries_BDDC);CHKERRQ(ierr); 2182 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDirichletBoundariesLocal_C",PCBDDCSetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2183 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundaries_C",PCBDDCSetNeumannBoundaries_BDDC);CHKERRQ(ierr); 2184 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetNeumannBoundariesLocal_C",PCBDDCSetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2185 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundaries_C",PCBDDCGetDirichletBoundaries_BDDC);CHKERRQ(ierr); 2186 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetDirichletBoundariesLocal_C",PCBDDCGetDirichletBoundariesLocal_BDDC);CHKERRQ(ierr); 2187 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundaries_C",PCBDDCGetNeumannBoundaries_BDDC);CHKERRQ(ierr); 2188 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCGetNeumannBoundariesLocal_C",PCBDDCGetNeumannBoundariesLocal_BDDC);CHKERRQ(ierr); 2189 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplitting_C",PCBDDCSetDofsSplitting_BDDC);CHKERRQ(ierr); 2190 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetDofsSplittingLocal_C",PCBDDCSetDofsSplittingLocal_BDDC);CHKERRQ(ierr); 2191 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCSetLocalAdjacencyGraph_C",PCBDDCSetLocalAdjacencyGraph_BDDC);CHKERRQ(ierr); 2192 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCCreateFETIDPOperators_C",PCBDDCCreateFETIDPOperators_BDDC);CHKERRQ(ierr); 2193 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetRHS_C",PCBDDCMatFETIDPGetRHS_BDDC);CHKERRQ(ierr); 2194 ierr = PetscObjectComposeFunction((PetscObject)pc,"PCBDDCMatFETIDPGetSolution_C",PCBDDCMatFETIDPGetSolution_BDDC);CHKERRQ(ierr); 2195 PetscFunctionReturn(0); 2196 } 2197 2198