pytorch
pytorch copied to clipboard
Unskip test_softmax_cuda_float32 and test_softmax_results_cuda_float16, test_softmax_results_cuda_float32 on ROCM 5.0
🐛 Describe the bug
The following tests are passing in ROCM5.0:
test_nn | test_softmax_cuda_float32 | (main.TestNNDeviceTypeCUDA) test_nn | test_softmax_results_cuda_float16 | (main.TestNNDeviceTypeCUDA) test_nn | test_softmax_results_cuda_float32 | (main.TestNNDeviceTypeCUDA)
We need to unskip them. It was disabled by the following PR:
https://github.com/pytorch/pytorch/pull/46793
Versions
ROCM5.0
Enabled by https://github.com/pytorch/pytorch/pull/74786