VITIS-12792 HIP Binding: Memory pool APIs

src/runtime_src/core/common/api/handle.h

@@ -40,6 +40,12 @@ class handle_map<HandleType, std::shared_ptr<ImplType>>
   }
 
 public:
+  const std::map<HandleType, std::shared_ptr<ImplType>>&
+  get_map() const
+  {
+    return handles;
+  }
+
   const std::shared_ptr<ImplType>&
   get_or_error(HandleType handle) const
   {
@@ -105,6 +111,12 @@ class handle_map<HandleType, std::unique_ptr<ImplType>>
   std::map<HandleType, std::unique_ptr<ImplType>> handles;
 
 public:
+  const std::map<HandleType, std::unique_ptr<ImplType>>&
+  get_map() const
+  {
+    return handles;
+  }
+
   ImplType*
   get_or_error(HandleType handle) const
   {
@@ -153,6 +165,13 @@ class handle_map<HandleType, std::unique_ptr<ImplType>>
     std::lock_guard<std::mutex> lk(mutex);
     return handles.count(handle);
   }
+
+  size_t
+  size() const
+  {
+    std::lock_guard<std::mutex> lk(mutex);
+    return handles.size();
+  }
 };
 
 } // xrt_core

src/runtime_src/hip/api/hip_device.cpp

@@ -5,6 +5,7 @@
 
 #include "hip/core/common.h"
 #include "hip/core/device.h"
+#include "hip/core/memory_pool.h"
 
 #include <cstring>
 #include <mutex>
@@ -46,8 +47,12 @@ device_init()
   for (uint32_t i = 0; i < dev_count; i++) {
     if (device_cache.count(static_cast<device_handle>(i)) > 0)
       continue;
-    auto dev = std::make_shared<xrt::core::hip::device>(i);
+    auto dev = std::make_unique<xrt::core::hip::device>(i);
     device_cache.add(i, std::move(dev));
+    auto default_mem_pool = std::make_shared<xrt::core::hip::memory_pool>(device_cache.get_or_error(i), MAX_MEMORY_POOL_SIZE_NPU, MEMORY_POOL_BLOCK_SIZE_NPU);
+    memory_pool_db[i].push_front(default_mem_pool);
+    current_memory_pool_db[i] = default_mem_pool;
+    insert_in_map(mem_pool_cache, default_mem_pool);
   }
   // make first device as default device
   if (dev_count > 0)