Skip to content

Commit 1e9f19c

Browse files
authored
[CI][AMD]BugFix] Fix deadlock occuring in test_moe_layer (vllm-project#40767)
Signed-off-by: Randall Smith <Randall.Smith@amd.com>
1 parent 6646c0c commit 1e9f19c

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tests/kernels/moe/test_moe_layer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1632,6 +1632,7 @@ def _parallel_worker(
16321632
if all2all_manager is not None:
16331633
all2all_manager.destroy()
16341634
total = total + 1
1635+
torch.distributed.barrier()
16351636

16361637
skipped = total - (passed + failed)
16371638

0 commit comments

Comments
 (0)