pytorch/torch/distributed/tensor/experimental
2025-05-09 20:40:54 +00:00
..
__init__.py PEP585 update - torch/distributed/tensor (#145141) 2025-01-18 20:01:59 +00:00
_attention.py [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
_func_map.py [DTensor] Fix local_map with multi-threading (#149070) 2025-03-13 10:58:59 +00:00
_register_sharding.py [dtensor] refactor sharding prop to handle cross mesh computation (#147869) 2025-03-04 18:30:44 +00:00
_tp_transform.py [BE][PYFMT] migrate PYFMT for torch.{distributed,distributions} to ruff format (#144547) 2025-02-28 07:35:56 +00:00