Skip to content

Commit e99614d

Browse files
committed
#2201: fixed the incorrect transfer type causing build error
1 parent 7c09e94 commit e99614d

File tree

2 files changed

+17
-3
lines changed

2 files changed

+17
-3
lines changed

src/vt/vrt/collection/balance/temperedlb/temperedlb.cc

+16-2
Original file line numberDiff line numberDiff line change
@@ -560,6 +560,7 @@ void TemperedLB::doLBStages(LoadType start_imb) {
560560
is_underloaded_ = true;
561561
}
562562

563+
// Perform requested type of information stage
563564
switch (inform_type_) {
564565
case InformTypeEnum::SyncInform:
565566
informSync();
@@ -571,7 +572,20 @@ void TemperedLB::doLBStages(LoadType start_imb) {
571572
vtAbort("TemperedLB:: Unsupported inform type");
572573
}
573574

574-
decide();
575+
// Execute transfer stage
576+
switch (transfer_type_) {
577+
case TransferTypeEnum::Original:
578+
originalTransfer();
579+
break;
580+
case TransferTypeEnum::Recursive:
581+
vtAbort("TemperedLB:: Unimplemented transfer type: Recursive");
582+
break;
583+
case TransferTypeEnum::SwapClusters:
584+
vtAbort("TemperedLB:: Unimplemented transfer type: SwapClusters");
585+
break;
586+
default:
587+
vtAbort("TemperedLB:: Unsupported transfer type");
588+
}
575589

576590
vt_debug_print(
577591
verbose, temperedlb,
@@ -1219,7 +1233,7 @@ std::vector<TemperedLB::ObjIDType> TemperedLB::orderObjects(
12191233
return ordered_obj_ids;
12201234
}
12211235

1222-
void TemperedLB::decide() {
1236+
void TemperedLB::originalTransfer() {
12231237
auto lazy_epoch = theTerm()->makeEpochCollective("TemperedLB: decide");
12241238

12251239
// Initialize transfer and rejection counters

src/vt/vrt/collection/balance/temperedlb/temperedlb.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ struct TemperedLB : BaseLB {
9191
void doLBStages(LoadType start_imb);
9292
void informAsync();
9393
void informSync();
94-
void decide();
94+
void originalTransfer();
9595
void migrate();
9696

9797
void propagateRound(uint8_t k_cur_async, bool sync, EpochType epoch = no_epoch);

0 commit comments

Comments
 (0)