#2209: location: properly erase from map when destroyed

DARMA-tasking · Apr 29, 2024 · 6aaccdd · 6aaccdd
1 parent 9542a42
commit 6aaccdd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/vt/topos/location/manager.impl.h b/src/vt/topos/location/manager.impl.h
@@ -79,8 +79,8 @@ LocationManager::makeCollectionLM(VirtualProxyType proxy) {
 
 template <typename IndexT>
 void LocationManager::destroyCollectionLM(VirtualProxyType proxy) {
-  if (auto it = collection_lms.find(proxy); it != collection_lms.end()) {
-    objgroup::proxy::Proxy<IndexedElmType<IndexT>> lm_proxy(it->second);
+  if (auto elm = collection_lms.extract(proxy); elm) {
+    objgroup::proxy::Proxy<IndexedElmType<IndexT>> lm_proxy(elm.mapped());
     lm_proxy.destroyCollective();
   } else {
     vtAbort("Could not find location manager for proxy");