pytorch

f5440a44 - CUDA BFloat16 i0 support (#44750)

CUDA BFloat16 i0 support (#44750) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44750 Reviewed By: glaringlee Differential Revision: D23764383 Pulled By: ngimel fbshipit-source-id: d0e784d89241e8028f97766fdac51fe1ab4c188c