Skip to content

GitLab

Unverified Commit 1e9f19ca authored by rasmith's avatar rasmith Committed by GitHub
Browse files

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


Signed-off-by: default avatarRandall Smith <Randall.Smith@amd.com>
parent 6646c0c7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/kernels/moe/test_moe_layer.py
View file @ 1e9f19ca
...@@ -1632,6 +1632,7 @@ def _parallel_worker( ...@@ -1632,6 +1632,7 @@ def _parallel_worker(
if all2all_manager is not None: if all2all_manager is not None:
all2all_manager.destroy() all2all_manager.destroy()
total = total + 1 total = total + 1
torch.distributed.barrier()
skipped = total - (passed + failed) skipped = total - (passed + failed)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment