提交 2bba58d0 authored 作者: Frederic Bastien's avatar Frederic Bastien

back port to python 2.4

上级 7ac84c38
......@@ -27,6 +27,6 @@ def test_no_shared_var_graph():
f = theano.function([a,b],[a+b], mode=mode_with_gpu)
l = f.maker.env.toposort()
assert len(l)==4
assert any(isinstance(x.op,cuda.GpuElemwise) for x in l)
assert any(isinstance(x.op,cuda.GpuFromHost) for x in l)
assert any(isinstance(x.op,cuda.HostFromGpu) for x in l)
assert numpy.any(isinstance(x.op,cuda.GpuElemwise) for x in l)
assert numpy.any(isinstance(x.op,cuda.GpuFromHost) for x in l)
assert numpy.any(isinstance(x.op,cuda.HostFromGpu) for x in l)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论