提交 723263f8 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make sure NanGuardMode works with lazy nodes.

上级 9c674006
......@@ -331,7 +331,8 @@ class NanGuardMode(Mode):
def nan_check(node, thunk, storage_map, compute_map):
for var in node.outputs:
if getattr(var.tag, 'nan_guard_mode_check', True):
if (compute_map[var][0] and
getattr(var.tag, 'nan_guard_mode_check', True)):
do_check_on(storage_map[var][0], node)
def nan_check_input(var, value):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论