mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-06 10:44:42 +08:00
Fix test
Signed-off-by: ilmarkov <markovilya197@gmail.com>
This commit is contained in:
parent
699ea97108
commit
6014dc26d3
@ -312,7 +312,7 @@ def _test_async_transfer_layer_without_mtp_worker(
|
|||||||
is_received_locally=is_received_locally,
|
is_received_locally=is_received_locally,
|
||||||
recv_metadata=recv_metadata,
|
recv_metadata=recv_metadata,
|
||||||
new_indices=new_indices_cpu[layer_idx],
|
new_indices=new_indices_cpu[layer_idx],
|
||||||
ep_group=ep_group,
|
ep_rank=ep_group.rank(),
|
||||||
)
|
)
|
||||||
|
|
||||||
verify_expert_weights_after_shuffle(
|
verify_expert_weights_after_shuffle(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user