fix: CI test

fengxiaoshuai · Apr 10, 2022 · 571ee3b · 571ee3b
1 parent 61837fd
commit 571ee3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_slice.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_slice.py
@@ -127,7 +127,7 @@ def generate_trt_nodes_num(attrs, dynamic_shape):
             program_config.ops[i].attrs
             for i in range(len(program_config.ops))
         ]
-
+        self.trt_param.max_batch_size = 9
         # for static_shape
         clear_dynamic_shape()
         self.trt_param.precision = paddle_infer.PrecisionType.Float32