CUDA BFloat16 infrastructure (#44925)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44925
Reviewed By: agolynski
Differential Revision: D23783910
Pulled By: ngimel
fbshipit-source-id: dacac2ad87d58056bdc68bfe0b7ab1de5c2af0d8
