LLM2D
学习比较高级综合中的硬件设计
Learning to Compare Hardware Designs for High-Level Synthesis
作者: Yunsheng Bai, Atefeh Sohrabizadeh, Zijian Ding, Rongjian Liang, Weikai Li, Ding Wang, Haoxing Ren, Yizhou Sun, Jason Cong
发布日期: 5/9/2025
arXiv ID: oai:arXiv.org:2409.13138v2

摘要

arXiv:2409.13138v2 宣告类型: replace-cross 摘要: 高级综合(HLS)是一种自动设计过程,它将高级代码转换为硬件设计,从而加快硬件加速器的快速开发。HLS 依赖于内嵌在源代码中的 pragma(指令),这些指令可以引导综合过程,并且 pra- mga 多种设置和值会严重影响最终的硬件设计。当前最先进的基于机器学习的 HLS 方法,如 HARP,首先训练一个深度学习模型,通常基于应用于基于图的代码和 pragma 表示的图神经网络(GNN)。然后他们进行设计空间探索(DSE),探索 pragma 设计空间,使用模型对候选设计进行排名,并返回最优设计。然而,传统的 DSE 方法由于 pragma 设置与性能指标之间的高度非线性关系以及影响性能的复杂互作而面临挑战。 为了应对这些挑战,我们提出了一种新颖的方法 compareXplore,它可以学习比较硬件设计以实现有效的 HLS 优化。compareXplore 引入了一种结合了成对偏好学习与点式性能预测的混合损失函数,使得模型能够捕获相对偏好和绝对性能。此外,我们引入了一种新颖的节点差异注意模块,该模块专注于设计之间的最有信息性的差异,使模型能够识别对性能有关键影响的pragma。compareXplore 采用两阶段的 DSE,在初始设计剪枝阶段使用点式预测模型,然后在精确性能验证阶段进行成对比较。在广泛的实验证明中,compareXplore 在排名指标上取得了显著改进,并为选定的设计生成了高质量的 HLS 结果,优于现有的 SOTA 方法。