EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Lint]Style: Convert vllm-ascend/ to ruff format(Batch #4) (#6200)

Browse Source 
### What this PR does / why we need it?
**Scope of Changes**:
| File Path |
| :--- |
| `vllm_ascend/distributed/kv_transfer/__init__.py` |
| `vllm_ascend/distributed/kv_transfer/kv_p2p/mooncake_connector.py` |
|
`vllm_ascend/distributed/kv_transfer/kv_p2p/mooncake_layerwise_connector.py`
|

### Does this PR introduce _any_ user-facing change?

### How was this patch tested?

- vLLM version: v0.14.0
- vLLM main:
d68209402d

Signed-off-by: MrZ20 <2609716663@qq.com>
This commit is contained in:
SILONG ZENG
2026-01-24 20:40:48 +08:00
committed by GitHub
parent fbae41697e
commit 153da1a669
4 changed files with 963 additions and 1041 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
vllm_ascend/distributed/kv_transfer/__init__.py
View File

@@ -15,31 +15,32 @@
# limitations under the License.
#
from vllm.distributed.kv_transfer.kv_connector.factory import \
KVConnectorFactory
from vllm.distributed.kv_transfer.kv_connector.factory import KVConnectorFactory
def register_connector():
KVConnectorFactory.register_connector(
"MooncakeConnectorV1",
"vllm_ascend.distributed.kv_transfer.kv_p2p.mooncake_connector",
"MooncakeConnector")
"MooncakeConnectorV1", "vllm_ascend.distributed.kv_transfer.kv_p2p.mooncake_connector", "MooncakeConnector"
)
KVConnectorFactory.register_connector(
"MooncakeConnectorStoreV1",
"vllm_ascend.distributed.kv_transfer.kv_pool.ascend_store.ascend_store_connector",
"AscendStoreConnector")
"AscendStoreConnector",
)
KVConnectorFactory.register_connector(
"AscendStoreConnector",
"vllm_ascend.distributed.kv_transfer.kv_pool.ascend_store.ascend_store_connector",
"AscendStoreConnector")
"AscendStoreConnector",
)
KVConnectorFactory.register_connector(
"MooncakeLayerwiseConnector",
"vllm_ascend.distributed.kv_transfer.kv_p2p.mooncake_layerwise_connector",
"MooncakeLayerwiseConnector")
"MooncakeLayerwiseConnector",
)
KVConnectorFactory.register_connector(
"UCMConnector", "vllm_ascend.distributed.kv_transfer.kv_pool.ucm_connector",
"UCMConnectorV1")
"UCMConnector", "vllm_ascend.distributed.kv_transfer.kv_pool.ucm_connector", "UCMConnectorV1"
)