mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
43ad172c54
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56549 This make the `kProcessGroupDefaultTimeout` be the same as the python side, and python side directly use the pybind value instead Test Plan: Imported from OSS Reviewed By: rohan-varma Differential Revision: D27899190 Pulled By: wanchaol fbshipit-source-id: 388a7f42358b0abed75cf4934fb7b311fd33fee6
8 lines
422 B
Python
8 lines
422 B
Python
from torch._C._distributed_c10d import _DEFAULT_PG_TIMEOUT
|
|
# Default process group wide timeout, if applicable.
|
|
# This only applies to the gloo and nccl backends
|
|
# (only if NCCL_BLOCKING_WAIT or NCCL_ASYNC_ERROR_HANDLING is set to 1).
|
|
# To make an attempt at backwards compatibility with THD, we use an
|
|
# extraordinarily high default timeout, given that THD did not have timeouts.
|
|
default_pg_timeout = _DEFAULT_PG_TIMEOUT
|