PCTELESCOPE#
Runs a KSP solver on a sub-communicator. MPI ranks not in the sub-communicator are idle during the solve.
Options Database Keys#
-pc_telescope_reduction_factor
- factor to reduce the communicator size by. e.g. with 64 MPI ranks and r=4, the new sub-communicator will have 64/4 = 16 ranks.-pc_telescope_ignore_dm - flag to indicate whether an attached DM should be ignored.
-pc_telescope_subcomm_type <interlaced,contiguous> - defines the selection of MPI ranks on the sub-communicator. see PetscSubcomm for more information.
-pc_telescope_ignore_kspcomputeoperators - flag to indicate whether
KSPSetComputeOperators()should be used on the sub-KSP.-pc_telescope_use_coarse_dm - flag to indicate whether the coarse
DMshould be used to define the sub-communicator.
Notes#
Assuming that the parent preconditioner PC is defined on a communicator c, this implementation
creates a child sub-communicator (c’) containing fewer MPI ranks than the original parent preconditioner PC.
The preconditioner is deemed telescopic as it only calls KSPSolve() on a single
sub-communicator, in contrast with PCREDUNDANT which calls KSPSolve() on N sub-communicators.
This means there will be MPI ranks which will be idle during the application of this preconditioner.
Additionally, in comparison with PCREDUNDANT, PCTELESCOPE can utilize an attached DM.
The default type of the sub KSP (the KSP defined on c’) is KSPPREONLY.
There are three setup mechanisms for PCTELESCOPE. Features support by each type are described below.
In the following, we will refer to the operators B and B’, these are the Bmat provided to the KSP on the
communicators c and c’ respectively.
[1] Default setup
The sub-communicator c’ is created via PetscSubcommCreate().
Explicitly defined nullspace and near nullspace vectors will be propagated from B to B’.
Currently there is no support define nullspaces via a user supplied method (e.g. as passed to MatNullSpaceSetFunction()).
No support is provided for KSPSetComputeOperators().
Currently there is no support for the flag -pc_use_amat.
[2] DM aware setup
If a DM is attached to the PC, it is re-partitioned on the sub-communicator c’.
c’ is created via PetscSubcommCreate().
Both the Bmat operator and the right hand side vector are permuted into the new DOF ordering defined by the re-partitioned DM.
Currently only support for re-partitioning a DMDA is provided.
Any explicitly defined nullspace or near nullspace vectors attached to the original Bmat operator (B) are extracted, re-partitioned and set on the re-partitioned Bmat operator (B’).
Currently there is no support define nullspaces via a user supplied method (e.g. as passed to MatNullSpaceSetFunction()).
Support is provided for KSPSetComputeOperators(). The user provided function and context is propagated to the sub KSP.
This is fragile since the user must ensure that their user context is valid for use on c’.
Currently there is no support for the flag -pc_use_amat.
[3] Coarse DM setup
If a DM (dmfine) is attached to the PC, dmfine is queried for a “coarse” DM (call this dmcoarse) via DMGetCoarseDM().
PCTELESCOPE will interpret the coarse DM as being defined on a sub-communicator of c.
The communicator associated with dmcoarse will define the c’ to be used within PCTELESCOPE.
PCTELESCOPE will check that c’ is in fact a sub-communicator of c. If it is not, an error will be reported.
The intention of this setup type is that PCTELESCOPE will use an existing (e.g. user defined) communicator hierarchy, say as would be
available with using multi-grid on unstructured meshes.
This setup will not use the command line options -pc_telescope_reduction_factor or -pc_telescope_subcomm_type.
Any explicitly defined nullspace or near nullspace vectors attached to the original Bmat operator (B) are extracted, scattered into the correct ordering consistent with dmcoarse and set on B’.
Currently there is no support define nullspaces via a user supplied method (e.g. as passed to MatNullSpaceSetFunction()).
There is no general method to permute field orderings, hence only KSPSetComputeOperators() is supported.
The user must use PetscObjectComposeFunction() with dmfine to define the method to scatter fields from dmfine to dmcoarse.
Propagation of the user context for KSPSetComputeOperators() on the sub KSP is attempted by querying the DM contexts associated with dmfine and dmcoarse. Alternatively, the user may use PetscObjectComposeFunction() with dmcoarse to define a method which will return the appropriate user context for KSPSetComputeOperators().
Currently there is no support for the flag -pc_use_amat.
This setup can be invoked by the option -pc_telescope_use_coarse_dm or by calling PCTelescopeSetUseCoarseDM(pc,PETSC_TRUE);
Further information about the user-provided methods required by this setup type are described here PCTelescopeSetUseCoarseDM().
Developer Notes#
During PCSetup(), the B operator is scattered onto c’.
Within PCApply(), the RHS vector (x) is scattered into a redundant vector, xred (defined on c’).
Then, KSPSolve() is executed on the c’ communicator.
The communicator used within the telescoping preconditioner is defined by a PetscSubcomm using the INTERLACED
creation routine by default (this can be changed with -pc_telescope_subcomm_type). We run the sub KSP on only the ranks within the communicator which have a color equal to zero.
The telescoping preconditioner is aware of nullspaces and near nullspaces which are attached to the B operator. In the case where B has a (near) nullspace attached, the (near) nullspace vectors are extracted from B and mapped into a new (near) nullspace, defined on the sub-communicator, which is attached to B’ (the B operator which was scattered to c’)
The telescoping preconditioner can re-partition an attached DM if it is a DMDA (2D or 3D -
support for 1D DMDAs is not provided). If a DMDA is found, a topologically equivalent DMDA is created on c’
and this new DM is attached the sub KSP. The design of telescope is such that it should be possible to extend support
for re-partitioning other to DM’s (e.g. DMPLEX). The user can supply a flag to ignore attached DMs.
Alternatively, user-provided re-partitioned DMs can be used via -pc_telescope_use_coarse_dm.
With the default setup mode, B’ is defined by fusing rows (in order) associated with MPI ranks common to c and c’.
When a DMDA is attached to the parent preconditioner, B’ is defined by: (i) performing a symmetric permutation of B
into the ordering defined by the DMDA on c’, (ii) extracting the local chunks via MatCreateSubMatrices(), (iii) fusing the
locally (sequential) matrices defined on the ranks common to c and c’ into B’ using MatCreateMPIMatConcatenateSeqMat()
Limitations/improvements include the following.
VecPlaceArray() could be used within PCApply() to improve efficiency and reduce memory usage.
A unified mechanism to query for user contexts as required by KSPSetComputeOperators() and MatNullSpaceSetFunction().
The symmetric permutation used when a DMDA is encountered is performed via explicitly assembling a permutation matrix P,
and performing P^T.A.P. Possibly it might be more efficient to use MatPermute(). We opted to use P^T.A.P as it appears
VecPermute() does not support the use case required here. By computing P, one can permute both the operator and RHS in a
consistent manner.
Mapping of vectors (default setup mode) is performed in the following way. Suppose the parent communicator size was 4, and we set a reduction factor of 2; this would give a comm size on c’ of 2. Using the interlaced creation routine, the ranks in c with color = 0 will be rank 0 and 2. We perform the scatter to the sub-communicator in the following way. [1] Given a vector x defined on communicator c
rank(c) local values of x
------- ----------------------------------------
0 [ 0.0, 1.0, 2.0, 3.0, 4.0, 5.0 ]
1 [ 6.0, 7.0, 8.0, 9.0, 10.0, 11.0 ]
2 [ 12.0, 13.0, 14.0, 15.0, 16.0, 17.0 ]
3 [ 18.0, 19.0, 20.0, 21.0, 22.0, 23.0 ]
scatter into xtmp defined also on comm c, so that we have the following values
rank(c) local values of xtmp
------- ----------------------------------------------------------------------------
0 [ 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0 ]
1 [ ]
2 [ 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0 ]
3 [ ]
The entries on rank 1 and 3 (ranks which do not have a color = 0 in c’) have no values
[2] Copy the values from ranks 0, 2 (indices with respect to comm c) into the vector xred which is defined on communicator c’. Ranks 0 and 2 are the only ranks in the subcomm which have a color = 0.
rank(c') local values of xred
-------- ----------------------------------------------------------------------------
0 [ 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0 ]
1 [ 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0 ]
Contributed by Dave May
Reference#
Dave A. May, Patrick Sanan, Karl Rupp, Matthew G. Knepley, and Barry F. Smith, “Extreme-Scale Multigrid Components within PETSc”. 2016. In Proceedings of the Platform for Advanced Scientific Computing Conference (PASC ‘16). DOI: 10.1145/2929908.2929913
See Also#
PCTelescopeGetKSP(), PCTelescopeGetDM(), PCTelescopeGetReductionFactor(), PCTelescopeSetReductionFactor(), PCTelescopeGetIgnoreDM(), PCTelescopeSetIgnoreDM(), PCREDUNDANT
Level#
advanced
Location#
src/ksp/pc/impls/telescope/telescope.c
Index of all PC routines
Table of Contents for all manual pages
Index of all manual pages