fix ut error.

PaddlePaddle · Aug 23, 2021 · 4a13c9b · 4a13c9b · paddle-bot-old · Aug 23, 2021
1 parent 69e7ff8
commit 4a13c9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/auto_scan_test.py b/python/paddle/fluid/tests/unittests/ir/inference/auto_scan_test.py
@@ -111,8 +111,8 @@ def run_test(self,
                         tensor_config.dtype)
                 results: List[Dict[str, Tensor]] = []
                 for pred_config in self.sample_predictor_configs():
-                    self.assert_op_size(trt_engine_num, paddle_op_num)
                     results.append(
                         self.run_test_config(model, params, prog_config,
                                              pred_config, feed_data))
                 self.assert_tensors_near(threshold=threshold, tensors=results)
+                self.assert_op_size(trt_engine_num, paddle_op_num)