recv_multiple_tensors#

abstractmethod TensorTransportManager.recv_multiple_tensors(obj_id: str, tensor_transport_metadata: TensorTransportMetadata, communicator_metadata: CommunicatorMetadata, target_buffers: List[Any] | None = None) → List[Any][source]#

Receives tensors on the destination actor. Ray calls this on the destination actor during the transfer.

Parameters:

  • obj_id (str) – The object ID for related GPU object.

  • tensor_transport_metadata (TensorTransportMetadata) – The tensor transport metadata for the GPU object.

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

  • target_buffers (List[Any] | None) – Pre-allocated buffers to receive the tensors into if possible.

Returns:

The received tensors.

Return type:

List[Any]