[CustomDevice] Fix error that query a destroyed event (#56745)

PaddlePaddle · Aug 30, 2023 · c5786be · c5786be
1 parent 31a9688
commit c5786be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/memory/allocation/stream_safe_custom_device_allocator.cc b/paddle/fluid/memory/allocation/stream_safe_custom_device_allocator.cc
@@ -88,8 +88,7 @@ bool StreamSafeCustomDeviceAllocation::CanBeFreed() {
   }
   std::call_once(once_flag_, [this] { phi::DeviceManager::SetDevice(place_); });
   for (auto it = outstanding_event_map_.begin();
-       it != outstanding_event_map_.end();
-       ++it) {
+       it != outstanding_event_map_.end();) {
     auto& event = it->second;
     if (!event->Query()) {
       VLOG(9) << "Event " << event->raw_event() << " for " << ptr()
@@ -98,6 +97,7 @@ bool StreamSafeCustomDeviceAllocation::CanBeFreed() {
     }
     VLOG(8) << "Destroy event " << event->raw_event();
     event->Destroy();
+    it = outstanding_event_map_.erase(it);
   }
   outstanding_event_map_.clear();
   return true;