adt/closure back to vm.adt/vm.closure

apache · Jan 16, 2020 · 040a532 · 040a532
1 parent a00a3fc
commit 040a532
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 9 deletions.
diff --git a/include/tvm/relay/interpreter.h b/include/tvm/relay/interpreter.h
@@ -25,7 +25,7 @@
  * Given a Relay module, and a Relay expression it produces a value.
  *
  * The interpreter's values are a naive representation of the values that
- * can be produced by a Relay program and are exposed via tvm's object
+ * can be produced by a Relay program and are exposed via TVM's object
  * protocol to Python for introspection and debugging.
  *
  * The interpreter's intent is to serve as a reference semantics for the Relay IR,

diff --git a/include/tvm/runtime/container.h b/include/tvm/runtime/container.h
@@ -169,8 +169,8 @@ class ADTObj : public Object, public InplaceArrayBase<ADTObj, ObjectRef> {
   uint32_t size;
   // The fields of the structure follows directly in memory.
 
-  static constexpr const uint32_t _type_index = TypeIndex::kADT;
-  static constexpr const char* _type_key = "ADT";
+  static constexpr const uint32_t _type_index = TypeIndex::kVMADT;
+  static constexpr const char* _type_key = "vm.ADT";
   TVM_DECLARE_FINAL_OBJECT_INFO(ADTObj, Object);
 
  private:

diff --git a/include/tvm/runtime/object.h b/include/tvm/runtime/object.h
@@ -51,8 +51,8 @@ namespace runtime {
 enum TypeIndex  {
   /*! \brief Root object type. */
   kRoot = 0,
-  kClosure = 1,
-  kADT = 2,
+  kVMClosure = 1,
+  kVMADT = 2,
   kRuntimeModule = 3,
   kStaticIndexEnd,
   /*! \brief Type index is allocated during runtime. */

diff --git a/include/tvm/runtime/vm.h b/include/tvm/runtime/vm.h
@@ -53,8 +53,8 @@ class ClosureObj : public Object {
   /*! \brief The free variables of the closure. */
   std::vector<ObjectRef> free_vars;
 
-  static constexpr const uint32_t _type_index = TypeIndex::kClosure;
-  static constexpr const char* _type_key = "Closure";
+  static constexpr const uint32_t _type_index = TypeIndex::kVMClosure;
+  static constexpr const char* _type_key = "vm.Closure";
   TVM_DECLARE_FINAL_OBJECT_INFO(ClosureObj, Object);
 };
 

diff --git a/python/tvm/container.py b/python/tvm/container.py
@@ -118,7 +118,7 @@ class LoweredFunc(Object):
     DeviceFunc = 2
 
 
-@register_object
+@register_object("vm.ADT")
 class ADT(Object):
     """Algebatic data type(ADT) object.
 

diff --git a/src/runtime/container.cc b/src/runtime/container.cc
@@ -19,7 +19,7 @@
 
 /*!
  * \file src/runtime/container.cc
- * \brief Implementation of common POD containers.
+ * \brief Implementations of common plain old data (POD) containers.
  */
 #include <tvm/runtime/container.h>
 #include <tvm/runtime/memory.h>