python fpga加速_基于FPGA的Tensorflow系统加速的设计方法与流程

本发明涉及异构计算开发领域,具体是,一种将Tensorflow人工智能系统中一些本身由CPU实现的算子,改用现场可编程门阵列(以下简称FPGA)来进行实现。

背景技术:

随着人工智能的发展,深度神经网络已经在计算机视觉、自然语言处理和其他跨学科研究领域日益普及。深度神经网络通过多个堆叠层自然地从输入中提取特征,并使用分类器来做出最终决策,这意味着其中蕴含着大量矩阵或卷积运算算子。最近的证据表明,神经网络的深度对于性能至关重要,这极大地增加了对计算能力的要求,传统的CPU已经无法满足日益增长的需求,完成一个深度神经网络的识别往往要很长的时间,已经不适用于现实应用场景中。解决该问题的常用方法是采用具有各种不同计算设备的异构分布式环境,例如DSP,GPU和诸如FPGA的定制硬件加速器。然而,尽管FPGA具有灵活性,高能效和成本效益,但并没有被纳入最先进的深度学习框架或系统。例如Tensorflow中作为计算加速器,这和其传统的开发方式速度非常缓慢有关系。因此,将FPGA作为计算加速设备内置于tensorflow中并通过高层次综合来改善FPGA加速器的开发效率是一个很有研究意义且具有创新性的。

技术实现要素:

为了解决现有技术中无法很好地使用CPU来支持人工智能神经网络的开发和运行速度,并且当下Tensorflow人工智能系统没有对于FPGA的设备支持这一系列问题,本发明提供了一种基于FPGA的Tensorflow系统加速的设计方法,将FPGA作为计算加速器添加进入Tensorflow系统中。通过将FPGA这一可编程的逻辑器件作为基本计算设备并入Tensorflow,使其完成Tensorflow中的一些计算。同时,实现FPGA器件的一组相应的操作内核,以获取FPGA上各种张量流操作的计算速度来确保加速效果。

本发明的技术解决方案如下:

一种基于FPGA的Tensorflow系统的加速方法,其特点征在于,包括如下步骤:

第一步,以Python作为上层客户端程序库

第二步,提供合适的C语言接口

将OpenCL实现的模块封装后作为接口供上层的客户端程序调用;

第三步,通过OpenCL准备FPGA算子开发环境

第四步,通过OpenCL开发FPGA算子

先用C语言来编写需要的内核函数,再通过设备商提供的BSP以及OpenCL内核编译环境即可自动生成相应的二进制流文件,将生成的二进制流文件烧入FPGA后即可在FPGA上运行算子;

第五步,通过OpenCL编译内核算子

将C语言描述的算子作为内核,然后通过设备商提供的BSP来进行映射的过程;

第六步,将FPGA设备并入整个系统

将OpenCL的C语言代码加入C语言接口层,并添加一些注册设备ID的代码以方便上层的Python,C++客户端直接调用。

所述的通过OpenCL准备FPGA算子开发环境,具体是

通过查询平台,查询设备确定系统中的FPGA型号和数量;

通过命令队列和创建缓存来完成待执行的操作类似和带操作数据的存储;

通过BSP将C语言完成的内核映射成FPGA可以执行的二进制流文件;

设置执行参数,OpenCL的主机端按照实际需求执行内核。

所述的执行参数包括执行内核的种类和次数。

所述的通过OpenCL编译内核算子,具体是

确定FPGA的型号种类和其对应的BSP;

将要实现的算子用C语言开发完成;

在设备商提供的环境里执行编译指令;

会生成相应的网表文件和能够烧写进入FPGA生成相应硬件的二进制流文件。

与现有技术相比,本发明的优势在于:使用FPGA加入Tensorflow系统来加速一些神经网络的运算,相较于传统的对于FPGA无支持的Tensorflow系统提高了神经网络运算速度。并且通过OpenCL来开发相应的算子,利用OpenCL本身的特性,相较于传统的用硬件描述语言开发算子,再去一步一步开发相应的数据流管理,通信管理,大大减少了开发难度,也让FPGA作为Tensorflow系统中的一个设备有了稳定性和实用性的保证。

附图说明

图1是本发明基于FPGA的Tensorflow系统加速的设计方法的流程图;

图2是FPGA算子实现架构图

图3OpenCL内核实现流程

具体实施方式

下面结合附图和实施例对本发明作进一步解释,但不应此限制本发明的保护范围。

