pytorch

15abc56b - Graph break on step closure in optimizer (#121777)

Graph break on step closure in optimizer (#121777) Fixes https://github.com/pytorch/pytorch/issues/116494 Pull Request resolved: https://github.com/pytorch/pytorch/pull/121777 Approved by: https://github.com/yanboliang