src/surf/plugins/dirty_page_tracking.cpp

   1 /* Copyright (c) 2017. 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/plugins/live_migration.h"
   7 #include "simgrid/s4u.hpp"
   8 #include "src/plugins/vm/VirtualMachineImpl.hpp"
   9 #include <map>
  10
  11 namespace simgrid {
  12 namespace vm {
  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;
  17
  18 public:
  19   void startTracking();
  20   void stopTracking() { dp_tracking = false; }
  21   bool isTracking() { return dp_tracking; }
  22   void track(kernel::activity::ExecImplPtr exec, double amount) { dp_objs.insert({exec, amount}); }
  23   void untrack(kernel::activity::ExecImplPtr exec) { dp_objs.erase(exec); }
  24   double getStoredRemains(kernel::activity::ExecImplPtr exec) { return dp_objs.at(exec); }
  25   void updateDirtyPageCount(double delta) { dp_updated_by_deleted_tasks += delta; }
  26   double computedFlopsLookup();
  27
  28   static simgrid::xbt::Extension<VirtualMachineImpl, VmDirtyPageTrackingExt> EXTENSION_ID;
  29   virtual ~VmDirtyPageTrackingExt() = default;
  30   VmDirtyPageTrackingExt()          = default;
  31 };
  32
  33 simgrid::xbt::Extension<VirtualMachineImpl, VmDirtyPageTrackingExt> VmDirtyPageTrackingExt::EXTENSION_ID;
  34
  35 void VmDirtyPageTrackingExt::startTracking()
  36 {
  37   dp_tracking = true;
  38   for (auto const& elm : dp_objs)
  39     dp_objs[elm.first] = elm.first->remains();
  40 }
  41
  42 double VmDirtyPageTrackingExt::computedFlopsLookup()
  43 {
  44   double total = 0;
  45
  46   for (auto const& elm : dp_objs) {
  47     total += elm.second - elm.first->remains();
  48     dp_objs[elm.first] = elm.first->remains();
  49   }
  50   total += dp_updated_by_deleted_tasks;
  51
  52   dp_updated_by_deleted_tasks = 0;
  53
  54   return total;
  55 }
  56 }
  57 }
  58
  59 static void onVirtualMachineCreation(simgrid::vm::VirtualMachineImpl* vm)
  60 {
  61   vm->extension_set<simgrid::vm::VmDirtyPageTrackingExt>(new simgrid::vm::VmDirtyPageTrackingExt());
  62 }
  63
  64 static void onExecCreation(simgrid::kernel::activity::ExecImplPtr exec)
  65 {
  66   simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(exec->host_);
  67   if (vm == nullptr)
  68     return;
  69
  70   if (vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->isTracking()) {
  71     vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->track(exec, exec->remains());
  72   } else {
  73     vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->track(exec, 0.0);
  74   }
  75 }
  76
  77 static void onExecDestruction(simgrid::kernel::activity::ExecImplPtr exec)
  78 {
  79   simgrid::s4u::VirtualMachine* vm = dynamic_cast<simgrid::s4u::VirtualMachine*>(exec->host_);
  80   if (vm == nullptr)
  81     return;
  82
  83   /* If we are in the middle of dirty page tracking, we record how much computation has been done until now, and keep
  84    * the information for the lookup_() function that will called soon. */
  85   if (vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->isTracking()) {
  86     double delta =
  87         vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->getStoredRemains(exec) - exec->remains();
  88     vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->updateDirtyPageCount(delta);
  89   }
  90   vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->untrack(exec);
  91 }
  92
  93 SG_BEGIN_DECL()
  94
  95 void sg_vm_live_migration_plugin_init()
  96 {
  97   if (not simgrid::vm::VmDirtyPageTrackingExt::EXTENSION_ID.valid()) {
  98     simgrid::vm::VmDirtyPageTrackingExt::EXTENSION_ID =
  99         simgrid::vm::VirtualMachineImpl::extension_create<simgrid::vm::VmDirtyPageTrackingExt>();
 100     simgrid::vm::VirtualMachineImpl::onVmCreation.connect(&onVirtualMachineCreation);
 101     simgrid::kernel::activity::ExecImpl::onCreation.connect(&onExecCreation);
 102     simgrid::kernel::activity::ExecImpl::onDestruction.connect(&onExecDestruction);
 103   }
 104 }
 105
 106 void sg_vm_start_dirty_page_tracking(sg_vm_t vm)
 107 {
 108   vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->startTracking();
 109 }
 110
 111 void sg_vm_stop_dirty_page_tracking(sg_vm_t vm)
 112 {
 113   vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->stopTracking();
 114 }
 115
 116 double sg_vm_lookup_computed_flops(sg_vm_t vm)
 117 {
 118   return vm->pimpl_vm_->extension<simgrid::vm::VmDirtyPageTrackingExt>()->computedFlopsLookup();
 119 }
 120 SG_END_DECL()