Skip to content

Commit

Permalink
[MISC] fix typo in kv transfer send recv test (vllm-project#11983)
Browse files Browse the repository at this point in the history
Signed-off-by: Bowen Wang <[email protected]>
  • Loading branch information
yyccli authored and abmfy committed Jan 24, 2025
1 parent 9b84523 commit 2f5fa84
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/kv_transfer/test_send_recv.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,13 @@ def test_run(my_rank, pipe):
x2 = pipe.recv_tensor()
print(f"rank {my_rank} received x2 = ", x2)
y2 = pipe.recv_tensor()
print(f"rank {my_rank} received y2 = ", x2)
print(f"rank {my_rank} received y2 = ", y2)

else:
x2 = pipe.recv_tensor()
print(f"rank {my_rank} received x2 = ", x2)
y2 = pipe.recv_tensor()
print(f"rank {my_rank} received y2 = ", x2)
print(f"rank {my_rank} received y2 = ", y2)
pipe.send_tensor(x)
print(f"rank {my_rank} sent tensor x")
pipe.send_tensor(y)
Expand Down

0 comments on commit 2f5fa84

Please sign in to comment.