wehub-resource-sync
31 lines
826 B
Python
31 lines
826 B
Python
from ray.experimental.dynamic_resources import set_resource
|
|
from ray.experimental.locations import get_local_object_locations, get_object_locations
|
|
from ray.experimental.rdt import (
|
|
CommunicatorMetadata,
|
|
RDTManager,
|
|
TensorTransportManager,
|
|
TensorTransportMetadata,
|
|
deregister_nixl_memory,
|
|
register_nixl_memory,
|
|
register_nixl_memory_pool,
|
|
register_tensor_transport,
|
|
set_target_for_ref,
|
|
wait_tensor_freed,
|
|
)
|
|
|
|
__all__ = [
|
|
"get_object_locations",
|
|
"get_local_object_locations",
|
|
"set_resource",
|
|
"RDTManager",
|
|
"wait_tensor_freed",
|
|
"register_tensor_transport",
|
|
"register_nixl_memory",
|
|
"deregister_nixl_memory",
|
|
"register_nixl_memory_pool",
|
|
"TensorTransportManager",
|
|
"TensorTransportMetadata",
|
|
"CommunicatorMetadata",
|
|
"set_target_for_ref",
|
|
]
|