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