Fix code style according to lint.

apache · Dec 27, 2017 · 1473e52 · 1473e52
1 parent cdbf01a
commit 1473e52
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 7 deletions.
diff --git a/src/codegen/codegen_opengl.cc b/src/codegen/codegen_opengl.cc
@@ -137,5 +137,5 @@ void CodeGenOpenGL::VisitExpr_(const StringImm*, std::ostream& os) {
   LOG(FATAL) << "GLSL 3.3 doesn't support strings.";
 }
 
-}  // namespace tvm
 }  // namespace codegen
+}  // namespace tvm
diff --git a/src/codegen/codegen_opengl.h b/src/codegen/codegen_opengl.h
@@ -16,7 +16,7 @@ namespace codegen {
 
 class CodeGenOpenGL final : public CodeGenC {
  public:
-  explicit CodeGenOpenGL();
+  CodeGenOpenGL();
 
   void AddFunction(LoweredFunc f);
 

diff --git a/src/runtime/opengl/opengl_common.h b/src/runtime/opengl/opengl_common.h
@@ -13,6 +13,9 @@
 #include <dmlc/logging.h>
 #include <glad/glad.h>
 #include <GLFW/glfw3.h>
+#include <string>
+#include <utility>
+#include <vector>
 
 namespace tvm {
 namespace runtime {

diff --git a/src/runtime/opengl/opengl_module.cc b/src/runtime/opengl/opengl_module.cc
@@ -127,8 +127,12 @@ OpenGLWrappedFunc::OpenGLWrappedFunc(
                     << "nargs = " << arg_size_.size() << ", "
                     << "nthread_axis_tags = " << thread_axis_tags.size()
                     << ")";
-  for (auto& a: arg_size_) { LOG(INFO) << a; }
-  for (auto& t: thread_axis_tags) { LOG(INFO) << t; }
+  for (auto& a : arg_size_) {
+    LOG(INFO) << a;
+  }
+  for (auto& t : thread_axis_tags) {
+    LOG(INFO) << t;
+  }
 
   thread_axis_cfg_.Init(arg_size_.size(), thread_axis_tags);
 }
@@ -144,8 +148,7 @@ void OpenGLWrappedFunc::operator()(TVMArgs args, TVMRetValue* rv,
           {"A", *static_cast<gl::Texture**>(void_args[1])},
           {"B", *static_cast<gl::Texture**>(void_args[2])}
       },
-      *static_cast<gl::Texture**>(void_args[0])
-  );
+      *static_cast<gl::Texture**>(void_args[0]));
 }
 
 Module OpenGLModuleCreate(std::string data,

diff --git a/src/schedule/schedule_lang.cc b/src/schedule/schedule_lang.cc
@@ -401,7 +401,7 @@ Stage& Stage::opengl() {
   CHECK(!is_scheduled()) << "Must be a fresh schedule";
   StageNode *self = operator->();
 
-  auto all_iter_vars = self->all_iter_vars; // curr version of all_iter_vars
+  auto all_iter_vars = self->all_iter_vars;  // curr version of all_iter_vars
   CHECK(!all_iter_vars.empty()) << "At least one iter var";
 
   // Fuse all dimensions to 1.