Sisil Mehta 9d3ffed327 [FSDP] Sharded Grad Scaler (#76918) ... Summary: Adding in a shard aware grad scaler for FSDP+MixedPrecision support Test Plan: Tests added Differential Revision: D35988676 Pull Request resolved: https://github.com/pytorch/pytorch/pull/76918 Approved by: https://github.com/rohan-varma		2022-05-16 15:53:21 +00:00
..
_internal	[FSDP] Sharded Grad Scaler (#76918)	2022-05-16 15:53:21 +00:00
__init__.py	cleanup torch.testing namespace (#72708)	2022-02-25 06:30:31 +00:00
_comparison.py	remove special casing for sparse CSR shape comparison	2022-04-11 16:04:34 +00:00
_creation.py	[complex32] make_tensor	2022-03-30 01:05:34 +00:00
_deprecated.py	promote torch.testing to stable (#73348)	2022-02-25 06:30:31 +00:00
_legacy.py	[complex32] real and imag (also remove unused real and imag kernels)	2022-05-05 04:36:58 +00:00