[IR] Fix ir StorageManager bug (#54503)

* fix ir storage bug

* refine code

* refine code

* fix bug

* refine code

* refine code

Author: zhangbo9674

paddle/ir/core/enforce.h

@@ -25,8 +25,10 @@
 // there is no equivalent intrinsics in msvc.
 #define UNLIKELY(condition) (condition)
 #endif
-
-inline bool is_error(bool stat) { return !stat; }
+template <typename T>
+inline bool is_error(const T& stat) {
+  return !stat;
+}
 
 namespace ir {
 class IrNotMetException : public std::exception {
@@ -55,7 +57,7 @@ class IrNotMetException : public std::exception {
 
 #define IR_ENFORCE(COND, ...)                                               \
   do {                                                                      \
-    auto __cond__ = (COND);                                                 \
+    bool __cond__(COND);                                                    \
     if (UNLIKELY(is_error(__cond__))) {                                     \
       try {                                                                 \
         throw ir::IrNotMetException(                                        \

paddle/ir/core/storage_manager.cc

@@ -25,19 +25,20 @@ namespace ir {
 struct ParametricStorageManager {
   using StorageBase = StorageManager::StorageBase;
 
-  ParametricStorageManager() {}
+  explicit ParametricStorageManager(std::function<void(StorageBase *)> destroy)
+      : destroy_(destroy) {}
 
   ~ParametricStorageManager() {
     for (const auto &instance : parametric_instances_) {
-      delete instance.second;
+      destroy_(instance.second);
     }
     parametric_instances_.clear();
   }
 
   // Get the storage of parametric type, if not in the cache, create and
   // insert the cache.
   StorageBase *GetOrCreate(std::size_t hash_value,
-                           std::function<bool(const StorageBase *)> equal_func,
+                           std::function<bool(StorageBase *)> equal_func,
                            std::function<StorageBase *()> constructor) {
     if (parametric_instances_.count(hash_value) != 0) {
       auto pr = parametric_instances_.equal_range(hash_value);
@@ -62,6 +63,7 @@ struct ParametricStorageManager {
   // In order to prevent hash conflicts, the unordered_multimap data structure
   // is used for storage.
   std::unordered_multimap<size_t, StorageBase *> parametric_instances_;
+  std::function<void(StorageBase *)> destroy_;
 };
 
 StorageManager::StorageManager() {}
@@ -95,12 +97,13 @@ StorageManager::StorageBase *StorageManager::GetParameterlessStorageImpl(
   return parameterless_instance;
 }
 
-void StorageManager::RegisterParametricStorageImpl(TypeId type_id) {
+void StorageManager::RegisterParametricStorageImpl(
+    TypeId type_id, std::function<void(StorageBase *)> destroy) {
   std::lock_guard<ir::SpinLock> guard(parametric_instance_lock_);
   VLOG(4) << "Register a parametric storage of: [TypeId_hash="
           << std::hash<ir::TypeId>()(type_id) << "].";
-  parametric_instance_.emplace(type_id,
-                               std::make_unique<ParametricStorageManager>());
+  parametric_instance_.emplace(
+      type_id, std::make_unique<ParametricStorageManager>(destroy));
 }
 
 void StorageManager::RegisterParameterlessStorageImpl(

paddle/ir/core/storage_manager.h

@@ -100,7 +100,9 @@ class StorageManager {
   ///
   template <typename Storage>
   void RegisterParametricStorage(TypeId type_id) {
-    return RegisterParametricStorageImpl(type_id);
+    return RegisterParametricStorageImpl(type_id, [](StorageBase *storage) {
+      delete static_cast<Storage *>(storage);
+    });
   }
 
   ///
@@ -129,7 +131,8 @@ class StorageManager {
 
   StorageBase *GetParameterlessStorageImpl(TypeId type_id);
 
-  void RegisterParametricStorageImpl(TypeId type_id);
+  void RegisterParametricStorageImpl(
+      TypeId type_id, std::function<void(StorageBase *)> destroy);
 
   void RegisterParameterlessStorageImpl(
       TypeId type_id, std::function<StorageBase *()> constructor);