From cfe55f312dc5770cd14d847f32824037c189c0a3 Mon Sep 17 00:00:00 2001
From: Zhang Jun <ewalker@live.cn>
Date: Tue, 10 Jan 2023 19:03:08 +0800
Subject: [PATCH] update reduce ut (#49689)

---
 .../tests/unittests/ir/inference/test_trt_convert_reduce.py    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py
index 67580b6d7c..1c04e40fe6 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce.py
@@ -85,6 +85,9 @@ class TrtConvertReduceTest(TrtLayerAutoScanTest):
                             ]
                             dics2 = [
                                 {
+                                    "keep_dim": keep_dim,
+                                    "dim": dim,
+                                    "reduce_all": reduce_all,
                                     "out_dtype": out_dtype,
                                     "in_dtype": out_dtype,
                                 },
-- 
GitLab