模型运算方法及装置制造方法及图纸

技术编号:43488345 阅读:13 留言:0更新日期:2024-11-29 16:58
本发明专利技术提供一种模型运算方法及装置,所述方法包括:将待运算数据划分为多份批次数据,并将各批次数据加载至对应芯片上;在各芯片上,执行数据交换步骤,数据交换步骤包括在各芯片上基于批次数据进行运算时,各芯片与目标芯片进行数据交换,以使各芯片从目标芯片上获取各芯片的交换数据,目标芯片指除各芯片之外的其余芯片;基于各芯片上的批次数据的运算结果,以及各芯片的交换数据的运算结果,进行模型运算,得到模型运算结果。本发明专利技术提供的模型运算方法及装置,可以避免各批次数据加载至对应芯片的过程和各芯片与目标芯片进行数据交换的过程抢占L2带宽,进而避免计算延迟,提高模型运算速度。

【技术实现步骤摘要】

本专利技术涉及人工智能,尤其涉及一种模型运算方法及装置


技术介绍

1、在人工智能模型里通常包括计算类算子和数据交换规约类算子。计算类算子指进行实际计算或处理的操作符或指令,计算类算子通常涉及对数据进行数学运算、逻辑运算或函数计算。数据交换规约类算子主要涉及数据的移动、交换和规约操作,数据交换规约类算子通常处理数据在不同存储位置(如不同芯片)之间的传输、汇总或合并操作。

2、目前,在进行人工智能模型推理时,多是并行执行计算类算子和数据交换规约类算子,但该方法会导致计算类算子性能下降,进而影响模型推理效率。


技术实现思路

1、本专利技术提供一种模型运算方法及装置,用以解决相关技术中存在的缺陷。

2、本专利技术提供一种模型运算方法,包括如下步骤:

3、将待运算数据划分为多份批次数据,并将各批次数据加载至对应芯片上;

4、在各芯片上,执行数据交换步骤,所述数据交换步骤包括在各芯片上基于批次数据进行运算时,各芯片与目标芯片进行数据交换,以使各芯片从所述目标芯片上获取各芯本文档来自技高网...

【技术保护点】

1.一种模型运算方法,其特征在于,包括:

2.根据权利要求1所述的模型运算方法,其特征在于,所述各芯片与目标芯片进行数据交换,包括:

3.根据权利要求2所述的模型运算方法,其特征在于,各芯片的交换数据基于各芯片的运算逻辑确定。

4.根据权利要求1至3任一项所述的模型运算方法,其特征在于,所述在各芯片上基于批次数据进行运算时,各芯片与目标芯片进行数据交换,包括:

5.根据权利要求1至3任一项所述的模型运算方法,其特征在于,所述基于各芯片上的批次数据的运算结果,以及各芯片的交换数据的运算结果,进行模型运算,得到模型运算结果,包括:

...

【技术特征摘要】

1.一种模型运算方法,其特征在于,包括:

2.根据权利要求1所述的模型运算方法,其特征在于,所述各芯片与目标芯片进行数据交换,包括:

3.根据权利要求2所述的模型运算方法,其特征在于,各芯片的交换数据基于各芯片的运算逻辑确定。

4.根据权利要求1至3任一项所述的模型运算方法,其特征在于,所述在各芯片上基于批次数据进行运算时,各芯片与目标芯片进行数据交换,包括:

5.根据权利要求1至3任一项所述的模型运算方法,其特征在于,所述基于各芯片上的批次数据的运算结果,以及各芯片的交换数据的运算结果,进行模型运算,得到模型运算结果,包括:

6.根据权利要求1至3...

【专利技术属性】
技术研发人员:请求不公布姓名请求不公布姓名请求不公布姓名请求不公布姓名请求不公布姓名请求不公布姓名
申请(专利权)人:广州壁仞集成电路有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1