diff --git a/include/tvm/relay/attrs/device_copy.h b/include/tvm/relay/attrs/device_copy.h index 8bd05db5c9cd8..6d97ab79be4a2 100644 --- 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 { 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 index 8a78977ee1497..405926e209c69 100644 --- 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 { 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 index f612f0c952640..faed6b46ce024 100644 --- 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 index f0613092b9c93..35752e8895f82 100644 --- a/src/relay/backend/te_compiler.cc +++ b/src/relay/backend/te_compiler.cc @@ -357,8 +357,8 @@ using AnalysisRemapping = std::unordered_mapfunc_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 index 74d9a8f3aaae6..a7e46e2ad30cc 100644 --- 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 index 4b6141f573f1d..1372ff85a314f 100644 --- 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(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);