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 f522eb10d92e6b7e377f072cc153f8b275a053e5..0d79d9b07233f1ef43e1c66009aca7863342b4cc 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()