本发明所述具体实现主要包括以下几个步骤:

第一步,选择合适的上层客户端程序库

如果要直接通过机器级语言来实现神经网络或者别的向量操作必然是十分费时的。所以Tensorflow往往对面向高级语言例如python和c++提供已经封装好的集成库,本发明的实现也是利用这些库,因为这些库里提供的函数不论是实现还是声明都有非常好的抽象。

结合附图1来看,不管是哪种语言实现的库,最后的具体操作还是通过之后的一系列流程最终要落实到设备级的器件上,也就是说上层的高级语言是提供统一的编程模型来方便用户来进行使用,而本发明也是按照这一套规则,将FPGA实现的一些加速算子也逐级封装成函数让上层用户直接用高级语言来调用。

第二步,提供合适的C语言接口

结合图2来看,尽管高级语言可以方便上层使用者的开发,但是因为本发明的目标还是将FPGA加入整个系统,所以本发明还是需要对连接上层与之后的设备层的C语言接口层做相应的修改。

采用的方法是:参考已有的针对CPU或GPU实现的方法,保留可以进行复用的部分。随后使用OpenCL来进行开发,因为OpenCL本身是一个面向异构系统通用的开源标准,同时OpenCL本身也可以使用C语言来作为实现载体。而且新版本的TensorFlow也实现了对OpenCL以及支持OpenCL设备的支持,这样可以大大降低接口上的开发难度和提高了接口的稳定度和实用性。

第三步,通过OpenCL准备FPGA算子开发环境

如果要用FPGA来完成相应算子的加速,传统的方法是用硬件描述语言来写出相应算子功能。但这样的开发时间很长,而OpenCL提供了一种可以用C语言来实现算子功能,最后再通过相应设备公司提供的板级支持文件(一下简称BSP)来在FPGA上进行实现。

如图3所述,OpenCL已经提供了一套标准的开发流程,其提供了大量的C语言接口,可以在第二部分中复用。首先,通过查询平台,查询设备确定系统中的FPGA型号和数量,再通过命令队列和创建缓存来完成待执行的操作类似和带操作数据的存储,随后通过BSP将C语言完成的内核映射成FPGA可以执行的二进制流文件。最后设置一些执行参数例如执行内核的种类和次数,OpenCL的主机端就会按照实际需求执行内核从而将算子部分通过FPGA实现。从上述流程可以发现,OpenCL大大减少了将FPGA接入整个系统的难度。因为它的流程本身就可以完成大部分工作比如附图2中的数据传输迁移到最后的设备管理。

第四步,通过OpenCL开发FPGA算子

需要在FPGA上完成的算子都可以通过OpenCL提供的标准来用C语言来编写内核函数来完成。用C语言完成的内核函数通过设备商提供的BSP以及OpenCL内核编译环境即可自动生成可以在FPGA硬件上运行的算子。

第五步,通过OpenCL编译内核算子

图3中的编译内核是将C语言描述的算子作为内核,然后通过设备商提供的BSP来进行映射的过程。具体实现方法是确定好FPGA的型号种类和其对应的BSP,然后将要实现的算子用C语言开发完成,然后在设备商提供的环境例如英特尔公司的SDK for OpenCL里执行编译指令。随后会生成相应的网表和实际的硬件实现二进制流,这样就可以让FPGA直接实现相应的算子。

第六步,将FPGA设备并入整个系统

按照上述步骤完成后,附图1中的内核实现和设备层部分均已完成。而因为使用了OpenCL的缘故,不要再独立开发主机端和FPGA的通信,数据流运行器部分也已经完成。最后仅需将OpenCL的C语言代码加入C语言接口层,并添加一些关于设备注册ID的代码以方便上层的Python,C++客户端直接调用。经过上述封装,即可让上层用户在Tensorflow系统中用高级语言实现神经网络,并且让FPGA实现部分算子可以对网络进行加速。

本发明不仅扩展Tensorflow对于新的设备FPGA的支持,还在开发模式上进行了创新,通过OpenCL这一面向异构系统通用的并行编程的开放式、免费的编程环境设计来开发FPGA。这样不仅可让Tensorflow能充分利用FPGA的全新操作来探索FPGA的潜力,也通过OpenCL提供的环境加快了FPGA的开发进度,同时也减少了将FPGA接入Tensorflow带来的诸多接口问题。这样可使得算子运行速度获得提升,同时也增加一些新的算子由FPGA来完成以供Tensorflow来调用。

  • 1
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值