Skip to content

T
tic
Commit db3dcba2 by 雾雨魔理沙 Committed by Tianqi Chen
Options

[Relay] visit the span (#1990)

parent 55d26ae8
Showing with 6 additions and 1 deletions
include/tvm/relay/type.h
...@@ -161,6 +161,7 @@ class IncompleteTypeNode : public TypeNode { ...@@ -161,6 +161,7 @@ class IncompleteTypeNode : public TypeNode {
void VisitAttrs(tvm::AttrVisitor* v) final { void VisitAttrs(tvm::AttrVisitor* v) final {
v->Visit("kind", &kind); v->Visit("kind", &kind);
v->Visit("span", &span);
} }
TVM_DLL static IncompleteType make(TypeVarNode::Kind kind); TVM_DLL static IncompleteType make(TypeVarNode::Kind kind);
...@@ -243,7 +244,10 @@ class TupleTypeNode : public TypeNode { ...@@ -243,7 +244,10 @@ class TupleTypeNode : public TypeNode {
TupleTypeNode() {} 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); TVM_DLL static TupleType make(tvm::Array<Type> fields);
...@@ -361,6 +365,7 @@ class TypeRelationNode : public TypeConstraintNode { ...@@ -361,6 +365,7 @@ class TypeRelationNode : public TypeConstraintNode {
v->Visit("args", &args); v->Visit("args", &args);
v->Visit("num_inputs", &num_inputs); v->Visit("num_inputs", &num_inputs);
v->Visit("attrs", &attrs); v->Visit("attrs", &attrs);
v->Visit("span", &span);
} }
TVM_DLL static TypeRelation make(TypeRelationFn func, 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