fduwjj 487ebcac3b Clean up unsed MHA code to avoid confusion (#105956) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/105956 Approved by: https://github.com/wz337, https://github.com/ezyang, https://github.com/wanchaol		2023-07-27 17:10:17 +00:00
..
__init__.py	Clean up unsed MHA code to avoid confusion (#105956)	2023-07-27 17:10:17 +00:00
_utils.py	Get rid of dim_groups attribute from DeviceMesh (#103105)	2023-06-09 04:11:15 +00:00
_view_with_dim_change.py	[dtensor] use tracing for metadata prop (#95456)	2023-02-28 17:54:22 +00:00
api.py	Clean up unsed MHA code to avoid confusion (#105956)	2023-07-27 17:10:17 +00:00
fsdp.py	Get rid of dim_groups attribute from DeviceMesh (#103105)	2023-06-09 04:11:15 +00:00
input_reshard.py	[Reland] Update mypy to 1.4.1 (#105227)	2023-07-15 20:30:20 +00:00
style.py	[TP] Enable more generic attn in Tensor Parallelism (#100508)	2023-05-07 18:15:49 +00:00