Copyright Notice:

The documents distributed by this server have been provided by the contributing authors as a means to ensure timely dissemination of scholarly and technical work on a noncommercial basis. Copyright and all rights therein are maintained by the authors or by other copyright holders, notwithstanding that they have offered their works here electronically. It is understood that all persons copying this information will adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.

Publications of SPCL

L. Truemper, T. Ben-Nun, P. Schaad, A. Calotoiu, T. Hoefler:

 Performance Embeddings: A Similarity-Based Transfer Tuning Approach to Performance Optimization

(Jun. 2023)

Publisher Reference

Abstract

Performance optimization is an increasingly challenging but often repetitive task. While each platform has its quirks, the underlying code transformations rely on data movement and computational characteristics that recur across applications. This paper proposes to leverage those similarities by constructing an embedding space for subprograms. The continuous space captures both static and dynamic properties of loop nests via symbolic code analysis and performance profiling, respectively. Performance embeddings enable direct knowledge transfer of performance tuning between applications, which can result from autotuning or tailored improvements. We demonstrate this transfer tuning approach on case studies in deep neural networks, dense and sparse linear algebra compositions, and numerical weather prediction stencils. Transfer tuning reduces the search complexity by up to four orders of magnitude and outperforms the MKL library in sparse-dense matrix multiplication. The results exhibit clear correspondences between program characteristics and optimizations, outperforming prior specialized state-of-the-art approaches and generalizing beyond their capabilities.

Documents

download article:
access preprint on arxiv:
 

BibTeX

@inproceedings{,
  author={Lukas Truemper and Tal Ben-Nun and Philipp Schaad and Alexandru Calotoiu and Torsten Hoefler},
  title={{Performance Embeddings: A Similarity-Based Transfer Tuning Approach to Performance Optimization}},
  year={2023},
  month={06},
  doi={10.1145/3577193.3593714},
}