Skip to content

Commit

Permalink
Merge pull request #12642 from burlen/coll_tuned_doc_tuning
Browse files Browse the repository at this point in the history
improve documentation for the coll tuned component
  • Loading branch information
jsquyres authored Jul 10, 2024
2 parents b8bd82b + 55c9dda commit ca33c7f
Show file tree
Hide file tree
Showing 2 changed files with 388 additions and 0 deletions.
387 changes: 387 additions & 0 deletions docs/tuning-apps/coll-tuned.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,387 @@
Tuning Collectives
==================

Open MPI's ``coll`` framework provides a number of components implementing
collective communication, including: ``han``, ``libnbc``, ``self``, ``ucc`` ``base``,
``hcoll``, ``sync``, ``xhc``, ``accelerator``, ``basic``, ``ftagree``, ``inter``, ``portals4``,
and ``tuned``. Some of these components may not be available depending on how
Open MPI was compiled and what hardware is available on the system. A run-time
decision based on each component's self reported priority, selects which
component will be used. These priorities may be adjusted on the command line
or with any of the other usual ways of setting MCA variables, giving us a way
to influence or override component selection. In the end, which of the
available components is selected depends on a number of factors such as the
underlying hardware and the whether or not a specific collective is provided by
the component as not all components implement all collectives. However, there
is always a fallback ``base`` component that steps in and takes over when another
component fails to provide an implementation.

The remainder of this section describes the tuning options available in the
``tuned`` collective component.

Fixed, Forced, and Dynamic Decisions
------------------------------------
Open MPI's ``tuned`` collective component has three modes of operation, *fixed
decision*, *forced algorithm*, and *dynamic decision*. Since different
collective algorithms perform better in different situations, the purpose of
these modes is, when a collective is called, to select an appropriate
algorithm.

In the *fixed decision* mode a decision tree, essentially a large set of nested
if-then-else-if blocks with baked in comm and message size thresholds derived
by measuring performance on existing clusters, selects one of a number of
available algorithms for the specific collective being called. *Fixed
decision* is the default. The *fixed decision* mode can work well if your
cluster hardware is similar to the systems profiled when constructing the
decision tree. However, in some cases the *fixed decision* rules can yield
less than ideal performance. For instance, when your hardware differs
substantially from that which was used to derive the fixed rules decision
tree.

In the *dynamic decision* mode a user can provide a set of rules encoded in an
ASCII file that tell the ``tuned`` component which algorithm should be for one or
more collectives as a function of communicator and message size. For the
collectives which rules are not provided the *fixed decision* mode is used.
More detail about the dynamic decision mode and rules file can be found in
section :ref:`Rules File<RulesFile>`.

In the *forced algorithm* mode one can specify an algorithm for a specific
collective on the command line. This is discussed in detail in section
:ref:`Forcing an Algorithm<ForceAlg>`.

.. _ForceAlg:

Forcing an Algorithm
--------------------

The simplest method of tuning is to force the ``tuned`` collective component to
use a specific algorithm for all calls to a specific collective. This can be
done on the command line, or by any of the other usual ways of setting MCA
variables. For example, the following command line sets the algorithm to
linear for all MPI_Alltoall calls in the entire run.

.. code-block:: sh
shell$ mpirun ... --mca coll_tuned_use_dynamic_rules 1 \
--mca coll_tuned_alltoall_algorithm 1 ...
When an algorithm is selected this way, the *fixed decision* rules associated
with that collective are short circuited. This is an easy way to select a
collective algorithm and is most effective when there is a single communicator
and message sizes are static with in the run. On the other hand, when the
message size for a given collective varies during the run, or there are
multiple communicators of different sizes, the effectiveness of this approach
may be limited. The :ref:`rules file <RulesFile>` provides a more comprehensive
and flexible approach to tuning. For collectives where an algorithm is not
forced on the command line, the *fixed decision* rules are used.

.. _RulesFile:

Dynamic Decisions and the Rules File
------------------------------------

Given that the best choice of algorithm for a given collective depends on a
number of factors only known at run time, and that some of these factors may
vary with in a run, setting an algorithm on the command line often is an
ineffective means of tuning. The rules file provides a means of choosing
an algorithm at run time based on communicator and message size. The rules
file can be specified on the command line, or the other usual ways to set MCA
variables. The file is parsed during initialization and takes affect there
after.

