Andrew Gu a8b1767ae5 [DTensor] Fix local_map with multi-threading (#149070) ... Using `nonlocal device_mesh` is not safe with multi-threading Pull Request resolved: https://github.com/pytorch/pytorch/pull/149070 Approved by: https://github.com/wanchaol		2025-03-13 10:58:59 +00:00
..
__init__.py	PEP585 update - torch/distributed/tensor (#145141)	2025-01-18 20:01:59 +00:00
_attention.py	[dtensor] add CuDNN SDPA op support to DTensor (#148537)	2025-03-06 23:44:40 +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