Get the Total Amount of Umpire Memory Allocated

examples/size_limiter.cpp

@@ -7,6 +7,7 @@
 #include <iostream>
 
 #include "umpire/ResourceManager.hpp"
+#include "umpire/Umpire.hpp"
 #include "umpire/strategy/QuickPool.hpp"
 #include "umpire/strategy/SizeLimiter.hpp"
 #include "umpire/util/Macros.hpp"
@@ -18,15 +19,30 @@ int main(int, char**)
       rm.makeAllocator<umpire::strategy::SizeLimiter>("size_limited_alloc", rm.getAllocator("HOST"), 1024);
 
   auto pool = rm.makeAllocator<umpire::strategy::QuickPool>("pool", size_limited_alloc, 64, 64);
+  void* data;
 
   // This will throw an exception because the pool is limited to 1024 bytes.
   std::cout << "Attempting to allocate 2098 bytes..." << std::endl;
   try {
-    void* data = pool.allocate(2048);
+    data = pool.allocate(2048);
     UMPIRE_USE_VAR(data);
   } catch (...) {
     std::cout << "Exception caught! Pool is limited to 1024 bytes." << std::endl;
   }
+  std::cout << "The total amount of memory used was: " << umpire::get_total_memory_usage() << std::endl;
+
+  std::cout << "Attempting to allocate 512 bytes..." << std::endl;
+  try {
+    data = pool.allocate(512);
+    UMPIRE_USE_VAR(data);
+  } catch (...) {
+    std::cout << "Exception caught! Pool is limited to 1024 bytes." << std::endl;
+  }
+  std::cout << "The total amount of memory used was: " << umpire::get_total_memory_usage() << std::endl;
+
+  if (data != nullptr) {
+    pool.deallocate(data);
+  }
 
   return 0;
 }

src/umpire/Umpire.cpp

@@ -173,6 +173,19 @@ void mark_event(const std::string& event)
       [&](auto& e) { e.name("event").category(event::category::metadata).arg("name", event).tag("replay", "true"); });
 }
 
+std::size_t get_total_memory_usage()
+{
+  auto& rm = umpire::ResourceManager::getInstance();
+  std::size_t total_memory{0};
+
+  for (auto s : rm.getResourceNames()) {
+    umpire::Allocator alloc = rm.getAllocator(s);
+    total_memory += alloc.getActualSize();
+  }
+
+  return total_memory;
+}
+
 std::size_t get_device_memory_usage(int device_id)
 {
 #if defined(UMPIRE_ENABLE_CUDA)

src/umpire/Umpire.hpp

@@ -162,6 +162,11 @@ std::size_t get_process_memory_usage_hwm();
  */
 void mark_event(const std::string& event);
 
+/*!
+ * \brief Get the total umpire memory usage in bytes across all memory resources
+ */
+std::size_t get_total_memory_usage();
+
 /*!
  * \brief Get memory usage of device device_id, using appropriate underlying
  * vendor API.