From f6b6b05721dcbe93d0548578936cfabfed199cc0 Mon Sep 17 00:00:00 2001 From: Aganlengzi Date: Sun, 27 Mar 2022 09:28:01 +0800 Subject: [PATCH] [NPU] fix npu cast ut (#40982) * [NPU] fix npu cast ut * [NPU] fix npu cast ut --- python/paddle/fluid/tests/unittests/npu/test_cast_op_npu.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/npu/test_cast_op_npu.py b/python/paddle/fluid/tests/unittests/npu/test_cast_op_npu.py index f522eb10d92..0d79d9b0723 100755 --- a/python/paddle/fluid/tests/unittests/npu/test_cast_op_npu.py +++ b/python/paddle/fluid/tests/unittests/npu/test_cast_op_npu.py @@ -18,7 +18,7 @@ import numpy as np import unittest import sys sys.path.append("..") -from op_test import OpTest +from op_test import OpTest, skip_check_grad_ci import paddle import paddle.fluid as fluid import paddle.fluid.core as core @@ -27,6 +27,7 @@ paddle.enable_static() SEED = 2021 +@skip_check_grad_ci(reason="[skip NPU cast grad check] not implemented yet.") class TestCast1(OpTest): def setUp(self): self.set_npu() @@ -49,6 +50,7 @@ class TestCast1(OpTest): self.check_output_with_place(self.place) +@skip_check_grad_ci(reason="[skip NPU cast grad check] not implemented yet.") class TestCast2(OpTest): def setUp(self): self.set_npu() @@ -71,6 +73,7 @@ class TestCast2(OpTest): self.check_output_with_place(self.place, atol=1e-3) +@skip_check_grad_ci(reason="[skip NPU cast grad check] not implemented yet.") class TestCast3(OpTest): def setUp(self): self.set_npu() -- GitLab