1 /* Copyright (c) 2017-2018. The SimGrid Team. All rights reserved. */
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. */
6 #include "simgrid/plugins/live_migration.h"
7 #include "src/plugins/vm/VirtualMachineImpl.hpp"
11 class VmDirtyPageTrackingExt {
12 bool dp_tracking = false;
13 std::map<kernel::activity::ExecImplPtr, double> dp_objs;
14 double dp_updated_by_deleted_tasks = 0.0;
15 // Percentage of pages that get dirty compared to netspeed [0;1] bytes per 1 flop execution
16 double dp_intensity = 0.0;
17 sg_size_t working_set_memory = 0.0;
18 double max_downtime = 0.03;
19 double mig_speed = 0.0;
22 void start_tracking();
23 void stop_tracking() { dp_tracking = false; }
24 bool is_tracking() { return dp_tracking; }
25 void track(kernel::activity::ExecImplPtr exec, double amount) { dp_objs.insert({exec, amount}); }
26 void untrack(kernel::activity::ExecImplPtr exec) { dp_objs.erase(exec); }
27 double get_stored_remains(kernel::activity::ExecImplPtr exec) { return dp_objs.at(exec); }
28 void update_dirty_page_count(double delta) { dp_updated_by_deleted_tasks += delta; }
29 double computed_flops_lookup();
30 double get_intensity() { return dp_intensity; }
31 void set_intensity(double intensity) { dp_intensity = intensity; }
32 double get_working_set_memory() { return working_set_memory; }
33 void set_working_set_memory(sg_size_t size) { working_set_memory = size; }
34 void set_migration_speed(double speed) { mig_speed = speed; }
35 double get_migration_speed() { return mig_speed; }
36 double get_max_downtime() { return max_downtime; }
38 static simgrid::xbt::Extension<VirtualMachineImpl, VmDirtyPageTrackingExt> EXTENSION_ID;
39 virtual ~VmDirtyPageTrackingExt() = default;
40 VmDirtyPageTrackingExt() = default;
43 simgrid::xbt::Extension<VirtualMachineImpl, VmDirtyPageTrackingExt> VmDirtyPageTrackingExt::EXTENSION_ID;
45 void VmDirtyPageTrackingExt::start_tracking()
48 for (auto const& elm : dp_objs)
49 dp_objs[elm.first] = elm.first->get_remaining();
52 double VmDirtyPageTrackingExt::computed_flops_lookup()
56 for (auto const& elm : dp_objs) {
57 total += elm.second - elm.first->get_remaining();
58 dp_objs[elm.first] = elm.first->get_remaining();
60 total += dp_updated_by_deleted_tasks;
62 dp_updated_by_deleted_tasks = 0;
67 } // namespace simgrid
69 static void onVirtualMachineCreation(simgrid::vm::VirtualMachineImpl* vm)
71 vm->extension_set<simgrid::vm::VmDirtyPageTrackingExt>(new simgrid::vm::VmDirtyPageTrackingExt());
74 static void onExecCreation(simgrid::kernel::activity::ExecImplPtr exec)
76 simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(exec->host_);
80 if (vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->is_tracking()) {
81 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->track(exec, exec->get_remaining());
83 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->track(exec, 0.0);
87 static void onExecCompletion(simgrid::kernel::activity::ExecImplPtr exec)
89 simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(exec->host_);
93 /* If we are in the middle of dirty page tracking, we record how much computation has been done until now, and keep
94 * the information for the lookup_() function that will called soon. */
95 if (vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->is_tracking()) {
96 double delta = vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->get_stored_remains(exec) -
97 exec->get_remaining();
98 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->update_dirty_page_count(delta);
100 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->untrack(exec);
103 void sg_vm_dirty_page_tracking_init()
105 if (not simgrid::vm::VmDirtyPageTrackingExt::EXTENSION_ID.valid()) {
106 simgrid::vm::VmDirtyPageTrackingExt::EXTENSION_ID =
107 simgrid::vm::VirtualMachineImpl::extension_create<simgrid::vm::VmDirtyPageTrackingExt>();
108 simgrid::vm::VirtualMachineImpl::onVmCreation.connect(&onVirtualMachineCreation);
109 simgrid::kernel::activity::ExecImpl::onCreation.connect(&onExecCreation);
110 simgrid::kernel::activity::ExecImpl::onCompletion.connect(&onExecCompletion);
114 void sg_vm_start_dirty_page_tracking(sg_vm_t vm)
116 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->start_tracking();
119 void sg_vm_stop_dirty_page_tracking(sg_vm_t vm)
121 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->stop_tracking();
124 double sg_vm_lookup_computed_flops(sg_vm_t vm)
126 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->computed_flops_lookup();
129 void sg_vm_set_dirty_page_intensity(sg_vm_t vm, double intensity)
131 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->set_intensity(intensity);
134 double sg_vm_get_dirty_page_intensity(sg_vm_t vm)
136 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->get_intensity();
139 void sg_vm_set_working_set_memory(sg_vm_t vm, sg_size_t size)
141 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->set_working_set_memory(size);
144 sg_size_t sg_vm_get_working_set_memory(sg_vm_t vm)
146 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->get_working_set_memory();
149 void sg_vm_set_migration_speed(sg_vm_t vm, double speed)
151 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->set_migration_speed(speed);
154 double sg_vm_get_migration_speed(sg_vm_t vm)
156 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->get_migration_speed();
159 double sg_vm_get_max_downtime(sg_vm_t vm)
161 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->get_max_downtime();