[Relay] Fix linting errors

apache · Jul 20, 2021 · 452815d · 452815d
1 parent 9fd6552
commit 452815d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 9 deletions.
diff --git a/src/relay/backend/vm/compiler.cc b/src/relay/backend/vm/compiler.cc
@@ -47,10 +47,10 @@
 #include "../../../target/source/codegen_source_base.h"
 #include "../../op/op_common.h"
 #include "../../transforms/pass_utils.h"
+#include "../te_compiler.h"
 #include "../te_compiler_cache.h"
 #include "../utils.h"
 #include "compiler.h"
-#include "../te_compiler.h"
 
 namespace tvm {
 namespace relay {
@@ -859,7 +859,7 @@ class VMFunctionCompiler : ExprFunctor<void(const Expr& expr)> {
   /*! \brief Total number of virtual registers allocated. */
   size_t registers_num_;
   /*! \brief Compiler engine to lower primitive functions. */
-  TECompiler compiler_;  
+  TECompiler compiler_;
   /*! \brief Global shared meta data */
   VMCompilerContext* context_;
   /*! \brief Target devices. */
@@ -1198,7 +1198,7 @@ void VMCompiler::Codegen() {
   }
   lib = codegen::CreateMetadataModule(params_, lib, ext_mods, target_host_, runtime::Metadata());
   exec_->SetLib(lib);
-} 
+}
 
 ExprDeviceMap VMCompiler::AnalyzeContext() const {
   Device default_device;

diff --git a/src/relay/backend/vm/compiler.h b/src/relay/backend/vm/compiler.h
@@ -44,9 +44,8 @@
 #include "../../../runtime/vm/naive_allocator.h"
 #include "../../../runtime/vm/profiler/vm.h"
 #include "../../transforms/pass_utils.h"
-#include "../te_compiler_cache.h"
 #include "../te_compiler.h"
-
+#include "../te_compiler_cache.h"
 
 namespace tvm {
 namespace relay {

diff --git a/src/relay/transforms/memory_alloc.cc b/src/relay/transforms/memory_alloc.cc
@@ -41,15 +41,14 @@
 #include <unordered_set>
 #include <vector>
 
+#include "../backend/te_compiler.h"
+#include "../backend/te_compiler_cache.h"
 #include "../op/memory/memory.h"
 #include "../op/vm/vm.h"
 #include "./pass_utils.h"
 #include "let_list.h"
 #include "pattern_utils.h"
 
-#include "../backend/te_compiler_cache.h"
-#include "../backend/te_compiler.h"
-
 using namespace tvm::runtime;
 using namespace tvm::relay::tec;
 
@@ -275,7 +274,7 @@ class DialectRewriter : public ExprMutator {
                             const std::vector<Expr>& new_args) {
     Array<Expr> shape_func_ins;
 
-   TECompiler compiler;
+    TECompiler compiler;
 
     CCacheKey key(func, target_host_);
     auto cfunc = compiler->LowerShapeFunc(key);