From 55418d3f4929f0b8c40c22ff6c326fa516575385 Mon Sep 17 00:00:00 2001 From: TTerror Date: Tue, 25 Jan 2022 13:45:23 +0800 Subject: [PATCH] fix test_refactor_op_xpu, *test=kunlun (#39168) --- .../fluid/tests/unittests/xpu/test_refactor_op_xpu.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/xpu/test_refactor_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_refactor_op_xpu.py index cb54d12488d..a1eb0af2bc9 100644 --- a/python/paddle/fluid/tests/unittests/xpu/test_refactor_op_xpu.py +++ b/python/paddle/fluid/tests/unittests/xpu/test_refactor_op_xpu.py @@ -52,8 +52,9 @@ class XPUTestArgsortOp1(XPUOpTestWrapper): classes = [] for descending in [True, False]: for axis in [0, 1, 2, -1, -2]: - class_name = 'XPUTestArgsortOp_axis_' + str(axis) - attr_dict = {'init_axis': axis, 'descending': descending} + class_name = 'XPUTestArgsortOp_axis_' + str(axis) + '_' + str( + descending) + attr_dict = {'init_axis': axis, 'init_descending': descending} classes.append([class_name, attr_dict]) return base_class, classes @@ -64,8 +65,9 @@ class XPUTestArgsortOp1(XPUOpTestWrapper): self.place = paddle.XPUPlace(0) self.dtype = self.in_type self.input_shape = (2, 2, 2, 3, 3) - self.axis = -1 - self.descending = False + self.axis = -1 if not hasattr(self, 'init_axis') else self.init_axis + self.descending = False if not hasattr( + self, 'init_descending') else self.init_descending if self.in_type == 'float32': self.x = np.random.random(self.input_shape).astype(self.dtype) -- GitLab