[PIR] fix 3 case in cov (#67573)

wanghuancoder · web-flow · commit 0fd362f979d9 · 2024-08-20T18:51:18.000+08:00
* fix 3 case in cov

* refine
diff --git a/test/legacy_test/test_imperative_ptb_rnn.py b/test/legacy_test/test_imperative_ptb_rnn.py
@@ -458,7 +458,9 @@ def ptb_rnn_cpu_float32(self, is_sparse):
         for key, value in static_param_init.items():
             np.testing.assert_array_equal(value, dy_param_init[key])
         for key, value in static_param_updated.items():
-            np.testing.assert_array_equal(value, dy_param_updated[key])
+            np.testing.assert_allclose(
+                value, dy_param_updated[key], atol=1e-10, rtol=1e-6
+            )
 
 
 if __name__ == '__main__':
diff --git a/test/legacy_test/test_imperative_ptb_rnn_sorted_gradient.py b/test/legacy_test/test_imperative_ptb_rnn_sorted_gradient.py
@@ -241,7 +241,9 @@ def ptb_rnn_sort_gradient_cpu_float32(self, is_sparse):
         for key, value in static_param_init.items():
             np.testing.assert_array_equal(value, dy_param_init[key])
         for key, value in static_param_updated.items():
-            np.testing.assert_array_equal(value, dy_param_updated[key])
+            np.testing.assert_allclose(
+                value, dy_param_updated[key], atol=1e-10, rtol=1e-6
+            )
 
 
 if __name__ == '__main__':
diff --git a/test/legacy_test/test_nan_inf.py b/test/legacy_test/test_nan_inf.py
@@ -180,7 +180,7 @@ def test_check_stack(self):
         self.check_stack(" check_nan_inf_backward_stack.py")
 
     def test_static_check_stack(self):
-        if not paddle.framework.in_pir_mode() and not os.environ.get(
+        if not paddle.framework.use_pir_api() and not os.environ.get(
             "FLAGS_enable_pir_api"
         ):
             self.check_stack(" check_nan_inf_backward_static_stack.py")
diff --git a/test/rnn/test_rnn_api.py b/test/rnn/test_rnn_api.py
@@ -23,6 +23,8 @@
 
 from paddle import base
 
+paddle.enable_static()
+
 bidirectional_list = ["bidirectional", "bidirect"]
 
 
