买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
摘要:一种数据处理方法、处理器、电子设备和非瞬时性计算机可读存储介质。该数据处理方法包括:获取第一向量和第一矩阵以执行一个向量矩阵乘法操作,其中,第一向量为第一输入张量中的任一行,第一矩阵为第二输入张量中与第一向量对应的1个批次;执行第一向量与第一矩阵中各行的乘累加操作,确定H2个乘累加结果;使用线程束级别的数据存储指令将H2个乘累加结果存储至内存,其中,在存储H2个乘累加结果时,针对H2个乘累加结果中连续的M个乘累加结果,利用线程束内的数据交换操作使得M个乘累加结果由M条线程分别持有,以使用线程束级别的数据存储指令一次性将M个乘累加结果存储至内存,M条线程属于同一个线程束。
主权项:1.一种数据处理方法,其特征在于,用于执行第一输入张量和第二输入张量的张量乘法操作,其中,所述第一输入张量和所述第二输入张量的数据格式为[N,H,W],N表示批次数量,H表示高度,W表示宽度,所述第一输入张量表示为[1,H1,W1],所述第二输入张量表示为[H1,H2,W1],H1、H2和W1为正整数,所述张量乘法操作包括H1个向量矩阵乘法操作,每个向量矩阵乘法操作包括进行所述第一输入张量中的一行与所述第二输入张量中的1个批次的转置乘法,所述数据处理方法包括:获取第一向量和第一矩阵以执行一个向量矩阵乘法操作,其中,所述第一向量为所述第一输入张量中的任一行,所述第一矩阵为所述第二输入张量中与所述第一向量对应的1个批次;执行所述第一向量与所述第一矩阵中各行的乘累加操作,以确定H2个乘累加结果,其中,所述H2个乘累加结果按照所述第一矩阵中各行的位置关系排列成一行,作为所述第一向量和所述第一矩阵进行所述向量矩阵乘法操作的转置乘法结果;使用线程束级别的数据存储指令将所述H2个乘累加结果存储至内存,其中,在存储所述H2个乘累加结果时,针对所述H2个乘累加结果中连续的M个乘累加结果,利用线程束内的数据交换操作使得所述M个乘累加结果由M条线程分别持有,以使用所述线程束级别的数据存储指令一次性将所述M个乘累加结果存储至所述内存,所述M条线程属于同一个线程束,M为正整数。
全文数据:
权利要求:
百度查询: 北京壁仞科技开发有限公司 上海壁仞科技股份有限公司 数据处理方法、处理器、电子设备和存储介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。