20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[MISC] fix typo in kv transfer send recv test (#11983)

Browse Source
This commit is contained in:
Yangcheng Li 2025-01-13 13:07:48 +08:00 committed by GitHub
parent 619ae268c3
commit f7b3ba82c3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/kv_transfer/test_send_recv.py
View File

@ -22,13 +22,13 @@ def test_run(my_rank, pipe):
x2 = pipe.recv_tensor() x2 = pipe.recv_tensor()
print(f"rank {my_rank} received x2 = ", x2) print(f"rank {my_rank} received x2 = ", x2)
y2 = pipe.recv_tensor() y2 = pipe.recv_tensor()
print(f"rank {my_rank} received y2 = ", x2) print(f"rank {my_rank} received y2 = ", y2)
else: else:
x2 = pipe.recv_tensor() x2 = pipe.recv_tensor()
print(f"rank {my_rank} received x2 = ", x2) print(f"rank {my_rank} received x2 = ", x2)
y2 = pipe.recv_tensor() y2 = pipe.recv_tensor()
print(f"rank {my_rank} received y2 = ", x2) print(f"rank {my_rank} received y2 = ", y2)
pipe.send_tensor(x) pipe.send_tensor(x)
print(f"rank {my_rank} sent tensor x") print(f"rank {my_rank} sent tensor x")
pipe.send_tensor(y) pipe.send_tensor(y)