addressed upstream comments-3

Author: d-smirnov

include/tvm/ir/memory_pools.h

@@ -258,7 +258,7 @@ struct ConstantInfoNode : public Object {
     hash_reduce(data);
   }
 
-  static constexpr const char* _type_key = "tir.usmp.ConstantInfo";
+  static constexpr const char* _type_key = "ir.ConstantInfo";
   static constexpr bool _type_has_method_sequal_reduce = true;
   static constexpr bool _type_has_method_shash_reduce = true;
   TVM_DECLARE_FINAL_OBJECT_INFO(ConstantInfoNode, Object);

python/tvm/ir/memory_pools.py

@@ -74,15 +74,6 @@ class PoolInfo(Object):
 
     """
 
-    # The string parameter to indicate read and write access to a pool
-    # This needs to be kept in sync with kTargetPoolReadWriteAccess in
-    # include/tvm/ir/memory_pools.h
-    READ_WRITE_ACCESS = "rw"
-    # The string parameter to indicate read only access to a pool
-    # This needs to be kept in sync with kTargetPoolReadOnlyAccess in
-    # include/tvm/ir/memory_pools.h
-    READ_ONLY_ACCESS = "ro"
-
     def __init__(
         self,
         pool_name: str,
@@ -98,18 +89,15 @@ def __init__(
         if not target_burst_bytes:
             target_burst_bytes = dict()
 
-        self.__init_handle_by_constructor__(
-            _ffi_api.PoolInfo,  # type: ignore # pylint: disable=no-member
-            pool_name,
-            target_access,
-            size_hint_bytes,
-            clock_frequency_hz,
-            read_bandwidth_bytes_per_cycle,
-            write_bandwidth_bytes_per_cycle,
-            read_latency_cycles,
-            write_latency_cycles,
-            target_burst_bytes,
-        )
+        self.pool_name = pool_name
+        self.target_access = target_access
+        self.size_hint_bytes = size_hint_bytes
+        self.clock_frequency_hz = clock_frequency_hz
+        self.read_bandwidth_bytes_per_cycle = read_bandwidth_bytes_per_cycle
+        self.write_bandwidth_bytes_per_cycle = write_bandwidth_bytes_per_cycle
+        self.read_latency_cycles = read_latency_cycles
+        self.write_latency_cycles = write_latency_cycles
+        self.target_burst_bytes = target_burst_bytes
 
 
 @register_object("ir.PoolInfoProperties")
@@ -196,13 +184,13 @@ class WorkspacePoolInfo(PoolInfo):
         The properties of the pool.
     """
 
-    # pylint: disable=W0231
     def __init__(
         self,
         pool_name: str,
         targets,
         pool_info_properties=None,
     ):
+        super().__init__(pool_name, targets)
         if pool_info_properties is None:
             pool_info_properties = PoolInfoProperties()
 
@@ -231,14 +219,14 @@ class ConstantPoolInfo(PoolInfo):
         The properties of the pool.
     """
 
-    # pylint: disable=W0231
     def __init__(
         self,
         pool_name: str,
         targets,  # list[Target]
         constant_info_arr=None,  # list[ConstantInfo]
         pool_info_properties=None,
     ):
+        super().__init__(pool_name, targets)
         if constant_info_arr is None:
             constant_info_arr = []
         if pool_info_properties is None:

src/ir/memory_pools.cc

@@ -148,7 +148,7 @@ ConstantInfo::ConstantInfo(String name_hint, Integer byte_offset, runtime::NDArr
 }
 
 TVM_REGISTER_NODE_TYPE(ConstantInfoNode);
-TVM_REGISTER_GLOBAL("tir.usmp.ConstantInfo")
+TVM_REGISTER_GLOBAL("ir.ConstantInfo")
     .set_body_typed([](String name_hint, Integer byte_offset, runtime::NDArray data) {
       return ConstantInfo(name_hint, byte_offset, data);
     });