Fix fbcode weak ordering (#25026)
Summary:
Same FBCode-only weak ordering issue as previously encountered :( Internal assert fails a test
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25026
Differential Revision: D16966994
Pulled By: eellison
fbshipit-source-id: 649331ae1317df870f26a968e3f40f2b7a3a072a