提交 3811ac70 authored 作者: Frederic Bastien's avatar Frederic Bastien

Don't check inputs if it is the output of computation. If should be already tested

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