Andrew Gu 954bae8e53 [FSDP][Easy] Rename streams; add back stream sharing test (#104966) ... Purely out of preference, this PR renames the streams to `_unshard_stream` instead of `_streams_unshard` etc. since the former reads more naturally. The PR also removes some duplicated comments and adds back a unit test that streams are shared. Pull Request resolved: https://github.com/pytorch/pytorch/pull/104966 Approved by: https://github.com/rohan-varma		2023-07-13 00:24:41 +00:00
..
fully_shard	[FSDP][Easy] Rename streams; add back stream sharing test (#104966)	2023-07-13 00:24:41 +00:00
test_checkpoint.py	Fix composable checkpoint(use_reentrant=True) with multi args (#103590)	2023-06-14 21:53:30 +00:00
test_compose.py
test_contract.py
test_replicate.py