[ROCm] Fix for test_nadam

pruthvistony · pruthvistony · commit 3df9ca3ed7dc · 2023-09-10T15:10:23.000-07:00
- Cherry pick of dbd701b
diff --git a/test/optim/test_optim.py b/test/optim/test_optim.py
@@ -1306,6 +1306,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(
                 self._build_params_dict(weight, bias, lr=1e-2), lr=1e-3, foreach=foreach

Original file line number	Diff line number	Diff line change
`@@ -1306,6 +1306,8 @@ def test_adadelta_complex(self):`
`1306`	`1306`	`)`
`1307`	`1307`
`1308`	`1308`	`def test_nadam(self):`
	`1309`	`+ if TEST_WITH_ROCM:`
	`1310`	`+ self.rel_tol = 1e-5`
`1309`	`1311`	`self._test_basic_cases(`
`1310`	`1312`	`lambda weight, bias, foreach: optim.NAdam(`
`1311`	`1313`	`self._build_params_dict(weight, bias, lr=1e-2), lr=1e-3, foreach=foreach`