#2201: fixed the incorrect transfer type causing build error

DARMA-tasking · Nov 28, 2023 · a7376ab · a7376ab
1 parent b6e6bca
commit a7376ab
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc b/src/vt/vrt/collection/balance/temperedlb/temperedlb.cc
@@ -560,6 +560,7 @@ void TemperedLB::doLBStages(LoadType start_imb) {
         is_underloaded_ = true;
       }
 
+      // Perform requested type of information stage
       switch (inform_type_) {
       case InformTypeEnum::SyncInform:
         informSync();
@@ -571,7 +572,20 @@ void TemperedLB::doLBStages(LoadType start_imb) {
         vtAbort("TemperedLB:: Unsupported inform type");
       }
 
-      decide();
+      // Execute transfer stage
+      switch (transfer_type_) {
+      case TransferTypeEnum::Original:
+        originalTransfer();
+        break;
+      case TransferTypeEnum::Recursive:
+	vtAbort("TemperedLB:: Unimplemented transfer type: Recursive");
+        break;
+      case TransferTypeEnum::SwapClusters:
+	vtAbort("TemperedLB:: Unimplemented transfer type: SwapClusters");
+        break;
+      default:
+        vtAbort("TemperedLB:: Unsupported transfer type");
+      }
 
       vt_debug_print(
         verbose, temperedlb,
@@ -1219,7 +1233,7 @@ std::vector<TemperedLB::ObjIDType> TemperedLB::orderObjects(
   return ordered_obj_ids;
 }
 
-void TemperedLB::decide() {
+void TemperedLB::originalTransfer() {
   auto lazy_epoch = theTerm()->makeEpochCollective("TemperedLB: decide");
 
   // Initialize transfer and rejection counters

diff --git a/src/vt/vrt/collection/balance/temperedlb/temperedlb.h b/src/vt/vrt/collection/balance/temperedlb/temperedlb.h
@@ -91,7 +91,7 @@ struct TemperedLB : BaseLB {
   void doLBStages(LoadType start_imb);
   void informAsync();
   void informSync();
-  void decide();
+  void originalTransfer();
   void migrate();
 
   void propagateRound(uint8_t k_cur_async, bool sync, EpochType epoch = no_epoch);