BFloat16 support for torch.sort (#58196)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58196
Reviewed By: anjali411
Differential Revision: D28721364
Pulled By: ngimel
fbshipit-source-id: 0785f7100fb76d69da7a73022c7d2eb43c91fa6e