From ac894ced9a8f82916c76478c8482899d11d45f5f Mon Sep 17 00:00:00 2001 From: zlsh80826 Date: Wed, 16 Feb 2022 12:50:01 +0800 Subject: [PATCH] Test only trt group norm (#39561) --- .../ir/inference/test_trt_group_norm_op.py | 15 ++------------- 1 file changed, 2 insertions(+), 13 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_group_norm_op.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_group_norm_op.py index 85bd625413c..1bcbbc38c97 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_group_norm_op.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_group_norm_op.py @@ -28,25 +28,14 @@ class TRTGroupNormTest(InferencePassTest): with fluid.program_guard(self.main_program, self.startup_program): data = fluid.data( name="data", shape=[-1, 512, 12, 12], dtype="float32") - relu_out = fluid.layers.relu(data) - relu6_out = fluid.layers.relu6(relu_out) - tanh_out = fluid.layers.tanh(relu6_out) - conv_out = fluid.layers.conv2d( - input=tanh_out, - num_filters=512, - filter_size=3, - groups=1, - padding=[1, 1], - bias_attr=False, - act=None) - out = self.append_group_norm(conv_out) + out = self.append_group_norm(data) self.feeds = { "data": np.random.random([1, 512, 12, 12]).astype("float32"), } self.enable_trt = True self.trt_parameters = TRTGroupNormTest.TensorRTParam( - 1 << 30, 32, 1, AnalysisConfig.Precision.Float32, False, False) + 1 << 30, 1, 1, AnalysisConfig.Precision.Float32, False, False) self.dynamic_shape_params = TRTGroupNormTest.DynamicShapeParam({ 'data': [1, 512, 12, 12] }, {'data': [1, 512, 12, 12]}, {'data': [1, 512, 12, 12]}, False) -- GitLab