一种基于光电存算一体器件的GRU加速器及其方法

    公开(公告)号:CN118468953A

    公开(公告)日:2024-08-09

    申请号:CN202410495901.6

    申请日:2024-04-24

    Applicant: 南京大学

    Abstract: 本发明提供了一种基于光电存算一体器件的GRU加速器及其方法。该加速器包括:数据预处理模块,用于对输入的序列信息进行预处理,转换为一维输入向量并传输给门控循环计算模块;门控循环计算模块,用于在每个时刻根据输入向量与前一时刻的状态对更新门、重置门的计算,并通过更新门、重置门控制输入向量与前一时刻的状态进行记忆与遗忘,计算出当前时刻的状态进行寄存并传输给预测输出模块;预测输出模块,用于对计算结果进行序列转换或分类操作,输出最终的预测结果。本发明在GRU进行矩阵向量乘时进行加速与优化,提高了矩阵向量乘法的效率,减少了由于密集计算过程中频繁访问片外存储带来的功耗与时间。

Patent Agency Ranking