提交 0d29731c authored 作者: Frederic's avatar Frederic

fix test crash in debugmode.

上级 9cc893f1
......@@ -20,7 +20,7 @@ else:
mode_with_gpu = theano.compile.mode.get_default_mode().including('gpu')
# The GC need to be enabled for those tests to work correctly.
if mode_with_gpu.linker.allow_gc != True:
if not getattr(mode_with_gpu.linker, 'allow_gc', False):
mode_with_gpu.linker = copy.copy(mode_with_gpu.linker)
mode_with_gpu.linker.allow_gc = True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论