Fix CI (#26593)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26593
This broke due to a merge conflict between my diffs and ezyang's multi dispatch diff being reverted and then relanded.
ghstack-source-id: 90549856
Test Plan: unit tests
Differential Revision: D17515837
fbshipit-source-id: c0bfd5f159ee4de80035079a1a2f39d5beafec41