pytorch/torch/distributed/_tools
PyTorch MergeBot 6374332d33 Revert "PEP585 update - torch/distributed (#145164)"
This reverts commit 6cb186e279.

Reverted https://github.com/pytorch/pytorch/pull/145164 on behalf of https://github.com/huydhn due to Sorry for reverting your change but it is failing an inductor test ([comment](https://github.com/pytorch/pytorch/pull/145164#issuecomment-2602875679))
2025-01-20 16:46:46 +00:00
..
__init__.py Selective Activation Checkpointing (SAC) Estimator for estimating memory and recomputation time trade-offs. (#135208) 2024-10-14 13:56:40 +00:00
fsdp2_mem_tracker.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
ilp_utils.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
mem_tracker.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
memory_tracker.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
mod_tracker.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
runtime_estimator.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
sac_estimator.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00
sac_ilp.py Revert "PEP585 update - torch/distributed (#145164)" 2025-01-20 16:46:46 +00:00