#1659: lb: clear custom model when TemperedWMin is destroyed

DARMA-tasking · Dec 13, 2022 · 0570bd0 · 0570bd0
1 parent ffcb97b
commit 0570bd0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.cc b/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.cc
@@ -50,6 +50,10 @@
 
 namespace vt { namespace vrt { namespace collection { namespace lb {
 
+TemperedWMin::~TemperedWMin() {
+  theLBManager()->setCustomLoadModel(nullptr);
+}
+
 void TemperedWMin::init(objgroup::proxy::Proxy<TemperedWMin> in_proxy) {
   auto proxy_bits = in_proxy.getProxy();
   auto proxy = objgroup::proxy::Proxy<TemperedLB>(proxy_bits);

diff --git a/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.h b/src/vt/vrt/collection/balance/temperedwmin/temperedwmin.h
@@ -55,7 +55,7 @@ struct TemperedWMin : TemperedLB {
   TemperedWMin() { comm_aware_ = true; }
   TemperedWMin(TemperedWMin const&) = delete;
 
-  virtual ~TemperedWMin() { }
+  virtual ~TemperedWMin();
 
 public:
   void init(objgroup::proxy::Proxy<TemperedWMin> in_proxy);