[checkpoint] lints

apache · Oct 19, 2021 · 032e63f · 032e63f
1 parent 2e7e107
commit 032e63f
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 15 deletions.
diff --git a/include/tvm/relay/attrs/device_copy.h b/include/tvm/relay/attrs/device_copy.h
@@ -39,7 +39,6 @@ struct DeviceCopyAttrs : public tvm::AttrsNode<DeviceCopyAttrs> {
   SEScope src_se_scope = SEScope::FullyUnconstrained();
   SEScope dst_se_scope = SEScope::FullyUnconstrained();
 
-
   TVM_DECLARE_ATTRS(DeviceCopyAttrs, "relay.attrs.DeviceCopyAttrs") {
     TVM_ATTR_FIELD(src_se_scope)
         .describe("The (virtual) device and scope where the op copies data from.");

diff --git a/include/tvm/relay/attrs/on_device.h b/include/tvm/relay/attrs/on_device.h
@@ -86,8 +86,8 @@ struct OnDeviceAttrs : public tvm::AttrsNode<OnDeviceAttrs> {
   bool is_fixed = false;
 
   TVM_DECLARE_ATTRS(OnDeviceAttrs, "relay.attrs.OnDeviceAttrs") {
-    TVM_ATTR_FIELD(se_scope).describe(
-        "The (virtual) device and scope holding the expression result.")
+    TVM_ATTR_FIELD(se_scope)
+        .describe("The (virtual) device and scope holding the expression result.")
         .set_default(SEScope::FullyUnconstrained());
     TVM_ATTR_FIELD(is_fixed)
         .describe("If true, do not insert a \"device_copy\" call to respect this annotation.")

diff --git a/src/relay/backend/graph_plan_memory.cc b/src/relay/backend/graph_plan_memory.cc
@@ -58,9 +58,7 @@ struct StorageToken {
 
   bool is_valid() const { return !se_scope->is_fully_unconstrained(); }
 
-  bool is_compatible(const StorageToken& that) const {
-    return se_scope == that.se_scope;
-  }
+  bool is_compatible(const StorageToken& that) const { return se_scope == that.se_scope; }
 
   std::string ToString() const {
     std::ostringstream os;

diff --git a/src/relay/backend/te_compiler.cc b/src/relay/backend/te_compiler.cc
@@ -357,8 +357,8 @@ using AnalysisRemapping = std::unordered_map<Expr, Expr, ObjectHash, ObjectEqual
  */
 class LowerTensorExprMutator : public DeviceAwareExprMutator {
  public:
-  LowerTensorExprMutator(const IRModule& module, ProcessFn process_fn,
-                         const String& module_name, TECompiler compiler)
+  LowerTensorExprMutator(const IRModule& module, ProcessFn process_fn, const String& module_name,
+                         TECompiler compiler)
       : DeviceAwareExprMutator(module),
         module_(module),
         process_fn_(process_fn),

diff --git a/src/relay/backend/vm/lambda_lift.cc b/src/relay/backend/vm/lambda_lift.cc
@@ -195,8 +195,7 @@ class LambdaLifter : public transform::DeviceAwareExprMutator {
       lifted_func =
           Function(typed_captured_vars, rebound_body, /*ret_type=*/func->func_type_annotation(),
                    free_type_vars, /*attrs=*/{}, func->span);
-      lifted_func =
-          MaybeFunctionOnDevice(lifted_func, captured_var_se_scopes, result_se_scope);
+      lifted_func = MaybeFunctionOnDevice(lifted_func, captured_var_se_scopes, result_se_scope);
       lifted_func = MarkClosure(lifted_func);
     }
 

diff --git a/src/relay/transforms/device_domains.cc b/src/relay/transforms/device_domains.cc
@@ -221,9 +221,9 @@ SEScope DeviceDomains::SEScopeWithTarget(SEScope se_scope) const {
     return target->kind->device_type == device_type;
   });
   CHECK(itr != targets_.end()) << "The SEScope " << se_scope
-                                << " does not specify a target, yet no target of device type "
-                                << device_type << " is in the " << targets_.size()
-                                << " known targets";
+                               << " does not specify a target, yet no target of device type "
+                               << device_type << " is in the " << targets_.size()
+                               << " known targets";
   return SEScope::MakeSEScope(device_type, se_scope->virtual_device_id(), *itr,
                               se_scope->memory_scope());
 }

diff --git a/src/runtime/vm/vm.cc b/src/runtime/vm/vm.cc
@@ -300,8 +300,8 @@ ObjectRef VirtualMachine::Invoke(const VMFunction& func, const std::vector<Objec
   DLOG(INFO) << "Executing Function: " << std::endl << func;
   for (int i = 0; i < static_cast<int>(devices_.size()); ++i) {
     DLOG(INFO) << "Device " << i << " has device type " << devices_[i].device_type
-               << " and device id " << devices_[i].device_id <<
-        (i == exec_->host_device_index ? " (using as host device)" : "");
+               << " and device id " << devices_[i].device_id
+               << (i == exec_->host_device_index ? " (using as host device)" : "");
   }
 
   InvokeGlobal(func, args);