[发明专利]异构加速计算优化方法、装置、设备及可读存储介质

说明书

本申请公开了一种异构加速计算优化方法、装置、设备及可读存储介质，所述异构加速计算优化方法包括：获取明文数据，并基于第一内存拷贝算子集合，将所述明文数据由CPU内存拷贝至异构芯片内存；基于密态计算算子集合，对所述明文数据进行常驻于所述异构芯片内存的密态计算，获得密态计算结果；将所述密态计算结果反馈至所述CPU内存。本申请解决了异构联邦学习框架计算效率低的技术问题。

技术领域

本申请涉及通信技术领域，尤其涉及一种异构加速计算优化方法、装置、设备及可读存储介质。

背景技术

随着计算机技术的不断发展，人工智能的应用也越来越广泛，目前，在通过联邦学习联合多方数据进行建模时，联邦学习的参与方通常需要进行海量的密态计算，而在异构联邦学习框架中，为了提高数据的计算效果，在联邦学习的一轮迭代过程中，通常需要将密文数据在CPU内存和异构芯片内存之间多次来回进行拷贝，但是由于密文数据的位数通常较高，密文数据内存拷贝耗时较长，进而将导致在联邦学习的一轮迭代过程中，密文数据的内存拷贝时间将远大于密文数据的密态计算时间，使得异构联邦学习框架的计算效率较低。

发明内容

本申请的主要目的在于提供一种异构加速计算优化方法、装置、设备及可读存储介质，旨在解决现有技术中异构联邦学习框架计算效率低的技术问题。

为实现上述目的，本申请提供一种异构加速计算优化方法，所述异构加速计算优化方法应用于异构加速计算优化设备，所述异构加速计算优化方法包括：

获取明文数据，并基于第一内存拷贝算子集合，将所述明文数据由CPU内存拷贝至异构芯片内存；

基于密态计算算子集合，对所述明文数据进行常驻于所述异构芯片内存的密态计算，获得密态计算结果；

将所述密态计算结果反馈至所述CPU内存。

本申请还提供一种异构加速计算优化装置，所述异构加速计算优化装置为虚拟装置，且所述异构加速计算优化装置应用于异构加速计算优化设备，所述异构加速计算优化装置包括：

内存拷贝模块，用于获取明文数据，并基于第一内存拷贝算子集合，将所述明文数据由CPU内存拷贝至异构芯片内存；

密态计算模块，用于基于密态计算算子集合，对所述明文数据进行常驻于所述异构芯片内存的密态计算，获得密态计算结果；

反馈模块，用于将所述密态计算结果反馈至所述CPU内存。

本申请还提供一种异构加速计算优化设备，所述异构加速计算优化设备为实体设备，所述异构加速计算优化设备包括：存储器、处理器以及存储在所述存储器上并可在所述处理器上运行的所述异构加速计算优化方法的程序，所述异构加速计算优化方法的程序被处理器执行时可实现如上述的异构加速计算优化方法的步骤。

本申请还提供一种可读存储介质，所述可读存储介质上存储有实现异构加速计算优化方法的程序，所述异构加速计算优化方法的程序被处理器执行时实现如上述的异构加速计算优化方法的步骤。

本申请还提供一种计算机程序产品，包括计算机程序，所述计算机程序被处理器执行时实现如上述的异构加速计算优化方法的步骤。