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 "simgrid/s4u.hpp"
8 #include "src/plugins/vm/VirtualMachineImpl.hpp"
13 class VmDirtyPageTrackingExt {
14 bool dp_tracking = false;
15 std::map<kernel::activity::ExecImplPtr, double> dp_objs;
16 double dp_updated_by_deleted_tasks = 0.0;
17 // Percentage of pages that get dirty compared to netspeed [0;1] bytes per 1 flop execution
18 double dp_intensity = 0.0;
19 sg_size_t working_set_memory = 0.0;
20 double max_downtime = 0.03;
21 double mig_speed = 0.0;
25 void stopTracking() { dp_tracking = false; }
26 bool isTracking() { return dp_tracking; }
27 void track(kernel::activity::ExecImplPtr exec, double amount) { dp_objs.insert({exec, amount}); }
28 void untrack(kernel::activity::ExecImplPtr exec) { dp_objs.erase(exec); }
29 double getStoredRemains(kernel::activity::ExecImplPtr exec) { return dp_objs.at(exec); }
30 void updateDirtyPageCount(double delta) { dp_updated_by_deleted_tasks += delta; }
31 double computedFlopsLookup();
32 double getIntensity() { return dp_intensity; }
33 void setIntensity(double intensity) { dp_intensity = intensity; }
34 double getWorkingSetMemory() { return working_set_memory; }
35 void setWorkingSetMemory(sg_size_t size) { working_set_memory = size; }
36 void setMigrationSpeed(double speed) { mig_speed = speed; }
37 double getMigrationSpeed() { return mig_speed; }
38 double getMaxDowntime() { return max_downtime; }
40 static simgrid::xbt::Extension<VirtualMachineImpl, VmDirtyPageTrackingExt> EXTENSION_ID;
41 virtual ~VmDirtyPageTrackingExt() = default;
42 VmDirtyPageTrackingExt() = default;
45 simgrid::xbt::Extension<VirtualMachineImpl, VmDirtyPageTrackingExt> VmDirtyPageTrackingExt::EXTENSION_ID;
47 void VmDirtyPageTrackingExt::startTracking()
50 for (auto const& elm : dp_objs)
51 dp_objs[elm.first] = elm.first->remains();
54 double VmDirtyPageTrackingExt::computedFlopsLookup()
58 for (auto const& elm : dp_objs) {
59 total += elm.second - elm.first->remains();
60 dp_objs[elm.first] = elm.first->remains();
62 total += dp_updated_by_deleted_tasks;
64 dp_updated_by_deleted_tasks = 0;
71 static void onVirtualMachineCreation(simgrid::vm::VirtualMachineImpl* vm)
73 vm->extension_set<simgrid::vm::VmDirtyPageTrackingExt>(new simgrid::vm::VmDirtyPageTrackingExt());
76 static void onExecCreation(simgrid::kernel::activity::ExecImplPtr exec)
78 simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(exec->host_);
82 if (vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->isTracking()) {
83 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->track(exec, exec->remains());
85 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->track(exec, 0.0);
89 static void onExecCompletion(simgrid::kernel::activity::ExecImplPtr exec)
91 simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(exec->host_);
95 /* If we are in the middle of dirty page tracking, we record how much computation has been done until now, and keep
96 * the information for the lookup_() function that will called soon. */
97 if (vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->isTracking()) {
99 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->getStoredRemains(exec) - exec->remains();
100 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->updateDirtyPageCount(delta);
102 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->untrack(exec);
105 void sg_vm_dirty_page_tracking_init()
107 if (not simgrid::vm::VmDirtyPageTrackingExt::EXTENSION_ID.valid()) {
108 simgrid::vm::VmDirtyPageTrackingExt::EXTENSION_ID =
109 simgrid::vm::VirtualMachineImpl::extension_create<simgrid::vm::VmDirtyPageTrackingExt>();
110 simgrid::vm::VirtualMachineImpl::onVmCreation.connect(&onVirtualMachineCreation);
111 simgrid::kernel::activity::ExecImpl::onCreation.connect(&onExecCreation);
112 simgrid::kernel::activity::ExecImpl::onCompletion.connect(&onExecCompletion);
116 void sg_vm_start_dirty_page_tracking(sg_vm_t vm)
118 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->startTracking();
121 void sg_vm_stop_dirty_page_tracking(sg_vm_t vm)
123 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->stopTracking();
126 double sg_vm_lookup_computed_flops(sg_vm_t vm)
128 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->computedFlopsLookup();
131 void sg_vm_set_dirty_page_intensity(sg_vm_t vm, double intensity)
133 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->setIntensity(intensity);
136 double sg_vm_get_dirty_page_intensity(sg_vm_t vm)
138 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->getIntensity();
141 void sg_vm_set_working_set_memory(sg_vm_t vm, sg_size_t size)
143 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->setWorkingSetMemory(size);
146 sg_size_t sg_vm_get_working_set_memory(sg_vm_t vm)
148 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->getWorkingSetMemory();
151 void sg_vm_set_migration_speed(sg_vm_t vm, double speed)
153 vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->setMigrationSpeed(speed);
156 double sg_vm_get_migration_speed(sg_vm_t vm)
158 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->getMigrationSpeed();
161 double sg_vm_get_max_downtime(sg_vm_t vm)
163 return vm->getImpl()->extension<simgrid::vm::VmDirtyPageTrackingExt>()->getMaxDowntime();