基于FPGA的时空图神经网络加速器结构

摘要

本发明公开了一种基于FPGA的时空图神经网络加速器结构，设计了张量和向量双加速模块并行处理的方式，向量加速模块执行稀疏矩阵相乘或者执行逐元素乘加运算，张量加速模块执行稠密矩阵相乘、偏置项相加以及不同激活函数的运算，而且系统控制模块控制所述张量加速模块和向量加速模块按照设定的计算时序完成与所述计算时序对应的计算，数据流控制模块控制张量加速模块和所述向量加速模块之间的数据流向以便循环完成计算，整个架构的设计实现了对三维时空图神经网络的高效处理，以简洁的方式实现了多功能函数的计算；由于可以实现多种神经网络模式的加速计算，可以向下兼容图卷积神经网络GCN和门控递归单元GRU等多种网络模型的加速，具备了更强的普适性。