Fix LBFGS on GPU (#25909)
Summary:
Changelog:
- Fixes mismatch of device in LBFGS, and possibly that of data type as well.
Fixes https://github.com/pytorch/pytorch/issues/25854
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25909
Differential Revision: D17285583
Pulled By: soumith
fbshipit-source-id: 68df9326c1c40803494ee0693a0eddcd98c30ce7