From 2dcfd6134017010ce79ba650ab34e27b0620323e Mon Sep 17 00:00:00 2001
From: Junru Shao <junrushao1994@gmail.com>
Date: Fri, 19 Jun 2020 19:36:08 -0700
Subject: [PATCH] [Target] Introduce Target Id Registry (#5838)

---
 include/tvm/ir/op.h                  |   8 -
 include/tvm/node/attr_registry_map.h |   2 +
 include/tvm/runtime/registry.h       |   8 +
 include/tvm/target/target_id.h       | 306 +++++++++++++++++++++++++++
 src/node/attr_registry.h             |   1 +
 src/target/target_id.cc              | 164 ++++++++++++++
 tests/cpp/target_test.cc             |  88 ++++++++
 7 files changed, 569 insertions(+), 8 deletions(-)
 create mode 100644 include/tvm/target/target_id.h
 create mode 100644 src/target/target_id.cc
 create mode 100644 tests/cpp/target_test.cc

diff --git a/include/tvm/ir/op.h b/include/tvm/ir/op.h
index 2bc2c90c7854..9a91302a98b4 100644
--- a/include/tvm/ir/op.h
+++ b/include/tvm/ir/op.h
@@ -340,14 +340,6 @@ class OpAttrMap : public AttrRegistryMap<Op, ValueType> {
   explicit OpAttrMap(const AttrRegistryMapContainerMap<Op>& map) : TParent(map) {}
 };
 
-#define TVM_STRINGIZE_DETAIL(x) #x
-#define TVM_STRINGIZE(x) TVM_STRINGIZE_DETAIL(x)
-#define TVM_DESCRIBE(...) describe(__VA_ARGS__ "\n\nFrom:" __FILE__ ":" TVM_STRINGIZE(__LINE__))
-/*!
- * \brief Macro to include current line as string
- */
-#define TVM_ADD_FILELINE "\n\nDefined in " __FILE__ ":L" TVM_STRINGIZE(__LINE__)
-
 // internal macros to make
 #define TVM_OP_REGISTER_VAR_DEF static DMLC_ATTRIBUTE_UNUSED ::tvm::OpRegEntry& __make_##Op
 
diff --git a/include/tvm/node/attr_registry_map.h b/include/tvm/node/attr_registry_map.h
index 748b3a80969c..9c554af9bc21 100644
--- a/include/tvm/node/attr_registry_map.h
+++ b/include/tvm/node/attr_registry_map.h
@@ -23,6 +23,8 @@
 #ifndef TVM_NODE_ATTR_REGISTRY_MAP_H_
 #define TVM_NODE_ATTR_REGISTRY_MAP_H_
 
+#include <tvm/node/container.h>
+
 #include <utility>
 #include <vector>
 
diff --git a/include/tvm/runtime/registry.h b/include/tvm/runtime/registry.h
index 4a5a21088222..86e3706b2058 100644
--- a/include/tvm/runtime/registry.h
+++ b/include/tvm/runtime/registry.h
@@ -289,6 +289,14 @@ class Registry {
 #define TVM_REGISTER_GLOBAL(OpName) \
   TVM_STR_CONCAT(TVM_FUNC_REG_VAR_DEF, __COUNTER__) = ::tvm::runtime::Registry::Register(OpName)
 
+#define TVM_STRINGIZE_DETAIL(x) #x
+#define TVM_STRINGIZE(x) TVM_STRINGIZE_DETAIL(x)
+#define TVM_DESCRIBE(...) describe(__VA_ARGS__ "\n\nFrom:" __FILE__ ":" TVM_STRINGIZE(__LINE__))
+/*!
+ * \brief Macro to include current line as string
+ */
+#define TVM_ADD_FILELINE "\n\nDefined in " __FILE__ ":L" TVM_STRINGIZE(__LINE__)
+
 }  // namespace runtime
 }  // namespace tvm
 #endif  // TVM_RUNTIME_REGISTRY_H_
