(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111676480.X (22)申请日 2021.12.31 (71)申请人 曙光信息产业 （北京） 有限公司 地址 100089 北京市海淀区东北旺西路8号 院36号楼 (72)发明人 朱鹏飞　赵源　梁智　 (74)专利代理 机构 华进联合专利商标代理有限 公司 44224 专利代理师 景怀宇 (51)Int.Cl. G06F 9/50(2006.01) G06F 9/445(2018.01) (54)发明名称 数组计算方法、 装置、 设备、 介质和计算机程 序产品 (57)摘要 本申请涉及一种数组计算方法、 装置、 设备、 介质和计算机程序产品。 所述方法包括： 调用目 标核函数， 该目标核函数定义有目标本地变量， 该目标本地变量指向第一数 组的计算结果， 该目 标核函数用于对第二数组进行计算， 该第二数组 的计算依赖于该第一数组的计算结果； 通过该目 标核函数调用预先创建的目标设备函数， 以由该 目标设备函数对 该第一数组进行计算， 得到该第 一数组的计算结果； 在运行该目标核函数的过程 中， 读取该目标本地变量所指向的该第一数组的 计算结果， 以由该目标核函数基于该第一数组的 计算结果对 该第二数组进行计算。 采用本方法能 够降低计算时延。 权利要求书2页 说明书10页 附图6页 CN 114490041 A 2022.05.13 CN 114490041 A 1.一种数组计算方法， 其特 征在于， 用于GPU计算平台 中， 所述方法包括： 调用目标核函数， 所述目标核函数定义有目标本地变量， 所述目标本地变量指向第一 数组的计算结果， 所述 目标核函数用于对第二数组进行计算， 所述第二数组的计算依赖于 所述第一数组的计算结果； 通过所述目标核函数调用预先创建的目标设备函数， 以由所述目标设备函数对所述第 一数组进行计算， 得到所述第一数组的计算结果； 在运行所述目标核函数的过程中， 读取所述目标本地变量所指向的所述第 一数组的计 算结果， 以由所述目标核函数基于所述第一数组的计算结果对所述第二数组进行计算。 2.根据权利要求1所述的方法， 其特征在于， 所述通过所述目标核函数调用预先创建的 目标设备函数， 以由所述 目标设备函数对所述第一数组进行计算， 得到所述第一数组的计 算结果之后， 所述方法还 包括： 将所述第一数组的计算结果存 储于所述目标本地变量对应的线程寄存器； 对应地， 所述读取 所述目标本地变量所指向的所述第一数组的计算结果， 包括： 从所述目标本地变量对应的线程寄存器中读取 所述第一数组的计算结果。 3.根据权利要求1所述的方法， 其特 征在于， 所述方法还 包括： 在对所述第二数组进行计算之前， 所述目标核函数从内存中读取 所述第二数组； 在对所述第 二数组计算之后， 所述目标核函数将所述第 二数组的计算结果存储至 内存 中。 4.根据权利要求1所述的方法， 其特征在于， 所述目标核函数定义有多个所述目标本地 变量， 各所述目标本地变量指向所述第一数组的不同元 素的计算结果。 5.根据权利要求1所述的方法， 其特征在于， 所述GPU计算平台包括主机端和设备端， 所 述调用目标核函数， 包括： 所述主机端向所述设备端发送调用指令， 以调用所述目标核函数。 6.根据权利要求1至5任一所述的方法， 其特征在于， 所述GPU计算平台为ROCm计算平 台。 7.一种数组计算装置， 其特 征在于， 用于GPU计算平台 中， 所述装置包括： 第一调用模块， 用于调用目标核函数， 所述目标核函数定义有目标本地变量， 所述目标 本地变量指向第一数组的计算结果， 所述 目标核函数用于对第二数组进行计算， 所述第二 数组的计算依赖 于所述第一数组的计算结果； 第二调用模块， 用于通过所述目标核函数调用预先创建的目标设备函数， 以由所述目 标设备函数对所述第一数组进行计算， 得到所述第一数组的计算结果； 计算模块， 用于在运行所述目标核函数的过程中， 读取所述目标本地变量所指向的所 述第一数组的计算结果， 以由所述目标核函数基于所述第一数组的计算结果对所述第二数 组进行计算。 8.一种计算机设备， 包括存储器和处理器， 所述存储器存储有计算机程序， 其特征在 于， 所述处 理器执行所述计算机程序时实现权利要求1至 6中任一项所述的方法的步骤。 9.一种计算机可读存储介质， 其上存储有计算机程序， 其特征在于， 所述计算机程序被 处理器执行时实现权利要求1至 6中任一项所述的方法的步骤。 10.一种计算机程序产品， 包括计算机程序， 其特征在于， 该计算机程序被处理器执行权　利　要　求　书 1/2 页 2 CN 114490041 A 2时实现权利要求1至 6中任一项所述的方法的步骤。权　利　要　求　书 2/2 页 3 CN 114490041 A 3