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

fix test in debugmode

上级 078a645d
...@@ -186,8 +186,13 @@ def test_speed_lazy(): ...@@ -186,8 +186,13 @@ def test_speed_lazy():
def test_allow_gc_cvm(): def test_allow_gc_cvm():
mode = theano.config.mode
if mode in ['DEBUG_MODE', 'DebugMode']:
mode = "FAST_RUN"
v = theano.tensor.vector() v = theano.tensor.vector()
f = theano.function([v], v + 1) f = theano.function([v], v + 1, mode=mode)
f([1]) f([1])
n = list(f.maker.fgraph.apply_nodes)[0].outputs[0] n = list(f.maker.fgraph.apply_nodes)[0].outputs[0]
assert f.fn.storage_map[n][0] is None assert f.fn.storage_map[n][0] is None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论