pytorch

bbe1b9bb - Fix https://github.com/pytorch/pytorch/issues/96278 (#96299)

Fix https://github.com/pytorch/pytorch/issues/96278 (#96299) Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/96299 Approved by: https://github.com/ngimel