User:Armin Wiebigke/sandbox

The two-tree broadcast (abbreviated 2tree-broadcast or 23-broadcast) is an algorithm to implement a broadcast communication pattern. The algorithm can also be adapted to perform a reduction or prefix sum.

Algorithm

A broadcast sends a message from a specified root processor to all other processors. Binary tree broadcasting uses a binary tree to model the communication between the processors. Each processor corresponds to one node in the tree, and the root processor is the root of the tree. To broadcast a message M, the root sends M to its two children (child nodes). Each processor waits until it receives M and then sends M to its children. Because leaves have no children, they don't have to send any messages. The broadcasting process can be pipelined by splitting the message into k blocks, which are then broadcasted consecutively. In such a binary tree, the leaves of the tree only receive data, but never send any data themselves. If the communication is bidirectional (full-duplex), meaning each processor can send a message and receive a message at the same time, the leaves only use halve of the available bandwidth.

Two-tree broadcast with seven processors, including the edge coloring. T₁ in red, T₂ in blue. The last processor is the root.

The idea of the two-tree broadcast is to use two binary trees T₁ and T₂ and communicate on both concurrently. The trees are constructed so that the interior nodes of one tree correspond to leaf nodes of the other tree. The data that has to be broadcasted is split into blocks of equal size. In each step of the algorithm, each processor receives one block and sends the previous block to one of its children in the tree in which it is an interior node. A schedule is needed so that no processor has to send or receive two messages in the same step. To create such a schedule, the edges of both trees are colored with 0 and 1 such that

• no processor is connected to its parent nodes in T₁ and T₂ using edges of the same color
• no processor is connected to its children nodes in T₁ or T₂ using edges of the same color.

Edges with color 0 are used in even steps, edges with color 1 are used in odd steps. This schedule allows each processor to send one message and receive one message in each step, fully utilizing the available bandwidth.
Assume that processor i wants to broadcasts a message. The two trees are constructed for the remaining processors. Processor i sends blocks alternating to the roots of the two trees, so each tree broadcasts one half of the message.

Analysis

(maybe different communication models)

Let p be the number of processes, numbered from 0 to p - 1.

Construction of the trees

Two-trees of size 6, 12, 7, 9 using mirroring (top) and shifting (bottom). T₁ in red, T₂ in blue.

Let h = ⌈log(p + 2)⌉. T₁ and T₂ can be constructed as trees of height h - 1, such that both trees form an in-order numbering of the processors, with the following method:

If p = 2^h - 2, T₁ is a complete binary tree of height h - 1 expect that the rightmost leaf is missing. Otherwise, T₁ consists of a complete binary tree of height h - 2 covering PEs [0, 2^h-1 - 2], a recursively constructed tree covering PEs [2^h-1, p - 1], and a root at PE 2^h-1 - 1 whose children are the roots of the left and the right subtree.

There are two ways to construct T₂. With shifting, T₂ is first constructed like T₁, except that it contains an additional processor. Then T₂ is shifted by one position to the left and the leftmost leaf is removed. With mirroring, T₂ is the mirror image of T₁ (with the mirror axis between processes ⁠p/2⁠-1 and ⁠p/2⁠). Mirroring only works for even p.

It can be proven that a coloring with the desired properties exists for all p. When mirroring is used to construct T₂, each processor can independently compute the color of its incident edges in O(log p) time.

Communication Time

Communication model: A messeage of size n has a communication time of α + βn. α represents the startup overhead to send the message, β represents the transmission time per data element.

Suppose the message of size m is split into 2k blocks. Each communication step takes time α + β⁠m/2k⁠. Let h=log p be the height of the communication structure with the root at processor i and the two trees below it. After 2h steps, the first data block has reached every node in both trees. Afterwards, each processor receives one block in every step until it received all blocks. The total number of steps is 2h + 2k resulting in a total communication time of (2h + 2k)(α + β⁠m/2k⁠). Using an optimal k = k^* = (⁠βmh/2α⁠)^½,the total communication time is βm + 2αlog p + √8αβmlog p

Comparison to similar algorithms

In a linear pipeline broadcast, the message is split into k blocks. In each step, each processor i receives one block from the processor i-1 and sends one block to the processor i+1. Linear pipeline has optimal throughput, but has a startup time in O(p). For large p, the O(log p) startup time of the two-tree broadcast is faster while the throuput is identical.

A binomial tree broadcast communicates along a binomial tree. Each process receives the message that is broadcasted (the root already has the message) and then sends the message to its children. A binomial tree broadcast has only half the startup time of the Two-Tree broadcast, but a factor of log(p) more communication. The binomial tree broadcast is faster than the two-tree broadcast for small messages, but slower for large messages.

Fibonacci trees of height one to five

A pipelined binary tree broadcast splits the message into k blocks and broadcasts the blocks consecutively over a binary tree. By using a Fibonacci tree instead of a simple complete binary tree, the startup latency can be reduced to αlog(p). A Fibonacci tree of height h consists of a root that has a a Fibonacci tree of height h-1 as its left child and a Fibonacci tree of h-2 as its right child. The pipelined Fibonacci tree broadcast has half the startup time of the two-tree broadcast, but also only half of the bandwidth. It is faster for small messages, while the two-tree broadcast is faster for large messages.

ESBT

If p is a power of two, there is an elegant optimal broadcasting algorithm [11] based on log p edge disjoint spanning binomial trees (ESBT) in a hypercube.

Usage for other communication primitives

Reduction

Two-tree reduction with seven processors. The last processor is the root. T₁ in red, T₂ in blue.

A reduction (MPI_Reduce) computes $\bigoplus _{i=0}^{p-1}M_{i}$ where M_i is a vector of length m originally available at processor i and $\bigoplus$ is a binary operation that is associative, but not necessarily commutative. The result is stored at a specified root processor r.

Assume that r = 0 or r = p-1. In this case the communication is identical to the broadcast, except that the communication direction is reversed. Each process receives two blocks from its children, reduces them with its own block, and sends the result to its parent. The root takes turns receiving blocks from the roots of T₁ and T₂ and reduces them with its own data. The communication time is the same as for the Broadcast and the amount of data reduced per processor is 2m.
If the reduce operation is commutative, the result can be achieved for any root by renumbering the processors. If the operation is not commutative and the root is not 0 or p-1, then 2βm is a lower bound for the communitation time. In this case the result is first computed with processor 0 as the root, and then the result is send to processor r.

Prefix sum

A prefix sum (MPI_Scan) computes $\bigoplus _{i=0}^{j}M_{i}$ for each processor j where M_i is a vector of length m originally available at processor i and $\bigoplus$ is a binary associative operation. Using an inorder binary tree, a prefix sum can be computed by first performing an up-phase in which each interior node computes a partial sum $\bigoplus _{i=l}^{r}M_{i}$ for left- and rightmost leaves l and r, followed by a down-phase in which prefixes of the form $\bigoplus _{i=0}^{l-1}M_{i}$ are sent down the tree and allow each processor to finish computing its prefix sum. The communication in the up-phase is equivalent to a reduction to processor 0 and communication in the down-phase is equivalent to a broadcast from the processor 0. The total communication time is about twice the communication time of the two-tree broadcast.