wehub-resource-sync
30 lines
712 B
Python
30 lines
712 B
Python
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",
|
|
]
|