#1376: gossiplb: fix for serial load balancing tests

DARMA-tasking · May 3, 2021 · 2e5a7be · 2e5a7be
1 parent b80f04f
commit 2e5a7be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/vt/vrt/collection/balance/gossiplb/gossiplb.cc b/src/vt/vrt/collection/balance/gossiplb/gossiplb.cc
@@ -136,9 +136,9 @@ void GossipLB::inputParams(balance::SpecEntry* spec) {
       f_ = static_cast<uint16_t>(std::ceil(std::pow(num_nodes, 1.0/k_max_)));
     } else {
       // set both the fanout and the rounds
-      k_max_ = static_cast<uint16_t>(
+      k_max_ = static_cast<uint16_t>(std::max(1.0,
         std::round(std::sqrt(std::log(num_nodes)/std::log(2.0)))
-      );
+      ));
       f_ = static_cast<uint16_t>(std::ceil(std::pow(num_nodes, 1.0/k_max_)));
     }
   } else if (knowledge_ == KnowledgeEnum::Complete) {