pytorch/torch/distributed/tensor/parallel
2025-03-21 03:59:25 +00:00
..
__init__.py
_data_parallel_utils.py Migrate from Tuple -> tuple in torch/distributed (#144258) 2025-01-10 08:34:54 +00:00
_utils.py Migrate from Tuple -> tuple in torch/distributed (#144258) 2025-01-10 08:34:54 +00:00
api.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
ddp.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
fsdp.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
input_reshard.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00
loss.py [BE][Ez]: Remove extra copy in dtensor parallel loss (#148096) 2025-02-28 05:42:32 +00:00
style.py [Distributed] Add repr methods for ParallelStyles (#149478) 2025-03-21 03:59:25 +00:00