pytorch

2111ec3b - CUDA BFloat16 losses (#45011)

CUDA BFloat16 losses (#45011) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/45011 Reviewed By: mruberry Differential Revision: D23805840 Pulled By: ngimel fbshipit-source-id: 3eb60d4367c727100763879e20e9df9d58bf5ad6