[发明专利]使用具有交换机的虚拟信道向加速器安全地广播消息的系统和方法

说明书

实施方式公开了在虚拟DP加速器(DPA)之间广播消息的系统和方法。在一个实施方式中，响应于经由通信交换机从应用程序接收到广播指令，该广播指令指定多个虚拟DP加速器中的一个或多个虚拟DP加速器接收广播消息，系统基于用于广播通信会话的广播会话密钥对广播消息进行加密。系统确定各自与指定的虚拟DP加速器中的一个相关联的一个或多个安全密钥对的一个或多个公共密钥。系统基于所确定的一个或多个公共密钥对多个广播会话密钥进行加密。系统将加密的广播消息以及一个或多个加密的广播会话密钥广播给虚拟DP加速器。

技术领域

本公开的实施方式总体涉及人工智能模型训练和推断。更具体地，本公开的实施方式涉及在配置为彼此通信的数据处理加速器之间共享密钥。

背景技术

配置为与主机计算设备通信的数据处理加速器(DPA)通常不能彼此安全地通信。DPA之间的通信(例如，对等通信)是有用的，使得两个或更多个DPA可协作和协调以代表主机计算设备执行数据处理任务。然而，重要的是DPA安全地通信，从而安全地执行由协作DPA执行的处理任务，使得另一计算实体不能改变由通信DPA产生的结果，并且不能够从任何通信DPA窃取代码或数据。

发明内容

在第一方面，本公开提供了一种向一个或多个虚拟数据处理(DP)加速器广播消息的计算机实施方法，该方法包括：经由通信交换机从由主机主管的应用程序接收广播指令，以向多个虚拟DP加速器中的一个或多个发送广播消息，每个虚拟DP加速器均通过虚拟通信信道通信；使用与广播通信会话相关联的广播会话密钥对广播消息进行加密；确定一个或多个安全密钥对的一个或多个公共密钥，每个安全密钥对均与一个或多个虚拟DP加速器中的一个相关联；使用所确定的公共密钥中的每个对广播会话密钥进行加密，生成与一个或多个虚拟DP加速器相对应的一个或多个加密广播会话密钥；以及将加密的广播消息和一个或多个加密广播会话密钥向虚拟DP加速器广播，其中，虚拟DP加速器中的每个均使用与虚拟DP加速器相关联的相应私有密钥来对加密广播会话密钥进行解密，其中，基于广播会话密钥来解密消息。

在第二方面，本公开提供了一种存储有指令的非暂时性机器可读介质，该指令在由处理器执行时致使处理器执行一个或多个操作，该操作包括：经由通信交换机从由主机主管的应用程序接收广播指令，以向多个虚拟DP加速器中的一个或多个发送广播消息，每个虚拟DP加速器均通过虚拟通信信道通信；使用与广播通信会话相关联的广播会话密钥对广播消息进行加密；确定一个或多个安全密钥对的一个或多个公共密钥，每个安全密钥对均与一个或多个虚拟DP加速器中的一个相关联；使用所确定的公共密钥中的每个对广播会话密钥进行加密，生成与一个或多个虚拟DP加速器相对应的一个或多个加密广播会话密钥；以及将加密的广播消息和一个或多个加密广播会话密钥向虚拟DP加速器广播，其中，虚拟DP加速器中的每个均使用与虚拟DP加速器相关联的相应私有密钥来对加密广播会话密钥进行解密，其中，基于广播会话密钥来解密消息。

在第三方面，本公开提供了一种数据处理系统，该数据处理系统包括处理器和存储器，其中，存储器联接至处理器以存储指令，该指令在由处理器执行时致使处理器执行操作，该操作包括：经由通信交换机从由主机主管的应用程序接收广播指令，以向多个虚拟DP加速器中的一个或多个发送广播消息，每个虚拟DP加速器均通过虚拟通信信道通信；使用与广播通信会话相关联的广播会话密钥对广播消息进行加密；确定一个或多个安全密钥对的一个或多个公共密钥，每个安全密钥对均与一个或多个虚拟DP加速器中的一个相关联；使用所确定的公共密钥中的每个对广播会话密钥进行加密，生成与一个或多个虚拟DP加速器相对应的一个或多个加密广播会话密钥；以及将加密的广播消息和一个或多个加密广播会话密钥向虚拟DP加速器广播，其中，虚拟DP加速器中的每个均使用与虚拟DP加速器相关联的相应私有密钥来对加密广播会话密钥进行解密，其中，基于广播会话密钥来解密消息。

附图说明

本公开的实施方式在附图的各图中以举例而非限制的方式示出，附图中的相同参考标记指示相似元件。

图1是示出根据一个实施方式的安全处理系统的框图。