LIFT: LLM-Based Pragma Insertion for HLS via GNN Supervised Fine-Tuning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Prakriya, Neha, Ding, Zijian, Sun, Yizhou, Cong, Jason
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915266952167424
author Prakriya, Neha
Ding, Zijian
Sun, Yizhou
Cong, Jason
author_facet Prakriya, Neha
Ding, Zijian
Sun, Yizhou
Cong, Jason
contents FPGAs are increasingly adopted in datacenter environments for their reconfigurability and energy efficiency. High-Level Synthesis (HLS) tools have eased FPGA programming by raising the abstraction level from RTL to untimed C/C++, yet attaining high performance still demands expert knowledge and iterative manual insertion of optimization pragmas to modify the microarchitecture. To address this challenge, we propose LIFT, a large language model (LLM)-based coding assistant for HLS that automatically generates performance-critical pragmas given a C/C++ design. We fine-tune the LLM by tightly integrating and supervising the training process with a graph neural network (GNN), combining the sequential modeling capabilities of LLMs with the structural and semantic understanding of GNNs necessary for reasoning over code and its control/data dependencies. On average, LIFT produces designs that improve performance by 3.52x and 2.16x than prior state-of the art AutoDSE and HARP respectively, and 66x than GPT-4o.
format Preprint
id arxiv_https___arxiv_org_abs_2504_21187
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle LIFT: LLM-Based Pragma Insertion for HLS via GNN Supervised Fine-Tuning
Prakriya, Neha
Ding, Zijian
Sun, Yizhou
Cong, Jason
Machine Learning
FPGAs are increasingly adopted in datacenter environments for their reconfigurability and energy efficiency. High-Level Synthesis (HLS) tools have eased FPGA programming by raising the abstraction level from RTL to untimed C/C++, yet attaining high performance still demands expert knowledge and iterative manual insertion of optimization pragmas to modify the microarchitecture. To address this challenge, we propose LIFT, a large language model (LLM)-based coding assistant for HLS that automatically generates performance-critical pragmas given a C/C++ design. We fine-tune the LLM by tightly integrating and supervising the training process with a graph neural network (GNN), combining the sequential modeling capabilities of LLMs with the structural and semantic understanding of GNNs necessary for reasoning over code and its control/data dependencies. On average, LIFT produces designs that improve performance by 3.52x and 2.16x than prior state-of the art AutoDSE and HARP respectively, and 66x than GPT-4o.
title LIFT: LLM-Based Pragma Insertion for HLS via GNN Supervised Fine-Tuning
topic Machine Learning
url https://arxiv.org/abs/2504.21187