__init__.py 627 B

12345678910111213141516171819202122
  1. from ray.experimental.dynamic_resources import set_resource
  2. from ray.experimental.gpu_object_manager import (
  3. CommunicatorMetadata,
  4. GPUObjectManager,
  5. TensorTransportManager,
  6. TensorTransportMetadata,
  7. register_tensor_transport,
  8. wait_tensor_freed,
  9. )
  10. from ray.experimental.locations import get_local_object_locations, get_object_locations
  11. __all__ = [
  12. "get_object_locations",
  13. "get_local_object_locations",
  14. "set_resource",
  15. "GPUObjectManager",
  16. "wait_tensor_freed",
  17. "register_tensor_transport",
  18. "TensorTransportManager",
  19. "TensorTransportMetadata",
  20. "CommunicatorMetadata",
  21. ]