用于专用加速器的神经网络预编译算法

基本信息

申请号 CN202111557330.7 申请日 -
公开(公告)号 CN114219081A 公开(公告)日 2022-03-22
申请公布号 CN114219081A 申请公布日 2022-03-22
分类号 G06N3/063(2006.01)I 分类 计算;推算;计数;
发明人 赵梓豪;王艳红;邹乔莎;史传进 申请(专利权)人 复旦大学
代理机构 上海正旦专利代理有限公司 代理人 陆飞;陆尤
地址 200433上海市杨浦区邯郸路220号
法律状态 -

摘要

摘要 本发明属神经网络技术领域,具体为用于专用加速器的神经网络预编译算法。本发明算法包括:使用有向图对各种神经网络进行建模,神经网络的算子使用图节点表示,神经网络的输入输出关系使用有向边表示:对神经网络进行预处理,包括算子融合、合并算子拆解、大节点分解、小节点合并;按照专用加速器的静态随机存储器的大小,对经过预处理的有向图进行分割,使得每个计算图可以直接编译到专用加速器上运行。本发明通过将神经网络使用图来表示,经过对各种节点的预编译处理,使得计算图达到更适合映射到硬件的形式,实现硬件的加速效果最大化以及硬件的功耗最小化。