|
49 | 49 | #define TVM_ARITH_ITER_AFFINE_MAP_H_ |
50 | 50 |
|
51 | 51 | #include <tvm/arith/analyzer.h> |
| 52 | +#include <tvm/ffi/reflection/reflection.h> |
52 | 53 | #include <tvm/ir/diagnostic.h> |
53 | 54 | #include <tvm/ir/expr.h> |
54 | 55 | #include <tvm/tir/var.h> |
55 | | -#include <tvm/ffi/reflection/reflection.h> |
56 | 56 |
|
57 | 57 | namespace tvm { |
58 | 58 | namespace arith { |
@@ -102,8 +102,8 @@ class IterMarkNode : public Object { |
102 | 102 | static void RegisterReflection() { |
103 | 103 | namespace refl = tvm::ffi::reflection; |
104 | 104 | refl::ObjectDef<IterMarkNode>() |
105 | | - .def_ro("source", &IterMarkNode::source) |
106 | | - .def_ro("extent", &IterMarkNode::extent); |
| 105 | + .def_ro("source", &IterMarkNode::source) |
| 106 | + .def_ro("extent", &IterMarkNode::extent); |
107 | 107 | } |
108 | 108 |
|
109 | 109 | static constexpr bool _type_has_method_visit_attrs = false; |
@@ -161,10 +161,10 @@ class IterSplitExprNode : public IterMapExprNode { |
161 | 161 | static void RegisterReflection() { |
162 | 162 | namespace refl = tvm::ffi::reflection; |
163 | 163 | refl::ObjectDef<IterSplitExprNode>() |
164 | | - .def_ro("source", &IterSplitExprNode::source) |
165 | | - .def_ro("lower_factor", &IterSplitExprNode::lower_factor) |
166 | | - .def_ro("extent", &IterSplitExprNode::extent) |
167 | | - .def_ro("scale", &IterSplitExprNode::scale); |
| 164 | + .def_ro("source", &IterSplitExprNode::source) |
| 165 | + .def_ro("lower_factor", &IterSplitExprNode::lower_factor) |
| 166 | + .def_ro("extent", &IterSplitExprNode::extent) |
| 167 | + .def_ro("scale", &IterSplitExprNode::scale); |
168 | 168 | } |
169 | 169 |
|
170 | 170 | static constexpr bool _type_has_method_visit_attrs = false; |
@@ -231,8 +231,8 @@ class IterSumExprNode : public IterMapExprNode { |
231 | 231 | static void RegisterReflection() { |
232 | 232 | namespace refl = tvm::ffi::reflection; |
233 | 233 | refl::ObjectDef<IterSumExprNode>() |
234 | | - .def_ro("args", &IterSumExprNode::args) |
235 | | - .def_ro("base", &IterSumExprNode::base); |
| 234 | + .def_ro("args", &IterSumExprNode::args) |
| 235 | + .def_ro("base", &IterSumExprNode::base); |
236 | 236 | } |
237 | 237 |
|
238 | 238 | static constexpr bool _type_has_method_visit_attrs = false; |
@@ -302,9 +302,9 @@ class IterMapResultNode : public Object { |
302 | 302 | static void RegisterReflection() { |
303 | 303 | namespace refl = tvm::ffi::reflection; |
304 | 304 | refl::ObjectDef<IterMapResultNode>() |
305 | | - .def_ro("indices", &IterMapResultNode::indices) |
306 | | - .def_ro("errors", &IterMapResultNode::errors) |
307 | | - .def_ro("padding_predicate", &IterMapResultNode::padding_predicate); |
| 305 | + .def_ro("indices", &IterMapResultNode::indices) |
| 306 | + .def_ro("errors", &IterMapResultNode::errors) |
| 307 | + .def_ro("padding_predicate", &IterMapResultNode::padding_predicate); |
308 | 308 | } |
309 | 309 |
|
310 | 310 | static constexpr bool _type_has_method_visit_attrs = false; |
|
0 commit comments