Automated Accelerator Optimization Aided by Graph Neural Networks
Automated Accelerator Optimization Aided by Graph Neural Networks
复制标题
图神经网络辅助的自动加速器优化
DOI:
10.1145/3490422.3502330
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
and Cong, Jason
中科院分区:
文献类型:
--
作者:
Sohrabizadeh, Atefeh;Bai, Yunsheng;Sun, Yizhou;and Cong, Jason
Using High-Level Synthesis (HLS), the hardware designers must describe only a high-level behavioral flow of the design. However, it still can take weeks to develop a high-performance architecture mainly because there are many design choices at a higher level to explore. Besides, it takes several minutes to hours to evaluate the design with the HLS tool. To solve this problem, we model the HLS tool with a graph neural network that is trained to be used for a wide range of applications. The experimental results demonstrate that our model can estimate the quality of design in milliseconds with high accuracy, resulting in up to 79X speedup (with an average of 48X) for optimizing the design compared to the previous state-of-the-art work relying on the HLS tool.