Skip to content

Commit

Permalink
* test case
Browse files Browse the repository at this point in the history
  • Loading branch information
srkreddy1238 committed Jan 14, 2019
1 parent 4dfbec3 commit 4497b31
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions tests/python/relay/test_backend_compile_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,17 @@ def get_func(shape):
y.asnumpy(), x.asnumpy() * 3)
engine.dump()

def test_compile_placeholder_bypass():
engine = relay.backend.compile_engine.get()
x = relay.var("x", shape=(2, 3))
y = relay.var("y", shape=(2, 3))
z = relay.var("z", shape=(2, 3))
result = relay.Tuple([relay.op.concatenate([y, z], axis=0)])
#result = relay.Tuple([x, relay.op.concatenate([y, z], axis=0)])
func = relay.Function(relay.ir_pass.free_vars(result), result)
with relay.build_config(opt_level=3):
graph, lib, params = relay.build(func, 'llvm')

if __name__ == "__main__":
test_compile_engine()
test_compile_placeholder_bypass()

0 comments on commit 4497b31

Please sign in to comment.