1 files changed, 8 insertions, 6 deletions

diff --git a/src/task.c b/src/task.c
index ed1c6aa..9121f0c 100644
--- a/src/task.c
+++ b/src/task.c
@@ -40,11 +40,11 @@ int __launch_rt_task(rt_fn_t rt_prog, void *rt_arg, rt_setup_fn_t setup,
         return rt_task;
 }
 
-int __create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, lt_t wcet, lt_t period,
-                     unsigned int priority, task_class_t class)
+int __create_rt_task(rt_fn_t rt_prog, void *arg, int cluster, int cluster_size,
+                lt_t wcet, lt_t period, unsigned int priority, task_class_t class)
 {
         struct rt_task params;
-        params.cpu       = cpu;
+        params.cpu       = cluster_to_first_cpu(cluster, cluster_size);
         params.period    = period;
         params.exec_cost = wcet;
         params.cls       = class;
@@ -57,9 +57,11 @@ int __create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, lt_t wcet, lt_t period
                                 (rt_setup_fn_t) set_rt_task_param, &params);
 }
 
-int create_rt_task(rt_fn_t rt_prog, void *arg, int cpu, lt_t wcet, lt_t period,
-                   unsigned int priority) {
-        return __create_rt_task(rt_prog, arg, cpu, wcet, period, priority, RT_CLASS_HARD);
+int create_rt_task(rt_fn_t rt_prog, void *arg, int cluster, int cluster_size,
+                lt_t wcet, lt_t period, unsigned int prio)
+{
+        return __create_rt_task(rt_prog, arg, cluster, cluster_size, wcet, period,
+                                prio, RT_CLASS_HARD);
 }