pytorch

a64b03a5 - Move lr tensor to cuda if needed (#119073)

Move lr tensor to cuda if needed (#119073) Fixes https://github.com/pytorch/pytorch/issues/119026 Pull Request resolved: https://github.com/pytorch/pytorch/pull/119073 Approved by: https://github.com/eellison