From f50768e89a005378fcfb21eb9570fdfc4186aed9 Mon Sep 17 00:00:00 2001
From: zlsh80826 <zlsh80826@gmail.com>
Date: Thu, 23 Dec 2021 10:16:50 +0800
Subject: [PATCH] Revise input_data1 shape to same as input_data2 shape for
 non-broadcast cases (#38206)

---
 .../ir/inference/test_trt_convert_elementwise.py   | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_elementwise.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_elementwise.py
index b54b923d3b..1c5b640fe4 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_elementwise.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_elementwise.py
@@ -215,11 +215,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
                     "input_data2": [1]
                 }
                 self.dynamic_shape.max_input_shape = {
-                    "input_data1": [256],
+                    "input_data1": [128],
                     "input_data2": [128]
                 }
                 self.dynamic_shape.opt_input_shape = {
-                    "input_data1": [16],
+                    "input_data1": [32],
                     "input_data2": [32]
                 }
             elif self.dims == 2:
@@ -232,7 +232,7 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
                     "input_data2": [128, 256]
                 }
                 self.dynamic_shape.opt_input_shape = {
-                    "input_data1": [2, 16],
+                    "input_data1": [32, 64],
                     "input_data2": [32, 64]
                 }
             elif self.dims == 3:
@@ -241,11 +241,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
                     "input_data2": [1, 4, 4]
                 }
                 self.dynamic_shape.max_input_shape = {
-                    "input_data1": [128, 256, 128],
+                    "input_data1": [128, 128, 256],
                     "input_data2": [128, 128, 256]
                 }
                 self.dynamic_shape.opt_input_shape = {
-                    "input_data1": [2, 32, 16],
+                    "input_data1": [2, 64, 64],
                     "input_data2": [2, 64, 64]
                 }
             elif self.dims == 4:
@@ -254,11 +254,11 @@ class TrtConvertElementwiseTest_two_input_without_broadcast(
                     "input_data2": [1, 4, 4, 4]
                 }
                 self.dynamic_shape.max_input_shape = {
-                    "input_data1": [8, 32, 64, 64],
+                    "input_data1": [8, 128, 64, 128],
                     "input_data2": [8, 128, 64, 128]
                 }
                 self.dynamic_shape.opt_input_shape = {
-                    "input_data1": [2, 32, 32, 16],
+                    "input_data1": [2, 64, 32, 32],
                     "input_data2": [2, 64, 32, 32]
                 }
 
-- 
GitLab