Andrew Gu 04ad0134ae [FSDP] Use reduce_scatter_tensor() (#87240) ... Let us silence some more warnings 👍🏼 Pull Request resolved: https://github.com/pytorch/pytorch/pull/87240 Approved by: https://github.com/rohan-varma		2022-10-24 11:29:23 +00:00
..
_checkpoint	[RFC] Separate CPU offload activation to its own wrapper (#85459)	2022-10-15 05:19:28 +00:00
_comm_hooks	[FSDP] Use reduce_scatter_tensor() (#87240)	2022-10-24 11:29:23 +00:00
_optimizer_overlap	make fsdp folder to be public (#72084)	2022-02-02 15:50:14 +00:00
_quantization	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
ddp_comm_hooks	Add __all__ for a few distributed modules plus a little typing (reland) (#84872)	2022-09-13 21:57:49 +00:00
model_averaging	Update hierarchical_model_averager.py (#85648)	2022-10-03 06:15:20 +00:00
__init__.py	Make _Join, _Joinable, _JoinHook public (#62605)	2021-08-03 12:20:11 -07:00
join.py	Integrate xdoctest - Rebased (#82797)	2022-08-12 02:08:01 +00:00