From db3dcba285e727b4d8b2b9c5d0aabe5e743bd34f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9B=BE=E9=9B=A8=E9=AD=94=E7=90=86=E6=B2=99?=
<lolisa@marisa.moe>
Date: Thu, 25 Oct 2018 09:28:39 -0700
Subject: [PATCH] [Relay] visit the span (#1990)
---
include/tvm/relay/type.h | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/include/tvm/relay/type.h b/include/tvm/relay/type.h
index 2bb9b3070..5d8eca037 100644
--- a/include/tvm/relay/type.h
+++ b/include/tvm/relay/type.h
@@ -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,
--
GitLab