Enable caffe2 tests for RocM jobs (#41604)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41604
Reviewed By: ezyang
Differential Revision: D22603703
Pulled By: malfet
fbshipit-source-id: 789ccf2bb79668a5a68006bb877b2d88fb569809