From 81c6746cc6d0540e0f40909fac291e4316a0dc98 Mon Sep 17 00:00:00 2001 From: Tom Benson Date: Tue, 17 Oct 2023 17:54:23 -0400 Subject: [PATCH] Avoid a segmentation fault when clearing cached blocks (#297) --- hipcub/include/hipcub/backend/rocprim/util_allocator.hpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/hipcub/include/hipcub/backend/rocprim/util_allocator.hpp b/hipcub/include/hipcub/backend/rocprim/util_allocator.hpp index 8fb13620..31ed0ae5 100644 --- a/hipcub/include/hipcub/backend/rocprim/util_allocator.hpp +++ b/hipcub/include/hipcub/backend/rocprim/util_allocator.hpp @@ -417,9 +417,7 @@ struct CachingDeviceAllocator if (debug) _HipcubLog("\tDevice %d freed %lld bytes.\n\t\t %lld available blocks cached (%lld bytes), %lld live blocks (%lld bytes) outstanding.\n", device, (long long) block_itr->bytes, (long long) cached_blocks.size(), (long long) cached_bytes[device].free, (long long) live_blocks.size(), (long long) cached_bytes[device].live); - cached_blocks.erase(block_itr); - - block_itr++; + block_itr = cached_blocks.erase(block_itr); } // Unlock