← 返回列表
用于进行卷积运算的方法、电子设备和存储介质
摘要文本
本公开涉及一种用于进行卷积运算的方法、电子设备和存储介质。其中方法包括对卷积核进行预处理，以便得到目标卷积核；经由向量核，将卷积算子转换为矩阵乘的数据，并将矩阵乘的数据存储至目标存储区；以及经由张量核，针对目标卷积核和从目标存储区中读取的矩阵乘的数据进行矩阵乘加操作，以便基于矩阵乘加结果生成卷积运算结果。本公开的方案能够显著提高卷积运算的性能。
申请人信息
- 申请人:北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
- 申请人地址:100085 北京市海淀区上地信息路26号1层0106-508室
- 发明人: 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 用于进行卷积运算的方法、电子设备和存储介质 |
| 专利类型 | 发明授权 |
| 申请号 | CN202311751966.4 |
| 申请日 | 2023/12/19 |
| 公告号 | CN117435855B |
| 公开日 | 2024/3/19 |
| IPC主分类号 | G06F17/15 |
| 权利人 | 北京壁仞科技开发有限公司; 上海壁仞科技股份有限公司 |
| 发明人 | 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名; 请求不公布姓名 |
| 地址 | 北京市海淀区上地信息路26号1层0106-508室; 上海市闵行区陈行公路2388号16幢13层1302室 |
专利主权项内容
1.一种用于进行卷积运算的方法，其特征在于，包括：对卷积核进行预处理，以便得到目标卷积核；经由向量核，将卷积算子转换为矩阵乘的数据，并将矩阵乘的数据存储至目标存储区；以及经由张量核，针对目标卷积核和从目标存储区中读取的矩阵乘的数据进行矩阵乘加操作，以便基于矩阵乘加结果生成卷积运算结果；经由向量核将卷积算子转换为矩阵乘的数据并将矩阵乘的数据存储至目标存储区包括：基于卷积参数，将矩阵乘的数据划分为多个子基本块以写入片上高速缓存单元。。来自马克数据网