Enable arg_ops_test/unique_ops_test on AMD/rocm (#16853)
Summary:
Verified both tests are passing on rocm 2.1 env.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16853
Differential Revision: D13996279
Pulled By: bddppq
fbshipit-source-id: c0df610d7d9ca8d80ed2d1339cdadef59105a71c