Skip to content

P
Paddle

PaddlePaddle / Paddle
11 个月 前同步成功

通知 2289
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

提交 6ef4f1f8 编写于 作者: K Kexin Zhao
浏览文件
操作

small fix

上级 f3c5e815
隐藏空白更改
内联 并排
Showing with 4 addition and 4 deletion
python/paddle/fluid/tests/unittests/test_cast_op.py
浏览文件 @ 6ef4f1f8
...@@ -40,7 +40,7 @@ class TestCastOp2(op_test.OpTest): ...@@ -40,7 +40,7 @@ class TestCastOp2(op_test.OpTest):
def setUp(self): def setUp(self):
ipt = np.random.random(size=[10, 10]) ipt = np.random.random(size=[10, 10])
# numpy float16 is binded to fluid float16 via uint16 # numpy float16 is binded to fluid float16 via uint16
self.inputs = {'X': ipt.astype('float16').view(uint16)} self.inputs = {'X': ipt.astype('float16').view(np.uint16)}
self.outputs = {'Out': ipt.astype('float32')} self.outputs = {'Out': ipt.astype('float32')}
self.attrs = { self.attrs = {
'in_dtype': int(core.VarDesc.VarType.FP16), 'in_dtype': int(core.VarDesc.VarType.FP16),
...@@ -49,10 +49,10 @@ class TestCastOp2(op_test.OpTest): ...@@ -49,10 +49,10 @@ class TestCastOp2(op_test.OpTest):
self.op_type = 'cast' self.op_type = 'cast'
def test_check_output(self): def test_check_output(self):
self.check_output() self.check_output(atol=1e-3)
class TestCastOp2(op_test.OpTest): class TestCastOp3(op_test.OpTest):
def setUp(self): def setUp(self):
ipt = np.random.random(size=[10, 10]) ipt = np.random.random(size=[10, 10])
self.inputs = {'X': ipt.astype('float32')} self.inputs = {'X': ipt.astype('float32')}
...@@ -64,7 +64,7 @@ class TestCastOp2(op_test.OpTest): ...@@ -64,7 +64,7 @@ class TestCastOp2(op_test.OpTest):
self.op_type = 'cast' self.op_type = 'cast'
def test_check_output(self): def test_check_output(self):
self.check_output() self.check_output(atol=1e-3)
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或