[ROCm] Fix test_nadam

- Cherry-pick of 5fc3423

Author: pruthvistony

test/test_optim.py

@@ -37,6 +37,7 @@
 from torch.testing._internal.common_utils import (
     TestCase,
     run_tests,
+    TEST_WITH_ROCM,
     TEST_WITH_UBSAN,
     load_tests,
     parametrize,
@@ -1164,6 +1165,8 @@ def test_adadelta_complex(self):
             )
 
     def test_nadam(self):
+        if TEST_WITH_ROCM:
+            self.rel_tol = 1e-5
         self._test_basic_cases(
             lambda weight, bias, foreach: optim.NAdam(
                 [weight, bias], lr=1e-3, foreach=foreach