Properly assign cpuwip-edf-os

1 files changed, 3 insertions, 2 deletions

diff --git a/src/task.c b/src/task.c
index e618d9e..8b4287c 100644
--- a/src/task.c
+++ b/src/task.c
@@ -71,12 +71,13 @@ int __create_rt_task_edffm(rt_fn_t rt_prog, void *arg, int cpu, int wcet,
                                 (rt_setup_fn_t) set_rt_task_param, &params);
 }
 
-int __create_rt_task_edfos(rt_fn_t rt_prog, void *arg, int wcet,
+int __create_rt_task_edfos(rt_fn_t rt_prog, void *arg, int cpu, int wcet,
                         int period, lt_t fracs[][2], task_class_t class)
 {
         struct rt_task params;
         struct edfos_params os;
         int i;
+        params.cpu       = cpu;
         params.period    = period;
         params.exec_cost = wcet;
         params.cls       = class;
@@ -86,7 +87,7 @@ int __create_rt_task_edfos(rt_fn_t rt_prog, void *arg, int wcet,
 
         os.migrat = 0;
         os.first_cpu = UINT_MAX;
-        for (i = 0; i < NR_CPUS; i++) {
+        for (i = 0; i < NR_CPUS_EDF_OS; i++) {
                 if (fracs[i][0] > 0) {
                         os.migrat++;
                         if (os.first_cpu == UINT_MAX)