/usr/local/lib64/python3.6/site-packages/torch/distributed/rpc/_testing
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,
)