提交 5f152002 authored 作者: Melanie Ducoffe's avatar Melanie Ducoffe

add one assert in test_gpu_alloc_empty

上级 15f89be2
...@@ -160,7 +160,7 @@ def test_gpuallocempty(): ...@@ -160,7 +160,7 @@ def test_gpuallocempty():
mode=mode_with_gpu) mode=mode_with_gpu)
l_gpu = f_gpu.maker.fgraph.toposort() l_gpu = f_gpu.maker.fgraph.toposort()
#assert numpy.any([isinstance(x.op, basic_ops.GpuAllocEmpty) for x in l_gpu]) assert numpy.any([isinstance(x.op, basic_ops.GpuAllocEmpty) for x in l_gpu])
f_cpu = theano.function([], tensor.AllocEmpty('int32')(2,3), f_cpu = theano.function([], tensor.AllocEmpty('int32')(2,3),
mode=mode_with_gpu) mode=mode_with_gpu)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论