fix unittest random error (#30808)

4 years ago · 200ee33df8
parent db87087283
commit 200ee33df8
1 changed files with 4 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_slice_plugin.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_slice_plugin.py
@ -56,7 +56,10 @@ class SlicePluginTRTTest(InferencePassTest):
        if core.is_compiled_with_cuda():
            use_gpu.append(True)
        for i in range(len(use_gpu)):
-            self.check_output_with_option(use_gpu[i])
+            atol = 1e-5
+            if self.trt_parameters.precision == AnalysisConfig.Precision.Half:
+                atol = 1e-3
+            self.check_output_with_option(use_gpu[i], atol)


 #negative starts && ends