[发明专利]用于联邦学习的异构计算系统设备交互方案

说明书

本说明书的实施例提供用于控制异构计算系统的设备交互的方法。异构计算系统包括主处理设备和从处理设备，从处理设备用于执行主处理设备所调度任务的任务计算处理。从处理设备包括状态寄存器和配置寄存器。在执行设备间交互时，设备交互控制装置读取从处理设备处的状态寄存器的寄存器值。在所读取的寄存器值指示从处理设备处于处理准备就绪状态时，设备交互控制装置控制主处理设备与从处理设备执行所述处理所需的交互处理。利用该交互方法，可以在主处理设备和从处理设置之间实现可靠交互。

技术领域

本说明书实施例通常涉及异构计算领域，尤其涉及异构计算系统的设备交互控制方法、设备交互控制装置及异构计算系统。

背景技术

在联邦学习应用中，为了保护用户隐私数据，需要对用户隐私数据进行加密处理。通常采用的加密方法可以包括同态加密算法等。同态加密算法所涉及的计算比较复杂，并且中间计算结果数据量庞大。此外，同态加密算法还涉及对计算的实时性要求。这要求计算系统具有高效的计算性能。

为了满足上述针对计算系统的高效计算性能要求，提出了异构计算(Heterogeneous computer)系统。异构计算系统指使用不同体系架构的处理设备组成的系统。异构计算系统通常采用主-从处理设备架构。主处理设备负责异构计算系统中的处理任务调度和配置，并且通过比如PCIE的高速传输接口来将任务配置信息和处理任务源数据提供给从处理设备，以由从处理设备来具体执行。主-从处理设备之间的数据/信息交互效率，成为了影响异构计算系统的计算性能的重要因素。

发明内容

鉴于上述，本说明书的实施例提供一种异构计算系统的设备交互控制方法、设备交互控制装置及异构计算系统。利用该设备交互控制方法，可以实现异构计算系统中的主-从处理设备之间的可靠交互，从而提高异构计算系统的计算性能和计算效率。

根据本说明书的实施例的一个方面，提供一种用于控制异构计算系统的设备交互的方法，所述异构计算系统包括主处理设备和从处理设备，所述从处理设备用于执行所述主处理设备所调度任务的任务计算处理，所述从处理设备包括状态寄存器和配置寄存器，所述方法包括：读取所述从处理设备处的状态寄存器的寄存器值；以及在所读取的寄存器值指示所述从处理设备处于处理准备就绪状态时，控制所述主处理设备与所述从处理设备执行所述处理所需的交互处理。

可选地，在上述方面的一个示例中，所述处理可以包括写数据处理，以及所述状态寄存器包括写数据状态寄存器，所读取的寄存器值指示所述从处理设备处于数据接收就绪状态，控制所述主处理设备与所述从处理设备执行所述处理所需的交互处理可以包括：控制所述主处理设备将写数据配置信息写入所述从处理设备的配置寄存器，所述写数据配置信息包括任务配置信息以及任务计算源数据在所述从处理设备的从存储器的存储位置信息；以及控制所述主处理设备根据所述写数据配置信息，将所述任务计算源数据写入所述从处理设备的从存储器的相应存储位置。

可选地，在上述方面的一个示例中，所述处理可以包括任务计算处理，以及所述状态寄存器包括工作状态寄存器，所读取的寄存器值指示所述从处理设备处于任务计算就绪状态，控制所述主处理设备与所述从处理设备执行所述处理所需的交互处理可以包括：控制所述主处理设备将任务计算启动命令写入所述从处理设备的配置寄存器，所述从处理设备响应于接收到所述任务计算启动命令来启动任务计算处理。

可选地，在上述方面的一个示例中，所述处理可以包括读数据处理，以及所述状态寄存器包括读数据状态寄存器，所读取的寄存器值指示所述从处理设备处于数据输出就绪状态，控制所述主处理设备与所述从处理设备执行所述处理所需的交互处理可以包括：控制所述主处理设备从所述从处理设备的配置寄存器中读取读数据配置信息，所述读数据配置信息包括所述任务计算结果数据在所述从处理设备的从存储器中的存储位置信息；以及控制所述主处理设备根据所述读数据配置信息，从所述从处理设备的相应存储位置读取所述任务计算结果数据并写入所述主处理设备的主存储器中。