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