Segfault in CachingDeviceAllocator when out of memory about cub HOT 4 CLOSED

@orjgre the fix is merged so it targets our next release: 2.2.0.

from cub.

Hello @orjgre and thank you for reporting this!

The issue happens when the allocator is trying to clean up some cached blocks to fit the new allocation. In this case, we are trying to increment an iterator after erasing it, which is not right:

cached_blocks.erase(block_itr);
block_itr++;

It should've been:

block_itr = cached_blocks.erase(block_itr);

I'll create a PR soon.

from cub.

@senior-zero Thank you for fixing this. Do you know which release this will be included in?

from cub.

@orjgre we are migrating to a different repo. The fix is merged in NVIDIA/cccl#119. I'm closing the issue.

from cub.