ray.experimental.TensorTransportManager

class ray.experimental.TensorTransportManager

Bases: ABC

Interface with which to implement custom tensor transports.

Methods

__init__
abort_transport	Aborts an in-progress transfer.
actor_has_tensor_transport	Whether the actor has the tensor transport available.
can_abort_transport	Indicates whether your transport can safely abort an in-progress transfer.
extract_tensor_transport_metadata	Implement this method to create the TensorTransportMetadata you defined previously.
garbage_collect	Clean up resources for an RDT object.
get_communicator_metadata	Gets the CommunicatorMetadata for a send/recv.
is_one_sided	Indicates whether your transport uses one-sided communication where only the receiver initiates the transfer.
recv_multiple_tensors	Receives tensors on the destination actor.
send_multiple_tensors	Sends tensors from the source actor to the destination actor.
tensor_transport_backend	Returns the name of your tensor transport backend.