diff --git a/include/tvm/target/target_id.h b/include/tvm/target/target_id.h
new file mode 100644
index 000000000000..93c88c758006
--- /dev/null
+++ b/include/tvm/target/target_id.h
@@ -0,0 +1,306 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*!
+ * \file tvm/target/target_id.h
+ * \brief Target id registry
+ */
+#ifndef TVM_TARGET_TARGET_ID_H_
+#define TVM_TARGET_TARGET_ID_H_
+
+#include <tvm/ir/expr.h>
+#include <tvm/ir/transform.h>
+#include <tvm/node/attr_registry_map.h>
+#include <tvm/runtime/container.h>
+#include <tvm/runtime/packed_func.h>
+#include <tvm/support/with.h>
+
+#include <memory>
+#include <string>
+#include <unordered_map>
+#include <utility>
+#include <vector>
+
+namespace tvm {
+namespace detail {
+template <typename, typename, typename>
+struct ValueTypeInfoMaker;
+}
+
+/*! \brief Perform schema validation */
+TVM_DLL void TargetValidateSchema(const Map<String, ObjectRef>& config);
+
+template <typename>
+class TargetIdAttrMap;
+
+/*! \brief Target Id, specifies the kind of the target */
+class TargetIdNode : public Object {
+ public:
+  /*! \brief Name of the target id */
+  String name;
+  /*! \brief Stores the required type_key and type_index of a specific attr of a target */
+  struct ValueTypeInfo {
+    String type_key;
+    uint32_t type_index;
+    std::unique_ptr<ValueTypeInfo> key;
+    std::unique_ptr<ValueTypeInfo> val;
+  };
+
+  static constexpr const char* _type_key = "TargetId";
+  TVM_DECLARE_FINAL_OBJECT_INFO(TargetIdNode, Object);
+
+ private:
+  uint32_t AttrRegistryIndex() const { return index_; }
+  String AttrRegistryName() const { return name; }
+  /*! \brief Perform schema validation */
+  void ValidateSchema(const Map<String, ObjectRef>& config) const;
+  /*! \brief A hash table that stores the type information of each attr of the target key */
+  std::unordered_map<String, ValueTypeInfo> key2vtype_;
+  /*! \brief Index used for internal lookup of attribute registry */
+  uint32_t index_;
+  friend void TargetValidateSchema(const Map<String, ObjectRef>&);
+  friend class TargetId;
+  template <typename, typename>
+  friend class AttrRegistry;
+  template <typename>
+  friend class AttrRegistryMapContainerMap;
+  friend class TargetIdRegEntry;
+  template <typename, typename, typename>
+  friend struct detail::ValueTypeInfoMaker;
+};
+
+/*!
+ * \brief Managed reference class to TargetIdNode
+ * \sa TargetIdNode
+ */
+class TargetId : public ObjectRef {
+ public:
+  /*! \brief Get the attribute map given the attribute name */
+  template <typename ValueType>
+  static inline TargetIdAttrMap<ValueType> GetAttrMap(const String& attr_name);
+  /*!
+   * \brief Retrieve the TargetId given its name
+   * \param target_id_name Name of the target id
+   * \return The TargetId requested
+   */
+  TVM_DLL static const TargetId& Get(const String& target_id_name);
+  TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS(TargetId, ObjectRef, TargetIdNode);
+
+ private:
+  /*! \brief Mutable access to the container class  */
+  TargetIdNode* operator->() { return static_cast<TargetIdNode*>(data_.get()); }
+  TVM_DLL static const AttrRegistryMapContainerMap<TargetId>& GetAttrMapContainer(
+      const String& attr_name);
+  template <typename, typename>
+  friend class AttrRegistry;
+  friend class TargetIdRegEntry;
+};
+
+/*!
+ * \brief Map<TargetId, ValueType> used to store meta-information about TargetId
+ * \tparam ValueType The type of the value stored in map
+ */
+template <typename ValueType>
+class TargetIdAttrMap : public AttrRegistryMap<TargetId, ValueType> {
+ public:
+  using TParent = AttrRegistryMap<TargetId, ValueType>;
+  using TParent::count;
+  using TParent::get;
+  using TParent::operator[];
+  explicit TargetIdAttrMap(const AttrRegistryMapContainerMap<TargetId>& map) : TParent(map) {}
+};
+
+/*!
+ * \brief Helper structure to register TargetId
+ * \sa TVM_REGISTER_TARGET_ID
+ */
+class TargetIdRegEntry {
+ public:
+  /*!
+   * \brief Register additional attributes to target_id.
+   * \param attr_name The name of the attribute.
+   * \param value The value to be set.
+   * \param plevel The priority level of this attribute,
+   *  an higher priority level attribute
+   *  will replace lower priority level attribute.
+   *  Must be bigger than 0.
+   *
+   *  Cannot set with same plevel twice in the code.
+   *
+   * \tparam ValueType The type of the value to be set.
+   */
+  template <typename ValueType>
+  inline TargetIdRegEntry& set_attr(const String& attr_name, const ValueType& value,
+                                    int plevel = 10);
+  /*!
+   * \brief Register a valid configuration option and its ValueType for validation
+   * \param key The configuration key
+   * \tparam ValueType The value type to be registered
+   */
+  template <typename ValueType>
+  inline TargetIdRegEntry& add_attr_option(const String& key);
+  /*! \brief Set name of the TargetId to be the same as registry if it is empty */
+  inline TargetIdRegEntry& set_name();
+  /*!
+   * \brief Register or get a new entry.
+   * \param target_id_name The name of the TargetId.
+   * \return the corresponding entry.
+   */
+  TVM_DLL static TargetIdRegEntry& RegisterOrGet(const String& target_id_name);
+
+ private:
+  TargetId id_;
+  String name;
+
+  /*! \brief private constructor */
+  explicit TargetIdRegEntry(uint32_t reg_index) : id_(make_object<TargetIdNode>()) {
+    id_->index_ = reg_index;
+  }
+  /*!
+   * \brief update the attribute TargetIdAttrMap
+   * \param key The name of the attribute
+   * \param value The value to be set
+   * \param plevel The priority level
+   */
+  TVM_DLL void UpdateAttr(const String& key, TVMRetValue value, int plevel);
+  template <typename, typename>
+  friend class AttrRegistry;
+  friend class TargetId;
+};
+
+#define TVM_TARGET_ID_REGISTER_VAR_DEF \
+  static DMLC_ATTRIBUTE_UNUSED ::tvm::TargetIdRegEntry& __make_##TargetId
+
+/*!
+ * \def TVM_REGISTER_TARGET_ID
+ * \brief Register a new target id, or set attribute of the corresponding target id.
+ *
+ * \param TargetIdName The name of target id
+ *
+ * \code
+ *
+ *  TVM_REGISTER_TARGET_ID("llvm")
+ *  .set_attr<TPreCodegenPass>("TPreCodegenPass", a-pre-codegen-pass)
+ *  .add_attr_option<Bool>("system_lib")
+ *  .add_attr_option<String>("mtriple")
+ *  .add_attr_option<String>("mattr");
+ *
+ * \endcode
+ */
+#define TVM_REGISTER_TARGET_ID(TargetIdName)                    \
+  TVM_STR_CONCAT(TVM_TARGET_ID_REGISTER_VAR_DEF, __COUNTER__) = \
+      ::tvm::TargetIdRegEntry::RegisterOrGet(TargetIdName).set_name()
+
+namespace detail {
+template <typename Type, template <typename...> class Container>
+struct is_specialized : std::false_type {
+  using type = std::false_type;
+};
+
+template <template <typename...> class Container, typename... Args>
+struct is_specialized<Container<Args...>, Container> : std::true_type {
+  using type = std::true_type;
+};
+
+template <typename ValueType, typename IsArray = typename is_specialized<ValueType, Array>::type,
+          typename IsMap = typename is_specialized<ValueType, Map>::type>
+struct ValueTypeInfoMaker {};
+
+template <typename ValueType>
+struct ValueTypeInfoMaker<ValueType, std::false_type, std::false_type> {
+  using ValueTypeInfo = TargetIdNode::ValueTypeInfo;
+
+  ValueTypeInfo operator()() const {
+    uint32_t tindex = ValueType::ContainerType::_GetOrAllocRuntimeTypeIndex();
+    ValueTypeInfo info;
+    info.type_index = tindex;
+    info.type_key = runtime::Object::TypeIndex2Key(tindex);
+    info.key = nullptr;
+    info.val = nullptr;
+    return info;
+  }
+};
+
+template <typename ValueType>
+struct ValueTypeInfoMaker<ValueType, std::true_type, std::false_type> {
+  using ValueTypeInfo = TargetIdNode::ValueTypeInfo;
+
+  ValueTypeInfo operator()() const {
+    using key_type = ValueTypeInfoMaker<typename ValueType::value_type>;
+    uint32_t tindex = ValueType::ContainerType::_GetOrAllocRuntimeTypeIndex();
+    ValueTypeInfo info;
+    info.type_index = tindex;
+    info.type_key = runtime::Object::TypeIndex2Key(tindex);
+    info.key = std::unique_ptr<ValueTypeInfo>(new ValueTypeInfo(key_type()()));
+    info.val = nullptr;
+    return info;
+  }
+};
+
+template <typename ValueType>
+struct ValueTypeInfoMaker<ValueType, std::false_type, std::true_type> {
+  using ValueTypeInfo = TargetIdNode::ValueTypeInfo;
+  ValueTypeInfo operator()() const {
+    using key_type = ValueTypeInfoMaker<typename ValueType::key_type>;
+    using val_type = ValueTypeInfoMaker<typename ValueType::mapped_type>;
+    uint32_t tindex = ValueType::ContainerType::_GetOrAllocRuntimeTypeIndex();
+    ValueTypeInfo info;
+    info.type_index = tindex;
+    info.type_key = runtime::Object::TypeIndex2Key(tindex);
+    info.key = std::unique_ptr<ValueTypeInfo>(new ValueTypeInfo(key_type()()));
+    info.val = std::unique_ptr<ValueTypeInfo>(new ValueTypeInfo(val_type()()));
+    return info;
+  }
+};
+
+}  // namespace detail
+
+template <typename ValueType>
+inline TargetIdAttrMap<ValueType> TargetId::GetAttrMap(const String& attr_name) {
+  return TargetIdAttrMap<ValueType>(GetAttrMapContainer(attr_name));
+}
+
+template <typename ValueType>
+inline TargetIdRegEntry& TargetIdRegEntry::set_attr(const String& attr_name, const ValueType& value,
+                                                    int plevel) {
+  CHECK_GT(plevel, 0) << "plevel in set_attr must be greater than 0";
+  runtime::TVMRetValue rv;
+  rv = value;
+  UpdateAttr(attr_name, rv, plevel);
+  return *this;
+}
+
+template <typename ValueType>
+inline TargetIdRegEntry& TargetIdRegEntry::add_attr_option(const String& key) {
+  CHECK(!id_->key2vtype_.count(key))
+      << "AttributeError: add_attr_option failed because '" << key << "' has been set once";
+  id_->key2vtype_[key] = detail::ValueTypeInfoMaker<ValueType>()();
+  return *this;
+}
+
+inline TargetIdRegEntry& TargetIdRegEntry::set_name() {
+  if (id_->name.empty()) {
+    id_->name = name;
+  }
+  return *this;
+}
+
+}  // namespace tvm
+
+#endif  // TVM_TARGET_TARGET_ID_H_
diff --git a/src/node/attr_registry.h b/src/node/attr_registry.h
index 9cc5b4d410a7..6f8b005f95bb 100644
--- a/src/node/attr_registry.h
+++ b/src/node/attr_registry.h
@@ -24,6 +24,7 @@
 #ifndef TVM_NODE_ATTR_REGISTRY_H_
 #define TVM_NODE_ATTR_REGISTRY_H_
 
