[ROCm] torch.cuda.is_bf16_supported() returns True (#80410)
`torch.cuda.is_bf16_supported()` return False on ROCm which is not correct, since BF16 is supported on all AMD GPU arch - gfx906, gfx908 and gfx90a.
cc @jithunnair-amd
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80410
Approved by: https://github.com/jeffdaily, https://github.com/malfet