diff --git 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 index d9817c6fe1825d19bbbf8ab58df180dd382c81cf..6ea2335c7a1b1c456a2adf2d60ddcffeb59d9367 100644 --- 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