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

   1 /* Copyright (c) 2006-2022. 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::Activity::on_completion_cb([](sg4::Activity& activity) {
  22     const auto* exec = dynamic_cast<simgrid::s4u::Exec*>(&activity);
  23     if (exec == nullptr) // Only Execs are concerned here
  24       return;
  25     if (exec->get_state() == sg4::Activity::State::FINISHED)
  26       XBT_INFO("Activity '%s' is complete (start time: %f, finish time: %f)", exec->get_cname(), exec->get_start_time(),
  27                exec->get_finish_time());
  28     if (exec->get_state() == sg4::Activity::State::FAILED) {
  29       if (exec->is_parallel())
  30         XBT_INFO("Activity '%s' has failed. %.f %% remain to be done", exec->get_cname(),
  31                  100 * exec->get_remaining_ratio());
  32       else
  33         XBT_INFO("Activity '%s' has failed. %.f flops remain to be done", exec->get_cname(), exec->get_remaining());
  34     }
  35   });
  36
  37   /* creation of a single Exec that will poorly fail when the workstation will stop */
  38   XBT_INFO("First test: sequential Exec activity");
  39   sg4::ExecPtr exec = sg4::Exec::init()->set_name("Poor task")->set_flops_amount(2e10)->vetoable_start();
  40
  41   XBT_INFO("Schedule Activity '%s' on 'Faulty Host'", exec->get_cname());
  42   exec->set_host(faulty);
  43
  44   /* Add a child Exec that depends on the Poor task' */
  45   sg4::ExecPtr child = sg4::Exec::init()->set_name("Child")->set_flops_amount(2e10)->set_host(safe);
  46   exec->add_successor(child);
  47   child->vetoable_start();
  48
  49   XBT_INFO("Run the simulation");
  50   e.run();
  51
  52   XBT_INFO("let's unschedule Activity '%s' and reschedule it on the 'Safe Host'", exec->get_cname());
  53   exec->unset_host();
  54   exec->set_host(safe);
  55
  56   XBT_INFO("Run the simulation again");
  57   e.run();
  58
  59   XBT_INFO("Second test: parallel Exec activity");
  60   exec = sg4::Exec::init()->set_name("Poor parallel task")->set_flops_amounts({2e10, 2e10})->vetoable_start();
  61
  62   XBT_INFO("Schedule Activity '%s' on 'Safe Host' and 'Faulty Host'", exec->get_cname());
  63   exec->set_hosts({safe, faulty});
  64
  65   /* Add a child Exec that depends on the Poor parallel task' */
  66   child = sg4::Exec::init()->set_name("Child")->set_flops_amount(2e10)->set_host(safe);
  67   exec->add_successor(child);
  68   child->vetoable_start();
  69
  70   XBT_INFO("Run the simulation");
  71   e.run();
  72
  73   XBT_INFO("let's unschedule Activity '%s' and reschedule it only on the 'Safe Host'", exec->get_cname());
  74   exec->unset_hosts();
  75   exec->set_flops_amount(4e10)->set_host(safe);
  76
  77   XBT_INFO("Run the simulation again");
  78   e.run();
  79
  80   return 0;
  81 }