examples/cpp/dag-failure/s4u-dag-failure.cpp

   1 /* Copyright (c) 2006-2023. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include "simgrid/s4u.hpp"
   8
   9 XBT_LOG_NEW_DEFAULT_CATEGORY(dag_failure, "Logging specific to this example");
  10
  11 namespace sg4 = simgrid::s4u;
  12
  13 int main(int argc, char** argv)
  14 {
  15   sg4::Engine e(&argc, argv);
  16   sg4::Engine::set_config("host/model:ptask_L07");
  17   e.load_platform(argv[1]);
  18
  19   auto* faulty = e.host_by_name("Faulty Host");
  20   auto* safe   = e.host_by_name("Safe Host");
  21   sg4::Exec::on_completion_cb([](sg4::Exec const& exec) {
  22     if (exec.get_state() == sg4::Activity::State::FINISHED)
  23       XBT_INFO("Activity '%s' is complete (start time: %f, finish time: %f)", exec.get_cname(), exec.get_start_time(),
  24                exec.get_finish_time());
  25     if (exec.get_state() == sg4::Activity::State::FAILED) {
  26       if (exec.is_parallel())
  27         XBT_INFO("Activity '%s' has failed. %.f %% remain to be done", exec.get_cname(),
  28                  100 * exec.get_remaining_ratio());
  29       else
  30         XBT_INFO("Activity '%s' has failed. %.f flops remain to be done", exec.get_cname(), exec.get_remaining());
  31     }
  32   });
  33
  34   /* creation of a single Exec that will poorly fail when the workstation will stop */
  35   XBT_INFO("First test: sequential Exec activity");
  36   sg4::ExecPtr exec = sg4::Exec::init()->set_name("Poor task")->set_flops_amount(2e10)->start();
  37
  38   XBT_INFO("Schedule Activity '%s' on 'Faulty Host'", exec->get_cname());
  39   exec->set_host(faulty);
  40
  41   /* Add a child Exec that depends on the Poor task' */
  42   sg4::ExecPtr child = sg4::Exec::init()->set_name("Child")->set_flops_amount(2e10)->set_host(safe);
  43   exec->add_successor(child);
  44   child->start();
  45
  46   XBT_INFO("Run the simulation");
  47   e.run();
  48
  49   XBT_INFO("let's unschedule Activity '%s' and reschedule it on the 'Safe Host'", exec->get_cname());
  50   exec->unset_host();
  51   exec->set_host(safe);
  52
  53   XBT_INFO("Run the simulation again");
  54   e.run();
  55
  56   XBT_INFO("Second test: parallel Exec activity");
  57   exec = sg4::Exec::init()->set_name("Poor parallel task")->set_flops_amounts({2e10, 2e10})->start();
  58
  59   XBT_INFO("Schedule Activity '%s' on 'Safe Host' and 'Faulty Host'", exec->get_cname());
  60   exec->set_hosts({safe, faulty});
  61
  62   /* Add a child Exec that depends on the Poor parallel task' */
  63   child = sg4::Exec::init()->set_name("Child")->set_flops_amount(2e10)->set_host(safe);
  64   exec->add_successor(child);
  65   child->start();
  66
  67   XBT_INFO("Run the simulation");
  68   e.run();
  69
  70   XBT_INFO("let's unschedule Activity '%s' and reschedule it only on the 'Safe Host'", exec->get_cname());
  71   exec->unset_hosts();
  72   exec->set_flops_amount(4e10)->set_host(safe);
  73
  74   XBT_INFO("Run the simulation again");
  75   e.run();
  76
  77   return 0;
  78 }