Empirical performance model-driven data layout optimization and library call selection for tensor contraction expressions

  • Authors:

  • Qingda Lu;Xiaoyang Gao;Sriram Krishnamoorthy;Gerald Baumgartner;J. Ramanujam;P. Sadayappan

  • Affiliations:

  • Department of Computer Science and Engineering, The Ohio State University, Columbus, OH43210, USA;Department of Computer Science and Engineering, The Ohio State University, Columbus, OH43210, USA;Department of Computer Science and Engineering, The Ohio State University, Columbus, OH43210, USA;Department of Computer Science, Louisiana State University, Baton Rouge, LA70803, USA;Department of Electrical and Computer Engineering, Louisiana State University, Baton Rouge, LA70803, USA;Department of Computer Science and Engineering, The Ohio State University, Columbus, OH43210, USA

  • Venue:
  • Journal of Parallel and Distributed Computing
  • Year:
  • 2012

Quantified Score

Hi-index 0.00

Visualization

Abstract

Empirical optimizers like ATLAS have been very effective in optimizing computational kernels in libraries. The best choice of parameters such as tile size and degree of loop unrolling is determined in ATLAS by executing different versions of the computation. In contrast, optimizing compilers use a model-driven approach to program transformation. While the model-driven approach of optimizing compilers is generally orders of magnitude faster than ATLAS-like library generators, its effectiveness can be limited by the accuracy of the performance models used. In this paper, we describe an approach where a class of computations is modeled in terms of constituent operations that are empirically measured, thereby allowing modeling of the overall execution time. The performance model with empirically determined cost components is used to select library calls and choose data layout transformations in the context of the Tensor Contraction Engine, a compiler for a high-level domain-specific language for expressing computational models in quantum chemistry. The effectiveness of the approach is demonstrated through experimental measurements on representative computations from quantum chemistry.