面向权重可变的存算一体加速器的卷积算子调度

作     者：师紧想 SHI Jinxiang

作者机构：上海交通大学电子信息与电气工程学院上海200240 

出 版 物：《现代计算机》 (Modern Computer)

年 卷 期：2021年第27卷第10期

页      码：24-28页

摘      要：存算一体加速器因其高能效和低功耗的特点受到广泛关注,但在神经网络模型与存算一体加速平台之间缺乏高效的自动化映射工具。为了改善这一局面,本文提供一个面向忆阻器存算一体加速平台的自动代码生成软件栈,以将训练后的神经网络模型映射在存算一体加速器上。为了提高映射的效率,本文借助Halide调度卷积算子的循环空间,并设计优化连续内存访问、优化忆阻器阵列权重更新次数等优化策略。相比与默认映射方案,本文在典型卷积层中权重更新次数减少85%~96%。

主 题 词：存算一体 神经网络 自动映射 调度 

学科分类：080903[080903] 12[管理学] 1201[管理学-管理科学与工程类] 0809[工学-计算机类] 08[工学] 081104[081104] 0835[0835] 0811[工学-水利类] 0812[工学-测绘类] 

D　O　I：10.3969/j.issn.1007-1423.2021.10.005

馆 藏 号：203103361...