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

One more tricky arg-unpacking case.

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