Enable caffe2 softmax tests with ROCm 2.4 (#20280)
Summary:
cc xw285cornell petrex
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20280
Reviewed By: xw285cornell
Differential Revision: D15262695
Pulled By: bddppq
fbshipit-source-id: d72490ff599cdab0331230bc9b12075085386319
