total_load_init += load;
ctrl_close_pending = data_close_pending = neigh.size();
- if (neigh.size() == 1) {
- comm.next_close_on_ctrl_is_last();
- comm.next_close_on_data_is_last();
- }
close_received = false;
- may_receive = (neigh.size() > 0); // the same as (ctrl_close_pending ||
- // data_close_pending)
finalizing = false;
- if (may_receive)
- comm.listen();
e_xbt_log_priority_t logp = xbt_log_priority_verbose;
if (!LOG_ISENABLED(logp))
// block on receiving unless there is something to compute or
// to send
- bool recv_wait = (load == 0 &&
- ((opt::bookkeeping ? expected_load : load)
- == prev_load_broadcast));
- DEBUG1("CALL RECEIVE(%s)", recv_wait? "WAIT": "NO_WAIT");
- receive(recv_wait? WAIT: NO_WAIT);
+ bool wait = (load == 0 &&
+ ((opt::bookkeeping ? expected_load : load)
+ == prev_load_broadcast));
+ receive(wait);
// one of our neighbor is finalizing
if (opt::exit_on_close && close_received)
break;
// have no load and cannot receive anything
- if (load == 0.0 && !may_receive)
+ if (load == 0.0 && !may_receive())
break;
// fixme: this check should be implemented with a distributed
if (load > 0.0) {
double duration = opt::comp_cost(load);
m_task_t task = MSG_task_create("computation", duration, 0.0, NULL);
- DEBUG2("compute %g flop%s.", duration, ESSE(duration));
+ DEBUG2("compute %g flop%s", duration, ESSE(duration));
MSG_task_execute(task);
MSG_task_destroy(task);
} else {
bind(&process::send1_no_bookkeeping, this, _1));
prev_load_broadcast = load;
}
+ comm.flush(false);
}
-void process::receive(recv_wait_mode wait)
+void process::receive(bool wait)
{
- // DEBUG1("go for receive(%s)",
- // "NO_WAIT\0WAIT\0\0\0\0WAIT_FOR_CLOSE" + 8 * wait);
message* msg;
m_host_t from;
- bool do_wait = (wait != NO_WAIT);
- while (may_receive && comm.recv(msg, from, do_wait)) {
+
+ while (may_receive() && comm.recv(msg, from, wait)) {
switch (msg->get_type()) {
case message::INFO: {
neighbor* n = rev_neigh[from];
break;
}
case message::CTRL_CLOSE:
- if (--ctrl_close_pending == 1)
- comm.next_close_on_ctrl_is_last();
- // DEBUG1("ctrl_close_pending = %d", ctrl_close_pending);
+ ctrl_close_pending--;
close_received = true;
break;
case message::DATA_CLOSE:
- if (--data_close_pending == 1)
- comm.next_close_on_data_is_last();
- // DEBUG1("data_close_pending = %d", data_close_pending);
+ data_close_pending--;
close_received = true;
break;
}
delete msg;
- may_receive = (ctrl_close_pending || data_close_pending);
- do_wait = (wait == WAIT_FOR_CLOSE);
+ wait = false; // only wait on first recv
}
+ comm.flush(false);
}
void process::finalize1(neighbor& nb)
finalizing = true;
total_load_running -= load;
- DEBUG2("send CLOSE to %d neighbor%s.",
- (int )neigh.size(), ESSE(neigh.size()));
+ DEBUG2("send CLOSE to %lu neighbor%s",
+ (unsigned long )neigh.size(), ESSE(neigh.size()));
std::for_each(neigh.begin(), neigh.end(),
bind(&process::finalize1, this, _1));
- DEBUG2("wait for CLOSE from %d neighbor%s.",
- (int )neigh.size(), ESSE(neigh.size()));
- receive(WAIT_FOR_CLOSE);
+ DEBUG2("wait for CLOSE from %lu neighbor%s",
+ (unsigned long )neigh.size(), ESSE(neigh.size()));
+ while (may_receive()) {
+ comm.flush(false);
+ receive(true);
+ }
comm.flush(true);
}