pytorch

f2b34807 - CUDA BFloat softmax (#44837)

CUDA BFloat softmax (#44837) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44837 Reviewed By: glaringlee Differential Revision: D23767981 Pulled By: ngimel fbshipit-source-id: be92c25a1b66ed50a52e090db167079def6f6b39