Enable more caffe2 fp16 rocm tests (#15040)
Summary:
cc rohithkrn petrex
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15040
Reviewed By: houseroad
Differential Revision: D13413068
Pulled By: bddppq
fbshipit-source-id: b2967f16f8da0b9e80083138fb8632c14e9e9b63
