一种基于昇腾AI处理器的批量矩阵乘算子的优化方法
实质审查的生效
摘要

本发明公开了一种基于昇腾AI处理器的批量矩阵乘算子的优化方法,所述方法包括:获取第一输入数据与第二输入数据,并将所述第一输入数据与所述第二输入数据搬运到AI Core;获取所述第二输入数据的加载行数,并根据所述加载行数以及预设的缓冲区的双缓存机制对所述第一输入数据和第二输入数据进行划分;将划分后的所述第一输入数据和第二输入数据加载到所述缓冲区中进行计算,得到输出数据;将所述输出数据搬运到外部存储中进行输出。本申请能够利用双缓冲机制,能够缩短第一矩阵和第二矩阵相乘的运算时间,从而提高数据处理的效率。

基本信息
专利标题 :
一种基于昇腾AI处理器的批量矩阵乘算子的优化方法
专利标题(英):
暂无
公开(公告)号 :
CN114329325A
申请号 :
CN202111374829.4
公开(公告)日 :
2022-04-12
申请日 :
2021-11-19
授权号 :
暂无
授权日 :
暂无
发明人 :
马银萍李若淼董昊森樊春杨宏辉龙汀汀
申请人 :
北京大学;鹏城实验室
申请人地址 :
北京市海淀区颐和园路5号
代理机构 :
深圳市君胜知识产权代理事务所(普通合伙)
代理人 :
李可
优先权 :
CN202111374829.4
主分类号 :
G06F17/16
IPC分类号 :
G06F17/16  G06F9/54  G06F7/523  
IPC结构图谱
G
G部——物理
G06
计算;推算或计数
G06F
电数字数据处理
G06F17/00
特别适用于特定功能的数字计算设备或数据处理设备或数据处理方法
G06F17/10
复杂数学运算的
G06F17/16
矩阵或向量计算的
法律状态
2022-04-29 :
实质审查的生效
IPC(主分类) : G06F 17/16
申请日 : 20211119
2022-04-12 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载
  • 联系电话
    电话:023-6033-8768
    QQ:1493236332
  • 联系 Q Q
    电话:023-6033-8768
    QQ:1493236332
  • 关注微信
    电话:023-6033-8768
    QQ:1493236332
  • 收藏
    电话:023-6033-8768
    QQ:1493236332