Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Enable generic_lmm_action to be lazily updated by refactoring the corresponding fields:
[simgrid.git] / src / surf / cpu_cas01.c
index 3755de9..e922b65 100644 (file)
@@ -6,13 +6,14 @@
 
 #include "surf_private.h"
 #include "surf/surf_resource.h"
+#include "maxmin_private.h"
 
 surf_model_t surf_cpu_model = NULL;
 lmm_system_t cpu_maxmin_system = NULL;
 e_UM_t cpu_update_mechanism = UM_UNDEFINED;
 static int cpu_selective_update = 0;
 
-static xbt_swag_t cpu_modified_cpu = NULL;
+static xbt_swag_t cpu_modified_set= NULL;
 static xbt_heap_t cpu_action_heap = NULL;
 
 #undef GENERIC_LMM_ACTION
@@ -24,9 +25,6 @@ static xbt_heap_t cpu_action_heap = NULL;
 
 typedef struct surf_action_cpu_cas01 {
   s_surf_action_lmm_t generic_lmm_action;
-  s_xbt_swag_hookup_t cpu_list_hookup;
-  int index_heap;
-  void *cpu;
 } s_surf_action_cpu_Cas01_t, *surf_action_cpu_Cas01_t;
 
 typedef struct cpu_Cas01 {
@@ -39,8 +37,6 @@ typedef struct cpu_Cas01 {
   e_surf_resource_state_t state_current;
   tmgr_trace_event_t state_event;
   lmm_constraint_t constraint;
-  xbt_swag_t action_set;
-  double last_update;
 } s_cpu_Cas01_t, *cpu_Cas01_t;
 
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_cpu, surf,
@@ -51,7 +47,36 @@ XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_cpu, surf,
 static xbt_swag_t
     cpu_running_action_set_that_does_not_need_being_checked = NULL;
 
-static void* cpu_create_resource(const char *name, double power_peak,
+/* added to manage the communication action's heap */
+static void net_action_update_index_heap(void *action, int i)
+{
+  surf_action_cpu_Cas01_t a = action;
+  GENERIC_LMM_ACTION(a).index_heap = i;
+}
+
+/* insert action on heap using a given key and a hat (heap_action_type)
+ * a hat can be of three types for communications:
+ *
+ * NORMAL = this is a normal heap entry stating the date to finish transmitting
+ * LATENCY = this is a heap entry to warn us when the latency is payed
+ * MAX_DURATION =this is a heap entry to warn us when the max_duration limit is reached
+ */
+static void heap_insert(surf_action_cpu_Cas01_t action, double key,
+                        enum heap_action_type hat)
+{
+  GENERIC_LMM_ACTION(action).hat = hat;
+  xbt_heap_push(cpu_action_heap, action, key);
+}
+
+static void heap_remove(surf_action_cpu_Cas01_t action)
+{
+  GENERIC_LMM_ACTION(action).hat = NOTSET;
+  if (GENERIC_LMM_ACTION(action).index_heap >= 0) {
+    xbt_heap_remove(cpu_action_heap, GENERIC_LMM_ACTION(action).index_heap);
+  }
+}
+
+static void *cpu_create_resource(const char *name, double power_peak,
                                  double power_scale,
                                  tmgr_trace_t power_trace,
                                  int core,
@@ -60,18 +85,18 @@ static void* cpu_create_resource(const char *name, double power_peak,
                                  xbt_dict_t cpu_properties)
 {
   cpu_Cas01_t cpu = NULL;
-  s_surf_action_cpu_Cas01_t action;
 
   xbt_assert(!surf_cpu_resource_by_name(name),
-              "Host '%s' declared several times in the platform file",
-              name);
+             "Host '%s' declared several times in the platform file",
+             name);
   cpu = (cpu_Cas01_t) surf_resource_new(sizeof(s_cpu_Cas01_t),
-          surf_cpu_model, name,cpu_properties);
+                                        surf_cpu_model, name,
+                                        cpu_properties);
   cpu->power_peak = power_peak;
   xbt_assert(cpu->power_peak > 0, "Power has to be >0");
   cpu->power_scale = power_scale;
   cpu->core = core;
-  xbt_assert(core>0,"Invalid number of cores %d",core);
+  xbt_assert(core > 0, "Invalid number of cores %d", core);
 
   if (power_trace)
     cpu->power_event =
@@ -87,8 +112,6 @@ static void* cpu_create_resource(const char *name, double power_peak,
                          cpu->core * cpu->power_scale * cpu->power_peak);
 
   xbt_lib_set(host_lib, name, SURF_CPU_LEVEL, cpu);
-  if(cpu_update_mechanism == UM_LAZY)
-    cpu->action_set = xbt_swag_new(xbt_swag_offset(action, cpu_list_hookup));
 
   return cpu;
 }
@@ -96,14 +119,13 @@ static void* cpu_create_resource(const char *name, double power_peak,
 
 static void parse_cpu_init(sg_platf_host_cbarg_t host)
 {
-       cpu_create_resource(host->id,
-                         host->power_peak,
-                         host->power_scale,
-                         host->power_trace,
-                         host->core_amount,
-                         host->initial_state,
-                         host->state_trace,
-                         host->properties);
+  cpu_create_resource(host->id,
+                      host->power_peak,
+                      host->power_scale,
+                      host->power_trace,
+                      host->core_amount,
+                      host->initial_state,
+                      host->state_trace, host->properties);
 }
 
 static void cpu_add_traces_cpu(void)
@@ -159,13 +181,10 @@ static int cpu_action_unref(surf_action_t action)
     if (((surf_action_lmm_t) action)->variable)
       lmm_variable_free(cpu_maxmin_system,
                         ((surf_action_lmm_t) action)->variable);
-    if(cpu_update_mechanism == UM_LAZY){
-    /* remove from heap */
-    xbt_heap_remove(cpu_action_heap,
-                    ((surf_action_cpu_Cas01_t) action)->index_heap);
-    xbt_swag_remove(action,
-                    ((cpu_Cas01_t) ACTION_GET_CPU(action))->action_set);
-    xbt_swag_insert(ACTION_GET_CPU(action), cpu_modified_cpu);
+    if (cpu_update_mechanism == UM_LAZY) {
+      /* remove from heap */
+      heap_remove((surf_action_cpu_Cas01_t) action);
+      xbt_swag_remove(action, cpu_modified_set);
     }
 #ifdef HAVE_TRACING
     xbt_free(action->category);
@@ -179,17 +198,15 @@ static int cpu_action_unref(surf_action_t action)
 static void cpu_action_cancel(surf_action_t action)
 {
   surf_action_state_set(action, SURF_ACTION_FAILED);
-  if(cpu_update_mechanism == UM_LAZY){
-  xbt_heap_remove(cpu_action_heap,
-                  ((surf_action_cpu_Cas01_t) action)->index_heap);
-  xbt_swag_remove(action,
-                  ((cpu_Cas01_t) ACTION_GET_CPU(action))->action_set);
+  if (cpu_update_mechanism == UM_LAZY) {
+    xbt_swag_remove(action, cpu_modified_set);
+    heap_remove((surf_action_cpu_Cas01_t) action);
   }
   return;
 }
 
-static void cpu_cpu_action_state_set(surf_action_t action,
-                                        e_surf_action_state_t state)
+static void cpu_action_state_set(surf_action_t action,
+                                     e_surf_action_state_t state)
 {
 /*   if((state==SURF_ACTION_DONE) || (state==SURF_ACTION_FAILED)) */
 /*     if(((surf_action_lmm_t)action)->variable) { */
@@ -201,114 +218,131 @@ static void cpu_cpu_action_state_set(surf_action_t action,
   return;
 }
 
-static void cpu_update_remains(cpu_Cas01_t cpu, double now)
+static void update_action_remaining_lazy(double now)
 {
   surf_action_cpu_Cas01_t action;
+  double delta = 0.0;
 
-  if (cpu->last_update >= now)
-    return;
-  xbt_swag_foreach(action, cpu->action_set) {
+  xbt_swag_foreach(action, cpu_modified_set) {
     if (GENERIC_ACTION(action).state_set !=
         surf_cpu_model->states.running_action_set)
-      continue;
+      continue; // FIXME useless ????
 
     /* bogus priority, skip it */
     if (GENERIC_ACTION(action).priority <= 0)
       continue;
 
+    delta = now - GENERIC_LMM_ACTION(action).last_update;
     if (GENERIC_ACTION(action).remains > 0) {
+
       double_update(&(GENERIC_ACTION(action).remains),
-                    lmm_variable_getvalue(GENERIC_LMM_ACTION
-                                          (action).variable) * (now -
-                                                                cpu->last_update));
+                    lmm_variable_getvalue(GENERIC_LMM_ACTION(action).variable) * delta);
+
 #ifdef HAVE_TRACING
       if (TRACE_is_enabled()) {
+        cpu_Cas01_t cpu =
+            lmm_constraint_id(lmm_get_cnst_from_var
+                              (cpu_maxmin_system,
+                               GENERIC_LMM_ACTION(action).variable, 0));
         TRACE_surf_host_set_utilization(cpu->generic_resource.name,
-                                        action->
-                                        generic_lmm_action.generic_action.
-                                        data, (surf_action_t) action,
+                                        action->generic_lmm_action.
+                                        generic_action.data,
+                                        (surf_action_t) action,
                                         lmm_variable_getvalue
-                                        (GENERIC_LMM_ACTION
-                                         (action).variable),
-                                        cpu->last_update,
-                                        now - cpu->last_update);
+                                        (GENERIC_LMM_ACTION(action).
+                                         variable), GENERIC_LMM_ACTION(action).last_update,
+                                        now - GENERIC_LMM_ACTION(action).last_update);
       }
 #endif
       XBT_DEBUG("Update action(%p) remains %lf", action,
-             GENERIC_ACTION(action).remains);
+                GENERIC_ACTION(action).remains);
     }
+    GENERIC_LMM_ACTION(action).last_update = now;
   }
-  cpu->last_update = now;
+
 }
 
 static double cpu_share_resources_lazy(double now)
 {
-  surf_action_cpu_Cas01_t action;
-  double min;
+  surf_action_cpu_Cas01_t action = NULL;
+  double min = -1;
   double value;
-  cpu_Cas01_t cpu, cpu_next;
 
-  xbt_swag_foreach(cpu, cpu_modified_cpu)
-      cpu_update_remains(cpu, now);
+  XBT_DEBUG
+      ("Before share resources, the size of modified actions set is %d",
+       xbt_swag_size(cpu_modified_set));
+  update_action_remaining_lazy(now);
 
   lmm_solve(cpu_maxmin_system);
 
-  xbt_swag_foreach_safe(cpu, cpu_next, cpu_modified_cpu) {
-    xbt_swag_foreach(action, cpu->action_set) {
-      if (GENERIC_ACTION(action).state_set !=
-          surf_cpu_model->states.running_action_set)
-        continue;
-
-      /* bogus priority, skip it */
-      if (GENERIC_ACTION(action).priority <= 0)
-        continue;
-
-      min = -1;
-      value = lmm_variable_getvalue(GENERIC_LMM_ACTION(action).variable);
-      if (value > 0) {
-        if (GENERIC_ACTION(action).remains > 0) {
-          value = GENERIC_ACTION(action).remains / value;
-          min = now + value;
-        } else {
-          value = 0.0;
-          min = now;
-        }
-      }
+  XBT_DEBUG
+      ("After share resources, The size of modified actions set is %d",
+       xbt_swag_size(cpu_modified_set));
 
-      if ((GENERIC_ACTION(action).max_duration != NO_MAX_DURATION)
-          && (min == -1
-              || GENERIC_ACTION(action).start +
-              GENERIC_ACTION(action).max_duration < min))
-        min =
-            GENERIC_ACTION(action).start +
-            GENERIC_ACTION(action).max_duration;
-
-      XBT_DEBUG("Action(%p) Start %lf Finish %lf Max_duration %lf", action,
-             GENERIC_ACTION(action).start, now + value,
-             GENERIC_ACTION(action).max_duration);
-
-      if (action->index_heap >= 0) {
-        surf_action_cpu_Cas01_t heap_act =
-            xbt_heap_remove(cpu_action_heap, action->index_heap);
-        if (heap_act != action)
-          DIE_IMPOSSIBLE;
-      }
-      if (min != -1) {
-        xbt_heap_push(cpu_action_heap, action, min);
-        XBT_DEBUG("Insert at heap action(%p) min %lf", action, min);
+  xbt_swag_foreach(action, cpu_modified_set) {
+    int max_dur_flag = 0;
+
+    if (GENERIC_ACTION(action).state_set !=
+        surf_cpu_model->states.running_action_set)
+      continue;
+
+    /* bogus priority, skip it */
+    if (GENERIC_ACTION(action).priority <= 0)
+      continue;
+
+    min = -1;
+    value = lmm_variable_getvalue(GENERIC_LMM_ACTION(action).variable);
+    if (value > 0) {
+      if (GENERIC_ACTION(action).remains > 0) {
+        value = GENERIC_ACTION(action).remains / value;
+        min = now + value;
+      } else {
+        value = 0.0;
+        min = now;
       }
     }
-    xbt_swag_remove(cpu, cpu_modified_cpu);
+
+    if ((GENERIC_ACTION(action).max_duration != NO_MAX_DURATION)
+        && (min == -1
+            || GENERIC_ACTION(action).start +
+            GENERIC_ACTION(action).max_duration < min)) {
+      min = GENERIC_ACTION(action).start +
+          GENERIC_ACTION(action).max_duration;
+      max_dur_flag = 1;
+    }
+
+    XBT_DEBUG("Action(%p) Start %lf Finish %lf Max_duration %lf", action,
+        GENERIC_ACTION(action).start, now + value,
+        GENERIC_ACTION(action).max_duration);
+
+    if (min != -1) {
+      heap_remove(action);
+      heap_insert(action, min, max_dur_flag ? MAX_DURATION : NORMAL);
+      XBT_DEBUG("Insert at heap action(%p) min %lf now %lf", action, min,
+                now);
+    } else DIE_IMPOSSIBLE;
   }
-  return xbt_heap_size(cpu_action_heap) >
-      0 ? xbt_heap_maxkey(cpu_action_heap) - now : -1;
+
+  //hereafter must have already the min value for this resource model
+  if (xbt_heap_size(cpu_action_heap) > 0)
+    min = xbt_heap_maxkey(cpu_action_heap) - now;
+  else
+    min = -1;
+
+  XBT_DEBUG("The minimum with the HEAP %lf", min);
+
+  return min;
 }
 
-static double cpu_share_resources_full(double now) {
-  surf_action_cpu_Cas01_t action;
-  return generic_maxmin_share_resources(surf_cpu_model->states.running_action_set,
-      xbt_swag_offset(*action, generic_lmm_action.variable),
-      cpu_maxmin_system, lmm_solve);
+static double cpu_share_resources_full(double now)
+{
+  s_surf_action_cpu_Cas01_t action;
+  return generic_maxmin_share_resources(surf_cpu_model->states.
+                                        running_action_set,
+                                        xbt_swag_offset(action,
+                                                        generic_lmm_action.
+                                                        variable),
+                                        cpu_maxmin_system, lmm_solve);
 }
 
 static void cpu_update_actions_state_lazy(double now, double delta)
@@ -322,39 +356,36 @@ static void cpu_update_actions_state_lazy(double now, double delta)
     /* set the remains to 0 due to precision problems when updating the remaining amount */
 #ifdef HAVE_TRACING
     if (TRACE_is_enabled()) {
-      cpu_Cas01_t cpu = ((cpu_Cas01_t)(action->cpu));
+      cpu_Cas01_t cpu =
+          lmm_constraint_id(lmm_get_cnst_from_var
+                            (cpu_maxmin_system,
+                             GENERIC_LMM_ACTION(action).variable, 0));
       TRACE_surf_host_set_utilization(cpu->generic_resource.name,
-          GENERIC_LMM_ACTION(action).generic_action.data,
-          (surf_action_t) action,
-          lmm_variable_getvalue (GENERIC_LMM_ACTION(action).variable),
-          cpu->last_update,
-          now - cpu->last_update);
+                                      GENERIC_LMM_ACTION(action).generic_action.data,
+                                      (surf_action_t) action,
+                                      lmm_variable_getvalue(GENERIC_LMM_ACTION(action).variable),
+                                      GENERIC_LMM_ACTION(action).last_update,
+                                      now - GENERIC_LMM_ACTION(action).last_update);
     }
 #endif
     GENERIC_ACTION(action).remains = 0;
-    cpu_cpu_action_state_set((surf_action_t) action, SURF_ACTION_DONE);
-    cpu_update_remains(action->cpu, surf_get_clock());
+    cpu_action_state_set((surf_action_t) action, SURF_ACTION_DONE);
+    heap_remove(action);
   }
 #ifdef HAVE_TRACING
   if (TRACE_is_enabled()) {
     //defining the last timestamp that we can safely dump to trace file
     //without losing the event ascending order (considering all CPU's)
-    void **data;
-    cpu_Cas01_t cpu;
-    xbt_lib_cursor_t cursor;
-    char *key;
     double smaller = -1;
-    xbt_lib_foreach(host_lib, cursor, key, data){
-      if(data[SURF_CPU_LEVEL]){
-        cpu = data[SURF_CPU_LEVEL];
-        if (smaller < 0){
-          smaller = cpu->last_update;
+    xbt_swag_t running_actions = surf_cpu_model->states.running_action_set;
+    xbt_swag_foreach(action, running_actions) {
+        if (smaller < 0) {
+          smaller = GENERIC_LMM_ACTION(action).last_update;
           continue;
         }
-        if (cpu->last_update < smaller){
-          smaller = cpu->last_update;
+        if (GENERIC_LMM_ACTION(action).last_update < smaller) {
+          smaller = GENERIC_LMM_ACTION(action).last_update;
         }
-      }
     }
     if (smaller > 0) {
       TRACE_last_timestamp_to_dump = smaller;
@@ -373,30 +404,34 @@ static void cpu_update_actions_state_full(double now, double delta)
 #ifdef HAVE_TRACING
     if (TRACE_is_enabled()) {
       cpu_Cas01_t x =
-        lmm_constraint_id(lmm_get_cnst_from_var
-                          (cpu_maxmin_system, GENERIC_LMM_ACTION(action).variable, 0));
+          lmm_constraint_id(lmm_get_cnst_from_var
+                            (cpu_maxmin_system,
+                             GENERIC_LMM_ACTION(action).variable, 0));
 
       TRACE_surf_host_set_utilization(x->generic_resource.name,
                                       GENERIC_ACTION(action).data,
                                       (surf_action_t) action,
                                       lmm_variable_getvalue
-                                      (GENERIC_LMM_ACTION(action).variable), now - delta,
-                                      delta);
-      TRACE_last_timestamp_to_dump = now-delta;
+                                      (GENERIC_LMM_ACTION(action).
+                                       variable), now - delta, delta);
+      TRACE_last_timestamp_to_dump = now - delta;
     }
 #endif
     double_update(&(GENERIC_ACTION(action).remains),
-                  lmm_variable_getvalue(GENERIC_LMM_ACTION(action).variable) * delta);
-    if (GENERIC_LMM_ACTION(action).generic_action.max_duration != NO_MAX_DURATION)
+                  lmm_variable_getvalue(GENERIC_LMM_ACTION(action).
+                                        variable) * delta);
+    if (GENERIC_LMM_ACTION(action).generic_action.max_duration !=
+        NO_MAX_DURATION)
       double_update(&(GENERIC_ACTION(action).max_duration), delta);
     if ((GENERIC_ACTION(action).remains <= 0) &&
-        (lmm_get_variable_weight(GENERIC_LMM_ACTION(action).variable) > 0)) {
+        (lmm_get_variable_weight(GENERIC_LMM_ACTION(action).variable) >
+         0)) {
       GENERIC_ACTION(action).finish = surf_get_clock();
-      cpu_cpu_action_state_set((surf_action_t) action, SURF_ACTION_DONE);
+      cpu_action_state_set((surf_action_t) action, SURF_ACTION_DONE);
     } else if ((GENERIC_ACTION(action).max_duration != NO_MAX_DURATION) &&
                (GENERIC_ACTION(action).max_duration <= 0)) {
       GENERIC_ACTION(action).finish = surf_get_clock();
-      cpu_cpu_action_state_set((surf_action_t) action, SURF_ACTION_DONE);
+      cpu_action_state_set((surf_action_t) action, SURF_ACTION_DONE);
     }
   }
 
@@ -404,8 +439,8 @@ static void cpu_update_actions_state_full(double now, double delta)
 }
 
 static void cpu_update_resource_state(void *id,
-                                         tmgr_trace_event_t event_type,
-                                         double value, double date)
+                                      tmgr_trace_event_t event_type,
+                                      double value, double date)
 {
   cpu_Cas01_t cpu = id;
   lmm_variable_t var = NULL;
@@ -414,19 +449,20 @@ static void cpu_update_resource_state(void *id,
   if (event_type == cpu->power_event) {
     cpu->power_scale = value;
     lmm_update_constraint_bound(cpu_maxmin_system, cpu->constraint,
-                                cpu->core * cpu->power_scale * cpu->power_peak);
+                                cpu->core * cpu->power_scale *
+                                cpu->power_peak);
 #ifdef HAVE_TRACING
     TRACE_surf_host_set_power(date, cpu->generic_resource.name,
-                              cpu->core * cpu->power_scale * cpu->power_peak);
+                              cpu->core * cpu->power_scale *
+                              cpu->power_peak);
 #endif
     while ((var = lmm_get_var_from_cnst
             (cpu_maxmin_system, cpu->constraint, &elem))) {
-       surf_action_cpu_Cas01_t action = lmm_variable_id(var);
-       lmm_update_variable_bound(cpu_maxmin_system, GENERIC_LMM_ACTION(action).variable,
-                                  cpu->power_scale * cpu->power_peak);
+      surf_action_cpu_Cas01_t action = lmm_variable_id(var);
+      lmm_update_variable_bound(cpu_maxmin_system,
+                                GENERIC_LMM_ACTION(action).variable,
+                                cpu->power_scale * cpu->power_peak);
     }
-    if(cpu_update_mechanism == UM_LAZY)
-      xbt_swag_insert(cpu, cpu_modified_cpu);
     if (tmgr_trace_event_free(event_type))
       cpu->power_event = NULL;
   } else if (event_type == cpu->state_event) {
@@ -434,13 +470,10 @@ static void cpu_update_resource_state(void *id,
       cpu->state_current = SURF_RESOURCE_ON;
     else {
       lmm_constraint_t cnst = cpu->constraint;
-      lmm_variable_t var = NULL;
-      lmm_element_t elem = NULL;
 
       cpu->state_current = SURF_RESOURCE_OFF;
 
-      while ((var =
-              lmm_get_var_from_cnst(cpu_maxmin_system, cnst, &elem))) {
+      while ((var = lmm_get_var_from_cnst(cpu_maxmin_system, cnst, &elem))) {
         surf_action_t action = lmm_variable_id(var);
 
         if (surf_action_state_get(action) == SURF_ACTION_RUNNING ||
@@ -448,7 +481,7 @@ static void cpu_update_resource_state(void *id,
             surf_action_state_get(action) ==
             SURF_ACTION_NOT_IN_THE_SYSTEM) {
           action->finish = date;
-          cpu_cpu_action_state_set(action, SURF_ACTION_FAILED);
+          cpu_action_state_set(action, SURF_ACTION_FAILED);
         }
       }
     }
@@ -478,12 +511,11 @@ static surf_action_t cpu_execute(void *cpu, double size)
 
   GENERIC_LMM_ACTION(action).variable =
       lmm_variable_new(cpu_maxmin_system, action,
-                       GENERIC_ACTION(action).priority, CPU->power_scale * CPU->power_peak, 1);
-  if(cpu_update_mechanism == UM_LAZY){
-    action->index_heap = -1;
-    action->cpu = CPU;
-    xbt_swag_insert(CPU, cpu_modified_cpu);
-    xbt_swag_insert(action, CPU->action_set);
+                       GENERIC_ACTION(action).priority,
+                       CPU->power_scale * CPU->power_peak, 1);
+  if (cpu_update_mechanism == UM_LAZY) {
+    GENERIC_LMM_ACTION(action).index_heap = -1;
+    GENERIC_LMM_ACTION(action).last_update = surf_get_clock();
   }
   lmm_expand(cpu_maxmin_system, CPU->constraint,
              GENERIC_LMM_ACTION(action).variable, 1.0);
@@ -513,8 +545,14 @@ static surf_action_t cpu_action_sleep(void *cpu, double duration)
 
   lmm_update_variable_weight(cpu_maxmin_system,
                              GENERIC_LMM_ACTION(action).variable, 0.0);
-  if(cpu_update_mechanism == UM_LAZY)
-    xbt_swag_insert(cpu, cpu_modified_cpu);
+  if (cpu_update_mechanism == UM_LAZY) {     // remove action from the heap
+    heap_remove((surf_action_cpu_Cas01_t) action);
+    // this is necessary for a variable with weight 0 since such
+    // variables are ignored in lmm and we need to set its max_duration
+    // correctly at the next call to share_resources
+    xbt_swag_insert_at_head(action,cpu_modified_set);
+  }
+
   XBT_OUT();
   return (surf_action_t) action;
 }
@@ -527,11 +565,8 @@ static void cpu_action_suspend(surf_action_t action)
                                ((surf_action_lmm_t) action)->variable,
                                0.0);
     ((surf_action_lmm_t) action)->suspended = 1;
-    if(cpu_update_mechanism == UM_LAZY){
-      xbt_heap_remove(cpu_action_heap,
-                      ((surf_action_cpu_Cas01_t) action)->index_heap);
-      xbt_swag_insert(ACTION_GET_CPU(action), cpu_modified_cpu);
-    }
+    if (cpu_update_mechanism == UM_LAZY)
+      heap_remove((surf_action_cpu_Cas01_t) action);
   }
   XBT_OUT();
 }
@@ -545,8 +580,8 @@ static void cpu_action_resume(surf_action_t action)
                                ((surf_action_lmm_t) action)->variable,
                                action->priority);
     ((surf_action_lmm_t) action)->suspended = 0;
-    if(cpu_update_mechanism == UM_LAZY)
-      xbt_swag_insert(ACTION_GET_CPU(action), cpu_modified_cpu);
+    if (cpu_update_mechanism == UM_LAZY)
+      heap_remove((surf_action_cpu_Cas01_t) action);
   }
   XBT_OUT();
 }
@@ -557,19 +592,17 @@ static int cpu_action_is_suspended(surf_action_t action)
 }
 
 static void cpu_action_set_max_duration(surf_action_t action,
-                                           double duration)
+                                        double duration)
 {
   XBT_IN("(%p,%g)", action, duration);
 
   action->max_duration = duration;
-  /* insert cpu in modified_cpu set to notice the max duration change */
-  if(cpu_update_mechanism == UM_LAZY)
-    xbt_swag_insert(ACTION_GET_CPU(action), cpu_modified_cpu);
+  if (cpu_update_mechanism == UM_LAZY)
+    heap_remove((surf_action_cpu_Cas01_t) action);
   XBT_OUT();
 }
 
