#2033: lb: fix bugs in LBDataHolder with assertions too strong

DARMA-tasking · Dec 7, 2022 · 8b37df9 · 8b37df9
1 parent 8e1f80d
commit 8b37df9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/vt/vrt/collection/balance/lb_data_holder.cc b/src/vt/vrt/collection/balance/lb_data_holder.cc
@@ -80,7 +80,7 @@ std::unique_ptr<nlohmann::json> LBDataHolder::toJson(PhaseType phase) const {
       ElementIDStruct id = elm.first;
       TimeType time = elm.second.whole_phase_load;
       j["tasks"][i]["resource"] = "cpu";
-      j["tasks"][i]["node"] = theContext()->getNode();
+      j["tasks"][i]["node"] = id.getCurrNode();
       j["tasks"][i]["time"] = time;
       if (user_defined_json_.find(phase) != user_defined_json_.end()) {
         auto &user_def_this_phase = user_defined_json_.at(phase);
@@ -183,7 +183,6 @@ LBDataHolder::LBDataHolder(nlohmann::json const& j) {
           auto etype = task["entity"]["type"];
           vtAssertExpr(time.is_number());
           vtAssertExpr(node.is_number());
-          vtAssertExpr(node == this_node);
 
           if (etype == "object") {
             auto object = task["entity"]["id"];