System and method to securely broadcast a message to accelerators

A non-transitory machine-readable medium having instructions stored therein, which when executed by a processor, cause the processor to perform one or more operations, the operations comprising:

• receiving a broadcast request from a host that hosts an application that initiated a broadcast message to be broadcast to two or more DP accelerators of a plurality of DP accelerators coupled to the host, wherein the broadcast request includes two or more DP accelerator identifiers (IDs) identifying the two or more DP accelerators;
• receiving from the host a broadcast session key for a broadcast communication session to broadcast the broadcast message;
• and
• for each of the one or more DP accelerator IDs:
• identifying a public key of a security key pair corresponding to the DP accelerator ID,
• encrypting the broadcast message using the broadcast session key,
• encrypting the broadcast session key using the public key, and
• transmitting the encrypted broadcast message and the encrypted broadcast session key to the DP accelerator identified by the DP accelerator ID, wherein the DP accelerator is configured to decrypt the encrypted broadcast session key using a corresponding private key associated with the DP accelerator ID and to decrypt the encrypted broadcast message using the broadcast session key, wherein one or more adjacent DP accelerators of the two or more DP accelerators propagates the broadcast by receiving the broadcast and forwarding the broadcast to their adjacent DP accelerators and the broadcast message is communicated via peripheral component interconnect or peripheral component interconnect express channel.

A data processing system, comprising:

• a processor;
• and
• a memory coupled to the processor to store instructions, which when executed by the processor, cause the processor to perform operations, the operations including receiving a broadcast request from a host that hosts an application that initiated a broadcast message to be broadcast to two or more DP accelerators of a plurality of DP accelerators coupled to the host, wherein the broadcast request includes two or more DP accelerator identifiers (IDs) identifying the two or more DP accelerators;
• receiving from the host a broadcast session key for a broadcast communication session to broadcast the broadcast message;
• and
• for each of the one or more DP accelerator IDs:
• identifying a public key of a security key pair corresponding to the DP accelerator ID,
• encrypting the broadcast message using the broadcast session key,
• encrypting the broadcast session key using the public key, and
• transmitting the encrypted broadcast message and the encrypted broadcast session key to the DP accelerator identified by the DP accelerator ID, wherein the DP accelerator is configured to decrypt the encrypted broadcast session key using a corresponding private key associated with the DP accelerator ID and to decrypt the encrypted broadcast message using the broadcast session key, wherein one or more adjacent DP accelerators of the two or more DP accelerators propagates the broadcast by receiving the broadcast and forwarding the broadcast to their adjacent DP accelerators and the broadcast message is communicated via peripheral component interconnect or peripheral component interconnect express channel.