提交 09686c4e authored 作者: Matthew Rocklin's avatar Matthew Rocklin

use make_(depends/dependence) rather than old fn

上级 4346b4c5
......@@ -162,6 +162,7 @@ def sort_schedule_fn(*cmps):
See also:
sort_apply_nodes
"""
dependence = make_dependence_cmp()
cmps = (dependence,) + cmps
def schedule(fgraph):
""" Order nodes in a FunctionGraph """
......
......@@ -166,7 +166,7 @@ class TestWrapLinker(unittest.TestCase):
def test_sort_schedule_fn():
import theano
from theano.gof.sched import sort_schedule_fn, depends
from theano.gof.sched import sort_schedule_fn, make_depends
x = theano.tensor.matrix('x')
y = theano.tensor.dot(x[:5]*2, x.T+1).T
str_cmp = lambda a, b: cmp(str(a), str(b)) # lexicographical sort
......@@ -175,6 +175,7 @@ def test_sort_schedule_fn():
f = theano.function((x,), (y,), mode=mode)
nodes = f.maker.linker.make_all()[-1]
depends = make_depends()
for a, b in zip(nodes[:-1], nodes[1:]):
if not depends((b,a)):
assert str(a) < str(b)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论