-static void cpu_action_set_priority(surf_action_t action,
-                                       double priority)
+static void cpu_action_set_priority(surf_action_t action, double priority)
 {
   XBT_IN("(%p,%g)", action, priority);
   action->priority = priority;
@@ -577,17 +610,17 @@ static void cpu_action_set_priority(surf_action_t action,
                              ((surf_action_lmm_t) action)->variable,
                              priority);
 
-  if(cpu_update_mechanism == UM_LAZY)
-    xbt_swag_insert(ACTION_GET_CPU(action), cpu_modified_cpu);
+  if (cpu_update_mechanism == UM_LAZY)
+    heap_remove((surf_action_cpu_Cas01_t) action);
   XBT_OUT();
 }
 
 #ifdef HAVE_TRACING
 static void cpu_action_set_category(surf_action_t action,
-                                       const char *category)
+                                    const char *category)
 {
   XBT_IN("(%p,%s)", action, category);
-  action->category = xbt_strdup (category);
+  action->category = xbt_strdup(category);
   XBT_OUT();
 }
 #endif
@@ -596,8 +629,8 @@ static double cpu_action_get_remains(surf_action_t action)
 {
   XBT_IN("(%p)", action);
   /* update remains before return it */
-  if(cpu_update_mechanism == UM_LAZY)
-    cpu_update_remains(ACTION_GET_CPU(action), surf_get_clock());
+  if (cpu_update_mechanism == UM_LAZY)
+    update_action_remaining_lazy(surf_get_clock());
   XBT_OUT();
   return action->remains;
 }
