def test_sched_to_cmp():
sched = an, bn, cn = _test_sched()
cmp = sched_to_cmp(sched)
assert cmp(an, bn) < 0 and cmp(cn, an) > 0
d = theano.tensor.matrix('d')
dn = (d + d).owner
assert cmp(dn, an) == 0
def test_sched_to_cmp():
sched = an, bn, cn = _test_sched()
cmp = sched_to_cmp(sched)
assert cmp(an, bn) < 0 and cmp(cn, an) > 0
d = theano.tensor.matrix('d')
dn = (d+d).owner
assert cmp(dn, an) == 0
def test_make_scheduler():
fgraph = read_graph(ape_dir + 'ape/codegen/tests/test.fgraph')
sched = read_sched(ape_dir + 'ape/codegen/tests/test.sched')
sched_cmp = sched_to_cmp(sched)
scheduler = make_scheduler(sched_cmp)
nodes = scheduler(fgraph)
nodestrings = map(str, nodes)
print set(sched) - set(nodestrings)
assert set(sched).issubset(set(nodestrings))
indices = map(lambda line: nodestrings.index(line), sched)
assert sorted(indices) == indices
def test_make_scheduler():
fgraph = read_graph(ape_dir+'ape/codegen/tests/test.fgraph')
sched = read_sched(ape_dir+'ape/codegen/tests/test.sched')
sched_cmp = sched_to_cmp(sched)
scheduler = make_scheduler(sched_cmp)
nodes = scheduler(fgraph)
nodestrings = map(str, nodes)
print set(sched) - set(nodestrings)
assert set(sched).issubset(set(nodestrings))
indices = map(lambda line: nodestrings.index(line), sched)
assert sorted(indices) == indices