.. code-block:: sh
shell$ mpirun ... --mca coll_tuned_use_dynamic_rules 1 \
--mca coll_tuned_dynamic_rules_filename /path/to/my_rules.conf ...
The loaded set of rules then are used to select the algorithm
to use based on the collective, the communicator size, and the message size.
Collectives for which rules have not be specified in the file will make use of
the *fixed decision* rules as usual.

Dynamic tuning files are organized in this format:

.. code-block:: sh
:linenos:
1 # Number of collectives
1 # Collective ID
1 # Number of comm sizes
2 # Comm size
2 # Number of message sizes
0 1 0 0 # Message size 0, algorithm 1, topo and segmentation at 0
1024 2 0 0 # Message size 1024, algorithm 2, topo and segmentation at 0
The rules file effectively defines, for one or more collectives, a function of
two variables, which given communicator and message size, returns an algorithm
id to use for the call. This mechanism allows one to specify for each
collective, an algorithm for any number of ranges of message and communicator
sizes. As communicators are constructed, a search of the rules table is made
using the communicator size to select a set of message size rules to be used
with that communicator. Later as the collective is invoked, a search of the
message size rules associated with the communicator is made. The rule with the
nearest (less than) matching message size specifies the algorithm that is used.
The actual definition of *message size* is dependent on the collective in
question, see the section on :ref:`collectives and algorithms<CollectivesAndAlgorithms>`
for details.

