fix segment fault bug of allocator with cuda graph (#73103)

zyfncg · web-flow · commit 44bd40ecec63 · 2025-06-05T10:37:08.000+08:00
diff --git a/paddle/phi/core/memory/allocation/auto_growth_best_fit_allocator.cc b/paddle/phi/core/memory/allocation/auto_growth_best_fit_allocator.cc
@@ -200,7 +200,8 @@ void AutoGrowthBestFitAllocator::FreeImpl(phi::Allocation *allocation) {
   auto next_it = block_it;
   ++next_it;
 
-  if (next_it != blocks.end() && next_it->is_free_) {
+  // It's weird that using `next_it == blocks.end()` will cause a judgment fail.
+  if (block_it != (--blocks.end()) && next_it->is_free_) {
     free_blocks_.erase(std::make_pair(next_it->size_, next_it->ptr_));
     block_it->size_ += next_it->size_;
     blocks.erase(next_it);
