Fix lint

apache · Aug 4, 2022 · 4c010ca · 4c010ca
1 parent 5215df4
commit 4c010ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/include/tvm/script/printer/traced_object_functor.h b/include/tvm/script/printer/traced_object_functor.h
@@ -16,15 +16,17 @@
  * specific language governing permissions and limitations
  * under the License.
  */
-#ifndef TVM_SCRIPT_PRINTER_REGISTRY_H_
-#define TVM_SCRIPT_PRINTER_REGISTRY_H_
+#ifndef TVM_SCRIPT_PRINTER_TRACED_OBJECT_FUNCTOR_H_
+#define TVM_SCRIPT_PRINTER_TRACED_OBJECT_FUNCTOR_H_
 
 #include <tvm/node/node.h>
 #include <tvm/runtime/logging.h>
 #include <tvm/runtime/packed_func.h>
 #include <tvm/script/printer/traced_object.h>
 
+#include <string>
 #include <type_traits>
+#include <unordered_map>
 #include <utility>
 #include <vector>
 
@@ -163,4 +165,4 @@ class TracedObjectFunctor {
 }  // namespace printer
 }  // namespace script
 }  // namespace tvm
-#endif  // TVM_SCRIPT_PRINTER_REGISTRY_H_
+#endif  // TVM_SCRIPT_PRINTER_TRACED_OBJECT_FUNCTOR_H_
diff --git a/src/script/printer/traced_object_functor.cc b/src/script/printer/traced_object_functor.cc
@@ -40,7 +40,7 @@ const runtime::PackedFunc* GetDispatchFunctionForToken(const DispatchTable& tabl
   } else {
     return nullptr;
   }
-};
+}
 
 const runtime::PackedFunc& GetDispatchFunction(const DispatchTable& dispatch_table,
                                                const String& token, uint32_t type_index) {