pytorch/torch/utils/data/_utils
erjia 3ec9d34f21 Fix distributed store to use add for the counter of DL shared seed (#80348)
In order to get the result of `_shared_seed_recv_cnt` properly, switch from `store.get` to `store.add(key, 0)`.

See the comment from distributed team for the reason:
590d3e5774/torch/distributed/distributed_c10d.py (L242-L246)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80348
Approved by: https://github.com/VitalyFedyunin, https://github.com/NivekT
2022-06-27 21:59:17 +00:00
..
__init__.py Fix distributed store to use add for the counter of DL shared seed (#80348) 2022-06-27 21:59:17 +00:00
collate.py
fetch.py
pin_memory.py
serialization.py
signal_handling.py
worker.py