Skip to content

GitLab

Commit 5755f841 authored by Ville Pietilä's avatar Ville Pietilä
Browse files

Optimize memory pool insertion.

parent c6ba9545
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
include/ck/utility/host_memory_allocator.hpp
View file @ 5755f841
......@@ -128,7 +128,9 @@ namespace memory {
std::cout << "[ DynamicMemPool ] Creating new pool queue for size " << sizeInBytes << std::endl;
}
#endif
memory_pool_[sizeInBytes].push(p);
std::queue<void*> q;
q.push(p);
memory_pool_.insert({sizeInBytes, std::move(q)});
memPoolSizeInBytes_ += sizeInBytes;
}
#ifdef ENABLE_MEM_POOL_LOGGING
......@@ -230,7 +232,7 @@ namespace memory {
else {
std::queue<void*> q;
q.push(p);
memory_pool_[sizeInBytes] = std::move(q);
memory_pool_.insert({sizeInBytes, std::move(q)});
#ifdef ENABLE_MEM_POOL_LOGGING
if (enableLogging_)
{
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment