[ROCm] add exact_dtype=False to bfloat16 test (#38381)
Summary:
CC rohithkrn ezyang xw285cornell
Fixes
- TestNNDeviceTypeCUDA.test_activations_bfloat16_cuda
- TestNNDeviceTypeCUDA.test_pooling_bfloat16_cuda
- TestNNDeviceTypeCUDA.test_softmax_bfloat16_cuda
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38381
Differential Revision: D21549636
Pulled By: ezyang
fbshipit-source-id: acb290c57eff4077b040a696267ecde613f0a433