提交 eee714ef authored 作者: Frederic's avatar Frederic

Update GPU test. Now we duplicate the node later

上级 d5f310f0
......@@ -389,7 +389,7 @@ def test_alloc_empty():
assert out[1].shape == (2, 3)
assert out[1].dtype == 'float32'
assert len([node for node in f.maker.fgraph.apply_nodes
if isinstance(node.op, cuda.basic_ops.GpuAllocEmpty)]) == 2
if isinstance(node.op, cuda.basic_ops.GpuAllocEmpty)]) == 1
def test_elemwise_empty():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论