From 1ba73815303ac8a5d38697098efd364bddb15686 Mon Sep 17 00:00:00 2001 From: Junjie Zhang <1356732652@qq.com> Date: Tue, 21 Jan 2025 20:18:20 +0800 Subject: [PATCH] Update test_converter_manipulation.py --- test/tensorrt/test_converter_manipulation.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/test/tensorrt/test_converter_manipulation.py b/test/tensorrt/test_converter_manipulation.py index 1b27b83e95da2..f6a3836480fa5 100644 --- a/test/tensorrt/test_converter_manipulation.py +++ b/test/tensorrt/test_converter_manipulation.py @@ -641,13 +641,7 @@ def test_trt_result(self): self.check_trt_result() -def wrapper_pad_error1(x): - return paddle.nn.functional.pad( - x, pad=[0, 0, 0, 0, 0, 0, 1, 1, 0, 0], mode='constant', value=0 - ) - - -def wrapper_pad_error2(x, padding, mode, pad_value): +def wrapper_pad_error(x, padding, mode, pad_value): return paddle.nn.functional.pad( x=paddle.to_tensor(np.random.randn(1, 1, 1, 2, 3).astype("float32")), pad=[0, 0, 0, 0, 0, 0, 1, 1, 0, 0], @@ -697,7 +691,7 @@ def test_trt_result(self): class TestPadError2TRTPattern(TensorRTBaseTest): def setUp(self): - self.python_api = wrapper_pad_error2 + self.python_api = wrapper_pad_error self.api_args = { "x": np.random.randn(1, 1, 1, 2, 3).astype("float32"), "paddings": [1, 1, 1, 0, 0, 0, 1, 1, 0, 0], @@ -715,7 +709,7 @@ def test_trt_result(self): class TestPadError3TRTPattern(TensorRTBaseTest): def setUp(self): - self.python_api = wrapper_pad_error2 + self.python_api = wrapper_pad_error self.api_args = { "x": np.random.randn(1, 1).astype("float32"), "paddings": [0, 0, 0, 0, 0, 0, 1, 1, 0, 0],