def test_tuple(): x = relay.Var("x") assert well_formed(x) v = relay.Constant(tvm.nd.array(10)) let = relay.Let(x, v, x) assert well_formed(let) assert well_formed(relay.Tuple([v, v])) assert not well_formed(relay.Tuple([let, relay.Let(x, v, x)]))
def test_adt(): mod = tvm.IRModule() p = Prelude(mod) x = relay.Var("x") some_case = relay.Clause( relay.PatternConstructor(p.some, [relay.PatternVar(x)]), x) default_case = relay.Clause(relay.PatternVar(x), x) m0 = relay.Match(p.none(), [default_case]) m1 = relay.Match(p.none(), [some_case, default_case]) assert well_formed(m0) assert not well_formed(m1)
def test_let(): x = relay.Var("x") assert well_formed(x) v = relay.Constant(tvm.nd.array(10)) ty = None let = relay.Let(x, v, x) assert well_formed(let) assert not well_formed(relay.Let(x, v, let)) f = relay.Function([x], x, ty) assert well_formed(f) assert well_formed( relay.Let(relay.Var("y"), f, relay.Let(relay.Var("z"), f, v)))
def test_adt(): mod = tvm.IRModule() p = Prelude(mod) _, none, some = p.mod.get_type("Option") x = relay.Var("x") some_case = relay.Clause( relay.PatternConstructor(some, [relay.PatternVar(x)]), x) default_case = relay.Clause(relay.PatternVar(x), x) m0 = relay.Match(none(), [default_case]) m1 = relay.Match(none(), [some_case, default_case]) assert well_formed(m0) assert not well_formed(m1)
def test_tuple_get_item(): t = relay.Var("t") assert well_formed(relay.TupleGetItem(t, 2))