One may provide rules for as many collectives, communicator sizes, and message
sizes as desired. Simply repeat the sections as needed and adjust the relevant
count parameters. One must always provide a rule for message size of zero.
Message size rules are expected in ascending order. The last two parameters in
the rule may or may not be used and have different meaning depending on the
collective and algorithm. As of writing not all of the relevant control
parameters can be set by the rules file (See issue #12589).

.. _CollectivesAndAlgorithms:

Collectives and their Algorithms
--------------------------------
The following table lists the collectives
implemented by the ``tuned`` collective component along with the enumeration value identifying it.
It is this value that must be used in the rules file when specifying a set of rules.
The definition of *message size* is dependent on the collective and is given in the table.
Tables describing the algorithms available for each collective, and there identifiers, are linked.

.. csv-table:: Collectives
:header: "Collective", "Id", "Message Size"
:widths: 20, 10, 65

:ref:`Allgather<Allgather>`, 0, "datatype size * comm size * number of elements in send buffer"
:ref:`Allgatherv<Allgatherv>`, 1, "datatype size * sum of number of elements that are to be received from each process (sum of recvcounts)"
:ref:`Allreduce<Allreduce>`, 2, "datatype size * number of elements in send buffer"
:ref:`Alltoall<Alltoall>`, 3, "datatype size * comm size * number of elements to send to each process"
:ref:`Alltoallv<Alltoallv>`, 4, "not used"
:ref:`Barrier<Barrier>`, 6, "not used"
:ref:`Bcast<Bcast>`, 7, "datatype size * number of entries in buffer"
:ref:`Exscan<Exscan>`, 8, "datatype size * comm size"
:ref:`Gather<Gather>`, 9, "datatype size * comm size * number of elements in send buffer"
:ref:`Reduce<Reduce>`, 11, "datatype size * number of elements in send buffer"
:ref:`Reduce_scatter<Reduce_scatter>`, 12, "datatype size * sum of number of elements in result distributed to each process (sum of recvcounts)"
:ref:`Reduce_scatter_block<Reduce_scatter_block>`, 13, "datatype size * comm size * element count per block"
:ref:`Scan<Scan>`, 14, "datatype size * comm size"
:ref:`Scatter<Scatter>`, 15, "datatype size * number of elements in send buffer"

.. _Allgather:

Allgather (Id=0)
~~~~~~~~~~~~~~~~

.. csv-table:: Allgather Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "linear", "..."
2, "bruck-k-fanout", "..."
3, "recursive_doubling", "..."
4, "ring", "..."
5, "neighbor", "..."
6, "two_proc", "..."
7, "sparbit", "..."
8, "direct-messaging", "..."

.. _Allgatherv:

Allgatherv (Id=1)
~~~~~~~~~~~~~~~~~

.. csv-table:: Allgatherv Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "default", "..."
2, "bruck", "..."
3, "ring", "..."
4, "neighbor", "..."
5, "two_proc", "..."
6, "sparbit", "..."

.. _Allreduce:

Allreduce (Id=2)
~~~~~~~~~~~~~~~~

.. csv-table:: Allreduce Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "basic_linear", "..."
2, "nonoverlapping", "..."
3, "recursive_doubling", "..."
4, "ring", "..."
5, "segmented_ring", "..."
6, "rabenseifner", "..."
7, "allgather_reduce", "..."

.. _Alltoall:

Alltoall (Id=3)
~~~~~~~~~~~~~~~

.. csv-table:: Alltoall Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "linear", "Launches all non-blocking send/recv pairs and wait for them to complete."
2, "pairwise", "For comm size P, implemented as P rounds of blocking MPI_Sendrecv"
3, "modified_bruck", "An algorithm exploiting network packet quantization to achieve O(log) time complexity. Typically best for very small message sizes."
4, "linear_sync", "Keep N non-blocking MPI_Isend/Irecv pairs in flight at all times. N is set by the coll_tuned_alltoall_max_requests MCA variable."
5, "two_proc", "An implementation tailored for alltoall between 2 ranks, otherwise it is not used."

.. _Alltoallv:

Alltoallv (Id=4)
~~~~~~~~~~~~~~~~

.. csv-table:: Alltoallv Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "basic_linear", "..."
2, "pairwise", "..."

.. _Barrier:

Barrier (Id=6)
~~~~~~~~~~~~~~

.. csv-table:: Barrier Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "linear", "..."
2, "double_ring", "..."
3, "recursive_doubling", "..."
4, "bruck", "..."
5, "two_proc", "..."
6, "tree", "..."

.. _Bcast:

Bcast (Id=7)
~~~~~~~~~~~~

.. csv-table:: Bcast Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "basic_linear", "..."
2, "chain", "..."
3, "pipeline", "..."
4, "split_binary_tree", "..."
5, "binary_tree", "..."
6, "binomial", "..."
7, "knomial", "..."
8, "scatter_allgather", "..."
9, "scatter_allgather_ring", "..."

.. _Exscan:

Exscan (Id=8)
~~~~~~~~~~~~~

.. csv-table:: Exscan Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "linear", "..."
2, "recursive_doubling", "..."

.. _Gather:

Gather (Id=9)
~~~~~~~~~~~~~

.. csv-table:: Gather Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "basic_linear", "..."
2, "binomial", "..."
3, "linear_sync", "..."

.. _Reduce:

Reduce (Id=11)
~~~~~~~~~~~~~~

.. csv-table:: Reduce Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "linear", "..."
2, "chain", "..."
3, "pipeline", "..."
4, "binary", "..."
5, "binomial", "..."
6, "in-order_binary", "..."
7, "rabenseifner", "..."
8, "knomial", "..."

.. _Reduce_scatter:

Reduce_scatter (Id=12)
~~~~~~~~~~~~~~~~~~~~~~

.. csv-table:: Reduce_scatter Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "non-overlapping", "..."
2, "recursive_halving", "..."
3, "ring", "..."
4, "butterfly", "..."

.. _Reduce_scatter_block:

Reduce_scatter_block (Id=13)
~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. csv-table:: Reduce_scatter_block Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "basic_linear", "..."
2, "recursive_doubling", "..."
3, "recursive_halving", "..."
4, "butterfly", "..."

.. _Scan:

Scan (Id=14)
~~~~~~~~~~~~

.. csv-table:: Scan Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "linear", "..."
2, "recursive_doubling", "..."

.. _Scatter:

Scatter (Id=15)
~~~~~~~~~~~~~~~

.. csv-table:: Scatter Algorithms
:header: "Id", "Name", "Description"
:widths: 10, 25, 60

0, "ignore", "Use fixed rules"
1, "basic_linear", "..."
2, "binomial", "..."
3, "linear_nb", "..."

1 change: 1 addition & 0 deletions docs/tuning-apps/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -16,5 +16,6 @@ components that can be tuned to affect behavior at run time.
large-clusters/index
affinity
mpi-io/index
coll-tuned
benchmarking
heterogeneity

0 comments on commit ca33c7f

Please sign in to comment.