用于专用加速器的神经网络预编译算法
基本信息
申请号 | CN202111557330.7 | 申请日 | - |
公开(公告)号 | CN114219081A | 公开(公告)日 | 2022-03-22 |
申请公布号 | CN114219081A | 申请公布日 | 2022-03-22 |
分类号 | G06N3/063(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 赵梓豪;王艳红;邹乔莎;史传进 | 申请(专利权)人 | 复旦大学 |
代理机构 | 上海正旦专利代理有限公司 | 代理人 | 陆飞;陆尤 |
地址 | 200433上海市杨浦区邯郸路220号 | ||
法律状态 | - |
摘要
摘要 | 本发明属神经网络技术领域,具体为用于专用加速器的神经网络预编译算法。本发明算法包括:使用有向图对各种神经网络进行建模,神经网络的算子使用图节点表示,神经网络的输入输出关系使用有向边表示:对神经网络进行预处理,包括算子融合、合并算子拆解、大节点分解、小节点合并;按照专用加速器的静态随机存储器的大小,对经过预处理的有向图进行分割,使得每个计算图可以直接编译到专用加速器上运行。本发明通过将神经网络使用图来表示,经过对各种节点的预编译处理,使得计算图达到更适合映射到硬件的形式,实现硬件的加速效果最大化以及硬件的功耗最小化。 |
