ROCM: Enable few more tests for ROCM (#77669)
This re-enables the test_reference_numerics_extremal
tests for atan and atanh.
This PR limits the tests to be not run with complex128 as
its still failing.
These tests were disabled previously in this PR:
https://github.com/pytorch/pytorch/pull/75951
There is no open issue for these tests.
Signed-off-by: Arindam Roy <rarindam@gmail.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77669
Approved by: https://github.com/mruberry