#1830: TemperedWMin: exclude current node from potential recipients

DARMA-tasking · Oct 20, 2022 · db8b936 · db8b936
1 parent 231eaf7
commit db8b936
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.cc b/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.cc
@@ -104,13 +104,20 @@ void TemperedWMin::inputParams(balance::ConfigEntry* config) {
 }
 
 std::vector<NodeType> TemperedWMin::getPotentialRecipients() const {
+  auto const this_node = theContext()->getNode();
   std::vector<NodeType> nodes = {};
+
   for (auto&& elm : load_info_) {
-    nodes.push_back(elm.first);
+    auto const node = elm.first;
+    if (node != this_node) {
+      nodes.push_back(node);
+    }
   }
+
   if (isDeterministic()) {
     std::sort(nodes.begin(), nodes.end());
   }
+
   return nodes;
 }