Open yolo loss uts (#59154)

PaddlePaddle · Nov 21, 2023 · 5547702 · 5547702
1 parent ccb30dd
commit 5547702
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/legacy_test/test_yolov3_loss_op.py b/test/legacy_test/test_yolov3_loss_op.py
@@ -273,11 +273,13 @@ def setUp(self):
 
     def test_check_output(self):
         place = core.CPUPlace()
-        self.check_output_with_place(place, atol=2e-3)
+        self.check_output_with_place(place, atol=2e-3, check_pir=True)
 
     def test_check_grad_ignore_gtbox(self):
         place = core.CPUPlace()
-        self.check_grad_with_place(place, ['X'], 'Loss', max_relative_error=0.2)
+        self.check_grad_with_place(
+            place, ['X'], 'Loss', max_relative_error=0.2, check_pir=True
+        )
 
     def initTestCase(self):
         self.anchors = [