pytorch
3ca36179 - TearDown function for ProcessGroupNCCLTest Initializer (#42209)

Commit
4 years ago
TearDown function for ProcessGroupNCCLTest Initializer (#42209) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/42209 This PR adds a TearDown function to the testing superclass to ensure that the NCCL_BLOCKING_WAIT environment variable is reset after each test case. ghstack-source-id: 109097247 Test Plan: Working on devGPU and devvm. Reviewed By: jiayisuse Differential Revision: D22782672 fbshipit-source-id: 8f919a96d7112f9f167e90ce3df59886c88f3514
Author
Parents
Loading