ray.experimental.TensorTransportManager.garbage_collect#

abstract TensorTransportManager.garbage_collect(obj_id: str, tensor_transport_meta: TensorTransportMetadata, tensors: List[Any])[source]#

Clean up resources for an RDT object. Ray calls this on the source actor after Ray’s distributed reference counting protocol determines the object is out of scope.

Use this to release any resources your transport allocated, such as deregistering memory buffers. On the receiver side, no cleanup is needed — Ray does not hold onto the tensor after returning it to the user, so it is garbage collected normally when the user releases it.

Parameters:

  • obj_id – The ID of the GPU object to garbage collect.

  • tensor_transport_meta – The tensor transport metadata.

  • tensors – The tensors that are contained in the ObjectRef that is being freed.