diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.h b/src/vt/vrt/collection/balance/temperedlb/temperedlb.h index 5af5b501fb..117f30da7a 100644 --- a/src/vt/vrt/collection/balance/temperedlb/temperedlb.h +++ b/src/vt/vrt/collection/balance/temperedlb/temperedlb.h @@ -111,7 +111,7 @@ struct TemperedLB : BaseLB { void propagateRound(uint8_t k_cur_async, bool sync, EpochType epoch = no_epoch); void propagateIncomingAsync(LoadMsgAsync* msg); void propagateIncomingSync(LoadMsgSync* msg); - bool isUnderloaded(LoadType load) const; + virtual bool isUnderloaded(LoadType load) const; bool isUnderloadedRelaxed(LoadType over, LoadType under) const; bool isOverloaded(LoadType load) const; diff --git a/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.h b/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.h index 3630d1cc0d..d82bff8c31 100644 --- a/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.h +++ b/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.h @@ -74,6 +74,10 @@ struct TemperedWMin : TemperedLB { * All ranks are allowed to initiate the information propagation stage */ bool canPropagate() const override { return true; } + /** + * TemperedWMin does not care about underloaded + */ + bool isUnderloaded(LoadType load) const override { return true; } std::vector getPotentialRecipients() const override;