fduwjj d4380edb9b [TP] Add API logging for TP high level API (#102209) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/102209 Approved by: https://github.com/wz337, https://github.com/wanchaol		2023-05-25 03:33:00 +00:00
..
__init__.py	[TP] Enable more generic attn in Tensor Parallelism (#100508)	2023-05-07 18:15:49 +00:00
_utils.py	[BE][DTensor] validate the mesh argument in DeviceMesh construction (#99094)	2023-04-21 23:47:51 +00:00
_view_with_dim_change.py	[dtensor] use tracing for metadata prop (#95456)	2023-02-28 17:54:22 +00:00
api.py	[TP] Add API logging for TP high level API (#102209)	2023-05-25 03:33:00 +00:00
fsdp.py	[TP] Add API logging for TP high level API (#102209)	2023-05-25 03:33:00 +00:00
multihead_attention_tp.py	Fix typos under torch/distributed directory (#95638)	2023-03-27 21:13:44 +00:00
style.py	[TP] Enable more generic attn in Tensor Parallelism (#100508)	2023-05-07 18:15:49 +00:00