The invention relates to parallel computing and more particularly to a method and system for optimizing the large vector global reduce operation among non power-of-two processes to enhance parallel computing performance.
In parallel computing, the global reduce operation is widely used for parallel applications. Data are selectively shared among processes in a group to minimize the communications being exchanged among the processes. The Message Passing Interface (MPI) standard defines several collective interfaces for the reduce operation, most notably MPI_REDUCE and MPI_ALLREDUCE. The global reduce operations could be expensive; therefore, efficient MPI_REDUCE and MPI_ALLREDUCE are important. One long-term profiling study demonstrated that the time spent by parallel applications in performing MPI_REDUCE and MPI_ALLREDUCE operations accounted for more than 40% of the time that the applications spent in all MPI functions.
A global reduce operation is collectively performed across all members (i.e., all processes) of a process group defined by the MPI communicator. For purposes of the instant invention, commutative reduce operations, which are most widely used, will be detailed. In a communicator group, each process has an equal length input vector of data. The global reduce operation combines all input vectors using the specified operation. Processes exchange data and perform local reduce operations to yield partial reduce results. Partial reduce results are exchanged and combined until complete reduce results are generated. In MPI_REDUCE, the full vector with the complete reduce result for the reduce operation returns at one process, known as the root. In MPI_ALLREDUCE, complete reduce results return at every process of the communicator group.
For a small message MPI_REDUCE the best known algorithm is the Minimum Spanning Tree (MST) algorithm in which the root of the MPI_REDUCE is the root of a process MST. A process on the tree first receives messages containing vectors from all of its children and then combines the received data with its own input vector. The process then sends the result vector to its parent process. In MPI_ALLREDUCE, one process is selected as the root. Once the root receives and combines vectors and gets the final result, it broadcasts the result to other member of the communicator group. The cost of the MST MPI_REDUCE algorithm can be modeled by the following if the MST is a binomial tree:
T=log(N)*(α+L*β+L*γ),
where;
α is the latency of each message
β is the communication cost per byte
γ is the local reduce cost per byte
N is the number of process is the communicator; and
L is the length of a process's input vector to the reduce operation.
Clearly for large messages, the MST algorithm is not efficient since every processor would have to perform the reduce operation on full vectors all the time, rather then taking advantage of the parallelism. Better algorithms have been developed for large message MPI_REDUCE and MPI_ALLREDUCE operations. One such algorithm is the Recursive Halving Recursive Doubling (RHRD) algorithm. In the RHRD algorithm for MPI_REDUCE, each process takes log(N) steps of computation and communication. The computation stage of the steps is preceded by a distribution stage during which the processes exchange data (i.e., vector information) with other processes. In the first step of the distribution stage, also referred to as the preparation stage, process i exchanges with process j half of their input vectors where j=(i ̂ mask). Symbol ̂ denotes a bitwise exclusive “OR” operation, and “mask” equals 1's binary representation. If i<j, process j sends the first half of its vector to process i and receives the second half of process i's vector. Process i combines the first half of the two vectors and process j combines the second half of the two vectors.
In the second step, process i and process j exchange half of their intermediate reduce results from the first step and combine the received data with the half that was not sent, where j=i ̂ mask and mask is 1's binary representation left-shifting 1 bit. At step k, process i and process j exchange half of their intermediate reduce results from step (k−1), and combine the received part with the half that was not sent, where j=i ̂ mask and mask is 1's binary representation left shifting (k−1) bits, if i<j, process i sends the second half of the intermediate result and receives the first half.
This procedure continues recursively, halving the size of exchanged and combined data at each step, for a total of log(N) steps. Each process owns 1/N of the result vector in the end: process 0 owns the first 1/N, process 1 owns the second 1/N, process i owns the (i−1)th 1/N and so on. In MPI_REDUCE, the root then performs a gather operation to gather the rest of the complete reduce results hack from other processes. The vectors are exchanged between processes by passing messages. Clearly the larger the vector, the larger the message and the higher the overhead and the greater the potential latency for message exchange.
In MPI_ALLREDUCE, an allgather step is performed instead of a gather step, in which every process gathers the complete reduce results from other processes. The cost of the RHRD algorithm can be modeled by:
T=2*log(N)*α+2*(N−1)/N*L*β+(N−1)/N*L*γ.
When the message size is large based on the larger vector length, the RHRD algorithm performs better than the MST algorithm, since the latency term “α” is small compared to the size of the other two terms in the algorithm and can essentially be ignored.
The RHRD algorithm applies only to “power of two” processes. Studies have shown that a reduce operation is widely performed on non-power of two processes as well. For non-power of two N′ processes, the prior art solution performs one extra preparation step at the beginning of the operation than the RHRD algorithm. In addition, for the MPI_ALLREDUCE case, another extra step is performed at the end of the operation. The beginning step is to exclude r processes from the algorithm, where r=N′−N, with N being the largest power of two less than N′. The first 2*r processes exchange input vectors. Of the first 2*r processes, those with even ranks send the second half of their vectors to their right neighbors and those with odd ranks send the first half of their vectors to their left neighbors. Once the exchange of vector information is completed, the first 2*r processes perform a local reduce operation. Those with odd ranks then send their local reduce results to their left neighbors and do not participate in the rest the algorithm. Other processes, N of them, follow the algorithm described previously for power of two processes. In MPI_ALLREDUCE, those processes excluded by the first step receive final results from their left neighbors in the final step.
The cost of the prior art approach on non-power of two processes can be modeled by:
T=(2+2*log(N))*α+(1+2*(N−1)/N)*L*β+(1/2+(N−1)/N)*L*γ. (1)
T=(3+2*log(N))*α+(2+2*(N−1)/N)*L*β+(1/2+(N−1)/N)*L*γ. (2)
The extra step or steps for non-power of two processes increase bandwidth requirements by more than 50% for MPI_REDUCE and 100% for MPI_ALLREDUCE. Clearly, therefore, it is desirable, and an object of the present invention, to provide an improved method of performing large message MPI_REDUCE/MPI_ALLREDUCE operations for non-power of two processes.
It is another object of the present invention to improve performance of large message MPI_REDUCE and MPI_ALLREDUCE operations by removing unnecessary data forwarding in the preparation step of the prior art algorithm.
Yet another object is to fully utilize interconnect bandwidth in the final step for MPI_ALLREDUCE by rearranging the flow of partial reduce results.
The foregoing and other objects are realized by the inventive system and method that optimizes reduce operations by consolidating the operation into a limited number of participating processes and then distributing the results back to all processes to optimize large message reduce operations on non power-of-two processes.
The method comprises the steps of dividing a group of processes into two subgroups, S0 and S1, performing paired exchange and local reduce operations at the processes in S0, such that each process in S0 gets a half vector of partial reduce results; consolidating the partial reduce results into a set of remaining processes, said set excluding a subset of processes, such that only power-of-two processes with a half vector of the partial results comprise the set of remaining processes; performing successive recursive halving and recursive doubling at the set of remaining processes until each process in the set of remaining process has a half vector of the complete result; and providing a full complete result at every process.
The invention will now be described in greater detail with, reference to the appended drawings wherein:
In accordance with the invention, in a first step, 201 of
Once the subgroups have been established, each process i of subgroup S0 exchanges half of its vector, comprising half of the data elements of the input vector, with process i+1 if i is even, or process i−1 if i is odd, during the preparation phase at step 202. Local reduce operations are performed on the half vectors at all of the S0 subgroup processes at step 203. Results of the local reduce operation are not, however, sent by processes with odd ranks to processes with even ranks, as had been done in the prior art processing.
Rather, at this point in the reduce operation, the subgrouping of processes dictates the computation and communication behavior of each process. Accordingly, the half vectors of the partial reduce results are consolidated into a set of remaining processes, where the set comprises X processes with X being a power-of-two integer and wherein the set comprises a plurality of processes from S0 and at least one process from S0.
If r is even, processes of S0 can be further organized into r/2 4-process subgroups ((4*i, 4*i+1, 4*i+2, 4*i+3)|i=0, 1, . . . , r/2−1). In each of the 4-process subgroups, process 4*i+2 sends its local reduce result to process 4*i. Process 4*i+3 sends its local reduce result to process 4*i+1. Process 4*i and process 4*i+1 perform local reduce operation on the received intermediate results and its own local reduce results at step 205 and move on to the next step. Process 4*i+2 and process 4*i+3, hereinafter referred to as the non-participating processes, are excluded from the next steps of the global reduce. Each process i of subgroup S1 exchanges half of its input vector with process i+1 if i is even, or process i−1 if i is odd, and performs local reduce operation at step 207. After this step, the remaining subset of processes, each with a half vector of a partial reduce results, participate in step 208, carrying out recursive halving until each gets (1/N)th of the complete result vector. Then the remaining set of processes perform the recursive doubling at step 209, such that for MPI_ALLREDUCE, each gets a half vector of the complete reduce result, or for MPI_REDUCE, two remaining processes, including the root, each gets a half vector of the complete result. For MPI_REDUCE, the full complete results can be provided to one process in one final step in which the other remaining process sends its half vector to the root. For MPI_ALLREDUCE, exchange of half vector results can then be done in successive steps, 210 and 212 to ensure that complete vector results are available at every process. At step 210, remaining processes that belong to S0, send their half results to the non-participating processes of S0: process 4*i sends to process 4*i+2, and process 4*i+1 sends to process 4*i+3, i=0, 1, . . . , r/2−1. Process i and i−1 of S1 exchange their results, i=2*r, 2*(r+1), . . . , N′−2. After step 210, each process in S0 has a half vector of the complete result and each process in S1 gets the full complete result. Finally, at step 212, processes in S0 perform an exchange, similar to what processes in S1 did at step 210, to get the full complete results.
In
At
If r is odd, the first 2*(r−1) processes of S0 can be organized into (r−1)/2 4-process subgroups, while the last two processes 2*r−2, 2*r−1 of S0 are regrouped into a subset with process 2*r, the first process of S1. Processes of subgroup S0, except for process 2*r−2 and 2*r−1, behave the same as in the case when r is even. Processes of subgroup S1, except for process 2*r, also behave the same as in the case when r is even. Process 2*r−1 of subgroup S1 sends its local reduce result to process 2*r. While receiving from process 2*r−1, process 2*r also sends the first half of its input vector to process 2*r−2. Process 2*r−2 and process 2*r then perform local reduce operations and move on to the next step. Process 2*r−1 is excluded from the rest of the global reduce as are the other non-participating processes.
Again, after this step, the remaining processes in the subset each have half vector partial reduce results and participate in the step 208 of recursive halving and steps 209 of recursive doubling, resulting in each process getting a half vector of the complete reduce results for MPI_ALLREDUCE, or in two regaining processes, including the root, each getting a half vector of the complete result for MPI_REDUCE. For MPI_REDUCE, the full complete results can be provided to one process in one final step in which the other process sends its half vector to the root. For MPI_ALLREDUCE, exchanging of half vector results can then be done in successive steps, 210 and 212, to ensure that full complete result vectors are available at every process. At step 210, remaining processes that belong to S0, except for process 2*r−2, send their half results to the non-participating processes of S0: process 4*i sends to process 4*i+2 and process 4*i+1 sends to process 4*i+3, i=0, 1, 2, . . . , r/2−2. Process i and i+1 of S1 exchange their results, i=2*r+1, 2*(r+1)+1, . . . , N′−2. Furthermore, process 2*r−2 sends its result to process 2*r, whereas process 2*r sends its result to process 2*r−1. Therefore, after step 210, each process in S0 has a half vector of the complete result; each process in S1 gets the full complete result. Processes in S0 then, at step 212, perform an exchange similar to what processes in S1 did in step 210 to get the complete final results.
In
At this point, process 6 has complete results for the full vector but all other processes still only have complete results for a half vector. For an MPI_ALLREDUCE, as shown at
The benefit of the inventive approach is two-fold. In the prior art approach, local reduce results of the preparation step are sent by odd rank processes to even rank processes of S0, only to be again forwarded in the next step. Those unnecessary data forwards are avoided by the new approach. Further, processes of the S0 subgroup only send or receive a half vector instead of sending and receiving a full vector in the step following the preparation step. The current approach will put less of a burden on adaptors and alleviate the serialization at CPU and memory. Further, as detailed above, the last two steps for MPI_ALLREDUCE on N′ processes reduce bandwidth requirements.
The prior art approach takes 2+2*log(N) steps, numbered from 0 to 2*log(N)+1. Before step 2*log(N), each of N/2 processes has the first half of the final complete reduce result, and the other N/2 processes ach has the second half of the final complete reduce result. Those processes can be referred to as subgroups S2 and S3. The other r processes (subgroup S4) were excluded after the first two steps and do not have any partial results. A one-to-one correspondence can be formed between processes of S2 and S3. With the prior art approach, the process of S2 and its partner in S3 exchange half of the final result during step 2*log(N). After this step, each N processes of S2 and S3 has the entire final reduce result. The first r of those N processes then sends the entire final result to processes of S4 in step 2*log(N)+1.
Under the presented approach as illustrated in
On N′ processes, with the presented approach, the cost of MPI_REDUCE is:
T=(2+2*log(N))*α+(1/2+2*(N−1)/N)*L*β+(1/2+(N−1)/N)*L*γ. (3)
and MPI_ALLREDUCE time is:
T=(3+2*log(N))*α+(1+2*(N−1)/N)*L*β+(1/2+(N−1)/N)*L*γ. (4)
Compared to the prior approaches, modeled by equations (1) and (2) above, the bandwidth requirement drops by more than 16% and 25%, respectively, for MPI_REDUCE and MPI_ALLREDUCE operations.
The invention has been described with reference to specific embodiments and processes for purposes of illustration and explanation and is not intended to be an exhaustive description. Modifications will be apparent to those of ordinary skill in the art and should be understood to be within the spirit and scope of the appended claims.