pytorch

301257c7 - rebase for autocast updates to include device_type and dtype flags (#61002)

rebase for autocast updates to include device_type and dtype flags (#61002) Summary: Fixes #{55374} https://github.com/pytorch/pytorch/issues/55374 Pull Request resolved: https://github.com/pytorch/pytorch/pull/61002 Reviewed By: malfet, mruberry Differential Revision: D30016812 Pulled By: ngimel fbshipit-source-id: 6e09a29f539d28e9aea5cd9489b1e633cc588033