ROCm: Fix test_nadam (#1006)

Change the rtol level

Signed-off-by: Jagadish Krishnamoorthy <[email protected]>

Author: jaglinux

test/test_optim.py

@@ -39,6 +39,7 @@
     TestCase,
     run_tests,
     TEST_WITH_UBSAN,
+    TEST_WITH_ROCM,
     load_tests,
     parametrize,
     instantiate_parametrized_tests,
@@ -1174,6 +1175,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