MLKAPS: Machine Learning and Adaptive Sampling for HPC Kernel Auto-tuning - Laboratoire d'Informatique Parallélisme Réseaux Algorithmes Distribués
Pré-Publication, Document De Travail Année : 2024

MLKAPS: Machine Learning and Adaptive Sampling for HPC Kernel Auto-tuning

Résumé

Many High-Performance Computing (HPC) libraries rely on decision trees to select the best kernel hyperparameters at runtime, depending on the input and environment. However, finding optimized configurations for each input and environment is challenging and requires significant manual effort and computational resources. This paper presents MLKAPS, a tool that automates this task using machine learning and adaptive sampling techniques. MLKAPS generates decision trees that tune HPC kernels’ design parameters to achieve efficient performance for any user input. MLKAPS scales to large input and design spaces, outperforming similar state-of-the-art auto-tuning tools in tuning time and mean speedup. We demonstrate the benefits of MLKAPS on the highly optimized Intel MKL dgetrf LU kernel and show that MLKAPS finds blindspots in the manual tuning of HPC experts. It improves over 85% of the inputs with a geomean speedup of ×1.30. On the Intel MKL dgeqrf QR kernel, MLKAPS improves performance on 85% of the inputs with a geomean speedup of ×1.18.
Fichier principal
Vignette du fichier
main.pdf (2.93 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04851397 , version 1 (06-01-2025)

Identifiants

  • HAL Id : hal-04851397 , version 1

Citer

Mathys Jam, Eric Petit, Pablo de Oliveira Castro, David Defour, Greg Henry, et al.. MLKAPS: Machine Learning and Adaptive Sampling for HPC Kernel Auto-tuning. 2024. ⟨hal-04851397⟩
0 Consultations
0 Téléchargements

Partager

More