公开(公告)号：CN114281874A

公开(公告)日：2022-04-05

申请号：CN202111373725.1

申请日：2021-11-19

Applicant: 北京大学 ,  鹏城实验室

Inventor： 龙汀汀 ,  樊春 ,  马银萍 ,  杨宏辉 ,  李若淼 ,  董昊森

IPC: G06F16/25 ,  G06F16/22 ,  G06F17/16

Abstract: 本发明涉及数据索引领域，具体是涉及一种基于昇腾AI处理器的索引转换方法及装置。本发明根据张量的数据排布格式，对算子的索引信息进行转换，使得转换之后的索引信息与张量的数据排布格式所匹配，由于对索引信息进行转换只涉及到对数字的简单运算，从而在不改变张量数据排布格式的前提下，依然能够通过转换之后的索引信息在张量中索引到相应的数据。由于本发明不涉及到改变张量数据排布格式，从而避免了因改变张量数据排布格式而耗费的时间，进而提高了算子的计算速度。

公开(公告)号：CN113704689A

公开(公告)日：2021-11-26

申请号：CN202110984626.0

申请日：2021-08-25

Applicant: 北京大学

Inventor： 马银萍 ,  李若淼 ,  杨宏辉 ,  樊春 ,  董昊森

IPC: G06F17/16 ,  G06F7/52 ,  G06F5/06

Abstract: 本发明公开了一种基于昇腾AI处理器的矩阵乘算子的处理方法及装置。其中，该方法包括：获取矩阵乘算子的第一源操作数和第二源操作数；对第一源操作数进行数据转换处理，得到第一转换结果，以及对第二源操作数进行数据转换处理，得到第二转换结果；对第一转换结果进行分块处理，得到第一分块结果，以及对第二转换结果进行分块处理，得到第二分块结果；采用与人工智能处理器中计算核心数量对应的多个计算核心对第一分块结果与第二分块结果进行矩阵乘运算，得到中间运算结果；对中间运算结果进行数据转换处理，得到矩阵乘算子的输出结果，其中，输出结果的输出格式为二维格式。本发明解决了矩阵乘运算的运算效率低的技术问题。