计算机工程与科学

北大核心,INSPEC,JST,CSCD扩展版,WJCI

国内刊号:43-1258/TP

国际刊号:1007-130X

计算机工程与科学杂志2023年第5期:基于脉动阵列的层融合注意力模型加速器结构

发布日期:

作者:刘晓航, 姜晶菲, 许金伟

单位:(1.国防科技大学研究生院,湖南 长沙 410073;2.国防科技大学并行与分布处理国家重点实验室,湖南 长沙 410073)

关键词:脉动阵列,注意力机制,层融合,加速器结构,矩阵分块,柔性最大值传输函数,

基金:国家国防科技工业局国防科技重点实验室稳定支持重点项目(WDZC20215250103)

注意力机制最近在深度神经网络中表现出优越的性能,但其计算包含复杂的数据流,内存开销和计算量大,需要定制加速器来优化推理计算。提出一种针对注意力机制计算的加速器结构。采用基于硬件控制的灵活分块方法,将模型中的巨大矩阵分成硬件亲和的计算块,使块矩阵的计算匹配加速器脉动阵列;提出基于双步softmax函数分解计算的层融合计算方法,有效减少了注意力模型计算对内存的访问。采用硬件描述语言HDL设计实现了细粒度计算调度的层融合注意力模型加速器结构。基于XILINX FPGA器件和HLS工具进行了性能评估。相同设置下,与CPU相比延迟加速了4.9倍,与GPU相比能效提升了1.24倍。

来源:2023年第5期

《计算机工程与科学》期刊编辑部

查看计算机工程与科学杂志2023年第5期

声明

严正声明:本站非期刊官网,非中介代理。

本站仅提供学术规范服务:快速预审、润色编辑服务、中英文查重、降重、去重服务、推荐合适的期刊投稿等学术规范服务。 如需提供学术规范服务请联系在线编辑。

联系我们

  • 地址:湖南省长沙市开福区德雅路109号
  • 电话:86-0731-87002567
  • E-mail:jsjgcykx@vip.163.com

咨询工作人员