Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
move surf_presolve, surf_solve, and surf_get_clock to EngineImpl
[simgrid.git] / src / surf / cpu_cas01.cpp
1 /* Copyright (c) 2009-2021. 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/kernel/routing/NetZoneImpl.hpp>
7 #include <simgrid/s4u/Engine.hpp>
8
9 #include "simgrid/sg_config.hpp"
10 #include "src/kernel/EngineImpl.hpp"
11 #include "src/kernel/resource/profile/Event.hpp"
12 #include "src/surf/cpu_cas01.hpp"
13 #include "src/surf/cpu_ti.hpp"
14 #include "src/surf/surf_interface.hpp"
15
16 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(cpu_cas, res_cpu, "CPU resource, CAS01 model (used by default)");
17
18 /***********
19  * Options *
20  ***********/
21
22 static simgrid::config::Flag<std::string>
23     cpu_optim_opt("cpu/optim", "Optimization algorithm to use for CPU resources. ", "Lazy",
24
25                   std::map<std::string, std::string, std::less<>>({
26                       {"Lazy", "Lazy action management (partial invalidation in lmm + heap in action remaining)."},
27                       {"TI", "Trace integration. Highly optimized mode when using availability traces (only available "
28                              "for the Cas01 CPU model for now)."},
29                       {"Full", "Full update of remaining and variables. Slow but may be useful when debugging."},
30                   }),
31
32                   [](std::string const&) {
33                     xbt_assert(_sg_cfg_init_status < 2,
34                                "Cannot change the optimization algorithm after the initialization");
35                   });
36
37 /*********
38  * Model *
39  *********/
40 void surf_cpu_model_init_Cas01()
41 {
42   if (cpu_optim_opt == "TI") {
43     simgrid::kernel::resource::CpuTiModel::create_pm_models();
44     return;
45   }
46
47   auto cpu_model_pm = std::make_shared<simgrid::kernel::resource::CpuCas01Model>("Cpu_Cas01");
48   auto* engine      = simgrid::kernel::EngineImpl::get_instance();
49   engine->add_model(cpu_model_pm);
50   engine->get_netzone_root()->set_cpu_pm_model(cpu_model_pm);
51 }
52
53 namespace simgrid {
54 namespace kernel {
55 namespace resource {
56
57 CpuCas01Model::CpuCas01Model(const std::string& name) : CpuModel(name)
58 {
59   if (config::get_value<std::string>("cpu/optim") == "Lazy")
60     set_update_algorithm(Model::UpdateAlgo::LAZY);
61
62   bool select = config::get_value<bool>("cpu/maxmin-selective-update");
63
64   if (is_update_lazy()) {
65     xbt_assert(select || config::is_default("cpu/maxmin-selective-update"),
66                "You cannot disable cpu selective update when using the lazy update mechanism");
67     select = true;
68   }
69
70   set_maxmin_system(new lmm::System(select));
71 }
72
73 CpuImpl* CpuCas01Model::create_cpu(s4u::Host* host, const std::vector<double>& speed_per_pstate)
74 {
75   return (new CpuCas01(host, speed_per_pstate))->set_model(this);
76 }
77
78 /************
79  * Resource *
80  ************/
81 /** @brief take into account changes of speed (either load or max) */
82 void CpuCas01::on_speed_change()
83 {
84   const lmm::Element* elem = nullptr;
85
86   get_model()->get_maxmin_system()->update_constraint_bound(get_constraint(),
87                                                             get_core_count() * speed_.scale * speed_.peak);
88
89   while (const auto* var = get_constraint()->get_variable(&elem)) {
90     const auto* action = static_cast<CpuCas01Action*>(var->get_id());
91     double bound       = action->requested_core() * speed_.scale * speed_.peak;
92     if (action->get_user_bound() > 0) {
93       bound = std::min(bound, action->get_user_bound());
94     }
95
96     get_model()->get_maxmin_system()->update_variable_bound(action->get_variable(), bound);
97   }
98
99   CpuImpl::on_speed_change();
100 }
101
102 void CpuCas01::apply_event(profile::Event* event, double value)
103 {
104   if (event == speed_.event) {
105     speed_.scale = value;
106     on_speed_change();
107
108     tmgr_trace_event_unref(&speed_.event);
109   } else if (event == get_state_event()) {
110     if (value > 0) {
111       if (not is_on()) {
112         XBT_VERB("Restart actors on host %s", get_iface()->get_cname());
113         get_iface()->turn_on();
114       }
115     } else {
116       const lmm::Element* elem = nullptr;
117       double date              = EngineImpl::get_clock();
118
119       get_iface()->turn_off();
120
121       while (const auto* var = get_constraint()->get_variable(&elem)) {
122         Action* action = var->get_id();
123
124         if (action->get_state() == Action::State::INITED || action->get_state() == Action::State::STARTED ||
125             action->get_state() == Action::State::IGNORED) {
126           action->set_finish_time(date);
127           action->set_state(Action::State::FAILED);
128         }
129       }
130     }
131     unref_state_event();
132
133   } else {
134     xbt_die("Unknown event!\n");
135   }
136 }
137
138 /** @brief Start a new execution on this CPU lasting @param size flops and using one core */
139 CpuAction* CpuCas01::execution_start(double size, double user_bound)
140 {
141   return execution_start(size, 1, user_bound);
142 }
143
144 CpuAction* CpuCas01::execution_start(double size, int requested_cores, double user_bound)
145 {
146   auto* action =
147       new CpuCas01Action(get_model(), size, not is_on(), speed_.scale * speed_.peak, get_constraint(), requested_cores);
148   action->set_user_bound(user_bound);
149   if (user_bound > 0 && user_bound < action->get_bound()) {
150     get_model()->get_maxmin_system()->update_variable_bound(action->get_variable(), user_bound);
151   }
152   if (factor_cb_) {
153     action->set_rate_factor(factor_cb_(size));
154   }
155
156   return action;
157 }
158
159 CpuAction* CpuCas01::sleep(double duration)
160 {
161   if (duration > 0)
162     duration = std::max(duration, sg_surf_precision);
163
164   XBT_IN("(%s, %g)", get_cname(), duration);
165   auto* action = new CpuCas01Action(get_model(), 1.0, not is_on(), speed_.scale * speed_.peak, get_constraint(), 1);
166
167   // FIXME: sleep variables should not consume 1.0 in System::expand()
168   action->set_max_duration(duration);
169   action->set_suspend_state(Action::SuspendStates::SLEEPING);
170   if (duration == NO_MAX_DURATION)
171     action->set_state(Action::State::IGNORED);
172
173   get_model()->get_maxmin_system()->update_variable_penalty(action->get_variable(), 0.0);
174   if (get_model()->is_update_lazy()) { // remove action from the heap
175     get_model()->get_action_heap().remove(action);
176     // this is necessary for a variable with weight 0 since such variables are ignored in lmm and we need to set its
177     // max_duration correctly at the next call to share_resources
178     get_model()->get_modified_set()->push_front(*action);
179   }
180
181   XBT_OUT();
182   return action;
183 }
184
185 void CpuCas01::set_factor_cb(const std::function<s4u::Host::CpuFactorCb>& cb)
186 {
187   xbt_assert(not is_sealed(), "Cannot set CPU factor callback in an already sealed CPU(%s)", get_cname());
188   factor_cb_ = cb;
189 }
190
191 /**********
192  * Action *
193  **********/
194 CpuCas01Action::CpuCas01Action(Model* model, double cost, bool failed, double speed, lmm::Constraint* constraint,
195                                int requested_core)
196     : CpuAction(model, cost, failed,
197                 model->get_maxmin_system()->variable_new(this, 1.0 / requested_core, requested_core * speed, 1))
198     , requested_core_(requested_core)
199 {
200   if (model->is_update_lazy())
201     set_last_update();
202   model->get_maxmin_system()->expand(constraint, get_variable(), 1.0);
203 }
204
205 } // namespace resource
206 } // namespace kernel
207 } // namespace simgrid