Files
vllm-omni/vllm_omni/distributed/__init__.py
T
e1b0b83238 add moriio transfer engine (#1742)
Signed-off-by: inkcherry <mingzhi.liu@amd.com>
Signed-off-by: zejwang  <zejwang@amd.com>
Signed-off-by: junkang1991 <junkangchow@gmail.com>
Signed-off-by: Zejian Wang <zejianwang@sjtu.edu.cn>
Co-authored-by: zejwang <zejwang@amd.com>
Co-authored-by: junkang1991 <junkangchow@gmail.com>
Co-authored-by: Zejian Wang <zejianwang@sjtu.edu.cn>
Co-authored-by: Zejian Wang <Zejian.Wang@amd.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
2026-06-10 08:37:33 +08:00

36 lines
904 B
Python

# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from .omni_connectors import (
ConnectorSpec,
MooncakeConnector,
MooncakeStoreConnector,
MooncakeTransferEngineConnector,
MoriTransferEngineConnector,
OmniConnectorBase,
OmniConnectorFactory,
OmniTransferConfig,
SharedMemoryConnector,
YuanrongConnector,
YuanrongTransferEngineConnector,
load_omni_transfer_config,
)
__all__ = [
# Config
"ConnectorSpec",
"OmniTransferConfig",
# Connectors
"OmniConnectorBase",
"OmniConnectorFactory",
"MooncakeConnector", # compat alias
"MooncakeStoreConnector",
"MooncakeTransferEngineConnector",
"MoriTransferEngineConnector",
"SharedMemoryConnector",
"YuanrongConnector",
"YuanrongTransferEngineConnector",
# Utilities
"load_omni_transfer_config",
]