-/* Copyright (c) 2009-2021. The SimGrid Team.
+/* Copyright (c) 2009-2022. The SimGrid Team.
* All rights reserved. */
/* This program is free software; you can redistribute it and/or modify it
#include <simgrid/s4u.hpp>
#include <string.h>
-#if SIMGRID_HAVE_JEDULE
-#include "simgrid/jedule/jedule_sd_binding.h"
-#endif
-
XBT_LOG_NEW_DEFAULT_CATEGORY(dag_scheduling, "Logging specific to this example");
-typedef struct _HostAttribute* HostAttribute;
-struct _HostAttribute {
+struct HostAttribute {
/* Earliest time at which a host is ready to execute a task */
- double available_at;
- simgrid::s4u::Exec* last_scheduled_task;
+ double available_at = 0.0;
+ simgrid::s4u::Exec* last_scheduled_task = nullptr;
};
static double sg_host_get_available_at(const simgrid::s4u::Host* host)
{
- return static_cast<HostAttribute>(host->get_data())->available_at;
+ return host->get_data<HostAttribute>()->available_at;
}
-static void sg_host_set_available_at(simgrid::s4u::Host* host, double time)
+static void sg_host_set_available_at(const simgrid::s4u::Host* host, double time)
{
- auto* attr = static_cast<HostAttribute>(host->get_data());
- attr->available_at = time;
- host->set_data(attr);
+ host->get_data<HostAttribute>()->available_at = time;
}
static simgrid::s4u::Exec* sg_host_get_last_scheduled_task(const simgrid::s4u::Host* host)
{
- return static_cast<HostAttribute>(host->get_data())->last_scheduled_task;
+ return host->get_data<HostAttribute>()->last_scheduled_task;
}
-static void sg_host_set_last_scheduled_task(simgrid::s4u::Host* host, simgrid::s4u::ExecPtr task)
+static void sg_host_set_last_scheduled_task(const simgrid::s4u::Host* host, simgrid::s4u::ExecPtr task)
{
- auto* attr = static_cast<HostAttribute>(host->get_data());
- attr->last_scheduled_task = task.get();
- host->set_data(attr);
+ host->get_data<HostAttribute>()->last_scheduled_task = task.get();
}
-static bool dependency_exists(simgrid::s4u::Exec* src, simgrid::s4u::Exec* dst)
+static bool dependency_exists(const simgrid::s4u::Exec* src, simgrid::s4u::Exec* dst)
{
const auto& dependencies = src->get_dependencies();
const auto& successors = src->get_successors();
dependencies.find(dst) != dependencies.end());
}
-static std::vector<simgrid::s4u::Exec*> get_ready_tasks(const std::vector<simgrid::s4u::ActivityPtr> dax)
+static std::vector<simgrid::s4u::Exec*> get_ready_tasks(const std::vector<simgrid::s4u::ActivityPtr>& dax)
{
std::vector<simgrid::s4u::Exec*> ready_tasks;
std::map<simgrid::s4u::Exec*, unsigned int> candidate_execs;
ready_tasks.push_back(exec);
// if it a comm, we consider its successor as a candidate. If a candidate solves all its dependencies,
// i.e., get all its input data, it's ready
- auto* comm = dynamic_cast<simgrid::s4u::Comm*>(a.get());
+ const auto* comm = dynamic_cast<simgrid::s4u::Comm*>(a.get());
if (comm != nullptr) {
auto* next_exec = static_cast<simgrid::s4u::Exec*>(comm->get_successors().front().get());
candidate_execs[next_exec]++;
last_data_available = -1.0;
for (const auto& parent : parents) {
/* normal case */
- auto* comm = dynamic_cast<simgrid::s4u::Comm*>(parent.get());
+ const auto* comm = dynamic_cast<simgrid::s4u::Comm*>(parent.get());
if (comm != nullptr) {
auto source = comm->get_source();
XBT_DEBUG("transfer from %s to %s", source->get_cname(), host->get_cname());
}
// We use the user data field to store the finish time of the predecessor of the comm, i.e., its potential start
// time
- data_available = *(static_cast<double*>(comm->get_data())) + redist_time;
+ data_available = *comm->get_data<double>() + redist_time;
}
- auto* exec = dynamic_cast<simgrid::s4u::Exec*>(parent.get());
+ const auto* exec = dynamic_cast<simgrid::s4u::Exec*>(parent.get());
/* no transfer, control dependency */
if (exec != nullptr) {
data_available = exec->get_finish_time();
return best_host;
}
-int main(int argc, char** argv)
+static void schedule_on(simgrid::s4u::ExecPtr exec, simgrid::s4u::Host* host)
{
- double min_finish_time = -1.0;
- simgrid::s4u::Exec* selected_task = nullptr;
- simgrid::s4u::Host* selected_host = nullptr;
- char* tracefilename = nullptr;
+ exec->set_host(host);
+ // we can also set the destination of all the input comms of this exec
+ for (const auto& pred : exec->get_dependencies()) {
+ auto* comm = dynamic_cast<simgrid::s4u::Comm*>(pred.get());
+ if (comm != nullptr) {
+ comm->set_destination(host);
+ delete comm->get_data<double>();
+ }
+ }
+ // we can also set the source of all the output comms of this exec
+ for (const auto& succ : exec->get_successors()) {
+ auto* comm = dynamic_cast<simgrid::s4u::Comm*>(succ.get());
+ if (comm != nullptr)
+ comm->set_source(host);
+ }
+}
+int main(int argc, char** argv)
+{
simgrid::s4u::Engine e(&argc, argv);
std::set<simgrid::s4u::Activity*> vetoed;
e.track_vetoed_activities(&vetoed);
- simgrid::s4u::Activity::on_completion.connect([](simgrid::s4u::Activity& activity) {
+ simgrid::s4u::Activity::on_completion_cb([](simgrid::s4u::Activity const& activity) {
// when an Exec completes, we need to set the potential start time of all its ouput comms
- const auto* exec = dynamic_cast<simgrid::s4u::Exec*>(&activity);
+ const auto* exec = dynamic_cast<simgrid::s4u::Exec const*>(&activity);
if (exec == nullptr) // Only Execs are concerned here
return;
for (const auto& succ : exec->get_successors()) {
auto* comm = dynamic_cast<simgrid::s4u::Comm*>(succ.get());
if (comm != nullptr) {
- double* finish_time = new double(exec->get_finish_time());
+ auto* finish_time = new double(exec->get_finish_time());
// We use the user data field to store the finish time of the predecessor of the comm, i.e., its potential start
// time
comm->set_data(finish_time);
}
}
});
- /* Check our arguments */
- xbt_assert(argc > 2,
- "Usage: %s platform_file dax_file [jedule_file]\n"
- "\tExample: %s simulacrum_7_hosts.xml Montage_25.xml Montage_25.jed",
- argv[0], argv[0]);
-
- if (argc == 4)
- tracefilename = xbt_strdup(argv[3]);
e.load_platform(argv[1]);
/* Allocating the host attribute */
- unsigned int total_nhosts = e.get_host_count();
+ unsigned long total_nhosts = e.get_host_count();
const auto hosts = e.get_all_hosts();
-
- for (unsigned int i = 0; i < total_nhosts; i++)
- hosts[i]->set_data(xbt_new0(struct _HostAttribute, 1));
+ std::vector<HostAttribute> host_attributes(total_nhosts);
+ for (unsigned long i = 0; i < total_nhosts; i++)
+ hosts[i]->set_data(&host_attributes[i]);
/* load the DAX file */
- std::vector<simgrid::s4u::ActivityPtr> dax = simgrid::s4u::create_DAG_from_DAX(argv[2]);
+ auto dax = simgrid::s4u::create_DAG_from_DAX(argv[2]);
/* Schedule the root first */
auto* root = static_cast<simgrid::s4u::Exec*>(dax.front().get());
auto host = get_best_host(root);
- root->set_host(host);
- // we can also set the source of all the output comms of the root node
- for (const auto& succ : root->get_successors()) {
- auto* comm = dynamic_cast<simgrid::s4u::Comm*>(succ.get());
- if (comm != nullptr)
- comm->set_source(host);
- }
+ schedule_on(root, host);
e.run();
vetoed.clear();
if (ready_tasks.empty()) {
- ready_tasks.clear();
/* there is no ready task, let advance the simulation */
e.run();
continue;
* get the host that minimizes the completion time.
* select the task that has the minimum completion time on its best host.
*/
+ double min_finish_time = -1.0;
+ simgrid::s4u::Exec* selected_task = nullptr;
+ simgrid::s4u::Host* selected_host = nullptr;
+
for (auto task : ready_tasks) {
XBT_DEBUG("%s is ready", task->get_cname());
host = get_best_host(task);
}
XBT_INFO("Schedule %s on %s", selected_task->get_cname(), selected_host->get_cname());
- selected_task->set_host(selected_host);
- // we can also set the destination of all the input comms of the selected task
- for (const auto& pred : selected_task->get_dependencies()) {
- auto* comm = dynamic_cast<simgrid::s4u::Comm*>(pred.get());
- if (comm != nullptr) {
- comm->set_destination(selected_host);
- delete static_cast<double*>(comm->get_data());
- }
- }
- // we can also set the source of all the output comms of the selected task
- for (const auto& succ : selected_task->get_successors()) {
- auto* comm = dynamic_cast<simgrid::s4u::Comm*>(succ.get());
- if (comm != nullptr)
- comm->set_source(selected_host);
- }
+ schedule_on(selected_task, selected_host);
/*
* tasks can be executed concurrently when they can by default.
sg_host_set_available_at(selected_host, min_finish_time);
ready_tasks.clear();
- /* reset the min_finish_time for the next set of ready tasks */
- min_finish_time = -1.;
e.run();
}
XBT_INFO("Simulation Time: %f", simgrid_get_clock());
- XBT_INFO("------------------- Produce the trace file---------------------------");
- XBT_INFO("Producing a jedule output (if active) of the run into %s",
- tracefilename ? tracefilename : "minmin_test.jed");
-#if SIMGRID_HAVE_JEDULE
- jedule_sd_dump(tracefilename);
-#endif
- free(tracefilename);
-
- for (auto h : hosts) {
- xbt_free(h->get_data());
- h->set_data(nullptr);
- }
return 0;
}
