From 383256364aa2bd783bf97e5c51c439a2f266e16b Mon Sep 17 00:00:00 2001
From: wenbin <wang3323032@qq.com>
Date: Thu, 8 Sep 2022 19:00:55 +0800
Subject: [PATCH] copyright (#45866)

---
 .../plugin/layernorm_shift_partition_op.cu    |  1 +
 .../test_layernorm_shift_partition_pass.py    | 22 +++++++++++++++++--
 2 files changed, 21 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu b/paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu
index 0605b39216..19128256bc 100644
--- a/paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu
+++ b/paddle/fluid/inference/tensorrt/plugin/layernorm_shift_partition_op.cu
@@ -1,4 +1,5 @@
 // Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
+// Copyright (c) 2019-2022, NVIDIA CORPORATION.  All rights reserved.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_layernorm_shift_partition_pass.py b/python/paddle/fluid/tests/unittests/ir/inference/test_layernorm_shift_partition_pass.py
index 2d5dd9fe4b..a4d74611fe 100644
--- a/python/paddle/fluid/tests/unittests/ir/inference/test_layernorm_shift_partition_pass.py
+++ b/python/paddle/fluid/tests/unittests/ir/inference/test_layernorm_shift_partition_pass.py
@@ -61,6 +61,24 @@ class TestLayernormShiftPartitionPass(PassAutoScanTest):
         })
         yield config, ['layernorm_shift_partition'], (1e-5, 1e-5)
 
+        # trt dynamic_shape
+        config = self.create_trt_inference_config()
+        config.enable_tensorrt_engine(
+            max_batch_size=1,
+            workspace_size=102400,
+            min_subgraph_size=0,
+            precision_mode=paddle_infer.PrecisionType.Half,
+            use_static=False,
+            use_calib_mode=False)
+        config.set_trt_dynamic_shape_info({
+            "input_data": [1, 9, 96],
+        }, {
+            "input_data": [4, 3136, 768],
+        }, {
+            "input_data": [1, 784, 384],
+        })
+        yield config, ['layernorm_shift_partition'], (1e-3, 1e-3)
+
     def sample_program_config(self, draw):
         axis = [0, 1, 3, 2, 4, 5]
         epsilon = draw(st.floats(min_value=0.0000001, max_value=0.001))
@@ -198,10 +216,10 @@ class TestLayernormShiftPartitionPass(PassAutoScanTest):
 
     def test(self):
         self.run_and_statis(quant=False,
-                            max_examples=20,
+                            max_examples=50,
                             passes=["layernorm_shift_partition_fuse_pass"],
                             max_duration=250,
-                            min_success_num=20)
+                            min_success_num=50)
 
 
 if __name__ == "__main__":
-- 
GitLab