chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
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",
|
||||
]
|
||||
Reference in New Issue
Block a user