Can Large Language Models Predict Parallel Code Performance?

  • Gregory Bolet
  • , Giorgis Georgakoudis
  • , Harshitha Menon
  • , Konstantinos Parasyris
  • , Niranjan Hasabnis
  • , Hayden Estes
  • , Kirk Cameron
  • , Gal Oren

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

1 Scopus citations

Abstract

Accurate determination of the performance of parallel GPU code typically requires execution-time profiling on target hardware - an increasingly prohibitive step due to limited access to high-end GPUs. This paper explores whether Large Language Models (LLMs) can offer an alternative approach for GPU performance prediction without relying on hardware. We frame the problem as a roofline classification task: given the source code of a GPU kernel and the hardware specifications of a target GPU, can an LLM predict whether the GPU kernel is compute-bound or bandwidth-bound?For this study, we build a balanced dataset of 340 GPU kernels, obtained from HeCBench benchmark and written in CUDA and OpenMP, along with their ground-truth labels obtained via empirical GPU profiling. We evaluate LLMs across four scenarios: (1) with access to profiling data of the kernel source, (2) zero-shot with source code only, (3) few-shot with code and label pairs, and (4) fine-tuned on a small custom dataset. Our results show that state-of-the-art LLMs have a strong understanding of the Roofline model, achieving 100% classification accuracy when provided with explicit profiling data. We also find that reasoning-capable LLMs significantly outperform standard LLMs in zero- and few-shot settings, achieving up to 64% classification accuracy of GPU source codes, without any profiling information. Lastly, we find that model accuracy does not benefit meaningfully from few-shot prompting compared to zero-shot, and that LLM fine-tuning will require much more data than what we currently have available. This work is among the first to use LLMs for source-level roofline performance prediction via classification, and illustrates their potential to guide optimization efforts when runtime profiling is infeasible. Our findings suggest that with better datasets and prompt strategies, LLMs could become practical tools for HPC performance analysis and performance portability. Code and datasets are publicly available at https://github.com/Scientific-Computing-Lab/ParallelCodeEstimation.

Original languageEnglish
Title of host publicationHPDC 2025 - Proceedings of the 34th International Symposium on High-Performance Parallel and Distributed Computing
PublisherAssociation for Computing Machinery, Inc
ISBN (Electronic)9798400718694
DOIs
StatePublished - 9 Sep 2025
Externally publishedYes
Event34th International Symposium on High-Performance Parallel and Distributed Computing - Notre Dame, United States
Duration: 20 Jul 202523 Jul 2025

Publication series

NameHPDC 2025 - Proceedings of the 34th International Symposium on High-Performance Parallel and Distributed Computing

Conference

Conference34th International Symposium on High-Performance Parallel and Distributed Computing
Country/TerritoryUnited States
CityNotre Dame
Period20/07/2523/07/25

Bibliographical note

Publisher Copyright:
© 2025 Copyright held by the owner/author(s).

Keywords

  • CUDA
  • GPU
  • LLMs
  • OpenMP
  • performance
  • roofline model

Fingerprint

Dive into the research topics of 'Can Large Language Models Predict Parallel Code Performance?'. Together they form a unique fingerprint.

Cite this