fix WAR race (#20182)
Summary:
was flagged by racecheck.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20182
Differential Revision: D15393536
Pulled By: ezyang
fbshipit-source-id: ad4849c9fb2c8feb966be1c4ca0dadd7360f58fe
Author
Natalia Gimelshein