Skip to content

GitLab

Unverified Commit f80b303c authored by Tom Birch's avatar Tom Birch Committed by GitHub
Browse files

fix bug (#193)

parent d85acf72
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/nn/pipe_process/test_bugs.py
View file @ f80b303c
...@@ -65,6 +65,7 @@ def python_autograd_function(pipeline_style): ...@@ -65,6 +65,7 @@ def python_autograd_function(pipeline_style):
if model.group.rank() == 1: if model.group.rank() == 1:
assert torch.allclose(x, y) assert torch.allclose(x, y)
torch.distributed.rpc.shutdown()
torch.distributed.barrier() torch.distributed.barrier()
......
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