/usr/local/lib64/python3.6/site-packages/torch/distributed/rpc
NameSizeModeActions
_testing/-0755rm
__pycache__/-0755rm
api.py343680644editdlrm
backend_registry.py120500644editdlrm
constants.py7830644editdlrm
functions.py72130644editdlrm
internal.py105480644editdlrm
options.py69800644editdlrm
rref_proxy.py15900644editdlrm
server_process_global_profiler.py81540644editdlrm
__init__.py92850644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/distributed/rpc/constants.py (783B)
from datetime import timedelta from torch._C._distributed_rpc import ( _DEFAULT_INIT_METHOD, _DEFAULT_NUM_WORKER_THREADS, _DEFAULT_RPC_TIMEOUT_SEC, _UNSET_RPC_TIMEOUT, ) # For any RpcAgent. DEFAULT_RPC_TIMEOUT_SEC: float = _DEFAULT_RPC_TIMEOUT_SEC DEFAULT_INIT_METHOD: str = _DEFAULT_INIT_METHOD DEFAULT_SHUTDOWN_TIMEOUT: float = 5.0 # For TensorPipeAgent. DEFAULT_NUM_WORKER_THREADS: int = _DEFAULT_NUM_WORKER_THREADS # Ensure that we don't time out when there are long periods of time without # any operations against the underlying ProcessGroup. DEFAULT_PROCESS_GROUP_TIMEOUT: timedelta = timedelta(milliseconds=2 ** 31 - 1) # Value indicating that timeout is not set for RPC call, and the default should be used. UNSET_RPC_TIMEOUT: float = _UNSET_RPC_TIMEOUT