def test_span():
x = relay.var("x", shape=(3, 2))
y = relay.var("y")
one = relay.const(10e10, dtype="float32")
z = relay.add(x, one)
z = relay.Call(z.op, z.args, z.attrs, z.type_args,
relay.Span(relay.SourceName("Add0"), 0, 0, 0, 0))
z = relay.add(z, z)
z = relay.Call(z.op, z.args, z.attrs, z.type_args,
relay.Span(relay.SourceName("Add1"), 0, 0, 0, 0))
f = relay.Function([x, y], z)
txt = astext(f)
assert "Add0" in txt
assert "Add1" in txt
def test_span():
span = relay.Span(None, 1, 1)
assert span.source == None
assert span.lineno == 1
assert span.col_offset == 1
assert span.same_as(span)
assert span == span
assert isinstance(span, relay.base.Span)
str(span)
def test_span():
span = relay.Span(None, 1, 1)
assert span.source == None
assert span.lineno == 1
assert span.col_offset == 1
assert span.same_as(span)
assert span == span
assert isinstance(span, relay.base.Span)
str(span)
# span is not a node so we can't use graph_equal
# to test the round trip
back = tvm.load_json(tvm.save_json(span))
assert back.source == span.source
assert back.lineno == span.lineno
assert back.col_offset == span.col_offset
def test_span():
span = relay.Span(None, 1, 2, 3, 4)
assert span.source_name == None
assert span.line == 1
assert span.end_line == 2
assert span.column == 3
assert span.end_column == 4
assert span.same_as(span)
assert span == span
assert isinstance(span, relay.base.Span)
str(span)
# span is not a node so we can't use graph_equal
# to test the round trip
back = tvm.ir.load_json(tvm.ir.save_json(span))
assert back.source_name == span.source_name
assert back.line == span.line
assert back.end_line == span.end_line
assert back.column == span.column
assert back.end_column == span.end_column
