ray.experimental.TensorTransportManager.send_multiple_tensors#

abstract TensorTransportManager.send_multiple_tensors(tensors: List[Any], tensor_transport_metadata: TensorTransportMetadata, communicator_metadata: CommunicatorMetadata)[source]#

Sends tensors from the source actor to the destination actor. Ray calls this on the source actor during the transfer. Implement this to perform the actual data transfer using your transport’s send mechanism. For one-sided transports, you can simply avoid implementing this method or even raise a NotImplementedError to ensure it’s not being called.

Parameters:

  • tensors – The tensors or jax arrays to send.

  • tensor_transport_metadata – The tensor transport metadata for the RDT object.

  • communicator_metadata – The communicator metadata for the send/recv operation.