mathematics
and
computer
science

LANS

LANS Publications

"Toward Adjoinable MPI"

J. Utke, L. Hascoet, C. Hill, P. Hovland, and U. Naumann

Preprint ANL/MCS-P1530-0808

Preprint Version: [pdf]

Automatic di erentiation is the primary means of obtaining analytic
6 derivatives from a numerical model given as a computer program. Therefore, it is an essential productivity tool in numerous computational science and engineering domains. Computing gradients with the adjoint (also called reverse) mode via source transformation is a particularly beneficial but also challenging use of automatic differentiation. To date only ad hoc solutions for adjoint differentiation of MPI programs have been available, forcing automatic differentiation tool users to reason about parallel communication data
ow and dependencies and manually develop adjoint communication code. Using the communication graph as a model we characterize the principal problems of adjoining the most frequently used communication idioms. We propose solutions to cover these idioms and consider the consequences for the MPI implementation, the MPI user and MPI-aware program analysis. The MIT general circulation model serves as a use case to illustrate the viability of our approach.