mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
e9b3bc2ead
Exit seed receiving section only when all ranks received seed, otherwise we are at risk that current rank will reach same section of the code again while rank zero still in the previous iteration Fixes: #80845 Differential Revision: [D37702557](https://our.internmc.facebook.com/intern/diff/D37702557) Pull Request resolved: https://github.com/pytorch/pytorch/pull/81071 Approved by: https://github.com/msaroufim, https://github.com/ejguan |
||
|---|---|---|
| .. | ||
| backcompat | ||
| benchmark | ||
| bottleneck | ||
| data | ||
| hipify | ||
| jit | ||
| model_dump | ||
| tensorboard | ||
| __init__.py | ||
| _cpp_extension_versioner.py | ||
| _crash_handler.py | ||
| _freeze.py | ||
| _mode_utils.py | ||
| _python_dispatch.py | ||
| _pytree.py | ||
| _zip.py | ||
| bundled_inputs.py | ||
| checkpoint.py | ||
| collect_env.py | ||
| cpp_extension.py | ||
| dlpack.py | ||
| file_baton.py | ||
| hooks.py | ||
| mkldnn.py | ||
| mobile_optimizer.py | ||
| model_zoo.py | ||
| show_pickle.py | ||
| throughput_benchmark.py | ||