未验证 提交 f6b6b057 编写于 作者: A Aganlengzi 提交者: GitHub

[NPU] fix npu cast ut (#40982)

* [NPU] fix npu cast ut

* [NPU] fix npu cast ut
上级 25591674
...@@ -18,7 +18,7 @@ import numpy as np ...@@ -18,7 +18,7 @@ import numpy as np
import unittest import unittest
import sys import sys
sys.path.append("..") sys.path.append("..")
from op_test import OpTest from op_test import OpTest, skip_check_grad_ci
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.core as core import paddle.fluid.core as core
...@@ -27,6 +27,7 @@ paddle.enable_static() ...@@ -27,6 +27,7 @@ paddle.enable_static()
SEED = 2021 SEED = 2021
@skip_check_grad_ci(reason="[skip NPU cast grad check] not implemented yet.")
class TestCast1(OpTest): class TestCast1(OpTest):
def setUp(self): def setUp(self):
self.set_npu() self.set_npu()
...@@ -49,6 +50,7 @@ class TestCast1(OpTest): ...@@ -49,6 +50,7 @@ class TestCast1(OpTest):
self.check_output_with_place(self.place) self.check_output_with_place(self.place)
@skip_check_grad_ci(reason="[skip NPU cast grad check] not implemented yet.")
class TestCast2(OpTest): class TestCast2(OpTest):
def setUp(self): def setUp(self):
self.set_npu() self.set_npu()
...@@ -71,6 +73,7 @@ class TestCast2(OpTest): ...@@ -71,6 +73,7 @@ class TestCast2(OpTest):
self.check_output_with_place(self.place, atol=1e-3) 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): class TestCast3(OpTest):
def setUp(self): def setUp(self):
self.set_npu() self.set_npu()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册