[reland] Enable TF32 support for cuBLAS (#41498)
Summary:
fix rocm
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41498
Reviewed By: mruberry
Differential Revision: D22560572
Pulled By: ngimel
fbshipit-source-id: 5ee79e96cb29e70d9180830d058efb53d1c6c041