11 lines
333 B
Python
11 lines
333 B
Python
|
import pytest
|
||
|
|
import torch.distributed
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(autouse=True)
|
||
|
|
def teardown_process_group():
|
||
|
|
"""Ensures that the distributed process group gets closed before the next test runs."""
|
||
|
|
yield
|
||
|
|
if torch.distributed.is_available() and torch.distributed.is_initialized():
|
||
|
|
torch.distributed.destroy_process_group()
|