Added support for CPU pinning in EXT-RES

2 files changed, 8 insertions, 6 deletions

diff --git a/bin/rt_launch.c b/bin/rt_launch.c
index 384ea64..77dbc31 100644
--- a/bin/rt_launch.c
+++ b/bin/rt_launch.c
@@ -155,12 +155,12 @@ int main(int argc, char** argv)
         param.cls = class;
         param.budget_policy = (want_enforcement) ?
                         PRECISE_ENFORCEMENT : NO_ENFORCEMENT;
-        if (migrate) {
-                if (reservation >= 0)
-                        param.cpu = reservation;
-                else
-                        param.cpu = domain_to_first_cpu(cluster);
-        }
+
+        if (reservation >= 0)
+                param.cpu = reservation;
+        else if (migrate)
+                param.cpu = domain_to_first_cpu(cluster);
+
         ret = set_rt_task_param(gettid(), &param);
         if (ret < 0)
                 bail_out("could not setup rt task params");
diff --git a/src/litmus.c b/src/litmus.c
index 4d9e986..8ddc3d9 100644
--- a/src/litmus.c
+++ b/src/litmus.c
@@ -89,6 +89,7 @@ void init_rt_task_param(struct rt_task* tp)
          *  - fixed priority = LITMUS_LOWEST_PRIORITY
          *  - release policy = TASK_SPORADIC
          *  - cpu assignment = 0
+         *  - pinned cpu = NO_CPU(-1)
          *
          * User must still set the following fields to non-zero values:
          *  - tp->exec_cost
@@ -105,6 +106,7 @@ void init_rt_task_param(struct rt_task* tp)
         tp->priority = LITMUS_LOWEST_PRIORITY;
         tp->budget_policy = NO_ENFORCEMENT;
         tp->release_policy = TASK_SPORADIC;
+        tp->pinned_cpu = -1;
 }
 
 task_class_t str2class(const char* str)