examples/cpp/dag-simple/s4u-dag-simple.cpp

   1 /* Copyright (c) 2007-2023. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #include "simgrid/s4u.hpp"
   7 #include <vector>
   8
   9 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "Messages specific for this s4u example");
  10 namespace sg4 = simgrid::s4u;
  11
  12 int main(int argc, char* argv[])
  13 {
  14   sg4::Engine e(&argc, argv);
  15   e.load_platform(argv[1]);
  16   std::set<sg4::Activity*> vetoed;
  17   e.track_vetoed_activities(&vetoed);
  18
  19   auto* fafard = e.host_by_name("Fafard");
  20
  21   // Display the details on vetoed activities
  22   sg4::Exec::on_veto_cb([](sg4::Exec const& exec) {
  23     XBT_INFO("Execution '%s' vetoed. Dependencies: %s; Ressources: %s", exec.get_cname(),
  24              (exec.dependencies_solved() ? "solved" : "NOT solved"),
  25              (exec.is_assigned() ? "assigned" : "NOT assigned"));
  26   });
  27
  28   sg4::Exec::on_completion_cb([](sg4::Exec const& exec) {
  29     XBT_INFO("Execution '%s' is complete (start time: %f, finish time: %f)", exec.get_cname(), exec.get_start_time(),
  30              exec.get_finish_time());
  31   });
  32
  33   // Define an amount of work that should take 1 second to execute.
  34   double computation_amount = fafard->get_speed();
  35
  36   // Create a small DAG: Two parents and a child
  37   sg4::ExecPtr first_parent  = sg4::Exec::init();
  38   sg4::ExecPtr second_parent = sg4::Exec::init();
  39   sg4::ExecPtr child         = sg4::Exec::init();
  40   first_parent->add_successor(child);
  41   second_parent->add_successor(child);
  42
  43   // Set the parameters (the name is for logging purposes only)
  44   // + First parent ends after 1 second and the Second parent after 2 seconds.
  45   first_parent->set_name("parent 1")->set_flops_amount(computation_amount);
  46   second_parent->set_name("parent 2")->set_flops_amount(2 * computation_amount);
  47   child->set_name("child")->set_flops_amount(computation_amount);
  48
  49   // Only the parents are scheduled so far
  50   first_parent->set_host(fafard);
  51   second_parent->set_host(fafard);
  52
  53   // Start all activities that can actually start.
  54   first_parent->start();
  55   second_parent->start();
  56   child->start();
  57
  58   while (child->get_state() != sg4::Activity::State::FINISHED) {
  59     e.run();
  60     for (auto* a : vetoed) {
  61       auto* exec = static_cast<sg4::Exec*>(a);
  62
  63       // In this simple case, we just assign the child task to a resource when its dependencies are solved
  64       if (exec->dependencies_solved() && not exec->is_assigned()) {
  65         XBT_INFO("Activity %s's dependencies are resolved. Let's assign it to Fafard.", exec->get_cname());
  66         exec->set_host(fafard);
  67       } else {
  68         XBT_INFO("Activity %s not ready.", exec->get_cname());
  69       }
  70     }
  71     vetoed.clear(); // DON'T FORGET TO CLEAR this set between two calls to run
  72   }
  73
  74   XBT_INFO("Simulation time %g", sg4::Engine::get_clock());
  75
  76   return 0;
  77 }