Signed-off-by: ilmarkov <markovilya197@gmail.com>
This commit is contained in:
ilmarkov 2025-12-16 15:02:53 +00:00
parent 699ea97108
commit 6014dc26d3

View File

@ -312,7 +312,7 @@ def _test_async_transfer_layer_without_mtp_worker(
is_received_locally=is_received_locally,
recv_metadata=recv_metadata,
new_indices=new_indices_cpu[layer_idx],
ep_group=ep_group,
ep_rank=ep_group.rank(),
)
verify_expert_weights_after_shuffle(