@@ -618,42 +651,24 @@ static double cpu_get_available_speed(void *cpu)
   return ((cpu_Cas01_t) cpu)->power_scale;
 }
 
-static void cpu_action_update_index_heap(void *action, int i)
-{
-  ((surf_action_cpu_Cas01_t) action)->index_heap = i;
-}
-
 static void cpu_finalize(void)
 {
-  void **cpu;
-  xbt_lib_cursor_t cursor;
-  char *key;
-
-  xbt_lib_foreach(host_lib, cursor, key, cpu){
-    if(cpu[SURF_CPU_LEVEL])
-    {
-        cpu_Cas01_t CPU = cpu[SURF_CPU_LEVEL];
-        xbt_swag_free(CPU->action_set);
-    }
-  }
-
   lmm_system_free(cpu_maxmin_system);
   cpu_maxmin_system = NULL;
 
   surf_model_exit(surf_cpu_model);
   surf_cpu_model = NULL;
 
-  xbt_swag_free
-      (cpu_running_action_set_that_does_not_need_being_checked);
+  xbt_swag_free(cpu_running_action_set_that_does_not_need_being_checked);
   cpu_running_action_set_that_does_not_need_being_checked = NULL;
-  if(cpu_action_heap)  xbt_heap_free(cpu_action_heap);
-  if(cpu_modified_cpu) xbt_swag_free(cpu_modified_cpu);
+  if (cpu_action_heap)
+    xbt_heap_free(cpu_action_heap);
 }
 
 static void surf_cpu_model_init_internal()
 {
   s_surf_action_t action;
-  s_cpu_Cas01_t cpu;
+  s_surf_action_cpu_Cas01_t comp;
 
   surf_cpu_model = surf_model_init();
 
@@ -664,16 +679,20 @@ static void surf_cpu_model_init_internal()
 
   surf_cpu_model->action_unref = cpu_action_unref;
   surf_cpu_model->action_cancel = cpu_action_cancel;
-  surf_cpu_model->action_state_set = cpu_cpu_action_state_set;
+  surf_cpu_model->action_state_set = cpu_action_state_set;
 
   surf_cpu_model->model_private->resource_used = cpu_resource_used;
 
-  if(cpu_update_mechanism == UM_LAZY) {
-    surf_cpu_model->model_private->share_resources = cpu_share_resources_lazy;
-    surf_cpu_model->model_private->update_actions_state = cpu_update_actions_state_lazy;
+  if (cpu_update_mechanism == UM_LAZY) {
+    surf_cpu_model->model_private->share_resources =
+        cpu_share_resources_lazy;
+    surf_cpu_model->model_private->update_actions_state =
+        cpu_update_actions_state_lazy;
   } else if (cpu_update_mechanism == UM_FULL) {
-    surf_cpu_model->model_private->share_resources = cpu_share_resources_full;
-    surf_cpu_model->model_private->update_actions_state = cpu_update_actions_state_full;
+    surf_cpu_model->model_private->share_resources =
+        cpu_share_resources_full;
+    surf_cpu_model->model_private->update_actions_state =
+        cpu_update_actions_state_full;
   } else
     xbt_die("Invalid update mechanism!");
 
@@ -704,12 +723,13 @@ static void surf_cpu_model_init_internal()
   if (!cpu_maxmin_system) {
     cpu_maxmin_system = lmm_system_new(cpu_selective_update);
   }
-  if(cpu_update_mechanism == UM_LAZY){
+  if (cpu_update_mechanism == UM_LAZY) {
     cpu_action_heap = xbt_heap_new(8, NULL);
     xbt_heap_set_update_callback(cpu_action_heap,
-                                 cpu_action_update_index_heap);
-    cpu_modified_cpu =
-        xbt_swag_new(xbt_swag_offset(cpu, modified_cpu_hookup));
+        net_action_update_index_heap);
+    cpu_modified_set =
+        xbt_swag_new(xbt_swag_offset(comp, generic_lmm_action.action_list_hookup));
+    cpu_maxmin_system->keep_track = cpu_modified_set;
   }
 }
 
@@ -732,21 +752,25 @@ static void surf_cpu_model_init_internal()
 void surf_cpu_model_init_Cas01()
 {
   char *optim = xbt_cfg_get_string(_surf_cfg_set, "cpu/optim");
-  int select = xbt_cfg_get_int(_surf_cfg_set, "cpu/maxmin_selective_update");
+  int select =
+      xbt_cfg_get_int(_surf_cfg_set, "cpu/maxmin_selective_update");
 
-  if(!strcmp(optim,"Full")) {
+  if (!strcmp(optim, "Full")) {
     cpu_update_mechanism = UM_FULL;
     cpu_selective_update = select;
-  } else if (!strcmp(optim,"Lazy")) {
+  } else if (!strcmp(optim, "Lazy")) {
     cpu_update_mechanism = UM_LAZY;
     cpu_selective_update = 1;
-    xbt_assert((select==1) || (xbt_cfg_is_default_value(_surf_cfg_set,"cpu/maxmin_selective_update")),
-        "Disabling selective update while using the lazy update mechanism is dumb!");
-  } else if (!strcmp(optim,"TI")) {
+    xbt_assert((select == 1)
+               ||
+               (xbt_cfg_is_default_value
+                (_surf_cfg_set, "cpu/maxmin_selective_update")),
+               "Disabling selective update while using the lazy update mechanism is dumb!");
+  } else if (!strcmp(optim, "TI")) {
     surf_cpu_model_init_ti();
     return;
   } else {
-    xbt_die("Unsupported optimization (%s) for this model",optim);
+    xbt_die("Unsupported optimization (%s) for this model", optim);
   }
 
   if (surf_cpu_model)