提交 da304dbd authored 作者: David Warde-Farley's avatar David Warde-Farley

One more tricky arg-unpacking case.

上级 7ce2ab47
......@@ -36,7 +36,7 @@ def memodict(f):
def make_depends():
@memodict
def depends((a, b)):
def depends(a, b):
""" Returns True if a depends on b """
return (any(bout in a.inputs for bout in b.outputs)
or any(depends((ainp.owner, b)) for ainp in a.inputs
......@@ -56,9 +56,9 @@ def make_dependence_cmp():
Returns negative number if b depends on a
Returns 0 otherwise
"""
if depends((a, b)):
if depends(a, b):
return 1
if depends((b, a)):
if depends(b, a):
return -1
return 0
......
......@@ -184,7 +184,7 @@ def test_sort_schedule_fn():
nodes = f.maker.linker.make_all()[-1]
depends = make_depends()
for a, b in zip(nodes[:-1], nodes[1:]):
if not depends((b, a)):
if not depends(b, a):
assert str(a) < str(b)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论