A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 6, 15:45 UTC
[CI][BugFix][MORI][AMD] Add transfer_id to kv transfer params for test (#37213)
Signed-off-by: Randall Smith <Randall.Smith@amd.com>
rasmith committed
Mar 16, 2026 at 15:22 UTC
2cc26c3a9973257d5fcc582f063915d52dded86f
1 file changed
+4
-1
tests/v1/kv_connector/unit/test_moriio_connector.py
+4
-1
@@ -84,10 +84,13 @@ def mock_parallel_groups():
84
yield mock_group
85
86
87
-def _setup_kv_transfer_request(request, remote_host="127.0.0.1", fake_port=4789):
87
+def _setup_kv_transfer_request(
88
+ request, remote_host="127.0.0.1", fake_port=4789, fake_transfer_id="0"
89
+):
90
"""Setup KV transfer parameters for a request."""
91
request.kv_transfer_params.update(
92
{
93
+ "transfer_id": fake_transfer_id,
94
"remote_notify_port": fake_port,
95
"remote_block_ids": None,
96
"remote_host": remote_host,