pytorch

0e543b2b - Provide a decorator to set/unset nccl blocking wait for tests (#54740)

Provide a decorator to set/unset nccl blocking wait for tests (#54740) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54740 Adds a simple helper decorator to set/unset nccl blocking wait for tests. This will make it easier than having to manually set/unset the os.environ vars every time. ghstack-source-id: 125233693 Test Plan: CI Reviewed By: SciPioneer Differential Revision: D27277222 fbshipit-source-id: c289b9d05e2f6328d672810b07501979b6e177c6