pytorch

7f06c65a - [torch/elastic] Improve the implementation of the utility functions and add their unit tests. (#54804)

[torch/elastic] Improve the implementation of the utility functions and add their unit tests. (#54804) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54804 Improve the implementation of the utility functions to handle more edge cases and also have a new set of unit tests to cover their usage. Test Plan: Run the existing and newly introduced unit tests. Reviewed By: kiukchung Differential Revision: D27327898 fbshipit-source-id: 96b6fe2d910e3de69f44947a0e8a9f687ab50633