mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
34051d74da
Summary: Action based on https://github.com/pytorch/pytorch/issues/66232 cc pietern mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse SciPioneer H-Huang Pull Request resolved: https://github.com/pytorch/pytorch/pull/66797 Reviewed By: gchanan Differential Revision: D31761389 Pulled By: janeyx99 fbshipit-source-id: c27c9ab4acec1eb71d5edd4538cd113b770dfc6c
38 lines
962 B
Python
38 lines
962 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_IN_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_IN_CI and torch.cuda.is_available()):
|
|
globals().update(
|
|
generate_tests(
|
|
"Faulty",
|
|
FaultyRpcAgentTestFixture,
|
|
FAULTY_AGENT_TESTS,
|
|
__name__,
|
|
)
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|