From b34b54db7d042c8c3d2643731b203e5dc18d418d Mon Sep 17 00:00:00 2001 From: zhangxiaoci Date: Tue, 28 Jun 2022 13:43:46 +0800 Subject: [PATCH] fix squeeze2/unsqueeze2 unittest, *test=kunlun (#43859) --- .../paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py | 1 + .../paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py index cdc67cf4647..6996554d427 100644 --- a/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py +++ b/python/paddle/fluid/tests/unittests/xpu/test_squeeze2_op_xpu.py @@ -38,6 +38,7 @@ class XPUTestSqueeze2Op(XPUOpTestWrapper): def setUp(self): self.op_type = "squeeze2" + self.__class__.op_type = "squeeze2" self.use_mkldnn = False self.init_dtype() self.init_test_case() diff --git a/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py index 6daa4739412..e9fc66ca4fc 100644 --- a/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py +++ b/python/paddle/fluid/tests/unittests/xpu/test_unsqueeze2_op_xpu.py @@ -39,6 +39,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper): def setUp(self): self.op_type = "unsqueeze2" + self.__class__.op_type = "unsqueeze2" self.use_mkldnn = False self.init_dtype() self.init_test_case() @@ -117,6 +118,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper): def setUp(self): self.op_type = "unsqueeze2" + self.__class__.op_type = "unsqueeze2" self.use_mkldnn = False self.init_dtype() self.init_test_case() @@ -191,6 +193,7 @@ class XPUTestUnsqueeze2Op(XPUOpTestWrapper): def setUp(self): self.op_type = "unsqueeze2" + self.__class__.op_type = "unsqueeze2" self.use_mkldnn = False self.init_test_case() self.init_dtype() -- GitLab