Fix lint

apache · Dec 18, 2021 · 09e1f9d · 09e1f9d
1 parent f7f1cb3
commit 09e1f9d
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
@@ -405,14 +405,15 @@ TVM_REGISTER_TARGET_KIND("composite", kDLCPU)  // line break
 
 /**********  Registry  **********/
 
-TVM_REGISTER_GLOBAL("target.TargetKindGetAttr").set_body_typed([](TargetKind kind, String attr_name) -> TVMRetValue {
-    auto target_attr_map = TargetKind::GetAttrMap<TVMRetValue>(attr_name);
-    TVMRetValue rv;
-    if (target_attr_map.count(kind)) {
-      rv = target_attr_map[kind];
-    }
-    return rv;
-  });
+TVM_REGISTER_GLOBAL("target.TargetKindGetAttr")
+    .set_body_typed([](TargetKind kind, String attr_name) -> TVMRetValue {
+      auto target_attr_map = TargetKind::GetAttrMap<TVMRetValue>(attr_name);
+      TVMRetValue rv;
+      if (target_attr_map.count(kind)) {
+        rv = target_attr_map[kind];
+      }
+      return rv;
+    });
 TVM_REGISTER_GLOBAL("target.ListTargetKinds").set_body_typed(TargetKindRegEntry::ListTargetKinds);
 TVM_REGISTER_GLOBAL("target.ListTargetKindOptions")
     .set_body_typed(TargetKindRegEntry::ListTargetKindOptions);