Commit db3dcba2 by 雾雨魔理沙 Committed by Tianqi Chen

[Relay] visit the span (#1990)

parent 55d26ae8
......@@ -161,6 +161,7 @@ class IncompleteTypeNode : public TypeNode {
void VisitAttrs(tvm::AttrVisitor* v) final {
v->Visit("kind", &kind);
v->Visit("span", &span);
}
TVM_DLL static IncompleteType make(TypeVarNode::Kind kind);
......@@ -243,7 +244,10 @@ class TupleTypeNode : public TypeNode {
TupleTypeNode() {}
void VisitAttrs(tvm::AttrVisitor* v) final { v->Visit("fields", &fields); }
void VisitAttrs(tvm::AttrVisitor* v) final {
v->Visit("fields", &fields);
v->Visit("span", &span);
}
TVM_DLL static TupleType make(tvm::Array<Type> fields);
......@@ -361,6 +365,7 @@ class TypeRelationNode : public TypeConstraintNode {
v->Visit("args", &args);
v->Visit("num_inputs", &num_inputs);
v->Visit("attrs", &attrs);
v->Visit("span", &span);
}
TVM_DLL static TypeRelation make(TypeRelationFn func,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment