fix build typo

apache · Jul 13, 2020 · c61ac20 · c61ac20
1 parent 1a617ba
commit c61ac20
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 5 deletions.
diff --git a/src/relay/op/nn/convolution.cc b/src/relay/op/nn/convolution.cc
@@ -22,7 +22,6 @@
  * \brief Convolution operators
  */
 #include "convolution.h"
-#include "convolution_make.h"
 
 #include <tvm/relay/attrs/nn.h>
 #include <tvm/relay/op.h>
@@ -32,6 +31,7 @@
 
 #include "../../transforms/infer_layout_util.h"
 #include "../op_common.h"
+#include "convolution_make.h"
 
 namespace tvm {
 namespace relay {

diff --git a/src/relay/op/nn/convolution_make.h b/src/relay/op/nn/convolution_make.h
@@ -24,7 +24,8 @@
 #ifndef TVM_RELAY_OP_NN_CONVOLUTION_MAKE_H_
 #define TVM_RELAY_OP_NN_CONVOLUTION_MAKE_H_
 
-#include <tvm/tir/analysis.h>
+#include <tvm/relay/attrs/nn.h>
+#include <tvm/relay/op.h>
 
 #include <string>
 #include <utility>

diff --git a/src/relay/op/nn/pooling.cc b/src/relay/op/nn/pooling.cc
@@ -21,6 +21,8 @@
  * \file pooling.cc
  * \brief Pooling operators
  */
+#include "pooling.h"
+
 #include <topi/nn/pooling.h>
 #include <tvm/relay/attrs/nn.h>
 #include <tvm/relay/op.h>
@@ -29,7 +31,6 @@
 
 #include <vector>
 
-#include "pooling.h"
 #include "../../transforms/infer_layout_util.h"
 
 namespace tvm {

diff --git a/src/relay/op/nn/pooling.h b/src/relay/op/nn/pooling.h
@@ -24,6 +24,9 @@
 #ifndef TVM_RELAY_OP_NN_POOLING_H_
 #define TVM_RELAY_OP_NN_POOLING_H_
 
+#include <tvm/relay/attrs/nn.h>
+#include <tvm/relay/op.h>
+
 #include <utility>
 
 namespace tvm {

diff --git a/src/relay/transforms/dynamic_to_static.cc b/src/relay/transforms/dynamic_to_static.cc
@@ -33,9 +33,10 @@ namespace relay {
 
 class DynamicToStaticMutator : public MixedModeMutator {
  public:
-  DynamicToStaticMutator()
+  DynamicToStaticMutator() {}
 
-      private : Expr Rewrite_(const CallNode* pre, const Expr& post) override {
+ private:
+  Expr Rewrite_(const CallNode* pre, const Expr& post) override {
     const CallNode* call_node = post.as<CallNode>();
     if (call_node->op == Op::Get("dyn.reshape")) {
       if (const ConstantNode* shape = call_node->args[1].as<ConstantNode>()) {