runshenzhu / palmtree Goto Github PK

void redistribute_inner_tasks(int layer, NodeModsMapType &cur_mods) {
  cur_mods = node_mods_[layer];

  // discard
  for (int i = 0; i < worker_id_; i++) {
    auto &wthread = palmtree_->workers_[i];
    for (auto other_itr = wthread.node_mods_[layer].begin();
         other_itr != wthread.node_mods_[layer].end(); other_itr++) {
      cur_mods.erase(other_itr->first);
    }
  }

  // Steal work from other threads
  for (int i = worker_id_ + 1; i < palmtree_->NUM_WORKER; i++) {
    auto &wthread = palmtree_->workers_[i];
    for (auto other_itr = wthread.node_mods_[layer].begin();
         other_itr != wthread.node_mods_[layer].end(); other_itr++) {
      auto itr = cur_mods.find(other_itr->first);
      if (itr != cur_mods.end()) {
        auto &my_mods = itr->second;
        auto &other_mods = other_itr->second;
        my_mods.insert(my_mods.end(), other_mods.begin(), other_mods.end());
      }
    }
  }
}