广告

实例!详解FPGA如何实现FP16格式点积级联运算

2020-08-14 杨宇,Achronix资深现场应用工程师 阅读:
通过使用Achronix Speedster7t FPGA中的机器学习加速器MLP72,开发人员可以轻松选择浮点/定点格式和多种位宽,或快速应用块浮点,并通过内部级联可以达到理想性能。

神经网络架构中的核心之一就是卷积层,卷积的最基本操作就是点积。向量乘法的结果是向量的每个元素的总和相乘在一起,通常称之为点积。此向量乘法如下所示:cE5ednc

cE5ednc

图 1 点积操作cE5ednc

该总和S由每个矢量元素的总和相乘而成,因此S=a1b1+a2b2+a3b3+...cE5ednc

本文讲述的是使用FP16格式的点积运算实例,展示了MLP72支持的数字类型和乘数的范围。cE5ednc

此设计实现了同时处理8对FP16输入的点积。该设计包含四个MLP72,使用MLP内部的级联路径连接。每个MLP72将两个并行乘法的结果相加(即aibi+ai+1bi+1),每个乘法都是i_a输入乘以i_b输入(均为FP16格式)的结果。来自每个MLP72的总和沿着MLP72的列级联到上面的下一个MLP72块。在最后一个MLP72中,在每个周期上,计算八个并行FP16乘法的总和。cE5ednc

最终结果是多个输入周期内的累加总和,其中累加由i_first和i_last输入控制。 i_first输入信号指示累加和归零的第一组输入。 i_last信号指示要累加和加到累加的最后一组输入。最终的i_last值可在之后的六个周期使用,并使用i_last o_valid进行限定。两次运算之间可以无空拍。cE5ednc

  • 配置说明

cE5ednc

表 1 FP16点积配置表cE5ednc

  • 端口说明

cE5ednc

表 2 FP16点积端口说明表cE5ednc

  • 时序图

cE5ednc

图 2 FP16点积时序图cE5ednc

其中,cE5ednc

cE5ednc

那么,以上运算功能如何对应到MLP内部呢?其后的细节已分为MLP72中的多个功能阶段进行说明。cE5ednc

  • 进位链

首先请看下图,MLP之间的进位链结构,这是MLP内部的专用走线,可以保证级联的高效执行。cE5ednc

cE5ednc

图 3 MLP进位链cE5ednc

  • 乘法阶段

下图是MLP中浮点乘法功能阶段,其中寄存器代表一级可选延迟。cE5ednc

cE5ednc

图 4 MLP乘法功能阶段框图cE5ednc

MLP72浮点乘法级包括两个24位全浮点乘法器和一个24位全浮点加法器。两个乘法器执行A×B和C×D的并行计算。加法器将两个结果相加得到A×B + C×D。cE5ednc

乘法阶段有两个输出。下半部分输出可以在A×B或(A×B + C×D)之间选择。上半部分输出始终为C×D。cE5ednc

乘法器和加法器使用的数字格式由字节选择参数以及和参数设置的格式确定。cE5ednc

浮点输出具有与整数输出级相同的路径和结构。MLP72可以配置为在特定阶段选择整数或等效浮点输入。输出支持两个24位全浮点加法器,可以对其进行加法或累加配置。 进一步可以加载加法器(开始累加),可以将其设置为减法,并支持可选的舍入模式。cE5ednc

最终输出阶段支持将浮点输出格式化为MLP72支持的三种浮点格式中的任何一种。 此功能使MLP72可以外部支持大小一致的浮点输入和输出(例如fp16或bfloat16),而在内部以fp24执行所有计算。cE5ednc

cE5ednc

图 5 MLP浮点输出阶段框图cE5ednc

需要强调的是本设计输入和输出都是FP16格式,中间计算过程,即进位链上的fwdo_out和fwdi_dout 都是FP24格式。具体逻辑框图如下所示:cE5ednc

cE5ednc

图 6 FP16点积逻辑框图cE5ednc

MLP内部数据流示意图:cE5ednc

cE5ednc

图 7 FP16点积在MLP内部数据流图cE5ednc

最终ACE的时序结果如下:cE5ednc

cE5ednc

如需了解更多产品细节,请发送邮件到Dawson.Guo@Achronix.comcE5ednc

本文为EDN电子技术设计 原创文章,禁止转载。请尊重知识产权,违者本司保留追究责任的权利。
  • 微信扫一扫
    一键转发
  • 最前沿的电子设计资讯
    请关注“电子技术设计微信公众号”
广告
热门推荐
广告
广告
EE直播间
在线研讨会
广告
广告
面包芯语
广告
向右滑动:上一篇 向左滑动:下一篇 我知道了