From 1e9f19ca3fd29f83442ab83b08d4642e691c95bd Mon Sep 17 00:00:00 2001 From: rasmith Date: Sat, 25 Apr 2026 08:34:14 -0500 Subject: [PATCH] [CI][AMD]BugFix] Fix deadlock occuring in test_moe_layer (#40767) Signed-off-by: Randall Smith --- tests/kernels/moe/test_moe_layer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/kernels/moe/test_moe_layer.py b/tests/kernels/moe/test_moe_layer.py index 14cfd00c2b..89e28d950f 100644 --- a/tests/kernels/moe/test_moe_layer.py +++ b/tests/kernels/moe/test_moe_layer.py @@ -1632,6 +1632,7 @@ def _parallel_worker( if all2all_manager is not None: all2all_manager.destroy() total = total + 1 + torch.distributed.barrier() skipped = total - (passed + failed)