mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
c978b609f7
The conventional env var to set is CI. Both circle and GHA set it, so IN_CI is unnecessary Pull Request resolved: https://github.com/pytorch/pytorch/pull/79229 Approved by: https://github.com/janeyx99
38 lines
956 B
Python
38 lines
956 B
Python
#!/usr/bin/env python3
|
|
# Owner(s): ["oncall: distributed"]
|
|
|
|
import sys
|
|
|
|
import torch
|
|
import torch.distributed as dist
|
|
|
|
if not dist.is_available():
|
|
print("Distributed not available, skipping tests", file=sys.stderr)
|
|
sys.exit(0)
|
|
|
|
from torch.testing._internal.common_utils import IS_CI, run_tests
|
|
from torch.testing._internal.distributed.rpc.faulty_rpc_agent_test_fixture import (
|
|
FaultyRpcAgentTestFixture,
|
|
)
|
|
from torch.testing._internal.distributed.rpc_utils import (
|
|
FAULTY_AGENT_TESTS,
|
|
generate_tests,
|
|
)
|
|
|
|
|
|
# On CircleCI these tests are already run on CPU jobs, thus to save resources do
|
|
# not run them on GPU jobs, since thet wouldn't provide additional test signal.
|
|
if not (IS_CI and torch.cuda.is_available()):
|
|
globals().update(
|
|
generate_tests(
|
|
"Faulty",
|
|
FaultyRpcAgentTestFixture,
|
|
FAULTY_AGENT_TESTS,
|
|
__name__,
|
|
)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|