[CI][AMD]BugFix] Fix deadlock occuring in test_moe_layer (#40767)

Signed-off-by: Randall Smith <Randall.Smith@amd.com>
2026-04-25 09:34:14 -04:00
parent 6646c0c7e0
commit 1e9f19ca3f
1 changed files with 1 additions and 0 deletions
@@ -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)