fix unittest random error (#30808)

200ee33d · Shang Zhizhou · GitHub · db870872 · 200ee33d
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

python/paddle/fluid/tests/unittests/ir/inference/test_trt_slice_plugin.py ...uid/tests/unittests/ir/inference/test_trt_slice_plugin.py +4 -1

未找到文件。
--- 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