pytorch

184f6b57 - Fix perf bug in #90528 (#90630)

Fix perf bug in #90528 (#90630) Fixes a minor I noticed in #90528 also a follow up to #89000. @ezyang Pull Request resolved: https://github.com/pytorch/pytorch/pull/90630 Approved by: https://github.com/ezyang