+#include <tvm/node/attr_registry_map.h>
 #include <tvm/runtime/container.h>
 #include <tvm/runtime/packed_func.h>
 
diff --git a/src/target/target_id.cc b/src/target/target_id.cc
new file mode 100644
index 000000000000..dc1255c45faf
--- /dev/null
+++ b/src/target/target_id.cc
@@ -0,0 +1,164 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*!
+ * \file src/target/target_id.cc
+ * \brief Target id registry
+ */
+#include <tvm/target/target_id.h>
+
+#include "../node/attr_registry.h"
+#include "../runtime/object_internal.h"
+
+namespace tvm {
+
+using TargetIdRegistry = AttrRegistry<TargetIdRegEntry, TargetId>;
+
+TargetIdRegEntry& TargetIdRegEntry::RegisterOrGet(const String& target_id_name) {
+  return TargetIdRegistry::Global()->RegisterOrGet(target_id_name);
+}
+
+void TargetIdRegEntry::UpdateAttr(const String& key, TVMRetValue value, int plevel) {
+  TargetIdRegistry::Global()->UpdateAttr(key, id_, value, plevel);
+}
+
+const AttrRegistryMapContainerMap<TargetId>& TargetId::GetAttrMapContainer(
+    const String& attr_name) {
+  return TargetIdRegistry::Global()->GetAttrMap(attr_name);
+}
+
+const TargetId& TargetId::Get(const String& target_id_name) {
+  const TargetIdRegEntry* reg = TargetIdRegistry::Global()->Get(target_id_name);
+  CHECK(reg != nullptr) << "TargetId " << target_id_name << " is not registered";
+  return reg->id_;
+}
+
+void VerifyTypeInfo(const ObjectRef& obj, const TargetIdNode::ValueTypeInfo& info) {
+  CHECK(obj.defined()) << "Object is None";
+  if (!runtime::ObjectInternal::DerivedFrom(obj.get(), info.type_index)) {
+    LOG(FATAL) << "AttributeError: expect type " << info.type_key << " but get "
+               << obj->GetTypeKey();
+    throw;
+  }
+  if (info.type_index == ArrayNode::_type_index) {
+    int i = 0;
+    for (const auto& e : *obj.as<ArrayNode>()) {
+      try {
+        VerifyTypeInfo(e, *info.key);
+      } catch (const tvm::Error& e) {
+        LOG(FATAL) << "The i-th element of array failed type checking, where i = " << i
+                   << ", and the error is:\n"
+                   << e.what();
+        throw;
+      }
+      ++i;
+    }
+  } else if (info.type_index == MapNode::_type_index) {
+    for (const auto& kv : *obj.as<MapNode>()) {
+      try {
+        VerifyTypeInfo(kv.first, *info.key);
+      } catch (const tvm::Error& e) {
+        LOG(FATAL) << "The key of map failed type checking, where key = " << kv.first
+                   << ", value = " << kv.second << ", and the error is:\n"
+                   << e.what();
+        throw;
+      }
+      try {
+        VerifyTypeInfo(kv.second, *info.val);
+      } catch (const tvm::Error& e) {
+        LOG(FATAL) << "The value of map failed type checking, where key = " << kv.first
+                   << ", value = " << kv.second << ", and the error is:\n"
+                   << e.what();
+        throw;
+      }
+    }
+  }
+}
+
+void TargetIdNode::ValidateSchema(const Map<String, ObjectRef>& config) const {
+  const String kTargetId = "id";
+  for (const auto& kv : config) {
+    const String& name = kv.first;
+    const ObjectRef& obj = kv.second;
+    if (name == kTargetId) {
+      CHECK(obj->IsInstance<StringObj>())
+          << "AttributeError: \"id\" is not a string, but its type is " << obj->GetTypeKey();
+      CHECK(Downcast<String>(obj) == this->name)
+          << "AttributeError: \"id\" = " << obj << " is inconsistent with TargetId " << this->name;
+      continue;
+    }
+    auto it = key2vtype_.find(name);
+    if (it == key2vtype_.end()) {
+      std::ostringstream os;
+      os << "AttributeError: Invalid config option, cannot recognize \'" << name
+         << "\'. Candidates are:";
+      for (const auto& kv : key2vtype_) {
+        os << "\n  " << kv.first;
+      }
+      LOG(FATAL) << os.str();
+      throw;
+    }
+    const auto& info = it->second;
+    try {
+      VerifyTypeInfo(obj, info);
+    } catch (const tvm::Error& e) {
+      LOG(FATAL) << "AttributeError: Schema validation failed for TargetId " << this->name
+                 << ", details:\n"
+                 << e.what() << "\n"
+                 << "The config is:\n"
+                 << config;
+      throw;
+    }
+  }
+}
+
+inline String GetId(const Map<String, ObjectRef>& target, const char* name) {
+  const String kTargetId = "id";
+  CHECK(target.count(kTargetId)) << "AttributeError: \"id\" does not exist in " << name << "\n"
+                                 << name << " = " << target;
+  const ObjectRef& obj = target[kTargetId];
+  CHECK(obj->IsInstance<StringObj>()) << "AttributeError: \"id\" is not a string in " << name
+                                      << ", but its type is " << obj->GetTypeKey() << "\n"
+                                      << name << " = " << target;
+  return Downcast<String>(obj);
+}
+
+void TargetValidateSchema(const Map<String, ObjectRef>& config) {
+  try {
+    const String kTargetHost = "target_host";
+    Map<String, ObjectRef> target = config;
+    Map<String, ObjectRef> target_host;
+    String target_id = GetId(target, "target");
+    String target_host_id;
+    if (config.count(kTargetHost)) {
+      target.erase(kTargetHost);
+      target_host = Downcast<Map<String, ObjectRef>>(config[kTargetHost]);
+      target_host_id = GetId(target_host, "target_host");
+    }
+    TargetId::Get(target_id)->ValidateSchema(target);
+    if (!target_host.empty()) {
+      TargetId::Get(target_host_id)->ValidateSchema(target_host);
+    }
+  } catch (const tvm::Error& e) {
+    LOG(INFO) << e.what();
+    throw e;
+  }
+}
+
+}  // namespace tvm
diff --git a/tests/cpp/target_test.cc b/tests/cpp/target_test.cc
new file mode 100644
index 000000000000..6b8e0b159139
--- /dev/null
+++ b/tests/cpp/target_test.cc
@@ -0,0 +1,88 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+#include <dmlc/logging.h>
+#include <gtest/gtest.h>
+#include <tvm/target/target_id.h>
+
+#include <cmath>
+#include <string>
+
+using namespace tvm;
+
+TVM_REGISTER_TARGET_ID("TestTargetId")
+    .set_attr<std::string>("Attr1", "Value1")
+    .add_attr_option<Bool>("my_bool")
+    .add_attr_option<Array<String>>("your_names")
+    .add_attr_option<Map<String, Integer>>("her_maps");
+
+TEST(TargetId, GetAttrMap) {
+  auto map = tvm::TargetId::GetAttrMap<std::string>("Attr1");
+  auto target_id = tvm::TargetId::Get("TestTargetId");
+  std::string result = map[target_id];
+  CHECK_EQ(result, "Value1");
+}
+
+TEST(TargetId, SchemaValidation) {
+  tvm::Map<String, ObjectRef> target;
+  {
+    tvm::Array<String> your_names{"junru", "jian"};
+    tvm::Map<String, Integer> her_maps{
+        {"a", 1},
+        {"b", 2},
+    };
+    target.Set("my_bool", Bool(true));
+    target.Set("your_names", your_names);
+    target.Set("her_maps", her_maps);
+    target.Set("id", String("TestTargetId"));
+  }
+  TargetValidateSchema(target);
+  tvm::Map<String, ObjectRef> target_host(target.begin(), target.end());
+  target.Set("target_host", target_host);
+  TargetValidateSchema(target);
+}
+
+TEST(TargetId, SchemaValidationFail) {
+  tvm::Map<String, ObjectRef> target;
+  {
+    tvm::Array<String> your_names{"junru", "jian"};
+    tvm::Map<String, Integer> her_maps{
+        {"a", 1},
+        {"b", 2},
+    };
+    target.Set("my_bool", Bool(true));
+    target.Set("your_names", your_names);
+    target.Set("her_maps", her_maps);
+    target.Set("ok", ObjectRef(nullptr));
+    target.Set("id", String("TestTargetId"));
+  }
+  bool failed = false;
+  try {
+    TargetValidateSchema(target);
+  } catch (...) {
+    failed = true;
+  }
+  ASSERT_EQ(failed, true);
+}
+
+int main(int argc, char** argv) {
+  testing::InitGoogleTest(&argc, argv);
+  testing::FLAGS_gtest_death_test_style = "threadsafe";
+  return RUN_ALL_TESTS();
+}