The search functionality is under construction.

IEICE TRANSACTIONS on Information

Guide Automatic Vectorization by means of Machine Learning: A Case Study of Tensor Contraction Kernels

Antoine TROUVÉ, Arnaldo J. CRUZ, Kazuaki J. MURAKAMI, Masaki ARAI, Tadashi NAKAHIRA, Eiji YAMANAKA

  • Full Text Views

    0

  • Cite this

Summary :

Modern optimizing compilers tend to be conservative and often fail to vectorize programs that would have benefited from it. In this paper, we propose a way to predict the relevant command-line options of the compiler so that it chooses the most profitable vectorization strategy. Machine learning has proven to be a relevant approach for this matter: fed with features that describe the software to the compiler, a machine learning device is trained to predict an appropriate optimization strategy. The related work relies on the control and data flow graphs as software features. In this article, we consider tensor contraction programs, useful in various scientific simulations, especially chemistry. Depending on how they access the memory, different tensor contraction kernels may yield very different performance figures. However, they exhibit identical control and data flow graphs, making them completely out of reach of the related work. In this paper, we propose an original set of software features that capture the important properties of the tensor contraction kernels. Considering the Intel Merom processor architecture with the Intel Compiler, we model the problem as a classification problem and we solve it using a support vector machine. Our technique predicts the best suited vectorization options of the compiler with a cross-validation accuracy of 93.4%, leading to up to a 3-times speedup compared to the default behavior of the Intel Compiler. This article ends with an original qualitative discussion on the performance of software metrics by means of visualization. All our measurements are made available for the sake of reproducibility.

Publication
IEICE TRANSACTIONS on Information Vol.E99-D No.6 pp.1585-1594
Publication Date
2016/06/01
Publicized
2016/03/22
Online ISSN
1745-1361
DOI
10.1587/transinf.2015EDP7440
Type of Manuscript
PAPER
Category
Artificial Intelligence, Data Mining

Authors

Antoine TROUVÉ
  Engineering Department of Kyushu University
Arnaldo J. CRUZ
  Engineering Department of Kyushu University
Kazuaki J. MURAKAMI
  Engineering Department of Kyushu University
Masaki ARAI
  Fujitsu Laboratories Limited
Tadashi NAKAHIRA
  Fujitsu Laboratories Limited
Eiji YAMANAKA
  Fujitsu Limited

Keyword