+double Task::get_finish_time() const
+{
+ if (surf_action_) /* should never happen as actions are destroyed right after their completion */
+ return surf_action_->get_finish_time();
+ else
+ return finish_time_;
+}
+
+void Task::set_sender_side_allocation(unsigned long count, const std::vector<s4u::Host*>* sender)
+{
+ for (unsigned long i = 0; i < count; i++)
+ allocation_->push_back(sender->at(i));
+}
+
+void Task::set_receiver_side_allocation(unsigned long count, const std::vector<s4u::Host*>* receiver)
+{
+ for (unsigned long i = 0; i < count; i++)
+ allocation_->insert(allocation_->begin() + i, receiver->at(i));
+}
+
+void Task::watch(e_SD_task_state_t state)
+{
+ if (state & SD_NOT_SCHEDULED)
+ throw std::invalid_argument("Cannot add a watch point for state SD_NOT_SCHEDULED");
+
+ watch_points_ = watch_points_ | state;
+}
+
+void Task::unwatch(e_SD_task_state_t state)
+{
+ xbt_assert(state != SD_NOT_SCHEDULED, "SimDag error: Cannot have a watch point for state SD_NOT_SCHEDULED");
+ watch_points_ = watch_points_ & ~state;
+}
+
+void Task::dump() const
+{
+ XBT_INFO("Displaying task %s", get_cname());
+ if (state_ == SD_RUNNABLE)
+ XBT_INFO(" - state: runnable");
+ else if (state_ < SD_RUNNABLE)
+ XBT_INFO(" - state: %s not runnable", __get_state_name(state_));
+ else
+ XBT_INFO(" - state: not runnable %s", __get_state_name(state_));
+
+ if (kind_ != 0) {
+ switch (kind_) {
+ case SD_TASK_COMM_E2E:
+ XBT_INFO(" - kind: end-to-end communication");
+ break;
+ case SD_TASK_COMP_SEQ:
+ XBT_INFO(" - kind: sequential computation");
+ break;
+ case SD_TASK_COMP_PAR_AMDAHL:
+ XBT_INFO(" - kind: parallel computation following Amdahl's law");
+ break;
+ case SD_TASK_COMM_PAR_MXN_1D_BLOCK:
+ XBT_INFO(" - kind: MxN data redistribution assuming 1D block distribution");
+ break;
+ default:
+ XBT_INFO(" - (unknown kind %d)", kind_);
+ }
+ }
+
+ XBT_INFO(" - amount: %.0f", amount_);
+ if (kind_ == SD_TASK_COMP_PAR_AMDAHL)
+ XBT_INFO(" - alpha: %.2f", alpha_);
+ XBT_INFO(" - Dependencies to satisfy: %lu", has_unsolved_dependencies());
+ if (has_unsolved_dependencies() > 0) {
+ XBT_INFO(" - pre-dependencies:");
+ for (auto const& it : predecessors_)
+ XBT_INFO(" %s", it->get_cname());
+
+ for (auto const& it : inputs_)
+ XBT_INFO(" %s", it->get_cname());
+ }
+ if (is_waited_by() > 0) {
+ XBT_INFO(" - post-dependencies:");
+
+ for (auto const& it : successors_)
+ XBT_INFO(" %s", it->get_cname());
+ for (auto const& it : outputs_)
+ XBT_INFO(" %s", it->get_cname());
+ }
+}
+
+void Task::released_by(Task* pred)
+{
+ predecessors_.erase(pred);
+ inputs_.erase(pred);
+ XBT_DEBUG("Release dependency on %s: %lu remain(s). Becomes schedulable if %zu=0", get_cname(),
+ has_unsolved_dependencies(), predecessors_.size());
+
+ if (state_ == SD_NOT_SCHEDULED && predecessors_.empty())
+ set_state(SD_SCHEDULABLE);
+
+ if (state_ == SD_SCHEDULED && has_unsolved_dependencies() == 0)
+ set_state(SD_RUNNABLE);
+
+ if (state_ == SD_RUNNABLE && not sd_global->watch_point_reached)
+ run();
+}
+
+void Task::produced_by(Task* pred)
+{
+ start_time_ = pred->get_finish_time();
+ predecessors_.erase(pred);
+ if (state_ == SD_SCHEDULED)
+ set_state(SD_RUNNABLE);
+ else
+ set_state(SD_SCHEDULABLE);
+
+ Task* comm_dst = *(successors_.begin());
+ if (comm_dst->get_state() == SD_NOT_SCHEDULED && comm_dst->get_predecessors().empty()) {
+ XBT_DEBUG("%s is a transfer, %s may be ready now if %zu=0", get_cname(), comm_dst->get_cname(),
+ comm_dst->get_predecessors().size());
+ comm_dst->set_state(SD_SCHEDULABLE);
+ }
+ if (state_ == SD_RUNNABLE && not sd_global->watch_point_reached)
+ run();
+}
+
+void Task::do_schedule()
+{
+ if (state_ > SD_SCHEDULABLE)
+ throw std::invalid_argument(simgrid::xbt::string_printf("Task '%s' has already been scheduled", get_cname()));
+
+ if (has_unsolved_dependencies() == 0)
+ set_state(SD_RUNNABLE);
+ else
+ set_state(SD_SCHEDULED);
+}
+
+void Task::schedule(const std::vector<s4u::Host*>& hosts, const double* flops_amount, const double* bytes_amount,
+ double rate)
+{
+ unsigned long host_count = hosts.size();
+ rate_ = rate;
+
+ if (flops_amount) {
+ flops_amount_ = static_cast<double*>(xbt_realloc(flops_amount_, sizeof(double) * host_count));
+ memcpy(flops_amount_, flops_amount, sizeof(double) * host_count);
+ } else {
+ xbt_free(flops_amount_);
+ flops_amount_ = nullptr;
+ }
+
+ unsigned long communication_nb = host_count * host_count;
+ if (bytes_amount) {
+ bytes_amount_ = static_cast<double*>(xbt_realloc(bytes_amount_, sizeof(double) * communication_nb));
+ memcpy(bytes_amount_, bytes_amount, sizeof(double) * communication_nb);
+ } else {
+ xbt_free(bytes_amount_);
+ bytes_amount_ = nullptr;
+ }
+
+ for (unsigned long i = 0; i < host_count; i++)
+ allocation_->push_back(hosts[i]);
+
+ do_schedule();
+}
+
+void Task::schedulev(const std::vector<s4u::Host*>& hosts)
+{
+ xbt_assert(kind_ == SD_TASK_COMP_SEQ || kind_ == SD_TASK_COMP_PAR_AMDAHL,
+ "Task %s is not typed. Cannot automatically schedule it.", get_cname());
+
+ for (unsigned long i = 0; i < hosts.size(); i++)
+ allocation_->push_back(hosts[i]);
+
+ XBT_VERB("Schedule computation task %s on %zu host(s)", get_cname(), allocation_->size());
+
+ if (kind_ == SD_TASK_COMP_SEQ) {
+ if (not flops_amount_) { /*This task has failed and is rescheduled. Reset the flops_amount*/
+ flops_amount_ = xbt_new0(double, 1);
+ flops_amount_[0] = amount_;
+ }
+ XBT_VERB("It costs %.f flops", flops_amount_[0]);
+ }
+
+ if (kind_ == SD_TASK_COMP_PAR_AMDAHL) {
+ distribute_comp_amdahl(hosts.size());
+ XBT_VERB("%.f flops will be distributed following Amdahl's Law", flops_amount_[0]);
+ }
+
+ do_schedule();
+
+ /* Iterate over all inputs and outputs to say where I am located (and start them if runnable) */
+ for (auto const& input : inputs_) {
+ unsigned long src_nb = input->get_allocation_size();
+ unsigned long dst_nb = hosts.size();
+ if (src_nb == 0)
+ XBT_VERB("Sender side of '%s' not scheduled. Set receiver side to '%s''s allocation", input->get_cname(),
+ get_cname());
+ input->set_sender_side_allocation(dst_nb, allocation_);
+
+ if (input->get_allocation_size() > allocation_->size()) {
+ if (kind_ == SD_TASK_COMP_PAR_AMDAHL)
+ input->build_MxN_1D_block_matrix(src_nb, dst_nb);
+
+ input->do_schedule();
+ XBT_VERB("Auto-Schedule Communication task '%s'. Send %.f bytes from %zu hosts to %zu hosts.", input->get_cname(),
+ input->get_amount(), src_nb, dst_nb);
+ }
+ }
+
+ for (auto const& output : outputs_) {
+ unsigned long src_nb = hosts.size();
+ unsigned long dst_nb = output->get_allocation_size();
+ if (dst_nb == 0)
+ XBT_VERB("Receiver side of '%s' not scheduled. Set sender side to '%s''s allocation", output->get_cname(),
+ get_cname());
+ output->set_receiver_side_allocation(src_nb, allocation_);
+
+ if (output->get_allocation_size() > allocation_->size()) {
+ if (kind_ == SD_TASK_COMP_PAR_AMDAHL)
+ output->build_MxN_1D_block_matrix(src_nb, dst_nb);
+
+ output->do_schedule();
+ XBT_VERB("Auto-Schedule Communication task %s. Send %.f bytes from %lu hosts to %lu hosts.", output->get_cname(),
+ output->get_amount(), src_nb, dst_nb);
+ }
+ }
+}
+
+void Task::unschedule()
+{
+ if (state_ == SD_NOT_SCHEDULED || state_ == SD_SCHEDULABLE)
+ throw std::invalid_argument(xbt::string_printf(
+ "Task %s: the state must be SD_SCHEDULED, SD_RUNNABLE, SD_RUNNING or SD_FAILED", get_cname()));
+
+ if (state_ == SD_SCHEDULED || state_ == SD_RUNNABLE) /* if the task is scheduled or runnable */ {
+ allocation_->clear();
+ if (kind_ == SD_TASK_COMP_PAR_AMDAHL || kind_ == SD_TASK_COMM_PAR_MXN_1D_BLOCK) {
+ /* Don't free scheduling data for typed tasks */
+ xbt_free(flops_amount_);
+ xbt_free(bytes_amount_);
+ bytes_amount_ = nullptr;
+ flops_amount_ = nullptr;
+ }
+ }
+
+ if (state_ == SD_RUNNING)
+ /* the task should become SD_FAILED */
+ surf_action_->cancel();
+ else {
+ if (has_unsolved_dependencies() == 0)
+ set_state(SD_SCHEDULABLE);
+ else
+ set_state(SD_NOT_SCHEDULED);
+ }
+ start_time_ = -1.0;
+}
+
+void Task::run()
+{
+ xbt_assert(state_ == SD_RUNNABLE, "Task '%s' is not runnable! Task state: %d", get_cname(), (int)state_);
+ xbt_assert(not allocation_->empty(), "Task '%s': host_list is empty!", get_cname());
+
+ XBT_VERB("Executing task '%s'", get_cname());
+
+ /* Beware! The scheduling data are now used by the surf action directly! no copy was done */
+ auto host_model = allocation_->front()->get_netpoint()->get_englobing_zone()->get_host_model();
+ surf_action_ = host_model->execute_parallel(*allocation_, flops_amount_, bytes_amount_, rate_);
+
+ surf_action_->set_data(this);
+
+ XBT_DEBUG("surf_action = %p", surf_action_);
+
+ set_state(SD_RUNNING);
+ sd_global->return_set.insert(this);
+}
+
+void Task::destroy()
+{
+ XBT_DEBUG("Destroying task %s...", get_cname());
+
+ /* First Remove all dependencies associated with the task. */
+ while (not predecessors_.empty())
+ SD_task_dependency_remove(*(predecessors_.begin()), this);
+ while (not inputs_.empty())
+ SD_task_dependency_remove(*(inputs_.begin()), this);
+ while (not successors_.empty())
+ SD_task_dependency_remove(this, *(successors_.begin()));
+ while (not outputs_.empty())
+ SD_task_dependency_remove(this, *(outputs_.begin()));
+
+ if (state_ == SD_SCHEDULED || state_ == SD_RUNNABLE) {
+ xbt_free(flops_amount_);
+ xbt_free(bytes_amount_);
+ bytes_amount_ = nullptr;
+ flops_amount_ = nullptr;
+ }
+
+ xbt_free(flops_amount_);
+ xbt_free(bytes_amount_);
+
+ delete allocation_;
+
+ if (surf_action_ != nullptr)
+ surf_action_->unref();
+
+ XBT_DEBUG("Task destroyed.");
+}
+} // namespace sd
+} // namespace simgrid
+
+/* **************************** Public C interface *************************** */
+