chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
from ray.experimental.rdt.rdt_manager import (
|
||||
RDTManager,
|
||||
set_target_for_ref,
|
||||
wait_tensor_freed,
|
||||
)
|
||||
from ray.experimental.rdt.tensor_transport_manager import (
|
||||
CommunicatorMetadata,
|
||||
TensorTransportManager,
|
||||
TensorTransportMetadata,
|
||||
)
|
||||
from ray.experimental.rdt.util import (
|
||||
deregister_nixl_memory,
|
||||
register_nixl_memory,
|
||||
register_nixl_memory_pool,
|
||||
register_tensor_transport,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
"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