fix core: add spin lock when auto_growth_allocator release memory (#68631)

joseflv · web-flow · commit 899ace552eba · 2024-10-12T10:58:56.000+08:00
diff --git a/paddle/phi/core/memory/allocation/auto_growth_best_fit_allocator.h b/paddle/phi/core/memory/allocation/auto_growth_best_fit_allocator.h
@@ -46,6 +46,7 @@ class AutoGrowthBestFitAllocator : public Allocator {
 
   // Release the memory block which is not used in pool.
   uint64_t ReleaseImpl(const phi::Place &place) override {
+    std::lock_guard<SpinLock> guard(spinlock_);
     return FreeIdleChunks();
   }
 

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,7 @@ class AutoGrowthBestFitAllocator : public Allocator {`
`46`	`46`
`47`	`47`	`// Release the memory block which is not used in pool.`
`48`	`48`	`uint64_t ReleaseImpl(const phi::Place &place) override {`
	`49`	`+ std::lock_guard<SpinLock> guard(spinlock_);`
`49`	`50`	`return FreeIdleChunks();`
`50`	`51`	`}`
`51`	`52`