Adjust the relative error of QR's grad (#42221)

PaddlePaddle · Apr 27, 2022 · 4c80385 · 4c80385
1 parent acca035
commit 4c80385
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_qr_op.py b/python/paddle/fluid/tests/unittests/test_qr_op.py
@@ -27,7 +27,7 @@
 class TestQrOp(OpTest):
     def setUp(self):
         paddle.enable_static()
-        np.random.seed(4)
+        np.random.seed(7)
         self.op_type = "qr"
         a, q, r = self.get_input_and_output()
         self.inputs = {"X": a}
@@ -74,7 +74,8 @@ def test_check_output(self):
         self.check_output()
 
     def test_check_grad_normal(self):
-        self.check_grad(['X'], ['Q', 'R'])
+        self.check_grad(
+            ['X'], ['Q', 'R'], numeric_grad_delta=1e-5, max_relative_error=1e-6)
 
 
 class TestQrOpCase1(TestQrOp):
@@ -116,6 +117,7 @@ def get_shape(self):
 class TestQrAPI(unittest.TestCase):
     def test_dygraph(self):
         paddle.disable_static()
+        np.random.seed(7)
 
         def run_qr_dygraph(shape, mode, dtype):
             if dtype == "float32":
@@ -180,6 +182,7 @@ def run_qr_dygraph(shape, mode, dtype):
 
     def test_static(self):
         paddle.enable_static()
+        np.random.seed(7)
 
         def run_qr_static(shape, mode, dtype):
             if dtype == "float32":

diff --git a/python/paddle/fluid/tests/unittests/white_list/op_threshold_white_list.py b/python/paddle/fluid/tests/unittests/white_list/op_threshold_white_list.py
@@ -51,6 +51,7 @@
     'matrix_power', \
     'cholesky_solve', \
     'solve', \
+    'qr', \
 ]
 
 NEED_FIX_FP64_CHECK_OUTPUT_THRESHOLD_OP_LIST = ['bilinear_interp',\