Skip to content

Commit 2c79399

Browse files
MarisaKirisamejroesch
MarisaKirisame
authored and
jroesch
committed
override all the stuff (apache#20)
1 parent 3fe4cfa commit 2c79399

File tree

3 files changed

+51
-51
lines changed

3 files changed

+51
-51
lines changed

‎relay/include/relay/evaluator.h‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -24,22 +24,22 @@ class Evaluator : public ExprFunctor<Value(const Expr& n)> {
2424
Evaluator();
2525
Evaluator(Environment env) : env(env) {}
2626
Value Eval(const Expr& expr);
27-
Value VisitExpr_(const LocalIdNode* op);
28-
Value VisitExpr_(const GlobalIdNode* op);
29-
Value VisitExpr_(const IntrinsicIdNode* op);
30-
Value VisitExpr_(const FloatLitNode* op);
31-
Value VisitExpr_(const BoolLitNode* op);
32-
Value VisitExpr_(const IntLitNode* op);
33-
Value VisitExpr_(const TensorLitNode* op);
34-
Value VisitExpr_(const ProductLitNode* op);
35-
Value VisitExpr_(const CastNode* op);
36-
Value VisitExpr_(const ParamNode* op);
37-
Value VisitExpr_(const FunctionNode* op);
38-
Value VisitExpr_(const CallNode* op);
39-
Value VisitExpr_(const DebugNode* op);
40-
Value VisitExpr_(const UnaryOpNode* op);
41-
Value VisitExpr_(const BinaryOpNode* op);
42-
Value VisitExpr_(const AssignmentNode* op);
27+
Value VisitExpr_(const LocalIdNode* op) override;
28+
Value VisitExpr_(const GlobalIdNode* op) override;
29+
Value VisitExpr_(const IntrinsicIdNode* op) override;
30+
Value VisitExpr_(const FloatLitNode* op) override;
31+
Value VisitExpr_(const BoolLitNode* op) override;
32+
Value VisitExpr_(const IntLitNode* op) override;
33+
Value VisitExpr_(const TensorLitNode* op) override;
34+
Value VisitExpr_(const ProductLitNode* op) override;
35+
Value VisitExpr_(const CastNode* op) override;
36+
Value VisitExpr_(const ParamNode* op) override;
37+
Value VisitExpr_(const FunctionNode* op) override;
38+
Value VisitExpr_(const CallNode* op) override;
39+
Value VisitExpr_(const DebugNode* op) override;
40+
Value VisitExpr_(const UnaryOpNode* op) override;
41+
Value VisitExpr_(const BinaryOpNode* op) override;
42+
Value VisitExpr_(const AssignmentNode* op) override;
4343
};
4444

4545
} // namespace relay

‎relay/include/relay/typechecker.h‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -20,22 +20,22 @@ class Typechecker : public ExprFunctor<Type(const Expr & n)> {
2020
Typechecker();
2121
Typechecker(Environment env) : env(env) {}
2222
Type Check(const Expr & expr);
23-
Type VisitExpr_(const LocalIdNode* op);
24-
Type VisitExpr_(const GlobalIdNode* op);
25-
Type VisitExpr_(const IntrinsicIdNode* op);
26-
Type VisitExpr_(const FloatLitNode* op);
27-
Type VisitExpr_(const BoolLitNode* op);
28-
Type VisitExpr_(const IntLitNode* op);
29-
Type VisitExpr_(const TensorLitNode* op);
30-
Type VisitExpr_(const ProductLitNode* op);
31-
Type VisitExpr_(const CastNode* op);
32-
Type VisitExpr_(const ParamNode* op);
33-
Type VisitExpr_(const FunctionNode* op);
34-
Type VisitExpr_(const CallNode* op);
35-
Type VisitExpr_(const DebugNode* op);
36-
Type VisitExpr_(const UnaryOpNode* op);
37-
Type VisitExpr_(const BinaryOpNode* op);
38-
Type VisitExpr_(const AssignmentNode* op);
23+
Type VisitExpr_(const LocalIdNode* op) override;
24+
Type VisitExpr_(const GlobalIdNode* op) override;
25+
Type VisitExpr_(const IntrinsicIdNode* op) override;
26+
Type VisitExpr_(const FloatLitNode* op) override;
27+
Type VisitExpr_(const BoolLitNode* op) override;
28+
Type VisitExpr_(const IntLitNode* op) override;
29+
Type VisitExpr_(const TensorLitNode* op) override;
30+
Type VisitExpr_(const ProductLitNode* op) override;
31+
Type VisitExpr_(const CastNode* op) override;
32+
Type VisitExpr_(const ParamNode* op) override;
33+
Type VisitExpr_(const FunctionNode* op) override;
34+
Type VisitExpr_(const CallNode* op) override;
35+
Type VisitExpr_(const DebugNode* op) override;
36+
Type VisitExpr_(const UnaryOpNode* op) override;
37+
Type VisitExpr_(const BinaryOpNode* op) override;
38+
Type VisitExpr_(const AssignmentNode* op) override;
3939
};
4040

