未验证 提交 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
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()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册