Forward allocation names in the NamedAllocationStrategy

examples/cookbook/recipe_shared_memory.cpp

@@ -16,6 +16,7 @@
 #include "umpire/Umpire.hpp"
 #include "umpire/config.hpp"
 #include "umpire/resource/HostSharedMemoryResource.hpp"
+#include "umpire/strategy/NamedAllocationStrategy.hpp"
 #include "umpire/util/MemoryResourceTraits.hpp"
 
 //
@@ -58,6 +59,9 @@ int main(int ac, char** av)
   //
   auto node_allocator{rm.makeResource("SHARED::node_allocator", traits)};
 
+  auto named_node_allocator{
+      rm.makeAllocator<umpire::strategy::NamedAllocationStrategy>("My Node Allocator", node_allocator)};
+
   //
   // Resource of this allocator is SHARED
   //
@@ -82,6 +86,7 @@ int main(int ac, char** av)
   // Allocate shared memory
   //
   void* ptr{node_allocator.allocate("allocation_name_2", sizeof(uint64_t))};
+  void* ptr2{named_node_allocator.allocate("allocation two", 1024)};
   uint64_t* data{static_cast<uint64_t*>(ptr)};
 
   if (shared_rank == foreman_rank)
@@ -98,6 +103,7 @@ int main(int ac, char** av)
   UMPIRE_ASSERT(*data == 0xDEADBEEF);
 
   node_allocator.deallocate(ptr);
+  named_node_allocator.deallocate(ptr2);
 
   if (use_mpi) {
     MPI_Finalize();

src/umpire/strategy/NamedAllocationStrategy.cpp

@@ -22,6 +22,11 @@ void* NamedAllocationStrategy::allocate(std::size_t bytes)
   return m_allocator->allocate_internal(bytes);
 }
 
+void* NamedAllocationStrategy::allocate_named(const std::string& name, std::size_t bytes)
+{
+  return m_allocator->allocate_named_internal(name, bytes);
+}
+
 void NamedAllocationStrategy::deallocate(void* ptr, std::size_t size)
 {
   return m_allocator->deallocate_internal(ptr, size);

src/umpire/strategy/NamedAllocationStrategy.hpp

@@ -18,6 +18,7 @@ class NamedAllocationStrategy : public AllocationStrategy {
   NamedAllocationStrategy(const std::string& name, int id, Allocator allocator);
 
   void* allocate(std::size_t bytes) override;
+  void* allocate_named(const std::string& name, std::size_t bytes) override;
   void deallocate(void* ptr, std::size_t size) override;
 
   Platform getPlatform() noexcept override;

tests/integration/strategy_tests.cpp

@@ -763,4 +763,22 @@ TEST(NamingShimTests, TestAllocateDeallocate)
     shim.deallocate(ptr);
   }
 }
+
+TEST(NamedAllocatorTest, ForwardName)
+{
+  auto& rm = umpire::ResourceManager::getInstance();
+
+  auto traits{umpire::get_default_resource_traits("SHARED")};
+  traits.size = 1 * 1024 * 1024;
+  traits.scope = umpire::MemoryResourceTraits::shared_scope::node;
+
+  auto node_allocator{rm.makeResource("SHARED::allocator_for_named_test", traits)};
+
+  auto allocator{rm.makeAllocator<umpire::strategy::NamedAllocationStrategy>("shared named alloc", node_allocator)};
+  {
+    void* ptr = allocator.allocate("test", 1024);
+    EXPECT_NE(ptr, nullptr);
+    allocator.deallocate(ptr);
+  }
+}
 #endif