pytorch
2f8c0cb2 - [FSDP][Easy] Use `run_subtests` for hybrid shard test (#90859)

Commit
2 years ago
[FSDP][Easy] Use `run_subtests` for hybrid shard test (#90859) This PR uses `self.run_subtests` which exactly contains the `self.subTest` and `dist.barrier()` boilerplate. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90859 Approved by: https://github.com/rohan-varma
Author
Committer
Parents
Loading