| 12345678910111213141516171819202122 |
- from ray.experimental.dynamic_resources import set_resource
- from ray.experimental.gpu_object_manager import (
- CommunicatorMetadata,
- GPUObjectManager,
- TensorTransportManager,
- TensorTransportMetadata,
- register_tensor_transport,
- wait_tensor_freed,
- )
- from ray.experimental.locations import get_local_object_locations, get_object_locations
- __all__ = [
- "get_object_locations",
- "get_local_object_locations",
- "set_resource",
- "GPUObjectManager",
- "wait_tensor_freed",
- "register_tensor_transport",
- "TensorTransportManager",
- "TensorTransportMetadata",
- "CommunicatorMetadata",
- ]
|