pytorch/torch/distributed/_tools
Yuanyuan Chen 030de07aff [2/N] Use 'is' in callable comparisons (#166685)
It is generally advised to use `is/is not` for comparisons against torch functions.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166685
Approved by: https://github.com/xmfan, https://github.com/mlazos
2025-10-31 08:08:07 +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
common_utils.py Add support for non functional collectives under FakeTensorMode and fake_pg for memory tracking (#147566) 2025-03-08 18:00:49 +00:00
fake_collectives.py Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)" 2025-09-25 13:47:46 +00:00
fsdp2_mem_tracker.py [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
ilp_utils.py [2/N] More ruff SIM fixes (#165031) 2025-10-14 14:22:54 +00:00
mem_tracker.py [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
memory_tracker.py [2/N] Use 'is' in callable comparisons (#166685) 2025-10-31 08:08:07 +00:00
mod_tracker.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
runtime_estimator.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
sac_estimator.py Use correct pyrefly syntax in suppressions distributed/... (#166241) 2025-10-26 04:16:41 +00:00
sac_ilp.py PEP585 update - torch/distributed (#145164) 2025-01-21 04:23:29 +00:00