用于矩阵计算的方法及装置

基本信息

申请号 CN202110804122.6 申请日 -
公开(公告)号 CN113268708B 公开(公告)日 2021-10-15
申请公布号 CN113268708B 申请公布日 2021-10-15
分类号 G06F17/16;G06F7/523 分类 计算;推算;计数;
发明人 不公告发明人 申请(专利权)人 上海壁仞智能科技有限公司
代理机构 北京市金杜律师事务所 代理人 王茂华
地址 100085 北京市海淀区上地信息路26号1层0106-508室
法律状态 -

摘要

摘要 本公开的实施例涉及用于矩阵计算的方法及装置,涉及集成电路领域。该方法包括:经由包括至少两行至少两列计算单元的阵列中每一行的至少两个计算单元,按行顺序获取第一矩阵中对应行的至少两个第一子矩阵,计算单元用于执行矩阵乘法运算,阵列中横向相邻和纵向相邻的计算单元之间配置有数据通道;经由横向的数据通道传输对应行的至少两个第一子矩阵;经由阵列中每一列的至少两个计算单元,按列顺序获取第二矩阵中对应列的至少两个第二子矩阵;以及经由纵向的数据通道传输对应列的至少两个第二子矩阵,以用于第一矩阵和第二矩阵之间的矩阵乘法运算。由此,能够灵活实现不同尺寸的矩阵乘法运算。