fix elementwise_mod UT error *test=kunlun

PaddlePaddle · Apr 26, 2022 · ba70104 · ba70104
1 parent 9691f86
commit ba70104
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/paddle/fluid/platform/device/xpu/xpu1_op_list.h b/paddle/fluid/platform/device/xpu/xpu1_op_list.h
@@ -102,8 +102,6 @@ XPUOpMap& get_kl1_ops() {
        XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
       {"elementwise_min_grad",
        XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
-      {"elementwise_mod",
-       XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
       {"elementwise_min",
        XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})},
       {"elementwise_mul_grad",

diff --git a/python/paddle/fluid/tests/unittests/xpu/test_elementwise_mod_op_xpu.py b/python/paddle/fluid/tests/unittests/xpu/test_elementwise_mod_op_xpu.py
@@ -89,7 +89,7 @@ def init_input_output(self):
 
     class TestRemainderOp(unittest.TestCase):
         def test_dygraph(self):
-            with fluid.dygraph.guard(fluid.XPUPlace(0)):
+            with fluid.dygraph.guard():
                 np_x = np.random.rand(22, 128, 3).astype('int64')
                 np_y = np.random.rand(22, 128, 3).astype('int64')
                 x = paddle.to_tensor(np_x)