Fix bc test (#36588)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36588
-
ghstack-source-id: 102122179
(Note: this ignores all push blocking failures!)
Test Plan: -
Differential Revision: D21021653
fbshipit-source-id: b3693a1a5e27d28dc2fc772cbef5787ab4ceafaa