未验证 提交 4b17659c 编写于 作者: W wangguanzhong 提交者: GitHub

fix unittest in ext_op (#689)

上级 bb42d6de
...@@ -83,11 +83,7 @@ class TestRightPoolOp(unittest.TestCase): ...@@ -83,11 +83,7 @@ class TestRightPoolOp(unittest.TestCase):
place = fluid.CUDAPlace(0) place = fluid.CUDAPlace(0)
with fluid.program_guard(tp, sp): with fluid.program_guard(tp, sp):
x = fluid.data( x = fluid.data(name=self.name, shape=x_shape, dtype=x_type)
name=self.name,
shape=x_shape,
dtype=x_type,
append_batch_size=False)
y = self.func_map[self.name][0](x) y = self.func_map[self.name][0](x)
np.random.seed(0) np.random.seed(0)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册