提交 6104d3f9 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix crash in profile_memory=True. Fix gh-4130

上级 5f89fb9c
......@@ -822,7 +822,7 @@ class ProfileStats(object):
running_memory_size[1])
# Mimic the combination of Theano and Python gc
for ins in node.inputs:
for ins in set(node.inputs):
assert not (ins in view_of and viewed_by[ins])
# we trac the original var, so this shouldn't happen
if isinstance(ins.type, CudaNdarrayType):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论