提交 407f0f5c authored 作者: Frederic Bastien's avatar Frederic Bastien

Don't check input if we told not

上级 aaab570f
......@@ -318,7 +318,8 @@ class NanGuardMode(Mode):
# If the input is the result of computation, then we
# don't need to check it. It is already done after the
# computation.
if var.owner is not None:
if (var.owner is not None and
getattr(var.tag, 'nan_guard_mode_check', True)):
do_check_on(x[0], node, fn, True)
fn()
outputs = fn.outputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论