4141
} // namespace relay

‎relay/include/relay/visitor.h‎

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -45,15 +45,15 @@ class FunctorNode
4545

4646
void VisitAttrs(tvm::AttrVisitor* v) final {}
4747

48-
NodeRef VisitExpr_(const LocalIdNode* local, tvm::Array<NodeRef> args) {
48+
NodeRef VisitExpr_(const LocalIdNode* local, tvm::Array<NodeRef> args) override {
4949
if (visit_local_id != nullptr) {
5050
return visit_local_id(local->name, args);
5151
} else {
5252
return LocalIdNode::make(local->name);
5353
}
5454
}
5555

56-
NodeRef VisitExpr_(const GlobalIdNode* global, tvm::Array<NodeRef> args) {
56+
NodeRef VisitExpr_(const GlobalIdNode* global, tvm::Array<NodeRef> args) override {
5757
if (visit_global_id != nullptr) {
5858
return visit_global_id(global->name, args);
5959
} else {
@@ -62,31 +62,31 @@ class FunctorNode
6262
}
6363

6464
NodeRef VisitExpr_(const IntrinsicIdNode* intrinsic,
65-
tvm::Array<NodeRef> args) {
65+
tvm::Array<NodeRef> args) override {
6666
if (visit_intrinsic_id != nullptr) {
6767
return visit_intrinsic_id(intrinsic->name, args);
6868
} else {
6969
return IntrinsicIdNode::make(intrinsic->name);
7070
}
7171
}
7272

73-
NodeRef VisitExpr_(const FloatLitNode* float_lit, tvm::Array<NodeRef> args) {
73+
NodeRef VisitExpr_(const FloatLitNode* float_lit, tvm::Array<NodeRef> args) override {
7474
if (visit_float_lit != nullptr) {
7575
return visit_float_lit(float_lit->value, args);
7676
} else {
7777
return FloatLitNode::make(float_lit->value);
7878
}
7979
}
8080

81-
NodeRef VisitExpr_(const BoolLitNode* bool_lit, tvm::Array<NodeRef> args) {
81+
NodeRef VisitExpr_(const BoolLitNode* bool_lit, tvm::Array<NodeRef> args) override {
8282
if (visit_bool_lit != nullptr) {
8383
return visit_bool_lit(bool_lit->value, args);
8484
} else {
8585
return BoolLitNode::make(bool_lit->value);
8686
}
8787
}
8888

89-
NodeRef VisitExpr_(const IntLitNode* int_lit, tvm::Array<NodeRef> args) {
89+
NodeRef VisitExpr_(const IntLitNode* int_lit, tvm::Array<NodeRef> args) override {
9090
if (visit_int_lit != nullptr) {
9191
return visit_int_lit(int_lit->value, args);
9292
} else {
@@ -95,7 +95,7 @@ class FunctorNode
9595
}
9696

9797
NodeRef VisitExpr_(const TensorLitNode* tensor_lit,
98-
tvm::Array<NodeRef> args) {
98+
tvm::Array<NodeRef> args) override {
9999
if (visit_tensor_lit != nullptr) {
100100
return visit_tensor_lit(tensor_lit->data, args);
101101
} else {
@@ -104,63 +104,63 @@ class FunctorNode
104104
}
105105

106106
NodeRef VisitExpr_(const ProductLitNode* product_lit,
107-
tvm::Array<NodeRef> args) {
107+
tvm::Array<NodeRef> args) override {
108108
if (visit_product_lit != nullptr) {
109109
return visit_product_lit(product_lit->fields, args);
110110
} else {
111111
return ProductLitNode::make(product_lit->fields);
112112
}
113113
}
114114

115-
NodeRef VisitExpr_(const CastNode* cast, tvm::Array<NodeRef> args) {
115+
NodeRef VisitExpr_(const CastNode* cast, tvm::Array<NodeRef> args) override {
116116
if (visit_cast != nullptr) {
117117
return visit_cast(cast->target, cast->node, args);
118118
} else {
119119
return CastNode::make(cast->target, cast->node);
120120
}
121121
}
122122

123-
NodeRef VisitExpr_(const ParamNode* param, tvm::Array<NodeRef> args) {
123+
NodeRef VisitExpr_(const ParamNode* param, tvm::Array<NodeRef> args) override {
124124
if (visit_param != nullptr) {
125125
return visit_param(param->id, param->type, args);
126126
} else {
127127
return ParamNode::make(param->id, param->type);
128128
}
129129
}
130130

131-
NodeRef VisitExpr_(const FunctionNode* fn, tvm::Array<NodeRef> args) {
131+
NodeRef VisitExpr_(const FunctionNode* fn, tvm::Array<NodeRef> args) override {
132132
if (visit_function != nullptr) {
133133
return visit_function(fn->params, fn->body, args);
134134
} else {
135135
return FunctionNode::make(fn->params, fn->body);
136136
}
137137
}
138138

139-
NodeRef VisitExpr_(const CallNode* call, tvm::Array<NodeRef> args) {
139+
NodeRef VisitExpr_(const CallNode* call, tvm::Array<NodeRef> args) override {
140140
if (visit_call != nullptr) {
141141
return visit_call(call->fn, call->args, args);
142142
} else {
143143
return CallNode::make(call->fn, call->args);
144144
}
145145
}
146146

147-
NodeRef VisitExpr_(const DebugNode* debug, tvm::Array<NodeRef> args) {
147+
NodeRef VisitExpr_(const DebugNode* debug, tvm::Array<NodeRef> args) override {
148148
if (visit_debug != nullptr) {
149149
return visit_debug(debug->node, args);
150150
} else {
151151
return DebugNode::make(debug->node);
152152
}
153153
}
154154

155-
NodeRef VisitExpr_(const UnaryOpNode* uop, tvm::Array<NodeRef> args) {
155+
NodeRef VisitExpr_(const UnaryOpNode* uop, tvm::Array<NodeRef> args) override {
156156
if (visit_unary_op != nullptr) {
157157
return visit_unary_op(static_cast<int>(uop->op), uop->node, args);
158158
} else {
159159
return UnaryOpNode::make(uop->op, uop->node);
160160
}
161161
}
162162

163-
NodeRef VisitExpr_(const BinaryOpNode* bop, tvm::Array<NodeRef> args) {
163+
NodeRef VisitExpr_(const BinaryOpNode* bop, tvm::Array<NodeRef> args) override {
164164
if (visit_binary_op != nullptr) {
165165
return visit_binary_op(static_cast<int>(bop->op), bop->left, bop->right,
166166
args);
@@ -169,7 +169,7 @@ class FunctorNode
169169
}
170170
}
171171

172-
NodeRef VisitExpr_(const AssignmentNode* bop, tvm::Array<NodeRef> args) {
172+
NodeRef VisitExpr_(const AssignmentNode* bop, tvm::Array<NodeRef> args) override {
173173
throw "foo";
174174
// if (visit_assignment != nullptr) {
175175
// return visit_assignment(intrinsic->name, args);
@@ -178,23 +178,23 @@ class FunctorNode
178178
// }
179179
}
180180

181-
NodeRef VisitExpr_(const ReverseNode* rev, tvm::Array<NodeRef> args) {
181+
NodeRef VisitExpr_(const ReverseNode* rev, tvm::Array<NodeRef> args) override {
182182
if (visit_reverse != nullptr) {
183183
return visit_reverse(rev->node, args);
184184
} else {
185185
return ReverseNode::make(rev->node);
186186
}
187187
}
188188

189-
NodeRef VisitExpr_(const AccumulateNode* acc, tvm::Array<NodeRef> args) {
189+
NodeRef VisitExpr_(const AccumulateNode* acc, tvm::Array<NodeRef> args) override {
190190
if (visit_accumulate != nullptr) {
191191
return visit_accumulate(acc->update_binders, acc->value, args);
192192
} else {
193193
return AccumulateNode::make(acc->update_binders, acc->value);
194194
}
195195
}
196196

197-
NodeRef VisitExpr_(const ZeroNode* zero, tvm::Array<NodeRef> args) {
197+
NodeRef VisitExpr_(const ZeroNode* zero, tvm::Array<NodeRef> args) override {
198198
if (visit_zero != nullptr) {
199199
return visit_zero(zero->type, args);
200200
} else {

0 commit comments

Comments
 (0)