/usr/local/lib64/python3.6/site-packages/torch/distributed/rpc/_testing
NameSizeModeActions
__pycache__/-0755rm
faulty_agent_backend_registry.py26600644editdlrm
__init__.py4710644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributed/rpc/_testing/__init__.py (471B)
import torch def is_available(): return hasattr(torch._C, "_faulty_agent_init") if is_available() and not torch._C._faulty_agent_init(): raise RuntimeError("Failed to initialize torch.distributed.rpc._testing") if is_available(): # Registers FAULTY_TENSORPIPE RPC backend. from . import faulty_agent_backend_registry from torch._C._distributed_rpc_testing import ( FaultyTensorPipeRpcBackendOptions, FaultyTensorPipeAgent, )